Checkout ADK sample from adk-sample-9-29

Author: aabmass

adk-sql-agent/.dockerignore

@@ -0,0 +1 @@
+Dockerfile

adk-sql-agent/.gitignore

@@ -0,0 +1 @@
+*.db

adk-sql-agent/.python-version

@@ -0,0 +1 @@
+3.13

adk-sql-agent/Dockerfile

@@ -0,0 +1,21 @@
+# NOTE: this must be run from the repo root as build context
+
+# Adapted from https://docs.astral.sh/uv/guides/integration/docker/#intermediate-layers
+# Install uv
+FROM python:3.13-slim
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
+
+RUN apt-get update && apt-get install -y git
+
+# Change the working directory to the `app` directory
+WORKDIR /app
+
+# Copy the project into the image
+COPY . ./
+
+WORKDIR /app/adk-sql-agent
+
+# Sync the project
+RUN --mount=type=cache,target=/root/.cache/uv uv sync --locked
+
+ENTRYPOINT [ "uv", "run", "--locked", "--env-file=main.env", "main.py" ]

adk-sql-agent/README.md

@@ -0,0 +1,58 @@
+# OpenTelemetry ADK instrumentation example
+
+<!-- TODO: link to devsite doc once it is published -->
+
+This demo shows does remote uploading and with `.ref` attributes in the telemetry, following OTel semconv 1.37 for GenAI: https://github.com/open-telemetry/semantic-conventions/blob/v1.37.0/docs/gen-ai/gen-ai-events.md
+
+## Configure upload
+This demo uses the [fsspec](https://filesystem-spec.readthedocs.io/) library to do the upload
+in a vendor agnostic way, which is configured in an environment variable
+`OTEL_PYTHON_GENAI_UPLOADER_PATH`. For example, to upload to GCS bucket `my-bucket` with
+subdirectory `v1/`:
+
+```console
+export OTEL_PYTHON_GENAI_UPLOADER_PATH="gs://my-bucket/v1"
+```
+
+This will work with any installed fsspec implementations and also allows using any fsspec
+configuration like URL Chaining:
+- [Built in implementations](https://filesystem-spec.readthedocs.io/en/latest/api.html#built-in-implementations)
+- [Known third party impls](https://filesystem-spec.readthedocs.io/en/latest/api.html#external-implementations)
+- [URL Chaining](https://filesystem-spec.readthedocs.io/en/latest/features.html#url-chaining)
+
+## Run the demo
+
+To run the app, set some environment variables (alternatively can be put in `main.env`):
+
+```console
+# Port if not default
+export PORT=8000
+# Project for ADC
+export GOOGLE_CLOUD_PROJECT=my-project
+
+# Configure the upload path
+export OTEL_PYTHON_GENAI_UPLOADER_PATH="gs://${GOOGLE_CLOUD_PROJECT}/v1"
+
+uv run --env-file main.env main.py
+```
+
+## Dockerfile
+
+Build
+
+```console
+# from this directory, run with repo root build context
+docker build -f ./Dockerfile -t adk-sql-agent-write-aside ..
+```
+
+Run:
+```console
+docker run --rm \
+    -p 8000:8000 \
+    -e PORT=8000 \
+    -e GOOGLE_APPLICATION_CREDENTIALS \
+    -e GOOGLE_CLOUD_PROJECT \
+    -e OTEL_PYTHON_GENAI_UPLOADER_PATH="gs://otel-starter-project-genai-refs/v1" \
+    -v "$GOOGLE_APPLICATION_CREDENTIALS:$GOOGLE_APPLICATION_CREDENTIALS" \
+    adk-sql-agent-write-aside:latest
+```

adk-sql-agent/image_describer_agent/.env

@@ -0,0 +1,3 @@
+GOOGLE_GENAI_USE_VERTEXAI=1
+GOOGLE_CLOUD_PROJECT=otel-starter-project
+GOOGLE_CLOUD_LOCATION=us-central1

adk-sql-agent/image_describer_agent/__init__.py

@@ -0,0 +1 @@
+from . import agent

adk-sql-agent/image_describer_agent/agent.py

@@ -0,0 +1,18 @@
+from textwrap import dedent
+
+from google.adk.agents.llm_agent import Agent
+
+root_agent = Agent(
+    model="gemini-2.5-flash",
+    name="root_agent",
+    description="An agent that can describe images",
+    instruction=dedent("""\
+    You are an agent that can describe images. You don't have any tools at your disposal.
+    Some guidelines for you:
+
+    - Don't answer any questions that are opinion based or involve emotions.
+    - Keep it brief and succinct
+    - But also be descriptive
+    - Answer in a professional tone
+    """),
+)

adk-sql-agent/main.env

@@ -0,0 +1,7 @@
+OTEL_PYTHON_LOGGING_AUTO_INSTRUMENTATION_ENABLED=true
+OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT=true
+
+GOOGLE_GENAI_USE_VERTEXAI=TRUE
+GOOGLE_CLOUD_LOCATION=us-central1
+
+ENABLE_GCS_PYTHON_CLIENT_OTEL_TRACES=true

adk-sql-agent/main.py

@@ -0,0 +1,167 @@
+# 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
+#
+#      https://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.
+
+import logging
+import logging.handlers
+import os
+
+import debugpy
+import google.auth
+import google.auth.transport.requests
+import grpc
+import uvicorn
+from fastapi import FastAPI
+from google.adk.cli.fast_api import get_fast_api_app
+from google.auth.transport.grpc import AuthMetadataPlugin
+
+from opentelemetry import _events as events
+from opentelemetry import _logs as logs
+from opentelemetry import metrics, trace
+from opentelemetry.exporter.cloud_logging import CloudLoggingExporter
+from opentelemetry.exporter.cloud_monitoring import (
+    CloudMonitoringMetricsExporter,
+)
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+    OTLPSpanExporter,
+)
+from opentelemetry.instrumentation.aiohttp_client import (
+    AioHttpClientInstrumentor,
+)
+from opentelemetry.instrumentation.google_genai import (
+    GoogleGenAiSdkInstrumentor,
+)
+from opentelemetry.instrumentation.vertexai import VertexAIInstrumentor
+from opentelemetry.sdk._events import EventLoggerProvider
+from opentelemetry.sdk._logs import LoggerProvider
+from opentelemetry.sdk._logs.export import (
+    BatchLogRecordProcessor,
+    ConsoleLogExporter,
+)
+from opentelemetry.sdk._logs import LoggingHandler
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.sdk.metrics.export import PeriodicExportingMetricReader
+from opentelemetry.sdk.resources import SERVICE_NAME, Resource
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+
+# Get the directory where main.py is located
+AGENT_DIR = os.path.dirname(os.path.abspath(__file__))
+# Example session DB URL (e.g., SQLite)
+SESSION_DB_URL = "sqlite:///./sessions.db"
+# Example allowed origins for CORS
+ALLOWED_ORIGINS = ["http://localhost", "http://localhost:8080", "*"]
+# Set web=True if you intend to serve a web interface, False otherwise
+SERVE_WEB_INTERFACE = True
+
+# logging.basicConfig(level=logging.INFO, handlers=[LoggingHandler()])
+logging.getLogger().addHandler(LoggingHandler())
+logging.getLogger("gcsfs").setLevel(logging.DEBUG)
+logging.getLogger("fsspec").setLevel(logging.DEBUG)
+logging.getLogger("fsspec.cached").setLevel(logging.DEBUG)
+logging.getLogger("opentelemetry").setLevel(logging.DEBUG)
+logging.getLogger("opentelemetry").addHandler(logging.StreamHandler())
+
+# debugpy.listen(5678)
+
+
+# [START opentelemetry_adk_otel_setup]
+def setup_opentelemetry() -> None:
+    credentials, project_id = google.auth.default()
+    resource = Resource.create(
+        attributes={
+            SERVICE_NAME: "adk-sql-agent",
+            # The project to send spans to
+            "gcp.project_id": project_id,
+        }
+    )
+
+    # Set up OTLP auth
+    request = google.auth.transport.requests.Request()
+    auth_metadata_plugin = AuthMetadataPlugin(
+        credentials=credentials, request=request
+    )
+    channel_creds = grpc.composite_channel_credentials(
+        grpc.ssl_channel_credentials(),
+        grpc.metadata_call_credentials(auth_metadata_plugin),
+    )
+
+    # Set up OpenTelemetry Python SDK
+    tracer_provider = TracerProvider(resource=resource)
+    tracer_provider.add_span_processor(
+        BatchSpanProcessor(
+            OTLPSpanExporter(
+                credentials=channel_creds,
+                endpoint="https://telemetry.googleapis.com:443/v1/traces",
+            )
+        )
+    )
+    trace.set_tracer_provider(tracer_provider)
+
+    logger_provider = LoggerProvider(resource=resource)
+    logger_provider.add_log_record_processor(
+        BatchLogRecordProcessor(CloudLoggingExporter())
+    )
+    # logger_provider.add_log_record_processor(
+    #     BatchLogRecordProcessor(
+    #         ConsoleLogExporter(out=open("./logs.jsonl", "w+"))
+    #     )
+    # )
+    logs.set_logger_provider(logger_provider)
+
+    event_logger_provider = EventLoggerProvider(logger_provider)
+    events.set_event_logger_provider(event_logger_provider)
+
+    reader = PeriodicExportingMetricReader(CloudMonitoringMetricsExporter())
+    meter_provider = MeterProvider(metric_readers=[reader], resource=resource)
+    metrics.set_meter_provider(meter_provider)
+
+    # Set up prompt uploader
+    # if OTEL_PYTHON_UPLOADER_PATH:
+    #     set_uploader(GcsUploader(bucket_path=OTEL_PYTHON_GCS_UPLOAD_BUCKET))
+
+    # Load instrumentors
+    # SQLite3Instrumentor().instrument()
+    VertexAIInstrumentor().instrument()
+    GoogleGenAiSdkInstrumentor().instrument()
+    AioHttpClientInstrumentor().instrument()
+
+
+# [END opentelemetry_adk_otel_setup]
+
+
+def main() -> None:
+    # Make sure to set:
+    # OTEL_PYTHON_LOGGING_AUTO_INSTRUMENTATION_ENABLED=true
+    # OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT=true
+    # in order to full prompts and responses and logs messages.
+    setup_opentelemetry()
+
+    # Call the function to get the FastAPI app instance
+    # Ensure the agent directory name ('capital_agent') matches your agent folder
+    app: FastAPI = get_fast_api_app(
+        agents_dir=AGENT_DIR,
+        session_service_uri=SESSION_DB_URL,
+        allow_origins=ALLOWED_ORIGINS,
+        web=SERVE_WEB_INTERFACE,
+    )
+
+    uvicorn.run(
+        app,
+        host="0.0.0.0",
+        port=int(os.environ.get("PORT", 8080)),
+    )
+
+
+if __name__ == "__main__":
+    main()