Update rag_pipeline.py

rag_pipeline.py

@@ -35,9 +35,10 @@ class RAGPipeline:
         prompt = f"لخص النص التالي:\n{text}"
         try:
             inputs = self.tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
-            summary_ids = self.model.generate(inputs["input_ids"], max_length=256)
+            summary_ids = self.model.generate(inputs["input_ids"], max_length=128)
             return self.tokenizer.decode(summary_ids[0], skip_special_tokens=True).strip()
-        except:
+        except Exception as e:
+            print(f"[RAG][ERROR] أثناء التلخيص: {e}")
             return ""
 
     def generate_answer_from_passages(self, question, passages_text):
@@ -48,7 +49,8 @@ class RAGPipeline:
             inputs = self.tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
             output_ids = self.model.generate(inputs["input_ids"], max_length=200)
             answer = self.tokenizer.decode(output_ids[0], skip_special_tokens=True).strip()
-        except:
+        except Exception as e:
+            print(f"[RAG][ERROR] أثناء توليد الإجابة: {e}")
             answer = ""
 
         return answer, summary