fix bug

app.py

@@ -39,7 +39,13 @@ def today_date():
 
 
 def clean_json_string(json_str):
-    return re.sub(r'[ ,}\s]+$', '', json_str) + '}'
+    try: 
+        data = json.loads(json_str)
+        if type(data) == list:
+            return json.dumps(data[0])
+        return json_str
+    except:
+        return re.sub(r'[ ,}\s]+$', '', json_str) + '}'
 
 def completion(history, model, system_prompt: str, tools=None):
     messages = [{"role": "system", "content": system_prompt.format(date=today_date())}]
@@ -107,14 +113,15 @@ def llm_in_loop(history, system_prompt, recursive):
             history[-1].content += chunk.choices[0].delta.content
             yield history[recursive:]
     
-    arguments = clean_json_string(arguments) if arguments else "{}"
-    print(name, arguments)
-    arguments = json.loads(arguments)
-    print(name, arguments)
-    print("====================")
-    if appended:
-        recursive -= 1
+    
+    
     if name:
+        print("------------------------")
+        print(name, arguments)
+        arguments = clean_json_string(arguments) if arguments else "{}"
+        print(name, arguments)
+        print("====================")
+        arguments = json.loads(arguments)
         result = f"💥 Error using tool {name}, tool doesn't exist" if name not in tools else str(tools[name].invoke(input=arguments))
         result = json.dumps({name: result}, ensure_ascii=False)
         # msg = ChatMessage(
@@ -123,7 +130,13 @@ def llm_in_loop(history, system_prompt, recursive):
         #             metadata= {"title": f"🛠️ Using tool '{name}', arguments: {json.dumps(json_arguments, ensure_ascii=False)}"},
         #             options=[{"label":"tool_calls", "value": json.dumps([{"id": "call_FthC9qRpsL5kBpwwyw6c7j4k","function": {"arguments": arguments,"name": name},"type": "function"}])}]
         #         )
-        history.append(ChatMessage(role="assistant", content=result, metadata={"title": json.dumps([{"id": "call_id", "function": {"arguments": json.dumps(arguments, ensure_ascii=False), "name": name}, "type": "function"}], ensure_ascii=False)}))
+        msg = ChatMessage(role="assistant", content=result, metadata={"title": json.dumps([{"id": "call_id", "function": {"arguments": json.dumps(arguments, ensure_ascii=False), "name": name}, "type": "function"}], ensure_ascii=False)})
+        if appended:
+            print("Text with function", history[-1].content)
+            msg.content = history[-1].content + "\n" + msg.content
+            history[-1] = msg
+        else:
+            history.append(msg)
         yield history[recursive:]
         yield from llm_in_loop(history, system_prompt, recursive - 1)