agentos-python 0.1.0__tar.gz

agentos_python-0.1.0/.github/workflows/ci.yml

@@ -0,0 +1,40 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        run: uv python install ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: uv sync && uv pip install -e ".[dev]"
+
+      - name: Lint
+        run: uv run ruff check .
+
+      - name: Format check
+        run: uv run ruff format --check .
+
+      - name: Run tests
+        run: uv run pytest -v --ignore=tests/test_client.py --ignore=tests/test_decorators.py
+
+      - name: Run client tests
+        run: uv run pytest tests/test_client.py tests/test_decorators.py -v
+        timeout-minutes: 5

agentos_python-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,55 @@
+name: Publish to PyPI
+
+on:
+  push:
+    tags:
+      - "v*"
+
+permissions:
+  contents: read
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.12"]
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        run: uv python install ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: uv sync && uv pip install -e ".[dev]"
+
+      - name: Lint
+        run: uv run ruff check .
+
+      - name: Run tests
+        run: uv run pytest -v
+
+  publish:
+    needs: test
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      contents: read
+      id-token: write
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+
+      - name: Set up Python
+        run: uv python install 3.12
+
+      - name: Build package
+        run: uv build
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

agentos_python-0.1.0/CHANGELOG.md

@@ -0,0 +1,18 @@
+# Changelog
+
+## 0.1.0 (2026-04-05)
+
+Initial release.
+
+### Features
+
+- **Core client** — `AgentOS` (sync) and `AsyncAgentOS` (async) with batched delivery, retry, and graceful shutdown
+- **Event builders** — typed helpers for all 8 event types: `llm_call`, `tool_call`, `handoff`, `retrieval_query`, `eval`, `security_alert`, `flag_check`, `business_event`
+- **Tracing** — `trace()` and `span()` context managers with automatic context propagation via `contextvars`
+- **Decorators** — `@observe()` (Langfuse-compatible), `@trace_agent`, `@trace_tool`
+- **`wrap_openai()`** — one-liner auto-instrumentation for OpenAI SDK
+- **`wrap_anthropic()`** — one-liner auto-instrumentation for Anthropic SDK
+- **Framework integrations** — LangChain `CallbackHandler`, CrewAI `@trace_crew`/`@trace_task`, Pydantic AI `wrap_agent`, LlamaIndex `CallbackHandler`, LiteLLM `patch_litellm`
+- **Migration compat layers** — drop-in shims for Langfuse, LangSmith, Arize Phoenix (OTLP), Helicone, Braintrust, W&B Weave
+- **Privacy mode** — `capture_content=False` strips message content
+- **Env-var auto-init** — set `AGENTOS_API_KEY` for zero-code activation

agentos_python-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Agent OS
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

agentos_python-0.1.0/PKG-INFO

@@ -0,0 +1,155 @@
+Metadata-Version: 2.4
+Name: agentos-python
+Version: 0.1.0
+Summary: Python SDK for Agent OS — AI agent observability platform
+Project-URL: Homepage, https://agentos.dev
+Project-URL: Documentation, https://docs.agentos.dev
+Project-URL: Repository, https://github.com/Amitabh1998/agentos-python
+Author-email: Agent OS <hello@agentos.dev>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: agents,ai,llm,observability,tracing
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Requires-Dist: httpx<1,>=0.24
+Provides-Extra: all
+Requires-Dist: anthropic>=0.20; extra == 'all'
+Requires-Dist: langchain-core>=0.2; extra == 'all'
+Requires-Dist: litellm>=1.0; extra == 'all'
+Requires-Dist: openai>=1.0; extra == 'all'
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.20; extra == 'all'
+Requires-Dist: opentelemetry-sdk>=1.20; extra == 'all'
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.20; extra == 'anthropic'
+Provides-Extra: crewai
+Requires-Dist: crewai>=0.50; (python_version >= '3.10') and extra == 'crewai'
+Provides-Extra: dev
+Requires-Dist: mypy>=1.11; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
+Requires-Dist: pytest>=8; extra == 'dev'
+Requires-Dist: respx>=0.21; extra == 'dev'
+Requires-Dist: ruff>=0.6; extra == 'dev'
+Provides-Extra: langchain
+Requires-Dist: langchain-core>=0.2; extra == 'langchain'
+Provides-Extra: litellm
+Requires-Dist: litellm>=1.0; extra == 'litellm'
+Provides-Extra: llamaindex
+Requires-Dist: llama-index-core>=0.10; (python_version >= '3.10') and extra == 'llamaindex'
+Provides-Extra: openai
+Requires-Dist: openai>=1.0; extra == 'openai'
+Provides-Extra: otel
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.20; extra == 'otel'
+Requires-Dist: opentelemetry-sdk>=1.20; extra == 'otel'
+Provides-Extra: pydantic-ai
+Requires-Dist: pydantic-ai>=0.1; (python_version >= '3.10') and extra == 'pydantic-ai'
+Description-Content-Type: text/markdown
+
+# Agent OS Python SDK
+
+Python SDK for [Agent OS](https://agentos.dev) — AI agent observability, experimentation, and security. "PostHog for AI Agents."
+
+## Install
+
+```bash
+pip install agentos
+```
+
+## Quick Start
+
+```python
+from agentos import wrap_openai
+import openai
+
+client = openai.OpenAI()
+client = wrap_openai(client, api_key="aos_...", agent_id="my-agent")
+
+# Every LLM call is now automatically traced
+response = client.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Hello!"}],
+)
+```
+
+That's it. Open your [Agent OS dashboard](https://app.agentos.dev) to see traces, token usage, costs, and latency.
+
+## Other Ways to Instrument
+
+### `@observe` decorator (Langfuse-compatible)
+
+```python
+from agentos import observe
+
+@observe()
+def my_agent(query: str) -> str:
+    result = search(query)       # auto-traced as child span
+    return summarize(result)     # auto-traced as child span
+
+@observe(as_type="generation")
+def summarize(text: str) -> str:
+    return openai_client.chat.completions.create(...)
+```
+
+### Manual event capture
+
+```python
+from agentos import AgentOS
+
+client = AgentOS(api_key="aos_...")
+
+client.llm_call("my-agent", model="gpt-4o", system="openai", input_tokens=100, output_tokens=50)
+client.tool_call("my-agent", tool_name="web-search", status="success", duration_ms=120)
+client.eval("my-agent", eval_name="accuracy", score=0.95)
+client.business_event("my-agent", event_name="task_completed")
+
+client.flush()
+```
+
+### Environment variable (zero-code)
+
+```bash
+export AGENTOS_API_KEY=aos_...
+export AGENTOS_ENABLED=true
+python my_agent.py
+```
+
+## Framework Integrations
+
+| Framework | Integration | One-liner |
+|-----------|------------|-----------|
+| **OpenAI** | `wrap_openai(client)` | `from agentos import wrap_openai` |
+| **Anthropic** | `wrap_anthropic(client)` | `from agentos import wrap_anthropic` |
+| **LangChain** | `AgentOSCallbackHandler` | `from agentos.integrations.langchain import AgentOSCallbackHandler` |
+| **LlamaIndex** | `AgentOSCallbackHandler` | `from agentos.integrations.llamaindex import AgentOSCallbackHandler` |
+| **CrewAI** | `@trace_crew` / `instrument_crewai()` | `from agentos.integrations.crewai import trace_crew` |
+| **Pydantic AI** | `wrap_agent()` / `instrument_pydantic_ai()` | `from agentos.integrations.pydantic_ai import wrap_agent` |
+| **LiteLLM** | `patch_litellm()` | `from agentos.integrations.litellm import patch_litellm` |
+| **OpenTelemetry** | OTLP SpanExporter | `from agentos.compat.otel import register` |
+
+## Migrating From Another Platform?
+
+Change one import and keep your existing code:
+
+| Platform | Migration |
+|----------|-----------|
+| **Langfuse** | `from agentos.compat.langfuse import Langfuse` |
+| **LangSmith** | `from agentos.compat.langsmith import traceable` |
+| **Arize Phoenix** | `from agentos.compat.otel import register` (swap endpoint) |
+| **Helicone** | `from agentos.compat.helicone import get_proxy_config` |
+| **Braintrust** | `from agentos.compat.braintrust import traced, Eval` |
+| **W&B Weave** | `from agentos.compat.weave import op` |
+
+## Documentation
+
+Full docs, guides, and API reference: **[docs.agentos.dev](https://docs.agentos.dev)**
+
+## License
+
+MIT

agentos_python-0.1.0/README.md

@@ -0,0 +1,101 @@
+# Agent OS Python SDK
+
+Python SDK for [Agent OS](https://agentos.dev) — AI agent observability, experimentation, and security. "PostHog for AI Agents."
+
+## Install
+
+```bash
+pip install agentos
+```
+
+## Quick Start
+
+```python
+from agentos import wrap_openai
+import openai
+
+client = openai.OpenAI()
+client = wrap_openai(client, api_key="aos_...", agent_id="my-agent")
+
+# Every LLM call is now automatically traced
+response = client.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Hello!"}],
+)
+```
+
+That's it. Open your [Agent OS dashboard](https://app.agentos.dev) to see traces, token usage, costs, and latency.
+
+## Other Ways to Instrument
+
+### `@observe` decorator (Langfuse-compatible)
+
+```python
+from agentos import observe
+
+@observe()
+def my_agent(query: str) -> str:
+    result = search(query)       # auto-traced as child span
+    return summarize(result)     # auto-traced as child span
+
+@observe(as_type="generation")
+def summarize(text: str) -> str:
+    return openai_client.chat.completions.create(...)
+```
+
+### Manual event capture
+
+```python
+from agentos import AgentOS
+
+client = AgentOS(api_key="aos_...")
+
+client.llm_call("my-agent", model="gpt-4o", system="openai", input_tokens=100, output_tokens=50)
+client.tool_call("my-agent", tool_name="web-search", status="success", duration_ms=120)
+client.eval("my-agent", eval_name="accuracy", score=0.95)
+client.business_event("my-agent", event_name="task_completed")
+
+client.flush()
+```
+
+### Environment variable (zero-code)
+
+```bash
+export AGENTOS_API_KEY=aos_...
+export AGENTOS_ENABLED=true
+python my_agent.py
+```
+
+## Framework Integrations
+
+| Framework | Integration | One-liner |
+|-----------|------------|-----------|
+| **OpenAI** | `wrap_openai(client)` | `from agentos import wrap_openai` |
+| **Anthropic** | `wrap_anthropic(client)` | `from agentos import wrap_anthropic` |
+| **LangChain** | `AgentOSCallbackHandler` | `from agentos.integrations.langchain import AgentOSCallbackHandler` |
+| **LlamaIndex** | `AgentOSCallbackHandler` | `from agentos.integrations.llamaindex import AgentOSCallbackHandler` |
+| **CrewAI** | `@trace_crew` / `instrument_crewai()` | `from agentos.integrations.crewai import trace_crew` |
+| **Pydantic AI** | `wrap_agent()` / `instrument_pydantic_ai()` | `from agentos.integrations.pydantic_ai import wrap_agent` |
+| **LiteLLM** | `patch_litellm()` | `from agentos.integrations.litellm import patch_litellm` |
+| **OpenTelemetry** | OTLP SpanExporter | `from agentos.compat.otel import register` |
+
+## Migrating From Another Platform?
+
+Change one import and keep your existing code:
+
+| Platform | Migration |
+|----------|-----------|
+| **Langfuse** | `from agentos.compat.langfuse import Langfuse` |
+| **LangSmith** | `from agentos.compat.langsmith import traceable` |
+| **Arize Phoenix** | `from agentos.compat.otel import register` (swap endpoint) |
+| **Helicone** | `from agentos.compat.helicone import get_proxy_config` |
+| **Braintrust** | `from agentos.compat.braintrust import traced, Eval` |
+| **W&B Weave** | `from agentos.compat.weave import op` |
+
+## Documentation
+
+Full docs, guides, and API reference: **[docs.agentos.dev](https://docs.agentos.dev)**
+
+## License
+
+MIT

agentos_python-0.1.0/pyproject.toml

@@ -0,0 +1,68 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "agentos-python"
+version = "0.1.0"
+description = "Python SDK for Agent OS — AI agent observability platform"
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [{ name = "Agent OS", email = "hello@agentos.dev" }]
+keywords = ["ai", "agents", "observability", "tracing", "llm"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Software Development :: Libraries",
+    "Typing :: Typed",
+]
+dependencies = ["httpx>=0.24,<1"]
+
+[project.optional-dependencies]
+openai = ["openai>=1.0"]
+anthropic = ["anthropic>=0.20"]
+litellm = ["litellm>=1.0"]
+langchain = ["langchain-core>=0.2"]
+crewai = ["crewai>=0.50; python_version>='3.10'"]
+pydantic-ai = ["pydantic-ai>=0.1; python_version>='3.10'"]
+llamaindex = ["llama-index-core>=0.10; python_version>='3.10'"]
+otel = ["opentelemetry-sdk>=1.20", "opentelemetry-exporter-otlp-proto-http>=1.20"]
+all = ["agentos-python[openai,anthropic,litellm,langchain,otel]"]
+dev = [
+    "pytest>=8",
+    "pytest-asyncio>=0.23",
+    "respx>=0.21",
+    "mypy>=1.11",
+    "ruff>=0.6",
+]
+
+[project.urls]
+Homepage = "https://agentos.dev"
+Documentation = "https://docs.agentos.dev"
+Repository = "https://github.com/Amitabh1998/agentos-python"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/agentos"]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 100
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "UP", "B", "SIM"]
+
+[tool.mypy]
+python_version = "3.10"
+strict = true
+warn_return_any = true
+warn_unused_configs = true
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]

agentos_python-0.1.0/src/agentos/__init__.py

@@ -0,0 +1,58 @@
+"""Agent OS Python SDK — AI agent observability platform.
+
+Usage::
+
+    from agentos import AgentOS
+
+    client = AgentOS(api_key="aos_...")
+    client.llm_call("my-agent", model="gpt-4o", system="openai")
+    client.flush()
+"""
+
+from agentos.client import AgentOS, AsyncAgentOS, get_client
+from agentos.config import AgentOSConfig
+from agentos.decorators import observe, trace_agent, trace_context, trace_tool
+from agentos.integrations.anthropic import wrap_anthropic
+from agentos.integrations.openai import wrap_openai
+from agentos.tracing import TraceContext, span, trace
+from agentos.types import (
+    ErrorInfo,
+    EvalEvaluator,
+    EventPayload,
+    EventType,
+    FinishReason,
+    LLMMessage,
+    SecurityAction,
+    SecurityAlertType,
+    Severity,
+    ToolStatus,
+)
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "__version__",
+    "AgentOS",
+    "AgentOSConfig",
+    "AsyncAgentOS",
+    "ErrorInfo",
+    "EvalEvaluator",
+    "EventPayload",
+    "EventType",
+    "FinishReason",
+    "LLMMessage",
+    "SecurityAction",
+    "SecurityAlertType",
+    "Severity",
+    "ToolStatus",
+    "TraceContext",
+    "get_client",
+    "observe",
+    "span",
+    "trace",
+    "trace_agent",
+    "trace_context",
+    "trace_tool",
+    "wrap_anthropic",
+    "wrap_openai",
+]