fix: Catch bad LLM function call

microsoft · Feb 22, 2024 · 737566f · 737566f
1 parent b5123da
commit 737566f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/models/message.py b/models/message.py
@@ -89,7 +89,14 @@ async def execute_function(self, plugins: object) -> str:
 
         logger = build_logger(__name__)
         name = self.function_name
-        args = json.loads(self.function_arguments)
+
+        try:
+            args = json.loads(self.function_arguments)
+        except json.JSONDecodeError:
+            logger.warn(
+                f"Error decoding JSON args for function {name}: {self.function_arguments}"
+            )
+            return f"Bad JSON format, impossible to execute function {name}"
 
         try:
             res = await getattr(plugins, name)(**args)