chore: Reduce polling interval for lightweight agent engine creation.

PiperOrigin-RevId: 778200996

Author: shawn-yang-google

tests/unit/vertexai/genai/test_agent_engines.py

@@ -983,6 +983,10 @@ def test_create_agent_engine(self, mock_await_operation, mock_prepare):
                     env_vars=_TEST_AGENT_ENGINE_ENV_VARS_INPUT,
                 ),
             )
+            mock_await_operation.assert_called_once_with(
+                operation_name=None,
+                poll_interval_seconds=10,
+            )
             request_mock.assert_called_with(
                 "post",
                 "reasoningEngines",
@@ -1023,6 +1027,10 @@ def test_create_agent_engine_lightweight(
                     description=_TEST_AGENT_ENGINE_DESCRIPTION,
                 )
             )
+            mock_await_operation.assert_called_once_with(
+                operation_name=None,
+                poll_interval_seconds=1,
+            )
             request_mock.assert_called_with(
                 "post",
                 "reasoningEngines",

vertexai/_genai/agent_engines.py

@@ -1988,7 +1988,15 @@ def create(
             env_vars=config.env_vars,
         )
         operation = self._create(config=api_config)
-        operation = self._await_operation(operation_name=operation.name)
+        if agent_engine is None:
+            poll_interval_seconds = 1  # Lightweight agent engine resource creation.
+        else:
+            poll_interval_seconds = 10
+        operation = self._await_operation(
+            operation_name=operation.name,
+            poll_interval_seconds=poll_interval_seconds,
+        )
+
         agent = types.AgentEngine(
             api_client=self,
             api_async_client=AsyncAgentEngines(api_client_=self._api_client),