tokenetc 0.1.2

Auto-track AI SDK token usage with cost calculation and statistics

tokenetc

自动追踪 OpenAI / Anthropic / Gemini SDK 的 token 用量，支持成本计算与多渠道统计。

安装

pip install tokenetc

使用

import tokenetc

# 一行 patch，自动追踪所有已安装 SDK 的调用（同步 + 异步）
tokenetc.patch()

# 之后正常使用 openai / anthropic / gemini，用量自动记录
import openai
client = openai.OpenAI()
response = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "hello"}],
)

# 手动录入（Cursor / 网页版等无法自动追踪的渠道）
tokenetc.record(
    channel="cursor",
    model="claude-sonnet-4",
    input_tokens=3000,
    output_tokens=500,
)

# 计算费用
cost = tokenetc.cost("gpt-4o", input_tokens=1000, output_tokens=500)
print(f"${cost:.6f}")  # $0.007500

# 查询统计
stats = tokenetc.stats(days=7)
print(stats["total_tokens"])
print(stats["cost_usd"])
print(stats["by_channel"])
print(stats["by_model"])

# 查看所有支持的模型价格
tokenetc.list_models()

数据存储

数据保存在 ~/.tokenetc/data.db（SQLite）。