support extra_headers in SkyvernAgent (#2080)

This commit is contained in:
Shuchang Zheng
2025-04-02 17:03:39 -04:00
committed by GitHub
parent 0d6f8f4f4f
commit 4f401e81b6
2 changed files with 11 additions and 2 deletions

View File

@@ -30,7 +30,9 @@ class SkyvernAgent:
cdp_url: str | None = None,
browser_path: str | None = None,
browser_type: str | None = None,
extra_headers: dict[str, str] | None = None,
) -> None:
self.extra_headers = extra_headers
self.client: SkyvernClient | None = None
if base_url is None and api_key is None:
# TODO: run at the root wherever the code is initiated
@@ -69,7 +71,7 @@ class SkyvernAgent:
self.client = SkyvernClient(
base_url=base_url,
api_key=api_key,
extra_headers={"X-User-Agent": "skyvern-mcp"},
extra_headers=self.extra_headers,
)
else:
raise ValueError("base_url and api_key must be both provided")