agent-starter-pack 0.0.7b0__py3-none-any.whl → 0.0.8b0__py3-none-any.whl

{agent_starter_pack-0.0.7b0.dist-info → agent_starter_pack-0.0.8b0.dist-info}/METADATA

@@ -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.dist-info → agent_starter_pack-0.0.8b0.dist-info}/RECORD

@@ -31,7 +31,7 @@ agents/multimodal_live_api/tests/unit/test_server.py,sha256=_TjlgQgNkjerIaBGnu8P
 src/base_template/.gitignore,sha256=mJKTZIcVdAFiIUQicRfPNGUg6WvwcfTEC2xbmAaU34g,2579
 src/base_template/Makefile,sha256=YQXymaikoFP3r7MN1EiNbQv8VZKjrVzeRoksIPHNX0E,1568
 src/base_template/README.md,sha256=LS07ZhCLj2SXew3j3xEWaaXwsf3aIjyvQpkp0S0l8nA,4194
-src/base_template/pyproject.toml,sha256=Yyk30u_QBtwHmFaEC40PyJe3Smt6snzSZxd5usoO9tA,2777
+src/base_template/pyproject.toml,sha256=Pv4nC4uh_9_sGA90Tom0ITJUAbGolhseCA9-AOaEIpY,2799
 src/base_template/app/utils/tracing.py,sha256=ovp6tgwgo_tqibpbhhHl8cWIqqUWVYfV0AL7XPDEPpM,5661
 src/base_template/app/utils/typing.py,sha256=UUFm74qiKGtT8pLuE_3yRpcmntRCt91wH_Q-LKdjQuQ,3194
 src/base_template/deployment/README.md,sha256=8AorSeF2Aj0kvPmuQLFrmAAUWVbdTzg0S-rVBbfdirQ,4835
@@ -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=oupbVv8zfMho4JowBCvgr0YpS2I7mYWCDR3We8VxqjI,18497
+src/cli/commands/create.py,sha256=8ELvElRExVa1KqDNhGINvui_xRyKKFJaSLtylbmX9pc,19421
 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=nZP9wBfzJXuATbX-e8gVY9FyF23b3q-bXFosgkW3eBU,4128
+src/cli/utils/gcp.py,sha256=f-XTq9AN2_KrCR8pf_gSPOng8KrvtFlNP2Ok9VtTk_Q,4118
 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
@@ -135,7 +135,7 @@ src/frontends/streamlit/frontend/utils/local_chat_history.py,sha256=9wc8L8j4tk10
 src/frontends/streamlit/frontend/utils/message_editing.py,sha256=YWoPe2KeWMuL3YVTm0am6MK3kzjEIYVmdkdwTQpmGdQ,2263
 src/frontends/streamlit/frontend/utils/multimodal_utils.py,sha256=v6YbCkz_YcnEo-9YvRjwBNt0SzU4M39bYxJGmKk69vE,7313
 src/frontends/streamlit/frontend/utils/stream_handler.py,sha256=J7MMsOR2GvDEsyJbsv8VQeZbMs7mNySrclZNfuzsZD4,11578
-src/frontends/streamlit/frontend/utils/title_summary.py,sha256=Po8FY1pfIc9rnWK8mCW76LwtQJANiDXV7GU9TXPXno4,2859
+src/frontends/streamlit/frontend/utils/title_summary.py,sha256=B0cadS_KPW-tsbABauI4J681aqjEtuKFDa25e9R1WKc,3030
 src/resources/containers/data_processing/Dockerfile,sha256=uXjqDiSrr7YpoX-vtzaIJIVXZ2qeUpu1Nzju9qWZ9oI,859
 src/resources/locks/uv-agentic_rag_vertexai_search-agent_engine.lock,sha256=6muWlgt4dduqGDiGUvTg-d2zzY5rgW9zP9DbDnt3krc,474353
 src/resources/locks/uv-agentic_rag_vertexai_search-cloud_run.lock,sha256=ViUzp35RGWZiiQvvVi6ntFTT-bEbHg4_HDDQznD4ML8,756574
@@ -150,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.7b0.dist-info/METADATA,sha256=xxRQaJNJXMnflrDzzJ0USRzAVFHsY-ICYEJm5e6j7PQ,6805
-agent_starter_pack-0.0.7b0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-agent_starter_pack-0.0.7b0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
-agent_starter_pack-0.0.7b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agent_starter_pack-0.0.7b0.dist-info/RECORD,,
+agent_starter_pack-0.0.8b0.dist-info/METADATA,sha256=aA8BBrtZBbPQuI9aeP6GEb7XSJT6Go8gR2FS5aIp0ns,7229
+agent_starter_pack-0.0.8b0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+agent_starter_pack-0.0.8b0.dist-info/entry_points.txt,sha256=U7uCxR7YulIhZ0L8R8Hui0Bsy6J7oyESBeDYJYMrQjA,56
+agent_starter_pack-0.0.8b0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agent_starter_pack-0.0.8b0.dist-info/RECORD,,

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]

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

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

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()