Skip to content

Commit 1d2d1ed

Browse files
chelsealongcopybara-github
authored andcommitted
fix: point to Gemini Enterprise registration docs after agent_engine deploy
Merge #6634 Fixes #6633 PiperOrigin-RevId: 964955729
1 parent b0cdecf commit 1d2d1ed

2 files changed

Lines changed: 23 additions & 0 deletions

File tree

src/google/adk/cli/cli_deploy.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -878,6 +878,15 @@ def _print_agent_engine_url(resource_name: str) -> None:
878878
)
879879

880880

881+
def _print_gemini_enterprise_hint() -> None:
882+
"""Prints a pointer to the Gemini Enterprise registration docs."""
883+
click.secho(
884+
'To make this agent available in Gemini Enterprise, register it by'
885+
' following:\nhttps://docs.cloud.google.com/gemini/enterprise/docs/register-and-manage-an-adk-agent\n',
886+
fg='cyan',
887+
)
888+
889+
881890
def to_agent_engine(
882891
*,
883892
agent_folder: str,
@@ -1322,6 +1331,7 @@ def create_dockerfile_for_agent_engine(resource_name: str) -> None:
13221331
click.secho(f'Cleaned up the instance: {resource_name}', fg='green')
13231332
raise e
13241333
_print_agent_engine_url(resource_name)
1334+
_print_gemini_enterprise_hint()
13251335
finally:
13261336
temp_folder_path = os.path.join(parent_folder, temp_folder)
13271337
click.echo(f'Cleaning up the temp folder: {temp_folder_path}')

tests/unittests/cli/utils/test_cli_deploy.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,6 +250,19 @@ def test_print_agent_engine_url() -> None:
250250
assert "playground" in call_args
251251

252252

253+
def test_print_gemini_enterprise_hint() -> None:
254+
"""It should print a pointer to the Gemini Enterprise registration docs."""
255+
with mock.patch("click.secho") as mocked_secho:
256+
cli_deploy._print_gemini_enterprise_hint()
257+
mocked_secho.assert_called_once()
258+
call_args = mocked_secho.call_args[0][0]
259+
assert "Gemini Enterprise" in call_args
260+
assert (
261+
"https://docs.cloud.google.com/gemini/enterprise/docs/register-and-manage-an-adk-agent"
262+
in call_args
263+
)
264+
265+
253266
@pytest.mark.parametrize("include_requirements", [True, False])
254267
def test_to_agent_engine_happy_path(
255268
monkeypatch: pytest.MonkeyPatch,

0 commit comments

Comments
 (0)