agent-starter-pack 0.0.13b0__py3-none-any.whl → 0.1.0__py3-none-any.whl

{agent_starter_pack-0.0.13b0.dist-info → agent_starter_pack-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.0.13b0
+Version: 0.1.0
 Summary: CLI tool to create GCP-based AI agent projects from templates
 Author-email: Google LLC <agent-starter-pack@google.com>
 License: Apache-2.0
@@ -51,7 +51,7 @@ pip install agent-starter-pack
 agent-starter-pack create my-awesome-agent
 ```
 
-**That's it!** You now have a fully functional agent project, complete with a backend, frontend, and deployment infrastructure, ready to explore and customize.
+**That's it!** You now have a fully functional agent project—complete with backend, frontend, and deployment infrastructure—ready for you to explore and customize.
 For more installation options, see the [Installation Guide](docs/installation.md).
 
 ---
@@ -119,7 +119,6 @@ See the [documentation](docs/) for more details:
 - [Data Ingestion](docs/data-ingestion.md)
 - [Observability](docs/observability.md)
 - [CLI Reference](docs/cli/README.md)
-- [CLI Reference](docs/cli/README.md)
 - [Troubleshooting](docs/troubleshooting.md)
 
 **Video Walkthrough:** A [20-Minute Video Walkthrough](https://youtu.be/kwRG7cnqSu0) is available, showcasing the previous `e2e-gen-ai-app-starter-pack`. A new walkthrough for the `agent-starter-pack` will be released soon!
@@ -138,7 +137,7 @@ If you encounter any issues or have specific suggestions, please first consider
 
 ### Share Your Experience
 
-For other types of feedback, or if you'd like to share a positive experience or success story using this starter pack, we'd love to hear from you! You can reach out to us at <a href="mailto:e2e-gen-ai-app-starter-pack@google.com">e2e-gen-ai-app-starter-pack@google.com</a>.
+For other types of feedback, or if you'd like to share a positive experience or success story using this starter pack, we'd love to hear from you! You can reach out to us at <a href="mailto:agent-starter-pack@google.com">agent-starter-pack@google.com</a>.
 
 Thank you for your contributions!
 

{agent_starter_pack-0.0.13b0.dist-info → agent_starter_pack-0.1.0.dist-info}/RECORD

@@ -21,7 +21,7 @@ agents/langgraph_base_react/template/.templateconfig.yaml,sha256=NpQRZbXtCOeX0Xi
 agents/langgraph_base_react/tests/integration/test_agent.py,sha256=wg2Y8V4l7dh7rhWy7Z8zioDW5ILRsz7RZJ-mz0AhK3U,1619
 agents/multimodal_live_api/README.md,sha256=t19QIXtYX8OuHy4po8dD875hDp0E5kRJOxaxGJ-i2zQ,2741
 agents/multimodal_live_api/app/agent.py,sha256=WFB7lvDiImvNyWFlds_f5TtzyrbwcquJe6JyJo8HGSs,2690
-agents/multimodal_live_api/app/server.py,sha256=RelFBb2iPzAUaI4huM7_5yRCXwNCtpoiT6AfYlQZvqI,7016
+agents/multimodal_live_api/app/server.py,sha256=LdJcBRcUEqjAPk2pwsxwLx8wMULUzvMNDYZf_dx4yl0,7063
 agents/multimodal_live_api/app/templates.py,sha256=HgA4QGIle4Q-Nr6xAAXk1pTX7eNbZqlQUD9_fbCvDWA,3301
 agents/multimodal_live_api/app/vector_store.py,sha256=AYRSNpfevbVTt-nnWna_dp2pn4T7o1nz9-9wOoWaxsA,2045
 agents/multimodal_live_api/template/.templateconfig.yaml,sha256=gkXZxV15yY6OpJ64VWRhWlCtPhJxyacrv_e2kY3-Wo0,1082
@@ -30,16 +30,16 @@ agents/multimodal_live_api/tests/load_test/load_test.py,sha256=HHZyfC4gqiQtZVF_C
 agents/multimodal_live_api/tests/unit/test_server.py,sha256=_TjlgQgNkjerIaBGnu8P8_KB8ZlSolDcivALpUOn_Rw,4786
 src/base_template/.gitignore,sha256=mJKTZIcVdAFiIUQicRfPNGUg6WvwcfTEC2xbmAaU34g,2579
 src/base_template/Makefile,sha256=PNmyhdIRf0p6eRLYqWHfCskLmuDMG673YPEeu51XLVQ,1720
-src/base_template/README.md,sha256=7HDBqs6gvoUJNagg48_apRuDkrgb0_uRkAei2OTR_II,8094
+src/base_template/README.md,sha256=A2AsGhPVfbMTjiPnuuEui7V8UVWa7hE5m0zut--QbbU,8196
 src/base_template/pyproject.toml,sha256=Pv4nC4uh_9_sGA90Tom0ITJUAbGolhseCA9-AOaEIpY,2799
 src/base_template/app/utils/tracing.py,sha256=ovp6tgwgo_tqibpbhhHl8cWIqqUWVYfV0AL7XPDEPpM,5661
 src/base_template/app/utils/typing.py,sha256=UUFm74qiKGtT8pLuE_3yRpcmntRCt91wH_Q-LKdjQuQ,3194
 src/base_template/deployment/README.md,sha256=8AorSeF2Aj0kvPmuQLFrmAAUWVbdTzg0S-rVBbfdirQ,4835
-src/base_template/deployment/cd/deploy-to-prod.yaml,sha256=TbIJc0YTPwH5lQUERiDlC98muaw6e0QjYOfjNtCx9os,3308
-src/base_template/deployment/cd/staging.yaml,sha256=9DtPZ2A0yiPXA07XbBa-yurT9os5OF28eEU-YXlmTk4,7121
-src/base_template/deployment/ci/pr_checks.yaml,sha256=6eNl5O-9zrbAcYGxWl_4-8EPKb7xxUiWHxC4npNEEvc,1442
+src/base_template/deployment/cd/deploy-to-prod.yaml,sha256=sy9IkWsQ9Pk_qBfiamsFuYpc0SUQpWEkII8JVqOehNc,3323
+src/base_template/deployment/cd/staging.yaml,sha256=BuK-Z08yEILfKUFlZQDSBx-hXmCH1m9h4A4KL4uOsRw,7141
+src/base_template/deployment/ci/pr_checks.yaml,sha256=FGa2ab43Y1KgYkxK7XovyQrQHY5ypeND5VewZ4lpBw0,1457
 src/base_template/deployment/terraform/apis.tf,sha256=NoQP9HUJ5eXJMSKerdhypFYSEpWgLLhKK6Bz8hJgPTo,1221
-src/base_template/deployment/terraform/build_triggers.tf,sha256=Gd51HQ-b2IGupfZmsMlm3M_c8C4pI-o_IBh1OkQsanw,5105
+src/base_template/deployment/terraform/build_triggers.tf,sha256=P-AbWegeDXoFkBeHSn0bBb-4MP5pQ15vc0b5DH-vcDc,5097
 src/base_template/deployment/terraform/iam.tf,sha256=6lsvBGLl0JRxR3pZMS38yw-pq6MNA0mxK54s19qr5-M,5600
 src/base_template/deployment/terraform/locals.tf,sha256=mrmOigExLk5g734-2VodDj8qQUTBq20e7FgQD0KUF8E,1390
 src/base_template/deployment/terraform/log_sinks.tf,sha256=pxZcQ-FyT7RVuGDORB_023SoGj96iqWtmh5kgJMG1u4,3011
@@ -60,7 +60,7 @@ src/cli/main.py,sha256=vNEV1Fdl1BhJAFbKHcN3MN1v7dPRUcAgXMnB3uCnlI0,1591
 src/cli/commands/create.py,sha256=38wc5DjUOelLtmCnkVLIq9aVc6c1EK10O0fd-refh6I,18963
 src/cli/commands/setup_cicd.py,sha256=EgVkDdA94npDHHPTuYDmWNIiRscnsXtN7CbTuelrTVU,28347
 src/cli/utils/__init__.py,sha256=x-k6jfcv68_zYS88LrA1mr0yB5y5SekJ0nc5Tvt6iDA,1064
-src/cli/utils/cicd.py,sha256=W44Dt39mZMzfDqA16gG69C8vc6VvBy99QVBJxEIjMI4,26631
+src/cli/utils/cicd.py,sha256=nrb46lJ3bR_yVJRlUPsBN-To4v-xo5qO5WmnIi6qcYM,27076
 src/cli/utils/gcp.py,sha256=f-XTq9AN2_KrCR8pf_gSPOng8KrvtFlNP2Ok9VtTk_Q,4118
 src/cli/utils/logging.py,sha256=0lHe4EPi1A8sOx9xkA7gS4UNl0GsIyp2ahydkkuCzLY,1570
 src/cli/utils/template.py,sha256=p7uKv5yHkXIOWo8EJJ0IjUh57lOSDL7kYHA9OZ7j-z4,25447
@@ -72,7 +72,7 @@ src/data_ingestion/data_ingestion_pipeline/submit_pipeline.py,sha256=i5TI-NMGaMz
 src/data_ingestion/data_ingestion_pipeline/components/ingest_data.py,sha256=EfAUQh4WEkc-hfZ0E0_mb5F-WkZWw4RXdgD6IByvslU,5969
 src/data_ingestion/data_ingestion_pipeline/components/process_data.py,sha256=TQEQ6obbERs-dhsTLn-GA7M1zEjitqg-MiBR80yzUO4,11770
 src/deployment_targets/agent_engine/deployment_metadata.json,sha256=G_t_n-UNrFsBgH1Xrw5-ZqYzUGwZ4X6ipsIm_yiSq-w,72
-src/deployment_targets/agent_engine/app/agent_engine_app.py,sha256=k1dIgjpcAmkAM0xD5IRVYFyWLf9UXSj-lwzYlSgwZyY,8699
+src/deployment_targets/agent_engine/app/agent_engine_app.py,sha256=Bf54uYOe9x7V_1Oe12k0eEvUbRDpI5ke1XrorulcG0A,9473
 src/deployment_targets/agent_engine/app/utils/gcs.py,sha256=voQNs8sbvLDH0PX3avpn8RFNTq7QWNgXfMWUKU7hBTA,1495
 src/deployment_targets/agent_engine/notebooks/intro_reasoning_engine.ipynb,sha256=DUlLvBjaPxHXNAxMw8pkHMWCDuyPleiAsiiyRJ-D1YA,929968
 src/deployment_targets/agent_engine/tests/integration/test_agent_engine_app.py,sha256=maYeJgeZceMgD4TI8ZID6SKAqHPlQuW2CksT7MGile4,3938
@@ -86,7 +86,7 @@ src/deployment_targets/cloud_run/app/server.py,sha256=h6H6xwK_VVlOOfD6sOSbxX3WqR
 src/deployment_targets/cloud_run/deployment/terraform/artifact_registry.tf,sha256=7KgheT1UuCUUQK2pfb8cmS_H8qftzy7HHKu1HrCS0_0,970
 src/deployment_targets/cloud_run/deployment/terraform/dev/service_accounts.tf,sha256=CBavl84WbuQCGz0bQt5B7Sh9ZlpQVMRNHQZAnyWxppg,820
 src/deployment_targets/cloud_run/tests/integration/test_server_e2e.py,sha256=KKtdtBTxaAfHQodm4FdyQHa7hBAwWCCda5aEF26-bjk,5978
-src/deployment_targets/cloud_run/tests/load_test/README.md,sha256=y-SLVLiYZp4Lt8xI5lhz5svAsIwMK8IWIZMwF27SLro,2663
+src/deployment_targets/cloud_run/tests/load_test/README.md,sha256=JP47idadGBiLaBxBPOBJBEY4W87Xh3-wv5Bn8avhvHU,2804
 src/deployment_targets/cloud_run/tests/load_test/load_test.py,sha256=HsxmcNKyEopeGER4QUmnjiW4Bqp6h1TXsCUkyrq6inw,3317
 src/deployment_targets/cloud_run/tests/load_test/.results/.placeholder,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/deployment_targets/cloud_run/tests/unit/test_server.py,sha256=f3nJIw2JCcxvV2WDwq9oZv_VUVYpTCm6R2kw5ch8BRc,4379
@@ -148,10 +148,10 @@ src/resources/setup_cicd/cicd_variables.tf,sha256=OHNKglkpN2TSTnfgpd-fM-eNUbWhs6
 src/resources/setup_cicd/github.tf,sha256=g7gMQqPZI69XteCSruNfiT00GBGawuXANHj-F0fc-GE,2727
 src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5jB7QBfF0,1122
 src/utils/generate_locks.py,sha256=xu5IAhGGBPkVQGSJX4kk7_JNDwWJUEaXAHbmaQIohbg,4386
-src/utils/lock_utils.py,sha256=plw3vWkRViCMxBK3IrLRHtF_I0kbbP043T9KIYxcb20,2423
+src/utils/lock_utils.py,sha256=_QdzQtgIbCmJ87s046_i1g966slVNmvr3bJDeHbRQSM,2419
 src/utils/watch_and_rebuild.py,sha256=IdbTLReYknTXXu5KWPz3sc7Zg5HTEMS5UVlMSnuhs5E,6067
-agent_starter_pack-0.0.13b0.dist-info/METADATA,sha256=JCEZ64tCOLyVW6BDUo1hLzP3bB0e0RUut2B1-1EWYvE,7379
-agent_starter_pack-0.0.13b0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.0.13b0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.0.13b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.0.13b0.dist-info/RECORD,,
+agent_starter_pack-0.1.0.dist-info/METADATA,sha256=f1w4h-F89pFoC5hN7sxzYRHUbYD6cFEf2Ab-2kXaO_Y,7328
+agent_starter_pack-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.1.0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.1.0.dist-info/RECORD,,

agents/multimodal_live_api/app/server.py

@@ -77,7 +77,9 @@ class GeminiSession:
                 elif "setup" in data:
                     self.run_id = data["setup"]["run_id"]
                     self.user_id = data["setup"]["user_id"]
-                    logger.log_struct({**data['setup'],"type": "setup"}, severity="INFO")
+                    logger.log_struct(
+                        {**data["setup"], "type": "setup"}, severity="INFO"
+                    )
                 else:
                     logging.warning(f"Received unexpected input from client: {data}")
             except ConnectionClosedError as e:

src/base_template/README.md

@@ -2,7 +2,7 @@
 
 {{cookiecutter.agent_description}}
 
-Template generated with `googleCloudPlatform/agent-starter-pack`
+Agent generated with [`googleCloudPlatform/agent-starter-pack`](https://github.com/GoogleCloudPlatform/agent-starter-pack)
 
 ## Project Structure
 
@@ -113,7 +113,9 @@ For full command options and usage, refer to the [Makefile](Makefile).
 3.  **Interact with the Agent**
     Once both the backend and frontend are running, click the play button in the frontend UI to establish a connection with the backend. You can now interact with the Multimodal Live Agent! You can try asking questions such as "Using the tool you have, define Governance in the context MLOPs" to allow the agent to use the documentation it was provided to.
 
-**Cloud Shell usage**
+<details>
+<summary><b>Cloud Shell usage</b></summary>
+
 To use the `multimodal_live_api` agent in Cloud Shell, follow these steps:
 
 1.  **Start the Frontend:**
@@ -153,6 +155,7 @@ To use the `multimodal_live_api` agent in Cloud Shell, follow these steps:
     *   Click the "Play button" to connect. Start interacting with it!
 
 *   When using Cloud Shell there is a known limitation when using the feedback feature in the Frontend. Feedback submission might fail due to different origins between the frontend and backend in the Cloud Shell environment.
+</details>
 {%- else %}
 ## Usage
 

src/base_template/deployment/cd/deploy-to-prod.yaml

@@ -14,7 +14,7 @@
 
 steps:
 {%- if cookiecutter.data_ingestion %}
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: deploy-data-ingestion-pipeline-prod
     entrypoint: bash
     args:
@@ -63,7 +63,7 @@ steps:
       - "--set-env-vars"
       - "COMMIT_SHA=${COMMIT_SHA}{%- if cookiecutter.data_ingestion %},DATA_STORE_ID=${_DATA_STORE_ID},DATA_STORE_REGION=${_DATA_STORE_REGION}{%- endif %}"
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
-  - name: "python:3.11" 
+  - name: "python:3.11-slim" 
     id: install-dependencies
     entrypoint: /bin/bash
     args:
@@ -73,7 +73,7 @@ steps:
     env:
       - 'PATH=/usr/local/bin:/usr/bin:~/.local/bin'
 
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: trigger-deployment
     entrypoint: /bin/bash
     args:

src/base_template/deployment/cd/staging.yaml

@@ -14,7 +14,7 @@
 
 steps:
 {%- if cookiecutter.data_ingestion %}
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: deploy-data-ingestion-pipeline-prod
     entrypoint: bash
     args:
@@ -96,7 +96,7 @@ steps:
       - |
         echo $(gcloud auth print-identity-token -q) > id_token.txt
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
-  - name: "python:3.11" 
+  - name: "python:3.11-slim" 
     id: install-dependencies
     entrypoint: /bin/bash
     args:
@@ -106,7 +106,7 @@ steps:
     env:
       - 'PATH=/usr/local/bin:/usr/bin:~/.local/bin'
 
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: deploy-staging
     entrypoint: /bin/bash
     args:
@@ -131,7 +131,7 @@ steps:
 {%- endif %}
 
   # Load Testing
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: load_test
     entrypoint: /bin/bash
     args:

src/base_template/deployment/ci/pr_checks.yaml

@@ -14,7 +14,7 @@
 
 steps:
   # Install uv package manager and sync dependencies
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: install-dependencies
     entrypoint: /bin/bash
     args:
@@ -25,7 +25,7 @@ steps:
       - 'PATH=/usr/local/bin:/usr/bin:~/.local/bin'
 
   # Run unit tests using pytest
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: unit-tests
     entrypoint: /bin/bash
     args:
@@ -36,7 +36,7 @@ steps:
       - 'PATH=/usr/local/bin:/usr/bin:~/.local/bin'
 
   # Run integration tests
-  - name: "python:3.11"
+  - name: "python:3.11-slim"
     id: integration-tests
     entrypoint: /bin/bash
     args:

src/base_template/deployment/terraform/build_triggers.tf

@@ -33,7 +33,7 @@ resource "google_cloudbuild_trigger" "pr_checks" {
     "data_ingestion/**",
     "tests/**",
     "deployment/**",
-    "poetry.lock",
+    "uv.lock",
   {%- if cookiecutter.data_ingestion %}
     "data_ingestion/**",
   {%- endif %}
@@ -62,7 +62,7 @@ resource "google_cloudbuild_trigger" "cd_pipeline" {
     "data_ingestion/**",
     "tests/**",
     "deployment/**",
-    "poetry.lock"
+    "uv.lock"
   ]
   substitutions = {
     _STAGING_PROJECT_ID            = var.staging_project_id

src/cli/utils/cicd.py

@@ -259,15 +259,30 @@ def create_github_connection(
                     console.print("⏳ Waiting for GitHub authorization...")
                     # Extract and print the action URI for user authentication
                     try:
-                        action_uri = json.loads(result.stdout).get("installationState", {}).get("actionUri")
+                        action_uri = (
+                            json.loads(result.stdout)
+                            .get("installationState", {})
+                            .get("actionUri")
+                        )
                         if action_uri:
-                            console.print("\n🔑 Authentication Required:", style="bold yellow")
-                            console.print(f"Please visit [link={action_uri}]this page[/link] to authenticate Cloud Build with GitHub:")
+                            console.print(
+                                "\n🔑 Authentication Required:", style="bold yellow"
+                            )
+                            console.print(
+                                f"Please visit [link={action_uri}]this page[/link] to authenticate Cloud Build with GitHub:"
+                            )
                             console.print(f"{action_uri}", style="bold blue")
-                            console.print("(Copy and paste the link into your browser if clicking doesn't work)")
-                            console.print("After completing authentication, return here to continue the setup.\n")
+                            console.print(
+                                "(Copy and paste the link into your browser if clicking doesn't work)"
+                            )
+                            console.print(
+                                "After completing authentication, return here to continue the setup.\n"
+                            )
                     except (json.JSONDecodeError, KeyError) as e:
-                        console.print(f"⚠️ Could not extract authentication link: {e}", style="yellow")
+                        console.print(
+                            f"⚠️ Could not extract authentication link: {e}",
+                            style="yellow",
+                        )
                     time.sleep(10)
                 continue
             else:

src/deployment_targets/agent_engine/app/agent_engine_app.py

@@ -32,20 +32,21 @@ from app.utils.gcs import create_bucket_if_not_exists
 from app.utils.tracing import CloudTraceLoggingSpanExporter
 from app.utils.typing import Feedback, InputChat, dumpd, ensure_valid_config
 
-
 logging.basicConfig(
     level=logging.INFO,
 )
 
+
 class AgentEngineApp:
     """Class for managing agent engine functionality."""
 
-    def __init__(self, project_id: str | None = None, env_vars: dict[str, str] | None = None) -> None:
+    def __init__(
+        self, project_id: str | None = None, env_vars: dict[str, str] | None = None
+    ) -> None:
         """Initialize the AgentEngineApp variables"""
         self.project_id = project_id
         self.env_vars = env_vars if env_vars is not None else {}
 
-
     def set_up(self) -> None:
         """The set_up method is used to define application initialization logic"""
         import os
@@ -144,8 +145,8 @@ class AgentEngineApp:
 
 
 def deploy_agent_engine_app(
-    project: str | None = None,
-    location: str | None = None,
+    project: str,
+    location: str,
     agent_name: str | None = None,
     requirements_file: str = ".requirements.txt",
     extra_packages: list[str] = ["./app"],
@@ -169,13 +170,12 @@ def deploy_agent_engine_app(
     # Common configuration for both create and update operations
     agent_config = {
         "reasoning_engine": agent,
-        "requirements": requirements,
         "display_name": agent_name,
         "description": "This is a sample custom application in Reasoning Engine that uses LangGraph",
         "extra_packages": extra_packages,
     }
-
     logging.info(f"Agent config: {agent_config}")
+    agent_config["requirements"] = requirements
 
     # Check if an agent with this name already exists
     existing_agents = reasoning_engines.ReasoningEngine.list(
@@ -207,14 +207,36 @@ def deploy_agent_engine_app(
 
 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=["./app"], 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(
+        "--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=["./app"],
+        help="Additional packages to include",
+    )
+    parser.add_argument(
+        "--set-env-vars",
+        help="Comma-separated list of environment variables in KEY=VALUE format",
+    )
     args = parser.parse_args()
 
     # Parse environment variables if provided
@@ -228,11 +250,19 @@ if __name__ == "__main__":
     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
+        env_vars=env_vars,
     )

src/deployment_targets/cloud_run/tests/load_test/README.md

@@ -11,13 +11,13 @@ Follow these steps to execute load tests on your local machine:
 Launch the FastAPI server in a separate terminal:
 
 ```bash
-poetry run uvicorn app.server:app --host 0.0.0.0 --port 8000 --reload
+uv run uvicorn app.server:app --host 0.0.0.0 --port 8000 --reload
 ```
 
 **2. (In another tab) Create virtual environment with Locust**
 Using another terminal tab, This is suggested to avoid conflicts with the existing application python environment.
 
-```commandline
+```bash
 python3 -m venv locust_env && source locust_env/bin/activate && pip install locust==2.31.1
 ```
 
@@ -67,10 +67,14 @@ export _ID_TOKEN=$(gcloud auth print-identity-token -q)
 ```
 
 **3. Execute the Load Test:**
-The following command executes the same load test parameters as the local test but targets your remote Cloud Run instance.
+Create virtual environment with Locust:
+```bash
+python3 -m venv locust_env && source locust_env/bin/activate && pip install locust==2.31.1
+```
 
+Execute load tests. The following command executes the same load test parameters as the local test but targets your remote Cloud Run instance.
 ```bash
-poetry run locust -f tests/load_test/load_test.py \
+locust -f tests/load_test/load_test.py \
 -H $RUN_SERVICE_URL \
 --headless \
 -t 30s -u 60 -r 2 \

src/utils/lock_utils.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Utilities for managing Poetry lock files and dependencies."""
+"""Utilities for managing uv lock files and dependencies."""
 
 import pathlib
 from pathlib import Path