From 12ec2502085c378d0bfe1e294d2d16bd30c0c342 Mon Sep 17 00:00:00 2001 From: Desiree Sng Date: Tue, 18 Mar 2025 14:31:05 -0700 Subject: [PATCH] Set origin request in headers --- agentstack/_tools/agentql/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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(