From b7a8091fa487935dbb88e2e3b58517983b4c285e Mon Sep 17 00:00:00 2001
From: A Vertex SDK engineer <vertex-sdk-bot@google.com>
Date: Thu, 19 Mar 2026 08:14:42 -0700
Subject: [PATCH] fix: add timeout for message send requests

PiperOrigin-RevId: 886188090
---
 vertexai/_genai/_agent_engines_utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/vertexai/_genai/_agent_engines_utils.py b/vertexai/_genai/_agent_engines_utils.py
index 1b9fc8ca79..6291ba11a4 100644
--- a/vertexai/_genai/_agent_engines_utils.py
+++ b/vertexai/_genai/_agent_engines_utils.py
@@ -1822,7 +1822,9 @@ async def _method(self, **kwargs) -> Any:  # type: ignore[no-untyped-def]
                     "Authorization": (
                         f"Bearer {self.api_client._api_client._credentials.token}"
                     )
-                }
+                },
+                timeout=self.api_client._api_client._http_options.timeout / 1000.0
+                or 60.0,
             ),
         )
         factory = ClientFactory(config)