"src/lib/vscode:/vscode.git/clone" did not exist on "3488b7f00644c2be7554ddb6bfef749457ff1d48"
Unverified Commit 3cc0e3ec authored by Que Nguyen's avatar Que Nguyen Committed by GitHub
Browse files

Refactor rag/main.py

Renamed function get_filtered_results
parent d8beed13
......@@ -3,13 +3,13 @@ from urllib.parse import urlparse
from pydantic import BaseModel
def filter_by_whitelist(results, whitelist):
if not whitelist:
def get_filtered_results(results, filter_list):
if not filter_list:
return results
filtered_results = []
for result in results:
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)
return filtered_results
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment