From 6cb3a7d995e22e2831982082ad8c248ed1a08e2a Mon Sep 17 00:00:00 2001 From: Ben Kazemi Date: Wed, 1 Apr 2026 11:48:05 -0700 Subject: [PATCH] chore: Internal PiperOrigin-RevId: 893039628 --- vertexai/_genai/types/common.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/vertexai/_genai/types/common.py b/vertexai/_genai/types/common.py index 7dd6062b3e..3bb1b2de0f 100644 --- a/vertexai/_genai/types/common.py +++ b/vertexai/_genai/types/common.py @@ -15722,8 +15722,14 @@ class RunQueryJobAgentEngineConfig(_common.BaseModel): query: Optional[str] = Field( default=None, description="""The query to send to the agent engine.""" ) - gcs_bucket: Optional[str] = Field( - default=None, description="""The GCS bucket to use for the query.""" + output_gcs_uri: Optional[str] = Field( + default=None, + description="""The GCS URI to use for the output. + If it is a file, the system use this file to store the response. + If it represents a directory, the system automatically generate a file + for the response. + In both cases, the input query will be stored in the same directory under + the same file name prefix as the output file.""", ) @@ -15736,8 +15742,13 @@ class RunQueryJobAgentEngineConfigDict(TypedDict, total=False): query: Optional[str] """The query to send to the agent engine.""" - gcs_bucket: Optional[str] - """The GCS bucket to use for the query.""" + output_gcs_uri: Optional[str] + """The GCS URI to use for the output. + If it is a file, the system use this file to store the response. + If it represents a directory, the system automatically generate a file + for the response. + In both cases, the input query will be stored in the same directory under + the same file name prefix as the output file.""" RunQueryJobAgentEngineConfigOrDict = Union[