OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

perf: Multi-thread web searching

Browse Source
This commit is contained in:
tth37 2025-04-21 22:18:05 +08:00
parent 48637893b1
commit 4b451b984b
1 changed files with 22 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
backend/open_webui/utils/middleware.py
View File

@ -355,21 +355,20 @@ async def chat_web_search_handler(
all_results = [] all_results = []
for searchQuery in queries: await event_emitter(
await event_emitter( {
{ "type": "status",
"type": "status", "data": {
"data": { "action": "web_search",
"action": "web_search", "description": "Searching the web",
"description": 'Searching "{{searchQuery}}"', "done": False,
"query": searchQuery, },
"done": False, }
}, )
}
)
try: web_search_tasks = [
results = await process_web_search( asyncio.create_task(
process_web_search(
request, request,
SearchForm( SearchForm(
**{ **{
@ -378,6 +377,15 @@ async def chat_web_search_handler(
), ),
user=user, user=user,
) )
)
for searchQuery in queries
]
gathered_results = await asyncio.gather(*web_search_tasks, return_exceptions=True)
for searchQuery, results in zip(queries, gathered_results):
try:
if isinstance(results, Exception):
raise Exception(f"Error searching {searchQuery}: {str(results)}")
if results: if results:
all_results.append(results) all_results.append(results)