From 43a0ab68b734806dbac634aba97f830c58cfec03 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 2 Sep 2024 09:24:05 +0200 Subject: [PATCH] Update personality.py --- lollms/personality.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lollms/personality.py b/lollms/personality.py index 3ef7882..84dd2a1 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -899,13 +899,13 @@ class AIPersonality: self.database_path = self.data_path / "db.sqlite" from lollmsvectordb.lollms_tokenizers.tiktoken_tokenizer import TikTokenTokenizer - if vectorizer == "semantic": + if self.config.rag_vectorizer == "semantic": from lollmsvectordb.lollms_vectorizers.semantic_vectorizer import SemanticVectorizer v = SemanticVectorizer() - elif vectorizer == "tfidf": + elif self.config.rag_vectorizer == "tfidf": from lollmsvectordb.lollms_vectorizers.tfidf_vectorizer import TFIDFVectorizer v = TFIDFVectorizer() - elif vectorizer == "openai": + elif self.config.rag_vectorizer == "openai": from lollmsvectordb.lollms_vectorizers.openai_vectorizer import OpenAIVectorizer v = OpenAIVectorizer()