Update calendar_rag.py

calendar_rag.py

@@ -517,28 +517,18 @@ class PipelineConfig:
         )
 
 def create_default_config(api_key: str) -> PipelineConfig:
-    """Create default pipeline configuration"""
-    model_config = ModelConfig(openai_api_key=api_key)
-    
-    # Create default configuration objects
-    retriever_config = RetrieverConfig()
-    cache_config = CacheConfig()
-    processing_config = ProcessingConfig()
-    monitoring_config = MonitoringConfig()
-    localization_config = LocalizationConfig()
-    
-    # Return complete pipeline configuration
+    """Create a default configuration with the given API key"""
+    model_config = ModelConfig(
+        openai_api_key=api_key,
+        embedder_model="sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
+    )
     return PipelineConfig(
         model=model_config,
-        retriever=retriever_config,
-        cache=cache_config,
-        processing=processing_config,
-        monitoring=monitoring_config,
-        localization=localization_config,
-        rate_limit_enabled=True,
-        requests_per_minute=60,
-        debug_mode=False,
-        development_mode=False
+        retriever=RetrieverConfig(),
+        cache=CacheConfig(),
+        processing=ProcessingConfig(),
+        monitoring=MonitoringConfig(),
+        localization=LocalizationConfig()
     )
 
 class CalendarDataProcessor:
@@ -603,9 +593,9 @@ class EnhancedDocumentStore:
     
     def __init__(self, config: PipelineConfig):
         self.store = InMemoryDocumentStore()
-        # Updated embedder initialization without model_name_or_path parameter
-        self.embedder = SentenceTransformersDocumentEmbedder()
-        
+        self.embedder = SentenceTransformersDocumentEmbedder(
+            model=config.model.embedder_model
+        )
         self.cache_manager = CacheManager(
             cache_dir=config.cache.cache_dir,
             ttl=config.cache.embeddings_cache_ttl
@@ -929,10 +919,9 @@ class ResponseGenerator:
             return "ขออภัย ไม่สามารถประมวลผลคำตอบได้ในขณะนี้"
 
 class AcademicCalendarRAG:
-    """Main RAG pipeline implementation"""
+    """Main RAG pipeline for academic calendar queries"""
     
     def __init__(self, config: PipelineConfig):
-        """Initialize pipeline with configurations"""
         self.config = config
         self.document_store = EnhancedDocumentStore(config)
         self.query_processor = AdvancedQueryProcessor(config)
@@ -978,7 +967,7 @@ class AcademicCalendarRAG:
                 "documents": [],
                 "query_info": {}
             }
-
+    
 # def main():
 #     """Main function for processing real calendar queries"""
 #     try:
@@ -1003,7 +992,7 @@ class AcademicCalendarRAG:
 #         pipeline.load_data(calendar_data)
         
 #         # Real queries to process
-#         queries = input in web
+#         queries = ["นิสิตที่เข้าศึกษาในภาคเรียนที่ 1 ปีการศึกษา 2567 สามารถถอนรายวิชาได้หรือไม่? เพราะเหตุใด?"]
         
 #         print("Processing calendar queries...")
 #         print("=" * 80)