diff --git a/agentstack/_tools/agentql/__init__.py b/agentstack/_tools/agentql/__init__.py index 3b46d02e..77369b65 100644 --- a/agentstack/_tools/agentql/__init__.py +++ b/agentstack/_tools/agentql/__init__.py @@ -9,12 +9,17 @@ API_KEY = os.getenv("AGENTQL_API_KEY") -def extract_data(url: str, query: Optional[str], prompt: Optional[str]) -> dict: +def extract_data( + url: str, + query: Optional[str], + prompt: Optional[str], + is_stealth_mode_enabled: bool = False, +) -> dict: """ url: url of website to scrape query: described below prompt: Natural language description of the data you want to scrape - + is_stealth_mode_enabled: Enable stealth mode for web scraping (default: False) AgentQL query to scrape the url. @@ -45,7 +50,14 @@ def extract_data(url: str, query: Optional[str], prompt: Optional[str]) -> dict: } ``` """ - payload = {"url": url, "query": query, "prompt": prompt} + payload = { + "url": url, + "query": query, + "prompt": prompt, + "metadata": { + "experimental_stealth_mode_enabled": is_stealth_mode_enabled, + }, + } headers = {"X-API-Key": f"{API_KEY}", "Content-Type": "application/json"}