support agent user header (#2077)

skyvern/agent/agent.py

@@ -66,7 +66,11 @@ class SkyvernAgent:
 
             settings.BROWSER_TYPE = browser_type
         elif base_url and api_key:
-            self.client = SkyvernClient(base_url=base_url, api_key=api_key)
+            self.client = SkyvernClient(
+                base_url=base_url,
+                api_key=api_key,
+                extra_headers={"X-User-Agent": "skyvern-mcp"},
+            )
         else:
             raise ValueError("base_url and api_key must be both provided")
 

skyvern/agent/client.py

@@ -8,10 +8,12 @@ class SkyvernClient:
         self,
         base_url: str = settings.SKYVERN_BASE_URL,
         api_key: str = settings.SKYVERN_API_KEY,
+        extra_headers: dict[str, str] | None = None,
     ) -> None:
         self.base_url = base_url
         self.api_key = api_key
         self.client = AsyncSkyvern(base_url=base_url, api_key=api_key)
+        self.extra_headers = extra_headers
 
     async def run_task(
         self,