raindrop-google-adk 0.0.1__tar.gz

raindrop_google_adk-0.0.1/.gitignore

@@ -0,0 +1,90 @@
+.zodprompt-eval.sqlite
+.topicmatch-eval.sqlite
+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+lerna-debug.log*
+env/
+temp
+.env*
+node_modules
+dist
+dist-ssr
+*.local
+.env
+.yarn
+.yarnrc.yml
+.pnpm-store
+package-lock.json
+.turbo
+*.tsbuildinfo
+apps/**/secrets/**
+*/*/eval-results
+apps/dashboard/eval-results/metadata/
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+!.vscode/settings.json
+!.vscode/tasks.json
+!.vscode/launch.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?
+*bite-sight-prompts.txt
+
+# Python files
+*.pyc
+__pycache__/
+.venv/
+
+myenv/
+
+build/
+dawnai.egg-info/
+
+.next/
+
+.vercel
+
+.extra
+
+packages/scripts/local-configs
+
+gcp-auth.json
+
+.crush/
+mcp.json
+
+# Don't commit local dev setup SST state
+infra/.sst
+plans/
+.evals/
+apps/dawn/app/api/internal/evals/
+.sst/
+sample-*.json
+results.txt
+
+# Dev logs
+.logs/
+tmp/
+
+# Supabase local
+**/supabase/.branches
+**/supabase/.temp
+**/supabase/seeds/
+
+# Coding Agents
+.claude/
+**/opencode/
+**/opencode/**
+
+.zed/
+

raindrop_google_adk-0.0.1/PKG-INFO

@@ -0,0 +1,201 @@
+Metadata-Version: 2.4
+Name: raindrop-google-adk
+Version: 0.0.1
+Summary: Raindrop integration for Google ADK (Agent Development Kit)
+Author-email: Raindrop AI <sdk@raindrop.ai>
+License: MIT
+Requires-Python: >=3.10
+Requires-Dist: raindrop-ai>=0.0.42
+Provides-Extra: adk
+Requires-Dist: google-adk>=1.14.1; extra == 'adk'
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: requests>=2.28; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# raindrop-google-adk
+
+Raindrop integration for [Google ADK](https://github.com/google/adk-python) (Agent Development Kit) — Google's framework for building AI agents powered by Gemini models.
+
+Wraps Google ADK `Runner` objects to automatically capture agent invocations, tool calls, and multi-step reasoning, shipping telemetry to Raindrop.
+
+## Installation
+
+```bash
+pip install raindrop-google-adk google-adk
+```
+
+## Quick Start
+
+```python
+from raindrop_google_adk import setup_google_adk
+from google.adk import Runner
+from google.adk.agents import LlmAgent
+from google.adk.sessions import InMemorySessionService
+from google.genai import types
+
+# Enable automatic tracing for all ADK Runner interactions
+rd = setup_google_adk(api_key="your-write-key")
+
+# Create your ADK agent as normal
+def get_weather(city: str) -> dict:
+    """Get weather for a city."""
+    return {"temperature": 72, "condition": "sunny", "city": city}
+
+agent = LlmAgent(
+    name="weather_assistant",
+    tools=[get_weather],
+    model="gemini-2.5-flash",
+    instruction="You are a helpful assistant that can check weather.",
+)
+
+# Create session and runner
+session_service = InMemorySessionService()
+runner = Runner(app_name="weather_app", agent=agent, session_service=session_service)
+
+# Use the runner as normal — all interactions are automatically traced
+user_msg = types.Content(
+    parts=[types.Part(text="What's the weather in New York?")],
+    role="user",
+)
+for event in runner.run(user_id="user123", session_id="session123", new_message=user_msg):
+    print(event)
+```
+
+## What Gets Traced
+
+The Google ADK integration automatically captures:
+
+- **Runner invocations** — input message, user_id, session_id, app_name
+- **Agent responses** — final output text from the agent
+- **Token usage** — prompt_tokens, completion_tokens, total_tokens from usage metadata
+- **Tool calls** — count of function calls in the response
+- **Model info** — model version when available
+- **Agent identity** — agent name, author from events
+- **Errors** — captured (with error message) and re-raised to the caller
+- **Async support** — both `run()` (sync) and `run_async()` (async) are instrumented
+
+## Configuration
+
+### Auto-instrumentation (recommended)
+
+```python
+from raindrop_google_adk import setup_google_adk
+
+rd = setup_google_adk(
+    api_key="your-write-key",  # Required: your Raindrop API key (or set RAINDROP_API_KEY env var)
+    user_id="user-123",    # Optional: default user ID for all events
+    convo_id="convo-456",  # Optional: conversation/thread ID
+)
+
+# All Runner.run() and Runner.run_async() calls are now automatically traced
+# Call rd.shutdown() before process exit
+```
+
+### Manual wrapping
+
+```python
+from raindrop_google_adk import create_raindrop_google_adk
+
+rd = create_raindrop_google_adk(
+    api_key="your-write-key",
+    user_id="user-123",
+)
+
+# Wrap a specific runner instance
+wrapped_runner = rd.wrap(runner)
+
+# Use wrapped_runner as normal
+for event in wrapped_runner.run(...):
+    print(event)
+
+rd.shutdown()
+```
+
+### Class-based API
+
+```python
+from raindrop_google_adk import RaindropGoogleADK
+
+rd = RaindropGoogleADK(api_key="your-write-key", user_id="user-123")
+
+# Auto-patch all Runner instances
+rd.setup()
+
+# Or wrap a specific runner
+wrapped = rd.wrap(runner)
+
+# Identify users and track signals
+rd.identify(user_id="user-123", properties={"plan": "pro"})
+rd.track_signal(user_id="user-123", signal_name="thumbs_up")
+
+# Cleanup
+rd.shutdown()
+```
+
+## Async Usage
+
+The wrapper supports both sync and async runner usage:
+
+```python
+import asyncio
+
+async def main():
+    user_msg = types.Content(
+        parts=[types.Part(text="What's the weather?")],
+        role="user",
+    )
+    async for event in runner.run_async(
+        user_id="user123",
+        session_id="session123",
+        new_message=user_msg,
+    ):
+        if event.is_final_response():
+            print(event.content.parts[0].text)
+
+    rd.shutdown()
+
+asyncio.run(main())
+```
+
+## Captured Properties
+
+Each event includes the following properties when available:
+
+| Property | Description |
+|----------|-------------|
+| `ai.usage.prompt_tokens` | Input token count |
+| `ai.usage.completion_tokens` | Output token count |
+| `ai.usage.total_tokens` | Total token count |
+| `google_adk.app_name` | Runner app name |
+| `google_adk.user_id` | User ID from the runner call |
+| `google_adk.session_id` | Session ID from the runner call |
+| `google_adk.author` | Event author (agent name) |
+| `google_adk.agent_name` | Agent name from the event |
+| `google_adk.tool_calls_count` | Number of tool calls in the run |
+| `google_adk.tool_call_names` | JSON list of tool names called |
+| `google_adk.error` | Whether a Python exception occurred |
+| `google_adk.error_message` | Python exception message if applicable |
+| `google_adk.error_code` | LLM-level error code (e.g. SAFETY, QUOTA) |
+| `google_adk.llm_error_message` | LLM-level error message |
+| `google_adk.agent_branch` | Agent hierarchy path for multi-agent setups |
+| `google_adk.finish_reason` | Why generation stopped (STOP, SAFETY, etc.) |
+| `ai.usage.cached_tokens` | Cached content token count |
+| `ai.usage.thoughts_tokens` | Thinking/reasoning token count |
+
+## Flushing and Shutdown
+
+Always call `shutdown()` before your process exits to ensure all telemetry is shipped:
+
+```python
+rd.shutdown()  # flush + release resources
+```
+
+## Testing
+
+```bash
+cd packages/google-adk-python
+pip install -e ".[dev]"
+pytest
+```

raindrop_google_adk-0.0.1/README.md

@@ -0,0 +1,185 @@
+# raindrop-google-adk
+
+Raindrop integration for [Google ADK](https://github.com/google/adk-python) (Agent Development Kit) — Google's framework for building AI agents powered by Gemini models.
+
+Wraps Google ADK `Runner` objects to automatically capture agent invocations, tool calls, and multi-step reasoning, shipping telemetry to Raindrop.
+
+## Installation
+
+```bash
+pip install raindrop-google-adk google-adk
+```
+
+## Quick Start
+
+```python
+from raindrop_google_adk import setup_google_adk
+from google.adk import Runner
+from google.adk.agents import LlmAgent
+from google.adk.sessions import InMemorySessionService
+from google.genai import types
+
+# Enable automatic tracing for all ADK Runner interactions
+rd = setup_google_adk(api_key="your-write-key")
+
+# Create your ADK agent as normal
+def get_weather(city: str) -> dict:
+    """Get weather for a city."""
+    return {"temperature": 72, "condition": "sunny", "city": city}
+
+agent = LlmAgent(
+    name="weather_assistant",
+    tools=[get_weather],
+    model="gemini-2.5-flash",
+    instruction="You are a helpful assistant that can check weather.",
+)
+
+# Create session and runner
+session_service = InMemorySessionService()
+runner = Runner(app_name="weather_app", agent=agent, session_service=session_service)
+
+# Use the runner as normal — all interactions are automatically traced
+user_msg = types.Content(
+    parts=[types.Part(text="What's the weather in New York?")],
+    role="user",
+)
+for event in runner.run(user_id="user123", session_id="session123", new_message=user_msg):
+    print(event)
+```
+
+## What Gets Traced
+
+The Google ADK integration automatically captures:
+
+- **Runner invocations** — input message, user_id, session_id, app_name
+- **Agent responses** — final output text from the agent
+- **Token usage** — prompt_tokens, completion_tokens, total_tokens from usage metadata
+- **Tool calls** — count of function calls in the response
+- **Model info** — model version when available
+- **Agent identity** — agent name, author from events
+- **Errors** — captured (with error message) and re-raised to the caller
+- **Async support** — both `run()` (sync) and `run_async()` (async) are instrumented
+
+## Configuration
+
+### Auto-instrumentation (recommended)
+
+```python
+from raindrop_google_adk import setup_google_adk
+
+rd = setup_google_adk(
+    api_key="your-write-key",  # Required: your Raindrop API key (or set RAINDROP_API_KEY env var)
+    user_id="user-123",    # Optional: default user ID for all events
+    convo_id="convo-456",  # Optional: conversation/thread ID
+)
+
+# All Runner.run() and Runner.run_async() calls are now automatically traced
+# Call rd.shutdown() before process exit
+```
+
+### Manual wrapping
+
+```python
+from raindrop_google_adk import create_raindrop_google_adk
+
+rd = create_raindrop_google_adk(
+    api_key="your-write-key",
+    user_id="user-123",
+)
+
+# Wrap a specific runner instance
+wrapped_runner = rd.wrap(runner)
+
+# Use wrapped_runner as normal
+for event in wrapped_runner.run(...):
+    print(event)
+
+rd.shutdown()
+```
+
+### Class-based API
+
+```python
+from raindrop_google_adk import RaindropGoogleADK
+
+rd = RaindropGoogleADK(api_key="your-write-key", user_id="user-123")
+
+# Auto-patch all Runner instances
+rd.setup()
+
+# Or wrap a specific runner
+wrapped = rd.wrap(runner)
+
+# Identify users and track signals
+rd.identify(user_id="user-123", properties={"plan": "pro"})
+rd.track_signal(user_id="user-123", signal_name="thumbs_up")
+
+# Cleanup
+rd.shutdown()
+```
+
+## Async Usage
+
+The wrapper supports both sync and async runner usage:
+
+```python
+import asyncio
+
+async def main():
+    user_msg = types.Content(
+        parts=[types.Part(text="What's the weather?")],
+        role="user",
+    )
+    async for event in runner.run_async(
+        user_id="user123",
+        session_id="session123",
+        new_message=user_msg,
+    ):
+        if event.is_final_response():
+            print(event.content.parts[0].text)
+
+    rd.shutdown()
+
+asyncio.run(main())
+```
+
+## Captured Properties
+
+Each event includes the following properties when available:
+
+| Property | Description |
+|----------|-------------|
+| `ai.usage.prompt_tokens` | Input token count |
+| `ai.usage.completion_tokens` | Output token count |
+| `ai.usage.total_tokens` | Total token count |
+| `google_adk.app_name` | Runner app name |
+| `google_adk.user_id` | User ID from the runner call |
+| `google_adk.session_id` | Session ID from the runner call |
+| `google_adk.author` | Event author (agent name) |
+| `google_adk.agent_name` | Agent name from the event |
+| `google_adk.tool_calls_count` | Number of tool calls in the run |
+| `google_adk.tool_call_names` | JSON list of tool names called |
+| `google_adk.error` | Whether a Python exception occurred |
+| `google_adk.error_message` | Python exception message if applicable |
+| `google_adk.error_code` | LLM-level error code (e.g. SAFETY, QUOTA) |
+| `google_adk.llm_error_message` | LLM-level error message |
+| `google_adk.agent_branch` | Agent hierarchy path for multi-agent setups |
+| `google_adk.finish_reason` | Why generation stopped (STOP, SAFETY, etc.) |
+| `ai.usage.cached_tokens` | Cached content token count |
+| `ai.usage.thoughts_tokens` | Thinking/reasoning token count |
+
+## Flushing and Shutdown
+
+Always call `shutdown()` before your process exits to ensure all telemetry is shipped:
+
+```python
+rd.shutdown()  # flush + release resources
+```
+
+## Testing
+
+```bash
+cd packages/google-adk-python
+pip install -e ".[dev]"
+pytest
+```

raindrop_google_adk-0.0.1/pyproject.toml

@@ -0,0 +1,32 @@
+[project]
+name = "raindrop-google-adk"
+version = "0.0.1"
+description = "Raindrop integration for Google ADK (Agent Development Kit)"
+readme = "README.md"
+license = {text = "MIT"}
+requires-python = ">=3.10"
+authors = [
+    {name = "Raindrop AI", email = "sdk@raindrop.ai"},
+]
+dependencies = [
+    "raindrop-ai>=0.0.42",
+]
+
+[project.optional-dependencies]
+adk = ["google-adk>=1.14.1"]
+dev = ["pytest>=8.0", "pytest-asyncio>=0.23.0", "requests>=2.28"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+addopts = ["-v", "--tb=short"]
+asyncio_mode = "auto"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["raindrop_google_adk"]

raindrop_google_adk-0.0.1/raindrop_google_adk/__init__.py

@@ -0,0 +1,4 @@
+from .wrapper import RaindropGoogleADK, create_raindrop_google_adk, setup_google_adk
+
+__version__ = "0.0.1"
+__all__ = ["RaindropGoogleADK", "create_raindrop_google_adk", "setup_google_adk", "__version__"]