fix custom code

Author: pirate

src/stagehand/types/session_execute_params.py

@@ -56,11 +56,8 @@ class AgentConfig(TypedDict, total=False):
     mode: Literal["dom", "hybrid", "cua"]
     """Tool mode for the agent (dom, hybrid, cua). If set, overrides cua."""
 
-    model: ModelConfigParam
-    """
-    Model name string with provider prefix (e.g., 'openai/gpt-5-nano',
-    'anthropic/claude-4.5-opus')
-    """
+    model: AgentConfigModel
+    """Model configuration object or model name string (e.g., 'openai/gpt-5-nano')"""
 
     provider: Literal["openai", "anthropic", "google", "microsoft", "bedrock"]
     """AI provider for the agent (legacy, use model: openai/gpt-5-nano instead)"""

src/stagehand/types/session_extract_params.py

@@ -3,12 +3,18 @@
 from __future__ import annotations
 
 from typing import Dict, Union, Optional
-from typing_extensions import Literal, Required, Annotated, TypedDict
+from typing_extensions import Literal, Required, Annotated, TypeAlias, TypedDict
 
 from .._utils import PropertyInfo
 from .model_config_param import ModelConfigParam
 
-__all__ = ["SessionExtractParamsBase", "Options", "SessionExtractParamsNonStreaming", "SessionExtractParamsStreaming"]
+__all__ = [
+    "SessionExtractParamsBase",
+    "Options",
+    "OptionsModel",
+    "SessionExtractParamsNonStreaming",
+    "SessionExtractParamsStreaming",
+]
 
 
 class SessionExtractParamsBase(TypedDict, total=False):
@@ -27,12 +33,12 @@ class SessionExtractParamsBase(TypedDict, total=False):
     """Whether to stream the response via SSE"""
 
 
+OptionsModel: TypeAlias = Union[ModelConfigParam, str]
+
+
 class Options(TypedDict, total=False):
-    model: ModelConfigParam
-    """
-    Model name string with provider prefix (e.g., 'openai/gpt-5-nano',
-    'anthropic/claude-4.5-opus')
-    """
+    model: OptionsModel
+    """Model configuration object or model name string (e.g., 'openai/gpt-5-nano')"""
 
     selector: str
     """CSS selector to scope extraction to a specific element"""

src/stagehand/types/session_observe_params.py

@@ -3,14 +3,15 @@
 from __future__ import annotations
 
 from typing import Dict, Union, Optional
-from typing_extensions import Literal, Required, Annotated, TypedDict
+from typing_extensions import Literal, Required, Annotated, TypeAlias, TypedDict
 
 from .._utils import PropertyInfo
 from .model_config_param import ModelConfigParam
 
 __all__ = [
     "SessionObserveParamsBase",
     "Options",
+    "OptionsModel",
     "OptionsVariables",
     "OptionsVariablesUnionMember3",
     "SessionObserveParamsNonStreaming",
@@ -30,6 +31,10 @@ class SessionObserveParamsBase(TypedDict, total=False):
     x_stream_response: Annotated[Literal["true", "false"], PropertyInfo(alias="x-stream-response")]
     """Whether to stream the response via SSE"""
 
+
+OptionsModel: TypeAlias = Union[ModelConfigParam, str]
+
+
 class OptionsVariablesUnionMember3(TypedDict, total=False):
     value: Required[Union[str, float, bool]]
 
@@ -40,11 +45,8 @@ class OptionsVariablesUnionMember3(TypedDict, total=False):
 
 
 class Options(TypedDict, total=False):
-    model: ModelConfigParam
-    """
-    Model name string with provider prefix (e.g., 'openai/gpt-5-nano',
-    'anthropic/claude-4.5-opus')
-    """
+    model: OptionsModel
+    """Model configuration object or model name string (e.g., 'openai/gpt-5-nano')"""
 
     selector: str
     """CSS selector to scope observation to a specific element"""