File size: 876 Bytes
da9a42e 7603dff da9a42e 3e43a4b da9a42e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from langgraph.graph import MessagesState
from typing import List, Annotated, Optional, Dict, Any
from src.schemas import PlannerPlan, ComplexityLevel, CritiqueFeedback, ExecutionReport
from typing import Sequence
from langchain_core.messages import BaseMessage
from langgraph.graph.message import add_messages
class AgentState(MessagesState):
query: str
final_answer: str
plan: Optional[PlannerPlan]
complexity_assessment: ComplexityLevel
current_step: int
reasoning_done: bool
messages : Annotated[Sequence[BaseMessage], add_messages]
files: List[str]
file_contents: Dict[str, Any]
critique_feedback: Optional[CritiqueFeedback]
iteration_count :int
max_iterations: int
execution_report : ExecutionReport
previous_tool_results: Dict[str, str] # НОВОЕ ПОЛЕ для сохранения результатов
|