formatting

Author: derekmeegan

stagehand/agent/google_cua.py

@@ -368,7 +368,9 @@ def _process_provider_response(
                 try:
                     # Directly construct the AgentActionType using the payload.
                     # Pydantic will use the 'type' field in action_payload_dict to discriminate the Union.
-                    action_payload_for_agent_action_type = TypeAdapter(AgentActionType).validate_python(action_payload_dict)
+                    action_payload_for_agent_action_type = TypeAdapter(
+                        AgentActionType
+                    ).validate_python(action_payload_dict)
 
                     agent_action = AgentAction(
                         action_type=action_type_str,  # This should match the 'type' in action_payload_dict

stagehand/agent/openai_cua.py

@@ -7,7 +7,10 @@
 from openai import (
     OpenAI as OpenAISDK,  # Renamed to avoid conflict with a potential class name
 )
-from pydantic import BaseModel, TypeAdapter  # Ensure BaseModel is imported for isinstance check
+from pydantic import (
+    BaseModel,
+    TypeAdapter,
+)  # Ensure BaseModel is imported for isinstance check
 
 from ..handlers.cua_handler import CUAHandler
 from ..types.agent import (