Tachi67 commited on
Commit
b4e4404
·
1 Parent(s): 54f092c

Update ControllerFlow_ExtLib.py

Browse files
Files changed (1) hide show
  1. ControllerFlow_ExtLib.py +4 -2
ControllerFlow_ExtLib.py CHANGED
@@ -113,8 +113,10 @@ class ControllerFlow_ExtLib(ChatAtomicFlow):
113
  while True:
114
  api_output = super().run(input_data)["api_output"].strip()
115
  try:
116
- response = json.loads(api_output)
117
- return response
 
 
118
  except (json.decoder.JSONDecodeError, json.JSONDecodeError):
119
  updated_system_message_content = self._get_message(self.system_message_prompt_template, input_data)
120
  self._state_update_add_chat_message(content=updated_system_message_content,
 
113
  while True:
114
  api_output = super().run(input_data)["api_output"].strip()
115
  try:
116
+ start = api_output.index("{")
117
+ end = api_output.rindex("}") + 1
118
+ json_str = api_output[start:end]
119
+ return json.loads(json_str)
120
  except (json.decoder.JSONDecodeError, json.JSONDecodeError):
121
  updated_system_message_content = self._get_message(self.system_message_prompt_template, input_data)
122
  self._state_update_add_chat_message(content=updated_system_message_content,