agentcost 0.1.0__tar.gz

agentcost-0.1.0/.env.example

@@ -0,0 +1,29 @@
+# AgentCost SDK - Environment Variables
+# Copy this file to .env and update with your values
+# ⚠️ NEVER commit the actual .env file to version control!
+
+# ===========================================
+# LLM API KEYS (for demos/testing)
+# ===========================================
+
+# OpenAI (https://platform.openai.com/api-keys)
+OPENAI_API_KEY=sk-your-openai-key-here
+
+# Groq (https://console.groq.com/keys) - FREE!
+GROQ_API_KEY=gsk_your-groq-key-here
+
+# Anthropic (https://console.anthropic.com/)
+ANTHROPIC_API_KEY=sk-ant-your-anthropic-key-here
+
+# ===========================================
+# AgentCost Configuration
+# ===========================================
+
+# AgentCost backend URL
+AGENTCOST_API_URL=http://localhost:8000
+
+# Project API key (create via backend API or dashboard)
+AGENTCOST_API_KEY=sk_your-agentcost-project-key-here
+
+# Project ID (shown when creating project)
+AGENTCOST_PROJECT_ID=your-project-uuid-here

agentcost-0.1.0/.gitignore

@@ -0,0 +1,60 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual environments
+venv/
+ENV/
+env/
+.venv/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+*~
+
+# Environment variables - NEVER COMMIT
+.env
+.env.local
+.env.*.local
+
+# Logs
+*.log
+logs/
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+.tox/
+
+# mypy
+.mypy_cache/
+
+# OS files
+.DS_Store
+Thumbs.db

agentcost-0.1.0/CONTRIBUTING.md

@@ -0,0 +1,30 @@
+# Contributing to AgentCost
+
+Thank you for your interest in contributing!
+
+## How to Contribute
+
+1. Fork the repository
+2. Create a feature branch (`git checkout -b feature/amazing-feature`)
+3. Commit your changes (`git commit -m 'Add amazing feature'`)
+4. Push to the branch (`git push origin feature/amazing-feature`)
+5. Open a Pull Request
+
+## Development Setup
+```bash
+git clone https://github.com/agentcost-ai/agentcost-sdk.git
+cd agentcost-sdk
+pip install -r requirements.txt
+pytest  # Run tests
+```
+
+## Code Style
+
+We use:
+- Black for code formatting
+- Flake8 for linting
+- Type hints for all functions
+
+## Issues
+
+Check existing issues or create a new one before starting work.

agentcost-0.1.0/LICENSE

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

agentcost-0.1.0/MANIFEST.in

@@ -0,0 +1,3 @@
+include README.md
+include LICENSE
+recursive-include agentcost *.py

agentcost-0.1.0/PKG-INFO

@@ -0,0 +1,294 @@
+Metadata-Version: 2.4
+Name: agentcost
+Version: 0.1.0
+Summary: Track LLM costs in LangChain applications with zero code changes
+Home-page: https://github.com/agentcost-ai/agentcost-sdk
+Author: Kushagra Agrawal
+Author-email: kushagraagrawal128@gmail.com
+License: MIT
+Project-URL: Bug Tracker, https://github.com/agentcost-ai/agentcost-sdk/issues
+Project-URL: Documentation, https://www.agentcost.tech/docs/sdk
+Keywords: llm,langchain,openai,anthropic,cost-tracking,tokens,ai,monitoring,observability,langchain ai agents cost tracking monitoring
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+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
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: tiktoken>=0.5.0
+Requires-Dist: requests>=2.28.0
+Requires-Dist: langchain-core>=0.1.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Provides-Extra: langchain
+Requires-Dist: langchain>=0.1.0; extra == "langchain"
+Requires-Dist: langchain-openai>=0.0.5; extra == "langchain"
+Requires-Dist: langchain-anthropic>=0.1.0; extra == "langchain"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# AgentCost SDK
+
+**Zero-friction LLM cost tracking for LangChain applications.**
+
+## Installation
+
+```bash
+pip install agentcost
+```
+
+Or install from source:
+
+```bash
+cd agentcost-sdk
+pip install -e .
+```
+
+## Quick Start
+
+```python
+from agentcost import track_costs
+
+# 2 lines to add cost tracking!
+track_costs.init(api_key="your_api_key", project_id="my-project")
+
+# Your existing code works unchanged
+from langchain_openai import ChatOpenAI
+
+llm = ChatOpenAI(model="gpt-4")
+response = llm.invoke("Hello!")  # Automatically tracked
+```
+
+## Features
+
+- **Zero Code Changes**: Monkey patches LangChain - your code works as-is
+- **Automatic Tracking**: Captures all `invoke()`, `ainvoke()`, `stream()`, `astream()` calls
+- **Accurate Tokens**: Uses `tiktoken` for precise token counting
+- **Real-Time Costs**: Calculates costs using up-to-date model pricing
+- **Batched Sending**: Efficient network usage (size-based + time-based batching)
+- **Rate Limiting**: Built-in rate limiter to protect your backend
+- **Local Mode**: Test without a backend
+
+## Configuration
+
+```python
+track_costs.init(
+    # Required for cloud mode
+    api_key="sk_...",
+    project_id="my-project",
+
+    # Optional settings
+    base_url="https://api.agentcost.tech",   # Your backend URL
+    batch_size=10,                          # Events before auto-flush
+    flush_interval=5.0,                     # Seconds between flushes
+    debug=True,                             # Enable debug logging
+    default_agent_name="my-agent",          # Default agent tag
+    local_mode=False,                       # Store locally (no backend)
+    enabled=True,                           # Enable/disable tracking
+
+    # Custom pricing (overrides defaults)
+    custom_pricing={
+        "my-custom-model": {"input": 0.001, "output": 0.002}
+    },
+
+    # Global metadata (attached to all events)
+    global_metadata={
+        "environment": "production",
+        "version": "1.0.0"
+    }
+)
+```
+
+## Agent Tagging
+
+Tag LLM calls by agent for granular analytics:
+
+```python
+# Option 1: Set default agent
+track_costs.set_agent_name("router-agent")
+
+# Option 2: Context manager (recommended)
+with track_costs.agent("technical-agent"):
+    llm.invoke("How do I fix this?")  # Tagged as "technical-agent"
+
+with track_costs.agent("billing-agent"):
+    llm.invoke("What's my balance?")  # Tagged as "billing-agent"
+```
+
+## Metadata
+
+Attach custom metadata for filtering/grouping:
+
+```python
+# Persistent metadata
+track_costs.add_metadata("user_id", "user_123")
+track_costs.add_metadata("tenant_id", "acme_corp")
+
+# Temporary metadata (context manager)
+with track_costs.metadata(conversation_id="conv_456", step="routing"):
+    llm.invoke("Route this query")
+```
+
+## Local Testing
+
+Test without running a backend:
+
+```python
+track_costs.init(local_mode=True, debug=True)
+
+# Make LLM calls
+llm.invoke("Hello!")
+llm.invoke("World!")
+
+# Retrieve captured events
+events = track_costs.get_local_events()
+for event in events:
+    print(f"Model: {event['model']}")
+    print(f"Tokens: {event['total_tokens']}")
+    print(f"Cost: ${event['cost']:.6f}")
+```
+
+## Streaming Support
+
+Streaming calls are automatically tracked:
+
+```python
+# Sync streaming
+for chunk in llm.stream("Tell me a story"):
+    print(chunk.content, end="")
+# Event recorded after stream completes
+
+# Async streaming
+async for chunk in llm.astream("Tell me a story"):
+    print(chunk.content, end="")
+# Event recorded after stream completes
+```
+
+## Event Structure
+
+Each tracked event contains:
+
+```python
+{
+    "agent_name": "my-agent",
+    "model": "gpt-4",
+    "input_tokens": 150,
+    "output_tokens": 80,
+    "total_tokens": 230,
+    "cost": 0.0093,            # USD, real-time calculated
+    "latency_ms": 1234,        # Measured latency
+    "timestamp": "2026-01-23T10:30:45.123Z",
+    "success": True,
+    "error": None,
+    "streaming": False,
+    "metadata": {"conversation_id": "conv_456"}
+}
+```
+
+## Dynamic Pricing (Real-Time Updates)
+
+The SDK automatically fetches the latest pricing from the backend. This means:
+
+- **No code changes** when model prices change
+- Pricing is **cached for 24 hours** (efficient)
+- Falls back to built-in defaults if backend is unavailable
+
+### How It Works
+
+```python
+# SDK automatically fetches pricing from backend
+track_costs.init(
+    api_key="...",
+    project_id="...",
+    base_url="http://localhost:8000",  # Pricing fetched from here
+)
+
+# Prices are fetched once and cached
+# GET http://localhost:8000/v1/pricing → {"pricing": {"gpt-4": {"input": 0.03, ...}}}
+```
+
+### Manually Update Pricing
+
+```python
+from agentcost.cost_calculator import refresh_pricing, update_pricing
+
+# Force refresh from backend
+refresh_pricing()
+
+# Or manually set pricing (doesn't require backend)
+update_pricing({
+    "my-custom-model": {"input": 0.001, "output": 0.002}
+})
+```
+
+### Backend Pricing API
+
+```bash
+# Get all pricing
+curl http://localhost:8000/v1/pricing
+
+# Get specific model
+curl http://localhost:8000/v1/pricing/gpt-4
+
+# Update pricing (admin)
+curl -X POST http://localhost:8000/v1/pricing \
+  -H "Content-Type: application/json" \
+  -d '{"gpt-4": {"input": 0.025, "output": 0.05}}'
+```
+
+## Supported Models (30+)
+
+| Provider  | Models                                                              |
+| --------- | ------------------------------------------------------------------- |
+| OpenAI    | gpt-4, gpt-4-turbo, gpt-4o, gpt-4o-mini, gpt-3.5-turbo, o1, o1-mini |
+| Anthropic | claude-3-opus/sonnet/haiku, claude-3.5-sonnet/haiku                 |
+| Google    | gemini-pro, gemini-1.5-pro/flash, gemini-2.0-flash                  |
+| Groq      | llama-3.1-8b/70b, llama-3.3-70b, mixtral-8x7b                       |
+| DeepSeek  | deepseek-chat, deepseek-coder, deepseek-reasoner                    |
+| Cohere    | command, command-r, command-r-plus                                  |
+| Mistral   | mistral-small/medium/large                                          |
+
+## Statistics
+
+```python
+stats = track_costs.get_stats()
+print(f"Events sent: {stats['batcher']['events_sent']}")
+print(f"Batches sent: {stats['batcher']['batches_sent']}")
+```
+
+## Graceful Shutdown
+
+```python
+track_costs.flush()     # Send pending events
+track_costs.shutdown()  # Full shutdown
+```
+
+## License
+
+MIT License

agentcost-0.1.0/README.md

@@ -0,0 +1,239 @@
+# AgentCost SDK
+
+**Zero-friction LLM cost tracking for LangChain applications.**
+
+## Installation
+
+```bash
+pip install agentcost
+```
+
+Or install from source:
+
+```bash
+cd agentcost-sdk
+pip install -e .
+```
+
+## Quick Start
+
+```python
+from agentcost import track_costs
+
+# 2 lines to add cost tracking!
+track_costs.init(api_key="your_api_key", project_id="my-project")
+
+# Your existing code works unchanged
+from langchain_openai import ChatOpenAI
+
+llm = ChatOpenAI(model="gpt-4")
+response = llm.invoke("Hello!")  # Automatically tracked
+```
+
+## Features
+
+- **Zero Code Changes**: Monkey patches LangChain - your code works as-is
+- **Automatic Tracking**: Captures all `invoke()`, `ainvoke()`, `stream()`, `astream()` calls
+- **Accurate Tokens**: Uses `tiktoken` for precise token counting
+- **Real-Time Costs**: Calculates costs using up-to-date model pricing
+- **Batched Sending**: Efficient network usage (size-based + time-based batching)
+- **Rate Limiting**: Built-in rate limiter to protect your backend
+- **Local Mode**: Test without a backend
+
+## Configuration
+
+```python
+track_costs.init(
+    # Required for cloud mode
+    api_key="sk_...",
+    project_id="my-project",
+
+    # Optional settings
+    base_url="https://api.agentcost.tech",   # Your backend URL
+    batch_size=10,                          # Events before auto-flush
+    flush_interval=5.0,                     # Seconds between flushes
+    debug=True,                             # Enable debug logging
+    default_agent_name="my-agent",          # Default agent tag
+    local_mode=False,                       # Store locally (no backend)
+    enabled=True,                           # Enable/disable tracking
+
+    # Custom pricing (overrides defaults)
+    custom_pricing={
+        "my-custom-model": {"input": 0.001, "output": 0.002}
+    },
+
+    # Global metadata (attached to all events)
+    global_metadata={
+        "environment": "production",
+        "version": "1.0.0"
+    }
+)
+```
+
+## Agent Tagging
+
+Tag LLM calls by agent for granular analytics:
+
+```python
+# Option 1: Set default agent
+track_costs.set_agent_name("router-agent")
+
+# Option 2: Context manager (recommended)
+with track_costs.agent("technical-agent"):
+    llm.invoke("How do I fix this?")  # Tagged as "technical-agent"
+
+with track_costs.agent("billing-agent"):
+    llm.invoke("What's my balance?")  # Tagged as "billing-agent"
+```
+
+## Metadata
+
+Attach custom metadata for filtering/grouping:
+
+```python
+# Persistent metadata
+track_costs.add_metadata("user_id", "user_123")
+track_costs.add_metadata("tenant_id", "acme_corp")
+
+# Temporary metadata (context manager)
+with track_costs.metadata(conversation_id="conv_456", step="routing"):
+    llm.invoke("Route this query")
+```
+
+## Local Testing
+
+Test without running a backend:
+
+```python
+track_costs.init(local_mode=True, debug=True)
+
+# Make LLM calls
+llm.invoke("Hello!")
+llm.invoke("World!")
+
+# Retrieve captured events
+events = track_costs.get_local_events()
+for event in events:
+    print(f"Model: {event['model']}")
+    print(f"Tokens: {event['total_tokens']}")
+    print(f"Cost: ${event['cost']:.6f}")
+```
+
+## Streaming Support
+
+Streaming calls are automatically tracked:
+
+```python
+# Sync streaming
+for chunk in llm.stream("Tell me a story"):
+    print(chunk.content, end="")
+# Event recorded after stream completes
+
+# Async streaming
+async for chunk in llm.astream("Tell me a story"):
+    print(chunk.content, end="")
+# Event recorded after stream completes
+```
+
+## Event Structure
+
+Each tracked event contains:
+
+```python
+{
+    "agent_name": "my-agent",
+    "model": "gpt-4",
+    "input_tokens": 150,
+    "output_tokens": 80,
+    "total_tokens": 230,
+    "cost": 0.0093,            # USD, real-time calculated
+    "latency_ms": 1234,        # Measured latency
+    "timestamp": "2026-01-23T10:30:45.123Z",
+    "success": True,
+    "error": None,
+    "streaming": False,
+    "metadata": {"conversation_id": "conv_456"}
+}
+```
+
+## Dynamic Pricing (Real-Time Updates)
+
+The SDK automatically fetches the latest pricing from the backend. This means:
+
+- **No code changes** when model prices change
+- Pricing is **cached for 24 hours** (efficient)
+- Falls back to built-in defaults if backend is unavailable
+
+### How It Works
+
+```python
+# SDK automatically fetches pricing from backend
+track_costs.init(
+    api_key="...",
+    project_id="...",
+    base_url="http://localhost:8000",  # Pricing fetched from here
+)
+
+# Prices are fetched once and cached
+# GET http://localhost:8000/v1/pricing → {"pricing": {"gpt-4": {"input": 0.03, ...}}}
+```
+
+### Manually Update Pricing
+
+```python
+from agentcost.cost_calculator import refresh_pricing, update_pricing
+
+# Force refresh from backend
+refresh_pricing()
+
+# Or manually set pricing (doesn't require backend)
+update_pricing({
+    "my-custom-model": {"input": 0.001, "output": 0.002}
+})
+```
+
+### Backend Pricing API
+
+```bash
+# Get all pricing
+curl http://localhost:8000/v1/pricing
+
+# Get specific model
+curl http://localhost:8000/v1/pricing/gpt-4
+
+# Update pricing (admin)
+curl -X POST http://localhost:8000/v1/pricing \
+  -H "Content-Type: application/json" \
+  -d '{"gpt-4": {"input": 0.025, "output": 0.05}}'
+```
+
+## Supported Models (30+)
+
+| Provider  | Models                                                              |
+| --------- | ------------------------------------------------------------------- |
+| OpenAI    | gpt-4, gpt-4-turbo, gpt-4o, gpt-4o-mini, gpt-3.5-turbo, o1, o1-mini |
+| Anthropic | claude-3-opus/sonnet/haiku, claude-3.5-sonnet/haiku                 |
+| Google    | gemini-pro, gemini-1.5-pro/flash, gemini-2.0-flash                  |
+| Groq      | llama-3.1-8b/70b, llama-3.3-70b, mixtral-8x7b                       |
+| DeepSeek  | deepseek-chat, deepseek-coder, deepseek-reasoner                    |
+| Cohere    | command, command-r, command-r-plus                                  |
+| Mistral   | mistral-small/medium/large                                          |
+
+## Statistics
+
+```python
+stats = track_costs.get_stats()
+print(f"Events sent: {stats['batcher']['events_sent']}")
+print(f"Batches sent: {stats['batcher']['batches_sent']}")
+```
+
+## Graceful Shutdown
+
+```python
+track_costs.flush()     # Send pending events
+track_costs.shutdown()  # Full shutdown
+```
+
+## License
+
+MIT License

agentcost-0.1.0/VERSION

@@ -0,0 +1 @@
+0.1.0