agent-starter-pack 0.17.1__tar.gz → 0.17.3__tar.gz

{agent_starter_pack-0.17.1 → agent_starter_pack-0.17.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.17.1
+Version: 0.17.3
 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.1 → agent_starter_pack-0.17.3}/agent_starter_pack/cli/commands/create.py

@@ -470,7 +470,9 @@ def create(
         # Load template configuration based on whether it's remote or local
         if template_source_path:
             # Prepare CLI overrides for remote template config
-            cli_overrides = {}
+            # Initialize cli_overrides if not provided (e.g., from enhance command)
+            if cli_overrides is None:
+                cli_overrides = {}
             if base_template:
                 cli_overrides["base_template"] = base_template
 
@@ -491,7 +493,7 @@ def create(
                 logging.debug(f"Using base template: {base_template_name}")
 
             base_template_path = (
-                pathlib.Path(__file__).parent.parent.parent.parent
+                pathlib.Path(__file__).parent.parent.parent
                 / "agents"
                 / base_template_name
                 / ".template"
@@ -504,7 +506,7 @@ def create(
             template_path = template_source_path / ".template"
         else:
             template_path = (
-                pathlib.Path(__file__).parent.parent.parent.parent
+                pathlib.Path(__file__).parent.parent.parent
                 / "agents"
                 / final_agent
                 / ".template"

{agent_starter_pack-0.17.1 → agent_starter_pack-0.17.3}/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:")
 
@@ -358,6 +364,8 @@ def enhance(
             selected_base_template = display_base_template_selection(
                 original_base_template_name
             )
+            # Always set base_template to the selected value (even if unchanged)
+            base_template = selected_base_template
             if selected_base_template != original_base_template_name:
                 # Update CLI overrides with the selected base template
                 cli_overrides["base_template"] = selected_base_template
@@ -365,7 +373,6 @@ def enhance(
                 if agent_directory:
                     cli_overrides["settings"] = cli_overrides.get("settings", {})
                     cli_overrides["settings"]["agent_directory"] = agent_directory
-                base_template = selected_base_template
                 console.print(
                     f"✅ Selected base template: [cyan]{selected_base_template}[/cyan]"
                 )
@@ -421,7 +428,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(
@@ -573,9 +580,10 @@ def enhance(
         final_cli_overrides["base_template"] = base_template
 
     # For current directory templates, ensure agent_directory is included in cli_overrides
-    if template_path == pathlib.Path(".") and agent_directory:
+    # final_agent_directory is set from interactive selection or CLI/detection
+    if template_path == pathlib.Path(".") and final_agent_directory:
         final_cli_overrides["settings"] = final_cli_overrides.get("settings", {})
-        final_cli_overrides["settings"]["agent_directory"] = agent_directory
+        final_cli_overrides["settings"]["agent_directory"] = final_agent_directory
 
     # Call the create command with in-folder mode enabled
     ctx.invoke(

{agent_starter_pack-0.17.1 → agent_starter_pack-0.17.3}/agent_starter_pack/cli/utils/template.py

@@ -579,8 +579,8 @@ def process_template(
             )
             # Get agent directory from config early for use in file copying
             # Load config early to get agent_directory
-            if is_remote:
-                early_config = remote_config or {}
+            if remote_config:
+                early_config = remote_config
             else:
                 template_path = pathlib.Path(template_dir)
                 early_config = load_template_config(template_path)
@@ -620,25 +620,16 @@ def process_template(
                 )
                 copy_data_ingestion_files(project_template, datastore)
 
-            # 4. Process frontend files
-            # Load template config
-            template_config = load_template_config(pathlib.Path(template_dir))
-            frontend_type = template_config.get("settings", {}).get(
-                "frontend_type", DEFAULT_FRONTEND
-            )
-            copy_frontend_files(frontend_type, project_template)
-            logging.debug(f"4. Processed frontend files for type: {frontend_type}")
-
-            # 6. Skip remote template files during cookiecutter processing
+            # 4. Skip remote template files during cookiecutter processing
             # Remote files will be copied after cookiecutter to avoid Jinja conflicts
             if is_remote and remote_template_path:
                 logging.debug(
-                    "6. Skipping remote template files during cookiecutter processing - will copy after templating"
+                    "4. Skipping remote template files during cookiecutter processing - will copy after templating"
                 )
 
             # Load and validate template config first
-            if is_remote:
-                config = remote_config or {}
+            if remote_config:
+                config = remote_config
             else:
                 template_path = pathlib.Path(template_dir)
                 config = load_template_config(template_path)
@@ -659,7 +650,14 @@ def process_template(
             # Use the already loaded config
             template_config = config
 
-            # 5. Copy agent-specific files to override base template (using final config)
+            # Process frontend files (after config is properly loaded with CLI overrides)
+            frontend_type = template_config.get("settings", {}).get(
+                "frontend_type", DEFAULT_FRONTEND
+            )
+            copy_frontend_files(frontend_type, project_template)
+            logging.debug(f"5. Processed frontend files for type: {frontend_type}")
+
+            # 6. Copy agent-specific files to override base template (using final config)
             if agent_path.exists():
                 agent_directory = get_agent_directory(template_config, cli_overrides)
 
@@ -673,7 +671,7 @@ def process_template(
                 target_agent_folder = project_template / agent_directory
                 if source_agent_folder.exists():
                     logging.debug(
-                        f"5. Copying agent folder {template_agent_directory} -> {agent_directory} with override"
+                        f"6. Copying agent folder {template_agent_directory} -> {agent_directory} with override"
                     )
                     copy_files(
                         source_agent_folder,
@@ -689,7 +687,7 @@ def process_template(
                     agent_folder = agent_path / folder
                     project_folder = project_template / folder
                     if agent_folder.exists():
-                        logging.debug(f"5. Copying {folder} folder with override")
+                        logging.debug(f"6. Copying {folder} folder with override")
                         copy_files(
                             agent_folder,
                             project_folder,

{agent_starter_pack-0.17.1 → agent_starter_pack-0.17.3}/pyproject.toml

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