Update rag_pipeline.py

rag_pipeline.py

@@ -493,7 +493,7 @@ class LLMJudge:
             temperature: float = 0.1) -> str:
         try:
             r = self.client.chat_completion(
-                messages=[{"role": "user", "content": prompt}],
+                messages=[ {"role": "system", "content": "/no_think"},{"role": "user", "content": prompt}],
                 max_tokens=max_tokens,
                 temperature=temperature,
             )
@@ -551,7 +551,7 @@ class LLMJudge:
             "صرف ایک لفظ میں جواب دیں: \"ہاں\" یا \"نہیں\"\n\n"
             f"متن:\n{context[:1800]}\n\nدعویٰ: {claim}\n\nجواب:"
         )
-        raw = self._judge_call(prompt, max_tokens=100).lower().strip()  # ← 20 → 100
+        raw = self._judge_call(prompt, max_tokens=500, temperature=0.4).lower().strip()  # ← 20 → 100
         dbg("verify_claim raw response:", repr(raw))
 
         if not raw:
@@ -643,15 +643,14 @@ class LLMJudge:
 
     def generate_questions_from_answer(self, answer: str) -> List[str]:
         prompt = (
-            "نیچے دیے گئے جواب سے 3 سوالات بنائیں۔\n"
-            "صرف JSON array میں جواب دیں، کوئی اور متن نہ لکھیں۔\n"
-            "مثال: [\"سوال 1\", \"سوال 2\", \"سوال 3\"]\n\n"
-            "جواب:\n" + answer + "\n\nJSON:"
+            "صرف JSON array میں تین سوالات لکھیں۔ کوئی وضاحت نہ کریں۔\n"
+            "فارمیٹ بالکل ایسا ہو: [\"سوال\", \"سوال\", \"سوال\"]\n\n"
+            "جواب: " + answer + "\n\n"
+            "JSON array:"
         )
-        raw = self._judge_call(prompt, max_tokens=300, temperature=0.4)
-        dbg("generate_questions raw response:", repr(raw))  # ← ADD THIS
-
-        # Try JSON parse first
+        raw = self._judge_call(prompt, max_tokens=500, temperature=0.4)
+        dbg("generate_questions raw response:", repr(raw))
+    
         try:
             match = re.search(r'\[.*?\]', raw, re.DOTALL)
             if match:
@@ -659,15 +658,19 @@ class LLMJudge:
                 return [q.strip() for q in questions if isinstance(q, str) and len(q.strip()) > 5][:3]
         except (json.JSONDecodeError, ValueError):
             pass
-
-        # Fallback: line-by-line parsing
-        questions = []
+    
+        # Handle truncated JSON
+        questions = re.findall(r'"([^"]{6,})"', raw)
+        if questions:
+            return questions[:3]
+    
+        # Last resort line-by-line
+        result = []
         for line in raw.split("\n"):
-            clean = re.sub(r"^[\d\.\)\-\s]+", "", line.strip()).strip()
-            clean = clean.strip('"').strip("'").strip()
+            clean = re.sub(r'^[\d\.\)\-\[\]"\']+\s*', "", line.strip()).strip().strip('"')
             if clean and len(clean) > 5:
-                questions.append(clean)
-        return questions[:3]
+                result.append(clean)
+        return result[:3]