This commit is contained in:
Saifeddine ALOUI 2024-12-09 00:29:32 +01:00
parent ae9f595bd0
commit 5c1ecc21db

View File

@ -77,11 +77,11 @@ def get_user_vectorizer(user_key: str):
v = OpenAIVectorizer(lollmsElfServer.config.rag_vectorizer_openai_key) v = OpenAIVectorizer(lollmsElfServer.config.rag_vectorizer_openai_key)
return VectorDatabase( return VectorDatabase(
str(user_folder / f"rag_db_{small_key}.sqlite"), "",
v, TikTokenTokenizer(), v, TikTokenTokenizer(),
lollmsElfServer.model,
chunk_size=lollmsElfServer.config.rag_chunk_size, chunk_size=lollmsElfServer.config.rag_chunk_size,
overlap=lollmsElfServer.config.rag_overlap overlap=lollmsElfServer.config.rag_overlap,
model=lollmsElfServer.model,
) )
async def validate_key(key: str): async def validate_key(key: str):