root/ailine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加 tool_call_history, stop, tools_used 等新状态字段

Browse Source 
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
root
2026-05-08 01:29:55 +08:00
parent 46cd7abcc6
commit 527d7a0b1d
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/app/main_graph/state.py
View File

@@ -9,7 +9,7 @@
- 统计：llm_calls, last_token_usage, last_elapsed_time - 统计：llm_calls, last_token_usage, last_elapsed_time
""" """
from typing import Annotated, Sequence, Optional, Dict, Any from typing import Annotated, Sequence, Optional, Dict, Any, List
from dataclasses import dataclass, field from dataclasses import dataclass, field
from langgraph.graph import add_messages from langgraph.graph import add_messages
from langchain_core.messages import BaseMessage from langchain_core.messages import BaseMessage
@@ -35,3 +35,14 @@ class AgentState:
llm_calls: int = 0 llm_calls: int = 0
last_token_usage: Dict[str, Any] = field(default_factory=dict) last_token_usage: Dict[str, Any] = field(default_factory=dict)
last_elapsed_time: float = 0.0 last_elapsed_time: float = 0.0
# ========== 新增字段: 工具调用历史 ==========
tool_call_history: List[dict] = field(default_factory=list)
tool_result_history: List[str] = field(default_factory=list)
# ========== 新增字段: 停止控制 ==========
stop: bool = False
stop_reason: str = ""
# ========== 新增字段: 本轮使用的工具 ==========
tools_used: List[str] = field(default_factory=list)