Update calendar_rag.py

calendar_rag.py

@@ -914,7 +914,7 @@ class HybridDocumentStore:
     
     def _format_selection_steps(self, steps: List[SelectionStep]) -> str:
         """Format selection process steps"""
-        return "\n".join("{0}. {1}".format(step.step_number, step.description) for step in steps)
+        return "\n".join(f"{step.step_number}. {step.description}" for step in steps)
 
     def add_events(self, 
                     events: List[CalendarEvent], 
@@ -1014,7 +1014,7 @@ class HybridDocumentStore:
                 for c in course.structure['หมวดวิชาปรับพื้นฐาน'].courses:
                     foundation_courses.append(f"- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต")
                 text += "\n".join(foundation_courses)
-        
+
                 text += f"""
                 
                 หมวดวิชาบังคับ/วิชาหลัก:
@@ -1026,7 +1026,7 @@ class HybridDocumentStore:
                 for c in course.structure['หมวดวิชาบังคับ'].courses:
                     core_courses.append(f"- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต")
                 text += "\n".join(core_courses)
-        
+
                 text += f"""
                 
                 หมวดวิชาเลือก:
@@ -1038,7 +1038,7 @@ class HybridDocumentStore:
                 for c in course.structure['หมวดวิชาเลือก'].courses:
                     elective_courses.append(f"- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต")
                 text += "\n".join(elective_courses)
-        
+
                 text += f"""
                 
                 หมวดวิชาการค้นคว้าอิสระ:
@@ -1610,4 +1610,39 @@ class AcademicCalendarRAG:
             "query": query,
             "answer": "ขออภัย ไม่สามารถประมวลผลคำตอบได้ในขณะนี้",
             "error": "Maximum retry attempts reached"
-        }
+        }
+            
+# def main():
+#     """Main function demonstrating hybrid retrieval"""
+#     try:
+#         # Load API key
+#         with open("key.txt", "r") as f:
+#             openai_api_key = f.read().strip()
+        
+#         # Create config with hybrid retrieval settings
+#         config = create_default_config(openai_api_key)
+#         pipeline = AcademicCalendarRAG(config)
+        
+#         # Load and process data
+#         with open("raw-data.json", "r", encoding="utf-8") as f:
+#             raw_data = json.load(f)
+        
+#         pipeline.load_data(raw_data)
+        
+#         # Test queries with different semantic weights
+#         queries = ["ค่าเทอมเท่าไหร่","เปิดเรียนวันไหน","ขั้นตอนการสมัครที่สาขานี้มีอะไรบ้าง","ต้องใช้ระดับภาษาอังกฤษเท่าไหร่ในการสมัครเรียนที่นี้","ถ้าจะไปติดต่อมาหลายต้องลง mrt อะไร","มีวิชาหลักเเละวิชาเลือกออะไรบ้าง", "ปีที่ 1 เทอม 1 ต้องเรียนอะไรบ้าง", "ปีที่ 2 เทอม 1 ต้องเรียนอะไรบ้าง"]
+#         # queries = ["สอบปลายภาควันไหน"]
+#         print("=" * 80)
+        
+#         for query in queries:
+#             print(f"\nQuery: {query}")
+#             result = pipeline.process_query(query)
+#             print(f"Answer: {result['answer']}")
+#             print("-" * 40)
+            
+#     except Exception as e:
+#         logger.error(f"Pipeline execution failed: {str(e)}")
+#         raise
+ 
+# if __name__ == "__main__":
+#     main()