Merge pull request #345 from desi003/add-stealth-agentql

Add stealth mode in AgentQL tool

Author: tcdent

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"}