from typing import get_type_hints, List, Dict, Any
from ea_chatbot.graph.state import AgentState

def test_agent_state_extensions():
    """Verify that AgentState has the new DeepAgents fields."""
    hints = get_type_hints(AgentState)
    
    assert "checklist" in hints
    # checklist: List[Dict[str, Any]] (or similar for tasks)
    
    assert "current_step" in hints
    assert hints["current_step"] == int
    
    assert "vfs" in hints
    # vfs: Dict[str, Any]