Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: bocha filter list

Browse Source 
Co-Authored-By: zengxy <11961641+zengxy@users.noreply.github.com>
This commit is contained in:
Timothy Jaeryang Baek
2025-12-20 17:24:24 +04:00
co-authored by zengxy
parent 823b9a6dd9
commit b5bd8704fe
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/retrieval/web/bocha.py
+5 -5
View File
@@ -9,13 +9,13 @@ log = logging.getLogger(__name__)
def _parse_response(response): def _parse_response(response):
result = {} results = []
if "data" in response: if "data" in response:
data = response["data"] data = response["data"]
if "webPages" in data: if "webPages" in data:
webPages = data["webPages"] webPages = data["webPages"]
if "value" in webPages: if "value" in webPages:
result["webpage"] = [ results = [
{ {
"id": item.get("id", ""), "id": item.get("id", ""),
"name": item.get("name", ""), "name": item.get("name", ""),
@@ -29,7 +29,7 @@ def _parse_response(response):
} }
for item in webPages["value"] for item in webPages["value"]
] ]
return result return results
def search_bocha( def search_bocha(
@@ -51,7 +51,7 @@ def search_bocha(
response = requests.post(url, headers=headers, data=payload, timeout=5) response = requests.post(url, headers=headers, data=payload, timeout=5)
response.raise_for_status() response.raise_for_status()
results = _parse_response(response.json()) results = _parse_response(response.json())
print(results)
if filter_list: if filter_list:
results = get_filtered_results(results, filter_list) results = get_filtered_results(results, filter_list)
@@ -59,5 +59,5 @@ def search_bocha(
SearchResult( SearchResult(
link=result["url"], title=result.get("name"), snippet=result.get("summary") link=result["url"], title=result.get("name"), snippet=result.get("summary")
) )
for result in results.get("webpage", [])[:count] for result in results[:count]
] ]