fix: Decrease polling interval for GenerateMemories and CreateSession (currently 10s)

vertex-sdk-bot · copybara-github · commit 77a39330473b · 2025-09-03T13:07:44.000-07:00
PiperOrigin-RevId: 802674616
diff --git a/vertexai/_genai/memories.py b/vertexai/_genai/memories.py
@@ -1283,6 +1283,7 @@ def generate(
             operation = _agent_engines_utils._await_operation(
                 operation_name=operation.name,
                 get_operation_fn=self._get_generate_memories_operation,
+                poll_interval_seconds=0.5,
             )
             if operation.error:
                 raise RuntimeError(f"Failed to generate memory: {operation.error}")
diff --git a/vertexai/_genai/sessions.py b/vertexai/_genai/sessions.py
@@ -770,6 +770,7 @@ def create(
             operation = _agent_engines_utils._await_operation(
                 operation_name=operation.name,
                 get_operation_fn=self._get_session_operation,
+                poll_interval_seconds=0.5,
             )
             if operation.response:
                 operation.response = self.get(name=operation.response.name)

Original file line number	Diff line number	Diff line change
`@@ -1283,6 +1283,7 @@ def generate(`
`1283`	`1283`	`operation = _agent_engines_utils._await_operation(`
`1284`	`1284`	`operation_name=operation.name,`
`1285`	`1285`	`get_operation_fn=self._get_generate_memories_operation,`
	`1286`	`+ poll_interval_seconds=0.5,`
`1286`	`1287`	`)`
`1287`	`1288`	`if operation.error:`
`1288`	`1289`	`raise RuntimeError(f"Failed to generate memory: {operation.error}")`
Original file line number	Diff line number	Diff line change
`@@ -770,6 +770,7 @@ def create(`
`770`	`770`	`operation = _agent_engines_utils._await_operation(`
`771`	`771`	`operation_name=operation.name,`
`772`	`772`	`get_operation_fn=self._get_session_operation,`
	`773`	`+ poll_interval_seconds=0.5,`
`773`	`774`	`)`
`774`	`775`	`if operation.response:`
`775`	`776`	`operation.response = self.get(name=operation.response.name)`