Update app.py

app.py

@@ -502,7 +502,7 @@ async def generate_function_call(prompt):
         try:
             # Attempt to parse the trimmed string as JSON
             parsed_json = json.loads(json_string)
-            return parsed_json.get("output",None)
+            return parsed_json
         except json.JSONDecodeError:
             return {"error": "Invalid JSON format"}
     else: