Update Controller_JarvisFlow.py

Controller_JarvisFlow.py

@@ -111,9 +111,11 @@ class Controller_JarvisFlow(ChatAtomicFlow):
         while True:
             api_output = super().run(input_data)["api_output"].strip()
             try:
-                response = json.loads(api_output)
-                return response
-            except (json.decoder.JSONDecodeError, json.JSONDecodeError):
+                start = api_output.index("{")
+                end = api_output.rindex("}") + 1
+                json_str = api_output[start:end]
+                return json.loads(json_str)
+            except (ValueError, json.decoder.JSONDecodeError, json.JSONDecodeError):
                 updated_system_message_content = self._get_message(self.system_message_prompt_template, input_data)
                 self._state_update_add_chat_message(content=updated_system_message_content,
                                                     role=self.flow_config["system_name"])