axonpush 0.1.0__tar.gz

axonpush-0.1.0/.gitignore

@@ -0,0 +1,13 @@
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+.venv
+.env
+data/*.db
+data/resumes/
+rendercv_output/
+migrations/versions/
+.claude/settings.local.json

axonpush-0.1.0/.python-version

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

axonpush-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 AxonPush
+
+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.

axonpush-0.1.0/PKG-INFO

@@ -0,0 +1,204 @@
+Metadata-Version: 2.4
+Name: axonpush
+Version: 0.1.0
+Summary: Python SDK for AxonPush — real-time event infrastructure for AI agent systems
+Author: AxonPush
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Development Status :: 4 - Beta
+Classifier: Framework :: Pydantic :: 2
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Requires-Dist: httpx-sse<1.0,>=0.4.0
+Requires-Dist: httpx<1.0,>=0.25.0
+Requires-Dist: pydantic<3.0,>=2.0
+Provides-Extra: all
+Requires-Dist: anthropic>=0.30.0; extra == 'all'
+Requires-Dist: crewai>=0.50.0; extra == 'all'
+Requires-Dist: langchain-core>=0.1.0; extra == 'all'
+Requires-Dist: openai-agents>=0.1.0; extra == 'all'
+Requires-Dist: python-socketio[asyncio-client]<6.0,>=5.10; extra == 'all'
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.30.0; extra == 'anthropic'
+Provides-Extra: crewai
+Requires-Dist: crewai>=0.50.0; extra == 'crewai'
+Provides-Extra: dev
+Requires-Dist: mypy>=1.10; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: respx>=0.21; extra == 'dev'
+Requires-Dist: ruff>=0.4; extra == 'dev'
+Provides-Extra: langchain
+Requires-Dist: langchain-core>=0.1.0; extra == 'langchain'
+Provides-Extra: openai-agents
+Requires-Dist: openai-agents>=0.1.0; extra == 'openai-agents'
+Provides-Extra: websocket
+Requires-Dist: python-socketio[asyncio-client]<6.0,>=5.10; extra == 'websocket'
+Description-Content-Type: text/markdown
+
+# axonpush
+
+Python SDK for [AxonPush](https://axonpush.com) — real-time event infrastructure for AI agent systems.
+
+Publish, subscribe, trace, and deliver agent events with sub-100ms latency. Drop-in integrations for LangChain, OpenAI Agents SDK, Claude/Anthropic, and CrewAI.
+
+## Install
+
+```bash
+pip install axonpush
+```
+
+With framework integrations:
+
+```bash
+pip install axonpush[langchain]       # LangChain/LangGraph
+pip install axonpush[openai-agents]   # OpenAI Agents SDK
+pip install axonpush[anthropic]       # Claude/Anthropic
+pip install axonpush[crewai]          # CrewAI
+pip install axonpush[all]             # Everything
+```
+
+## Quick Start
+
+```python
+from axonpush import AxonPush, EventType
+
+with AxonPush(api_key="ak_...", tenant_id="1") as client:
+    # Publish an event
+    event = client.events.publish(
+        "web_search",
+        {"query": "AI agent frameworks"},
+        channel_id=1,
+        agent_id="researcher",
+        trace_id="tr_run_42",
+        event_type=EventType.AGENT_TOOL_CALL_START,
+    )
+
+    # List events
+    events = client.events.list(channel_id=1)
+
+    # Get a trace summary
+    summary = client.traces.get_summary("tr_run_42")
+```
+
+### Async
+
+```python
+from axonpush import AsyncAxonPush
+
+async with AsyncAxonPush(api_key="ak_...", tenant_id="1") as client:
+    event = await client.events.publish(
+        "web_search",
+        {"query": "AI agents"},
+        channel_id=1,
+        agent_id="researcher",
+        event_type="agent.tool_call.start",
+    )
+```
+
+## Framework Integrations
+
+### LangChain / LangGraph
+
+```python
+from axonpush import AxonPush
+from axonpush.integrations.langchain import AxonPushCallbackHandler
+
+client = AxonPush(api_key="ak_...", tenant_id="1")
+handler = AxonPushCallbackHandler(client, channel_id=1, agent_id="my-agent")
+
+# All chain/tool/LLM events are published automatically
+chain.invoke({"input": "..."}, config={"callbacks": [handler]})
+```
+
+### OpenAI Agents SDK
+
+```python
+from axonpush import AsyncAxonPush
+from axonpush.integrations.openai_agents import AxonPushRunHooks
+
+client = AsyncAxonPush(api_key="ak_...", tenant_id="1")
+hooks = AxonPushRunHooks(client, channel_id=1)
+
+result = await Runner.run(agent, input="...", hooks=hooks)
+```
+
+### Claude / Anthropic
+
+```python
+from axonpush import AxonPush
+from axonpush.integrations.anthropic import AxonPushAnthropicTracer
+
+client = AxonPush(api_key="ak_...", tenant_id="1")
+tracer = AxonPushAnthropicTracer(client, channel_id=1)
+
+# Wraps messages.create() — auto-emits events for tool_use, text, turns
+response = tracer.create_message(
+    anthropic_client,
+    model="claude-sonnet-4-20250514",
+    messages=[{"role": "user", "content": "Hello"}],
+)
+```
+
+### CrewAI
+
+```python
+from axonpush import AxonPush
+from axonpush.integrations.crewai import AxonPushCrewCallbacks
+
+client = AxonPush(api_key="ak_...", tenant_id="1")
+callbacks = AxonPushCrewCallbacks(client, channel_id=1)
+
+callbacks.on_crew_start()
+result = Crew(
+    agents=[...],
+    tasks=[...],
+    step_callback=callbacks.on_step,
+    task_callback=callbacks.on_task_complete,
+).kickoff()
+callbacks.on_crew_end(result)
+```
+
+## Real-Time Subscriptions
+
+### SSE
+
+```python
+from axonpush.realtime.sse import SSESubscription
+
+with SSESubscription(
+    client._transport, channel_id=1, event_type="agent.error"
+) as sub:
+    for event in sub:
+        print(f"[{event.agent_id}] {event.identifier}: {event.payload}")
+```
+
+### WebSocket
+
+```python
+ws = client.connect_websocket()
+ws.on_event(lambda e: print(e.agent_id, e.payload))
+ws.subscribe(channel_id=1, event_type="agent.tool_call.start")
+ws.wait()
+```
+
+## Resources
+
+The client exposes Stripe-style resource objects:
+
+| Resource | Methods |
+|---|---|
+| `client.events` | `publish()`, `list()` |
+| `client.channels` | `create()`, `get()`, `update()`, `delete()`, `subscribe_sse()` |
+| `client.apps` | `create()`, `get()`, `list()`, `update()`, `delete()` |
+| `client.webhooks` | `create_endpoint()`, `list_endpoints()`, `delete_endpoint()`, `get_deliveries()` |
+| `client.traces` | `list()`, `get_events()`, `get_summary()` |
+
+## License
+
+MIT

axonpush-0.1.0/README.md

@@ -0,0 +1,161 @@
+# axonpush
+
+Python SDK for [AxonPush](https://axonpush.com) — real-time event infrastructure for AI agent systems.
+
+Publish, subscribe, trace, and deliver agent events with sub-100ms latency. Drop-in integrations for LangChain, OpenAI Agents SDK, Claude/Anthropic, and CrewAI.
+
+## Install
+
+```bash
+pip install axonpush
+```
+
+With framework integrations:
+
+```bash
+pip install axonpush[langchain]       # LangChain/LangGraph
+pip install axonpush[openai-agents]   # OpenAI Agents SDK
+pip install axonpush[anthropic]       # Claude/Anthropic
+pip install axonpush[crewai]          # CrewAI
+pip install axonpush[all]             # Everything
+```
+
+## Quick Start
+
+```python
+from axonpush import AxonPush, EventType
+
+with AxonPush(api_key="ak_...", tenant_id="1") as client:
+    # Publish an event
+    event = client.events.publish(
+        "web_search",
+        {"query": "AI agent frameworks"},
+        channel_id=1,
+        agent_id="researcher",
+        trace_id="tr_run_42",
+        event_type=EventType.AGENT_TOOL_CALL_START,
+    )
+
+    # List events
+    events = client.events.list(channel_id=1)
+
+    # Get a trace summary
+    summary = client.traces.get_summary("tr_run_42")
+```
+
+### Async
+
+```python
+from axonpush import AsyncAxonPush
+
+async with AsyncAxonPush(api_key="ak_...", tenant_id="1") as client:
+    event = await client.events.publish(
+        "web_search",
+        {"query": "AI agents"},
+        channel_id=1,
+        agent_id="researcher",
+        event_type="agent.tool_call.start",
+    )
+```
+
+## Framework Integrations
+
+### LangChain / LangGraph
+
+```python
+from axonpush import AxonPush
+from axonpush.integrations.langchain import AxonPushCallbackHandler
+
+client = AxonPush(api_key="ak_...", tenant_id="1")
+handler = AxonPushCallbackHandler(client, channel_id=1, agent_id="my-agent")
+
+# All chain/tool/LLM events are published automatically
+chain.invoke({"input": "..."}, config={"callbacks": [handler]})
+```
+
+### OpenAI Agents SDK
+
+```python
+from axonpush import AsyncAxonPush
+from axonpush.integrations.openai_agents import AxonPushRunHooks
+
+client = AsyncAxonPush(api_key="ak_...", tenant_id="1")
+hooks = AxonPushRunHooks(client, channel_id=1)
+
+result = await Runner.run(agent, input="...", hooks=hooks)
+```
+
+### Claude / Anthropic
+
+```python
+from axonpush import AxonPush
+from axonpush.integrations.anthropic import AxonPushAnthropicTracer
+
+client = AxonPush(api_key="ak_...", tenant_id="1")
+tracer = AxonPushAnthropicTracer(client, channel_id=1)
+
+# Wraps messages.create() — auto-emits events for tool_use, text, turns
+response = tracer.create_message(
+    anthropic_client,
+    model="claude-sonnet-4-20250514",
+    messages=[{"role": "user", "content": "Hello"}],
+)
+```
+
+### CrewAI
+
+```python
+from axonpush import AxonPush
+from axonpush.integrations.crewai import AxonPushCrewCallbacks
+
+client = AxonPush(api_key="ak_...", tenant_id="1")
+callbacks = AxonPushCrewCallbacks(client, channel_id=1)
+
+callbacks.on_crew_start()
+result = Crew(
+    agents=[...],
+    tasks=[...],
+    step_callback=callbacks.on_step,
+    task_callback=callbacks.on_task_complete,
+).kickoff()
+callbacks.on_crew_end(result)
+```
+
+## Real-Time Subscriptions
+
+### SSE
+
+```python
+from axonpush.realtime.sse import SSESubscription
+
+with SSESubscription(
+    client._transport, channel_id=1, event_type="agent.error"
+) as sub:
+    for event in sub:
+        print(f"[{event.agent_id}] {event.identifier}: {event.payload}")
+```
+
+### WebSocket
+
+```python
+ws = client.connect_websocket()
+ws.on_event(lambda e: print(e.agent_id, e.payload))
+ws.subscribe(channel_id=1, event_type="agent.tool_call.start")
+ws.wait()
+```
+
+## Resources
+
+The client exposes Stripe-style resource objects:
+
+| Resource | Methods |
+|---|---|
+| `client.events` | `publish()`, `list()` |
+| `client.channels` | `create()`, `get()`, `update()`, `delete()`, `subscribe_sse()` |
+| `client.apps` | `create()`, `get()`, `list()`, `update()`, `delete()` |
+| `client.webhooks` | `create_endpoint()`, `list_endpoints()`, `delete_endpoint()`, `get_deliveries()` |
+| `client.traces` | `list()`, `get_events()`, `get_summary()` |
+
+## License
+
+MIT

axonpush-0.1.0/pyproject.toml

@@ -0,0 +1,64 @@
+[project]
+name = "axonpush"
+version = "0.1.0"
+description = "Python SDK for AxonPush — real-time event infrastructure for AI agent systems"
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [{ name = "AxonPush" }]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Typing :: Typed",
+    "Framework :: Pydantic :: 2",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+]
+
+dependencies = [
+    "httpx>=0.25.0,<1.0",
+    "pydantic>=2.0,<3.0",
+    "httpx-sse>=0.4.0,<1.0",
+]
+
+[project.optional-dependencies]
+websocket = ["python-socketio[asyncio_client]>=5.10,<6.0"]
+langchain = ["langchain-core>=0.1.0"]
+openai-agents = ["openai-agents>=0.1.0"]
+anthropic = ["anthropic>=0.30.0"]
+crewai = ["crewai>=0.50.0"]
+all = [
+    "axonpush[websocket]",
+    "axonpush[langchain]",
+    "axonpush[openai-agents]",
+    "axonpush[anthropic]",
+    "axonpush[crewai]",
+]
+dev = [
+    "pytest>=8.0",
+    "pytest-asyncio>=0.23",
+    "respx>=0.21",
+    "ruff>=0.4",
+    "mypy>=1.10",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/axonpush"]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 100
+
+[tool.mypy]
+python_version = "3.10"
+strict = true
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]

axonpush-0.1.0/src/axonpush/__init__.py

@@ -0,0 +1,48 @@
+"""AxonPush — Python SDK for real-time event infrastructure for AI agent systems."""
+
+from axonpush._tracing import TraceContext, get_or_create_trace
+from axonpush._version import __version__
+from axonpush.client import AsyncAxonPush, AxonPush
+from axonpush.exceptions import (
+    AuthenticationError,
+    AxonPushError,
+    ForbiddenError,
+    NotFoundError,
+    RateLimitError,
+    ServerError,
+    ValidationError,
+)
+from axonpush.models.apps import App
+from axonpush.models.channels import Channel
+from axonpush.models.events import Event, EventType
+from axonpush.models.traces import TraceListItem, TraceSummary
+from axonpush.models.webhooks import DeliveryStatus, WebhookDelivery, WebhookEndpoint
+
+__all__ = [
+    # Clients
+    "AxonPush",
+    "AsyncAxonPush",
+    # Models
+    "App",
+    "Channel",
+    "DeliveryStatus",
+    "Event",
+    "EventType",
+    "TraceListItem",
+    "TraceSummary",
+    "WebhookDelivery",
+    "WebhookEndpoint",
+    # Tracing
+    "TraceContext",
+    "get_or_create_trace",
+    # Exceptions
+    "AuthenticationError",
+    "AxonPushError",
+    "ForbiddenError",
+    "NotFoundError",
+    "RateLimitError",
+    "ServerError",
+    "ValidationError",
+    # Meta
+    "__version__",
+]

axonpush-0.1.0/src/axonpush/_auth.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+
+class AuthConfig:
+    """Immutable auth configuration. Thread-safe (read-only after construction)."""
+
+    __slots__ = ("api_key", "tenant_id", "base_url")
+
+    def __init__(
+        self,
+        api_key: str,
+        tenant_id: str,
+        base_url: str = "https://api.axonpush.com",
+    ) -> None:
+        self.api_key = api_key
+        self.tenant_id = tenant_id
+        self.base_url = base_url.rstrip("/")
+
+    def headers(self) -> dict[str, str]:
+        return {
+            "X-API-Key": self.api_key,
+            "x-tenant-id": self.tenant_id,
+            "Content-Type": "application/json",
+        }

axonpush-0.1.0/src/axonpush/_http.py

@@ -0,0 +1,121 @@
+from __future__ import annotations
+
+from contextlib import contextmanager
+from typing import Any, Dict, Generator, Iterator, Optional
+
+import httpx
+from httpx_sse import EventSource, connect_sse, aconnect_sse
+
+from axonpush._auth import AuthConfig
+from axonpush.exceptions import (
+    AuthenticationError,
+    AxonPushError,
+    ForbiddenError,
+    NotFoundError,
+    RateLimitError,
+    ServerError,
+    ValidationError,
+)
+
+_ERROR_MAP: Dict[int, type] = {
+    400: ValidationError,
+    401: AuthenticationError,
+    403: ForbiddenError,
+    404: NotFoundError,
+    429: RateLimitError,
+}
+
+
+def _raise_for_status(response: httpx.Response) -> None:
+    if response.is_success:
+        return
+
+    status = response.status_code
+    try:
+        body = response.json()
+        message = body.get("message", response.text)
+        if isinstance(message, list):
+            message = "; ".join(str(m) for m in message)
+    except Exception:
+        message = response.text or f"HTTP {status}"
+
+    if status == 429:
+        retry_after_raw = response.headers.get("Retry-After")
+        retry_after = float(retry_after_raw) if retry_after_raw else None
+        raise RateLimitError(str(message), retry_after=retry_after)
+
+    exc_cls = _ERROR_MAP.get(status)
+    if exc_cls is not None:
+        raise exc_cls(str(message), status_code=status)
+
+    if status >= 500:
+        raise ServerError(str(message), status_code=status)
+
+    raise AxonPushError(str(message), status_code=status)
+
+
+class SyncTransport:
+    """Synchronous HTTP transport backed by httpx.Client."""
+
+    def __init__(self, auth: AuthConfig, timeout: float = 30.0) -> None:
+        self._auth = auth
+        self._client = httpx.Client(
+            base_url=auth.base_url,
+            headers=auth.headers(),
+            timeout=timeout,
+        )
+
+    def request(
+        self,
+        method: str,
+        path: str,
+        *,
+        json: Any = None,
+        params: Optional[Dict[str, Any]] = None,
+    ) -> Any:
+        response = self._client.request(method, path, json=json, params=params)
+        _raise_for_status(response)
+        if not response.content:
+            return None
+        return response.json()
+
+    @contextmanager
+    def stream_sse(
+        self, path: str, params: Optional[Dict[str, Any]] = None
+    ) -> Generator[EventSource, None, None]:
+        with connect_sse(
+            self._client, "GET", path, params=params or {}
+        ) as event_source:
+            yield event_source
+
+    def close(self) -> None:
+        self._client.close()
+
+
+class AsyncTransport:
+    """Asynchronous HTTP transport backed by httpx.AsyncClient."""
+
+    def __init__(self, auth: AuthConfig, timeout: float = 30.0) -> None:
+        self._auth = auth
+        self._client = httpx.AsyncClient(
+            base_url=auth.base_url,
+            headers=auth.headers(),
+            timeout=timeout,
+        )
+
+    async def request(
+        self,
+        method: str,
+        path: str,
+        *,
+        json: Any = None,
+        params: Optional[Dict[str, Any]] = None,
+    ) -> Any:
+        response = await self._client.request(method, path, json=json, params=params)
+        _raise_for_status(response)
+        if not response.content:
+            return None
+        return response.json()
+
+    async def close(self) -> None:
+        await self._client.aclose()