agent-cost-attribution-layer

agent-cost-attribution-layer

A FastAPI-based middleware proxy that automatically attributes every AI request to agents, users, and tasks—generating compliance-ready audit trails for multi-agent deployments.

What is this?

This is a transparent proxy layer that sits between your AI agents and LLM APIs (OpenAI, Anthropic, local models, etc.). It automatically tags every request with agent identity, user attribution, task context, and cost, then logs everything to a queryable audit database. It solves the core problem revealed by the VS Code Copilot attribution drama: "Who authorized this AI action and what data did it use?" Unlike cost-monitoring tools, this focuses on consent tracking, regulatory compliance, and multi-agent accountability.

Features

• Transparent proxy — Drop-in replacement for OpenAI/Anthropic client calls; no agent code changes required
• Automatic attribution — Tags every request with agent ID, user ID, task context, and timestamp
• Audit trails — SQLite-backed compliance log with prompt hashes, token usage, and costs
• REST API — Query attribution history: "Show all AI actions by user X in the last 30 days"
• Export compliance — CSV (legal), JSON (monitoring), and GDPR-compliant deletion endpoints
• CLI tooling — agent-attr proxy start --port 8080 --log-db audit.db
• Rate limiting — Per-user, per-agent quotas with configurable tiers
• Multi-tier pricing — Freemium (local), Pro (cloud-hosted), Enterprise (SOC2 audit trails)

Quick Start

Installation

# Clone and install
git clone https://github.com/yourusername/agent-cost-attribution-layer.git
cd agent-cost-attribution-layer
pip install -e .

Basic Setup

# Copy environment template
cp .env.example .env

# Start the proxy server
agent-attr proxy start --port 8080 --log-db ./audit.db

Docker Deployment

docker-compose up -d

The proxy is now listening on http://localhost:8080.

Usage

Redirect agent requests through the proxy

Before:

from openai import OpenAI
client = OpenAI(api_key="sk-...")
response = client.chat.completions.create(model="gpt-4", messages=[...])

After (one-line change):

from openai import OpenAI
client = OpenAI(api_key="sk-...", base_url="http://localhost:8080/v1")
# Add attribution headers
client.default_headers = {
    "X-Agent-ID": "code-review-bot",
    "X-User-ID": "user-123",
    "X-Task-Context": "PR review #456"
}
response = client.chat.completions.create(model="gpt-4", messages=[...])

Query audit logs via REST API

# Get all actions by a user in the last 30 days
curl http://localhost:8080/audit/user/user-123?days=30

# Get all actions by an agent
curl http://localhost:8080/audit/agent/code-review-bot

# Export as CSV (for compliance)
curl http://localhost:8080/audit/export/csv?start_date=2026-05-01&end_date=2026-05-31 > audit.csv

CLI commands

# Start proxy
agent-attr proxy start --port 8080 --log-db audit.db

# Query logs
agent-attr audit query --agent code-review-bot --days 7

# Export for compliance
agent-attr audit export --format csv --output audit.csv

# GDPR deletion
agent-attr audit delete --user user-123

Tech Stack

• FastAPI — High-performance async proxy server
• SQLite — Local-first audit log (no external DB required)
• Pydantic — Request/response validation
• pytest — Test suite
• Docker — Container deployment
• Python 3.10+

License

MIT