agent-starter-pack 0.5.2__tar.gz → 0.5.3__tar.gz

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-starter-pack
-Version: 0.5.2
+Version: 0.5.3
 Summary: CLI to bootstrap production-ready Google Cloud GenAI agent projects from templates.
 Author-email: Google LLC <agent-starter-pack@google.com>
 License: Apache-2.0
@@ -9,7 +9,7 @@ Requires-Python: >=3.10
 Requires-Dist: backoff>=2.2.1
 Requires-Dist: click~=8.1.7
 Requires-Dist: cookiecutter~=2.5.0
-Requires-Dist: google-cloud-aiplatform>=1.95.0
+Requires-Dist: google-cloud-aiplatform>=1.87.0
 Requires-Dist: pyyaml~=6.0.1
 Requires-Dist: rich~=13.7.0
 Provides-Extra: jupyter

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/agents/adk_base/template/.templateconfig.yaml

@@ -17,6 +17,6 @@ example_question: "What's the weather in San Francisco?"
 settings:
   requires_data_ingestion: false
   deployment_targets: ["agent_engine", "cloud_run"]
-  extra_dependencies: ["google-adk~=1.2.0"]
+  extra_dependencies: ["google-adk~=1.3.0"]
   tags: ["adk"]
   frontend_type: "None"

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/agents/agentic_rag/template/.templateconfig.yaml

@@ -18,7 +18,7 @@ settings:
   requires_data_ingestion: true
   deployment_targets: ["agent_engine", "cloud_run"]
   extra_dependencies: [
-    "google-adk~=1.2.0",
+    "google-adk~=1.3.0",
     "langchain-google-vertexai~=2.0.7",
     "langchain~=0.3.24",
     "langchain-core~=0.3.55",

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/agents/crewai_coding_crew/template/.templateconfig.yaml

@@ -21,7 +21,7 @@ settings:
     "langchain~=0.3.14",
     "langchain-community~=0.3.17", 
     "langchain-openai~=0.3.5",
-    "langgraph~=0.2.63",
+    "langgraph~=0.4.8",
     "crewai~=0.119.0"
   ]
 example_question: "How can I implement a function to sort a list in Python?"

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/agents/langgraph_base_react/template/.templateconfig.yaml

@@ -19,7 +19,7 @@ settings:
   extra_dependencies: [
     "langchain-google-vertexai~=2.0.7",
     "langchain~=0.3.14",
-    "langgraph~=0.2.63",
+    "langgraph~=0.4.8",
     "langchain-google-vertexai~=2.0.22",
     "langchain~=0.3.14", 
     "langchain-community~=0.3.17",

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-starter-pack"
-version = "0.5.2"
+version = "0.5.3"
 description = "CLI to bootstrap production-ready Google Cloud GenAI agent projects from templates."
 authors = [
     { name = "Google LLC", email = "agent-starter-pack@google.com" },
@@ -11,7 +11,7 @@ requires-python = ">=3.10"
 dependencies = [
     "click~=8.1.7",
     "cookiecutter~=2.5.0",
-    "google-cloud-aiplatform>=1.95.0",
+    "google-cloud-aiplatform>=1.87.0",
     "rich~=13.7.0",
     "pyyaml~=6.0.1",
     "backoff>=2.2.1",

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/src/data_ingestion/data_ingestion_pipeline/components/process_data.py

@@ -15,7 +15,7 @@
 
 """
 This component is derived from the notebook:
-https://github.com/GoogleCloudPlatform/generative-ai/blob/main/gemini/use-cases/retrieval-augmented_generation/scalable_rag_with_bigframes.ipynb
+https://github.com/GoogleCloudPlatform/generative-ai/blob/main/gemini/use-cases/retrieval-augmented-generation/scalable_rag_with_bigframes.ipynb
 
 It leverages BigQuery for data processing. We also suggest looking at remote functions for enhanced scalability.
 """

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/src/deployment_targets/agent_engine/app/agent_engine_app.py

@@ -24,6 +24,7 @@ from typing import Any
 
 import google.auth
 import vertexai
+from google.adk.artifacts import GcsArtifactService
 from google.cloud import logging as google_cloud_logging
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider, export
@@ -210,18 +211,29 @@ def deploy_agent_engine_app(
 ) -> agent_engines.AgentEngine:
     """Deploy the agent engine app to Vertex AI."""
 
-    staging_bucket = f"gs://{project}-agent-engine"
-
+    staging_bucket_uri = f"gs://{project}-agent-engine"
+{%- if "adk" in cookiecutter.tags %}
+    artifacts_bucket_name = f"{project}-{{cookiecutter.project_name}}-logs-data"
     create_bucket_if_not_exists(
-        bucket_name=staging_bucket, project=project, location=location
+        bucket_name=artifacts_bucket_name, project=project, location=location
     )
-    vertexai.init(project=project, location=location, staging_bucket=staging_bucket)
+{%- endif %}
+    create_bucket_if_not_exists(
+        bucket_name=staging_bucket_uri, project=project, location=location
+    )
+
+    vertexai.init(project=project, location=location, staging_bucket=staging_bucket_uri)
 
     # Read requirements
     with open(requirements_file) as f:
         requirements = f.read().strip().split("\n")
 {% if "adk" in cookiecutter.tags %}
-    agent_engine = AgentEngineApp(agent=root_agent)
+    agent_engine = AgentEngineApp(
+        agent=root_agent,
+        artifact_service_builder=lambda: GcsArtifactService(
+            bucket_name=artifacts_bucket_name
+        ),
+    )
 {% else %}
     agent_engine = AgentEngineApp(project_id=project)
 {% endif %}

{agent_starter_pack-0.5.2 → agent_starter_pack-0.5.3}/src/deployment_targets/cloud_run/app/server.py

@@ -14,26 +14,35 @@
 {% if "adk" in cookiecutter.tags %}
 import os
 
+import google.auth
 from fastapi import FastAPI
 from google.adk.cli.fast_api import get_fast_api_app
 from google.cloud import logging as google_cloud_logging
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider, export
 
+from app.utils.gcs import create_bucket_if_not_exists
 from app.utils.tracing import CloudTraceLoggingSpanExporter
 from app.utils.typing import Feedback
 
+_, project_id = google.auth.default()
 logging_client = google_cloud_logging.Client()
 logger = logging_client.logger(__name__)
 
+bucket_name = f"gs://{project_id}-{{cookiecutter.project_name}}-logs-data"
+create_bucket_if_not_exists(
+    bucket_name=bucket_name, project=project_id, location="us-central1"
+)
+
 provider = TracerProvider()
 processor = export.BatchSpanProcessor(CloudTraceLoggingSpanExporter())
 provider.add_span_processor(processor)
 trace.set_tracer_provider(provider)
 
 AGENT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-app: FastAPI = get_fast_api_app(agents_dir=AGENT_DIR, web=True)
-
+app: FastAPI = get_fast_api_app(
+    agents_dir=AGENT_DIR, web=True, artifact_service_uri=bucket_name
+)
 app.title = "{{cookiecutter.project_name}}"
 app.description = "API for interacting with the Agent {{cookiecutter.project_name}}"
 {%- else %}