Upload generator.py

src/rag/generator.py

@@ -1,29 +1,29 @@
 """
-Generator module for Norwegian RAG chatbot.
-Generates responses using retrieved context and LLM.
+Generator module for Norwegian RAG chatbot with GPT-4o integration.
+Generates responses using retrieved context and GPT-4o.
 """
 
 from typing import List, Dict, Any, Optional
 
-from ..api.huggingface_api import HuggingFaceAPI, create_rag_prompt
+from ..api.openai_api import OpenAIAPI, create_rag_prompt
 
 class Generator:
     """
-    Generates responses using retrieved context and LLM.
-    Uses Hugging Face Inference API for text generation.
+    Generates responses using retrieved context and GPT-4o.
+    Uses OpenAI API for text generation.
     """
     
     def __init__(
         self,
-        api_client: Optional[HuggingFaceAPI] = None,
+        api_client: Optional[OpenAIAPI] = None,
     ):
         """
         Initialize the generator.
         
         Args:
-            api_client: HuggingFaceAPI client for text generation
+            api_client: OpenAIAPI client for text generation
         """
-        self.api_client = api_client or HuggingFaceAPI()
+        self.api_client = api_client or OpenAIAPI()
     
     def generate(
         self,