feat(graph): Extend AgentState with checklist, current_step, and vfs

This commit is contained in:
Yunxiao Xu
2026-02-23 04:34:28 -08:00
parent 45fe122580
commit 8957e93f3d
2 changed files with 25 additions and 0 deletions

View File

@@ -34,3 +34,13 @@ class AgentState(AS):
# Number of execution attempts
iterations: int
# --- DeepAgents Extensions ---
# High-level plan/checklist: List of Task objects (dicts)
checklist: List[Dict[str, Any]]
# Current active step in the checklist
current_step: int
# Virtual File System (VFS): Map of filenames to content/metadata
vfs: Dict[str, Any]

View File

@@ -0,0 +1,15 @@
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]