Skip to content

Commit f1fb91f

Browse files
committed
fix: bugname with suffixes
1 parent 8c3a6b6 commit f1fb91f

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

agentic_rag/local_rag_agent.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,6 @@ def __init__(self, model_name: str):
5757
Args:
5858
model_name: Name of the Ollama model to use
5959
"""
60-
# Ensure model name has :latest suffix
61-
if not model_name.endswith(":latest"):
62-
model_name = f"{model_name}:latest"
63-
6460
self.model_name = model_name
6561
self._check_ollama_running()
6662

@@ -174,10 +170,6 @@ def __init__(self, vector_store: VectorStore = None, model_name: str = None,
174170
self.is_ollama = not (model_name and "mistral" in model_name.lower())
175171

176172
if self.is_ollama:
177-
# Add :latest suffix if not present
178-
if not model_name.endswith(":latest"):
179-
model_name = f"{model_name}:latest"
180-
181173
# Load Ollama model
182174
print("\nLoading Ollama model...")
183175
print(f"Model: {model_name}")

0 commit comments

Comments
 (0)