agent-starter-pack 0.5.2__py3-none-any.whl → 0.6.0__py3-none-any.whl

agents/adk_gemini_fullstack/template/.templateconfig.yaml

@@ -0,0 +1,37 @@
+# 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 production-ready fullstack research agent that uses Gemini to strategize, research, and synthesize comprehensive reports with human-in-the-loop collaboration"
+example_question: "A report on the latest Google I/O event"
+settings:
+  requires_data_ingestion: false
+  deployment_targets: ["agent_engine", "cloud_run"]
+  extra_dependencies: ["google-adk~=1.3.0"]
+  tags: ["adk"]
+  frontend_type: "adk_gemini_fullstack"
+  commands:
+    override:
+      install: "npm --prefix frontend install"
+    extra:
+      dev: 
+        command: 'make dev-backend & make dev-frontend'
+        description: "Start the ADK API server and React frontend development server simultaneously"
+      dev-backend:
+        command: 
+          agent_engine: 'uv run adk api_server app --allow_origins="*"'
+          cloud_run: 'ALLOW_ORIGINS="*" uv run uvicorn app.server:app --host 0.0.0.0 --port 8000 --reload'
+        description: "Start the ADK API server"
+      dev-frontend:
+        command: 'npm --prefix frontend run dev'
+        description: "Start the React frontend development server"

agents/adk_gemini_fullstack/tests/integration/test_agent.py

@@ -0,0 +1,58 @@
+# 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.
+
+# mypy: disable-error-code="union-attr"
+from google.adk.agents.run_config import RunConfig, StreamingMode
+from google.adk.runners import Runner
+from google.adk.sessions import InMemorySessionService
+from google.genai import types
+
+from app.agent import root_agent
+
+
+def test_agent_stream() -> None:
+    """
+    Integration test for the agent stream functionality.
+    Tests that the agent returns valid streaming responses.
+    """
+
+    session_service = InMemorySessionService()
+
+    session = session_service.create_session_sync(user_id="test_user", app_name="test")
+    runner = Runner(agent=root_agent, session_service=session_service, app_name="test")
+
+    message = types.Content(
+        role="user", parts=[types.Part.from_text(text="Why is the sky blue?")]
+    )
+
+    events = list(
+        runner.run(
+            new_message=message,
+            user_id="test_user",
+            session_id=session.id,
+            run_config=RunConfig(streaming_mode=StreamingMode.SSE),
+        )
+    )
+    assert len(events) > 0, "Expected at least one message"
+
+    has_text_content = False
+    for event in events:
+        if (
+            event.content
+            and event.content.parts
+            and any(part.text for part in event.content.parts)
+        ):
+            has_text_content = True
+            break
+    assert has_text_content, "Expected at least one message with text content"

agents/agentic_rag/notebooks/adk_app_testing.ipynb

@@ -261,7 +261,7 @@
    "source": [
     "### Local Testing\n",
     "\n",
-    "> You can run the application locally via the `make backend` command."
+    "> You can run the application locally via the `make local-backend` command."
    ]
   },
   {

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",

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?"

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",

src/base_template/Makefile

@@ -1,11 +1,26 @@
 install:
 	@command -v uv >/dev/null 2>&1 || { echo "uv is not installed. Installing uv..."; curl -LsSf https://astral.sh/uv/0.6.12/install.sh | sh; source ~/.bashrc; }
 	uv sync --dev{% if cookiecutter.agent_name != 'live_api' and "adk" not in cookiecutter.tags %} --extra streamlit{%- endif %} --extra jupyter --frozen{% if cookiecutter.agent_name == 'live_api' %} && npm --prefix frontend install{%- endif %}
+{%- if cookiecutter.settings.get("commands", {}).get("override", {}).get("install") %} && {{cookiecutter.settings.get("commands", {}).get("override", {}).get("install")}}{%- endif %}
+{%- if cookiecutter.settings.get("commands", {}).get("extra", {}) %}
+{%- for cmd_name, cmd_value in cookiecutter.settings.get("commands", {}).get("extra", {}).items() %}
 
-test:
-	uv run pytest tests/unit && uv run pytest tests/integration
+{{ cmd_name }}:
+{%- if cmd_value is mapping %}
+{%- if cmd_value.command is mapping and cookiecutter.deployment_target in cmd_value.command %}
+	{{ cmd_value.command[cookiecutter.deployment_target] }}
+{%- else %}
+	{{ cmd_value.command if cmd_value.command is string else "" }}
+{%- endif %}
+{%- else %}
+	{{ cmd_value }}
+{%- endif %}
+{%- endfor %}{%- endif %}
 
 playground:
+{%- if cookiecutter.settings.get("commands", {}).get("override", {}).get("playground") %}
+	{{cookiecutter.settings.get("commands", {}).get("override", {}).get("playground")}}
+{%- else %}
 	@echo "==============================================================================="
 	@echo "| 🚀 Starting your agent playground...                                        |"
 	@echo "|                                                                             |"
@@ -27,6 +42,7 @@ playground:
 	{% if cookiecutter.deployment_target == 'agent_engine' %}PYTHONPATH=. {% endif %}uv run streamlit run frontend/streamlit_app.py --browser.serverAddress=localhost --server.enableCORS=false --server.enableXsrfProtection=false
 {%- endif %}
 {%- endif %}
+{%- endif %}
 
 backend:
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
@@ -82,6 +98,9 @@ data-ingestion:
 		--pipeline-name="data-ingestion-pipeline")
 {%- endif %}
 
+test:
+	uv run pytest tests/unit && uv run pytest tests/integration
+
 lint:
 	uv run codespell
 	uv run ruff check . --diff

src/base_template/README.md

@@ -46,6 +46,11 @@ make install && make playground
 | Command              | Description                                                                                 |
 | -------------------- | ------------------------------------------------------------------------------------------- |
 | `make install`       | Install all required dependencies using uv                                                  |
+{%- if cookiecutter.settings.get("commands", {}).get("extra", {}) %}
+{%- for cmd_name, cmd_value in cookiecutter.settings.get("commands", {}).get("extra", {}).items() %}
+| `make {{ cmd_name }}`       | {% if cmd_value is mapping %}{% if cmd_value.description %}{{ cmd_value.description }}{% else %}{% if cookiecutter.deployment_target in cmd_value %}{{ cmd_value[cookiecutter.deployment_target] }}{% else %}{{ cmd_value.command if cmd_value.command is string else "" }}{% endif %}{% endif %}{% else %}{{ cmd_value }}{% endif %} |
+{%- endfor %}
+{%- endif %}
 {%- if cookiecutter.deployment_target == 'cloud_run' %}
 | `make playground`    | Launch local development environment with backend and frontend{%- if "adk" in cookiecutter.tags %} - leveraging `adk web` command. {%- endif %}|
 | `make backend`       | Deploy agent to Cloud Run |
@@ -84,7 +89,7 @@ Here’s the recommended workflow for local development:
 2.  **Start the Backend Server:**
     Open a terminal and run:
     ```bash
-    make backend
+    make local-backend
     ```
     The backend is ready when you see `INFO:     Application startup complete.` Wait for this message before starting the frontend.
 
@@ -96,7 +101,7 @@ Here’s the recommended workflow for local development:
     ```bash
     export VERTEXAI=false
     export GOOGLE_API_KEY="your-google-api-key" # Replace with your actual key
-    make backend
+    make local-backend
     ```
     Ensure `GOOGLE_API_KEY` is set correctly in your environment.
     </details>
@@ -130,7 +135,7 @@ To run the agent using Google Cloud Shell:
 2.  **Start the Backend:**
     Open a *new* Cloud Shell tab. Set your project: `gcloud config set project [PROJECT_ID]`. Then run:
     ```bash
-    make backend
+    make local-backend
     ```
 
 3.  **Configure Backend Web Preview:**

src/base_template/pyproject.toml

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

src/cli/commands/create.py

@@ -191,6 +191,13 @@ def create(
         if debug:
             logging.debug(f"Selected agent: {agent}")
 
+        template_path = (
+            pathlib.Path(__file__).parent.parent.parent.parent
+            / "agents"
+            / final_agent
+            / "template"
+        )
+        config = load_template_config(template_path)
         # Data ingestion and datastore selection
         if include_data_ingestion or datastore:
             # If datastore is specified but include_data_ingestion is not, set it to True
@@ -206,13 +213,6 @@ def create(
                 logging.debug(f"Selected datastore type: {datastore}")
         else:
             # Check if the agent requires data ingestion
-            template_path = (
-                pathlib.Path(__file__).parent.parent.parent.parent
-                / "agents"
-                / final_agent
-                / "template"
-            )
-            config = load_template_config(template_path)
             if config and config.get("settings", {}).get("requires_data_ingestion"):
                 include_data_ingestion = True
                 datastore = prompt_datastore_selection(final_agent)
@@ -322,9 +322,16 @@ def create(
         )
         # Determine the correct path to display based on whether output_dir was specified
         console.print("\n🚀 To get started, run the following command:")
-        console.print(
-            f"   [bold bright_green]cd {cd_path} && make install && make playground[/]"
-        )
+
+        # Check if the agent has a 'dev' command in its settings
+        if config["settings"].get("commands", {}).get("extra", {}).get("dev"):
+            console.print(
+                f"   [bold bright_green]cd {cd_path} && make install && make dev[/]"
+            )
+        else:
+            console.print(
+                f"   [bold bright_green]cd {cd_path} && make install && make playground[/]"
+            )
     except Exception:
         if debug:
             logging.exception(

src/cli/utils/template.py

@@ -80,7 +80,12 @@ def get_available_agents(deployment_target: str | None = None) -> dict:
         deployment_target: Optional deployment target to filter agents
     """
     # Define priority agents that should appear first
-    PRIORITY_AGENTS = ["adk_base", "agentic_rag", "langgraph_base_react"]
+    PRIORITY_AGENTS = [
+        "adk_base",
+        "adk_gemini_fullstack",
+        "agentic_rag",
+        "langgraph_base_react",
+    ]
 
     agents_list = []
     priority_agents_dict = dict.fromkeys(PRIORITY_AGENTS)  # Track priority agents
@@ -499,15 +504,12 @@ def process_template(
                 copy_data_ingestion_files(project_template, datastore)
 
             # Create cookiecutter.json in the template root
-            # Process extra dependencies
-            extra_deps = template_config.get("settings", {}).get(
-                "extra_dependencies", []
-            )
-            # Get frontend type from template config
-            frontend_type = template_config.get("settings", {}).get(
-                "frontend_type", DEFAULT_FRONTEND
-            )
-            tags = template_config.get("settings", {}).get("tags", ["None"])
+            # Get settings from template config
+            settings = template_config.get("settings", {})
+            extra_deps = settings.get("extra_dependencies", [])
+            frontend_type = settings.get("frontend_type", DEFAULT_FRONTEND)
+            tags = settings.get("tags", ["None"])
+
             cookiecutter_config = {
                 "project_name": "my-project",
                 "agent_name": agent_name,
@@ -516,6 +518,7 @@ def process_template(
                 "example_question": template_config.get("example_question", "").ljust(
                     61
                 ),
+                "settings": settings,
                 "tags": tags,
                 "deployment_target": deployment_target or "",
                 "frontend_type": frontend_type,

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.
 """

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 %}

src/deployment_targets/cloud_run/app/server.py

@@ -14,17 +14,28 @@
 {% 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__)
+allow_origins = (
+    os.getenv("ALLOW_ORIGINS", "").split(",") if os.getenv("ALLOW_ORIGINS") else None
+)
+
+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())
@@ -32,8 +43,12 @@ 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,
+    allow_origins=allow_origins,
+)
 app.title = "{{cookiecutter.project_name}}"
 app.description = "API for interacting with the Agent {{cookiecutter.project_name}}"
 {%- else %}

src/deployment_targets/cloud_run/tests/integration/test_server_e2e.py

@@ -157,7 +157,7 @@ def test_chat_stream(server_fixture: subprocess.Popen[str]) -> None:
     }
 {% endif %}
     response = requests.post(
-        STREAM_URL, headers=HEADERS, json=data, stream=True, timeout=10
+        STREAM_URL, headers=HEADERS, json=data, stream=True, timeout=60
     )
     assert response.status_code == 200