fix: point to Gemini Enterprise registration docs after agent_engine deploy

Merge https://github.com/google/adk-python/pull/6634

Fixes #6633

PiperOrigin-RevId: 964955729
This commit is contained in:
chelsealong
2026-08-14 16:40:39 -07:00
committed by Copybara-Service
parent b0cdecfc3f
commit 1d2d1eda3c
2 changed files with 23 additions and 0 deletions
+10
View File
@@ -878,6 +878,15 @@ def _print_agent_engine_url(resource_name: str) -> None:
)
def _print_gemini_enterprise_hint() -> None:
"""Prints a pointer to the Gemini Enterprise registration docs."""
click.secho(
'To make this agent available in Gemini Enterprise, register it by'
' following:\nhttps://docs.cloud.google.com/gemini/enterprise/docs/register-and-manage-an-adk-agent\n',
fg='cyan',
)
def to_agent_engine(
*,
agent_folder: str,
@@ -1322,6 +1331,7 @@ def to_agent_engine(
click.secho(f'Cleaned up the instance: {resource_name}', fg='green')
raise e
_print_agent_engine_url(resource_name)
_print_gemini_enterprise_hint()
finally:
temp_folder_path = os.path.join(parent_folder, temp_folder)
click.echo(f'Cleaning up the temp folder: {temp_folder_path}')
@@ -250,6 +250,19 @@ def test_print_agent_engine_url() -> None:
assert "playground" in call_args
def test_print_gemini_enterprise_hint() -> None:
"""It should print a pointer to the Gemini Enterprise registration docs."""
with mock.patch("click.secho") as mocked_secho:
cli_deploy._print_gemini_enterprise_hint()
mocked_secho.assert_called_once()
call_args = mocked_secho.call_args[0][0]
assert "Gemini Enterprise" in call_args
assert (
"https://docs.cloud.google.com/gemini/enterprise/docs/register-and-manage-an-adk-agent"
in call_args
)
@pytest.mark.parametrize("include_requirements", [True, False])
def test_to_agent_engine_happy_path(
monkeypatch: pytest.MonkeyPatch,