15 lines
465 B
Python
15 lines
465 B
Python
|
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 {}
|