Update skills_database.py

This commit is contained in:
Saifeddine ALOUI 2024-12-16 10:23:46 +01:00 committed by GitHub
parent f27b2ea24f
commit 3a277a282f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -23,7 +23,7 @@ class SkillsLibrary:
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