ソースを参照

删除执行过程中的api_data的生成函数,保留最后chat的api_data的生成函数.

wangxq 1 ヶ月 前
コミット
698c852d73
1 ファイル変更5 行追加22 行削除
  1. 5 22
      test/custom_react_agent/agent.py

+ 5 - 22
test/custom_react_agent/agent.py

@@ -624,18 +624,7 @@ class CustomReactAgent:
         # 注释掉前缀添加,直接使用原始内容
         # last_message.content = f"[Formatted Output]\n{last_message.content}"
         
-        # 生成API格式的数据
-        api_data = await self._async_generate_api_data(state)
-
-        # 打印api_data
-        print("-"*20+"api_data_start"+"-"*20)
-        print(api_data)
-        print("-"*20+"api_data_end"+"-"*20)
-
-        return {
-            "messages": [last_message],
-            "api_data": api_data  # 新增:API格式数据
-        }
+        return {"messages": [last_message]}
 
     async def _async_generate_api_data(self, state: AgentState) -> Dict[str, Any]:
         """异步生成API格式的数据结构"""
@@ -874,16 +863,10 @@ class CustomReactAgent:
                 result["sql_data"] = sql_data
                 logger.info("   📊 已包含SQL原始数据")
             
-            # 🔧 修复:检查 api_data 是否在 final_state 中
-            if "api_data" in final_state:
-                result["api_data"] = final_state["api_data"]
-                logger.info("   🔌 已包含API格式数据")
-            else:
-                # 🔧 备用方案:如果 final_state 中没有 api_data,手动生成
-                logger.warning("   ⚠️ final_state 中未找到 api_data,手动生成...")
-                api_data = await self._async_generate_api_data(final_state)
-                result["api_data"] = api_data
-                logger.info("   🔌 已手动生成API格式数据")
+            # 生成API格式数据
+            api_data = await self._async_generate_api_data(final_state)
+            result["api_data"] = api_data
+            logger.info("   🔌 已生成API格式数据")
             
             return result