Explorar o código

refac: Implicit asyncio create_task

tth37 hai 5 meses
pai
achega
cf2d7de873
Modificáronse 1 ficheiros con 7 adicións e 11 borrados
  1. 7 11
      backend/open_webui/utils/middleware.py

+ 7 - 11
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: