From 12b33c8cf78d63b52742c2598d96c11ad64da54e Mon Sep 17 00:00:00 2001 From: fiatrete Date: Fri, 16 Jun 2023 15:58:57 +0800 Subject: [PATCH] do not fix json if the arguments is a valid json (event if it's an empty object) --- agent_jarvis/jarvis/ai_agent/gpt_agent.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/agent_jarvis/jarvis/ai_agent/gpt_agent.py b/agent_jarvis/jarvis/ai_agent/gpt_agent.py index c6b344b..443fb8e 100644 --- a/agent_jarvis/jarvis/ai_agent/gpt_agent.py +++ b/agent_jarvis/jarvis/ai_agent/gpt_agent.py @@ -55,7 +55,11 @@ class GptAgent(BaseAgent): "speak": assistant_reply, } elif reply_type == "function_call": - arguments = await fix_json_using_multiple_techniques(assistant_reply["arguments"]) + arguments_string = assistant_reply["arguments"] + try: + arguments = json.loads(arguments_string) + except: + arguments = await fix_json_using_multiple_techniques() return { "function": assistant_reply["name"], "arguments": arguments