agentspend-tracker 0.1.0__py3-none-any.whl

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.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,5 @@
+agentspend/__init__.py,sha256=DZfJLRPN_TmPod-oldf6XGjublKuWBMy22GAHpHPx68,3008
+agentspend_tracker-0.1.0.dist-info/METADATA,sha256=_ZyLrAMdj2q_TCH77UDHgr_yXUJk1CsUZJJVoXDZ0l8,271
+agentspend_tracker-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+agentspend_tracker-0.1.0.dist-info/top_level.txt,sha256=NxRmonI3i5h_eqUQt0XwwkEkl9AFWHQn6NkvoAxgiUw,11
+agentspend_tracker-0.1.0.dist-info/RECORD,,

agentspend_tracker-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

agentspend_tracker-0.1.0.dist-info/top_level.txt

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