agent-starter-pack 0.11.2__py3-none-any.whl → 0.12.0__py3-none-any.whl

src/deployment_targets/cloud_run/Dockerfile

@@ -20,7 +20,7 @@ WORKDIR /code
 
 COPY ./pyproject.toml ./README.md ./uv.lock* ./
 
-COPY ./app ./app
+COPY ./{{cookiecutter.agent_directory}} ./{{cookiecutter.agent_directory}}
 
 RUN uv sync --frozen
 
@@ -29,4 +29,4 @@ ENV COMMIT_SHA=${COMMIT_SHA}
 
 EXPOSE 8080
 
-CMD ["uv", "run", "uvicorn", "app.server:app", "--host", "0.0.0.0", "--port", "8080"]
+CMD ["uv", "run", "uvicorn", "{{cookiecutter.agent_directory}}.server:app", "--host", "0.0.0.0", "--port", "8080"]

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

@@ -54,7 +54,7 @@ def start_server() -> subprocess.Popen[str]:
         sys.executable,
         "-m",
         "uvicorn",
-        "app.server:app",
+        "{{cookiecutter.agent_directory}}.server:app",
         "--host",
         "0.0.0.0",
         "--port",
@@ -129,7 +129,7 @@ def test_chat_stream(server_fixture: subprocess.Popen[str]) -> None:
     user_id = "test_user_123"
     session_data = {"state": {"preferred_language": "English", "visit_count": 1}}
 
-    session_url = f"{BASE_URL}/apps/app/users/{user_id}/sessions"
+    session_url = f"{BASE_URL}/apps/{{cookiecutter.agent_directory}}/users/{user_id}/sessions"
     session_response = requests.post(
         session_url,
         headers=HEADERS,
@@ -142,7 +142,7 @@ def test_chat_stream(server_fixture: subprocess.Popen[str]) -> None:
 
     # Then send chat message
     data = {
-        "app_name": "app",
+        "app_name": "{{cookiecutter.agent_directory}}",
         "user_id": user_id,
         "session_id": session_id,
         "new_message": {

src/deployment_targets/cloud_run/tests/load_test/README.md

@@ -11,7 +11,7 @@ Follow these steps to execute load tests on your local machine:
 Launch the FastAPI server in a separate terminal:
 
 ```bash
-uv run uvicorn app.server:app --host 0.0.0.0 --port 8000 --reload
+uv run uvicorn {{cookiecutter.agent_directory}}.server:app --host 0.0.0.0 --port 8000 --reload
 ```
 
 **2. (In another tab) Create virtual environment with Locust**

src/deployment_targets/cloud_run/tests/load_test/load_test.py

@@ -45,7 +45,7 @@ class ChatStreamUser(HttpUser):
         user_id = f"user_{uuid.uuid4()}"
         session_data = {"state": {"preferred_language": "English", "visit_count": 1}}
 
-        session_url = f"{self.client.base_url}/apps/app/users/{user_id}/sessions"
+        session_url = f"{self.client.base_url}/apps/{{cookiecutter.agent_directory}}/users/{user_id}/sessions"
         session_response = requests.post(
             session_url,
             headers=headers,
@@ -58,7 +58,7 @@ class ChatStreamUser(HttpUser):
 
         # Send chat message
         data = {
-            "app_name": "app",
+            "app_name": "{{cookiecutter.agent_directory}}",
             "user_id": user_id,
             "session_id": session_id,
             "new_message": {

{agents/live_api/app → src/deployment_targets/cloud_run/{{cookiecutter.agent_directory}}}/server.py

@@ -11,7 +11,7 @@
 # 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.
-
+{% if cookiecutter.agent_name == "live_api" %}
 import asyncio
 import json
 import logging
@@ -27,7 +27,7 @@ from google.genai.types import LiveServerToolCall
 from pydantic import BaseModel
 from websockets.exceptions import ConnectionClosedError
 
-from app.agent import MODEL_ID, genai_client, live_connect_config, tool_functions
+from .agent import MODEL_ID, genai_client, live_connect_config, tool_functions
 
 app = FastAPI()
 app.add_middleware(
@@ -197,16 +197,195 @@ class Feedback(BaseModel):
     run_id: str
     user_id: str | None
     log_type: Literal["feedback"] = "feedback"
+{% elif "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
+{%- if cookiecutter.session_type == "agent_engine" %}
+from vertexai import agent_engines
+{%- endif %}
+
+from {{cookiecutter.agent_directory}}.utils.gcs import create_bucket_if_not_exists
+from {{cookiecutter.agent_directory}}.utils.tracing import CloudTraceLoggingSpanExporter
+from {{cookiecutter.agent_directory}}.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())
+provider.add_span_processor(processor)
+trace.set_tracer_provider(provider)
+
+AGENT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+
+{%- if cookiecutter.session_type == "alloydb" %}
+# AlloyDB session configuration
+db_user = os.environ.get("DB_USER", "postgres")
+db_name = os.environ.get("DB_NAME", "postgres")
+db_pass = os.environ.get("DB_PASS")
+db_host = os.environ.get("DB_HOST")
+
+# Set session_service_uri if database credentials are available
+session_service_uri = None
+if db_host and db_pass:
+    session_service_uri = f"postgresql://{db_user}:{db_pass}@{db_host}:5432/{db_name}"
+{%- elif cookiecutter.session_type == "agent_engine" %}
+# Agent Engine session configuration
+# Use environment variable for agent name, default to project name
+agent_name = os.environ.get("AGENT_ENGINE_SESSION_NAME", "{{cookiecutter.project_name}}")
+
+# Check if an agent with this name already exists
+existing_agents = list(agent_engines.list(filter=f"display_name={agent_name}"))
+
+if existing_agents:
+    # Use the existing agent
+    agent_engine = existing_agents[0]
+else:
+    # Create a new agent if none exists
+    agent_engine = agent_engines.create(display_name=agent_name)
+
+session_service_uri = f"agentengine://{agent_engine.resource_name}"
+{%- else %}
+# In-memory session configuration - no persistent storage
+session_service_uri = None
+{%- endif %}
+
+app: FastAPI = get_fast_api_app(
+    agents_dir=AGENT_DIR,
+    web=True,
+    artifact_service_uri=bucket_name,
+    allow_origins=allow_origins,
+    session_service_uri=session_service_uri,
+)
+app.title = "{{cookiecutter.project_name}}"
+app.description = "API for interacting with the Agent {{cookiecutter.project_name}}"
+{% else %}
+import logging
+import os
+from collections.abc import Generator
+
+from fastapi import FastAPI
+from fastapi.responses import RedirectResponse, StreamingResponse
+from google.cloud import logging as google_cloud_logging
+from langchain_core.runnables import RunnableConfig
+from traceloop.sdk import Instruments, Traceloop
+
+from {{cookiecutter.agent_directory}}.agent import agent
+from {{cookiecutter.agent_directory}}.utils.tracing import CloudTraceLoggingSpanExporter
+from {{cookiecutter.agent_directory}}.utils.typing import Feedback, InputChat, Request, dumps, ensure_valid_config
+
+# Initialize FastAPI app and logging
+app = FastAPI(
+    title="{{cookiecutter.project_name}}",
+    description="API for interacting with the Agent {{cookiecutter.project_name}}",
+)
+logging_client = google_cloud_logging.Client()
+logger = logging_client.logger(__name__)
+
+# Initialize Telemetry
+try:
+    Traceloop.init(
+        app_name=app.title,
+        disable_batch=False,
+        exporter=CloudTraceLoggingSpanExporter(),
+        instruments={Instruments.LANGCHAIN, Instruments.CREW},
+    )
+except Exception as e:
+    logging.error("Failed to initialize Telemetry: %s", str(e))
+
+
+def set_tracing_properties(config: RunnableConfig) -> None:
+    """Sets tracing association properties for the current request.
+
+    Args:
+        config: Optional RunnableConfig containing request metadata
+    """
+    Traceloop.set_association_properties(
+        {
+            "log_type": "tracing",
+            "run_id": str(config.get("run_id", "None")),
+            "user_id": config["metadata"].pop("user_id", "None"),
+            "session_id": config["metadata"].pop("session_id", "None"),
+            "commit_sha": os.environ.get("COMMIT_SHA", "None"),
+        }
+    )
+
+
+def stream_messages(
+    input: InputChat,
+    config: RunnableConfig | None = None,
+) -> Generator[str, None, None]:
+    """Stream events in response to an input chat.
+
+    Args:
+        input: The input chat messages
+        config: Optional configuration for the runnable
+
+    Yields:
+        JSON serialized event data
+    """
+    config = ensure_valid_config(config=config)
+    set_tracing_properties(config)
+    input_dict = input.model_dump()
+
+    for data in agent.stream(input_dict, config=config, stream_mode="messages"):  # type: ignore[arg-type]
+        yield dumps(data) + "\n"
+
+
+# Routes
+@app.get("/", response_class=RedirectResponse)
+def redirect_root_to_docs() -> RedirectResponse:
+    """Redirect the root URL to the API documentation."""
+    return RedirectResponse(url="/docs")
+
+
+@app.post("/stream_messages")
+def stream_chat_events(request: Request) -> StreamingResponse:
+    """Stream chat events in response to an input request.
+
+    Args:
+        request: The chat request containing input and config
+
+    Returns:
+        Streaming response of chat events
+    """
+    return StreamingResponse(
+        stream_messages(input=request.input, config=request.config),
+        media_type="text/event-stream",
+    )
+{% endif %}
 
 @app.post("/feedback")
-async def collect_feedback(feedback_dict: Feedback) -> None:
-    """Collect and log feedback."""
-    feedback_data = feedback_dict.model_dump()
-    logger.log_struct(feedback_data, severity="INFO")
+def collect_feedback(feedback: Feedback) -> dict[str, str]:
+    """Collect and log feedback.
+
+    Args:
+        feedback: The feedback data to log
+
+    Returns:
+        Success message
+    """
+    logger.log_struct(feedback.model_dump(), severity="INFO")
+    return {"status": "success"}
 
 
+# Main execution
 if __name__ == "__main__":
     import uvicorn
 
-    uvicorn.run(app, host="0.0.0.0", port=8000, log_level="debug")
+    uvicorn.run(app, host="0.0.0.0", port=8000)

src/resources/docs/adk-cheatsheet.md

@@ -1193,7 +1193,7 @@ import asyncio
 
 from google.adk.runners import Runner
 from google.adk.sessions import InMemorySessionService
-from app.agent import root_agent
+from {{cookiecutter.agent_directory}}.agent import root_agent
 from google.genai import types as genai_types
 
 
@@ -1201,10 +1201,10 @@ async def main():
     """Runs the agent with a sample query."""
     session_service = InMemorySessionService()
     await session_service.create_session(
-        app_name="app", user_id="test_user", session_id="test_session"
+        app_name="{{cookiecutter.agent_directory}}", user_id="test_user", session_id="test_session"
     )
     runner = Runner(
-        agent=root_agent, app_name="app", session_service=session_service
+        agent=root_agent, app_name="{{cookiecutter.agent_directory}}", session_service=session_service
     )
     query = "I want a recipe for pancakes"
     async for event in runner.run_async(

src/base_template/app/__init__.py

@@ -1,3 +0,0 @@
-from app.agent import root_agent
-
-__all__ = ["root_agent"]

src/deployment_targets/cloud_run/app/server.py

@@ -1,206 +0,0 @@
-# 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.
-{% 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
-{%- if cookiecutter.session_type == "agent_engine" %}
-from vertexai import agent_engines
-{%- endif %}
-
-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())
-provider.add_span_processor(processor)
-trace.set_tracer_provider(provider)
-
-AGENT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-
-{%- if cookiecutter.session_type == "alloydb" %}
-# AlloyDB session configuration
-db_user = os.environ.get("DB_USER", "postgres")
-db_name = os.environ.get("DB_NAME", "postgres")
-db_pass = os.environ.get("DB_PASS")
-db_host = os.environ.get("DB_HOST")
-
-# Set session_service_uri if database credentials are available
-session_service_uri = None
-if db_host and db_pass:
-    session_service_uri = f"postgresql://{db_user}:{db_pass}@{db_host}:5432/{db_name}"
-{%- elif cookiecutter.session_type == "agent_engine" %}
-# Agent Engine session configuration
-# Use environment variable for agent name, default to project name
-agent_name = os.environ.get("AGENT_ENGINE_SESSION_NAME", "{{cookiecutter.project_name}}")
-
-# Check if an agent with this name already exists
-existing_agents = list(agent_engines.list(filter=f"display_name={agent_name}"))
-
-if existing_agents:
-    # Use the existing agent
-    agent_engine = existing_agents[0]
-else:
-    # Create a new agent if none exists
-    agent_engine = agent_engines.create(display_name=agent_name)
-
-session_service_uri = f"agentengine://{agent_engine.resource_name}"
-{%- else %}
-# In-memory session configuration - no persistent storage
-session_service_uri = None
-{%- endif %}
-
-app: FastAPI = get_fast_api_app(
-    agents_dir=AGENT_DIR,
-    web=True,
-    artifact_service_uri=bucket_name,
-    allow_origins=allow_origins,
-    session_service_uri=session_service_uri,
-)
-app.title = "{{cookiecutter.project_name}}"
-app.description = "API for interacting with the Agent {{cookiecutter.project_name}}"
-{%- else %}
-import logging
-import os
-from collections.abc import Generator
-
-from fastapi import FastAPI
-from fastapi.responses import RedirectResponse, StreamingResponse
-from google.cloud import logging as google_cloud_logging
-from langchain_core.runnables import RunnableConfig
-from traceloop.sdk import Instruments, Traceloop
-
-from app.agent import agent
-from app.utils.tracing import CloudTraceLoggingSpanExporter
-from app.utils.typing import Feedback, InputChat, Request, dumps, ensure_valid_config
-
-# Initialize FastAPI app and logging
-app = FastAPI(
-    title="{{cookiecutter.project_name}}",
-    description="API for interacting with the Agent {{cookiecutter.project_name}}",
-)
-logging_client = google_cloud_logging.Client()
-logger = logging_client.logger(__name__)
-
-# Initialize Telemetry
-try:
-    Traceloop.init(
-        app_name=app.title,
-        disable_batch=False,
-        exporter=CloudTraceLoggingSpanExporter(),
-        instruments={Instruments.LANGCHAIN, Instruments.CREW},
-    )
-except Exception as e:
-    logging.error("Failed to initialize Telemetry: %s", str(e))
-
-
-def set_tracing_properties(config: RunnableConfig) -> None:
-    """Sets tracing association properties for the current request.
-
-    Args:
-        config: Optional RunnableConfig containing request metadata
-    """
-    Traceloop.set_association_properties(
-        {
-            "log_type": "tracing",
-            "run_id": str(config.get("run_id", "None")),
-            "user_id": config["metadata"].pop("user_id", "None"),
-            "session_id": config["metadata"].pop("session_id", "None"),
-            "commit_sha": os.environ.get("COMMIT_SHA", "None"),
-        }
-    )
-
-
-def stream_messages(
-    input: InputChat,
-    config: RunnableConfig | None = None,
-) -> Generator[str, None, None]:
-    """Stream events in response to an input chat.
-
-    Args:
-        input: The input chat messages
-        config: Optional configuration for the runnable
-
-    Yields:
-        JSON serialized event data
-    """
-    config = ensure_valid_config(config=config)
-    set_tracing_properties(config)
-    input_dict = input.model_dump()
-
-    for data in agent.stream(input_dict, config=config, stream_mode="messages"):  # type: ignore[arg-type]
-        yield dumps(data) + "\n"
-
-
-# Routes
-@app.get("/", response_class=RedirectResponse)
-def redirect_root_to_docs() -> RedirectResponse:
-    """Redirect the root URL to the API documentation."""
-    return RedirectResponse(url="/docs")
-
-
-@app.post("/stream_messages")
-def stream_chat_events(request: Request) -> StreamingResponse:
-    """Stream chat events in response to an input request.
-
-    Args:
-        request: The chat request containing input and config
-
-    Returns:
-        Streaming response of chat events
-    """
-    return StreamingResponse(
-        stream_messages(input=request.input, config=request.config),
-        media_type="text/event-stream",
-    )
-{%- endif %}
-
-
-@app.post("/feedback")
-def collect_feedback(feedback: Feedback) -> dict[str, str]:
-    """Collect and log feedback.
-
-    Args:
-        feedback: The feedback data to log
-
-    Returns:
-        Success message
-    """
-    logger.log_struct(feedback.model_dump(), severity="INFO")
-    return {"status": "success"}
-
-
-# Main execution
-if __name__ == "__main__":
-    import uvicorn
-
-    uvicorn.run(app, host="0.0.0.0", port=8000)

src/frontends/adk_gemini_fullstack/frontend/components.json

@@ -1,21 +0,0 @@
-{
-  "$schema": "https://ui.shadcn.com/schema.json",
-  "style": "new-york",
-  "rsc": false,
-  "tsx": true,
-  "tailwind": {
-    "config": "",
-    "css": "src/app.css",
-    "baseColor": "neutral",
-    "cssVariables": true,
-    "prefix": ""
-  },
-  "aliases": {
-    "components": "@/components",
-    "utils": "@/lib/utils",
-    "ui": "@/components/ui",
-    "lib": "@/lib",
-    "hooks": "@/hooks"
-  },
-  "iconLibrary": "lucide"
-}

src/frontends/adk_gemini_fullstack/frontend/eslint.config.js

@@ -1,28 +0,0 @@
-import js from '@eslint/js'
-import globals from 'globals'
-import reactHooks from 'eslint-plugin-react-hooks'
-import reactRefresh from 'eslint-plugin-react-refresh'
-import tseslint from 'typescript-eslint'
-
-export default tseslint.config(
-  { ignores: ['dist'] },
-  {
-    extends: [js.configs.recommended, ...tseslint.configs.recommended],
-    files: ['**/*.{ts,tsx}'],
-    languageOptions: {
-      ecmaVersion: 2020,
-      globals: globals.browser,
-    },
-    plugins: {
-      'react-hooks': reactHooks,
-      'react-refresh': reactRefresh,
-    },
-    rules: {
-      ...reactHooks.configs.recommended.rules,
-      'react-refresh/only-export-components': [
-        'warn',
-        { allowConstantExport: true },
-      ],
-    },
-  },
-)

src/frontends/adk_gemini_fullstack/frontend/index.html

@@ -1,12 +0,0 @@
-<!doctype html>
-<html lang="en">
-  <head>
-    <meta charset="UTF-8" />
-    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>ADK Fullstack Agent</title>
-  </head>
-  <body>
-    <div id="root"></div>
-    <script type="module" src="/src/main.tsx"></script>
-  </body>
-</html>