support agent user header (#2077)

2025-04-02 16:29:38 -04:00
parent 3d7788791c
commit a86864d072
3 changed files with 11 additions and 1 deletions
--- a/skyvern/agent/client.py
+++ b/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,