change fix to use original function name from gemini

Author: derekmeegan

stagehand/agent/google_cua.py

@@ -349,7 +349,7 @@ def _process_provider_response(
                 action_type_str = "function"
                 action_payload_dict = {
                     "type": "function",
-                    "name": "search",
+                    "name": "goto",
                     "arguments": {"url": "https://www.google.com"},
                 }
             else:
@@ -567,34 +567,19 @@ async def run_task(
                         )
                         current_url = self.handler.page.url
 
-                    function_name_called_for_feedback = ""
-                    if agent_action.action_type == "function" and isinstance(
-                        agent_action.action.root, FunctionAction
-                    ):
-                        function_name_called_for_feedback = (
-                            agent_action.action.root.name
-                        )
-                        self._format_action_feedback(
-                            function_name_called=function_name_called_for_feedback,
-                            action_result=action_result,
-                            new_screenshot_base64=current_screenshot_b64,
-                            current_url=current_url,
-                            function_call_args=function_call_args,
+                    if not invoked_function_name:
+                        self.logger.error(
+                            "Original Google function name not found for feedback loop (was None).",
+                            category="agent",
                         )
-                    else:
-                        if not invoked_function_name:
-                            self.logger.error(
-                                "Original Google function name not found for feedback loop (was None).",
-                                category="agent",
-                            )
-                        else:
-                            self._format_action_feedback(
-                                function_name_called=invoked_function_name,
-                                action_result=action_result,
-                                new_screenshot_base64=current_screenshot_b64,
-                                current_url=current_url,
-                                function_call_args=function_call_args,
-                            )
+
+                    self._format_action_feedback(
+                        function_name_called=invoked_function_name,
+                        action_result=action_result,
+                        new_screenshot_base64=current_screenshot_b64,
+                        current_url=current_url,
+                        function_call_args=function_call_args,
+                    )
 
             if task_completed:
                 self.logger.info(

stagehand/handlers/cua_handler.py

@@ -142,7 +142,7 @@ async def perform_action(self, action: AgentAction) -> ActionExecutionResult:
                 # specific_action_model is FunctionAction
                 name = specific_action_model.name
                 args = getattr(specific_action_model, "arguments", {})
-                if name == "search" and args.url:
+                if name == "goto" and args.url:
                     await self.page.goto(args.url)
                     return {"success": True}
                 elif name == "navigate_back":