feat: GenAI SDK client - Make operation polling interval configurable when creating agent engine sandbox

PiperOrigin-RevId: 867762024

Author: vertex-sdk-bot

tests/unit/vertexai/genai/replays/test_create_agent_engine_sandbox.py

@@ -25,6 +25,7 @@ def test_create_sandbox(client):
 
     operation = client.agent_engines.sandboxes.create(
         name=agent_engine.api_resource.name,
+        poll_interval_seconds=1,
         spec={
             "code_execution_environment": {
                 "machineConfig": "MACHINE_CONFIG_VCPU4_RAM4GIB"

vertexai/_genai/sandboxes.py

@@ -540,6 +540,7 @@ def create(
         self,
         *,
         name: str,
+        poll_interval_seconds: float = 0.1,
         spec: Optional[types.SandboxEnvironmentSpecOrDict] = None,
         config: Optional[types.CreateAgentEngineSandboxConfigOrDict] = None,
     ) -> types.AgentEngineSandboxOperation:
@@ -549,6 +550,9 @@ def create(
             name (str):
                 Required. The name of the agent engine to create sandbox for.
                 projects/{project}/locations/{location}/reasoningEngines/{resource_id}
+            poll_interval_seconds (float):
+                Optional. The interval in seconds to poll for sandbox creation
+                completion.
             spec (SandboxEnvironmentSpec):
                 Optional. The specification for the sandbox to create.
             config (CreateAgentEngineSandboxConfigOrDict):
@@ -582,7 +586,7 @@ def create(
                 operation = _agent_engines_utils._await_operation(
                     operation_name=operation.name,
                     get_operation_fn=self._get_sandbox_operation,
-                    poll_interval_seconds=0.1,
+                    poll_interval_seconds=poll_interval_seconds,
                 )
             # We need to make a call to get the sandbox because the operation
             # response might not contain the relevant fields.