final changes

src/main.py

@@ -294,7 +294,7 @@ class RAGSystem:
     def __init__(self):
         self.session_manager = SessionManager()
         self.crawler = WebCrawler()
-        self.llm = OllamaLLM(base_url="http://localhost:11434", model="mistral")
+        self.llm = Ollama(base_url="http://localhost:11434", model="mistral")
 
     def crawl_and_index(self, session_id: str, start_url: str) -> Dict[str, Any]:
         """Crawl website and index content in Qdrant"""
@@ -358,8 +358,9 @@ class RAGSystem:
 
             # Select appropriate LLM
             if model == "mistral" and ollama_url:
-                self.llm = OllamaLLM(base_url=ollama_url, model="mistral")
+                self.llm = Ollama(base_url=ollama_url, model="mistral")
             elif model == "gemini" and gemini_api_key:
+                from src.llm import GeminiProvider
                 self.llm = GeminiProvider()
 
             # Process the query