瀏覽代碼

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

tags/0.6.15
sino 1 年之前
父節點
當前提交
dfb6f4fec6
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 10 行新增2 行删除
  1. 10
    2
      api/core/agent/fc_agent_runner.py

+ 10
- 2
api/core/agent/fc_agent_runner.py 查看文件

""" """
tool_calls = [] tool_calls = []
for prompt_message in llm_result_chunk.delta.message.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(( tool_calls.append((
prompt_message.id, prompt_message.id,
prompt_message.function.name, prompt_message.function.name,
json.loads(prompt_message.function.arguments),
args,
)) ))


return tool_calls return tool_calls
""" """
tool_calls = [] tool_calls = []
for prompt_message in llm_result.message.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(( tool_calls.append((
prompt_message.id, prompt_message.id,
prompt_message.function.name, prompt_message.function.name,
json.loads(prompt_message.function.arguments),
args,
)) ))


return tool_calls return tool_calls

Loading…
取消
儲存