ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-15 15:12:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded vectorizer

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-12-22 00:46:29 +01:00
parent d441efff01
commit 3901f0b83c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lollms/server/endpoints/lollms_generator.py
View File

@ -258,8 +258,8 @@ async def lollms_embed(request: LollmsEmbed):
vdb = VectorDatabase("", vectorizer, None if elf_server.config.rag_vectorizer=="semantic" else elf_server.model if elf_server.model else TikTokenTokenizer(), n_neighbors=elf_server.config.rag_n_chunks)
vector = vdb.vectorizer.vectorize(request.text)
return {"vector":vector}
vector = vdb.vectorizer.vectorize([request.text])
return {"vector":vector[0].tolist()}
class LollmsGenerateRequest(BaseModel):
model_config = ConfigDict(protected_namespaces=())