From 1e10efabc50587c4769e365ea91416e4082ba017 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 9 Dec 2024 23:48:37 +0100 Subject: [PATCH] added ollama vectorization --- lollms/databases/discussions_database.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/lollms/databases/discussions_database.py b/lollms/databases/discussions_database.py index 094f960..0931964 100644 --- a/lollms/databases/discussions_database.py +++ b/lollms/databases/discussions_database.py @@ -783,9 +783,9 @@ class Discussion: elif self.lollms.config.rag_vectorizer=="openai": from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer 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 - 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.discussion_rag_folder/"db.sqli", @@ -957,6 +957,10 @@ class Discussion: elif self.lollms.config.rag_vectorizer == "openai": from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer 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.discussion_rag_folder/"db.sqli", v,