Update calendar_rag.py

calendar_rag.py

@@ -1279,7 +1279,7 @@ class HybridDocumentStore:
         return sorted_docs[:top_k]
              
 class ResponseGenerator:
-    """Generate responses with conversation context awareness"""
+    """Generate responses with enhanced conversation context awareness"""
     
     def __init__(self, config: PipelineConfig):
         self.generator = OpenAIGenerator(
@@ -1319,9 +1319,11 @@ class ResponseGenerator:
             5. สำหรับคำถามเกี่ยวกับข้อกำหนดภาษาอังกฤษหรือขั้นตอนการสมัคร ให้อธิบายข้อมูลอย่างละเอียด
             6. ใส่ข้อความ "หากมีข้อสงสัยเพิ่มเติม สามารถสอบถามได้" ท้ายคำตอบเสมอ
             7. คำนึงถึงประวัติการสนทนาและให้คำตอบที่ต่อเนื่องกับบทสนทนาก่อนหน้า
-            8. หากคำถามอ้างอิงถึงข้อมูลในบทสนทนาก่อนหน้า (เช่น "แล้วอันนั้นล่ะ", "มีอะไรอีกบ้าง") ให้พิจารณาบริบทและตอบคำถามอย่างตรงประเด็น
+            8. หากคำถามอ้างอิงถึงข้อมูลในบทสนทนาก่อนหน้า (เช่น "แล้วอันนั้นล่ะ", "มีอะไรอีกบ้าง", "คำถามก่อนหน้า") ให้พิจารณาบริบทและตอบคำถามอย่างตรงประเด็น
             9. กรณีคำถามมีความไม่ชัดเจน ใช้ประวัติการสนทนาเพื่อเข้าใจบริบทของคำถาม
 
+            สิ่งสำคัญพิเศษ: หากคำถามอ้างอิงถึงคำถามก่อนหน้า ให้แสดงคำถามก่อนหน้านั้นในคำตอบด้วย เช่น "คำถามก่อนหน้าคือ [คำถามก่อนหน้า] และคำตอบคือ..."
+
             กรุณาตอบเป็นภาษาไทย:
             """
         )
@@ -1333,12 +1335,23 @@ class ResponseGenerator:
                          conversation_history: List[Dict[str, str]] = None) -> str:
         """Generate response using retrieved documents and conversation history"""
         try:
+            # Enhanced handling of reference questions
+            reference_keywords = ["ก่อนหน้านี้", "ก่อนหน้า", "ที่ผ่านมา", "คำถามก่อนหน้า", "คำถามที่แล้ว", 
+                                 "previous", "earlier", "before", "last time", "last question"]
+            
+            is_reference_question = any(keyword in query.lower() for keyword in reference_keywords)
+            
+            # For reference questions, we'll add additional prompting
+            enhanced_context = conversation_history or []
+            
             result = self.prompt_builder.run(
                 query=query,
                 context=documents,
                 format=query_info.get("response_format", "detailed"),
-                conversation_history=conversation_history or []
+                conversation_history=enhanced_context,
+                is_reference_question=is_reference_question
             )
+            
             response = self.generator.run(prompt=result["prompt"])
             return response["replies"][0]