more clenaup

Author: pirate

examples/full_example.py

@@ -84,7 +84,9 @@ async def main() -> None:
 
             # Use the first result
             result = results[0]
-            print(f"Acting on: {result.description}")
+            if not isinstance(result, dict):
+                raise RuntimeError(f"Expected observe stream result item to be a dict, got {type(result).__name__}")
+            print(f"Acting on: {result['description']}")
 
             # Pass the action to Act
             act_stream = await session.act(

src/stagehand/_client.py

@@ -50,6 +50,7 @@
 
     ### <CUSTOM CODE HANDWRITTEN BY STAGEHAND TEAM (not codegen)>
     from ._custom.sea_server import SeaServerManager
+    from .resources.sessions import SessionsResource, AsyncSessionsResource
     ### </END CUSTOM CODE>
 
 __all__ = [
@@ -195,7 +196,7 @@ def close(self) -> None:
             ### </END CUSTOM CODE>
 
     @cached_property
-    def sessions(self) -> sessions.SessionsResource:
+    def sessions(self) -> SessionsResource:
         from .resources.sessions import SessionsResource
 
         return SessionsResource(self)
@@ -481,7 +482,7 @@ async def close(self) -> None:
             ### </END CUSTOM CODE>
 
     @cached_property
-    def sessions(self) -> sessions.AsyncSessionsResource:
+    def sessions(self) -> AsyncSessionsResource:
         from .resources.sessions import AsyncSessionsResource
 
         return AsyncSessionsResource(self)