Trying to handle the inference generatoin remaining incomplete

cloudzy/agents/image_analyzer.py

@@ -76,12 +76,20 @@ result: {
 
         # Try to extract JSON-like dict from model output
         try:
-            if "{" not in text_content or "}" not in text_content:
-                raise ValueError("Response does not contain valid JSON structure (missing braces)")
+            if "{" not in text_content:
+                raise ValueError("Response does not contain valid JSON structure (missing opening brace)")
             
             start = text_content.index("{")
-            end = text_content.rindex("}") + 1
-            json_str = text_content[start:end]
+            
+            # Try to find closing brace
+            if "}" not in text_content[start:]:
+                # No closing brace found, try adding one
+                print(f"[Warning] No closing brace found in response, attempting to add closing brace...")
+                json_str = text_content[start:] + "}"
+            else:
+                end = text_content.rindex("}") + 1
+                json_str = text_content[start:end]
+            
             result = json.loads(json_str)
         except ValueError as ve:
             raise ValueError(f"Failed to parse model output: {text_content}\nError: {ve}")

cloudzy/agents/image_analyzer_2.py

@@ -101,14 +101,19 @@ result: {
         response_text = str(response) if response is not None else ""
         
         # Extract JSON part from response
-        # Look for the pattern: result: { ... }
-        match = re.search(r'result:\s*(\{[\s\S]*\})', response_text)
+        # Look for the pattern: result: { ... } (or { ... if closing brace is missing)
+        match = re.search(r'result:\s*(\{[\s\S]*)', response_text)
         
         if not match:
             raise ValueError(f"Could not find JSON in response: {response_text}")
         
         json_str = match.group(1)
         
+        # If the extracted JSON doesn't end with }, try adding it
+        if not json_str.rstrip().endswith("}"):
+            print(f"[Warning] No closing brace found in JSON, attempting to add closing brace...")
+            json_str = json_str + "}"
+        
         try:
             # Parse the JSON string into a dictionary
             result_dict = json.loads(json_str)