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:
committed by
Copybara-Service
parent
b0cdecfc3f
commit
1d2d1eda3c
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user