AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 10:35:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix tool_inputs parse error in message that in CoT(ReAct) agent mode (#2949)

Browse Source
This commit is contained in:
listeng 2024-03-26 11:05:10 +08:00 committed by GitHub
parent 8e05261588
commit d70bd4aaa4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/features/assistant_base_runner.py
View File

@ -566,7 +566,11 @@ class BaseAssistantApplicationRunner(AppRunner):
tools = tools.split(';') tools = tools.split(';')
tool_calls: list[AssistantPromptMessage.ToolCall] = [] tool_calls: list[AssistantPromptMessage.ToolCall] = []
tool_call_response: list[ToolPromptMessage] = [] tool_call_response: list[ToolPromptMessage] = []
tool_inputs = json.loads(agent_thought.tool_input) try:
tool_inputs = json.loads(agent_thought.tool_input)
except Exception as e:
logging.warning("tool execution error: {}, tool_input: {}.".format(str(e), agent_thought.tool_input))
tool_inputs = { agent_thought.tool: agent_thought.tool_input }
for tool in tools: for tool in tools:
# generate a uuid for tool call # generate a uuid for tool call
tool_call_id = str(uuid.uuid4()) tool_call_id = str(uuid.uuid4())
@ -599,4 +603,4 @@ class BaseAssistantApplicationRunner(AppRunner):
db.session.close() db.session.close()
return result return result

4
api/core/features/assistant_cot_runner.py
View File

@ -182,7 +182,7 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
delta=LLMResultChunkDelta( delta=LLMResultChunkDelta(
index=0, index=0,
message=AssistantPromptMessage( message=AssistantPromptMessage(
content=json.dumps(chunk) content=json.dumps(chunk, ensure_ascii=False) # if ensure_ascii=True, the text in webui maybe garbled text
), ),
usage=None usage=None
) )
@ -667,4 +667,4 @@ class AssistantCotApplicationRunner(BaseAssistantApplicationRunner):
try: try:
return json.dumps(tools, ensure_ascii=False) return json.dumps(tools, ensure_ascii=False)
except json.JSONDecodeError: except json.JSONDecodeError:
return json.dumps(tools) return json.dumps(tools)