Pārlūkot izejas kodu

fix: extract tool calls correctly while arguments is empty (#6503)

tags/0.6.15
sino pirms 1 gada
vecāks
revīzija
dfb6f4fec6
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 10 papildinājumiem un 2 dzēšanām
  1. 10
    2
      api/core/agent/fc_agent_runner.py

+ 10
- 2
api/core/agent/fc_agent_runner.py Parādīt failu

@@ -342,10 +342,14 @@ class FunctionCallAgentRunner(BaseAgentRunner):
"""
tool_calls = []
for prompt_message in llm_result_chunk.delta.message.tool_calls:
args = {}
if prompt_message.function.arguments != '':
args = json.loads(prompt_message.function.arguments)

tool_calls.append((
prompt_message.id,
prompt_message.function.name,
json.loads(prompt_message.function.arguments),
args,
))

return tool_calls
@@ -359,10 +363,14 @@ class FunctionCallAgentRunner(BaseAgentRunner):
"""
tool_calls = []
for prompt_message in llm_result.message.tool_calls:
args = {}
if prompt_message.function.arguments != '':
args = json.loads(prompt_message.function.arguments)

tool_calls.append((
prompt_message.id,
prompt_message.function.name,
json.loads(prompt_message.function.arguments),
args,
))

return tool_calls

Notiek ielāde…
Atcelt
Saglabāt