From 43efff0fe62a0ac8e41342a9158dcc387a1ec6d5 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Tue, 22 Apr 2025 23:22:50 +0900 Subject: [PATCH] refac --- backend/open_webui/routers/retrieval.py | 27 ++++++++++++++----------- 1 file changed, 15 insertions(+), 12 deletions(-) diff --git a/backend/open_webui/routers/retrieval.py b/backend/open_webui/routers/retrieval.py index 0343d9300..bcb8aa413 100644 --- a/backend/open_webui/routers/retrieval.py +++ b/backend/open_webui/routers/retrieval.py @@ -1557,19 +1557,22 @@ async def process_web_search( collection_names = [] for doc_idx, doc in enumerate(docs): if doc and doc.page_content: - collection_name = f"web-search-{calculate_sha256_string(form_data.query + '-' + urls[doc_idx])}"[ - :63 - ] + try: + collection_name = f"web-search-{calculate_sha256_string(form_data.query + '-' + urls[doc_idx])}"[ + :63 + ] - collection_names.append(collection_name) - await run_in_threadpool( - save_docs_to_vector_db, - request, - [doc], - collection_name, - overwrite=True, - user=user, - ) + collection_names.append(collection_name) + await run_in_threadpool( + save_docs_to_vector_db, + request, + [doc], + collection_name, + overwrite=True, + user=user, + ) + except Exception as e: + log.debug(f"error saving doc {doc_idx}: {e}") return { "status": True,