UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-07 05:27:55 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Refactor rag/main.py

Browse Source 
Renamed function get_filtered_results
This commit is contained in:
Que Nguyen 2024-06-17 14:36:26 +07:00 committed by GitHub
parent d8beed13b4
commit 3cc0e3ecb6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/apps/rag/search/main.py
View File

@ -3,13 +3,13 @@ from urllib.parse import urlparse
from pydantic import BaseModel from pydantic import BaseModel
def filter_by_whitelist(results, whitelist): def get_filtered_results(results, filter_list):
if not whitelist: if not filter_list:
return results return results
filtered_results = [] filtered_results = []
for result in results: for result in results:
domain = urlparse(result["url"]).netloc domain = urlparse(result["url"]).netloc
if any(domain.endswith(whitelisted_domain) for whitelisted_domain in whitelist): if any(domain.endswith(filtered_domain) for filtered_domain in filter_list):
filtered_results.append(result) filtered_results.append(result)
return filtered_results return filtered_results