Skip to main content

Zero-config AI usage tracking — wrap any OpenAI/Anthropic/Gemini client and log to your TokenGauge dashboard

Project description

TokenGauge SDK

Zero-config AI usage tracking. Wrap your existing OpenAI, Anthropic, or Google Gemini client with one line — every call is automatically logged to your TokenGauge dashboard.

Your API keys stay with you. The SDK only reads token counts from API responses and sends them to TokenGauge. Nothing is proxied.

Install

pip install tokengauge

Quick start

  1. Sign up at tokengauge.onrender.com and copy your SDK token from Settings.

  2. Wrap your client:

from tokengauge import TokenGauge
import openai

tw = TokenGauge(token="your-sdk-token")
client = tw.wrap(openai.OpenAI(api_key="sk-..."))

# Use exactly as before — usage appears on your dashboard automatically
response = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=[{"role": "user", "content": "Hello!"}],
)
print(response.choices[0].message.content)

Anthropic

from tokengauge import TokenGauge
import anthropic

tw = TokenGauge(token="your-sdk-token")
client = tw.wrap(anthropic.Anthropic(api_key="sk-ant-..."))

response = client.messages.create(
    model="claude-3-5-sonnet-20241022",
    max_tokens=256,
    messages=[{"role": "user", "content": "Hello!"}],
)
print(response.content[0].text)

Google Gemini

from tokengauge import TokenGauge
from google import genai

tw = TokenGauge(token="your-sdk-token")
client = tw.wrap(genai.Client(api_key="your-gemini-key"))

response = client.models.generate_content(
    model="gemini-1.5-flash",
    contents="Hello!",
)
print(response.text)

Async clients

from tokengauge import TokenGauge
import openai, asyncio

tw = TokenGauge(token="your-sdk-token")
client = tw.wrap(openai.AsyncOpenAI(api_key="sk-..."))

async def main():
    response = await client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[{"role": "user", "content": "Hello!"}],
    )
    print(response.choices[0].message.content)

asyncio.run(main())

Tag calls by feature

summarizer = tw.wrap(openai.OpenAI(api_key="sk-..."), app_tag="summarizer")
chatbot    = tw.wrap(openai.OpenAI(api_key="sk-..."), app_tag="chatbot")

Login instead of pasting a token

tw = TokenGauge.login(email="you@example.com", password="your-password")

What gets tracked

Field Description
Provider openai / anthropic / google
Model e.g. gpt-4o-mini, claude-3-5-sonnet
Tokens in Prompt token count
Tokens out Completion token count
Cost (USD) Calculated from current model pricing
Latency End-to-end request time in ms
App tag Optional label you set per-client

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

tokengauge-0.2.2.tar.gz (7.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

tokengauge-0.2.2-py3-none-any.whl (7.1 kB view details)

Uploaded Python 3

File details

Details for the file tokengauge-0.2.2.tar.gz.

File metadata

  • Download URL: tokengauge-0.2.2.tar.gz
  • Upload date:
  • Size: 7.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.7

File hashes

Hashes for tokengauge-0.2.2.tar.gz
Algorithm Hash digest
SHA256 64000d92f6b1d398ff8f9bbb992ca93695ac4fd8fcbf198e55da1dee05de55dd
MD5 66ceaa9dd21bb5db66b3f9ae509648e6
BLAKE2b-256 7f5abf407d9194090efac7f038caa24f579f3cfebb37b478cd40fa6d11b6cc44

See more details on using hashes here.

File details

Details for the file tokengauge-0.2.2-py3-none-any.whl.

File metadata

  • Download URL: tokengauge-0.2.2-py3-none-any.whl
  • Upload date:
  • Size: 7.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.7

File hashes

Hashes for tokengauge-0.2.2-py3-none-any.whl
Algorithm Hash digest
SHA256 3360de5800275019951e2d0de289f9ad6ab95b1991ed36aeec7c2f85d3c9e0a0
MD5 2f5fe3db5cfb4598b628bd781b758100
BLAKE2b-256 31d1e9366f883863b7d60e8a123bce1610f2aaf97cc89b7103b6dc64d02e828a

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page