agent-starter-pack 0.17.1__py3-none-any.whl → 0.17.2__py3-none-any.whl

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.1.dist-info → agent_starter_pack-0.17.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.17.1
+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.1.dist-info → agent_starter_pack-0.17.2.dist-info}/RECORD

@@ -70,7 +70,7 @@ agent_starter_pack/base_template/{{cookiecutter.agent_directory}}/utils/tracing.
 agent_starter_pack/base_template/{{cookiecutter.agent_directory}}/utils/typing.py,sha256=cfirFUBZQOwfQjEs9EYYP2nubv6BEpVzFHPj85PTMmg,4221
 agent_starter_pack/cli/main.py,sha256=eKi-Oi6pv2-LNiN3RYdrmvDiLXInWHkNMn8lGlkmVP4,1982
 agent_starter_pack/cli/commands/create.py,sha256=wuRL9lX6_AL6EucRSlixP91iIo5ROeDlLUAOIGOoCbA,47322
-agent_starter_pack/cli/commands/enhance.py,sha256=AF58CFikNLPib8PULKkzM6r9JGXkTvRWspBaEvGnR3k,24237
+agent_starter_pack/cli/commands/enhance.py,sha256=zFnxuQbEVTUtkRS3XuuuHlZ7wKIrpaNzggYN2100oYM,24622
 agent_starter_pack/cli/commands/list.py,sha256=ZGol9eYB9Yon7JysMUCtpEOwdXzrApdTHzErx6KvT04,6856
 agent_starter_pack/cli/commands/setup_cicd.py,sha256=UnhU8p9emfBV-7_6uKYdUfTQBnsbSIFQbJoMHMq8pik,32507
 agent_starter_pack/cli/utils/__init__.py,sha256=_cTmsXGPqOtK0q8UW5164QTltbJRJFR_Efxq_BRL1-o,1311
@@ -172,8 +172,8 @@ agent_starter_pack/utils/generate_locks.py,sha256=LwR46w25CV0_ySF_MS9W2vp7S1Nf17
 agent_starter_pack/utils/lock_utils.py,sha256=vqFHTN7lju9to74MKOmLvz80BA47G0CVkxrd1MyLWYQ,2309
 agent_starter_pack/utils/watch_and_rebuild.py,sha256=1hMn29eLpVblfvixV9FEo46u8efmnLuRIn-bPrfdW3g,6694
 llm.txt,sha256=z9FTkAnj3cErCzHj3IES9eB35wW3JGaVsd8a0wA7IMs,15380
-agent_starter_pack-0.17.1.dist-info/METADATA,sha256=m18MZ3DYwCLYcuyhsZU0krkfLq-h8VuK_dh_NhGmkNY,11195
-agent_starter_pack-0.17.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.17.1.dist-info/entry_points.txt,sha256=QLSX_97UynG0vLpoZ3ePfyCQkaR-g0qglp2b26G4KHQ,71
-agent_starter_pack-0.17.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.17.1.dist-info/RECORD,,
+agent_starter_pack-0.17.2.dist-info/METADATA,sha256=yJSYoXd9l5G7IlDuSfpUOv28GDC3B8jyswf1j6hIqZo,11195
+agent_starter_pack-0.17.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.17.2.dist-info/entry_points.txt,sha256=QLSX_97UynG0vLpoZ3ePfyCQkaR-g0qglp2b26G4KHQ,71
+agent_starter_pack-0.17.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.17.2.dist-info/RECORD,,