agent-starter-pack 0.15.1__py3-none-any.whl → 0.15.3__py3-none-any.whl

{agent_starter_pack-0.15.1.dist-info → agent_starter_pack-0.15.3.dist-info}/METADATA

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

@@ -1,18 +1,18 @@
 agents/README.md,sha256=L8__lIXXjxLTZbkCmTC4taTFUMRVrO3QI2mCKZTdhJM,462
 agents/adk_base/README.md,sha256=_rnoIkHupZ4afDZEAtrDZzeYSS4WqtJWnvtX2H-RrCM,894
-agents/adk_base/.template/templateconfig.yaml,sha256=8mQ_IIgC0h4eiTUM7FEIPSxwI73aenspcHsnqUb4AfY,923
+agents/adk_base/.template/templateconfig.yaml,sha256=OwPdd3NoOpT4WpnlOHM5tk_PFdptDdqnS3v7UdwjeOQ,924
 agents/adk_base/app/__init__.py,sha256=wT1TtenBOdA4LTtf6mQjvAT5aRR1N2PCx_A-MrDLGsg,631
 agents/adk_base/app/agent.py,sha256=6Q6QegQDxEaAck989DPIb3ArHMZtw_YJBqhHFqg-UjQ,2182
-agents/adk_base/notebooks/adk_app_testing.ipynb,sha256=81L1kbeUlkC3DpTY9nGyB8h5ezCdEeNCVHkBPyG0fgc,10497
+agents/adk_base/notebooks/adk_app_testing.ipynb,sha256=hyPBwXOmZBWu1Dte6lUU0LDW_vV2VA57yQZ3AjBMzDU,10729
 agents/adk_base/notebooks/evaluating_adk_agent.ipynb,sha256=SdZC1htgXtXw3oyi8d70dNVpDgUmV9T5hqiwGxM9plk,57034
 agents/adk_base/tests/integration/test_agent.py,sha256=_zb3GMWMtMRq2rOyYC-uL1eNhpg2XBkm8Noj0O_UOng,1998
 agents/agentic_rag/README.md,sha256=cPeY_XGt79ava22PIM0x92oi67R-TnlEKBQGpK1MOkA,2262
-agents/agentic_rag/.template/templateconfig.yaml,sha256=Gfw3FOjj3XNxoXn6WGwC2PG59f4qB3p2cYTQ49ykvJc,1248
+agents/agentic_rag/.template/templateconfig.yaml,sha256=o8D1Yx7zWsIgI1zVOS3K1pDgc_Lczevw6cveaj2T7HQ,1249
 agents/agentic_rag/app/__init__.py,sha256=wT1TtenBOdA4LTtf6mQjvAT5aRR1N2PCx_A-MrDLGsg,631
 agents/agentic_rag/app/agent.py,sha256=joSAfrVcS9eRtOT3MXPSoiwElpwVq-hjpCHkqTj6vuM,4041
 agents/agentic_rag/app/retrievers.py,sha256=qfxwSEixMddvAu4FvUTZAFh1Gdp3RPY65s0Wp6gEFVk,4561
 agents/agentic_rag/app/templates.py,sha256=8N_uRchwiN7hlH9Y8rdJ1CAtOhpHUWZkXZZ7IU-YJJk,867
-agents/agentic_rag/notebooks/adk_app_testing.ipynb,sha256=81L1kbeUlkC3DpTY9nGyB8h5ezCdEeNCVHkBPyG0fgc,10497
+agents/agentic_rag/notebooks/adk_app_testing.ipynb,sha256=hyPBwXOmZBWu1Dte6lUU0LDW_vV2VA57yQZ3AjBMzDU,10729
 agents/agentic_rag/notebooks/evaluating_adk_agent.ipynb,sha256=SdZC1htgXtXw3oyi8d70dNVpDgUmV9T5hqiwGxM9plk,57034
 agents/agentic_rag/tests/integration/test_agent.py,sha256=YcwOZXa9uJqcqZN8UPikNm_dlSh9XduEfcuQa0wkby8,2170
 agents/crewai_coding_crew/README.md,sha256=4No7sfHQVHELARGiT3fWANJzDR0NP48ow4ct3FoPwlA,1672
@@ -30,7 +30,7 @@ agents/langgraph_base_react/app/agent.py,sha256=yzCc5JaOJgItoBiZOzHPiSy9aD2HGeGI
 agents/langgraph_base_react/notebooks/evaluating_langgraph_agent.ipynb,sha256=PlW41fL-ji9NKkimW6u25YRLGWN2pgoX2G8w5nzIEPE,58737
 agents/langgraph_base_react/tests/integration/test_agent.py,sha256=xy0A_3eGhxM8A5rc81revMztJN5GtFA-o5-CG5JzeQw,1630
 agents/live_api/README.md,sha256=c3lu5WguOUBc5hO_UtNZfVhfYxvSzMXuGFICn9M38mU,2897
-agents/live_api/.template/templateconfig.yaml,sha256=DD_hihMsuqmd5VWS18vVGZ3ZGVVpFWVXJ2dU07eKb6k,964
+agents/live_api/.template/templateconfig.yaml,sha256=Y_UQcxq5ql2YPrC5kd-SK7kfGb6Q6IqScaNDRv-vG5I,964
 agents/live_api/app/agent.py,sha256=dYE2qd64OhL-fOgwzrjqN50LYXAgJ1TWbx6ohqtV5Rc,2434
 agents/live_api/tests/integration/test_server_e2e.py,sha256=SBowHiVj-_Okbk1dqyITFVIG6yvr5BGNrfxHVpO4TFs,8635
 agents/live_api/tests/load_test/load_test.py,sha256=HHZyfC4gqiQtZVF_CbbxENGgWQccMLpwMv0IdoQ6cbQ,1275
@@ -40,7 +40,7 @@ src/base_template/.gitignore,sha256=Fq0w34x4sfbwP4RqDqh6hdHNYRUEsFNI-9XONzLWBIs,
 src/base_template/GEMINI.md,sha256=WzscHWlQeYkKORZ-453P8KM9IHuj1kAxW-69c7Ytuwk,133
 src/base_template/Makefile,sha256=1UKB6rlIbGStc-hUfnwKanOtxhz-vzaKxrCxTlPilqs,8087
 src/base_template/README.md,sha256=_TjEIWOK75GfaG3qfaOQdfC9J5AlPo_dckRY7e84dlM,10184
-src/base_template/pyproject.toml,sha256=HsLasS9vm6PZi7BiGrHe0tgBnwvgafa4eIj4k1P3DCU,2940
+src/base_template/pyproject.toml,sha256=Ch69IVTRzNn9zueU5TunZjb-dTmvCM_G_NwXDFUiIp8,2940
 src/base_template/deployment/README.md,sha256=gZJvSWdQh_Mi-bZ3dmuPv7fMezIw04fgN5tq7KgglPw,692
 src/base_template/deployment/terraform/apis.tf,sha256=KX8Oe2gjT-gh9Bkntz9yIAyRjPc1gZvwOhroJ6NZVp4,1513
 src/base_template/deployment/terraform/github.tf,sha256=KhfNBAzeUH9jDr8FOw_DiypesA5AEC8IZT1PRqntSt8,10838
@@ -72,14 +72,14 @@ src/base_template/{{cookiecutter.agent_directory}}/utils/gcs.py,sha256=jKblaWOGQ
 src/base_template/{{cookiecutter.agent_directory}}/utils/tracing.py,sha256=2rv1Ukh2jTBENDwoghCItJ28l-Sjz9gMlzdojlVgJa4,6052
 src/base_template/{{cookiecutter.agent_directory}}/utils/typing.py,sha256=DP5OZC3IGvqA1XbvWt8kI3gyAK3ZjzUSL5Ca17wNeLI,4249
 src/cli/main.py,sha256=Dya7Sw3ozMTaGDcwMh_-W7udkGZHGzgAj8aBdSZaZxI,1832
-src/cli/commands/create.py,sha256=R1KKf419Wd2QMi98LrVgm68OyaHKFd0rtm5P3_SIZhI,46779
+src/cli/commands/create.py,sha256=ceUJyzdUN-h-0-7jw6gSSgQWFNXmcsDS47C7Lt8yEew,47385
 src/cli/commands/enhance.py,sha256=AF58CFikNLPib8PULKkzM6r9JGXkTvRWspBaEvGnR3k,24237
 src/cli/commands/list.py,sha256=ZGol9eYB9Yon7JysMUCtpEOwdXzrApdTHzErx6KvT04,6856
 src/cli/commands/setup_cicd.py,sha256=3BtHChOPvp0I-4tlPqwz5hZFm13gtw9gvlWQwF-F7oE,32492
 src/cli/utils/__init__.py,sha256=_cTmsXGPqOtK0q8UW5164QTltbJRJFR_Efxq_BRL1-o,1311
 src/cli/utils/cicd.py,sha256=l9hPdqFXo5SxdtOG1yEoQdmRff5vNcqf0sqxBTtPQig,30704
 src/cli/utils/datastores.py,sha256=gv1V6eDcOEKx4MRNG5C3Y-VfixYq1AzQuaYMLp8QRNo,1058
-src/cli/utils/gcp.py,sha256=_Eq2bsNVS_Npj6ZPzjZA1J6SCu2srGUgsrgu1mIhKh0,9261
+src/cli/utils/gcp.py,sha256=wGgTOc1EWuq7FIHyXa80OUgE1ypGm7zwMF-dvI0jDOE,9261
 src/cli/utils/logging.py,sha256=61ulUY1hUrpxHDFi0szqsjPlrpxdBvwzfYEEV0o97GA,3530
 src/cli/utils/remote_template.py,sha256=ITN1a5GQmVIb1dHpExPLTz1k5c9j7M6fF-HXUDCl6i8,23967
 src/cli/utils/template.py,sha256=YIJu0xZJTyP0XmyNN08Fj7Wc7Wl9ROT45rcZvsvgLb0,52108
@@ -97,7 +97,8 @@ src/deployment_targets/agent_engine/tests/integration/test_agent_engine_app.py,s
 src/deployment_targets/agent_engine/tests/load_test/README.md,sha256=aQP7nDAqd2jKBET4noI39HC9PVHpgpned7cxucWpVJo,1302
 src/deployment_targets/agent_engine/tests/load_test/load_test.py,sha256=VbQjm8uR8JsdJe3upO1eUIz5SZR7IQsHYyJy88cAjy0,4516
 src/deployment_targets/agent_engine/tests/load_test/.results/.placeholder,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/deployment_targets/agent_engine/{{cookiecutter.agent_directory}}/agent_engine_app.py,sha256=G2X3hY5i1MqFxL6rQpxITZYaLT_mGHIfVh_FiKiziFY,12718
+src/deployment_targets/agent_engine/{{cookiecutter.agent_directory}}/agent_engine_app.py,sha256=a5QPypiAn_yJxTAEJLQwpbqJRzhP3vk21IoeIZvC0VI,12437
+src/deployment_targets/agent_engine/{{cookiecutter.agent_directory}}/utils/deployment.py,sha256=AY7awaGuFjri4Ha6M2IrPwpl1b1V7RTlJ_4uZ_zpfTo,2833
 src/deployment_targets/cloud_run/Dockerfile,sha256=rSpmFK7uJhZYwYMtxH8W7mywcCPvaoFn7gL_mXRYuF8,1449
 src/deployment_targets/cloud_run/deployment/terraform/service.tf,sha256=09F6jkkMtoQlbQNdzJl8OZs-ZlfmzZmgCaUFvniJnlQ,11146
 src/deployment_targets/cloud_run/deployment/terraform/dev/service.tf,sha256=l6fG-kMyYOOrRwnsUys_Vq46-76bMiRyXzHYkVD2XFM,6993
@@ -156,20 +157,20 @@ src/resources/docs/adk-cheatsheet.md,sha256=9VZpM5bG3pJq4PbNfym7LOLzgeLZfqkAGA0Y
 src/resources/idx/idx-template.json,sha256=07OQZCPp45Iqor2O7Tm1e1Gmsdd-AmmUofSvA7y-oRs,793
 src/resources/idx/idx-template.nix,sha256=sesHGev_PYtVDg0J5tHkg0OO7IR1Bz2iAtl_if3Ar3M,892
 src/resources/idx/.idx/dev.nix,sha256=XQ4T1xt5I-8SVwnI37v56RFFlxa58Ko-a9G2-oelYAI,1602
-src/resources/locks/uv-adk_base-agent_engine.lock,sha256=qFMwabOGCC6xVimE1UxtFuSO-fDyG6Bb1tGFSHTve0k,560006
-src/resources/locks/uv-adk_base-cloud_run.lock,sha256=QaNqo7hkJtAI1fEQe3px8kPadYCAM6PTV2w0GPgCtj8,740330
-src/resources/locks/uv-agentic_rag-agent_engine.lock,sha256=hysVuh-LMUV0zeEjOy4qN2pwxj6y2TvmudEcLe02hJI,629087
-src/resources/locks/uv-agentic_rag-cloud_run.lock,sha256=dK3burkuPD-WbUQ-eb6AVnodYdtIHeEZtettbm5KsXc,833605
-src/resources/locks/uv-crewai_coding_crew-agent_engine.lock,sha256=GWlR1nTnoFnM3djQ2HJBfPpZVINX8mrM3-sBglONZHo,932027
-src/resources/locks/uv-crewai_coding_crew-cloud_run.lock,sha256=ZmU5X8qsSFRdIMS1Wjq-QL48EqtEehUhS1RdNINZ4iw,1226879
-src/resources/locks/uv-langgraph_base_react-agent_engine.lock,sha256=PPQjq6WOwuKDYo2TtGfFr9HmrOkAXEeKzDGnfhZ3QGY,797526
-src/resources/locks/uv-langgraph_base_react-cloud_run.lock,sha256=E3qF-qnLPNfHTH_VajbD1V0nWDB_ePeGUo3fT-J11pY,1055674
-src/resources/locks/uv-live_api-cloud_run.lock,sha256=GDYYBrnGgZWBsCZm8dtP3iBNToyT1Wkn7cLeDg1zdkc,962006
+src/resources/locks/uv-adk_base-agent_engine.lock,sha256=3vxRrbyJ0ySW_VqjCL3RLFKAuZpYw4CYcL2CjgJWEVI,567595
+src/resources/locks/uv-adk_base-cloud_run.lock,sha256=2A1nvhxumnphEbs3jhbPPc3yqU7XJn7g01pEygsZQVQ,752779
+src/resources/locks/uv-agentic_rag-agent_engine.lock,sha256=j6GtLqMEjBPrMHLbYBP7Tx8u92Ubq8wm-QfvbvpVnOg,636673
+src/resources/locks/uv-agentic_rag-cloud_run.lock,sha256=NulH9sYfEV3hftBTkPFh0uSr-3Zwl4ojE0SeXiT6_KI,846052
+src/resources/locks/uv-crewai_coding_crew-agent_engine.lock,sha256=fcwOoEXj0AGZxYvI_KFbO9t3Opq6KnaLIn3eCRASKB0,934190
+src/resources/locks/uv-crewai_coding_crew-cloud_run.lock,sha256=iQiheltFkHQtF9tq992RiQyQKUx1hlaYs_qKMckuous,1233908
+src/resources/locks/uv-langgraph_base_react-agent_engine.lock,sha256=qcsSMY-_lHe3Fy2iY9RwJd2lGXBVhx_lJItDZjh2xy0,800009
+src/resources/locks/uv-langgraph_base_react-cloud_run.lock,sha256=WNSVBOeNpPi96j-1D6ONx7fLZaZOpHMsqwkVDTO_lOE,1063023
+src/resources/locks/uv-live_api-cloud_run.lock,sha256=jrcrlnpJfJ6MIyEz1Z7qS_nxy0F035M-6C0D80VWCIo,969381
 src/utils/generate_locks.py,sha256=6V1B8V2BEuevWnXUsxZVTrLjXwFRII8UfsIGrQqZxVs,4320
 src/utils/lock_utils.py,sha256=IFOMUWtb-ypm2Y8w8J5y2oI_-MaPuwPF_JOAAlnNudA,2275
 src/utils/watch_and_rebuild.py,sha256=vP4yIiA7E_lj5sfQdJUl8TXas6V7msDg8XWUutAC05Q,6679
-agent_starter_pack-0.15.1.dist-info/METADATA,sha256=m6CTm9YwO8u0zEvtnKnHVgjC4q66C-0okiDzskp1SmI,11482
-agent_starter_pack-0.15.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.15.1.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.15.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.15.1.dist-info/RECORD,,
+agent_starter_pack-0.15.3.dist-info/METADATA,sha256=q2e5-mEcZTW81pjmENcnXyTmECp7TxBwMHsqCOrJ8nc,11482
+agent_starter_pack-0.15.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.15.3.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.15.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.15.3.dist-info/RECORD,,

agents/adk_base/.template/templateconfig.yaml

@@ -18,7 +18,7 @@ settings:
   requires_data_ingestion: false
   requires_session: true
   deployment_targets: ["agent_engine", "cloud_run"]
-  extra_dependencies: ["google-adk~=1.8.0"]
+  extra_dependencies: ["google-adk~=1.14.0"]
   tags: ["adk"]
   frontend_type: "None"
     

agents/adk_base/notebooks/adk_app_testing.ipynb

@@ -4,15 +4,12 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# ADK Application Testing\n",
+    "# 🧪 ADK Application Testing\n",
     "\n",
     "This notebook demonstrates how to test an ADK (Agent Development Kit) application.\n",
     "It covers both local and remote testing, both with Agent Engine and Cloud Run.\n",
     "\n",
-    "> **Note**: This notebook assumes that the agent files are stored in the `app` folder. If your agent files are located in a different directory, please update all relevant file paths and references accordingly.\n",
-    "\n",
-    "<img src=\"https://github.com/GoogleCloudPlatform/agent-starter-pack/blob/main/docs/images/adk_logo.png?raw=true\" width=\"400\">\n",
-    "\n"
+    "> **Note**: This notebook assumes that the agent files are stored in the `app` folder. If your agent files are located in a different directory, please update all relevant file paths and references accordingly."
    ]
   },
   {
@@ -92,12 +89,21 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Replace PROJECT_ID and ENGINE_ID with your values\n",
-    "AGENT_ENGINE_ID = \"projects/PROJECT_ID/locations/us-central1/reasoningEngines/ENGINE_ID\"\n",
+    "# Set to None to auto-detect from ./deployment_metadata.json, or specify manually\n",
+    "# \"projects/PROJECT_ID/locations/us-central1/reasoningEngines/ENGINE_ID\"\n",
+    "REASONING_ENGINE_ID = None\n",
+    "\n",
+    "if REASONING_ENGINE_ID is None:\n",
+    "    try:\n",
+    "        with open(\"../deployment_metadata.json\") as f:\n",
+    "            metadata = json.load(f)\n",
+    "            REASONING_ENGINE_ID = metadata.get(\"remote_agent_engine_id\")\n",
+    "    except (FileNotFoundError, json.JSONDecodeError):\n",
+    "        pass\n",
     "\n",
+    "print(f\"Using REASONING_ENGINE_ID: {REASONING_ENGINE_ID}\")\n",
     "# Get the existing agent engine\n",
-    "remote_agent_engine = client.agent_engines.get(name=AGENT_ENGINE_ID)\n",
-    "print(f\"Retrieved agent engine: {remote_agent_engine.api_resource.name}\")"
+    "remote_agent_engine = client.agent_engines.get(name=REASONING_ENGINE_ID)"
    ]
   },
   {
@@ -354,7 +360,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "myagent-1757939450",
+   "display_name": "myagent-1758791726",
    "language": "python",
    "name": "python3"
   },

agents/agentic_rag/.template/templateconfig.yaml

@@ -19,7 +19,7 @@ settings:
   requires_session: true
   deployment_targets: ["agent_engine", "cloud_run"]
   extra_dependencies: [
-    "google-adk~=1.8.0",
+    "google-adk~=1.14.0",
     "langchain-google-vertexai~=2.0.7",
     "langchain~=0.3.24",
     "langchain-core~=0.3.55",

agents/agentic_rag/notebooks/adk_app_testing.ipynb

@@ -4,15 +4,12 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# ADK Application Testing\n",
+    "# 🧪 ADK Application Testing\n",
     "\n",
     "This notebook demonstrates how to test an ADK (Agent Development Kit) application.\n",
     "It covers both local and remote testing, both with Agent Engine and Cloud Run.\n",
     "\n",
-    "> **Note**: This notebook assumes that the agent files are stored in the `app` folder. If your agent files are located in a different directory, please update all relevant file paths and references accordingly.\n",
-    "\n",
-    "<img src=\"https://github.com/GoogleCloudPlatform/agent-starter-pack/blob/main/docs/images/adk_logo.png?raw=true\" width=\"400\">\n",
-    "\n"
+    "> **Note**: This notebook assumes that the agent files are stored in the `app` folder. If your agent files are located in a different directory, please update all relevant file paths and references accordingly."
    ]
   },
   {
@@ -92,12 +89,21 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Replace PROJECT_ID and ENGINE_ID with your values\n",
-    "AGENT_ENGINE_ID = \"projects/PROJECT_ID/locations/us-central1/reasoningEngines/ENGINE_ID\"\n",
+    "# Set to None to auto-detect from ./deployment_metadata.json, or specify manually\n",
+    "# \"projects/PROJECT_ID/locations/us-central1/reasoningEngines/ENGINE_ID\"\n",
+    "REASONING_ENGINE_ID = None\n",
+    "\n",
+    "if REASONING_ENGINE_ID is None:\n",
+    "    try:\n",
+    "        with open(\"../deployment_metadata.json\") as f:\n",
+    "            metadata = json.load(f)\n",
+    "            REASONING_ENGINE_ID = metadata.get(\"remote_agent_engine_id\")\n",
+    "    except (FileNotFoundError, json.JSONDecodeError):\n",
+    "        pass\n",
     "\n",
+    "print(f\"Using REASONING_ENGINE_ID: {REASONING_ENGINE_ID}\")\n",
     "# Get the existing agent engine\n",
-    "remote_agent_engine = client.agent_engines.get(name=AGENT_ENGINE_ID)\n",
-    "print(f\"Retrieved agent engine: {remote_agent_engine.api_resource.name}\")"
+    "remote_agent_engine = client.agent_engines.get(name=REASONING_ENGINE_ID)"
    ]
   },
   {
@@ -354,7 +360,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "myagent-1757939450",
+   "display_name": "myagent-1758791726",
    "language": "python",
    "name": "python3"
   },

agents/live_api/.template/templateconfig.yaml

@@ -19,6 +19,6 @@ settings:
   deployment_targets: ["cloud_run"]
   extra_dependencies: [
     "backoff~=2.2.1",
-    "google-genai~=1.16.0",
+    "google-genai>=1.16.0",
   ]
 example_question: "What's the weather in San Francisco?"

src/base_template/pyproject.toml

@@ -16,12 +16,12 @@ dependencies = [
 {%- endif %}
     "google-cloud-logging>=3.12.0",
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
-    "google-cloud-aiplatform[evaluation]~=1.113.0",
+    "google-cloud-aiplatform[evaluation]~=1.116.0",
     "fastapi~=0.115.8",
     "uvicorn~=0.34.0",
     "psycopg2-binary>=2.9.10",
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
-    "google-cloud-aiplatform[evaluation,agent-engines]~=1.113.0"
+    "google-cloud-aiplatform[evaluation,agent-engines]~=1.116.0"
 {%- endif %}
 ]
 {% if cookiecutter.deployment_target == 'cloud_run' %}

src/cli/commands/create.py

@@ -630,12 +630,13 @@ def create(
         # CI/CD runner selection
         final_cicd_runner = cicd_runner
         if not final_cicd_runner:
-            if auto_approve:
+            if auto_approve or agent_garden:
                 final_cicd_runner = "google_cloud_build"
-                console.print(
-                    "Info: --cicd-runner not specified. Defaulting to 'google_cloud_build' in auto-approve mode.",
-                    style="yellow",
-                )
+                if not agent_garden:
+                    console.print(
+                        "Info: --cicd-runner not specified. Defaulting to 'google_cloud_build' in auto-approve mode.",
+                        style="yellow",
+                    )
             else:
                 final_cicd_runner = prompt_cicd_runner_selection()
         if debug:
@@ -646,7 +647,13 @@ def create(
             not auto_approve
             and ctx.get_parameter_source("region") != ParameterSource.COMMANDLINE
         ):
-            region = prompt_region_confirmation(region)
+            # Show Agent Engine supported regions link if agent_garden flag is set
+            if agent_garden:
+                console.print(
+                    "\n📍 [blue]Agent Engine Supported Regions:[/blue]\n"
+                    "   [cyan]https://cloud.google.com/vertex-ai/generative-ai/docs/agent-engine/overview#supported-regions[/cyan]"
+                )
+            region = prompt_region_confirmation(region, agent_garden=agent_garden)
         if debug:
             logging.debug(f"Selected region: {region}")
 
@@ -777,7 +784,9 @@ def create(
         raise
 
 
-def prompt_region_confirmation(default_region: str = "us-central1") -> str:
+def prompt_region_confirmation(
+    default_region: str = "us-central1", agent_garden: bool = False
+) -> str:
     """Prompt user to confirm or change the default region."""
     new_region = Prompt.ask(
         "\nEnter desired GCP region (Gemini uses global endpoint by default)",
@@ -965,7 +974,8 @@ def setup_gcp_environment(
         logging.debug("Verifying GCP credentials...")
     creds_info = verify_credentials()
     # Handle credential verification and project selection
-    if not auto_approve:
+    # Skip interactive prompts if auto_approve or agent_garden is set
+    if not auto_approve and not agent_garden:
         creds_info = _handle_credential_verification(creds_info)
         # If user chose to skip verification, don't test Vertex AI connection
         if creds_info.get("skip_vertex_test", False):
@@ -976,7 +986,7 @@ def setup_gcp_environment(
                 creds_info["project"], region, agent_garden=agent_garden
             )
     else:
-        # Even with auto_approve, we should still set the GCP project
+        # Even with auto_approve or agent_garden, we should still set the GCP project
         set_gcp_project(creds_info["project"], set_quota_project=True)
         # Test Vertex AI connection
         _test_vertex_ai_connection(

src/cli/utils/gcp.py

@@ -147,7 +147,7 @@ def get_dummy_request(project_id: str) -> CountTokensRequest:
     """Creates a simple test request for Gemini."""
     return CountTokensRequest(
         contents=[{"role": "user", "parts": [{"text": "Hi"}]}],
-        endpoint=f"projects/{project_id}/locations/global/publishers/google/models/gemini-2.0-flash",
+        endpoint=f"projects/{project_id}/locations/global/publishers/google/models/gemini-2.5-flash",
     )
 
 

src/deployment_targets/agent_engine/{{cookiecutter.agent_directory}}/agent_engine_app.py

@@ -14,12 +14,11 @@
 
 # mypy: disable-error-code="attr-defined,arg-type"
 {%- if "adk" in cookiecutter.tags %}
-import datetime
-import json
 import logging
 import os
 from typing import Any
 
+import click
 import google.auth
 import vertexai
 from google.adk.artifacts import GcsArtifactService
@@ -30,6 +29,11 @@ from vertexai._genai.types import AgentEngine, AgentEngineConfig
 from vertexai.agent_engines.templates.adk import AdkApp
 
 from {{cookiecutter.agent_directory}}.agent import root_agent
+from {{cookiecutter.agent_directory}}.utils.deployment import (
+    parse_env_vars,
+    print_deployment_success,
+    write_deployment_metadata,
+)
 from {{cookiecutter.agent_directory}}.utils.gcs import create_bucket_if_not_exists
 from {{cookiecutter.agent_directory}}.utils.tracing import CloudTraceLoggingSpanExporter
 from {{cookiecutter.agent_directory}}.utils.typing import Feedback
@@ -68,8 +72,6 @@ class AgentEngineApp(AdkApp):
         return operations
 
 {%- else %}
-import datetime
-import json
 import logging
 import os
 from collections.abc import Iterable, Mapping
@@ -77,6 +79,7 @@ from typing import (
     Any,
 )
 
+import click
 import google.auth
 import vertexai
 from google.cloud import logging as google_cloud_logging
@@ -84,6 +87,11 @@ from langchain_core.runnables import RunnableConfig
 from traceloop.sdk import Instruments, Traceloop
 from vertexai._genai.types import AgentEngine, AgentEngineConfig
 
+from {{cookiecutter.agent_directory}}.utils.deployment import (
+    parse_env_vars,
+    print_deployment_success,
+    write_deployment_metadata,
+)
 from {{cookiecutter.agent_directory}}.utils.gcs import create_bucket_if_not_exists
 from {{cookiecutter.agent_directory}}.utils.tracing import CloudTraceLoggingSpanExporter
 from {{cookiecutter.agent_directory}}.utils.typing import Feedback, InputChat, dumpd, ensure_valid_config
@@ -190,17 +198,69 @@ class AgentEngineApp:
 {%- endif %}
 
 
+@click.command()
+@click.option(
+    "--project",
+    default=None,
+    help="GCP project ID (defaults to application default credentials)",
+)
+@click.option(
+    "--location",
+    default="us-central1",
+    help="GCP region (defaults to us-central1)",
+)
+@click.option(
+    "--agent-name",
+    default="{{cookiecutter.project_name}}",
+    help="Name for the agent engine",
+)
+@click.option(
+    "--requirements-file",
+    default=".requirements.txt",
+    help="Path to requirements.txt file",
+)
+@click.option(
+    "--extra-packages",
+    multiple=True,
+    default=["./{{cookiecutter.agent_directory}}"],
+    help="Additional packages to include",
+)
+@click.option(
+    "--set-env-vars",
+    default=None,
+    help="Comma-separated list of environment variables in KEY=VALUE format",
+)
+@click.option(
+    "--service-account",
+    default=None,
+    help="Service account email to use for the agent engine",
+)
 def deploy_agent_engine_app(
-    project: str,
+    project: str | None,
     location: str,
-    agent_name: str | None = None,
-    requirements_file: str = ".requirements.txt",
-    extra_packages: list[str] = ["./{{cookiecutter.agent_directory}}"],
-    env_vars: dict[str, str] = {},
-    service_account: str | None = None,
+    agent_name: str,
+    requirements_file: str,
+    extra_packages: tuple[str, ...],
+    set_env_vars: str | None,
+    service_account: str | None,
 ) -> AgentEngine:
     """Deploy the agent engine app to Vertex AI."""
+    # Parse environment variables if provided
+    env_vars = parse_env_vars(set_env_vars)
+
+    if not project:
+        _, project = google.auth.default()
+
+    print("""
+    ╔═══════════════════════════════════════════════════════════╗
+    ║                                                           ║
+    ║   🤖 DEPLOYING AGENT TO VERTEX AI AGENT ENGINE 🤖         ║
+    ║                                                           ║
+    ╚═══════════════════════════════════════════════════════════╝
+    """)
+
     logging.basicConfig(level=logging.INFO)
+    extra_packages_list = list(extra_packages)
     staging_bucket_uri = f"gs://{project}-agent-engine"
 {%- if "adk" in cookiecutter.tags %}
     artifacts_bucket_name = f"{project}-{{cookiecutter.project_name}}-logs-data"
@@ -236,14 +296,24 @@ def deploy_agent_engine_app(
     env_vars["NUM_WORKERS"] = "1"
 
     # Common configuration for both create and update operations
+    labels: dict[str, str] = {}
+{%- if cookiecutter.agent_garden %}
+    labels["deployed-with"] = "agent-garden"
+{%- if cookiecutter.agent_sample_id and cookiecutter.agent_sample_publisher %}
+    labels["vertex-agent-sample-id"] = "{{cookiecutter.agent_sample_id}}"
+    labels["vertex-agent-sample-publisher"] = "{{cookiecutter.agent_sample_publisher}}"
+{%- endif %}
+{%- endif %}
+
     config = AgentEngineConfig(
         display_name=agent_name,
         description="{{cookiecutter.agent_description}}",
-        extra_packages=extra_packages,
+        extra_packages=extra_packages_list,
         env_vars=env_vars,
         service_account=service_account,
         requirements=requirements,
         staging_bucket=staging_bucket_uri,
+        labels=labels,
     )
 
     agent_config = {
@@ -262,94 +332,20 @@ def deploy_agent_engine_app(
 
     if matching_agents:
         # Update the existing agent with new configuration
-        logging.info(f"Updating existing agent: {agent_name}")
+        logging.info(f"\n📝 Updating existing agent: {agent_name}")
         remote_agent = client.agent_engines.update(
             name=matching_agents[0].api_resource.name, **agent_config
         )
     else:
         # Create a new agent if none exists
-        logging.info(f"Creating new agent: {agent_name}")
+        logging.info(f"\n🚀 Creating new agent: {agent_name}")
         remote_agent = client.agent_engines.create(**agent_config)
 
-    metadata = {
-        "remote_agent_engine_id": remote_agent.api_resource.name,
-        "deployment_timestamp": datetime.datetime.now().isoformat(),
-    }
-    metadata_file = "deployment_metadata.json"
-
-    with open(metadata_file, "w") as f:
-        json.dump(metadata, f, indent=2)
-
-    logging.info(f"Agent Engine ID written to {metadata_file}")
+    write_deployment_metadata(remote_agent)
+    print_deployment_success(remote_agent, location, project)
 
     return remote_agent
 
 
 if __name__ == "__main__":
-    import argparse
-
-    parser = argparse.ArgumentParser(description="Deploy agent engine app to Vertex AI")
-    parser.add_argument(
-        "--project",
-        default=None,
-        help="GCP project ID (defaults to application default credentials)",
-    )
-    parser.add_argument(
-        "--location",
-        default="us-central1",
-        help="GCP region (defaults to us-central1)",
-    )
-    parser.add_argument(
-        "--agent-name",
-        default="{{cookiecutter.project_name}}",
-        help="Name for the agent engine",
-    )
-    parser.add_argument(
-        "--requirements-file",
-        default=".requirements.txt",
-        help="Path to requirements.txt file",
-    )
-    parser.add_argument(
-        "--extra-packages",
-        nargs="+",
-        default=["./{{cookiecutter.agent_directory}}"],
-        help="Additional packages to include",
-    )
-    parser.add_argument(
-        "--set-env-vars",
-        help="Comma-separated list of environment variables in KEY=VALUE format",
-    )
-    parser.add_argument(
-        "--service-account",
-        default=None,
-        help="Service account email to use for the agent engine",
-    )
-    args = parser.parse_args()
-
-    # Parse environment variables if provided
-    env_vars = {}
-    if args.set_env_vars:
-        for pair in args.set_env_vars.split(","):
-            key, value = pair.split("=", 1)
-            env_vars[key] = value
-
-    if not args.project:
-        _, args.project = google.auth.default()
-
-    print("""
-    ╔═══════════════════════════════════════════════════════════╗
-    ║                                                           ║
-    ║   🤖 DEPLOYING AGENT TO VERTEX AI AGENT ENGINE 🤖         ║
-    ║                                                           ║
-    ╚═══════════════════════════════════════════════════════════╝
-    """)
-
-    deploy_agent_engine_app(
-        project=args.project,
-        location=args.location,
-        agent_name=args.agent_name,
-        requirements_file=args.requirements_file,
-        extra_packages=args.extra_packages,
-        env_vars=env_vars,
-        service_account=args.service_account,
-    )
+    deploy_agent_engine_app()