diff --git a/src/google/adk/cli/cli_create.py b/src/google/adk/cli/cli_create.py index 3e5a5b0ed9..9d3f011880 100644 --- a/src/google/adk/cli/cli_create.py +++ b/src/google/adk/cli/cli_create.py @@ -23,7 +23,7 @@ from .utils import _onboarding _INIT_PY_TEMPLATE = """\ -from . import agent +from . import agent as agent """ _AGENT_PY_TEMPLATE = """\ diff --git a/tests/unittests/cli/utils/test_cli_create.py b/tests/unittests/cli/utils/test_cli_create.py index 98e0a92f88..bcb45218fe 100644 --- a/tests/unittests/cli/utils/test_cli_create.py +++ b/tests/unittests/cli/utils/test_cli_create.py @@ -71,7 +71,9 @@ def test_generate_files_with_api_key(agent_folder: Path) -> None: assert "GOOGLE_GENAI_USE_ENTERPRISE=0" in env_content assert (agent_folder / ".gitignore").read_text() == ".env\n" assert (agent_folder / "agent.py").exists() - assert (agent_folder / "__init__.py").exists() + assert ( + agent_folder / "__init__.py" + ).read_text() == "from . import agent as agent\n" def test_generate_files_with_gcp(agent_folder: Path) -> None: