kwargs agent execute

Author: miguelg719

examples/agent_example.py

@@ -36,8 +36,8 @@
 async def main():
     # Build a unified configuration object for Stagehand
     config = StagehandConfig(
-        env="BROWSERBASE",
-        # env="LOCAL",
+        # env="BROWSERBASE",
+        env="LOCAL",
         api_key=os.getenv("BROWSERBASE_API_KEY"),
         project_id=os.getenv("BROWSERBASE_PROJECT_ID"),
         model_name="gpt-4o",
@@ -58,13 +58,6 @@ async def main():
         console.print(
             f"🌐 [white]View your live browser:[/] [url]https://www.browserbase.com/sessions/{stagehand.session_id}[/]"
         )
-    
-    # Define the task for the agent
-    execute_options = AgentExecuteOptions(
-        instruction="Play a game of 2048",
-        max_steps=20,
-        auto_screenshot=True,
-    )
 
     console.print("\n▶️ [highlight] Navigating[/] to Google")
     await stagehand.page.goto("https://google.com/")
@@ -76,7 +69,11 @@ async def main():
         instructions="You are a helpful web navigation assistant that helps users find information. You are currently on the following page: google.com. Do not ask follow up questions, the user will trust your judgement.",
         options={"apiKey": os.getenv("MODEL_API_KEY")}
     )
-    agent_result = await agent.execute(execute_options)
+    agent_result = await agent.execute(
+        instruction="Play a game of 2048",
+        max_steps=20,
+        auto_screenshot=True,
+    )
 
     console.print("📊 [info]Agent execution result:[/]")
     console.print(f"✅ Success: [bold]{'Yes' if agent_result.success else 'No'}[/]")

stagehand/agent/agent.py

@@ -87,31 +87,40 @@ def _get_client(self) -> AgentClient:
         )
 
     async def execute(
-        self, options_or_instruction: Union[AgentExecuteOptions, str]
+        self,
+        options_or_instruction: Union[AgentExecuteOptions, str, dict, None] = None,
+        **kwargs,
     ) -> AgentResult:
         options: Optional[AgentExecuteOptions] = None
-        instruction: str
+        options_dict = {}
 
-        if isinstance(options_or_instruction, str):
-            instruction = options_or_instruction
-            options = AgentExecuteOptions(instruction=instruction)
+        if isinstance(options_or_instruction, AgentExecuteOptions):
+            options_dict = options_or_instruction.model_dump()
         elif isinstance(options_or_instruction, dict):
-            options = AgentExecuteOptions(**options_or_instruction)
-            instruction = options.instruction
-        else:
-            options = options_or_instruction
-            instruction = options.instruction
+            options_dict = options_or_instruction.copy()
+        elif isinstance(options_or_instruction, str):
+            options_dict["instruction"] = options_or_instruction
+
+        options_dict.update(kwargs)
+
+        try:
+            options = AgentExecuteOptions(**options_dict)
+        except Exception as e:
+            self.logger.error(f"Invalid agent execute options: {e}")
+            raise
+
+        if not options.instruction:
+            self.logger.error("No instruction provided for agent execution.")
+            return AgentResult(
+                message="No instruction provided.",
+                completed=True,
+                actions=[],
+                usage={},
+            )
 
-        if self.stagehand.env == "LOCAL":
-            if not instruction:
-                self.logger.error("No instruction provided for agent execution.")
-                return AgentResult(
-                    message="No instruction provided.",
-                    completed=True,
-                    actions=[],
-                    usage={},
-                )
+        instruction = options.instruction
 
+        if self.stagehand.env == "LOCAL":
             self.logger.info(
                 f"Agent starting execution for instruction: '{instruction}'",
                 category="agent",