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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.0.3b0
+Version: 0.0.4b0
 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.4b0/agents/agentic_rag_vertexai_search/template/.templateconfig.yaml

@@ -0,0 +1,28 @@
+# 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
+  deployment_targets: ["agent_engine", "cloud_run"]
+  extra_dependencies: [
+    "langchain-google-vertexai~=2.0.7",
+    "langchain~=0.3.14",
+    "langgraph~=0.2.63",
+    "langchain-google-vertexai~=2.0.7",
+    "langchain~=0.3.14",
+    "langchain-community~=0.3.17",
+    "langchain-openai~=0.3.5",
+    "langchain-google-community[vertexaisearch]~=2.0.2",
+  ]

agent_starter_pack-0.0.4b0/agents/crewai_coding_crew/template/.templateconfig.yaml

@@ -0,0 +1,26 @@
+# 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
+  deployment_targets: ["agent_engine", "cloud_run"]
+  extra_dependencies: [
+    "langchain-google-vertexai~=2.0.7",
+    "langchain~=0.3.14",
+    "langchain-community~=0.3.17", 
+    "langchain-openai~=0.3.5",
+    "langgraph~=0.2.63",
+    "crewai~=0.102.0"
+  ]

agent_starter_pack-0.0.4b0/agents/langgraph_base_react/template/.templateconfig.yaml

@@ -0,0 +1,27 @@
+# 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
+  deployment_targets: ["agent_engine", "cloud_run"]
+  extra_dependencies: [
+    "langchain-google-vertexai~=2.0.7",
+    "langchain~=0.3.14",
+    "langgraph~=0.2.63",
+    "langchain-google-vertexai~=2.0.7",
+    "langchain~=0.3.14", 
+    "langchain-community~=0.3.17",
+    "langchain-openai~=0.3.5",
+  ]

{agent_starter_pack-0.0.3b0 → agent_starter_pack-0.0.4b0}/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

agent_starter_pack-0.0.4b0/agents/multimodal_live_api/template/.templateconfig.yaml

@@ -0,0 +1,29 @@
+# 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
+  frontend_type: "live_api_react"
+  deployment_targets: ["cloud_run"]
+  extra_dependencies: [
+    "backoff~=2.2.1",
+    "beautifulsoup4~=4.12.3",
+    "google-genai~=1.2.0",
+    "jinja2~=3.1.4",
+    "langchain~=0.3.13",
+    "langchain-community~=0.3.13",
+    "langchain-google-vertexai~=2.0.9",
+    "scikit-learn>=1.0.0,<2.0.0",
+  ]

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

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.0.3b"
+version = "0.0.4b"
 description = "CLI tool to create GCP-based AI agent projects from templates"
 authors = [
     { name = "Google LLC", email = "agent-starter-pack@google.com" },

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

@@ -47,8 +47,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 +75,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,87 +160,18 @@ 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...")
-            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"}
+            
+            # Set up GCP environment
+            setup_gcp_environment(
+                auto_approve=auto_approve,
+                skip_checks=skip_checks,
+                region=region,
+                debug=debug
             )
 
         # Process template
@@ -332,74 +259,122 @@ 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_project_quota(project_id: str) -> None:
+    """Set the GCP project and application default quota project.
+    
+    Args:
+        project_id: The GCP project ID to set
+    """
     try:
-        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()
-
-        # 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?",
-                choices=["y", "n"],
-                default="n",
-            ).lower()
-            == "y"
+        subprocess.run(
+            ["gcloud", "config", "set", "project", project_id], check=True, capture_output=True, text=True
         )
-
-        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}")
+        console.print(f"\n> Error setting project to {project_id}:")
+        console.print(e.stderr)
         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
-
 
-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)
-
-        console.print("> Setting application default quota project...")
         subprocess.run(
-            ["gcloud", "auth", "application-default", "set-quota-project", new_project],
-            check=True,
+            ["gcloud", "auth", "application-default", "set-quota-project", project_id],
+            check=True, capture_output=True, text=True
         )
+    except subprocess.CalledProcessError as e:
+        console.print(f"> Error setting application default quota project:")
+        console.print(e.stderr)
+        raise
+    
+    console.print(f"> Successfully configured project: {project_id}")
 
-        console.print(f"> Successfully switched to project: {new_project}")
 
-        # Re-verify credentials one final time
-        final_creds_info = verify_credentials()
+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
+    """
+    if skip_checks:
+        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
+        return {"project": "unknown"}
+    
+    # Verify current GCP credentials
+    if debug:
+        logging.debug("Verifying GCP credentials...")
+    creds_info = verify_credentials()
+    
+    if not auto_approve:
+        # 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
+            creds_info = verify_credentials()
+            
+            # Prompt for project change
+            change_project = (
+                Prompt.ask(
+                    f"\n> You are now logged in with account '{creds_info['account']}'. "
+                    f"Current project is '{creds_info['project']}'. "
+                    "Do you wish to change the project?",
+                    choices=["y", "n"],
+                    default="n",
+                ).lower() == "y"
+            )
+            
+            if change_project:
+                # Prompt for new project ID
+                new_project = Prompt.ask("\n> Enter the new project ID")
+                creds_info["project"] = new_project
+    
+    # Always set project and quota, regardless of whether credentials were changed
+    set_project_quota(creds_info["project"])
+    
+    # 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"\n> Now using account '{final_creds_info['account']}' "
-            f"with project '{final_creds_info['project']}'"
+            f"> ✓ Successfully verified connection to Vertex AI in project {creds_info['project']}"
         )
-
-    except subprocess.CalledProcessError as e:
+    except Exception as e:
         console.print(
-            "\n> Error while changing project. Please verify the project ID and try again.",
-            style="bold red",
+            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
+    
+    return creds_info
 
 
 def replace_region_in_files(
@@ -485,11 +460,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 +477,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

{agent_starter_pack-0.0.3b0 → agent_starter_pack-0.0.4b0}/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]: