init

agent/state.py

@@ -0,0 +1,14 @@
+from typing import Any, Dict, List
+from langchain_core.messages import BaseMessage
+
+
+class AgentState:
+    """代理工作流的状态定义"""
+    messages: List[BaseMessage]
+    current_step: str
+    context: Dict[str, Any]
+
+    def __init__(self, messages: List[BaseMessage] = None, current_step: str = "start", context: Dict[str, Any] = None):
+        self.messages = messages or []
+        self.current_step = current_step
+        self.context = context or {}