withorbit-sdk 0.1.3__tar.gz

withorbit_sdk-0.1.3/PKG-INFO

@@ -0,0 +1,233 @@
+Metadata-Version: 2.4
+Name: withorbit-sdk
+Version: 0.1.3
+Summary: Orbit - AI Cost Analytics SDK. Track, monitor, and optimize your AI spend.
+Author-email: WithOrbit <support@withorbit.io>
+License: MIT
+Project-URL: Homepage, https://withorbit.io
+Project-URL: Documentation, https://withorbit.io/docs
+Project-URL: Repository, https://github.com/withorbit/sdk-python
+Keywords: ai,llm,analytics,monitoring,openai,anthropic,cost-tracking,observability,orbit,withorbit
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.24.0
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.18.0; extra == "anthropic"
+Provides-Extra: google
+Requires-Dist: google-genai>=1.0.0; extra == "google"
+Provides-Extra: all
+Requires-Dist: openai>=1.0.0; extra == "all"
+Requires-Dist: anthropic>=0.18.0; extra == "all"
+Requires-Dist: google-genai>=1.0.0; extra == "all"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+
+# Orbit SDK for Python
+
+Track, monitor, and optimize your AI spend across OpenAI, Anthropic, and other LLM providers.
+
+## Installation
+
+```bash
+pip install withorbit-sdk
+
+# With OpenAI support
+pip install withorbit-sdk[openai]
+
+# With Anthropic support
+pip install withorbit-sdk[anthropic]
+
+# With all providers
+pip install withorbit-sdk[all]
+```
+
+## Quick Start
+
+### 1. Get your API key
+
+Sign up at [Orbit](https://app.withorbit.io) and create an API key.
+
+### 2. Initialize the SDK
+
+```python
+from withorbit_sdk import Orbit
+
+orbit = Orbit(
+    api_key="orb_live_xxxxxxxxxxxxxxxxxxxxxxxx",
+    default_feature="my-app",  # Optional: default feature for all events
+)
+```
+
+### 3. Track your LLM calls
+
+#### Option A: Automatic tracking (Recommended)
+
+Wrap your OpenAI or Anthropic client for automatic tracking:
+
+```python
+from openai import OpenAI
+from withorbit_sdk import Orbit, WrapperOptions
+
+orbit = Orbit(api_key="orb_live_xxx")
+openai = orbit.wrap_openai(OpenAI(), WrapperOptions(feature="chat-assistant"))
+
+# All API calls are now automatically tracked!
+response = openai.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Hello, world!"}],
+)
+```
+
+Works with Anthropic too:
+
+```python
+from anthropic import Anthropic
+from withorbit_sdk import Orbit, WrapperOptions
+
+orbit = Orbit(api_key="orb_live_xxx")
+anthropic = orbit.wrap_anthropic(Anthropic(), WrapperOptions(feature="document-analysis"))
+
+message = anthropic.messages.create(
+    model="claude-3-opus-20240229",
+    max_tokens=1024,
+    messages=[{"role": "user", "content": "Analyze this document..."}],
+)
+```
+
+#### Option B: Manual tracking
+
+For other providers or custom implementations:
+
+```python
+from withorbit_sdk import Orbit
+
+orbit = Orbit(api_key="orb_live_xxx")
+
+# Track a successful request
+orbit.track(
+    model="gpt-4o",
+    input_tokens=150,
+    output_tokens=50,
+    latency_ms=1234,
+    feature="summarization",
+    environment="production",
+)
+
+# Track an error
+orbit.track_error(
+    model="gpt-4o",
+    error_type="rate_limit_exceeded",
+    error_message="Rate limit exceeded",
+    feature="chat-assistant",
+    input_tokens=150,
+)
+```
+
+## Configuration
+
+```python
+from withorbit_sdk import Orbit, OrbitConfig
+
+orbit = Orbit(config=OrbitConfig(
+    # Required
+    api_key="orb_live_xxx",
+
+    # Optional
+    base_url="https://app.withorbit.io/api/v1",  # Custom API endpoint
+    default_feature="my-app",                    # Default feature name
+    default_environment="production",            # 'production' | 'staging' | 'development'
+    debug=False,                                 # Enable debug logging
+
+    # Batching (for high-volume applications)
+    batch_events=True,       # Batch events before sending
+    batch_size=10,           # Max events per batch
+    batch_interval=5.0,      # Max seconds before sending batch
+
+    # Reliability
+    retry=True,              # Retry failed requests
+    max_retries=3,           # Max retry attempts
+))
+```
+
+## Feature Attribution
+
+Features are Orbit's key differentiator - they let you see exactly which parts of your application are consuming AI resources:
+
+```python
+# Track different features
+orbit.track(
+    model="gpt-4o",
+    input_tokens=100,
+    output_tokens=50,
+    feature="chat-assistant",  # Attribute to chat feature
+)
+
+orbit.track(
+    model="gpt-4o",
+    input_tokens=500,
+    output_tokens=200,
+    feature="document-analysis",  # Attribute to doc analysis
+)
+```
+
+Then in the Orbit dashboard, you'll see:
+- Cost breakdown by feature
+- Request volume by feature
+- Error rates by feature
+- And more!
+
+## Context Manager Support
+
+```python
+from withorbit_sdk import Orbit
+
+with Orbit(api_key="orb_live_xxx") as orbit:
+    orbit.track(model="gpt-4o", input_tokens=100, output_tokens=50)
+# Automatically flushes on exit
+```
+
+## Graceful Shutdown
+
+For long-running processes, flush events before exit:
+
+```python
+# Before your process exits
+orbit.shutdown()
+```
+
+## Event Properties
+
+| Property | Type | Required | Description |
+|----------|------|----------|-------------|
+| `model` | str | Yes | Model name (e.g., 'gpt-4o', 'claude-3-opus') |
+| `input_tokens` | int | Yes | Number of input tokens |
+| `output_tokens` | int | Yes | Number of output tokens |
+| `provider` | str | No | Provider name (auto-detected if not provided) |
+| `latency_ms` | int | No | Request latency in milliseconds |
+| `feature` | str | No | Feature name for attribution |
+| `environment` | str | No | Environment ('production', 'staging', 'development') |
+| `status` | str | No | Request status ('success', 'error', 'timeout') |
+| `error_type` | str | No | Error type if status is 'error' |
+| `error_message` | str | No | Error message if status is 'error' |
+| `user_id` | str | No | Your application's user ID |
+| `session_id` | str | No | Session ID for grouping requests |
+| `request_id` | str | No | Unique request ID for tracing |
+| `metadata` | dict | No | Additional key-value metadata |
+
+## License
+
+MIT

withorbit_sdk-0.1.3/README.md

@@ -0,0 +1,195 @@
+# Orbit SDK for Python
+
+Track, monitor, and optimize your AI spend across OpenAI, Anthropic, and other LLM providers.
+
+## Installation
+
+```bash
+pip install withorbit-sdk
+
+# With OpenAI support
+pip install withorbit-sdk[openai]
+
+# With Anthropic support
+pip install withorbit-sdk[anthropic]
+
+# With all providers
+pip install withorbit-sdk[all]
+```
+
+## Quick Start
+
+### 1. Get your API key
+
+Sign up at [Orbit](https://app.withorbit.io) and create an API key.
+
+### 2. Initialize the SDK
+
+```python
+from withorbit_sdk import Orbit
+
+orbit = Orbit(
+    api_key="orb_live_xxxxxxxxxxxxxxxxxxxxxxxx",
+    default_feature="my-app",  # Optional: default feature for all events
+)
+```
+
+### 3. Track your LLM calls
+
+#### Option A: Automatic tracking (Recommended)
+
+Wrap your OpenAI or Anthropic client for automatic tracking:
+
+```python
+from openai import OpenAI
+from withorbit_sdk import Orbit, WrapperOptions
+
+orbit = Orbit(api_key="orb_live_xxx")
+openai = orbit.wrap_openai(OpenAI(), WrapperOptions(feature="chat-assistant"))
+
+# All API calls are now automatically tracked!
+response = openai.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Hello, world!"}],
+)
+```
+
+Works with Anthropic too:
+
+```python
+from anthropic import Anthropic
+from withorbit_sdk import Orbit, WrapperOptions
+
+orbit = Orbit(api_key="orb_live_xxx")
+anthropic = orbit.wrap_anthropic(Anthropic(), WrapperOptions(feature="document-analysis"))
+
+message = anthropic.messages.create(
+    model="claude-3-opus-20240229",
+    max_tokens=1024,
+    messages=[{"role": "user", "content": "Analyze this document..."}],
+)
+```
+
+#### Option B: Manual tracking
+
+For other providers or custom implementations:
+
+```python
+from withorbit_sdk import Orbit
+
+orbit = Orbit(api_key="orb_live_xxx")
+
+# Track a successful request
+orbit.track(
+    model="gpt-4o",
+    input_tokens=150,
+    output_tokens=50,
+    latency_ms=1234,
+    feature="summarization",
+    environment="production",
+)
+
+# Track an error
+orbit.track_error(
+    model="gpt-4o",
+    error_type="rate_limit_exceeded",
+    error_message="Rate limit exceeded",
+    feature="chat-assistant",
+    input_tokens=150,
+)
+```
+
+## Configuration
+
+```python
+from withorbit_sdk import Orbit, OrbitConfig
+
+orbit = Orbit(config=OrbitConfig(
+    # Required
+    api_key="orb_live_xxx",
+
+    # Optional
+    base_url="https://app.withorbit.io/api/v1",  # Custom API endpoint
+    default_feature="my-app",                    # Default feature name
+    default_environment="production",            # 'production' | 'staging' | 'development'
+    debug=False,                                 # Enable debug logging
+
+    # Batching (for high-volume applications)
+    batch_events=True,       # Batch events before sending
+    batch_size=10,           # Max events per batch
+    batch_interval=5.0,      # Max seconds before sending batch
+
+    # Reliability
+    retry=True,              # Retry failed requests
+    max_retries=3,           # Max retry attempts
+))
+```
+
+## Feature Attribution
+
+Features are Orbit's key differentiator - they let you see exactly which parts of your application are consuming AI resources:
+
+```python
+# Track different features
+orbit.track(
+    model="gpt-4o",
+    input_tokens=100,
+    output_tokens=50,
+    feature="chat-assistant",  # Attribute to chat feature
+)
+
+orbit.track(
+    model="gpt-4o",
+    input_tokens=500,
+    output_tokens=200,
+    feature="document-analysis",  # Attribute to doc analysis
+)
+```
+
+Then in the Orbit dashboard, you'll see:
+- Cost breakdown by feature
+- Request volume by feature
+- Error rates by feature
+- And more!
+
+## Context Manager Support
+
+```python
+from withorbit_sdk import Orbit
+
+with Orbit(api_key="orb_live_xxx") as orbit:
+    orbit.track(model="gpt-4o", input_tokens=100, output_tokens=50)
+# Automatically flushes on exit
+```
+
+## Graceful Shutdown
+
+For long-running processes, flush events before exit:
+
+```python
+# Before your process exits
+orbit.shutdown()
+```
+
+## Event Properties
+
+| Property | Type | Required | Description |
+|----------|------|----------|-------------|
+| `model` | str | Yes | Model name (e.g., 'gpt-4o', 'claude-3-opus') |
+| `input_tokens` | int | Yes | Number of input tokens |
+| `output_tokens` | int | Yes | Number of output tokens |
+| `provider` | str | No | Provider name (auto-detected if not provided) |
+| `latency_ms` | int | No | Request latency in milliseconds |
+| `feature` | str | No | Feature name for attribution |
+| `environment` | str | No | Environment ('production', 'staging', 'development') |
+| `status` | str | No | Request status ('success', 'error', 'timeout') |
+| `error_type` | str | No | Error type if status is 'error' |
+| `error_message` | str | No | Error message if status is 'error' |
+| `user_id` | str | No | Your application's user ID |
+| `session_id` | str | No | Session ID for grouping requests |
+| `request_id` | str | No | Unique request ID for tracing |
+| `metadata` | dict | No | Additional key-value metadata |
+
+## License
+
+MIT

withorbit_sdk-0.1.3/pyproject.toml

@@ -0,0 +1,49 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "withorbit-sdk"
+version = "0.1.3"
+description = "Orbit - AI Cost Analytics SDK. Track, monitor, and optimize your AI spend."
+readme = "README.md"
+license = {text = "MIT"}
+authors = [
+    {name = "WithOrbit", email = "support@withorbit.io"}
+]
+keywords = ["ai", "llm", "analytics", "monitoring", "openai", "anthropic", "cost-tracking", "observability", "orbit", "withorbit"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+requires-python = ">=3.8"
+dependencies = [
+    "httpx>=0.24.0",
+]
+
+[project.optional-dependencies]
+openai = ["openai>=1.0.0"]
+anthropic = ["anthropic>=0.18.0"]
+google = ["google-genai>=1.0.0"]
+all = ["openai>=1.0.0", "anthropic>=0.18.0", "google-genai>=1.0.0"]
+dev = ["pytest>=7.0.0", "pytest-asyncio>=0.21.0", "mypy>=1.0.0"]
+
+[project.urls]
+Homepage = "https://withorbit.io"
+Documentation = "https://withorbit.io/docs"
+Repository = "https://github.com/withorbit/sdk-python"
+
+[tool.setuptools.packages.find]
+where = ["."]
+
+[tool.mypy]
+python_version = "3.8"
+strict = true

withorbit_sdk-0.1.3/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

withorbit_sdk-0.1.3/withorbit_sdk/__init__.py

@@ -0,0 +1,40 @@
+"""
+Orbit SDK
+Track, monitor, and optimize your AI spend
+
+Usage:
+    from withorbit_sdk import Orbit
+
+    orbit = Orbit(api_key="orb_live_xxx")
+
+    # Manual tracking
+    orbit.track(
+        model="gpt-4o",
+        input_tokens=100,
+        output_tokens=50,
+        feature="chat-assistant",
+    )
+
+    # Or wrap your OpenAI/Anthropic client for automatic tracking
+    from openai import OpenAI
+
+    openai = orbit.wrap_openai(OpenAI())
+    response = openai.chat.completions.create(
+        model="gpt-4o",
+        messages=[{"role": "user", "content": "Hello!"}],
+    )
+    # Usage automatically tracked!
+"""
+
+from .client import OrbitClient, Orbit
+from .types import OrbitConfig, OrbitEvent, OrbitResponse, WrapperOptions
+
+__version__ = "0.1.3"
+__all__ = [
+    "Orbit",
+    "OrbitClient",
+    "OrbitConfig",
+    "OrbitEvent",
+    "OrbitResponse",
+    "WrapperOptions",
+]