Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 11:57:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: async save docs to vector db

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-11-25 17:19:33 -05:00
parent c631659327
commit 4370dee79e
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
backend/open_webui/routers/retrieval.py
View File

@@ -1689,7 +1689,7 @@ async def process_text(
log.debug(f"text_content: {text_content}")
result = await run_in_threadpool(
save_docs_to_vector_db, request, docs, collection_name, user
save_docs_to_vector_db, request, docs, collection_name, user=user
)
if result:
return {
@@ -1721,7 +1721,12 @@ async def process_web(
if not request.app.state.config.BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL:
await run_in_threadpool(
save_docs_to_vector_db, request, docs, collection_name, True, user
save_docs_to_vector_db,
request,
docs,
collection_name,
overwrite=True,
user=user,
)
else:
collection_name = None
@@ -2464,7 +2469,12 @@ async def process_files_batch(
if all_docs:
try:
await run_in_threadpool(
save_docs_to_vector_db, request, all_docs, collection_name, True, user
save_docs_to_vector_db,
request,
all_docs,
collection_name,
add=True,
user=user,
)
# Update all files with collection name