agent-starter-pack 0.15.6__py3-none-any.whl → 0.16.0__py3-none-any.whl

agents/langgraph_base_react/app/agent.py

@@ -13,61 +13,22 @@
 # limitations under the License.
 
 # mypy: disable-error-code="union-attr"
-from langchain_core.messages import BaseMessage
-from langchain_core.runnables import RunnableConfig
-from langchain_core.tools import tool
 from langchain_google_vertexai import ChatVertexAI
-from langgraph.graph import END, MessagesState, StateGraph
-from langgraph.prebuilt import ToolNode
+from langgraph.prebuilt import create_react_agent
 
 LOCATION = "global"
 LLM = "gemini-2.5-flash"
 
+llm = ChatVertexAI(model=LLM, location=LOCATION, temperature=0)
 
-# 1. Define tools
-@tool
-def search(query: str) -> str:
+
+def get_weather(query: str) -> str:
     """Simulates a web search. Use it get information on weather"""
     if "sf" in query.lower() or "san francisco" in query.lower():
         return "It's 60 degrees and foggy."
     return "It's 90 degrees and sunny."
 
 
-tools = [search]
-
-# 2. Set up the language model
-llm = ChatVertexAI(
-    model=LLM, location=LOCATION, temperature=0, max_tokens=1024, streaming=True
-).bind_tools(tools)
-
-
-# 3. Define workflow components
-def should_continue(state: MessagesState) -> str:
-    """Determines whether to use tools or end the conversation."""
-    last_message = state["messages"][-1]
-    return "tools" if last_message.tool_calls else END
-
-
-def call_model(state: MessagesState, config: RunnableConfig) -> dict[str, BaseMessage]:
-    """Calls the language model and returns the response."""
-    system_message = "You are a helpful AI assistant."
-    messages_with_system = [{"type": "system", "content": system_message}] + state[
-        "messages"
-    ]
-    # Forward the RunnableConfig object to ensure the agent is capable of streaming the response.
-    response = llm.invoke(messages_with_system, config)
-    return {"messages": response}
-
-
-# 4. Create the workflow graph
-workflow = StateGraph(MessagesState)
-workflow.add_node("agent", call_model)
-workflow.add_node("tools", ToolNode(tools))
-workflow.set_entry_point("agent")
-
-# 5. Define graph edges
-workflow.add_conditional_edges("agent", should_continue)
-workflow.add_edge("tools", "agent")
-
-# 6. Compile the workflow
-agent = workflow.compile()
+agent = create_react_agent(
+    model=llm, tools=[get_weather], prompt="You are a helpful assistant"
+)

llm.txt

@@ -138,7 +138,7 @@ Templates for the `create` command (via `-a` or `--agent`):
 | `agentic_rag`          | RAG agent for document retrieval & Q&A       |
 | `langgraph_base_react` | Base ReAct agent (LangGraph)                 |
 | `crewai_coding_crew`   | Multi-agent collaborative coding assistance  |
-| `live_api`             | Real-time multimodal RAG agent               |
+| `adk_live`             | Real-time multimodal RAG agent               |
 
 ---
 

src/base_template/GEMINI.md

@@ -1,5 +1,5 @@
 Coding Agent guidance:
-{%- if "adk" in cookiecutter.tags %}
+{%- if cookiecutter.is_adk %}
 {{ cookiecutter.adk_cheatsheet }}
 {%- endif %}
 {{ cookiecutter.llm_txt }}

src/base_template/Makefile

@@ -1,99 +1,128 @@
+# ==============================================================================
+# Installation & Setup
+# ==============================================================================
+
 # Install dependencies using uv package manager
 install:
-	@command -v uv >/dev/null 2>&1 || { echo "uv is not installed. Installing uv..."; curl -LsSf https://astral.sh/uv/0.6.12/install.sh | sh; source $HOME/.local/bin/env; }
-{%- if cookiecutter.settings.get("commands", {}).get("override", {}).get("install") %} 
+	@command -v uv >/dev/null 2>&1 || { echo "uv is not installed. Installing uv..."; curl -LsSf https://astral.sh/uv/0.8.13/install.sh | sh; source $HOME/.local/bin/env; }
+{%- if cookiecutter.settings.get("commands", {}).get("override", {}).get("install") %}
 	{{cookiecutter.settings.get("commands", {}).get("override", {}).get("install")}}
 {%- else %}
-	uv sync --dev{% if cookiecutter.agent_name != 'live_api' and "adk" not in cookiecutter.tags %} --extra streamlit{%- endif %}{% if cookiecutter.agent_name == 'live_api' %} && (cd frontend && npm install){%- endif %}
-{%- endif %}
-
-{%- if cookiecutter.settings.get("commands", {}).get("extra", {}) %}
-{%- for cmd_name, cmd_value in cookiecutter.settings.get("commands", {}).get("extra", {}).items() %}
-
-# {{ cmd_value.get("description") }}
-{{ cmd_name }}:
-{%- if cmd_value is mapping %}
-{%- if cmd_value.command is mapping and cookiecutter.deployment_target in cmd_value.command %}
-	{{ cmd_value.command[cookiecutter.deployment_target] }}
-{%- else %}
-	{{ cmd_value.command if cmd_value.command is string else "" }}
-{%- endif %}
-{%- else %}
-	{{ cmd_value }}
+	uv sync --dev{% if not cookiecutter.is_adk and not cookiecutter.is_adk_live %} --extra streamlit{%- endif %}{% if cookiecutter.is_adk_live %} && (cd frontend && npm install){%- endif %}
 {%- endif %}
-{%- endfor %}{%- endif %}
 
-{%- if cookiecutter.agent_name == 'live_api' %}
-# Build the frontend for production
-build-frontend:
-	(cd frontend && npm run build)
+# ==============================================================================
+# Playground Targets
+# ==============================================================================
 
-{%- endif %}
 # Launch local dev playground
-playground:{%- if cookiecutter.agent_name == 'live_api' %} build-frontend{%- endif %}
+playground:{%- if cookiecutter.is_adk_live %} build-frontend-if-needed{%- endif %}
 {%- if cookiecutter.settings.get("commands", {}).get("override", {}).get("playground") %}
+{%- if cookiecutter.settings.get("commands", {}).get("override", {}).get("playground") is mapping %}
+	{{cookiecutter.settings.get("commands", {}).get("override", {}).get("playground").get(cookiecutter.deployment_target, "")}}
+{%- else %}
 	{{cookiecutter.settings.get("commands", {}).get("override", {}).get("playground")}}
+{%- endif %}
 {%- else %}
 	@echo "==============================================================================="
 	@echo "| 🚀 Starting your agent playground...                                        |"
 	@echo "|                                                                             |"
-{%- if cookiecutter.agent_name == 'live_api' %}
+{%- if cookiecutter.is_adk_live %}
 	@echo "| 🌐 Access your app at: http://localhost:8000                               |"
 {%- endif %}
 	@echo "| 💡 Try asking: {{cookiecutter.example_question}}|"
-{%- if "adk" in cookiecutter.tags %}
+{%- if cookiecutter.is_adk %}
 	@echo "|                                                                             |"
 	@echo "| 🔍 IMPORTANT: Select the '{{cookiecutter.agent_directory}}' folder to interact with your agent.          |"
 {%- endif %}
 	@echo "==============================================================================="
-{%- if "adk" in cookiecutter.tags %}
+{%- if cookiecutter.is_adk_live %}
+{%- if cookiecutter.deployment_target == 'agent_engine' %}
+	uv run python -m {{cookiecutter.agent_directory}}.utils.expose_app --mode local
+{%- else %}
+	uv run uvicorn {{cookiecutter.agent_directory}}.server:app --host localhost --port 8000 --reload
+{%- endif %}
+{%- elif cookiecutter.is_adk %}
 	uv run adk web . --port 8501 --reload_agents
 {%- else %}
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
-{%- if cookiecutter.agent_name == 'live_api' %}
-	uv run uvicorn {{cookiecutter.agent_directory}}.server:app --host 0.0.0.0 --port 8000 --reload
-{%- else %}
-	uv run uvicorn {{cookiecutter.agent_directory}}.server:app --host 0.0.0.0 --port 8000 --reload &
-{%- endif %}
+	uv run uvicorn {{cookiecutter.agent_directory}}.server:app --host localhost --port 8000 --reload &
 {%- endif %}
-{%- if cookiecutter.agent_name != 'live_api' %}
 	{% if cookiecutter.deployment_target == 'agent_engine' %}PYTHONPATH=. {% endif %}uv run streamlit run frontend/streamlit_app.py --browser.serverAddress=localhost --server.enableCORS=false --server.enableXsrfProtection=false
 {%- endif %}
 {%- endif %}
+{%- if cookiecutter.settings.get("commands", {}).get("extra", {}) %}
+
+# ==============================================================================
+# Agent-Specific Commands
+# ==============================================================================
+{%- for cmd_name, cmd_value in cookiecutter.settings.get("commands", {}).get("extra", {}).items() %}
+{%- if cmd_value is mapping %}
+{%- if cmd_value.command is mapping %}
+{%- if cookiecutter.deployment_target in cmd_value.command %}
+
+# {{ cmd_value.get("description") }}
+{{ cmd_name }}:
+	{{ cmd_value.command[cookiecutter.deployment_target] }}
 {%- endif %}
+{%- else %}
 
-# Deploy the agent remotely
-{%- if cookiecutter.deployment_target == 'cloud_run' %}
-# Usage: make backend [IAP=true] [PORT=8080] - Set IAP=true to enable Identity-Aware Proxy, PORT to specify container port
+# {{ cmd_value.get("description") }}
+{{ cmd_name }}:
+	{{ cmd_value.command if cmd_value.command is string else "" }}
 {%- endif %}
-backend:
-{%- if cookiecutter.deployment_target == 'cloud_run' %}
-	PROJECT_ID=$$(gcloud config get-value project) && \
-	gcloud beta run deploy {{cookiecutter.project_name}} \
-		--source . \
-		--memory "4Gi" \
-		--project $$PROJECT_ID \
-		--region "us-central1" \
-		--no-allow-unauthenticated \
-		--no-cpu-throttling \
-		--labels "{% if "adk" in cookiecutter.tags %}created-by=adk{% if cookiecutter.agent_garden %},{% endif %}{% endif %}{% if cookiecutter.agent_garden %}deployed-with=agent-garden{% if cookiecutter.agent_sample_id %},vertex-agent-sample-id={{cookiecutter.agent_sample_id}},vertex-agent-sample-publisher={{cookiecutter.agent_sample_publisher}}{% endif %}{% endif %}" \
-		--set-env-vars \
-		"COMMIT_SHA=$(shell git rev-parse HEAD){%- if cookiecutter.data_ingestion %}{%- if cookiecutter.datastore_type == "vertex_ai_search" %},DATA_STORE_ID={{cookiecutter.project_name}}-datastore,DATA_STORE_REGION=us{%- elif cookiecutter.datastore_type == "vertex_ai_vector_search" %},VECTOR_SEARCH_INDEX={{cookiecutter.project_name}}-vector-search,VECTOR_SEARCH_INDEX_ENDPOINT={{cookiecutter.project_name}}-vector-search-endpoint,VECTOR_SEARCH_BUCKET=$$PROJECT_ID-{{cookiecutter.project_name}}-vs{%- endif %}{%- endif %}" \
-		$(if $(IAP),--iap) \
-		$(if $(PORT),--port=$(PORT))
-{%- elif cookiecutter.deployment_target == 'agent_engine' %}
-	# Export dependencies to requirements file using uv export.
-	uv export --no-hashes --no-header --no-dev --no-emit-project --no-annotate > .requirements.txt 2>/dev/null || \
-	uv export --no-hashes --no-header --no-dev --no-emit-project > .requirements.txt && uv run {{cookiecutter.agent_directory}}/agent_engine_app.py
+{%- else %}
+
+# {{ cmd_value.get("description", "") }}
+{{ cmd_name }}:
+	{{ cmd_value }}
 {%- endif %}
+{%- endfor %}
+{%- endif %}
+{%- if cookiecutter.is_adk_live %}
+
+# ==============================================================================
+# ADK Live Commands
+# ==============================================================================
+
+# Build the frontend for production
+build-frontend:
+	(cd frontend && npm run build)
+
+# Build the frontend only if needed (conditional build)
+build-frontend-if-needed:
+	@if [ ! -d "frontend/build" ] || [ ! -f "frontend/build/index.html" ]; then \
+		echo "Frontend build directory not found or incomplete. Building..."; \
+		$(MAKE) build-frontend; \
+	elif [ "frontend/package.json" -nt "frontend/build/index.html" ] || \
+		 find frontend/src -newer frontend/build/index.html 2>/dev/null | head -1 | grep -q .; then \
+		echo "Frontend source files are newer than build. Rebuilding..."; \
+		$(MAKE) build-frontend; \
+	else \
+		echo "Frontend build is up to date. Skipping build..."; \
+	fi
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
 
 # Launch local development server with hot-reload
 local-backend:
-	uv run uvicorn {{cookiecutter.agent_directory}}.server:app --host 0.0.0.0 --port 8000 --reload
+	uv run uvicorn {{cookiecutter.agent_directory}}.server:app --host localhost --port 8000 --reload
+{%- endif %}
+{%- if cookiecutter.deployment_target == 'agent_engine' %}
+
+# Launch local development server with hot-reload
+local-backend:
+	uv run python -m {{cookiecutter.agent_directory}}.utils.expose_app --mode local --port 8000
+
+# Connect to remote deployed agent
+playground-remote: build-frontend-if-needed
+	@echo "==============================================================================="
+	@echo "| 🚀 Connecting to REMOTE agent...                                           |"
+	@echo "|                                                                             |"
+	@echo "| 🌐 Access your app at: http://localhost:8000                               |"
+	@echo "| ☁️  Connected to deployed agent engine                                      |"
+	@echo "==============================================================================="
+	uv run python -m {{cookiecutter.agent_directory}}.utils.expose_app --mode remote
 {%- endif %}
-{%- if cookiecutter.deployment_target == 'cloud_run' %}
-{%- if cookiecutter.agent_name == 'live_api' %}
 
 # Start the frontend UI separately for development (requires backend running separately)
 ui:
@@ -106,7 +135,7 @@ playground-dev:
 	@echo "|                                                                             |"
 	@echo "| 🌐 Frontend: http://localhost:8501                                         |"
 	@echo "| 🌐 Backend:  http://localhost:8000                                         |"
-	@echo "| 💡 Try asking: What's the weather in San Francisco?                         |"
+	@echo "| 💡 Try asking: {{cookiecutter.example_question}}|"
 	@echo "| 🔄 Both frontend and backend will auto-reload on changes                    |"
 	@echo "==============================================================================="
 	@echo "Starting backend server..."
@@ -114,15 +143,51 @@ playground-dev:
 	@echo "Starting frontend dev server..."
 	$(MAKE) ui
 {%- endif %}
+
+# ==============================================================================
+# Backend Deployment Targets
+# ==============================================================================
+
+# Deploy the agent remotely
+{%- if cookiecutter.deployment_target == 'cloud_run' %}
+# Usage: make backend [IAP=true] [PORT=8080] - Set IAP=true to enable Identity-Aware Proxy, PORT to specify container port
+{%- endif %}
+backend:
+{%- if cookiecutter.deployment_target == 'cloud_run' %}
+	PROJECT_ID=$$(gcloud config get-value project) && \
+	gcloud beta run deploy {{cookiecutter.project_name}} \
+		--source . \
+		--memory "4Gi" \
+		--project $$PROJECT_ID \
+		--region "us-central1" \
+		--no-allow-unauthenticated \
+		--no-cpu-throttling \
+		--labels "{% if cookiecutter.is_adk %}created-by=adk{% if cookiecutter.agent_garden %},{% endif %}{% endif %}{% if cookiecutter.agent_garden %}deployed-with=agent-garden{% if cookiecutter.agent_sample_id %},vertex-agent-sample-id={{cookiecutter.agent_sample_id}},vertex-agent-sample-publisher={{cookiecutter.agent_sample_publisher}}{% endif %}{% endif %}" \
+		--set-env-vars \
+		"COMMIT_SHA=$(shell git rev-parse HEAD){%- if cookiecutter.data_ingestion %}{%- if cookiecutter.datastore_type == "vertex_ai_search" %},DATA_STORE_ID={{cookiecutter.project_name}}-datastore,DATA_STORE_REGION=us{%- elif cookiecutter.datastore_type == "vertex_ai_vector_search" %},VECTOR_SEARCH_INDEX={{cookiecutter.project_name}}-vector-search,VECTOR_SEARCH_INDEX_ENDPOINT={{cookiecutter.project_name}}-vector-search-endpoint,VECTOR_SEARCH_BUCKET=$$PROJECT_ID-{{cookiecutter.project_name}}-vs{%- endif %}{%- endif %}" \
+		$(if $(IAP),--iap) \
+		$(if $(PORT),--port=$(PORT))
+{%- elif cookiecutter.deployment_target == 'agent_engine' %}
+	# Export dependencies to requirements file using uv export.
+	uv export --no-hashes --no-header --no-dev --no-emit-project --no-annotate > .requirements.txt 2>/dev/null || \
+	uv export --no-hashes --no-header --no-dev --no-emit-project > .requirements.txt && uv run {{cookiecutter.agent_directory}}/agent_engine_app.py
 {%- endif %}
 
+
+# ==============================================================================
+# Infrastructure Setup
+# ==============================================================================
+
 # Set up development environment resources using Terraform
 setup-dev-env:
 	PROJECT_ID=$$(gcloud config get-value project) && \
 	(cd deployment/terraform/dev && terraform init && terraform apply --var-file vars/env.tfvars --var dev_project_id=$$PROJECT_ID --auto-approve)
-
 {%- if cookiecutter.data_ingestion %}
 
+# ==============================================================================
+# Data Ingestion (RAG capabilities)
+# ==============================================================================
+
 # Run the data ingestion pipeline for RAG capabilities
 data-ingestion:
 	PROJECT_ID=$$(gcloud config get-value project) && \
@@ -142,6 +207,10 @@ data-ingestion:
 		--pipeline-name="data-ingestion-pipeline")
 {%- endif %}
 
+# ==============================================================================
+# Testing & Code Quality
+# ==============================================================================
+
 # Run unit and integration tests
 test:
 	uv run pytest tests/unit && uv run pytest tests/integration

src/base_template/README.md

@@ -58,11 +58,11 @@ make install && make playground
 {%- endfor %}
 {%- endif %}
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
-| `make playground`    | Launch local development environment with backend and frontend{%- if "adk" in cookiecutter.tags %} - leveraging `adk web` command. {%- endif %}|
+| `make playground`    | Launch local development environment with backend and frontend{%- if cookiecutter.is_adk %} - leveraging `adk web` command. {%- endif %}|
 | `make backend`       | Deploy agent to Cloud Run (use `IAP=true` to enable Identity-Aware Proxy) |
 | `make local-backend` | Launch local development server |
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
-{%- if cookiecutter.agent_name == 'live_api' %}
+{%- if cookiecutter.is_adk_live %}
 | `make ui`       | Launch Agent Playground front-end only |
 {%- endif %}
 {%- endif %}
@@ -80,7 +80,7 @@ make install && make playground
 
 For full command options and usage, refer to the [Makefile](Makefile).
 
-{% if cookiecutter.agent_name == 'live_api' %}
+{% if cookiecutter.is_adk_live %}
 ## Usage
 
 This template follows a "bring your own agent" approach - you focus on your business logic in `{{cookiecutter.agent_directory}}/agent.py`, and the template handles the surrounding components (UI, infrastructure, deployment, monitoring).
@@ -158,7 +158,7 @@ You can test deployment towards a Dev Environment using the following command:
 gcloud config set project <your-dev-project-id>
 make backend
 ```
-{% if cookiecutter.agent_name == 'live_api' %}
+{% if cookiecutter.is_adk_live %}
 **Note:** For secure access to your deployed backend, consider using Identity-Aware Proxy (IAP) by running `make backend IAP=true`.
 {%- endif %}
 
@@ -169,9 +169,9 @@ See [deployment/README.md](deployment/README.md) for instructions.
 
 The repository includes a Terraform configuration for the setup of a production Google Cloud project. Refer to [deployment/README.md](deployment/README.md) for detailed instructions on how to deploy the infrastructure and application.
 
-{% if cookiecutter.agent_name != 'live_api' %}
+{% if not cookiecutter.is_adk_live %}
 ## Monitoring and Observability
-> You can use [this Looker Studio dashboard]({%- if "adk" in cookiecutter.tags %}https://lookerstudio.google.com/reporting/46b35167-b38b-4e44-bd37-701ef4307418/page/tEnnC{%- else %}https://lookerstudio.google.com/c/reporting/fa742264-4b4b-4c56-81e6-a667dd0f853f/page/tEnnC{%- endif %}
+> You can use [this Looker Studio dashboard]({%- if cookiecutter.is_adk %}https://lookerstudio.google.com/reporting/46b35167-b38b-4e44-bd37-701ef4307418/page/tEnnC{%- else %}https://lookerstudio.google.com/c/reporting/fa742264-4b4b-4c56-81e6-a667dd0f853f/page/tEnnC{%- endif %}
 ) template for visualizing events being logged in BigQuery. See the "Setup Instructions" tab to getting started.
 
 The application uses OpenTelemetry for comprehensive observability with all events being sent to Google Cloud Trace and Logging for monitoring and to BigQuery for long term storage. 

src/base_template/deployment/terraform/dev/apis.tf

@@ -25,7 +25,7 @@ locals {
     "serviceusage.googleapis.com",
     "logging.googleapis.com",
     "cloudtrace.googleapis.com",
-{%- if "adk" in cookiecutter.tags and cookiecutter.session_type == "alloydb" %}
+{%- if cookiecutter.is_adk and cookiecutter.session_type == "alloydb" %}
     "compute.googleapis.com",
     "servicenetworking.googleapis.com",
     "alloydb.googleapis.com",

src/base_template/deployment/terraform/dev/variables.tf

@@ -32,7 +32,7 @@ variable "region" {
 variable "telemetry_logs_filter" {
   type        = string
   description = "Log Sink filter for capturing telemetry data. Captures logs with the `traceloop.association.properties.log_type` attribute set to `tracing`."
-{%- if "adk" in cookiecutter.tags %}
+{%- if cookiecutter.is_adk %}
   default     = "labels.service_name=\"{{cookiecutter.project_name}}\" labels.type=\"agent_telemetry\""
 {%- else %}
   default     = "jsonPayload.attributes.\"traceloop.association.properties.log_type\"=\"tracing\" jsonPayload.resource.attributes.\"service.name\"=\"{{cookiecutter.project_name}}\""

src/base_template/deployment/terraform/locals.tf

@@ -33,7 +33,7 @@ locals {
     "serviceusage.googleapis.com",
     "logging.googleapis.com",
     "cloudtrace.googleapis.com",
-{%- if "adk" in cookiecutter.tags and cookiecutter.session_type == "alloydb" %}
+{%- if cookiecutter.is_adk and cookiecutter.session_type == "alloydb" %}
     "compute.googleapis.com",
     "servicenetworking.googleapis.com",
     "alloydb.googleapis.com",

src/base_template/deployment/terraform/variables.tf

@@ -53,7 +53,7 @@ variable "repository_name" {
 variable "telemetry_logs_filter" {
   type        = string
   description = "Log Sink filter for capturing telemetry data. Captures logs with the `traceloop.association.properties.log_type` attribute set to `tracing`."
-{%- if "adk" in cookiecutter.tags %}
+{%- if cookiecutter.is_adk %}
   default     = "labels.service_name=\"{{cookiecutter.project_name}}\" labels.type=\"agent_telemetry\""
 {%- else %}
   default     = "jsonPayload.attributes.\"traceloop.association.properties.log_type\"=\"tracing\" jsonPayload.resource.attributes.\"service.name\"=\"{{cookiecutter.project_name}}\""

src/base_template/pyproject.toml

@@ -9,19 +9,20 @@ dependencies = [
 {%- for dep in cookiecutter.extra_dependencies %}
     "{{ dep }}",
 {%- endfor %}
-    "opentelemetry-exporter-gcp-trace~=1.9.0",
-{%- if "adk" not in cookiecutter.tags %}
+    "opentelemetry-exporter-gcp-trace>=1.9.0,<2.0.0",
+{%- if not cookiecutter.is_adk %}
     "langchain-core~=0.3.9",
     "traceloop-sdk~=0.38.7",
 {%- endif %}
-    "google-cloud-logging>=3.12.0",
+    "google-cloud-logging>=3.12.0,<4.0.0",
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
-    "google-cloud-aiplatform[evaluation]~=1.116.0",
+    "google-cloud-aiplatform[evaluation]>=1.118.0,<2.0.0",
     "fastapi~=0.115.8",
     "uvicorn~=0.34.0",
-    "psycopg2-binary>=2.9.10",
+    "psycopg2-binary>=2.9.10,<3.0.0",
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
-    "google-cloud-aiplatform[evaluation,agent-engines]~=1.116.0"
+    "google-cloud-aiplatform[evaluation,agent-engines]>=1.118.0,<2.0.0",
+    "protobuf>=6.31.1,<7.0.0"
 {%- endif %}
 ]
 {% if cookiecutter.deployment_target == 'cloud_run' %}
@@ -32,29 +33,29 @@ requires-python = ">=3.10,<3.13"
 
 [dependency-groups]
 dev = [
-    "pytest>=8.3.4",
-    "pytest-asyncio>=0.23.8",
-    "nest-asyncio>=1.6.0",
+    "pytest>=8.3.4,<9.0.0",
+    "pytest-asyncio>=0.23.8,<1.0.0",
+    "nest-asyncio>=1.6.0,<2.0.0",
 ]
 
 [project.optional-dependencies]
-{% if cookiecutter.agent_name != 'live_api' and "adk" not in cookiecutter.tags %}
+{% if not cookiecutter.is_adk %}
 streamlit = [
-    "streamlit~=1.42.0",
-    "streamlit-extras~=0.4.3",
-    "extra-streamlit-components~=0.1.71",
-    "streamlit-feedback~=0.1.3",
+    "streamlit>=1.42.0,<2.0.0",
+    "streamlit-extras>=0.4.3,<1.0.0",
+    "extra-streamlit-components>=0.1.71,<1.0.0",
+    "streamlit-feedback>=0.1.3,<1.0.0",
 ]
 {% endif %}
 jupyter = [
-    "jupyter~=1.0.0",
+    "jupyter>=1.0.0,<2.0.0",
 ]
 lint = [
-    "ruff>=0.4.6",
-    "mypy~=1.15.0",
-    "codespell~=2.2.0",
-    "types-pyyaml~=6.0.12.20240917",
-    "types-requests~=2.32.0.20240914",
+    "ruff>=0.4.6,<1.0.0",
+    "mypy>=1.15.0,<2.0.0",
+    "codespell>=2.2.0,<3.0.0",
+    "types-pyyaml>=6.0.12.20240917,<7.0.0",
+    "types-requests>=2.32.0.20240914,<3.0.0",
 ]
 
 [tool.ruff]
@@ -92,7 +93,7 @@ follow_imports = "silent"
 ignore_missing_imports = true
 explicit_package_bases = true
 disable_error_code = ["misc", "no-untyped-call", "no-any-return"]
-{% if cookiecutter.agent_name == 'live_api' %}
+{% if cookiecutter.is_adk_live %}
 exclude = [".venv","./frontend"]
 {% else %}
 exclude = [".venv"]

src/base_template/{% if cookiecutter.cicd_runner == 'github_actions' %}.github{% else %}unused_github{% endif %}/workflows/deploy-to-prod.yaml

@@ -67,14 +67,14 @@ jobs:
 {%- if cookiecutter.deployment_target == 'agent_engine' %}
       - name: Install uv and dependencies
         run: |
-          pip install uv==0.6.12
+          pip install uv==0.8.13
           uv sync --locked
 {%- endif %}
 
 {%- if cookiecutter.data_ingestion %}
       - name: Deploy data ingestion pipeline (Production)
         run: |
-          cd data_ingestion && pip install uv==0.6.12 && cd data_ingestion_pipeline && \
+          cd data_ingestion && pip install uv==0.8.13 && cd data_ingestion_pipeline && \
           uv sync --locked && uv run python submit_pipeline.py
         env:
           PIPELINE_ROOT: {% raw %}${{ vars.PIPELINE_GCS_ROOT_PROD }}{% endraw %}

src/base_template/{% if cookiecutter.cicd_runner == 'github_actions' %}.github{% else %}unused_github{% endif %}/workflows/pr_checks.yaml

@@ -55,7 +55,7 @@ jobs:
 
       - name: Install uv and dependencies
         run: |
-          pip install uv==0.6.12
+          pip install uv==0.8.13
           uv sync --locked
 
       - name: Run unit tests

src/base_template/{% if cookiecutter.cicd_runner == 'github_actions' %}.github{% else %}unused_github{% endif %}/workflows/staging.yaml

@@ -58,7 +58,7 @@ jobs:
 
       - name: Deploy data ingestion pipeline (Staging)
         run: |
-          cd data_ingestion && pip install uv==0.6.12 && cd data_ingestion_pipeline && \
+          cd data_ingestion && pip install uv==0.8.13 && cd data_ingestion_pipeline && \
           uv sync --locked && uv run python submit_pipeline.py
         env:
           PIPELINE_ROOT: {% raw %}${{ vars.PIPELINE_GCS_ROOT_STAGING }}{% endraw %}
@@ -113,7 +113,7 @@ jobs:
 
       - name: Install uv and dependencies
         run: |
-          pip install uv==0.6.12
+          pip install uv==0.8.13
           uv sync --locked
 
       - name: Deploy to Staging (Agent Engine)
@@ -135,23 +135,86 @@ jobs:
 
       - name: Run load test
         run: |
-{%- if cookiecutter.deployment_target == 'cloud_run' %}
+{%- if cookiecutter.deployment_target == 'cloud_run' and cookiecutter.agent_name == 'adk_live' %}
+          # Install dependencies for load test
+          pip install locust==2.31.1 websockets
+
+          # Install Cloud Run proxy component
+          gcloud components install cloud-run-proxy --quiet
+
+          # Start Cloud Run proxy in background
+          gcloud run services proxy {{cookiecutter.project_name}} \
+          --port=8080 \
+          --region {% raw %}${{ vars.REGION }}{% endraw %} \
+          --project {% raw %}${{ vars.STAGING_PROJECT_ID }}{% endraw %} \
+          --quiet &
+          PROXY_PID=$!
+
+          # Wait for proxy to be ready
+          echo "Waiting for proxy to start..."
+          sleep 10
+
+          # Run load test and capture exit code
+          locust -f tests/load_test/load_test.py \
+          --headless \
+          -H http://127.0.0.1:8080 \
+          -t 30s -u 2 -r 2 \
+          --csv=tests/load_test/.results/results \
+          --html=tests/load_test/.results/report.html
+          LOCUST_EXIT_CODE=$?
+
+          # Clean up proxy
+          kill $PROXY_PID || true
+
+          # Exit with load test result to fail build if tests failed
+          if [ $LOCUST_EXIT_CODE -ne 0 ]; then
+            echo "Load test failed with exit code $LOCUST_EXIT_CODE"
+            exit $LOCUST_EXIT_CODE
+          fi
+{%- elif cookiecutter.deployment_target == 'cloud_run' %}
           export _ID_TOKEN="{% raw %}${{ steps.fetch-token.outputs._id_token }}{% endraw %}"
           export _STAGING_URL="{% raw %}${{ steps.fetch-url.outputs._staging_url }}{% endraw %}"
-{%- elif cookiecutter.deployment_target == 'agent_engine' %}
-          export _AUTH_TOKEN="{% raw %}${{ steps.fetch-token.outputs._auth_token }}{% endraw %}"
-{%- endif %}
           pip install locust==2.31.1
           locust -f tests/load_test/load_test.py \
           --headless \
-{%- if cookiecutter.deployment_target == 'cloud_run' %}
           -H ${_STAGING_URL} \
           -t 30s -u 10 -r 0.5 \
+          --csv=tests/load_test/.results/results \
+          --html=tests/load_test/.results/report.html
+{%- elif cookiecutter.deployment_target == 'agent_engine' and cookiecutter.is_adk_live %}
+          # Start expose app in remote mode (uses deployment_metadata.json by default)
+          uv run python -m {{cookiecutter.agent_directory}}.utils.expose_app --mode remote &
+          EXPOSE_PID=$!
+
+          # Wait for expose app to be ready
+          sleep 10
+
+          # Run load test against local expose app
+          uv run --with locust==2.31.1 --with websockets locust -f tests/load_test/load_test.py \
+          -H http://127.0.0.1:8000 \
+          --headless \
+          -t 30s -u 2 -r 1 \
+          --csv=tests/load_test/.results/results \
+          --html=tests/load_test/.results/report.html
+          LOCUST_EXIT_CODE=$?
+
+          # Stop expose app
+          kill $EXPOSE_PID
+
+          # Exit with error if locust test failed
+          if [ $LOCUST_EXIT_CODE -ne 0 ]; then
+            echo "Load test failed with exit code $LOCUST_EXIT_CODE"
+            exit $LOCUST_EXIT_CODE
+          fi
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
+          export _AUTH_TOKEN="{% raw %}${{ steps.fetch-token.outputs._auth_token }}{% endraw %}"
+          pip install locust==2.31.1
+          locust -f tests/load_test/load_test.py \
+          --headless \
           -t 30s -u 2 -r 0.5 \
-{%- endif %}
           --csv=tests/load_test/.results/results \
           --html=tests/load_test/.results/report.html
+{%- endif %}
 
       - name: Export Load Test Results to GCS
         run: |