diff --git a/cognee/api/v1/add/add.py b/cognee/api/v1/add/add.py index c8720f33f..93ddc0b49 100644 --- a/cognee/api/v1/add/add.py +++ b/cognee/api/v1/add/add.py @@ -150,7 +150,7 @@ async def add( await cognee.add("https://example.com",extraction_rules=extraction_rules) # Add a single url and tavily extract ingestion method - Make sure to TAVILY_API_KEY = YOUR_TAVILY_API_KEY as a environment variable + Make sure to set TAVILY_API_KEY = YOUR_TAVILY_API_KEY as a environment variable await cognee.add("https://example.com") # Add multiple urls diff --git a/cognee/tasks/web_scraper/utils.py b/cognee/tasks/web_scraper/utils.py index 12d2338ef..f22a18aa5 100644 --- a/cognee/tasks/web_scraper/utils.py +++ b/cognee/tasks/web_scraper/utils.py @@ -36,7 +36,7 @@ async def fetch_page_content( extraction_rules are needed. Returns: - Dict[str, Union[str, Dict[str, str]]]: A dictionary mapping each URL to its + Dict[str, str]: A dictionary mapping each URL to its extracted content (as a string for BeautifulSoup or a dict for Tavily). Raises: