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

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

@ -355,21 +355,20 @@ async def chat_web_search_handler(
all_results = []
for searchQuery in queries:
await event_emitter(
{
"type": "status",
"data": {
"action": "web_search",
"description": 'Searching "{{searchQuery}}"',
"query": searchQuery,
"description": "Searching the web",
"done": False,
},
}
)
try:
results = await process_web_search(
web_search_tasks = [
asyncio.create_task(
process_web_search(
request,
SearchForm(
**{
@ -378,6 +377,15 @@ async def chat_web_search_handler(
),
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:
all_results.append(results)