agentspend-tracker 0.1.0__tar.gz

agentspend_tracker-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: agentspend-tracker
+Version: 0.1.0
+Summary: Track LLM costs across your AI agents
+Author: Sankar
+Requires-Python: >=3.8
+Requires-Dist: requests>=2.28.0
+Dynamic: author
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

agentspend_tracker-0.1.0/README.md

@@ -0,0 +1,66 @@
+# AgentSpend SDK
+
+Track LLM costs across your AI agents in real time.
+
+## Install
+
+```bash
+pip install agentspend
+```
+
+## Quick Start
+
+```python
+import agentspend
+
+agentspend.init("your-api-key")  # Get your key from the AgentSpend app
+```
+
+## Track Anthropic Claude
+
+```python
+import anthropic
+import agentspend
+
+agentspend.init("as_your_key_here")
+
+client = anthropic.Anthropic(api_key="sk-ant-...")
+response = client.messages.create(
+    model="claude-sonnet-4-6",
+    max_tokens=1024,
+    messages=[{"role": "user", "content": "Hello!"}]
+)
+
+agentspend.track_anthropic(response, agent_name="my-agent")
+```
+
+## Track OpenAI
+
+```python
+import openai
+import agentspend
+
+agentspend.init("as_your_key_here")
+
+client = openai.OpenAI(api_key="sk-...")
+response = client.chat.completions.create(
+    model="gpt-4o",
+    messages=[{"role": "user", "content": "Hello!"}]
+)
+
+agentspend.track_openai(response, agent_name="my-agent")
+```
+
+## Manual Tracking
+
+```python
+agentspend.track(
+    agent_name="my-agent",
+    model="claude-sonnet-4-6",
+    input_tokens=1000,
+    output_tokens=500,
+    project="my-project"  # optional
+)
+```
+
+Get your API key at agentspend.app

agentspend_tracker-0.1.0/agentspend/__init__.py

@@ -0,0 +1,99 @@
+import requests
+import threading
+
+_api_key = None
+_base_url = "https://agentspend-backend-production.up.railway.app"
+
+
+def init(api_key: str):
+    """Initialize AgentSpend with your API key."""
+    global _api_key
+    _api_key = api_key
+
+
+def track(
+    agent_name: str,
+    model: str,
+    input_tokens: int,
+    output_tokens: int,
+    cost_usd: float = None,
+    project: str = None,
+):
+    """Track an LLM call. Call this after every Claude/OpenAI/etc. request."""
+    if _api_key is None:
+        raise ValueError("Call agentspend.init('your-api-key') first.")
+
+    if cost_usd is None:
+        cost_usd = _estimate_cost(model, input_tokens, output_tokens)
+
+    payload = {
+        "agent_name": agent_name,
+        "model": model,
+        "input_tokens": input_tokens,
+        "output_tokens": output_tokens,
+        "cost_usd": round(cost_usd, 6),
+        "project": project,
+    }
+
+    # Send in background so it doesn't slow down the agent
+    thread = threading.Thread(target=_send, args=(payload,), daemon=True)
+    thread.start()
+
+
+def track_anthropic(response, agent_name: str, project: str = None):
+    """Auto-track an Anthropic Claude response object."""
+    model = response.model
+    input_tokens = response.usage.input_tokens
+    output_tokens = response.usage.output_tokens
+    cost_usd = _estimate_cost(model, input_tokens, output_tokens)
+    track(
+        agent_name=agent_name,
+        model=model,
+        input_tokens=input_tokens,
+        output_tokens=output_tokens,
+        cost_usd=cost_usd,
+        project=project,
+    )
+
+
+def track_openai(response, agent_name: str, project: str = None):
+    """Auto-track an OpenAI response object."""
+    model = response.model
+    input_tokens = response.usage.prompt_tokens
+    output_tokens = response.usage.completion_tokens
+    cost_usd = _estimate_cost(model, input_tokens, output_tokens)
+    track(
+        agent_name=agent_name,
+        model=model,
+        input_tokens=input_tokens,
+        output_tokens=output_tokens,
+        cost_usd=cost_usd,
+        project=project,
+    )
+
+
+def _estimate_cost(model: str, input_tokens: int, output_tokens: int) -> float:
+    pricing = {
+        "claude-opus-4-8":           (15.0, 75.0),
+        "claude-sonnet-4-6":         (3.0,  15.0),
+        "claude-haiku-4-5-20251001": (0.8,  4.0),
+        "claude-3-5-sonnet-20241022":(3.0,  15.0),
+        "claude-3-5-haiku-20241022": (0.8,  4.0),
+        "gpt-4o":                    (2.5,  10.0),
+        "gpt-4o-mini":               (0.15, 0.6),
+        "gpt-4-turbo":               (10.0, 30.0),
+    }
+    input_price, output_price = pricing.get(model, (3.0, 15.0))
+    return (input_tokens * input_price / 1_000_000) + (output_tokens * output_price / 1_000_000)
+
+
+def _send(payload: dict):
+    try:
+        requests.post(
+            f"{_base_url}/events",
+            headers={"Authorization": f"Bearer {_api_key}"},
+            json=payload,
+            timeout=5,
+        )
+    except Exception:
+        pass  # Never crash the user's agent

agentspend_tracker-0.1.0/agentspend_tracker.egg-info/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: agentspend-tracker
+Version: 0.1.0
+Summary: Track LLM costs across your AI agents
+Author: Sankar
+Requires-Python: >=3.8
+Requires-Dist: requests>=2.28.0
+Dynamic: author
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

agentspend_tracker-0.1.0/agentspend_tracker.egg-info/SOURCES.txt

@@ -0,0 +1,8 @@
+README.md
+setup.py
+agentspend/__init__.py
+agentspend_tracker.egg-info/PKG-INFO
+agentspend_tracker.egg-info/SOURCES.txt
+agentspend_tracker.egg-info/dependency_links.txt
+agentspend_tracker.egg-info/requires.txt
+agentspend_tracker.egg-info/top_level.txt

agentspend_tracker-0.1.0/agentspend_tracker.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

agentspend_tracker-0.1.0/agentspend_tracker.egg-info/requires.txt

@@ -0,0 +1 @@
+requests>=2.28.0

agentspend_tracker-0.1.0/agentspend_tracker.egg-info/top_level.txt

@@ -0,0 +1 @@
+agentspend

agentspend_tracker-0.1.0/setup.cfg

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

agentspend_tracker-0.1.0/setup.py

@@ -0,0 +1,11 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="agentspend-tracker",
+    version="0.1.0",
+    description="Track LLM costs across your AI agents",
+    author="Sankar",
+    packages=find_packages(),
+    install_requires=["requests>=2.28.0"],
+    python_requires=">=3.8",
+)