16 lines
471 B
Python
16 lines
471 B
Python
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]
|