outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: more robust toolcall argument load

Browse Source
This commit is contained in:
zhangmingyu1 2024-12-10 09:52:39 +08:00
parent fc1415d705
commit 6415b176ed
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
api/core/agent/fc_agent_runner.py
View File

@ -340,7 +340,27 @@ class FunctionCallAgentRunner(BaseAgentRunner):
for prompt_message in llm_result_chunk.delta.message.tool_calls:
args = {}
if prompt_message.function.arguments != "":
args = json.loads(prompt_message.function.arguments)
try_load_str = prompt_message.function.arguments
while True:
if len(try_load_str) == 0:
break
try:
args = json.loads(try_load_str)
break
except Exception as e:
logging.error("error try_load_str: " + try_load_str)
first_left_brace_index = try_load_str.find("{")
if len(try_load_str) >= first_left_brace_index + 1:
try_load_str = try_load_str[try_load_str.find("{") + 1 :]
# find next left brace
next_left_brace_index = try_load_str.find("{")
if next_left_brace_index == -1:
logging.error("error try_load_str not found next left brace: " + try_load_str)
break
try_load_str = try_load_str[next_left_brace_index:]
else:
logging.error("error try_load_str not found next left brace: " + try_load_str)
break
tool_calls.append(
(