Update pipeline/meta_review.py

pipeline/meta_review.py

@@ -14,6 +14,13 @@ client = OpenAI(
     api_key=os.getenv("OPENROUTER_API_KEY"),
 )
 
+# Priority list of models (Consistency with other files)
+MODELS = [
+    "deepseek/deepseek-r1",
+    "deepseek/deepseek-r1-distill-llama-70b",
+    "google/gemini-2.0-flash-exp:free"
+]
+
 class MetaReviewResult(BaseModel):
     meta_review: str
 
@@ -25,15 +32,6 @@ def construct_meta_review_prompt(
 ) -> tuple:
     """
     Construct prompt for meta-review generation
-    
-    Args:
-        paper_title: Paper title
-        paper_abstract: Paper abstract
-        resolutions: List of disagreement resolutions
-        search_results: Search and retrieval results
-    
-    Returns:
-        Tuple of (system_prompt, user_prompt)
     """
     # Aggregate all resolutions
     all_accepted = {}
@@ -110,20 +108,10 @@ async def generate_meta_review(
     paper_abstract: str,
     resolutions: List[Dict],
     search_results: Dict,
-    retries: int = 5
+    retries: int = 3 # Reduced retries per model
 ) -> str:
     """
-    Generate a meta-review using DeepSeek-R1
-    
-    Args:
-        paper_title: Paper title
-        paper_abstract: Paper abstract
-        resolutions: List of disagreement resolutions
-        search_results: Search and retrieval results
-        retries: Maximum retry attempts
-    
-    Returns:
-        Generated meta-review text
+    Generate a meta-review using DeepSeek-R1 (with fallback)
     """
     if not resolutions:
         return "Unable to generate meta-review: No disagreement resolutions available."
@@ -140,31 +128,51 @@ async def generate_meta_review(
         {"role": "user", "content": user_prompt},
     ]
     
-    for attempt in range(retries):
-        try:
-            response = await asyncio.to_thread(
-                client.chat.completions.create,
-                model="deepseek/deepseek-r1",
-                messages=messages,
-            )
-            
-            if not response.choices or not response.choices[0].message.content.strip():
-                raise ValueError("Empty response from DeepSeek-R1")
-            
-            meta_review_text = response.choices[0].message.content.strip()
-            
-            # Remove any JSON formatting if present
-            if meta_review_text.startswith("```"):
-                lines = meta_review_text.split("\n")
-                meta_review_text = "\n".join(lines[1:-1])
-            
-            return meta_review_text
-            
-        except Exception as e:
-            wait_time = 2 ** attempt
-            print(f"Meta-review generation attempt {attempt + 1} failed: {e}")
-            
-            if attempt < retries - 1:
-                await asyncio.sleep(wait_time)
-            else:
-                return f"Error generating meta-review: {str(e)}"
+    last_exception = None
+
+    for model in MODELS:
+        print(f"Generating Meta-Review with model: {model}")
+        
+        for attempt in range(retries):
+            try:
+                response = await asyncio.to_thread(
+                    client.chat.completions.create,
+                    model=model,
+                    messages=messages,
+                    # CRITICAL FIX: Limit max_tokens to prevent 402 Error
+                    # 8192 is plenty for a meta-review (~6000 words) but much cheaper than 64k
+                    max_tokens=8192, 
+                )
+                
+                if not response.choices or not response.choices[0].message.content.strip():
+                    raise ValueError("Empty response from AI")
+                
+                meta_review_text = response.choices[0].message.content.strip()
+                
+                # Remove any JSON formatting/thinking tags if present (cleanup)
+                if meta_review_text.startswith("```"):
+                    lines = meta_review_text.split("\n")
+                    # Simple heuristic to strip markdown fences
+                    if lines[0].startswith("```"):
+                        lines = lines[1:]
+                    if lines and lines[-1].startswith("```"):
+                        lines = lines[:-1]
+                    meta_review_text = "\n".join(lines)
+                
+                return meta_review_text
+                
+            except Exception as e:
+                last_exception = e
+                error_msg = str(e)
+                print(f"Model {model} - Attempt {attempt + 1} failed: {error_msg}")
+                
+                # Immediate fallback on payment errors
+                if "402" in error_msg or "insufficient_quota" in error_msg:
+                    print("Insufficient credits detected. Switching to cheaper model...")
+                    break 
+                
+                wait_time = 2 ** attempt
+                if attempt < retries - 1:
+                    await asyncio.sleep(wait_time)
+                    
+    return f"Error generating meta-review after trying all models: {str(last_exception)}"