ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-20 21:23:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added rag_dbs

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-18 01:36:43 +02:00
parent 8ba91e75c6
commit 3acd936fe8
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
lollms/app.py
View File

@ -89,7 +89,11 @@ class LollmsApplication(LoLLMsCom):
self.ttm = None
self.ttv = None
self.active_rag_dbs = []
try:
self.load_rag_dbs()
except Exception as ex:
trace_exception(ex)
for entry in self.config.rag_databases:
if "mounted" in entry:
parts = entry.split("::")
@ -265,6 +269,23 @@ class LollmsApplication(LoLLMsCom):
def get_uploads_path(self, client_id):
return self.lollms_paths.personal_uploads_path
def load_rag_dbs(self):
self.active_rag_dbs = []
for rag_db in self.config.rag_databases:
parts = rag_db.split("::")
if parts[-1]=="mounted":
if not PackageManager.check_package_installed("lollmsvectordb"):
PackageManager.install_package("lollmsvectordb")
from lollmsvectordb.vectorizers.bert_vectorizer import BERTVectorizer
from lollmsvectordb import VectorDatabase
from lollmsvectordb.text_document_loader import TextDocumentsLoader
v = BERTVectorizer()
vdb = VectorDatabase(Path(parts[1])/"db_name.sqlite", v)
vdb.build_index()
self.active_rag_dbs.append({"name":parts[0],"path":parts[1],"vectorizer":vdb})
def start_servers(self):
ASCIIColors.yellow("* - * - * - Starting services - * - * - *")