prompt revision 3

components/generators/daily_feed.py

@@ -2,7 +2,7 @@ import os
 import sys
 import json
 import requests
-from requests.exceptions import RequestException
+from typing import Optional
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
@@ -51,38 +51,50 @@ HEADERS = {
     "Content-Type": "application/json"
 }
 
-def call_mistral(prompt: str, max_new_tokens: int = 128, temperature: float = 0.7) -> str:
-    """
-    Call Hugging Face Inference Endpoint hosting Mistral-7B.
-    Returns the generated summary, or empty string on failure.
-    """
+def call_mistral(prompt: str) -> Optional[str]:
+    if not prompt or len(prompt.strip()) < 10:
+        print(f"⚠️ Skipping empty or invalid prompt:\n{prompt}\n")
+        return None
+
+    # Wrap the prompt properly
+    formatted_prompt = f"[INST] {prompt.strip()} [/INST]"
+
     payload = {
-        "inputs": prompt,
+        "inputs": formatted_prompt,
         "parameters": {
-            "max_new_tokens": max_new_tokens,
-            "temperature": temperature
+            "max_new_tokens": 120,
+            "temperature": 0.7,
         }
     }
 
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Content-Type": "application/json"
+    }
+
     try:
-        response = requests.post(MISTRAL_URL, headers=HEADERS, json=payload, timeout=60)
+        response = requests.post(MISTRAL_URL, headers=headers, json=payload, timeout=30)
         response.raise_for_status()
         data = response.json()
 
-        # Handle both list and dict output formats
-        if isinstance(data, list) and data:
-            return data[0].get("generated_text", "").strip()
-        if isinstance(data, dict) and "generated_text" in data:
+        # Handle both dict and list responses
+        if isinstance(data, list) and "generated_text" in data[0]:
+            return data[0]["generated_text"].strip()
+        elif isinstance(data, dict) and "generated_text" in data:
             return data["generated_text"].strip()
+        else:
+            print("⚠️ Unexpected response format:", data)
+            return None
 
-    except RequestException as e:
-        print("❌ Mistral HF request failed:", str(e))
+    except requests.RequestException as e:
+        print("❌ HF Endpoint error:", str(e))
         if e.response is not None:
-            print("↪️ Response:", e.response.text[:300])
+            print("❌ Response body:", e.response.text[:300])
     except Exception as e:
-        print("❌ Unexpected error:", str(e))
+        print("❌ Unknown error:", str(e))
+
+    return None
 
-    return ""
 
 
 # ✂️ Generate summaries per topic