Просмотр исходного кода

Refactor rag/main.py

Renamed function get_filtered_results
Que Nguyen 11 месяцев назад
Родитель
Сommit
3cc0e3ecb6
1 измененных файлов с 3 добавлено и 3 удалено
  1. 3 3
      backend/apps/rag/search/main.py

+ 3 - 3
backend/apps/rag/search/main.py

@@ -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