Update app.py

app.py

@@ -6,6 +6,8 @@ from openai import OpenAI
 from tools import tools, oitools
 from dotenv import load_dotenv
 import os
+import re
+
 load_dotenv(".env")
 HF_TOKEN = os.environ["HF_TOKEN"]
 BASE_URL = os.environ["BASE_URL"]
@@ -21,6 +23,10 @@ client = OpenAI(
     )
 
 
+def clean_json_string(json_str):
+    # Strip spaces and '}' from the end, then add back a single '}'
+    return re.sub(r'[ }\s]+$', '', json_str) + '}'
+
 def complation(history, model, system_prompt, tools=None):
     messages = [{"role": "system", "content": system_prompt}]
     for msg in history:
@@ -108,8 +114,9 @@ def respond(
     
     if not arguments:
         arguments = "{}"
-    elif arguments.strip() == "} }" or arguments.strip() == "}}":
-        arguments = arguments.strip()[:-1]
+    else:
+        arguments = clean_json_string(arguments)
+    
     if name:
         result = f"💥 Error using tool {name}, tools doesn't exists"
         print("arguments:", arguments)