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}") log.debug(f"text_content: {text_content}")
result = await run_in_threadpool( 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: if result:
return { return {
@@ -1721,7 +1721,12 @@ async def process_web(
if not request.app.state.config.BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL: if not request.app.state.config.BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL:
await run_in_threadpool( 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: else:
collection_name = None collection_name = None
@@ -2464,7 +2469,12 @@ async def process_files_batch(
if all_docs: if all_docs:
try: try:
await run_in_threadpool( 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 # Update all files with collection name