Update skills_database.py

ParisNeo · Dec 16, 2024 · 3a277a2 · 3a277a2
1 parent f27b2ea
commit 3a277a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lollms/databases/skills_database.py b/lollms/databases/skills_database.py
@@ -23,7 +23,7 @@ def __init__(self, db_path, chunk_size:int=512, overlap:int=0, n_neighbors:int=5
                 v = OpenAIVectorizer()
             elif self.config.rag_vectorizer == "ollama":
                 from lollmsvectordb.lollms_vectorizers.ollama_vectorizer import OllamaVectorizer
-                v = OllamaVectorizer()
+                v = OllamaVectorizer(self.config.rag_vectorizer_model, self.config.rag_service_url)
 
         else:
             from lollmsvectordb.lollms_vectorizers.semantic_vectorizer import SemanticVectorizer