use set for allowed domains

This commit is contained in:
Vadym Barda
2025-03-23 09:57:24 -04:00
parent f7556c9bd6
commit ef4c0c3dd0

View File

@@ -81,7 +81,7 @@ def create_server(
return content
# Parse the domain names in the llms.txt URLs
allowed_domains = [extract_domain(entry["llms_txt"]) for entry in doc_source]
allowed_domains = set(extract_domain(entry["llms_txt"]) for entry in doc_source)
@server.tool()
async def fetch_docs(url: str) -> str: