Преглед на файлове

修改agent执行时候,工具调用参数输出

xueyiming преди 4 месеца
родител
ревизия
9ee2574dc3
променени са 1 файла, в които са добавени 3 реда и са изтрити 3 реда
  1. 3 3
      pqai_agent/agents/simple_chat_agent.py

+ 3 - 3
pqai_agent/agents/simple_chat_agent.py

@@ -59,10 +59,10 @@ class SimpleOpenAICompatibleChatAgent:
                     function_name = tool_call.function.name
                     arguments = json.loads(tool_call.function.arguments)
                     call_list.append({
-                        "name": function_name,
-                        "parameters": arguments
+                        "toolName": function_name,
+                        "toolParameters": arguments
                     })
-                tool_message = json.dumps(call_list, ensure_ascii=False, indent=2)
+                tool_message = json.dumps(call_list, ensure_ascii=False)
                 agent_task_detail = AgentTaskDetail()
                 agent_task_detail.executor_type = 'llm'
                 agent_task_detail.executor_name = self.model