瀏覽代碼

修改State的字段,删除了四个不使用的字段.

wangxq 4 周之前
父節點
當前提交
5b9e10fad4
共有 2 個文件被更改,包括 0 次插入12 次删除
  1. 0 6
      agent/citu_agent.py
  2. 0 6
      agent/state.py

+ 0 - 6
agent/citu_agent.py

@@ -776,7 +776,6 @@ class CituLangGraphAgent:
             
             # 数据库查询流程状态
             sql=None,
-            sql_generation_attempts=0,
             query_result=None,
             summary=None,
             
@@ -801,11 +800,6 @@ class CituLangGraphAgent:
             # 流程控制
             current_step="initialized",
             execution_path=["start"],
-            retry_count=0,
-            max_retries=3,
-            
-            # 调试信息
-            debug_info={},
             
             # 路由模式
             routing_mode=effective_routing_mode

+ 0 - 6
agent/state.py

@@ -20,7 +20,6 @@ class AgentState(TypedDict):
     
     # 数据库查询流程状态
     sql: Optional[str]
-    sql_generation_attempts: int
     query_result: Optional[Dict[str, Any]]
     summary: Optional[str]
     
@@ -45,11 +44,6 @@ class AgentState(TypedDict):
     # 流程控制
     current_step: str
     execution_path: List[str]
-    retry_count: int
-    max_retries: int
-    
-    # 调试信息
-    debug_info: Dict[str, Any]
     
     # 路由模式相关
     routing_mode: Optional[str]  # 记录使用的路由模式