Fix K2 timeout detection and JSON parsing errors

app/services/k2_think_engine.py

@@ -139,7 +139,12 @@ Start directly with <think> if needed, then output the JSON inside [RESULT] tags
                 response = await chat.ainvoke([HumanMessage(content=instruction_prompt)])
                 raw_content = response.content
             except Exception as e:
-                if "524" in str(e) or "timeout" in str(e).lower():
+                is_timeout = (
+                    "524" in str(e) 
+                    or "timeout" in str(e).lower() 
+                    or type(e).__name__ in ["APITimeoutError", "Timeout", "ReadTimeout", "TimeoutError"]
+                )
+                if is_timeout:
                     logger.warning("K2 API Timeout detected. Retrying with reduced token budget...")
                     # Réduction drastique pour passer le timeout
                     chat_config["max_tokens"] = 4000
@@ -196,9 +201,17 @@ Start directly with <think> if needed, then output the JSON inside [RESULT] tags
                 try:
                     # Tentative 1: Standard
                     k2_analysis = json.loads(clean_json)
-                except json.JSONDecodeError:
-                    # Tentative 2: Réparation (Fermeture des balises)
-                    logger.warning("JSON parsing failed, attempting auto-repair...")
+                except json.JSONDecodeError as e:
+                    if "Extra data" in str(e) and hasattr(e, "pos"):
+                        try:
+                            logger.warning(f"Extra data found at pos {e.pos}, attempting to truncate...")
+                            k2_analysis = json.loads(clean_json[:e.pos].strip())
+                        except Exception:
+                            pass
+                            
+                    if not k2_analysis:
+                        # Tentative 2: Réparation (Fermeture des balises)
+                        logger.warning("JSON parsing failed, attempting auto-repair...")
                     repaired = clean_json.strip()
                     open_braces = repaired.count('{')
                     close_braces = repaired.count('}')

scratch/check_db.py

@@ -0,0 +1,43 @@
+
+import os
+import sys
+from uuid import UUID
+
+# Add the app directory to sys.path
+backend_path = 'c:/Users/user/Desktop/ai-scientific-coinvestigator-backend'
+if backend_path not in sys.path:
+    sys.path.append(backend_path)
+
+from app.db.session import SessionLocal
+from app.models.all_models import AnalysisRun
+
+def check_analysis(analysis_id):
+    db = SessionLocal()
+    try:
+        analysis = db.query(AnalysisRun).filter(AnalysisRun.id == UUID(analysis_id)).first()
+        if not analysis:
+            print(f"Analysis {analysis_id} not found in DB")
+            return
+        
+        print(f"ID: {analysis.id}")
+        print(f"Project ID: {analysis.project_id}")
+        print(f"Status: {analysis.status}")
+        print(f"Model: {analysis.model_used}")
+        print(f"Result Data Type: {type(analysis.result_data)}")
+        if analysis.result_data:
+             import json
+             print("Result Data JSON:")
+             print(json.dumps(analysis.result_data, indent=2))
+        else:
+             print("Result Data: None")
+             
+    except Exception as e:
+        print(f"Error: {e}")
+    finally:
+        db.close()
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        check_analysis(sys.argv[1])
+    else:
+        print("Usage: python check_db.py <analysis_id>")