agent-starter-pack 0.0.7b0__tar.gz → 0.0.8b0__tar.gz

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/CONTRIBUTING.md

@@ -47,15 +47,7 @@ Here are some additional things to keep in mind during the process:
 
 ## For Google Employees
 
-Complete the following steps to register your GitHub account and be added as a contributor to this repository.
-
-1. Register your GitHub account at [go/GitHub](http://go/github)
-
-1. Once you have registered, go to [go/github-googlecloudplatform](http://go/github-googlecloudplatform) and request to join the GoogleCloudPlatform organization. Check the box "I need write access on a public repository".
-
-1. You'll receive an email to your GitHub registered email to approve the request to join. Approve it.
-
-1. Request to join this team [GoogleCloudPlatform/teams/generative-ai-samples-contributors](https://github.com/orgs/GoogleCloudPlatform/teams/generative-ai-samples-contributors/members)
+If you're a Google employee, as a requirement, please follow the steps outlined in the [Google Cloud Platform Generative AI repository's contributing guide](https://github.com/GoogleCloudPlatform/generative-ai/blob/main/CONTRIBUTING.md#for-google-employees) for processes and requirements.
 
 ## Code Quality Checks
 

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.0.7b0
+Version: 0.0.8b0
 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
@@ -143,3 +143,7 @@ Thank you for your contributions!
 ## Disclaimer
 
 This repository is for demonstrative purposes only and is not an officially supported Google product.
+
+## Terms of Service
+
+The agent-starter-pack templating CLI and the templates in this starter pack leverage Google Cloud APIs. When you use this starter pack, you'll be deploying resources in your own Google Cloud project and will be responsible for those resources. Please review the [Google Cloud Service Terms](https://cloud.google.com/terms/service-terms) for details on the terms of service associated with these APIs.

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/README.md

@@ -117,3 +117,7 @@ Thank you for your contributions!
 ## Disclaimer
 
 This repository is for demonstrative purposes only and is not an officially supported Google product.
+
+## Terms of Service
+
+The agent-starter-pack templating CLI and the templates in this starter pack leverage Google Cloud APIs. When you use this starter pack, you'll be deploying resources in your own Google Cloud project and will be responsible for those resources. Please review the [Google Cloud Service Terms](https://cloud.google.com/terms/service-terms) for details on the terms of service associated with these APIs.

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/docs/observability.md

@@ -24,3 +24,7 @@ Events are forwarded to BigQuery through a [log router](https://cloud.google.com
 Once the data is written to BigQuery, it can be used to populate a [Looker Studio dashboard](https://lookerstudio.google.com/c/reporting/fa742264-4b4b-4c56-81e6-a667dd0f853f/page/tEnnC).
 
 This dashboard template provides a starting point for building custom visualizations on top of the captured data.
+
+## Disclaimer 
+
+**Note:** The templated agents are designed to enable *your* use-case observability in your Google Cloud Project. Google Cloud does not log, monitor, or otherwise access any data generated from the deployed resources. See the [Google Cloud Service Terms](https://cloud.google.com/terms/service-terms) for more details.

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.0.7b"
+version = "0.0.8b"
 description = "CLI tool to create GCP-based AI agent projects from templates"
 authors = [
     { name = "Google LLC", email = "agent-starter-pack@google.com" },
@@ -108,7 +108,7 @@ disable_error_code = ["misc", "no-untyped-call", "no-any-return"]
 
 [tool.codespell]
 ignore-words-list = "rouge"
-skip = "./locust_env/*,uv.lock,.venv"
+skip = "./locust_env/*,uv.lock,.venv,**/*.ipynb"
 
 [build-system]
 requires = ["hatchling"]

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/src/base_template/pyproject.toml

@@ -96,9 +96,9 @@ exclude = [".venv"]
 [tool.codespell]
 ignore-words-list = "rouge"
 {% if cookiecutter.agent_name == 'multimodal_live_api' %}
-skip = "./locust_env/*,uv.lock,.venv,./frontend"
+skip = "./locust_env/*,uv.lock,.venv,./frontend,**/*.ipynb"
 {% else %}
-skip = "./locust_env/*,uv.lock,.venv"
+skip = "./locust_env/*,uv.lock,.venv,**/*.ipynb"
 {%- endif %}
 
 [build-system]

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/src/cli/commands/create.py

@@ -330,16 +330,20 @@ def setup_gcp_environment(
     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)
+        # If user chose to skip verification, don't test Vertex AI connection
+        if creds_info.get("skip_vertex_test", False):
+            console.print("> Skipping Vertex AI connection test", style="yellow")
+        else:
+            # Test Vertex AI connection
+            _test_vertex_ai_connection(creds_info["project"], region)
     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)
+        # Test Vertex AI connection
+        _test_vertex_ai_connection(creds_info["project"], region)
 
     return creds_info
 
@@ -365,17 +369,23 @@ def _handle_credential_verification(creds_info: dict) -> dict:
             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"
-    )
+    # Ask user if current credentials are correct or if they want to skip
+    console.print(f"\n> You are logged in with account: '{creds_info['account']}'")
+    console.print(f"> You are using project: '{creds_info['project']}'")
+
+    choices = ["Y", "skip", "edit"]
+    response = Prompt.ask(
+        "> Do you want to continue? (The CLI will check if Vertex AI is enabled in this project)",
+        choices=choices,
+        default="Y",
+    ).lower()
+
+    if response == "skip":
+        console.print("> Skipping credential verification", style="yellow")
+        creds_info["skip_vertex_test"] = True
+        return creds_info
+
+    change_creds = response == "edit"
 
     if change_creds:
         # Handle credential change
@@ -387,18 +397,23 @@ def _handle_credential_verification(creds_info: dict) -> dict:
         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']}'. "
-                "Is this correct?",
-                choices=["y", "n"],
-                default="y",
-            ).lower()
-            == "n"
+        console.print(
+            f"\n> You are now logged in with account: '{creds_info['account']}'."
         )
+        console.print(f"> Current project is: '{creds_info['project']}'.")
+        choices = ["y", "skip", "edit"]
+        response = Prompt.ask(
+            "> Do you want to continue? (The CLI will verify Vertex AI access in this project)",
+            choices=choices,
+            default="y",
+        ).lower()
+
+        if response == "skip":
+            console.print("> Skipping project verification", style="yellow")
+            creds_info["skip_vertex_test"] = True
+            return creds_info
 
-        if change_project:
+        if response == "edit":
             # Prompt for new project ID
             new_project = Prompt.ask("\n> Enter the new project ID")
             creds_info["project"] = new_project

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/src/cli/utils/gcp.py

@@ -69,7 +69,7 @@ def verify_vertex_connection(
     client.count_tokens(request=request)
 
 
-def verify_credentials() -> dict[str, str]:
+def verify_credentials() -> dict:
     """Verify GCP credentials and return current project and account."""
     try:
         # Get credentials and project

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/src/frontends/streamlit/frontend/utils/title_summary.py

@@ -12,31 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import google.auth
+# mypy: disable-error-code="assignment"
 import os
+from typing import Any
 
+import google.auth
+from langchain_core.messages import AIMessage
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_google_vertexai import ChatVertexAI
 
-# Get default project from credentials
-try:
-    _, project_id = google.auth.default()
-    
-    llm = ChatVertexAI(
-        model_name="gemini-2.0-flash-001",
-        temperature=0,
-        project=project_id,
-        location=os.getenv("LOCATION", "us-central1"),
-    )
-except Exception:
-    # Fallback to a dummy LLM that just returns "conversation"
-    print("WARNING: Failed to initialize Vertex AI. Using dummy LLM instead.")
-    class DummyLLM:
-        def invoke(self, *args, **kwargs):
-            return {"content": "conversation"}
-    
-    llm = DummyLLM()
-
 title_template = ChatPromptTemplate.from_messages(
     [
         (
@@ -87,5 +71,24 @@ If there's not enough context in the conversation to create a meaningful title,
         MessagesPlaceholder(variable_name="messages"),
     ]
 )
+try:
+    # Initialize Vertex AI with default project credentials
+    _, project_id = google.auth.default()
+
+    llm = ChatVertexAI(
+        model_name="gemini-2.0-flash-001",
+        temperature=0,
+        project=project_id,
+        location=os.getenv("LOCATION", "us-central1"),
+    )
+    chain_title = title_template | llm
+
+except Exception:
+    # Fallback to a simple title generator when Vertex AI is unavailable
+    print("WARNING: Failed to initialize Vertex AI. Using dummy LLM instead.")
+
+    class DummyChain:
+        def invoke(*args: Any, **kwargs: Any) -> AIMessage:
+            return AIMessage(content="conversation")
 
-chain_title = title_template | llm
+    chain_title = DummyChain()

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/tests/cli/commands/test_create.py

@@ -378,15 +378,21 @@ class TestCreateCommand:
             {"account": "new@example.com", "project": "new-project"},  # After login
         ]
 
+        # Set up expected subprocess calls
+        mock_subprocess.side_effect = [
+            MagicMock(returncode=0),  # gcloud auth login
+            MagicMock(returncode=0),  # gcloud config set project
+            MagicMock(returncode=0),  # gcloud auth application-default set-quota-project
+        ]
+
         with (
             patch("pathlib.Path.exists", return_value=False),
             patch("rich.prompt.Prompt.ask") as mock_prompt,
         ):
             # Set up credential change responses
             mock_prompt.side_effect = [
-                "n",  # Change credentials when prompted
-                "y",  # Change project when prompted
-                "new-project",  # New project ID
+                "edit",  # Change credentials when prompted
+                "y",  # Continue after login
             ]
 
             result = runner.invoke(
@@ -404,9 +410,10 @@ class TestCreateCommand:
             )
         assert result.exit_code == 0
 
-        # Verify that gcloud auth login was called
+        # Verify subprocess was called with the right arguments for login
         mock_subprocess.assert_any_call(
-            ["gcloud", "auth", "login", "--update-adc"], check=True
+            ["gcloud", "auth", "login", "--update-adc"], 
+            check=True
         )
 
     def test_create_with_invalid_agent_name(

{agent_starter_pack-0.0.7b0 → agent_starter_pack-0.0.8b0}/uv.lock

@@ -11,7 +11,7 @@ resolution-markers = [
 
 [[package]]
 name = "agent-starter-pack"
-version = "0.0.6b0"
+version = "0.0.7b0"
 source = { editable = "." }
 dependencies = [
     { name = "click" },