agent-starter-pack 0.0.3b0__py3-none-any.whl → 0.0.5b0__py3-none-any.whl

{agent_starter_pack-0.0.3b0.dist-info → agent_starter_pack-0.0.5b0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.0.3b0
+Version: 0.0.5b0
 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

{agent_starter_pack-0.0.3b0.dist-info → agent_starter_pack-0.0.5b0.dist-info}/RECORD

@@ -3,7 +3,7 @@ agents/agentic_rag_vertexai_search/app/agent.py,sha256=ly4OZiOj_jKKSLvo2ZhtQgrxW
 agents/agentic_rag_vertexai_search/app/retrievers.py,sha256=h5mIa2wphDLefBxNQLH50_zFfbKnEwpz7p1S2rTqt28,2769
 agents/agentic_rag_vertexai_search/app/templates.py,sha256=h3dtaVScPNMLqEudh63ZURzZh59p4tclKusFP5rWPgA,1632
 agents/agentic_rag_vertexai_search/notebooks/evaluating_langgraph_agent.ipynb,sha256=INlcKJQ_2KB9XAXloG6wKB7cHBKDhW56w1bAQm2IeLA,58048
-agents/agentic_rag_vertexai_search/template/.templateconfig.yaml,sha256=XP3ZfJRvGmgO9w9lawxv-qlPCbSjObI1_5mZfaUTWRY,495
+agents/agentic_rag_vertexai_search/template/.templateconfig.yaml,sha256=Y1QvZ8Q0kilS_EN6N9YxKW7LOXABq2i2DgeYPwZJHEw,1070
 agents/agentic_rag_vertexai_search/tests/integration/test_agent.py,sha256=8Y4BUvfxPaiEcT8WvdMTl5hVYTWS_W_4aQtYmgrdgx8,1935
 agents/crewai_coding_crew/README.md,sha256=4No7sfHQVHELARGiT3fWANJzDR0NP48ow4ct3FoPwlA,1672
 agents/crewai_coding_crew/app/agent.py,sha256=GcmNrOyRPhnI7LICIV6GvBvcDfA_PKqPsQcOEzAORmY,3269
@@ -12,19 +12,19 @@ agents/crewai_coding_crew/app/crew/config/agents.yaml,sha256=Cn9zTcP3C-8MvlV19cY
 agents/crewai_coding_crew/app/crew/config/tasks.yaml,sha256=SSFuiSQPhi4dM0V5HdNuMqtN7JJr5RUCMhhBzUkPBpc,1511
 agents/crewai_coding_crew/notebooks/evaluating_crewai_agent.ipynb,sha256=OxUud-wX9CLhKEldFwU_jYgA8KALkQ8_MwLN8gvOai8,58007
 agents/crewai_coding_crew/notebooks/evaluating_langgraph_agent.ipynb,sha256=INlcKJQ_2KB9XAXloG6wKB7cHBKDhW56w1bAQm2IeLA,58048
-agents/crewai_coding_crew/template/.templateconfig.yaml,sha256=DwgofjCYxH3A7m6XtrX0sE8iPYiL4TARiCfLZcUbd94,399
+agents/crewai_coding_crew/template/.templateconfig.yaml,sha256=cl2idKg9NZKrkIkDhxCVSiDIJaoDjy5yMVmVc93Nlm8,974
 agents/crewai_coding_crew/tests/integration/test_agent.py,sha256=0uvfmM3FSdESIU7ICb2oqABsQqylQU2PE3UBwbpif-A,1629
 agents/langgraph_base_react/README.md,sha256=erfhoSbEqbqn1XM_sSjxXyWleBWIyAMuaN5xMIG6f3M,485
 agents/langgraph_base_react/app/agent.py,sha256=2HsbZaSLpMhJQ21geauKPfavcavPW50lx4WV5dXlvnQ,2514
 agents/langgraph_base_react/notebooks/evaluating_langgraph_agent.ipynb,sha256=INlcKJQ_2KB9XAXloG6wKB7cHBKDhW56w1bAQm2IeLA,58048
-agents/langgraph_base_react/template/.templateconfig.yaml,sha256=QfPpUQ1KkZQtQoMK1ktxcieNSvViOCVwKTB5gOtUxr0,416
+agents/langgraph_base_react/template/.templateconfig.yaml,sha256=NpQRZbXtCOeX0XiYtqyI73V6zH4tWWr1TpcTDsSTcjw,991
 agents/langgraph_base_react/tests/integration/test_agent.py,sha256=wg2Y8V4l7dh7rhWy7Z8zioDW5ILRsz7RZJ-mz0AhK3U,1619
-agents/multimodal_live_api/README.md,sha256=eTrjK-xB0yITMTEamUYX3zkGvPHdn7tCbsg52uI1Ces,3001
+agents/multimodal_live_api/README.md,sha256=v_RsHdcmw1ed5jQ65ZgdoDldCsoQuupPTLd9YvG_U8w,3435
 agents/multimodal_live_api/app/agent.py,sha256=WFB7lvDiImvNyWFlds_f5TtzyrbwcquJe6JyJo8HGSs,2690
 agents/multimodal_live_api/app/server.py,sha256=2l7WgNVQAHbnhWUyXBsIt4DI2_kuBoKiEEhcmg1Hom4,6991
 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=tcKgjWka1uLLkzzvDsR6bnEEGvbJqR7HSO5vyYHmBqc,507
+agents/multimodal_live_api/template/.templateconfig.yaml,sha256=gkXZxV15yY6OpJ64VWRhWlCtPhJxyacrv_e2kY3-Wo0,1082
 agents/multimodal_live_api/tests/integration/test_server_e2e.py,sha256=D2VETDIyTD2fQyQ6DXwLr-POSYyVzzzcpIFLt5Ppyx4,8398
 agents/multimodal_live_api/tests/load_test/load_test.py,sha256=HHZyfC4gqiQtZVF_CbbxENGgWQccMLpwMv0IdoQ6cbQ,1275
 agents/multimodal_live_api/tests/unit/test_server.py,sha256=_TjlgQgNkjerIaBGnu8P8_KB8ZlSolDcivALpUOn_Rw,4786
@@ -57,11 +57,11 @@ src/base_template/deployment/terraform/dev/vars/env.tfvars,sha256=xz5cYG29ZIgqM1
 src/base_template/deployment/terraform/vars/env.tfvars,sha256=2dxWRDzeolN2f7eKSHXcBiBt9fRcIMO6VH-2MAZevLQ,1633
 src/base_template/tests/unit/test_utils/test_tracing_exporter.py,sha256=JAb0vIB7wNFPm_kaDaHcxtPKNReypDHjdsMQyzpuePQ,4687
 src/cli/main.py,sha256=134KpcOQlexr_vR-KT3D3B1jKh1Y6S3qHGJ-9uD9Kkg,950
-src/cli/commands/create.py,sha256=y2Zw6Zx6tACWvgBnba55laYHZGGaQ1NGXd6H3iTOuJ8,19717
+src/cli/commands/create.py,sha256=oupbVv8zfMho4JowBCvgr0YpS2I7mYWCDR3We8VxqjI,18497
 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=x-8PKp4aevkUwAmtbEZVwu0UlrA6ifXGbouxICu1D0o,25664
-src/cli/utils/gcp.py,sha256=zs7JJP79IAO5zvUe4Ttwr_qlTC0jtBK87AP6Ug8Nj4Y,4164
+src/cli/utils/gcp.py,sha256=nZP9wBfzJXuATbX-e8gVY9FyF23b3q-bXFosgkW3eBU,4128
 src/cli/utils/logging.py,sha256=0lHe4EPi1A8sOx9xkA7gS4UNl0GsIyp2ahydkkuCzLY,1570
 src/cli/utils/template.py,sha256=p7uKv5yHkXIOWo8EJJ0IjUh57lOSDL7kYHA9OZ7j-z4,25447
 src/data_ingestion/README.md,sha256=QlomTIZINvZxf2ftCkm2_QVozlivJqAfFKyfiGMZDpY,4217
@@ -79,11 +79,6 @@ src/deployment_targets/agent_engine/tests/integration/test_agent_engine_app.py,s
 src/deployment_targets/agent_engine/tests/load_test/README.md,sha256=ckP2eu5_HpOjM390KXNwUk4YWAzWqir3YOTceyoeL64,1441
 src/deployment_targets/agent_engine/tests/load_test/load_test.py,sha256=msUkkqsNinCJ8loQHsEeGkQQqQ-Zvv7oAp5ni9vOsWs,3541
 src/deployment_targets/agent_engine/tests/load_test/.results/.placeholder,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/deployment_targets/agent_engine/tests/load_test/.results/report.html,sha256=QVJtVS8wsFhWltwaH1v2JSkdS6q4rdSs_DuaXEfYu9I,1475308
-src/deployment_targets/agent_engine/tests/load_test/.results/results_exceptions.csv,sha256=PnPo9mVzHuTkWmqRICoxsGmZ8sUyQPFK7iUy4OC0Kwk,31
-src/deployment_targets/agent_engine/tests/load_test/.results/results_failures.csv,sha256=7K2uMJ7c4_mPrMYsMQE9JyDAUClzKhuz2PKIY5_8NzY,31
-src/deployment_targets/agent_engine/tests/load_test/.results/results_stats.csv,sha256=IOlGa4JOaOOKjMjICLFvd5k80pUHzY6eYe38dF2-ljY,588
-src/deployment_targets/agent_engine/tests/load_test/.results/results_stats_history.csv,sha256=2WjXHaQb17ieFrcVZo-P94qpjSELzt1tAJdiPa6MBeI,3794
 src/deployment_targets/agent_engine/tests/unit/test_dummy.py,sha256=hL-JCmoQUJ4IgluliEo3KkYd47nmiCy3obiPuz_b0sU,712
 src/deployment_targets/cloud_run/Dockerfile,sha256=JTh6P_RXE3bkdsHS0zIgguDVVf0XyLViMcb-UNT3Da4,835
 src/deployment_targets/cloud_run/uv.lock,sha256=2RYrR89xDS1FHVb-g2JGZq8zZ-rzbm_jRvehVCYuXRQ,887957
@@ -155,8 +150,8 @@ src/resources/setup_cicd/providers.tf,sha256=Km4z6IJt7x7PLaa0kyZbBrO2m3lpuIJZFD5
 src/utils/generate_locks.py,sha256=xu5IAhGGBPkVQGSJX4kk7_JNDwWJUEaXAHbmaQIohbg,4386
 src/utils/lock_utils.py,sha256=plw3vWkRViCMxBK3IrLRHtF_I0kbbP043T9KIYxcb20,2423
 src/utils/watch_and_rebuild.py,sha256=IdbTLReYknTXXu5KWPz3sc7Zg5HTEMS5UVlMSnuhs5E,6067
-agent_starter_pack-0.0.3b0.dist-info/METADATA,sha256=qWWhxf8uhoeWK6SYIgWnYzCqm2aOhl79T5jCBe4E_c8,6805
-agent_starter_pack-0.0.3b0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.0.3b0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.0.3b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.0.3b0.dist-info/RECORD,,
+agent_starter_pack-0.0.5b0.dist-info/METADATA,sha256=bPlCq8nyiNrEuW_3fNR_i3LGViTs0MWwK9-qp_OL6PY,6805
+agent_starter_pack-0.0.5b0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.0.5b0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.0.5b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.0.5b0.dist-info/RECORD,,

agents/agentic_rag_vertexai_search/template/.templateconfig.yaml

@@ -1,3 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 description: "A RAG agent using Vertex AI Search and LangGraph for document retrieval and Q&A"
 settings:
   requires_data_ingestion: true

agents/crewai_coding_crew/template/.templateconfig.yaml

@@ -1,3 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 description: "A multi-agent system implemented with CrewAI created to support coding activities"
 settings:
   requires_data_ingestion: false

agents/langgraph_base_react/template/.templateconfig.yaml

@@ -1,3 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 description: "A agent implementing a base ReAct agent using LangGraph"
 settings:
   requires_data_ingestion: false

agents/multimodal_live_api/README.md

@@ -28,16 +28,32 @@ This pattern showcases a real-time conversational RAG agent powered by Google Ge
 
 2.  **Start the Backend and Frontend:**
 
-    Start the backend:
+    **Backend:**
     ```bash
     make backend
     ```
+    
+    The backend will be ready when you see `INFO:     Application startup complete.` in the console.
+
+    <details>
+    <summary><b>Click here if you want to use AI Studio and API Key instead of Vertex AI:</b></summary>
 
-    In a different shell, start the frontend:
     ```bash
+    export VERTEXAI=false
+    export GOOGLE_API_KEY=your-google-api-key
+    ```
+
+    </details>
+    <br>
+    
+    **Frontend:**
+    ```bash
+    # In a different shell
     make ui
     ```
 
+    This is the suggested mode for development as it allows you to see changes in real-time.
+
 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](https://cloud.google.com/architecture/deploy-operate-generative-ai-applications) it was provided to.
 
 ## Additional Resources for Multimodal Live API

agents/multimodal_live_api/template/.templateconfig.yaml

@@ -1,3 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 description: "A real-time multimodal RAG agent powered by Gemini, supporting audio/video/text chat with vector DB-backed responses"
 settings:
   requires_data_ingestion: false

src/cli/commands/create.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import logging
+import os
 import pathlib
 import subprocess
 
@@ -47,8 +48,6 @@ console = Console()
 @click.option(
     "--include-data-ingestion", "-i", is_flag=True, help="Include data pipeline"
 )
-@click.option("--gcp-account", help="GCP service account email")
-@click.option("--gcp-project", help="GCP project ID")
 @click.option("--debug", is_flag=True, help="Enable debug logging")
 @click.option(
     "--output-dir",
@@ -77,8 +76,6 @@ def create(
     agent: str | None,
     deployment_target: str | None,
     include_data_ingestion: bool,
-    gcp_account: str | None,
-    gcp_project: str | None,
     debug: bool,
     output_dir: str | None,
     auto_approve: bool,
@@ -164,88 +161,23 @@ def create(
             logging.debug(f"Selected region: {region}")
 
         # GCP Setup
-        with console.status("[bold blue]Setting up GCP environment...", spinner="dots"):
-            if debug:
-                logging.debug("Setting up GCP...")
-
-        # Handle GCP credentials
-        if gcp_account and gcp_project:
-            try:
-                subprocess.run(
-                    ["gcloud", "config", "set", "account", gcp_account], check=True
-                )
-                subprocess.run(
-                    ["gcloud", "config", "set", "project", gcp_project], check=True
-                )
-                subprocess.run(
-                    [
-                        "gcloud",
-                        "auth",
-                        "application-default",
-                        "set-quota-project",
-                        gcp_project,
-                    ],
-                    check=True,
-                )
-            except subprocess.CalledProcessError as e:
-                console.print(f"Error setting GCP credentials: {e!s}", style="bold red")
-                raise
+        logging.debug("Setting up GCP...")
 
+        # Check for uv installation if not skipping checks
         if not skip_checks:
-            # Verify GCP credentials
-            if debug:
-                logging.debug("Verifying GCP credentials...")
-            creds_info = verify_credentials()
-
-            if not auto_approve:
-                change_creds = (
-                    Prompt.ask(
-                        f"\n> You are logged in with account '{creds_info['account']}' "
-                        f"and using project '{creds_info['project']}'. "
-                        "Is this correct?",
-                        choices=["Y", "n"],
-                        default="Y",
-                    ).lower()
-                    == "n"
-                )
-
-                if change_creds:
-                    handle_credential_change()
-            else:
-                console.print(
-                    f"\n> Using account '{creds_info['account']}' "
-                    f"with project '{creds_info['project']}'"
-                )
-
-            # Check for uv installation
-            console.print("> Checking for uv installation...")
             check_and_install_uv()
 
-            # Test Vertex AI connection
-            console.print("> Testing GCP and Vertex AI Connection...")
+            # Set up GCP environment
             try:
-                verify_vertex_connection(
-                    project_id=creds_info["project"],
-                    location=region,
-                )
-                console.print(
-                    f"> ✓ Successfully verified connection to Vertex AI in project {creds_info['project']}"
+                setup_gcp_environment(
+                    auto_approve=auto_approve,
+                    skip_checks=skip_checks,
+                    region=region,
+                    debug=debug,
                 )
             except Exception as e:
-                console.print(
-                    f"> ✗ Failed to connect to Vertex AI: {e!s}\n"
-                    f"> Please check your authentication settings and permissions. "
-                    f"Visit https://cloud.google.com/vertex-ai/docs/authentication for help.", 
-                    style="bold red"
-                )
-        else:
-            if debug:
-                logging.debug("Skipping verification checks due to --skip-checks flag")
-            console.print("> Skipping verification checks", style="yellow")
-            # Set a default creds_info when skipping checks
-            creds_info = (
-                {"project": gcp_project} if gcp_project else {"project": "unknown"}
-            )
+                if debug:
+                    logging.warning(f"GCP environment setup failed: {e}")
 
         # Process template
         template_path = get_template_path(final_agent, debug=debug)
@@ -332,74 +264,172 @@ def display_agent_selection(deployment_target: str | None = None) -> str:
     return agents[choice]["name"]
 
 
-def handle_credential_change() -> None:
-    """Handle the process of changing GCP credentials."""
+def set_gcp_project(project_id: str, set_quota_project: bool = True) -> None:
+    """Set the GCP project and optionally the application default quota project.
+
+    Args:
+        project_id: The GCP project ID to set.
+        set_quota_project: Whether to set the application default quota project.
+    """
     try:
+        subprocess.run(
+            ["gcloud", "config", "set", "project", project_id],
+            check=True,
+            capture_output=True,
+            text=True,
+        )
+    except subprocess.CalledProcessError as e:
+        console.print(f"\n> Error setting project to {project_id}:")
+        console.print(e.stderr)
+        raise
+
+    if set_quota_project:
+        try:
+            subprocess.run(
+                [
+                    "gcloud",
+                    "auth",
+                    "application-default",
+                    "set-quota-project",
+                    project_id,
+                ],
+                check=True,
+                capture_output=True,
+                text=True,
+            )
+        except subprocess.CalledProcessError as e:
+            console.print("> Error setting application default quota project:")
+            console.print(e.stderr)
+            raise
+
+    console.print(f"> Successfully configured project: {project_id}")
+
+
+def setup_gcp_environment(
+    auto_approve: bool, skip_checks: bool, region: str, debug: bool
+) -> dict:
+    """Set up the GCP environment with proper credentials and project.
+
+    Args:
+        auto_approve: Whether to skip confirmation prompts
+        skip_checks: Whether to skip verification checks
+        region: GCP region for deployment
+        debug: Whether debug logging is enabled
+
+    Returns:
+        Dictionary with credential information
+    """
+    # Skip all verification if requested
+    if skip_checks:
+        if debug:
+            logging.debug("Skipping verification checks due to --skip-checks flag")
+        console.print("> Skipping verification checks", style="yellow")
+        return {"project": "unknown"}
+
+    # Verify current GCP credentials
+    if debug:
+        logging.debug("Verifying GCP credentials...")
+    creds_info = verify_credentials()
+
+    # Handle credential verification and project selection
+    if not auto_approve:
+        creds_info = _handle_credential_verification(creds_info)
+    else:
+        # Even with auto_approve, 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(creds_info["project"], region)
+
+    return creds_info
+
+
+def _handle_credential_verification(creds_info: dict) -> dict:
+    """Handle verification of credentials and project selection.
+
+    Args:
+        creds_info: Current credential information
+
+    Returns:
+        Updated credential information
+    """
+    # Check if running in Cloud Shell
+    if os.environ.get("CLOUD_SHELL") == "true":
+        if creds_info["project"] == "":
+            console.print("> It looks like you are running in Cloud Shell.")
+            console.print(
+                "> You need to set up a project ID to continue, but you haven't setup a project yet."
+            )
+            new_project = Prompt.ask("\n> Enter a project ID")
+            creds_info["project"] = new_project
+            set_gcp_project(creds_info["project"], set_quota_project=False)
+        return creds_info
+
+    # Ask user if current credentials are correct
+    change_creds = (
+        Prompt.ask(
+            f"\n> You are logged in with account '{creds_info['account']}' "
+            f"and using project '{creds_info['project']}'. "
+            "Is this correct?",
+            choices=["Y", "n"],
+            default="Y",
+        ).lower()
+        == "n"
+    )
+
+    if change_creds:
+        # Handle credential change
         console.print("\n> Initiating new login...")
         subprocess.run(["gcloud", "auth", "login", "--update-adc"], check=True)
         console.print("> Login successful. Verifying new credentials...")
 
         # Re-verify credentials after login
-        new_creds_info = verify_credentials()
+        creds_info = verify_credentials()
 
         # Prompt for project change
         change_project = (
             Prompt.ask(
-                f"\n> You are now logged in with account '{new_creds_info['account']}'. "
-                f"Current project is '{new_creds_info['project']}'. "
-                "Do you wish to change the project?",
+                f"\n> You are now logged in with account '{creds_info['account']}'. "
+                f"Current project is '{creds_info['project']}'. "
+                "Is this correct?",
                 choices=["y", "n"],
-                default="n",
+                default="y",
             ).lower()
-            == "y"
+            == "n"
         )
 
         if change_project:
-            handle_project_change()
-
-    except subprocess.CalledProcessError as e:
-        console.print(
-            "\n> Error during login process. Please try again.", style="bold red"
-        )
-        logging.debug(f"Login error: {e!s}")
-        raise
-    except Exception as e:
-        console.print(f"\n> Unexpected error: {e!s}", style="bold red")
-        logging.debug(f"Unexpected error during login: {e!s}")
-        raise
+            # Prompt for new project ID
+            new_project = Prompt.ask("\n> Enter the new project ID")
+            creds_info["project"] = new_project
 
+    set_gcp_project(creds_info["project"], set_quota_project=True)
+    return creds_info
 
-def handle_project_change() -> None:
-    """Handle the process of changing GCP project."""
-    try:
-        # Prompt for new project ID
-        new_project = Prompt.ask("\n> Enter the new project ID")
 
-        console.print(f"\n> Setting project to {new_project}...")
-        subprocess.run(["gcloud", "config", "set", "project", new_project], check=True)
+def _test_vertex_ai_connection(project_id: str, region: str) -> None:
+    """Test connection to Vertex AI.
 
-        console.print("> Setting application default quota project...")
-        subprocess.run(
-            ["gcloud", "auth", "application-default", "set-quota-project", new_project],
-            check=True,
+    Args:
+        project_id: GCP project ID
+        region: GCP region for deployment
+    """
+    console.print("> Testing GCP and Vertex AI Connection...")
+    try:
+        verify_vertex_connection(
+            project_id=project_id,
+            location=region,
         )
-
-        console.print(f"> Successfully switched to project: {new_project}")
-
-        # Re-verify credentials one final time
-        final_creds_info = verify_credentials()
         console.print(
-            f"\n> Now using account '{final_creds_info['account']}' "
-            f"with project '{final_creds_info['project']}'"
+            f"> ✓ Successfully verified connection to Vertex AI in project {project_id}"
         )
-
-    except subprocess.CalledProcessError as e:
+    except Exception as e:
         console.print(
-            "\n> Error while changing project. Please verify the project ID and try again.",
+            f"> ✗ Failed to connect to Vertex AI: {e!s}\n"
+            f"> Please check your authentication settings and permissions. "
+            f"Visit https://cloud.google.com/vertex-ai/docs/authentication for help.",
             style="bold red",
         )
-        logging.debug(f"Project change error: {e!s}")
-        raise
 
 
 def replace_region_in_files(
@@ -485,11 +515,11 @@ def replace_region_in_files(
 
 
 def check_and_install_uv() -> None:
-    """Check if uv is installed and install it if not present."""
+    """Check if uv is installed and inform the user if it's not."""
     try:
         # Use shell=True for Windows compatibility and add timeout
         process = subprocess.run(
-            "uv --version", shell=True, capture_output=True, text=True, timeout=10
+            "uv --version", shell=True, capture_output=True, text=True, timeout=5
         )
         if process.returncode == 0:
             logging.debug("uv is already installed")
@@ -502,43 +532,6 @@ def check_and_install_uv() -> None:
         console.print("> uv is not installed", style="yellow")
 
     console.print(
-        "\n> uv is required for the template to work. You can install it in several ways:"
-        "\n 1. Automatically install uv now"
-        "\n 2. Manual installation from: https://docs.astral.sh/uv/getting-started/installation"
-        "\n    This includes other options e.g PyPI (pip/pipx), Homebrew, Docker.."
-    )
-
-    install_choice = Prompt.ask(
-        "\n> Would you like to install uv automatically now?",
-        choices=["y", "n"],
-        default="y",
+        "\n> uv is recommended for managing virtual environments and dependencies."
+        "\n> For installation instructions, visit: https://docs.astral.sh/uv/getting-started/installation"
     )
-    if install_choice.lower() == "y":
-        console.print("> Installing uv...", style="yellow")
-        try:
-            install_command = "curl -LsSf https://astral.sh/uv/install.sh | sh"
-            # Add timeout to installation process as well
-            subprocess.run(
-                install_command,
-                shell=True,
-                check=True,
-                timeout=60,  # Give installation more time
-            )
-            console.print("> uv installed successfully!", style="green")
-        except subprocess.TimeoutExpired:
-            console.print("> uv installation timed out", style="bold red")
-            raise
-        except subprocess.CalledProcessError as e:
-            console.print("> Failed to install uv", style="bold red")
-            logging.debug(f"uv installation error: {e!s}")
-            console.print(
-                "> Please install uv manually using one of the methods listed above",
-                style="yellow",
-            )
-            raise
-    else:
-        console.print(
-            "> Please install uv manually using one of the methods listed above and try again",
-            style="yellow",
-        )
-        raise click.Abort() from None

src/cli/utils/gcp.py

@@ -24,7 +24,7 @@ from google.cloud.aiplatform_v1beta1.services.prediction_service import (
     PredictionServiceClient,
 )
 from google.cloud.aiplatform_v1beta1.types.prediction_service import (
-    GenerateContentRequest,
+    CountTokensRequest,
 )
 
 
@@ -43,12 +43,11 @@ def get_client_info() -> ClientInfo:
     return ClientInfo(client_library_version=version, user_agent=agent)
 
 
-def get_dummy_request(project_id: str, location: str) -> GenerateContentRequest:
+def get_dummy_request(project_id: str, location: str) -> CountTokensRequest:
     """Creates a simple test request for Gemini."""
-    return GenerateContentRequest(
-        contents=[{"role": "user", "parts": [{"text": "Hi there!"}]}],
-        generation_config={"candidate_count": 1},
-        model=f"projects/{project_id}/locations/{location}/publishers/google/models/gemini-2.0-flash-001",
+    return CountTokensRequest(
+        contents=[{"role": "user", "parts": [{"text": "Hi"}]}],
+        endpoint=f"projects/{project_id}/locations/{location}/publishers/google/models/gemini-1.5-flash-002",
     )
 
 
@@ -64,10 +63,10 @@ def verify_vertex_connection(
             api_endpoint=f"{location}-aiplatform.googleapis.com"
         ),
         client_info=get_client_info(),
-        transport="rest",
+        transport=initializer.global_config._api_transport,
     )
     request = get_dummy_request(project_id=project_id, location=location)
-    client.generate_content(request=request)
+    client.count_tokens(request=request)
 
 
 def verify_credentials() -> dict[str, str]: