Moving code out of playwright branch

backend/open_webui/retrieval/web/main.py

@@ -1,5 +1,3 @@
-import validators
-
 from typing import Optional
 from urllib.parse import urlparse
 
@@ -12,8 +10,6 @@ def get_filtered_results(results, filter_list):
     filtered_results = []
     for result in results:
         url = result.get("url") or result.get("link", "")
-        if not validators.url(url):
-            continue
         domain = urlparse(url).netloc
         if any(domain.endswith(filtered_domain) for filtered_domain in filter_list):
             filtered_results.append(result)

backend/open_webui/retrieval/web/utils.py

@@ -48,16 +48,6 @@ def validate_url(url: Union[str, Sequence[str]]):
     else:
         return False
 
-def safe_validate_urls(url: Sequence[str]) -> Sequence[str]:
-    valid_urls = []
-    for u in url:
-        try:
-            if validate_url(u):
-                valid_urls.append(u)
-        except ValueError:
-            continue
-    return valid_urls
-
 def resolve_hostname(hostname):
     # Get address information
     addr_info = socket.getaddrinfo(hostname, None)
@@ -253,11 +243,12 @@ def get_web_loader(
     verify_ssl: bool = True,
     requests_per_second: int = 2,
 ):
-    # Check if the URLs are valid
-    safe_urls = safe_validate_urls([urls] if isinstance(urls, str) else urls)
+    # Check if the URL is valid
+    if not validate_url(urls):
+        raise ValueError(ERROR_MESSAGES.INVALID_URL)
 
     web_loader_args = {
-        "urls": safe_urls,
+        "urls": urls,
         "verify_ssl": verify_ssl,
         "requests_per_second": requests_per_second,
         "continue_on_failure": True
@@ -270,6 +261,6 @@ def get_web_loader(
     WebLoaderClass = RAG_WEB_LOADERS[RAG_WEB_LOADER.value]
     web_loader = WebLoaderClass(**web_loader_args)
 
-    log.debug("Using RAG_WEB_LOADER %s for %s URLs", web_loader.__class__.__name__, len(safe_urls))
+    log.debug("Using RAG_WEB_LOADER %s for %s URLs", web_loader.__class__.__name__, len(urls))
 
     return web_loader