ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-22 14:02:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added ollama vectorization

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-12-09 23:48:37 +01:00
parent b515a38645
commit 1e10efabc5
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lollms/databases/discussions_database.py
View File

@ -783,9 +783,9 @@ class Discussion:
elif self.lollms.config.rag_vectorizer=="openai": elif self.lollms.config.rag_vectorizer=="openai":
from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer
vectorizer = OpenAIVectorizer(self.lollms.config.rag_vectorizer_model, self.lollms.config.rag_vectorizer_openai_key) vectorizer = OpenAIVectorizer(self.lollms.config.rag_vectorizer_model, self.lollms.config.rag_vectorizer_openai_key)
elif self.config.rag_vectorizer == "ollama": elif self.lollms.config.rag_vectorizer == "ollama":
from lollmsvectordb.lollms_vectorizers.ollama_vectorizer import OllamaVectorizer from lollmsvectordb.lollms_vectorizers.ollama_vectorizer import OllamaVectorizer
v = OllamaVectorizer(self.lollms.config.rag_vectorizer_model, self.lollms.config.rag_service_url) vectorizer = OllamaVectorizer(self.lollms.config.rag_vectorizer_model, self.lollms.config.rag_service_url)
self.vectorizer = VectorDatabase( self.vectorizer = VectorDatabase(
self.discussion_rag_folder/"db.sqli", self.discussion_rag_folder/"db.sqli",
@ -957,6 +957,10 @@ class Discussion:
elif self.lollms.config.rag_vectorizer == "openai": elif self.lollms.config.rag_vectorizer == "openai":
from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer
v = OpenAIVectorizer(self.lollms.config.rag_vectorizer_openai_key) v = OpenAIVectorizer(self.lollms.config.rag_vectorizer_openai_key)
elif self.lollms.config.rag_vectorizer == "ollama":
from lollmsvectordb.lollms_vectorizers.ollama_vectorizer import OllamaVectorizer
v = OllamaVectorizer(self.lollms.config.rag_vectorizer_model, self.lollms.config.rag_service_url)
self.vectorizer = VectorDatabase( self.vectorizer = VectorDatabase(
self.discussion_rag_folder/"db.sqli", self.discussion_rag_folder/"db.sqli",
v, v,