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

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>