consolidate schemas into types folder

README.md

@@ -190,7 +190,7 @@ if __name__ == "__main__":
 import os
 from stagehand.sync import Stagehand
 from stagehand import StagehandConfig
-from stagehand.schemas import AgentConfig, AgentExecuteOptions, AgentProvider
+from stagehand.types import AgentConfigAPI as AgentConfig, AgentExecuteOptionsAPI as AgentExecuteOptions, AgentProvider
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -246,7 +246,7 @@ if __name__ == "__main__":
 
   The `ActOptions` model takes an `action` field that tells the AI what to do on the page, plus optional fields such as `useVision` and `variables`:
   ```python
-  from stagehand.schemas import ActOptions
+  from stagehand.types import ActOptions
 
   # Example:
   await page.act(ActOptions(action="click on the 'Quickstart' button"))
@@ -256,7 +256,7 @@ if __name__ == "__main__":
 
   The `ObserveOptions` model lets you find elements on the page using natural language. The `onlyVisible` option helps limit the results:
   ```python
-  from stagehand.schemas import ObserveOptions
+  from stagehand.types import ObserveOptions
 
   # Example:
   await page.observe(ObserveOptions(instruction="find the button labeled 'News'", onlyVisible=True))
@@ -266,7 +266,7 @@ if __name__ == "__main__":
 
   The `ExtractOptions` model extracts structured data from the page. Pass your instructions and a schema defining your expected data format. **Note:** If you are using a Pydantic model for the schema, call its `.model_json_schema()` method to ensure JSON serializability.
   ```python
-  from stagehand.schemas import ExtractOptions
+  from stagehand.types import ExtractOptions
   from pydantic import BaseModel
 
   class DescriptionSchema(BaseModel):

examples/agent_example.py

@@ -8,7 +8,7 @@
 from rich.theme import Theme
 
 from stagehand import Stagehand, StagehandConfig, AgentConfig, configure_logging
-from stagehand.schemas import AgentExecuteOptions, AgentProvider
+from stagehand.types import AgentExecuteOptionsAPI as AgentExecuteOptions, AgentProvider
 
 # Create a custom theme for consistent styling
 custom_theme = Theme(

stagehand/__init__.py

@@ -8,11 +8,11 @@
 from .main import Stagehand
 from .metrics import StagehandFunctionName, StagehandMetrics
 from .page import StagehandPage
-from .schemas import (
+from .types import (
     ActOptions,
     ActResult,
-    AgentConfig,
-    AgentExecuteOptions,
+    AgentConfigAPI as AgentConfig,
+    AgentExecuteOptionsAPI as AgentExecuteOptions,
     AgentExecuteResult,
     AgentProvider,
     ExtractOptions,

stagehand/agent.py

@@ -1,6 +1,6 @@
-from .schemas import (
-    AgentConfig,
-    AgentExecuteOptions,
+from .types import (
+    AgentConfigAPI as AgentConfig,
+    AgentExecuteOptionsAPI as AgentExecuteOptions,
     AgentExecuteResult,
     AgentProvider,
 )

stagehand/config.py

@@ -3,7 +3,7 @@
 from browserbase.types import SessionCreateParams as BrowserbaseSessionCreateParams
 from pydantic import BaseModel, ConfigDict, Field
 
-from stagehand.schemas import AvailableModel
+from stagehand.types import AvailableModel
 
 
 class StagehandConfig(BaseModel):

stagehand/handlers/observe_handler.py

@@ -5,7 +5,7 @@
 from stagehand.a11y.utils import get_accessibility_tree, get_xpath_by_resolved_object_id
 from stagehand.llm.inference import observe as observe_inference
 from stagehand.metrics import StagehandFunctionName  # Changed import location
-from stagehand.schemas import ObserveOptions, ObserveResult
+from stagehand.types import ObserveOptions, ObserveResult
 from stagehand.utils import draw_observe_overlay
 
 

stagehand/main.py

@@ -28,7 +28,7 @@
 from .logging import StagehandLogger, default_log_handler
 from .metrics import StagehandFunctionName, StagehandMetrics
 from .page import StagehandPage
-from .schemas import AgentConfig
+from .types import AgentConfigAPI as AgentConfig
 from .utils import make_serializable
 
 load_dotenv()

stagehand/page.py

@@ -7,7 +7,7 @@
 from stagehand.handlers.extract_handler import ExtractHandler
 from stagehand.handlers.observe_handler import ObserveHandler
 
-from .schemas import (
+from .types import (
     DEFAULT_EXTRACT_SCHEMA,
     ActOptions,
     ActResult,