sensu-sdk 0.6.0__tar.gz

sensu_sdk-0.6.0/.gitignore

@@ -0,0 +1,93 @@
+# ============================================================
+# Environment & Secrets — NEVER commit these
+# ============================================================
+# Claude Code project settings may contain API keys in allowed-command lists
+.claude/
+.env
+.env.*
+**/.env
+**/.env.*
+!.env.example
+!**/.env.example
+
+# ============================================================
+# Node
+# ============================================================
+node_modules/
+.pnp
+.pnp.js
+
+# ============================================================
+# Build outputs
+# ============================================================
+dist/
+build/
+**/dist/
+**/build/
+apps/vscode-extension/out/
+apps/vscode-extension/.vsix-stage/
+apps/vscode-extension/*.vsix
+
+# ============================================================
+# Logs
+# ============================================================
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+
+# ============================================================
+# OS / Editor
+# ============================================================
+.DS_Store
+Thumbs.db
+.vscode/settings.json
+.idea/
+*.swp
+*.swo
+
+# ============================================================
+# Testing / Coverage
+# ============================================================
+coverage/
+.nyc_output/
+
+# ============================================================
+# Vite / Turbo caches
+# ============================================================
+.vite/
+.turbo/
+
+# ============================================================
+# VS Code extension
+# ============================================================
+apps/web/.env.local
+
+# ============================================================
+# pnpm
+# ============================================================
+.pnpm-store/
+
+
+COMPANY_INFO.md
+DEPLOYMENT_STRATEGY.md
+MONETIZATION_STRATEGY.md
+PLANNING_FOR_PHASE_1.md
+PLANNING_FOR_PHASE_*.md
+EVENT_TRACKING_VALUE_ADD.md
+STYLING_SUGGESTIONS.md
+PYTHON_DISCOVERY.md
+QUOTA_ENFORCEMENT_STRATEGY.md
+PRODUCT_DIFFERENTIATION_ENHANCEMENTS.md
+PLANNED_PRODUCT_ENHANCEMENTS.md
+FIX_CURRENTLY_BROKEN.md
+CURRENT_DEVELOPMENT_ITEMS.md
+DEMO_IDEAS.md
+FURTHER_RESEARCH_FOR_YCOMBINATOR.md
+LIVE_PRICING_FEATURE.md
+.npmrc
+
+**/.npmrc
+packages/.npmrc
+

sensu_sdk-0.6.0/PKG-INFO

@@ -0,0 +1,109 @@
+Metadata-Version: 2.4
+Name: sensu-sdk
+Version: 0.6.0
+Summary: Python SDK for the Sensu AI observability platform
+Project-URL: Homepage, https://sensu-ai.com
+Project-URL: Repository, https://github.com/sensu-inc/sensu
+Author: Sensu Inc
+License: MIT
+Keywords: agents,ai,llm,observability,telemetry
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.9
+Requires-Dist: httpx>=0.25.0
+Requires-Dist: typing-extensions>=4.0; python_version < '3.11'
+Provides-Extra: all
+Requires-Dist: anthropic>=0.20.0; extra == 'all'
+Requires-Dist: langchain>=0.1.0; extra == 'all'
+Requires-Dist: openai>=1.0.0; extra == 'all'
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.20.0; extra == 'anthropic'
+Provides-Extra: dev
+Requires-Dist: anyio[trio]; extra == 'dev'
+Requires-Dist: mypy>=1.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Requires-Dist: ruff; extra == 'dev'
+Provides-Extra: langchain
+Requires-Dist: langchain>=0.1.0; extra == 'langchain'
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == 'openai'
+Description-Content-Type: text/markdown
+
+# sensu-sdk
+
+Python SDK for the [Sensu](https://sensu-ai.com) AI observability platform.
+
+> Migrating from `senzu-sdk`? The package was renamed; install `sensu-sdk` and replace `from senzu import SenzuClient` with `from sensu import SensuClient`. The legacy `senzu-sdk` v0.5.x package will continue to work but receives no further updates.
+
+## Installation
+
+```bash
+pip install sensu-sdk                    # core
+pip install "sensu-sdk[anthropic]"       # + Anthropic auto-tracking
+pip install "sensu-sdk[openai]"          # + OpenAI auto-tracking
+pip install "sensu-sdk[langchain]"       # + LangChain callback handler
+pip install "sensu-sdk[all]"             # everything
+```
+
+## Quick start
+
+```python
+import sensu
+
+client = sensu.SensuClient({
+    "api_key": "your-api-key",
+    "agent_id": "your-agent-id",
+    "org_id": "your-org-id",
+})
+
+# High-level API with automatic context propagation
+async def handle_request():
+    async with client.run({"session_id": "abc"}) as run:
+        step = run.start_step({"name": "fetch", "step_type": "tool"})
+        result = await step.track_tool({"tool_name": "search", "fn": search})
+        await step.end()
+
+# Anthropic auto-tracking
+from anthropic import AsyncAnthropic
+from sensu.integrations.anthropic import WrapAnthropicOptions, wrap_anthropic
+
+anthropic = wrap_anthropic(
+    AsyncAnthropic(),
+    WrapAnthropicOptions(client=client),
+)
+
+async def chat():
+    async with client.run({}) as run:
+        # messages.create() is tracked automatically
+        return await anthropic.messages.create(
+            model="claude-sonnet-4-6",
+            max_tokens=1024,
+            messages=[{"role": "user", "content": "Hello"}],
+        )
+```
+
+## Environment variables
+
+| Variable | Description |
+|---|---|
+| `SENSU_API_KEY` | API key |
+| `SENSU_BASE_URL` | API base URL (default: `http://localhost:3001`) |
+| `SENSU_AGENT_ID` | Agent ID |
+| `SENSU_ORG_ID` | Organisation ID |
+
+> The legacy `SENZU_*` names are still read as a fallback and emit a deprecation warning. They will be removed in a future release.
+
+Use `from_env=True` to load from environment:
+
+```python
+client = sensu.SensuClient({"from_env": True})
+```
+
+## License
+
+MIT

sensu_sdk-0.6.0/README.md

@@ -0,0 +1,73 @@
+# sensu-sdk
+
+Python SDK for the [Sensu](https://sensu-ai.com) AI observability platform.
+
+> Migrating from `senzu-sdk`? The package was renamed; install `sensu-sdk` and replace `from senzu import SenzuClient` with `from sensu import SensuClient`. The legacy `senzu-sdk` v0.5.x package will continue to work but receives no further updates.
+
+## Installation
+
+```bash
+pip install sensu-sdk                    # core
+pip install "sensu-sdk[anthropic]"       # + Anthropic auto-tracking
+pip install "sensu-sdk[openai]"          # + OpenAI auto-tracking
+pip install "sensu-sdk[langchain]"       # + LangChain callback handler
+pip install "sensu-sdk[all]"             # everything
+```
+
+## Quick start
+
+```python
+import sensu
+
+client = sensu.SensuClient({
+    "api_key": "your-api-key",
+    "agent_id": "your-agent-id",
+    "org_id": "your-org-id",
+})
+
+# High-level API with automatic context propagation
+async def handle_request():
+    async with client.run({"session_id": "abc"}) as run:
+        step = run.start_step({"name": "fetch", "step_type": "tool"})
+        result = await step.track_tool({"tool_name": "search", "fn": search})
+        await step.end()
+
+# Anthropic auto-tracking
+from anthropic import AsyncAnthropic
+from sensu.integrations.anthropic import WrapAnthropicOptions, wrap_anthropic
+
+anthropic = wrap_anthropic(
+    AsyncAnthropic(),
+    WrapAnthropicOptions(client=client),
+)
+
+async def chat():
+    async with client.run({}) as run:
+        # messages.create() is tracked automatically
+        return await anthropic.messages.create(
+            model="claude-sonnet-4-6",
+            max_tokens=1024,
+            messages=[{"role": "user", "content": "Hello"}],
+        )
+```
+
+## Environment variables
+
+| Variable | Description |
+|---|---|
+| `SENSU_API_KEY` | API key |
+| `SENSU_BASE_URL` | API base URL (default: `http://localhost:3001`) |
+| `SENSU_AGENT_ID` | Agent ID |
+| `SENSU_ORG_ID` | Organisation ID |
+
+> The legacy `SENZU_*` names are still read as a fallback and emit a deprecation warning. They will be removed in a future release.
+
+Use `from_env=True` to load from environment:
+
+```python
+client = sensu.SensuClient({"from_env": True})
+```
+
+## License
+
+MIT

sensu_sdk-0.6.0/pyproject.toml

@@ -0,0 +1,53 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "sensu-sdk"
+version = "0.6.0"
+description = "Python SDK for the Sensu AI observability platform"
+readme = "README.md"
+requires-python = ">=3.9"
+license = { text = "MIT" }
+authors = [{ name = "Sensu Inc" }]
+keywords = ["ai", "observability", "telemetry", "llm", "agents"]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "License :: OSI Approved :: MIT License",
+]
+
+dependencies = [
+    "httpx>=0.25.0",
+    "typing_extensions>=4.0; python_version < '3.11'",
+]
+
+[project.optional-dependencies]
+anthropic = ["anthropic>=0.20.0"]
+openai    = ["openai>=1.0.0"]
+langchain = ["langchain>=0.1.0"]
+all       = ["anthropic>=0.20.0", "openai>=1.0.0", "langchain>=0.1.0"]
+dev       = [
+    "pytest>=7.0",
+    "pytest-asyncio>=0.21",
+    "anyio[trio]",
+    "mypy>=1.0",
+    "ruff",
+]
+
+[project.urls]
+Homepage   = "https://sensu-ai.com"
+Repository = "https://github.com/sensu-inc/sensu"
+
+[tool.hatch.build.targets.wheel]
+packages = ["sensu"]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+
+[tool.mypy]
+python_version = "3.9"
+strict = true

sensu_sdk-0.6.0/releases/CHANGELOG.md

@@ -0,0 +1,45 @@
+# Changelog
+
+All notable changes to `senzu-sdk` are documented here.
+
+Format: each release has a **Breaking Changes**, **Added**, **Changed**, and **Fixed** section. Sections are omitted when empty.
+
+---
+
+## [0.5.3] — 2026-05-07
+
+### Added
+- Initial public release of the Python SDK.
+- `SenzuClient` — core telemetry client with batched event flushing, async context propagation via `contextvars.ContextVar`, thread-safe buffer, and `atexit` process-exit flush.
+- `RunHandle` — represents an agent run; exposes `start_step()`, `record_feedback()`, `record_eval_score()`, `handoff()`, `end()`, and `async with` context manager support.
+- `StepHandle` — represents a step within a run; exposes `track_llm()`, `track_streaming_llm()`, `record_llm()`, `track_tool()`, `track_retrieval()`, `record_retrieval()`, `track_embedding()`, `record_embedding()`, `track_guardrail()`, `record_guardrail()`, `record_prompt_render()`, `end()`, and `async with` support.
+- `senzu.run()` high-level API — wraps an async function with automatic run start/end and ContextVar propagation so nested `track_*` calls resolve the active run implicitly.
+- Client-level shortcut methods — `track_tool()`, `track_retrieval()`, `track_embedding()`, `track_guardrail()` auto-find the active run from context.
+- Multi-agent support — `spawn_run()` (shared trace and session) and `handoff()`.
+- Session management — `start_session()` and `resume_session()`.
+- Prompt management — `deploy_prompt_version()` and `StepHandle.record_prompt_render()`.
+- Loop detection — configurable `loop_threshold` and `on_loop_detected` callback.
+- Dynamic pricing — `resolve_pricing()` fetches live rates from `GET /api/v1/pricing/models/{provider}/{model}` with per-client session cache; returns a `(0.0, 0.0)` sentinel when unavailable rather than fabricating a number.
+- `senzu.integrations.anthropic` — `wrap_anthropic()` patches `client.messages.create()` to auto-track LLM calls including cache token counts.
+- `senzu.integrations.openai` — `wrap_openai()` patches `client.chat.completions.create()` to auto-track LLM calls.
+- `senzu.integrations.langchain` — `SenzuCallbackHandler` for LangChain chains and agents; tracks LLM calls, tool calls, streaming TTFT, and retry detection.
+- Full `TypedDict` type annotations throughout; compatible with Python 3.9+.
+- GitHub Actions workflow for Trusted Publishing to PyPI on `sdk-python/v*` tag push.
+
+---
+
+## How to release
+
+1. Update `version` in [pyproject.toml](../pyproject.toml) and `__version__` in [senzu/\_\_init\_\_.py](../senzu/__init__.py).
+2. Add a release entry at the top of this file.
+3. Commit: `git commit -m "chore(sdk-python): release vX.Y.Z"`
+4. Tag and push: `git tag sdk-python/vX.Y.Z && git push origin sdk-python/vX.Y.Z`
+5. GitHub Actions builds and publishes to PyPI automatically via Trusted Publishing.
+
+## Version policy
+
+This SDK follows [Semantic Versioning](https://semver.org):
+
+- **Patch** (0.5.x) — bug fixes, non-breaking additions (new optional fields, new event types).
+- **Minor** (0.x.0) — new features, new methods, new integrations. Backwards compatible.
+- **Major** (x.0.0) — breaking changes to public method signatures or wire format.

sensu_sdk-0.6.0/sensu/__init__.py

@@ -0,0 +1,117 @@
+"""
+Sensu Python SDK — AI observability for agent systems.
+
+Quick start (async context propagation):
+
+    import sensu
+
+    client = sensu.SensuClient({"from_env": True})
+    anthropic = sensu.wrap_anthropic(Anthropic(), sensu.WrapAnthropicOptions(client=client))
+
+    async def handle_request(user_input: str) -> str:
+        async with client.run({}) as run:
+            step = run.start_step({"name": "main", "step_type": "llm"})
+            result = await anthropic.messages.create(model="claude-sonnet-4-6", ...)
+            await step.end()
+            return result.content[0].text
+
+Quick start (explicit run/step):
+
+    client = sensu.SensuClient({"api_key": "...", "agent_id": "...", "org_id": "..."})
+    run = client.start_run({"session_id": "abc"})
+    step = run.start_step({"name": "fetch", "step_type": "tool"})
+    result = await step.track_tool({"tool_name": "web_search", "fn": search})
+    await step.end()
+    await run.end()
+"""
+
+from sensu._client import RunHandle, SensuClient, StepHandle
+from sensu._types import (
+    ContextBreakdown,
+    DeployPromptVersionOptions,
+    GuardrailResult,
+    GuardrailType,
+    HandoffOptions,
+    MessageSnapshotItem,
+    RawEmbeddingOptions,
+    RawGuardrailOptions,
+    RawLlmCallOptions,
+    RawRetrievalOptions,
+    RecordEvalScoreOptions,
+    RecordFeedbackOptions,
+    RecordPromptRenderOptions,
+    ResumeSessionOptions,
+    RetrievalChunkInput,
+    RunOptions,
+    SensuClientOptions,
+    SpawnRunOptions,
+    StartRunOptions,
+    StartSessionOptions,
+    StartStepOptions,
+    TrackEmbeddingOptions,
+    TrackGuardrailOptions,
+    TrackLlmOptions,
+    TrackRetrievalOptions,
+    TrackStreamingLlmOptions,
+    TrackToolOptions,
+)
+
+# Integration option classes — re-exported for ergonomic imports
+from sensu.integrations.anthropic import WrapAnthropicOptions
+from sensu.integrations.openai import WrapOpenAIOptions
+
+
+def wrap_anthropic(anthropic_client: object, opts: WrapAnthropicOptions) -> object:
+    """Wrap an Anthropic client to auto-track messages.create() calls."""
+    from sensu.integrations.anthropic import wrap_anthropic as _wrap
+    return _wrap(anthropic_client, opts)
+
+
+def wrap_openai(openai_client: object, opts: WrapOpenAIOptions) -> object:
+    """Wrap an OpenAI client to auto-track chat.completions.create() calls."""
+    from sensu.integrations.openai import wrap_openai as _wrap
+    return _wrap(openai_client, opts)
+
+
+__version__ = "0.5.3"
+
+__all__ = [
+    # Core classes
+    "SensuClient",
+    "RunHandle",
+    "StepHandle",
+    # Integration helpers
+    "wrap_anthropic",
+    "wrap_openai",
+    "WrapAnthropicOptions",
+    "WrapOpenAIOptions",
+    # TypedDicts — option types
+    "SensuClientOptions",
+    "StartRunOptions",
+    "RunOptions",
+    "StartStepOptions",
+    "TrackLlmOptions",
+    "TrackStreamingLlmOptions",
+    "TrackToolOptions",
+    "TrackRetrievalOptions",
+    "TrackEmbeddingOptions",
+    "TrackGuardrailOptions",
+    "RawLlmCallOptions",
+    "RawRetrievalOptions",
+    "RawEmbeddingOptions",
+    "RawGuardrailOptions",
+    "RecordFeedbackOptions",
+    "RecordEvalScoreOptions",
+    "SpawnRunOptions",
+    "HandoffOptions",
+    "RecordPromptRenderOptions",
+    "DeployPromptVersionOptions",
+    "StartSessionOptions",
+    "ResumeSessionOptions",
+    # Supporting types
+    "ContextBreakdown",
+    "MessageSnapshotItem",
+    "RetrievalChunkInput",
+    "GuardrailResult",
+    "GuardrailType",
+]