ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-19 04:37:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update discussions_database.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-12-17 13:31:44 +01:00 committed by GitHub
parent 1244a8677e
commit 7d368776ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/databases/discussions_database.py
View File

@ -776,7 +776,7 @@ class Discussion:
if len(self.text_files)>0: if len(self.text_files)>0:
if self.lollms.config.rag_vectorizer=="semantic": if self.lollms.config.rag_vectorizer=="semantic":
from lollmsvectordb.lollms_vectorizers.semantic_vectorizer import SemanticVectorizer from lollmsvectordb.lollms_vectorizers.semantic_vectorizer import SemanticVectorizer
vectorizer = SemanticVectorizer(self.lollms.config.rag_vectorizer_model, self.config.rag_vectorizer_execute_remote_code) vectorizer = SemanticVectorizer(self.lollms.config.rag_vectorizer_model, self.lollms.config.rag_vectorizer_execute_remote_code)
elif self.lollms.config.rag_vectorizer=="tfidf": elif self.lollms.config.rag_vectorizer=="tfidf":
from lollmsvectordb.lollms_vectorizers.tfidf_vectorizer import TFIDFVectorizer from lollmsvectordb.lollms_vectorizers.tfidf_vectorizer import TFIDFVectorizer
vectorizer = TFIDFVectorizer() vectorizer = TFIDFVectorizer()