From cf2d7de8737825d19f6ef8d4ed521b3506ec425f Mon Sep 17 00:00:00 2001 From: tth37 Date: Tue, 29 Apr 2025 00:20:55 +0800 Subject: [PATCH] refac: Implicit asyncio create_task --- backend/open_webui/utils/middleware.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 4fc4df10d..b0e89bbf0 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -366,21 +366,17 @@ async def chat_web_search_handler( } ) - web_search_tasks = [ - asyncio.create_task( + gathered_results = await asyncio.gather( + *( process_web_search( request, - SearchForm( - **{ - "query": searchQuery, - } - ), + SearchForm(**{"query": searchQuery}), user=user, ) - ) - for searchQuery in queries - ] - gathered_results = await asyncio.gather(*web_search_tasks, return_exceptions=True) + for searchQuery in queries + ), + return_exceptions=True, + ) for searchQuery, results in zip(queries, gathered_results): try: