Fixed retry logic in chat handler

dougollerenshaw · Sep 19, 2024 · b587065 · b587065
1 parent b93a5bd
commit b587065
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/codeaide/logic/chat_handler.py b/codeaide/logic/chat_handler.py
@@ -41,14 +41,18 @@ def process_input(self, user_input):
                 self.cost_tracker.log_request(response)
 
                 try:
+                    parsed_response = parse_response(response)
+                    if parsed_response[0] is None:  # If parsing failed
+                        raise ValueError("Failed to parse JSON response")
+
                     (
                         text,
                         questions,
                         code,
                         code_version,
                         version_description,
                         requirements,
-                    ) = parse_response(response)
+                    ) = parsed_response
 
                     if (
                         code