diff --git a/.changeset/outrageous-adept-hyena.md b/.changeset/outrageous-adept-hyena.md new file mode 100644 index 00000000..0f23d6a8 --- /dev/null +++ b/.changeset/outrageous-adept-hyena.md @@ -0,0 +1,5 @@ +--- +"stagehand": patch +--- + +Fix draw_overlay on env:LOCAL diff --git a/stagehand/handlers/observe_handler.py b/stagehand/handlers/observe_handler.py index 251f857c..f0f29181 100644 --- a/stagehand/handlers/observe_handler.py +++ b/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 diff --git a/stagehand/utils.py b/stagehand/utils.py index 4c1bb85b..2383f46f 100644 --- a/stagehand/utils.py +++ b/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.