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(f"{step.step_number}. {step.description}" for step in steps)
+        return "\n".join("{0}. {1}".format(step.step_number, step.description) for step in steps)
 
     def add_events(self, 
                     events: List[CalendarEvent], 
@@ -1008,23 +1008,48 @@ class HybridDocumentStore:
                 คำอธิบาย: {course.structure['หมวดวิชาปรับพื้นฐาน'].description or 'ไม่ระบุ'}
                 หน่วยกิต: {course.structure['หมวดวิชาปรับพื้นฐาน'].credits}
                 รายวิชา:
-                {' '.join([f'- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต\n' for c in course.structure['หมวดวิชาปรับพื้นฐาน'].courses])}
+                """
+                # Add foundation courses
+                foundation_courses = []
+                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"""
                 
                 หมวดวิชาบังคับ/วิชาหลัก:
                 หน่วยกิตขั้นต่ำ: {course.structure['หมวดวิชาบังคับ'].minimum_credits}
                 รายวิชา:
-                {' '.join([f'- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต\n' for c in course.structure['หมวดวิชาบังคับ'].courses])}
+                """
+                # Add core courses
+                core_courses = []
+                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"""
                 
                 หมวดวิชาเลือก:
                 หน่วยกิตขั้นต่ำ: {course.structure['หมวดวิชาเลือก'].minimum_credits}
                 รายวิชา:
-                {' '.join([f'- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต\n' for c in course.structure['หมวดวิชาเลือก'].courses])}
+                """
+                # Add elective courses
+                elective_courses = []
+                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"""
                 
                 หมวดวิชาการค้นคว้าอิสระ:
                 หน่วยกิตขั้นต่ำ: {course.structure['หมวดวิชาการค้นคว้าอิสระ'].minimum_credits}
                 รายวิชา:
-                {' '.join([f'- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต\n' for c in course.structure['หมวดวิชาการค้นคว้าอิสระ'].courses])}
                 """
+                # Add research courses
+                research_courses = []
+                for c in course.structure['หมวดวิชาการค้นคว้าอิสระ'].courses:
+                    research_courses.append(f"- {c.code}: {c.title_th} ({c.title_en}) - {c.credits} หน่วยกิต")
+                text += "\n".join(research_courses)
                 
                 doc = Document(
                     id=self._generate_unique_id(),