mirror of
https://github.com/open-webui/open-webui.git
synced 2026-05-03 02:39:11 -05:00
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
import logging
|
|
from typing import Optional, List
|
|
|
|
import requests
|
|
|
|
from fastapi import Request
|
|
|
|
|
|
from open_webui.retrieval.web.main import SearchResult, get_filtered_results
|
|
from open_webui.utils.headers import include_user_info_headers
|
|
from open_webui.env import FORWARD_SESSION_INFO_HEADER_CHAT_ID
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def search_external(
|
|
request: Request,
|
|
external_url: str,
|
|
external_api_key: str,
|
|
query: str,
|
|
count: int,
|
|
filter_list: Optional[List[str]] = None,
|
|
user=None,
|
|
) -> List[SearchResult]:
|
|
try:
|
|
headers = {
|
|
'User-Agent': 'Open WebUI (https://github.com/open-webui/open-webui) RAG Bot',
|
|
'Authorization': f'Bearer {external_api_key}',
|
|
}
|
|
headers = include_user_info_headers(headers, user)
|
|
|
|
chat_id = getattr(request.state, 'chat_id', None)
|
|
if chat_id:
|
|
headers[FORWARD_SESSION_INFO_HEADER_CHAT_ID] = str(chat_id)
|
|
|
|
response = requests.post(
|
|
external_url,
|
|
headers=headers,
|
|
json={
|
|
'query': query,
|
|
'count': count,
|
|
},
|
|
)
|
|
response.raise_for_status()
|
|
results = response.json()
|
|
if filter_list:
|
|
results = get_filtered_results(results, filter_list)
|
|
results = [
|
|
SearchResult(
|
|
link=result.get('link'),
|
|
title=result.get('title'),
|
|
snippet=result.get('snippet'),
|
|
)
|
|
for result in results[:count]
|
|
]
|
|
log.info(f'External search results: {results}')
|
|
return results
|
|
except Exception as e:
|
|
log.error(f'Error in External search: {e}')
|
|
return []
|