Dump ObserveResult to dict before passing to draw_overlay (#198) (#201)

* Dump ObserveResult to dict before passing to draw_overlay (#198)

* changeset:

---------

Co-authored-by: Zach Loza <[email protected]>

Author: miguelg719

.changeset/outrageous-adept-hyena.md

@@ -0,0 +1,5 @@
+---
+"stagehand": patch
+---
+
+Fix draw_overlay on env:LOCAL

stagehand/handlers/observe_handler.py

@@ -118,7 +118,10 @@ async def observe(
 
         # Draw overlay if requested
         if options.draw_overlay:
-            await draw_observe_overlay(self.stagehand_page, elements_with_selectors)
+            await draw_observe_overlay(
+                page=self.stagehand_page,
+                elements=[el.model_dump() for el in elements_with_selectors],
+            )
 
         # Return the list of results without trying to attach _llm_response
         return elements_with_selectors

stagehand/utils.py

@@ -109,7 +109,7 @@ def format_simplified_tree(node: AccessibilityNode, level: int = 0) -> str:
     return result
 
 
-async def draw_observe_overlay(page, elements):
+async def draw_observe_overlay(page, elements: list[dict]):
     """
     Draw an overlay on the page highlighting the observed elements.