agent-starter-pack 0.0.1b0__tar.gz → 0.0.3b0__tar.gz

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/PKG-INFO

@@ -1,18 +1,18 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.0.1b0
+Version: 0.0.3b0
 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
 License-File: LICENSE
 Requires-Python: >=3.10
-Requires-Dist: click>=8.1.7
-Requires-Dist: cookiecutter>=2.5.0
-Requires-Dist: google-cloud-aiplatform>=1.38.1
-Requires-Dist: inquirer>=3.1.3
-Requires-Dist: pyyaml>=6.0.1
-Requires-Dist: rich>=13.7.0
-Requires-Dist: uv>=0.1.0
+Requires-Dist: click~=8.1.7
+Requires-Dist: cookiecutter~=2.5.0
+Requires-Dist: google-cloud-aiplatform~=1.81.0
+Requires-Dist: inquirer~=3.1.3
+Requires-Dist: pyyaml~=6.0.1
+Requires-Dist: rich~=13.7.0
+Requires-Dist: uv~=0.6.3
 Provides-Extra: jupyter
 Requires-Dist: ipykernel>=6.29.5; extra == 'jupyter'
 Requires-Dist: jupyter; extra == 'jupyter'
@@ -34,7 +34,7 @@ It accelerates development by providing a holistic, production-ready solution, a
 
 | ⚡️ Launch | 🧪 Experiment  | ✅ Deploy | 🛠️ Customize |
 |---|---|---|---|
-| Pre-built agent templates (ReAct, RAG, multi-agent, Live Multimodal API). | Vertex AI evaluation and an interactive playground. | Production-ready infrastructure with monitoring, observability, and CI/CD. | Extend and customize templates. |
+| [Pre-built agent templates](./agents/) (ReAct, RAG, multi-agent, Live Multimodal API). | [Vertex AI evaluation]() and an interactive playground. | Production-ready infra with [monitoring](./docs/monitoring.md), [observability](./docs/observability.md), and [CI/CD](./docs/deployment.md) on [Cloud Run](https://cloud.google.com/run) or [Agent Engine](https://cloud.google.com/vertex-ai/generative-ai/docs/reasoning-engine/overview). | Extend and customize templates according to your needs. |
 
 ---
  
@@ -43,10 +43,13 @@ It accelerates development by providing a holistic, production-ready solution, a
 Ready to build your AI agent? Simply run this command:
 
 ```bash
-# 1. Install the Agent Starter Pack CLI
+# Create and activate a Python virtual environment
+python -m venv venv && source venv/bin/activate
+
+# Install the agent starter pack
 pip install agent-starter-pack
 
-# 2. Create a new agent project (choose from available templates)
+# Create a new agent project
 agent-starter-pack create my-awesome-agent
 ```
 
@@ -105,7 +108,6 @@ This project represents the next evolution of the [e2e-gen-ai-app-starter-pack](
 
 - Python 3.10+
 - [Google Cloud SDK](https://cloud.google.com/sdk/docs/install)
-- [uv](https://github.com/astral-sh/uv)
 - [Terraform](https://developer.hashicorp.com/terraform/downloads) (for deployment)
 
 

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/README.md

@@ -8,7 +8,7 @@ It accelerates development by providing a holistic, production-ready solution, a
 
 | ⚡️ Launch | 🧪 Experiment  | ✅ Deploy | 🛠️ Customize |
 |---|---|---|---|
-| Pre-built agent templates (ReAct, RAG, multi-agent, Live Multimodal API). | Vertex AI evaluation and an interactive playground. | Production-ready infrastructure with monitoring, observability, and CI/CD. | Extend and customize templates. |
+| [Pre-built agent templates](./agents/) (ReAct, RAG, multi-agent, Live Multimodal API). | [Vertex AI evaluation]() and an interactive playground. | Production-ready infra with [monitoring](./docs/monitoring.md), [observability](./docs/observability.md), and [CI/CD](./docs/deployment.md) on [Cloud Run](https://cloud.google.com/run) or [Agent Engine](https://cloud.google.com/vertex-ai/generative-ai/docs/reasoning-engine/overview). | Extend and customize templates according to your needs. |
 
 ---
  
@@ -17,10 +17,13 @@ It accelerates development by providing a holistic, production-ready solution, a
 Ready to build your AI agent? Simply run this command:
 
 ```bash
-# 1. Install the Agent Starter Pack CLI
+# Create and activate a Python virtual environment
+python -m venv venv && source venv/bin/activate
+
+# Install the agent starter pack
 pip install agent-starter-pack
 
-# 2. Create a new agent project (choose from available templates)
+# Create a new agent project
 agent-starter-pack create my-awesome-agent
 ```
 
@@ -79,7 +82,6 @@ This project represents the next evolution of the [e2e-gen-ai-app-starter-pack](
 
 - Python 3.10+
 - [Google Cloud SDK](https://cloud.google.com/sdk/docs/install)
-- [uv](https://github.com/astral-sh/uv)
 - [Terraform](https://developer.hashicorp.com/terraform/downloads) (for deployment)
 
 

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/docs/cli/setup_cicd.md

@@ -1,4 +1,3 @@
-
 ## ⚙️ CI/CD Setup (Experimental)
 
 This starter pack provides an *experimental* command to automate the setup of a basic CI/CD pipeline.  This pipeline connects your agent to a GitHub repository and uses Google Cloud Build for automated testing and deployment.
@@ -7,7 +6,7 @@ This starter pack provides an *experimental* command to automate the setup of a
 
 *   **Experimental Feature:** The `setup-cicd` command is under active development.  Expect potential changes and report any issues you encounter.
 *   **Production Readiness:**  For production deployments, we *strongly* recommend following the manual setup instructions in the `deployment/README.md` file.  The manual approach provides greater control over security, customization, and environment-specific configurations.  This automated setup is primarily for development and testing.
-*  **GitHub Only:** This command is works with Github.
+*   **Git Provider:** Currently, only GitHub is supported, but support for other Git providers is planned.
 
 ### Quick CI/CD Setup (Automated)
 
@@ -15,12 +14,11 @@ This command streamlines the CI/CD setup process.  It handles:
 
 1.  **GitHub Repository:** Creates a new GitHub repository (or connects to an existing *empty* one).
 2.  **Cloud Build Connection:** Establishes a connection between your GitHub repository and Google Cloud Build.
-3.  **Development Environment:**  Sets up the infrastructure for your development environment using Terraform.
+3.  **Development Environment:**  Sets up the infrastructure for your development environment using Terraform (optional).
 4.  **CI/CD Triggers:** Configures basic Cloud Build triggers for pull request checks and deployments.
 5.  **Terraform State:**  By default, configures remote Terraform state management using a Google Cloud Storage (GCS) bucket.  You can opt for local state management if needed.
 
 **Command Usage:**
-
 ```bash
 agent-starter-pack setup-cicd \
     --dev-project <YOUR_DEV_PROJECT_ID> \
@@ -40,10 +38,10 @@ agent-starter-pack setup-cicd \
 
 **Options:**
 
-*   `--dev-project`:  (Optional) The Google Cloud project ID for your development environment.  If provided, the setup will also configure a development environment.
 *   `--staging-project`:  **Required.** The Google Cloud project ID for your staging environment.
 *   `--prod-project`:  **Required.** The Google Cloud project ID for your production environment.
-*   `--cicd-project`:  **Required.** The Google Cloud project ID where your CI/CD resources (Cloud Build, etc.) will reside.
+*   `--cicd-project`:  **Required.** The Google Cloud project ID where your CI/CD resources (Cloud Build, etc.) will reside. This can be the same as your staging or production project.
+*   `--dev-project`:  (Optional) The Google Cloud project ID for your development environment.  If provided, the setup will also configure a development environment.
 *   `--region`:  The GCP region to use (default: `us-central1`).
 *   `--repository-name`:  (Optional) The name of your GitHub repository.  If omitted, a name will be generated (e.g., `genai-app-1678886400`).
 *   `--repository-owner`: (Optional) Your GitHub username or organization name.  If omitted, it defaults to your currently authenticated GitHub user.
@@ -54,6 +52,10 @@ agent-starter-pack setup-cicd \
 *   `--debug`:  Enable debug logging for more verbose output.
 *   `--auto-approve`:  Skip interactive confirmation prompts.  Use with caution!
 
+**Project Requirements:**
+
+You need at least *two* Google Cloud projects: one for staging and one for production. The CI/CD project can be the same as either the staging or production project. A separate development project is optional.
+
 **Interactive Mode vs. Programmatic Mode:**
 
 The command operates in two modes:
@@ -95,4 +97,4 @@ For fine-grained control and production deployments, refer to the detailed instr
 *   **Enhanced Security:**  Configure stricter IAM permissions and network policies.
 *   **Custom Workflows:**  Implement custom build, test, and deployment steps.
 *   **Environment-Specific Settings:**  Tailor configurations for each environment (dev, staging, prod).
-*   **Advanced CI/CD:**  Integrate with other CI/CD tools and services.
+*   **Advanced CI/CD:**  Integrate with other CI/CD tools and services.

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.0.1b"
+version = "0.0.3b"
 description = "CLI tool to create GCP-based AI agent projects from templates"
 authors = [
     { name = "Google LLC", email = "agent-starter-pack@google.com" },
@@ -9,41 +9,41 @@ license = {text = "Apache-2.0"}
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "click>=8.1.7",
-    "cookiecutter>=2.5.0",
-    "google-cloud-aiplatform>=1.38.1",
-    "inquirer>=3.1.3",
-    "rich>=13.7.0",
-    "uv>=0.1.0",
-    "pyyaml>=6.0.1",
+    "click~=8.1.7",
+    "cookiecutter~=2.5.0",
+    "google-cloud-aiplatform~=1.81.0",
+    "inquirer~=3.1.3",
+    "rich~=13.7.0",
+    "uv~=0.6.3",
+    "pyyaml~=6.0.1",
 ]
 
 [dependency-groups]
 dev = [
-    "pytest>=7.4.3",
-    "pytest-mock>=3.12.0",
-    "pytest-cov>=4.1.0",
-    "black>=23.11.0",
-    "isort>=5.12.0",
-    "flake8>=6.1.0",
-    "watchdog>=6.0.0",
-    "sphinx>=7.1.2",
-    "sphinx-click>=5.1.0", 
-    "sphinx-rtd-theme>=2.0.0",
-    "sphinx-autoapi>=3.0.0",
-    "myst-parser>=2.0.0",
-    "rich>=13.7.0",
-    "streamlit>=1.42.0",
-    "langchain-core>=0.3.34",
-    "langchain-google-vertexai>=2.0.13",
-    "streamlit-feedback>=0.1.4",
-    "grpcio>=1.70.0",
-    "pytest-xdist>=3.6.1",
-    "langgraph>=0.2.72",
-    "crewai>=0.95.0",
-    "opentelemetry-instrumentation-langchain>=0.38.5",
-    "langchain-google-community[vertexaisearch]>=2.0.4",
-    "google-cloud-logging>=3.11.4",
+    "black~=23.11.0",
+    "crewai~=0.95.0",
+    "flake8~=6.1.0",
+    "google-cloud-logging~=3.11.4",
+    "grpcio~=1.70.0",
+    "isort~=5.12.0",
+    "langchain-core~=0.3.34",
+    "langchain-google-community[vertexaisearch]~=2.0.4",
+    "langchain-google-vertexai~=2.0.13",
+    "langgraph~=0.2.72",
+    "myst-parser~=2.0.0",
+    "opentelemetry-instrumentation-langchain~=0.38.5",
+    "pytest~=7.4.3",
+    "pytest-cov~=4.1.0",
+    "pytest-mock~=3.12.0",
+    "pytest-xdist~=3.6.1",
+    "rich~=13.7.0",
+    "sphinx~=7.1.2",
+    "sphinx-autoapi~=3.0.0",
+    "sphinx-click~=5.1.0",
+    "sphinx-rtd-theme~=2.0.0",
+    "streamlit~=1.42.0",
+    "streamlit-feedback~=0.1.4",
+    "watchdog~=6.0.0",
 ]
 
 [project.optional-dependencies]

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/base_template/deployment/terraform/dev/vars/env.tfvars

@@ -6,7 +6,7 @@ region = "us-central1"
 
 telemetry_bigquery_dataset_id = "telemetry_genai_app_sample_sink"
 telemetry_sink_name = "telemetry_logs_genai_app_sample"
-telemetry_logs_filter = "jsonPayload.attributes.\"traceloop.association.properties.log_type\"=\"tracing\" jsonPayload.resource.attributes.\"service.name\"=\"Sample Chatbot Application\""
+telemetry_logs_filter = "jsonPayload.attributes.\"traceloop.association.properties.log_type\"=\"tracing\" jsonPayload.resource.attributes.\"service.name\"=\"{{cookiecutter.project_name}}\""
 
 feedback_bigquery_dataset_id = "feedback_genai_app_sample_sink"
 feedback_sink_name = "feedback_logs_genai_app_sample"

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/base_template/deployment/terraform/vars/env.tfvars

@@ -22,7 +22,7 @@ cloud_run_app_sa_name = "{{cookiecutter.project_name}}-cr"
 
 telemetry_bigquery_dataset_id = "telemetry_genai_app_sample_sink"
 telemetry_sink_name = "telemetry_logs_genai_app_sample"
-telemetry_logs_filter = "jsonPayload.attributes.\"traceloop.association.properties.log_type\"=\"tracing\" jsonPayload.resource.attributes.\"service.name\"=\"Sample Chatbot Application\""
+telemetry_logs_filter = "jsonPayload.attributes.\"traceloop.association.properties.log_type\"=\"tracing\" jsonPayload.resource.attributes.\"service.name\"=\"{{cookiecutter.project_name}}\""
 
 feedback_bigquery_dataset_id = "feedback_genai_app_sample_sink"
 feedback_sink_name = "feedback_logs_genai_app_sample"

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/base_template/pyproject.toml

@@ -14,11 +14,11 @@ dependencies = [
     "traceloop-sdk~=0.38.7",
     "google-cloud-logging~=3.11.4",
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
-    "google-cloud-aiplatform[evaluation]~=1.80.0",
+    "google-cloud-aiplatform[evaluation]~=1.81.0",
     "fastapi~=0.115.8",
     "uvicorn~=0.34.0"
 {%- elif cookiecutter.deployment_target == 'agent_engine' %}
-    "google-cloud-aiplatform[evaluation,langchain,reasoningengine]~=1.77.0"
+    "google-cloud-aiplatform[evaluation,langchain,reasoningengine]~=1.81.0"
 {%- endif %}
 ]
 {% if cookiecutter.deployment_target == 'cloud_run' %}

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/cli/commands/create.py

@@ -149,11 +149,16 @@ def create(
             logging.debug(f"Selected deployment target: {final_deployment}")
 
         # Data pipeline selection
-        include_data_ingestion = include_data_ingestion or prompt_data_ingestion(final_agent)
+        include_data_ingestion = include_data_ingestion or prompt_data_ingestion(
+            final_agent
+        )
         if debug:
             logging.debug(f"Include data pipeline: {include_data_ingestion}")
         # Region confirmation (if not explicitly passed)
-        if not auto_approve and ctx.get_parameter_source("region") != ParameterSource.COMMANDLINE:
+        if (
+            not auto_approve
+            and ctx.get_parameter_source("region") != ParameterSource.COMMANDLINE
+        ):
             region = prompt_region_confirmation(region)
         if debug:
             logging.debug(f"Selected region: {region}")
@@ -197,11 +202,11 @@ def create(
                     Prompt.ask(
                         f"\n> You are logged in with account '{creds_info['account']}' "
                         f"and using project '{creds_info['project']}'. "
-                        "Do you wish to change this?",
-                        choices=["y", "n"],
-                        default="n",
+                        "Is this correct?",
+                        choices=["Y", "n"],
+                        default="Y",
                     ).lower()
-                    == "y"
+                    == "n"
                 )
 
                 if change_creds:
@@ -215,27 +220,32 @@ def create(
             # 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...")
+            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']}"
+                )
+            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"}
-
-        console.print("> Testing GCP and Vertex AI Connection...")
-        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']}"
-            )
-        except Exception as e:
-            console.print(
-                f"> ✗ Failed to connect to Vertex AI: {e!s}", style="bold red"
+            creds_info = (
+                {"project": gcp_project} if gcp_project else {"project": "unknown"}
             )
-            raise
 
         # Process template
         template_path = get_template_path(final_agent, debug=debug)
@@ -263,10 +273,14 @@ def create(
         console.print("\n> 👍 Done. Execute the following command to get started:")
         if output_dir:
             # If output_dir was specified, use the absolute path
-            console.print(f"cd {project_path} && make install && make playground")
+            console.print(
+                f"[bold bright_green]cd {project_path} && make install && make playground[/]"
+            )
         else:
             # If no output_dir specified, just use project name
-            console.print(f"cd {project_name} && make install && make playground")
+            console.print(
+                f"[bold bright_green]cd {project_name} && make install && make playground[/]"
+            )
 
         # Replace region in all files if a different region was specified
         if region != "us-central1":
@@ -475,11 +489,7 @@ def check_and_install_uv() -> None:
     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=10
         )
         if process.returncode == 0:
             logging.debug("uv is already installed")
@@ -512,7 +522,7 @@ def check_and_install_uv() -> None:
                 install_command,
                 shell=True,
                 check=True,
-                timeout=60  # Give installation more time
+                timeout=60,  # Give installation more time
             )
             console.print("> uv installed successfully!", style="green")
         except subprocess.TimeoutExpired:

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/cli/commands/setup_cicd.py

@@ -31,6 +31,8 @@ from src.cli.utils.cicd import (
     create_github_connection,
     create_github_repository,
     ensure_apis_enabled,
+    handle_github_authentication,
+    is_github_authenticated,
     print_cicd_summary,
     run_command,
 )
@@ -84,7 +86,7 @@ def setup_git_repository(config: ProjectConfig) -> str:
 
     # Initialize git if not already initialized
     if not (Path.cwd() / ".git").exists():
-        run_command(["git", "init"])
+        run_command(["git", "init", "-b", "main"])
         console.print("✅ Git repository initialized")
 
     # Get current GitHub username for the remote URL
@@ -470,6 +472,14 @@ def setup_cicd(
     if not git_provider:
         git_provider = prompt_for_git_provider()
 
+    # Check GitHub authentication if GitHub is selected
+    if git_provider == "github" and not (github_pat and github_app_installation_id):
+        if not is_github_authenticated():
+            console.print("\n⚠️ Not authenticated with GitHub CLI", style="yellow")
+            handle_github_authentication()
+        else:
+            console.print("✅ GitHub CLI authentication verified")
+
     # Only prompt for repository details if not provided via CLI
     if not (repository_name and repository_owner):
         repository_name, repository_owner = prompt_for_repository_details(
@@ -676,7 +686,7 @@ def setup_cicd(
 
         # Initialize git if not already initialized
         if not (Path.cwd() / ".git").exists():
-            run_command(["git", "init"])
+            run_command(["git", "init", "-b", "main"])
             console.print("✅ Git repository initialized")
 
         # Get current GitHub username for the remote URL

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/cli/utils/cicd.py

@@ -22,6 +22,7 @@ from dataclasses import dataclass
 from enum import Enum
 from pathlib import Path
 
+import click
 from rich.console import Console
 from rich.prompt import IntPrompt, Prompt
 
@@ -407,6 +408,7 @@ def run_command(
     cwd: Path | None = None,
     capture_output: bool = False,
     shell: bool = False,
+    input: str | None = None,
 ) -> subprocess.CompletedProcess:
     """Run a command and display it to the user"""
     # Format command for display
@@ -417,7 +419,13 @@ def run_command(
 
     # Run the command
     result = subprocess.run(
-        cmd, check=check, cwd=cwd, capture_output=capture_output, text=True, shell=shell
+        cmd,
+        check=check,
+        cwd=cwd,
+        capture_output=capture_output,
+        text=True,
+        shell=shell,
+        input=input,
     )
 
     # Display output if captured
@@ -429,6 +437,63 @@ def run_command(
     return result
 
 
+def is_github_authenticated() -> bool:
+    """Check if the user is authenticated with GitHub CLI.
+
+    Returns:
+        bool: True if authenticated, False otherwise
+    """
+    try:
+        # Try to get the current user, which will fail if not authenticated
+        result = run_command(["gh", "auth", "status"], check=False, capture_output=True)
+        return result.returncode == 0
+    except Exception:
+        return False
+
+
+def handle_github_authentication() -> None:
+    """Handle GitHub CLI authentication interactively."""
+    console.print("\n🔑 GitHub Authentication Required", style="bold yellow")
+    console.print("You need to authenticate with GitHub CLI to continue.")
+    console.print("Choose an authentication method:")
+    console.print("1. Login with browser")
+    console.print("2. Login with token")
+
+    choice = click.prompt(
+        "Select authentication method", type=click.Choice(["1", "2"]), default="1"
+    )
+
+    try:
+        if choice == "1":
+            # Browser-based authentication
+            run_command(["gh", "auth", "login", "--web"])
+        else:
+            # Token-based authentication
+            token = click.prompt(
+                "Enter your GitHub Personal Access Token", hide_input=True
+            )
+            # Use a subprocess with pipe to avoid showing the token in process list
+            process = subprocess.Popen(
+                ["gh", "auth", "login", "--with-token"],
+                stdin=subprocess.PIPE,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True,
+            )
+            stdout, stderr = process.communicate(input=token + "\n")
+
+            if process.returncode != 0:
+                console.print(f"❌ Authentication failed: {stderr}", style="bold red")
+                raise subprocess.CalledProcessError(
+                    process.returncode, ["gh", "auth", "login"], stdout, stderr
+                )
+
+        console.print("✅ Successfully authenticated with GitHub", style="green")
+    except subprocess.CalledProcessError as e:
+        console.print(f"❌ Authentication failed: {e}", style="bold red")
+        raise click.Abort() from e
+
+
 def create_github_repository(repository_owner: str, repository_name: str) -> None:
     """Create GitHub repository if it doesn't exist.
 

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/cli/utils/gcp.py

@@ -64,7 +64,7 @@ def verify_vertex_connection(
             api_endpoint=f"{location}-aiplatform.googleapis.com"
         ),
         client_info=get_client_info(),
-        transport=initializer.global_config._api_transport,
+        transport="rest",
     )
     request = get_dummy_request(project_id=project_id, location=location)
     client.generate_content(request=request)

{agent_starter_pack-0.0.1b0 → agent_starter_pack-0.0.3b0}/src/cli/utils/template.py

@@ -344,7 +344,9 @@ def process_template(
             requires_data_ingestion = template_config.get("settings", {}).get(
                 "requires_data_ingestion", False
             )
-            should_include_data_ingestion = include_data_ingestion or requires_data_ingestion
+            should_include_data_ingestion = (
+                include_data_ingestion or requires_data_ingestion
+            )
 
             if should_include_data_ingestion:
                 logging.debug("3. Including data processing files")
@@ -400,7 +402,9 @@ def process_template(
             requires_data_ingestion = template_config.get("settings", {}).get(
                 "requires_data_ingestion", False
             )
-            should_include_data_ingestion = include_data_ingestion or requires_data_ingestion
+            should_include_data_ingestion = (
+                include_data_ingestion or requires_data_ingestion
+            )
 
             if should_include_data_ingestion:
                 logging.debug(