demeter

 1from typing import List, TypedDict, Optional, Dict, Any
 2from langchain_core.messages import BaseMessage
 3
 4class AgentState(TypedDict):
 5    # Inputs
 6    sensors: Dict[str, float]
 7    strategy: str
 8    research_context: str
 9    history: str
10
11    image_b64: Optional[str]
12    
13    # Internal Processing
14    draft_plan: Optional[Dict[str, Any]]
15    simulation_result: Optional[Dict[str, Any]]
16    critique: Optional[str]
17    retry_count: int
18
19    messages: List[BaseMessage]
20    
21    # Final Output
22    final_action: Optional[Dict[str, Any]]