agent-starter-pack 0.15.7__py3-none-any.whl → 0.17.0__py3-none-any.whl

{src → agent_starter_pack}/base_template/deployment/terraform/dev/providers.tf

@@ -17,7 +17,7 @@ terraform {
   required_providers {
     google = {
       source  = "hashicorp/google"
-      version = "< 7.0.0"
+      version = "> 7.0.0"
     }
   }
 }

{src → agent_starter_pack}/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 → agent_starter_pack}/base_template/deployment/terraform/github.tf

@@ -119,6 +119,20 @@ resource "github_actions_variable" "app_sa_email_prod" {
   depends_on    = [github_repository.repo]
 }
 
+resource "github_actions_variable" "logs_bucket_name_staging" {
+  repository    = var.repository_name
+  variable_name = "LOGS_BUCKET_NAME_STAGING"
+  value         = google_storage_bucket.logs_data_bucket[var.staging_project_id].url
+  depends_on    = [github_repository.repo]
+}
+
+resource "github_actions_variable" "logs_bucket_name_prod" {
+  repository    = var.repository_name
+  variable_name = "LOGS_BUCKET_NAME_PROD"
+  value         = google_storage_bucket.logs_data_bucket[var.prod_project_id].url
+  depends_on    = [github_repository.repo]
+}
+
 {% if cookiecutter.deployment_target == 'cloud_run' %}
 resource "github_actions_variable" "container_name" {
   repository    = var.repository_name

{src → agent_starter_pack}/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 → agent_starter_pack}/base_template/deployment/terraform/log_sinks.tf

@@ -12,14 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-resource "google_project_iam_member" "bigquery_data_editor" {
-  for_each = local.deploy_project_ids
-
-  project = each.value
-  role    = "roles/bigquery.dataEditor"
-  member  = module.log_export_to_bigquery[each.key].writer_identity
-}
-
 resource "google_bigquery_dataset" "feedback_dataset" {
   for_each      = local.deploy_project_ids
   project       = each.value
@@ -38,33 +30,50 @@ resource "google_bigquery_dataset" "telemetry_logs_dataset" {
   depends_on    = [resource.google_project_service.cicd_services, resource.google_project_service.deploy_project_services]
 }
 
-module "log_export_to_bigquery" {
+resource "google_logging_project_sink" "feedback_export_to_bigquery" {
   for_each = local.deploy_project_ids
 
-  source  = "terraform-google-modules/log-export/google"
-  version = "10.0.0"
+  name        = "${var.project_name}_feedback"
+  project     = each.value
+  destination = "bigquery.googleapis.com/projects/${each.value}/datasets/${google_bigquery_dataset.feedback_dataset[each.key].dataset_id}"
+  filter      = var.feedback_logs_filter
+
+  bigquery_options {
+    use_partitioned_tables = true
+  }
 
-  log_sink_name          = "${var.project_name}_telemetry"
-  parent_resource_type   = "project"
-  parent_resource_id     = each.value
-  destination_uri        = "bigquery.googleapis.com/projects/${each.value}/datasets/${google_bigquery_dataset.telemetry_logs_dataset[each.key].dataset_id}"
-  filter                 = var.telemetry_logs_filter
-  bigquery_options       = { use_partitioned_tables = true }
   unique_writer_identity = true
-  depends_on             = [google_bigquery_dataset.telemetry_logs_dataset]
+  depends_on             = [google_bigquery_dataset.feedback_dataset]
 }
 
-module "feedback_export_to_bigquery" {
+resource "google_logging_project_sink" "log_export_to_bigquery" {
   for_each = local.deploy_project_ids
 
-  source                 = "terraform-google-modules/log-export/google"
-  version                = "10.0.0"
-  log_sink_name          = "${var.project_name}_feedback"
-  parent_resource_type   = "project"
-  parent_resource_id     = each.value
-  destination_uri        = "bigquery.googleapis.com/projects/${each.value}/datasets/${google_bigquery_dataset.feedback_dataset[each.key].dataset_id}"
-  filter                 = var.feedback_logs_filter
-  bigquery_options       = { use_partitioned_tables = true }
+  name        = "${var.project_name}_telemetry"
+  project     = each.value
+  destination = "bigquery.googleapis.com/projects/${each.value}/datasets/${google_bigquery_dataset.telemetry_logs_dataset[each.key].dataset_id}"
+  filter      = var.telemetry_logs_filter
+
+  bigquery_options {
+    use_partitioned_tables = true
+  }
+
   unique_writer_identity = true
-  depends_on             = [google_bigquery_dataset.feedback_dataset]
+  depends_on             = [google_bigquery_dataset.telemetry_logs_dataset]
+}
+
+resource "google_project_iam_member" "bigquery_data_editor" {
+  for_each = local.deploy_project_ids
+
+  project = each.value
+  role    = "roles/bigquery.dataEditor"
+  member  = google_logging_project_sink.log_export_to_bigquery[each.key].writer_identity
+}
+
+resource "google_project_iam_member" "feedback_bigquery_data_editor" {
+  for_each = local.deploy_project_ids
+
+  project = each.value
+  role    = "roles/bigquery.dataEditor"
+  member  = google_logging_project_sink.feedback_export_to_bigquery[each.key].writer_identity
 }

{src → agent_starter_pack}/base_template/deployment/terraform/providers.tf

@@ -18,7 +18,7 @@ terraform {
   required_providers {
     google = {
       source  = "hashicorp/google"
-      version = "< 7.0.0"
+      version = "> 7.0.0"
     }
     github = {
       source  = "integrations/github"

{src → agent_starter_pack}/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 → agent_starter_pack}/base_template/deployment/terraform/{% if cookiecutter.cicd_runner == 'google_cloud_build' %}build_triggers.tf{% else %}unused_build_triggers.tf{% endif %}

@@ -74,12 +74,13 @@ resource "google_cloudbuild_trigger" "cd_pipeline" {
   substitutions = {
     _STAGING_PROJECT_ID            = var.staging_project_id
     _BUCKET_NAME_LOAD_TEST_RESULTS = resource.google_storage_bucket.bucket_load_test_results.name
+    _LOGS_BUCKET_NAME_STAGING      = resource.google_storage_bucket.logs_data_bucket[var.staging_project_id].url
     _REGION                        = var.region
 {% if cookiecutter.deployment_target == 'cloud_run' %}
     _CONTAINER_NAME                = var.project_name
     _ARTIFACT_REGISTRY_REPO_NAME   = resource.google_artifact_registry_repository.repo-artifacts-genai.repository_id
 {% elif cookiecutter.deployment_target == 'agent_engine' %}
-    _APP_SA_EMAIL_STAGING          = resource.google_service_account.app_sa["staging"].email
+
 {% endif %}
 {% if cookiecutter.data_ingestion %}
     _PIPELINE_GCS_ROOT_STAGING     = "gs://${resource.google_storage_bucket.data_ingestion_pipeline_gcs_root["staging"].name}"
@@ -123,12 +124,13 @@ resource "google_cloudbuild_trigger" "deploy_to_prod_pipeline" {
   }
   substitutions = {
     _PROD_PROJECT_ID             = var.prod_project_id
+    _LOGS_BUCKET_NAME_PROD       = resource.google_storage_bucket.logs_data_bucket[var.prod_project_id].url
     _REGION                      = var.region
 {% if cookiecutter.deployment_target == 'cloud_run' %}
     _CONTAINER_NAME              = var.project_name
     _ARTIFACT_REGISTRY_REPO_NAME = resource.google_artifact_registry_repository.repo-artifacts-genai.repository_id
 {% elif cookiecutter.deployment_target == 'agent_engine' %}
-    _APP_SA_EMAIL_PROD             = resource.google_service_account.app_sa["prod"].email
+
 {% endif %}
 {% if cookiecutter.data_ingestion %}
     _PIPELINE_GCS_ROOT_PROD        = "gs://${resource.google_storage_bucket.data_ingestion_pipeline_gcs_root["prod"].name}"

{src → agent_starter_pack}/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 → agent_starter_pack}/base_template/{% if cookiecutter.cicd_runner == 'github_actions' %}.github{% else %}unused_github{% endif %}/workflows/deploy-to-prod.yaml

@@ -44,10 +44,10 @@ jobs:
       - name: Checkout code
         uses: actions/checkout@v4
 
-      - name: Set up Python 3.11
+      - name: Set up Python 3.12
         uses: actions/setup-python@v4
         with:
-          python-version: '3.11'
+          python-version: '3.12'
 
       - id: 'auth'
         name: 'Authenticate to Google Cloud'
@@ -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 %}
@@ -111,7 +111,7 @@ jobs:
           uv run {{cookiecutter.agent_directory}}/agent_engine_app.py \
             --project {% raw %}${{ vars.PROD_PROJECT_ID }}{% endraw %} \
             --location {% raw %}${{ vars.REGION }}{% endraw %} \
-            --service-account {% raw %}${{ vars.APP_SA_EMAIL_PROD }}{% endraw %} \
+            --artifacts-bucket-name {% raw %}${{ vars.LOGS_BUCKET_NAME_PROD }}{% endraw %} \
             --set-env-vars="COMMIT_SHA={% raw %}${{ github.sha }}{% endraw %}{%- if cookiecutter.data_ingestion %}{%- if cookiecutter.datastore_type == "vertex_ai_search" %},DATA_STORE_ID={% raw %}${{ vars.DATA_STORE_ID_PROD }}{% endraw %},DATA_STORE_REGION={% raw %}${{ vars.DATA_STORE_REGION }}{% endraw %}{%- elif cookiecutter.datastore_type == "vertex_ai_vector_search" %},VECTOR_SEARCH_INDEX={% raw %}${{ vars.VECTOR_SEARCH_INDEX_PROD }}{% endraw %},VECTOR_SEARCH_INDEX_ENDPOINT={% raw %}${{ vars.VECTOR_SEARCH_INDEX_ENDPOINT_PROD }}{% endraw %},VECTOR_SEARCH_BUCKET={% raw %}${{ vars.VECTOR_SEARCH_BUCKET_PROD }}{% endraw %}{%- endif %}{%- endif %}"
 {%- endif %}
 

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

@@ -48,14 +48,14 @@ jobs:
           project_id: {% raw %}${{ vars.CICD_PROJECT_ID }}{% endraw %}
 
 
-      - name: Set up Python 3.11
+      - name: Set up Python 3.12
         uses: actions/setup-python@v4
         with:
-          python-version: '3.11'
+          python-version: '3.12'
 
       - 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 → agent_starter_pack}/base_template/{% if cookiecutter.cicd_runner == 'github_actions' %}.github{% else %}unused_github{% endif %}/workflows/staging.yaml

@@ -36,10 +36,10 @@ jobs:
       - name: Checkout code
         uses: actions/checkout@v4
 
-      - name: Set up Python 3.11
+      - name: Set up Python 3.12
         uses: actions/setup-python@v4
         with:
-          python-version: '3.11'
+          python-version: '3.12'
 
       - id: 'auth'
         name: 'Authenticate to Google Cloud'
@@ -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)
@@ -122,7 +122,7 @@ jobs:
           uv run {{cookiecutter.agent_directory}}/agent_engine_app.py \
             --project {% raw %}${{ vars.STAGING_PROJECT_ID }}{% endraw %} \
             --location {% raw %}${{ vars.REGION }}{% endraw %} \
-            --service-account {% raw %}${{ vars.APP_SA_EMAIL_STAGING }}{% endraw %} \
+            --artifacts-bucket-name {% raw %}${{ vars.LOGS_BUCKET_NAME_STAGING }}{% endraw %} \
             --set-env-vars="COMMIT_SHA={% raw %}${{ github.sha }}{% endraw %}{%- if cookiecutter.data_ingestion %}{%- if cookiecutter.datastore_type == "vertex_ai_search" %},DATA_STORE_ID={% raw %}${{ vars.DATA_STORE_ID_STAGING }}{% endraw %},DATA_STORE_REGION={% raw %}${{ vars.DATA_STORE_REGION }}{% endraw %}{%- elif cookiecutter.datastore_type == "vertex_ai_vector_search" %},VECTOR_SEARCH_INDEX={% raw %}${{ vars.VECTOR_SEARCH_INDEX_STAGING }}{% endraw %},VECTOR_SEARCH_INDEX_ENDPOINT={% raw %}${{ vars.VECTOR_SEARCH_INDEX_ENDPOINT_STAGING }}{% endraw %},VECTOR_SEARCH_BUCKET={% raw %}${{ vars.VECTOR_SEARCH_BUCKET_STAGING }}{% endraw %}{%- endif %}{%- endif %}"
 
       - name: Fetch Auth Token
@@ -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: |

{src → agent_starter_pack}/base_template/{% if cookiecutter.cicd_runner == 'google_cloud_build' %}.cloudbuild{% else %}unused_.cloudbuild{% endif %}/deploy-to-prod.yaml

@@ -14,13 +14,13 @@
 
 steps:
 {%- if cookiecutter.data_ingestion %}
-  - name: "python:3.11-slim"
+  - name: "python:3.12-slim"
     id: deploy-data-ingestion-pipeline-prod
     entrypoint: bash
     args:
       - -c
       - |
-        cd data_ingestion && pip install uv==0.6.12 --user && cd data_ingestion_pipeline && \
+        cd data_ingestion && pip install uv==0.8.13 --user && cd data_ingestion_pipeline && \
         uv sync --locked && uv run python submit_pipeline.py
     env:
       - "PIPELINE_ROOT=${_PIPELINE_GCS_ROOT_PROD}"
@@ -57,17 +57,17 @@ steps:
       - $_PROD_PROJECT_ID
 
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
-  - name: "python:3.11-slim" 
+  - name: "python:3.12-slim"
     id: install-dependencies
     entrypoint: /bin/bash
     args:
       - "-c"
       - |
-        pip install uv==0.6.12 --user && uv sync --locked
+        pip install uv==0.8.13 --user && uv sync --locked
     env:
       - 'PATH=/usr/local/bin:/usr/bin:~/.local/bin'
 
-  - name: "python:3.11-slim"
+  - name: "python:3.12-slim"
     id: trigger-deployment
     entrypoint: /bin/bash
     args:
@@ -77,7 +77,7 @@ steps:
         uv run {{cookiecutter.agent_directory}}/agent_engine_app.py \
           --project ${_PROD_PROJECT_ID} \
           --location ${_REGION} \
-          --service-account ${_APP_SA_EMAIL_PROD} \
+          --artifacts-bucket-name ${_LOGS_BUCKET_NAME_PROD} \
           --set-env-vars="COMMIT_SHA=${COMMIT_SHA}{%- if cookiecutter.data_ingestion %}{%- if cookiecutter.datastore_type == "vertex_ai_search" %},DATA_STORE_ID=${_DATA_STORE_ID_PROD},DATA_STORE_REGION=${_DATA_STORE_REGION}{%- elif cookiecutter.datastore_type == "vertex_ai_vector_search" %},VECTOR_SEARCH_INDEX=${_VECTOR_SEARCH_INDEX_PROD},VECTOR_SEARCH_INDEX_ENDPOINT=${_VECTOR_SEARCH_INDEX_ENDPOINT_PROD},VECTOR_SEARCH_BUCKET=${_VECTOR_SEARCH_BUCKET_PROD}{%- endif %}{%- endif %}"
     env:
       - 'PATH=/usr/local/bin:/usr/bin:~/.local/bin'

{src → agent_starter_pack}/base_template/{% if cookiecutter.cicd_runner == 'google_cloud_build' %}.cloudbuild{% else %}unused_.cloudbuild{% endif %}/pr_checks.yaml

@@ -14,18 +14,18 @@
 
 steps:
   # Install uv package manager and sync dependencies
-  - name: "python:3.11-slim"
+  - name: "python:3.12-slim"
     id: install-dependencies
     entrypoint: /bin/bash
     args:
       - "-c"
       - |
-        pip install uv==0.6.12 --user && uv sync --locked
+        pip install uv==0.8.13 --user && uv sync --locked
     env:
       - 'PATH=/usr/local/bin:/usr/bin:~/.local/bin'
 
   # Run unit tests using pytest
-  - name: "python:3.11-slim"
+  - name: "python:3.12-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-slim"
+  - name: "python:3.12-slim"
     id: integration-tests
     entrypoint: /bin/bash
     args: