开源AI工作流：LLM预算控制和AI代理成本治理

Install

pip install shekel[openai]       # OpenAI
pip install shekel[anthropic]    # Anthropic
pip install shekel[all]          # OpenAI + Anthropic + LiteLLM + Gemini + HuggingFace
pip install shekel[cli]          # shekel run — enforce budgets without touching code

---

Shell / cron / Docker

AGENT_BUDGET_USD=5 shekel run agent.py

Docker — operator sets budget at runtime, no rebuild needed

ENTRYPOINT ["shekel", "run", "agent.py"]

docker run -e AGENT_BUDGET_USD=5 my-agent-image

Key flags:

---

BudgetConfigMismatchError if the same name is reused with different limits

```

Works with AsyncRedisBackend for async workflows. Circuit breaker built in — configurable threshold + cooldown. Fail-open or fail-closed.

Each user gets their own isolated $0.10 cap — same Redis, zero per-tenant config

```

Quota management: backend.set_tenant_limit(...), backend.get_tenant_spend(...), backend.reset_tenant(...), backend.list_tenants(...). Inspect from the command line with shekel tenants list --name api.

Enforce from the CLI — zero code changes

Don't want to touch the code at all? Don't.

```bash pip install shekel[cli]

shekel run agent.py --budget 5

API quick reference

budget(
    max_usd=5.00,           # hard USD cap
    warn_at=0.8,            # warn at 80%
    max_llm_calls=50,       # cap by call count
    max_tool_calls=100,     # cap tool dispatches
    tool_prices={"web_search": 0.01},  # charge per tool
    fallback={"at_pct": 0.8, "model": "gpt-4o-mini"},  # switch instead of crash
    name="my-agent",        # required for nesting + temporal budgets
    backend=RedisBackend(), # distributed enforcement across processes
)

budget("$5/hr + 100 calls/hr", name="api-tier")  # multi-cap rolling-window

Component caps — all chainable, all raise before the component executes:

b.node("fetch_data", max_usd=0.50)   # LangGraph node  → NodeBudgetExceededError
b.chain("retriever", max_usd=0.20)   # LangChain chain → ChainBudgetExceededError
b.agent("researcher", max_usd=1.00)  # CrewAI agent    → AgentBudgetExceededError
b.task("summarize", max_usd=0.50)    # CrewAI task     → TaskBudgetExceededError

Exceptions — all subclass BudgetExceededError, so one except catches everything:

---

pipeline: $4.80 / $10.00

Supported models

Built-in pricing for GPT-4o, GPT-4o-mini, o1, o3, Claude 3.5/3/3.7 Sonnet, Claude 3 Haiku/Opus, Gemini 2.0/2.5 Flash/Pro, and more.

pip install shekel[all-models]   # 400+ models via tokencost
shekel models                    # list all bundled models and pricing
shekel estimate --model gpt-4o --input-tokens 1000 --output-tokens 500

---