diff --git a/agentstack/_tools/agentql/__init__.py b/agentstack/_tools/agentql/__init__.py index 3b46d02e..14b7009d 100644 --- a/agentstack/_tools/agentql/__init__.py +++ b/agentstack/_tools/agentql/__init__.py @@ -47,7 +47,11 @@ def extract_data(url: str, query: Optional[str], prompt: Optional[str]) -> dict: """ payload = {"url": url, "query": query, "prompt": prompt} - headers = {"X-API-Key": f"{API_KEY}", "Content-Type": "application/json"} + headers = { + "X-API-Key": f"{API_KEY}", + "Content-Type": "application/json", + "X-TF-Request-Origin": "agentstack", + } try: response = httpx.post(