Function Calls: Handle Malformed JSON Errors

routers/tool_calls.py

@@ -1,4 +1,5 @@
 import json
+import logging
 from .tool_gpu_checker import gpu_checker_get_message
 from .tool_bpy_doc import bpy_doc_get_documentation
 from fastapi import APIRouter, Body
@@ -21,15 +22,24 @@ router = APIRouter()
 
 
 def process_tool_call(tool_call: ToolCallInput) -> Dict:
-    function_name = tool_call.function.name
-    function_args = json.loads(tool_call.function.arguments)
     output = {"tool_call_id": tool_call.id, "output": ""}
-    if function_name == "get_bpy_api_info":
-        output["output"] = bpy_doc_get_documentation(
-            function_args.get("api", ""))
-    elif function_name == "check_gpu":
-        output["output"] = gpu_checker_get_message(
-            function_args.get("gpu", ""))
+    function_name = tool_call.function.name
+
+    try:
+        function_args = json.loads(tool_call.function.arguments)
+        if function_name == "get_bpy_api_info":
+            output["output"] = bpy_doc_get_documentation(
+                function_args.get("api", ""))
+        elif function_name == "check_gpu":
+            output["output"] = gpu_checker_get_message(
+                function_args.get("gpu", ""))
+    except json.JSONDecodeError as e:
+        error_message = f"Malformed JSON encountered at position {e.pos}: {e.msg}\n {tool_call.function.arguments}"
+        output["output"] = error_message
+
+        # Logging the error for further investigation
+        logging.error(f"JSONDecodeError in process_tool_call: {error_message}")
+
     return output