Skip to content

Commit 4370dee

Browse files
committed
fix: async save docs to vector db
1 parent c631659 commit 4370dee

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

backend/open_webui/routers/retrieval.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1689,7 +1689,7 @@ async def process_text(
16891689
log.debug(f"text_content: {text_content}")
16901690

16911691
result = await run_in_threadpool(
1692-
save_docs_to_vector_db, request, docs, collection_name, user
1692+
save_docs_to_vector_db, request, docs, collection_name, user=user
16931693
)
16941694
if result:
16951695
return {
@@ -1721,7 +1721,12 @@ async def process_web(
17211721

17221722
if not request.app.state.config.BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL:
17231723
await run_in_threadpool(
1724-
save_docs_to_vector_db, request, docs, collection_name, True, user
1724+
save_docs_to_vector_db,
1725+
request,
1726+
docs,
1727+
collection_name,
1728+
overwrite=True,
1729+
user=user,
17251730
)
17261731
else:
17271732
collection_name = None
@@ -2464,7 +2469,12 @@ async def process_files_batch(
24642469
if all_docs:
24652470
try:
24662471
await run_in_threadpool(
2467-
save_docs_to_vector_db, request, all_docs, collection_name, True, user
2472+
save_docs_to_vector_db,
2473+
request,
2474+
all_docs,
2475+
collection_name,
2476+
add=True,
2477+
user=user,
24682478
)
24692479

24702480
# Update all files with collection name

0 commit comments

Comments
 (0)