fix: bugname with suffixes

Author: jasperan

agentic_rag/local_rag_agent.py

@@ -57,6 +57,10 @@ def __init__(self, model_name: str):
         Args:
             model_name: Name of the Ollama model to use
         """
+        # Remove 'ollama:' prefix if present
+        if model_name and model_name.startswith("ollama:"):
+            model_name = model_name.replace("ollama:", "")
+        
         self.model_name = model_name
         self._check_ollama_running()
 
@@ -170,6 +174,10 @@ def __init__(self, vector_store: VectorStore = None, model_name: str = None,
         self.is_ollama = not (model_name and "mistral" in model_name.lower())
 
         if self.is_ollama:
+            # Remove 'ollama:' prefix if present
+            if model_name and model_name.startswith("ollama:"):
+                model_name = model_name.replace("ollama:", "")
+            
             # Load Ollama model
             print("\nLoading Ollama model...")
             print(f"Model: {model_name}")