OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-04 03:37:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-05 10:05:12 -07:00
parent 1f9b5b6456
commit 61b147441c
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/open_webui/apps/retrieval/main.py
View File

@ -738,12 +738,18 @@ def process_file(
collection_name = f"file-{file.id}" collection_name = f"file-{file.id}"
if form_data.content: if form_data.content:
VECTOR_DB_CLIENT.delete(
collection_name=f"file-{file.id}",
filter={"file_id": file.id},
)
docs = [ docs = [
Document( Document(
page_content=form_data.content, page_content=form_data.content,
metadata={ metadata={
"name": file.meta.get("name", file.filename), "name": file.meta.get("name", file.filename),
"created_by": file.user_id, "created_by": file.user_id,
"file_id": file.id,
**file.meta, **file.meta,
}, },
) )
@ -755,7 +761,7 @@ def process_file(
collection_name=f"file-{file.id}", filter={"file_id": file.id} collection_name=f"file-{file.id}", filter={"file_id": file.id}
) )
if result: if len(result.ids[0]) > 0:
docs = [ docs = [
Document( Document(
page_content=result.documents[0][idx], page_content=result.documents[0][idx],
@ -770,6 +776,7 @@ def process_file(
metadata={ metadata={
"name": file.meta.get("name", file.filename), "name": file.meta.get("name", file.filename),
"created_by": file.user_id, "created_by": file.user_id,
"file_id": file.id,
**file.meta, **file.meta,
}, },
) )
@ -795,6 +802,7 @@ def process_file(
metadata={ metadata={
"name": file.filename, "name": file.filename,
"created_by": file.user_id, "created_by": file.user_id,
"file_id": file.id,
**file.meta, **file.meta,
}, },
) )