remove x-language and x-sdk-version (#272)

* remove x-language and x-sdk-version

* linting

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",

scripts/download-binary.py

@@ -14,15 +14,15 @@
 """
 from __future__ import annotations
 
-import sys
 import os
+import sys
 import json
 import argparse
 import platform
 import urllib.error
 import urllib.request
-from pathlib import Path
 from typing import Any
+from pathlib import Path
 
 
 def get_platform_info() -> tuple[str, str]:

src/stagehand/resources/sessions_helpers.py

@@ -12,6 +12,7 @@
 from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from ..session import Session, AsyncSession
 from .sessions import SessionsResource, AsyncSessionsResource
+from ..types.session_start_response import SessionStartResponse
 
 
 class SessionsResourceWithHelpers(SessionsResource):
@@ -29,8 +30,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 +49,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 +74,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 +93,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

@@ -16,9 +16,9 @@
     session_navigate_params,
 )
 from ._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
-from .types.session_start_response import SessionStartResponse, Data as SessionStartResponseData
 from .types.session_act_response import SessionActResponse
 from .types.session_end_response import SessionEndResponse
+from .types.session_start_response import Data as SessionStartResponseData, SessionStartResponse
 from .types.session_execute_response import SessionExecuteResponse
 from .types.session_extract_response import SessionExtractResponse
 from .types.session_observe_response import SessionObserveResponse
@@ -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,