feat(cli): add otel_to_cloud flag to adk deploy agent_engine command

Co-authored-by: Wiktoria Walczak <wwalczak@google.com>
PiperOrigin-RevId: 858546581
This commit is contained in:
Wiktoria Walczak
2026-01-20 05:41:39 -08:00
committed by Copybara-Service
parent 69ad605bc4
commit 21f63f66ee
3 changed files with 53 additions and 0 deletions
+11
View File
@@ -689,6 +689,7 @@ def to_agent_engine(
adk_app: str,
staging_bucket: Optional[str] = None,
trace_to_cloud: Optional[bool] = None,
otel_to_cloud: Optional[bool] = None,
api_key: Optional[str] = None,
adk_app_object: Optional[str] = None,
agent_engine_id: Optional[str] = None,
@@ -733,6 +734,8 @@ def to_agent_engine(
staging_bucket (str): Deprecated. This argument is no longer required or
used.
trace_to_cloud (bool): Whether to enable Cloud Trace.
otel_to_cloud (bool): Whether to enable exporting OpenTelemetry signals
to Google Cloud.
api_key (str): Optional. The API key to use for Express Mode.
If not provided, the API key from the GOOGLE_API_KEY environment variable
will be used. It will only be used if GOOGLE_GENAI_USE_VERTEXAI is true.
@@ -910,6 +913,14 @@ def to_agent_engine(
if 'GOOGLE_API_KEY' in env_vars:
api_key = env_vars['GOOGLE_API_KEY']
click.echo(f'api_key set by GOOGLE_API_KEY in {env_file}')
if otel_to_cloud:
if 'GOOGLE_CLOUD_AGENT_ENGINE_ENABLE_TELEMETRY' in env_vars:
click.secho(
'Ignoring GOOGLE_CLOUD_AGENT_ENGINE_ENABLE_TELEMETRY in .env'
' as `--otel_to_cloud` was explicitly passed and takes precedence',
fg='yellow',
)
env_vars['GOOGLE_CLOUD_AGENT_ENGINE_ENABLE_TELEMETRY'] = 'true'
if env_vars:
if 'env_vars' in agent_config:
click.echo(
+10
View File
@@ -1751,6 +1751,14 @@ def cli_migrate_session(
default=None,
help="Optional. Whether to enable Cloud Trace for Agent Engine.",
)
@click.option(
"--otel_to_cloud",
type=bool,
is_flag=True,
show_default=True,
default=None,
help="Optional. Whether to enable OpenTelemetry for Agent Engine.",
)
@click.option(
"--display_name",
type=str,
@@ -1842,6 +1850,7 @@ def cli_deploy_agent_engine(
staging_bucket: Optional[str],
agent_engine_id: Optional[str],
trace_to_cloud: Optional[bool],
otel_to_cloud: Optional[bool],
api_key: Optional[str],
display_name: str,
description: str,
@@ -1872,6 +1881,7 @@ def cli_deploy_agent_engine(
region=region,
agent_engine_id=agent_engine_id,
trace_to_cloud=trace_to_cloud,
otel_to_cloud=otel_to_cloud,
api_key=api_key,
adk_app_object=adk_app_object,
display_name=display_name,