agent-starter-pack 0.17.0__tar.gz → 0.17.2__tar.gz

{agent_starter_pack-0.17.0 → agent_starter_pack-0.17.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.17.0
+Version: 0.17.2
 Summary: CLI to bootstrap production-ready Google Cloud GenAI agent projects from templates.
 Author-email: Google LLC <agent-starter-pack@google.com>
 License: Apache-2.0

{agent_starter_pack-0.17.0 → agent_starter_pack-0.17.2}/agent_starter_pack/cli/commands/enhance.py

@@ -94,16 +94,22 @@ def display_base_template_selection(current_base: str) -> str:
 
 
 def display_agent_directory_selection(
-    current_dir: pathlib.Path, detected_directory: str
+    current_dir: pathlib.Path, detected_directory: str, base_template: str | None = None
 ) -> str:
     """Display available directories and prompt for agent directory selection."""
+    # Determine the required object name based on base template
+    is_adk = base_template and "adk" in base_template.lower()
+    required_object = "root_agent" if is_adk else "agent"
+
     while True:
         console.print()
         console.print("📁 [bold]Agent Directory Selection[/bold]")
         console.print()
         console.print("Your project needs an agent directory containing:")
         console.print("  • [cyan]agent.py[/cyan] file with your agent logic")
-        console.print("  • [cyan]root_agent[/cyan] variable defined in agent.py")
+        console.print(
+            f"  • [cyan]{required_object}[/cyan] variable defined in agent.py"
+        )
         console.print()
         console.print("Choose where your agent code is located:")
 
@@ -323,7 +329,10 @@ def enhance(
         console.print()
 
     # Determine agent specification based on template_path
-    if template_path == pathlib.Path("."):
+    # If base_template is specified, use it as the agent spec
+    if base_template:
+        agent_spec = base_template
+    elif template_path == pathlib.Path("."):
         # Current directory - use local@ syntax
         agent_spec = "local@."
     elif template_path.is_dir():
@@ -421,7 +430,7 @@ def enhance(
         # Interactive agent directory selection if not provided via CLI and not auto-approved
         if not agent_directory and not auto_approve:
             selected_agent_directory = display_agent_directory_selection(
-                current_dir, detected_agent_directory
+                current_dir, detected_agent_directory, base_template
             )
             final_agent_directory = selected_agent_directory
             console.print(

{agent_starter_pack-0.17.0 → agent_starter_pack-0.17.2}/agent_starter_pack/cli/utils/template.py

@@ -100,7 +100,7 @@ def get_overwrite_folders(agent_directory: str) -> list[str]:
 
 TEMPLATE_CONFIG_FILE = "templateconfig.yaml"
 DEPLOYMENT_FOLDERS = ["cloud_run", "agent_engine"]
-DEFAULT_FRONTEND = "streamlit"
+DEFAULT_FRONTEND = "None"
 
 
 def get_available_agents(deployment_target: str | None = None) -> dict:
@@ -1182,14 +1182,11 @@ def copy_files(
 
 def copy_frontend_files(frontend_type: str, project_template: pathlib.Path) -> None:
     """Copy files from the specified frontend folder directly to project root."""
-    # Skip copying if frontend_type is "None"
-    if frontend_type == "None":
-        logging.debug("Frontend type is 'None', skipping frontend files")
+    # Skip copying if frontend_type is "None" or empty
+    if not frontend_type or frontend_type == "None":
+        logging.debug("Frontend type is 'None' or empty, skipping frontend files")
         return
 
-    # Use default frontend if none specified
-    frontend_type = frontend_type or DEFAULT_FRONTEND
-
     # Get the frontends directory path
     frontends_path = (
         pathlib.Path(__file__).parent.parent.parent / "frontends" / frontend_type
@@ -1201,9 +1198,12 @@ def copy_frontend_files(frontend_type: str, project_template: pathlib.Path) -> N
         copy_files(frontends_path, project_template, overwrite=True)
     else:
         logging.warning(f"Frontend type directory not found: {frontends_path}")
-        if frontend_type != DEFAULT_FRONTEND:
+        # Don't fall back to default if it's "None" - just skip
+        if DEFAULT_FRONTEND != "None":
             logging.info(f"Falling back to default frontend: {DEFAULT_FRONTEND}")
             copy_frontend_files(DEFAULT_FRONTEND, project_template)
+        else:
+            logging.debug("No default frontend configured, skipping frontend files")
 
 
 def copy_deployment_files(

{agent_starter_pack-0.17.0 → agent_starter_pack-0.17.2}/docs/guide/template-config-reference.md

@@ -27,7 +27,7 @@ This object contains fields that control the generated project's features and be
 | --------------------------- | -------------- | ------------------------------------------------------------------------------------------------------------------------------------------- |
 | `deployment_targets`        | list(string)   | A list of deployment targets your template supports. Options: `agent_engine`, `cloud_run`.                                                  |
 | `tags`                      | list(string)   | A list of tags for categorization. The `adk` tag enables special integrations with the Agent Development Kit.                                 |
-| `frontend_type`             | string         | Specifies the frontend to use. Examples: `streamlit`, `adk_live_react`. Defaults to `streamlit`.                                             |
+| `frontend_type`             | string         | Specifies the frontend to use. Examples: `streamlit`, `adk_live_react`. Defaults to `None` (no frontend).                                    |
 | `agent_directory`           | string         | The name of the directory where agent code will be placed. Defaults to `app`. Can be overridden by the CLI `--agent-directory` parameter.    |
 | `requires_data_ingestion`   | boolean        | If `true`, the user will be prompted to configure a datastore.                                                                              |
 | `requires_session`          | boolean        | If `true`, the user will be prompted to choose a session storage type (e.g., `alloydb`) when using the `cloud_run` target.                    |

{agent_starter_pack-0.17.0 → agent_starter_pack-0.17.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.17.0"
+version = "0.17.2"
 description = "CLI to bootstrap production-ready Google Cloud GenAI agent projects from templates."
 authors = [
     { name = "Google LLC", email = "agent-starter-pack@google.com" },

{agent_starter_pack-0.17.0 → agent_starter_pack-0.17.2}/tests/cli/commands/test_enhance.py

@@ -283,3 +283,203 @@ packages = ["detected_agent", "frontend"]
                 assert cli_overrides is not None
                 assert cli_overrides["base_template"] == "langgraph_base_react"
                 assert cli_overrides["settings"]["agent_directory"] == "my_chatbot"
+
+
+class TestEnhanceAgentEngineAppGeneration:
+    """Test that enhance properly generates agent_engine_app.py with correct imports."""
+
+    @pytest.mark.parametrize(
+        "base_template,expected_import",
+        [
+            ("adk_base", "root_agent"),
+            ("adk_live", "root_agent"),
+            ("langgraph_base_react", "agent"),
+            ("agentic_rag", "root_agent"),  # agentic_rag is ADK-based
+        ],
+    )
+    def test_agent_engine_app_has_correct_import(
+        self, base_template: str, expected_import: str, tmp_path: pathlib.Path
+    ) -> None:
+        """Test that agent_engine_app.py imports the correct variable based on base template."""
+        runner = CliRunner()
+
+        with runner.isolated_filesystem(temp_dir=tmp_path):
+            # Create agent directory with agent.py
+            agent_dir = pathlib.Path("app")
+            agent_dir.mkdir()
+            agent_file = agent_dir / "agent.py"
+
+            # Create appropriate agent.py content based on template type
+            if "adk" in base_template or base_template == "agentic_rag":
+                agent_content = """from google.adk.agents import Agent
+
+root_agent = Agent(
+    name="test_agent",
+    model="gemini-2.0-flash-001",
+)
+"""
+            else:
+                agent_content = """from langchain_core.runnables import RunnablePassthrough
+
+agent = RunnablePassthrough()
+"""
+            agent_file.write_text(agent_content)
+
+            # Run enhance with the specified base template
+            result = runner.invoke(
+                enhance,
+                [
+                    ".",
+                    "--base-template",
+                    base_template,
+                    "--deployment-target",
+                    "agent_engine",
+                    "--auto-approve",
+                    "--skip-checks",
+                ],
+            )
+
+            # Check that enhance succeeded
+            assert result.exit_code == 0, (
+                f"Enhance failed with output:\n{result.output}"
+            )
+
+            # Verify agent_engine_app.py was created
+            agent_engine_app = agent_dir / "agent_engine_app.py"
+            assert agent_engine_app.exists(), (
+                f"agent_engine_app.py not created in {agent_dir}"
+            )
+
+            # Read the content and verify the correct import
+            content = agent_engine_app.read_text()
+            expected_import_line = f"from app.agent import {expected_import}"
+            assert expected_import_line in content, (
+                f"Expected '{expected_import_line}' in agent_engine_app.py but got:\n{content}"
+            )
+
+    def test_agent_engine_app_created_in_custom_agent_directory(
+        self, tmp_path: pathlib.Path
+    ) -> None:
+        """Test that agent_engine_app.py is created in custom agent directory."""
+        runner = CliRunner()
+
+        with runner.isolated_filesystem(temp_dir=tmp_path):
+            # Create custom agent directory
+            agent_dir = pathlib.Path("my_custom_agent")
+            agent_dir.mkdir()
+            agent_file = agent_dir / "agent.py"
+            agent_file.write_text(
+                """from google.adk.agents import Agent
+
+root_agent = Agent(
+    name="test_agent",
+    model="gemini-2.0-flash-001",
+)
+"""
+            )
+
+            # Run enhance with custom agent directory
+            result = runner.invoke(
+                enhance,
+                [
+                    ".",
+                    "--base-template",
+                    "adk_base",
+                    "--agent-directory",
+                    "my_custom_agent",
+                    "--deployment-target",
+                    "agent_engine",
+                    "--auto-approve",
+                    "--skip-checks",
+                ],
+            )
+
+            # Check that enhance succeeded
+            assert result.exit_code == 0, (
+                f"Enhance failed with output:\n{result.output}"
+            )
+
+            # Verify agent_engine_app.py was created in custom directory
+            agent_engine_app = agent_dir / "agent_engine_app.py"
+            assert agent_engine_app.exists(), (
+                f"agent_engine_app.py not created in {agent_dir}"
+            )
+
+            # Verify the import uses the custom directory name
+            content = agent_engine_app.read_text()
+            expected_import_line = "from my_custom_agent.agent import root_agent"
+            assert expected_import_line in content, (
+                f"Expected '{expected_import_line}' in agent_engine_app.py"
+            )
+
+
+class TestEnhanceAgentDirectoryPrompt:
+    """Test that enhance shows the correct required variable in prompts."""
+
+    @patch("agent_starter_pack.cli.commands.enhance.display_agent_directory_selection")
+    @patch("agent_starter_pack.cli.utils.remote_template.get_base_template_name")
+    @patch("agent_starter_pack.cli.utils.remote_template.load_remote_template_config")
+    def test_prompt_shows_root_agent_for_adk_templates(
+        self,
+        mock_load_config: MagicMock,
+        mock_get_base_name: MagicMock,
+        mock_display_selection: MagicMock,
+    ) -> None:
+        """Test that agent directory prompt shows 'root_agent' for ADK templates."""
+        runner = CliRunner()
+
+        # Mock the template config to return an ADK base template
+        mock_get_base_name.return_value = "adk_base"
+        mock_load_config.return_value = {"base_template": "adk_base"}
+        mock_display_selection.return_value = "app"
+
+        with runner.isolated_filesystem():
+            pathlib.Path("app").mkdir()
+            pathlib.Path("app/agent.py").write_text("root_agent = None")
+
+            with patch("agent_starter_pack.cli.commands.enhance.create"):
+                runner.invoke(
+                    enhance,
+                    [".", "--base-template", "adk_base"],
+                    input="n\n",  # Cancel enhancement
+                )
+
+                # Verify display_agent_directory_selection was called with base_template
+                if mock_display_selection.called:
+                    call_args = mock_display_selection.call_args
+                    # The base_template should be passed to the function
+                    assert call_args[0][2] == "adk_base"  # Third positional arg
+
+    @patch("agent_starter_pack.cli.commands.enhance.display_agent_directory_selection")
+    @patch("agent_starter_pack.cli.utils.remote_template.get_base_template_name")
+    @patch("agent_starter_pack.cli.utils.remote_template.load_remote_template_config")
+    def test_prompt_shows_agent_for_non_adk_templates(
+        self,
+        mock_load_config: MagicMock,
+        mock_get_base_name: MagicMock,
+        mock_display_selection: MagicMock,
+    ) -> None:
+        """Test that agent directory prompt shows 'agent' for non-ADK templates."""
+        runner = CliRunner()
+
+        # Mock the template config to return a non-ADK base template
+        mock_get_base_name.return_value = "langgraph_base_react"
+        mock_load_config.return_value = {"base_template": "langgraph_base_react"}
+        mock_display_selection.return_value = "app"
+
+        with runner.isolated_filesystem():
+            pathlib.Path("app").mkdir()
+            pathlib.Path("app/agent.py").write_text("agent = None")
+
+            with patch("agent_starter_pack.cli.commands.enhance.create"):
+                runner.invoke(
+                    enhance,
+                    [".", "--base-template", "langgraph_base_react"],
+                    input="n\n",  # Cancel enhancement
+                )
+
+                # Verify display_agent_directory_selection was called with base_template
+                if mock_display_selection.called:
+                    call_args = mock_display_selection.call_args
+                    # The base_template should be passed to the function
+                    assert call_args[0][2] == "langgraph_base_react"

{agent_starter_pack-0.17.0 → agent_starter_pack-0.17.2}/uv.lock

@@ -11,7 +11,7 @@ resolution-markers = [
 
 [[package]]
 name = "agent-starter-pack"
-version = "0.17.0"
+version = "0.17.2"
 source = { editable = "." }
 dependencies = [
     { name = "backoff" },