remove x-language and x-sdk-version

Author: miguelg719

pyproject.toml

@@ -67,6 +67,7 @@ dev = [
     "importlib-metadata>=6.7.0",
     "rich>=13.7.1",
     "pytest-xdist>=3.6.1",
+    "dotenv>=0.9.9",
 ]
 pydantic-v1 = [
     "pydantic>=1.9.0,<2",

src/stagehand/resources/sessions_helpers.py

@@ -29,8 +29,6 @@ def create(
         system_prompt: str | Omit = omit,
         verbose: Literal[0, 1, 2] | Omit = omit,
         wait_for_captcha_solves: bool | Omit = omit,
-        x_language: Literal["typescript", "python", "playground"] | Omit = omit,
-        x_sdk_version: str | Omit = omit,
         x_sent_at: Union[str, datetime] | Omit = omit,
         x_stream_response: Literal["true", "false"] | Omit = omit,
         extra_headers: Headers | None = None,
@@ -50,8 +48,6 @@ def create(
             system_prompt=system_prompt,
             verbose=verbose,
             wait_for_captcha_solves=wait_for_captcha_solves,
-            x_language=x_language,
-            x_sdk_version=x_sdk_version,
             x_sent_at=x_sent_at,
             x_stream_response=x_stream_response,
             extra_headers=extra_headers,
@@ -77,8 +73,6 @@ async def create(
         system_prompt: str | Omit = omit,
         verbose: Literal[0, 1, 2] | Omit = omit,
         wait_for_captcha_solves: bool | Omit = omit,
-        x_language: Literal["typescript", "python", "playground"] | Omit = omit,
-        x_sdk_version: str | Omit = omit,
         x_sent_at: Union[str, datetime] | Omit = omit,
         x_stream_response: Literal["true", "false"] | Omit = omit,
         extra_headers: Headers | None = None,
@@ -98,8 +92,6 @@ async def create(
             system_prompt=system_prompt,
             verbose=verbose,
             wait_for_captcha_solves=wait_for_captcha_solves,
-            x_language=x_language,
-            x_sdk_version=x_sdk_version,
             x_sent_at=x_sent_at,
             x_stream_response=x_stream_response,
             extra_headers=extra_headers,

src/stagehand/session.py

@@ -41,10 +41,10 @@ class Session(SessionStartResponse):
     """A Stagehand session bound to a specific `session_id`."""
 
     def __init__(self, client: Stagehand, id: str, data: SessionStartResponseData, success: bool) -> None:
+        # Must call super().__init__() first to initialize Pydantic's __pydantic_extra__ before setting attributes
+        super().__init__(data=data, success=success)
         self._client = client
         self.id = id
-        # in case user tries to use client.sessions.start(...) return value as a SessionStartResponse dataclass/dict
-        super().__init__(data=data, success=success)
 
 
     def navigate(
@@ -140,8 +140,6 @@ def execute(
     def end(
         self,
         *,
-        x_language: Literal["typescript", "python", "playground"] | Omit = omit,
-        x_sdk_version: str | Omit = omit,
         x_sent_at: Union[str, datetime] | Omit = omit,
         x_stream_response: Literal["true", "false"] | Omit = omit,
         extra_headers: Headers | None = None,
@@ -151,8 +149,6 @@ def end(
     ) -> SessionEndResponse:
         return self._client.sessions.end(
             id=self.id,
-            x_language=x_language,
-            x_sdk_version=x_sdk_version,
             x_sent_at=x_sent_at,
             x_stream_response=x_stream_response,
             extra_headers=extra_headers,
@@ -166,9 +162,10 @@ class AsyncSession(SessionStartResponse):
     """Async variant of `Session`."""
 
     def __init__(self, client: AsyncStagehand, id: str, data: SessionStartResponseData, success: bool) -> None:
+        # Must call super().__init__() first to initialize Pydantic's __pydantic_extra__ before setting attributes
+        super().__init__(data=data, success=success)
         self._client = client
         self.id = id
-        super().__init__(data=data, success=success)
 
     async def navigate(
         self,
@@ -263,8 +260,6 @@ async def execute(
     async def end(
         self,
         *,
-        x_language: Literal["typescript", "python", "playground"] | Omit = omit,
-        x_sdk_version: str | Omit = omit,
         x_sent_at: Union[str, datetime] | Omit = omit,
         x_stream_response: Literal["true", "false"] | Omit = omit,
         extra_headers: Headers | None = None,
@@ -274,8 +269,6 @@ async def end(
     ) -> SessionEndResponse:
         return await self._client.sessions.end(
             id=self.id,
-            x_language=x_language,
-            x_sdk_version=x_sdk_version,
             x_sent_at=x_sent_at,
             x_stream_response=x_stream_response,
             extra_headers=extra_headers,