tokenary 0.1.0__tar.gz

tokenary-0.1.0/.github/workflows/ci.yaml

@@ -0,0 +1,28 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
+        with:
+          enable-cache: true
+
+      - name: Set up Python
+        run: uv python install 3.12
+
+      - name: Install dependencies
+        run: uv sync --group dev
+
+      - name: Run tests
+        run: uv run pytest

tokenary-0.1.0/.gitignore

@@ -0,0 +1,96 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+temp/
+
+# Virtual environments
+venv/
+env/
+.venv/
+.env/
+
+# Mkdocs
+site/
+
+# Distribution / packaging
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Jupyter Notebook checkpoints
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre
+.pyre/
+
+# Pytype
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# VS Code
+.vscode/
+
+# JetBrains IDEs (PyCharm, etc.)
+.idea/
+
+# MacOS
+.DS_Store
+
+# Windows
+Thumbs.db
+ehthumbs.db
+desktop.ini
+
+# dotenv files
+.env
+
+docs/node_modules/
+
+.ruff_cache/
+bandit-report.json
+
+.claude/

tokenary-0.1.0/.pre-commit-config.yaml

@@ -0,0 +1,27 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.6.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+        exclude: ^mkdocs\.yml$
+      - id: check-toml
+      - id: check-merge-conflict
+      - id: check-added-large-files
+        args: ["--maxkb=1000"]
+        exclude: ^tokenary/_generated\.py$
+      - id: check-case-conflict
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.13.1
+    hooks:
+      - id: ruff
+        args: [--fix, --exit-non-zero-on-fix]
+      - id: ruff-format
+
+  - repo: https://github.com/asottile/pyupgrade
+    rev: v3.15.2
+    hooks:
+      - id: pyupgrade
+        args: [--py39-plus]

tokenary-0.1.0/PKG-INFO

@@ -0,0 +1,95 @@
+Metadata-Version: 2.4
+Name: tokenary
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.14
+Requires-Dist: pydantic>=2.13.4
+Provides-Extra: test
+Requires-Dist: pytest-cov>=7.0.0; extra == 'test'
+Requires-Dist: pytest>=9.0.0; extra == 'test'
+Description-Content-Type: text/markdown
+
+# tokenary
+
+Minimal Python library to calculate LLM API costs based on the LiteLLM model catalog.
+
+## Installation
+
+```bash
+pip install tokenary
+```
+
+## Usage
+
+### Functional API
+
+```python
+import tokenary
+from tokenary import ModelName
+
+result = tokenary.calculate(
+    model=ModelName.AZURE_GPT_3_5_TURBO,
+    input_tokens=1000,
+    output_tokens=500,
+)
+
+print(result.total_cost)
+print(result.model_dump())
+```
+
+### Request object
+
+```python
+from tokenary import ModelName, UsageCostRequest, calculate
+
+request = UsageCostRequest(
+    model=ModelName.AZURE_GPT_3_5_TURBO,
+    input_tokens=2000,
+    output_tokens=800,
+    reasoning_tokens=200,
+)
+
+result = calculate(request)
+print(result.model_dump())
+```
+
+### Reasoning tokens (e.g. o1)
+
+```python
+from tokenary import ModelName, calculate
+
+result = calculate(
+    model=ModelName.O1,
+    input_tokens=500,
+    output_tokens=200,
+    reasoning_tokens=300,
+)
+
+print(f"Total: ${result.total_cost:.6f}")
+print(f"  Reasoning: ${result.reasoning_cost:.6f}")
+```
+
+### All supported parameters
+
+| Parameter                   | Type        | Description                    |
+| --------------------------- | ----------- | ------------------------------ |
+| `model`                     | `ModelName` | Model identifier               |
+| `input_tokens`              | `int`       | Number of input tokens         |
+| `output_tokens`             | `int`       | Number of output tokens        |
+| `reasoning_tokens`          | `int`       | Reasoning tokens (e.g. o1)     |
+| `audio_input_tokens`        | `int`       | Audio input tokens             |
+| `generated_images`          | `int`       | Number of generated images     |
+| `code_interpreter_sessions` | `int`       | Code interpreter sessions      |
+| `file_search_calls`         | `int`       | File search API calls          |
+| `file_search_gb_days`       | `float`     | File search storage (GB-days)  |
+| `vector_store_gb_days`      | `float`     | Vector store storage (GB-days) |
+
+The returned `CostBreakdown` object contains per-category costs (`input_cost`, `output_cost`, `reasoning_cost`, …) and a `total_cost`, all in USD.
+
+## Generate pricing artifacts
+
+Re-generate the bundled pricing data from the LiteLLM catalog:
+
+```bash
+python -m tokenary.generator
+```

tokenary-0.1.0/README.md

@@ -0,0 +1,84 @@
+# tokenary
+
+Minimal Python library to calculate LLM API costs based on the LiteLLM model catalog.
+
+## Installation
+
+```bash
+pip install tokenary
+```
+
+## Usage
+
+### Functional API
+
+```python
+import tokenary
+from tokenary import ModelName
+
+result = tokenary.calculate(
+    model=ModelName.AZURE_GPT_3_5_TURBO,
+    input_tokens=1000,
+    output_tokens=500,
+)
+
+print(result.total_cost)
+print(result.model_dump())
+```
+
+### Request object
+
+```python
+from tokenary import ModelName, UsageCostRequest, calculate
+
+request = UsageCostRequest(
+    model=ModelName.AZURE_GPT_3_5_TURBO,
+    input_tokens=2000,
+    output_tokens=800,
+    reasoning_tokens=200,
+)
+
+result = calculate(request)
+print(result.model_dump())
+```
+
+### Reasoning tokens (e.g. o1)
+
+```python
+from tokenary import ModelName, calculate
+
+result = calculate(
+    model=ModelName.O1,
+    input_tokens=500,
+    output_tokens=200,
+    reasoning_tokens=300,
+)
+
+print(f"Total: ${result.total_cost:.6f}")
+print(f"  Reasoning: ${result.reasoning_cost:.6f}")
+```
+
+### All supported parameters
+
+| Parameter                   | Type        | Description                    |
+| --------------------------- | ----------- | ------------------------------ |
+| `model`                     | `ModelName` | Model identifier               |
+| `input_tokens`              | `int`       | Number of input tokens         |
+| `output_tokens`             | `int`       | Number of output tokens        |
+| `reasoning_tokens`          | `int`       | Reasoning tokens (e.g. o1)     |
+| `audio_input_tokens`        | `int`       | Audio input tokens             |
+| `generated_images`          | `int`       | Number of generated images     |
+| `code_interpreter_sessions` | `int`       | Code interpreter sessions      |
+| `file_search_calls`         | `int`       | File search API calls          |
+| `file_search_gb_days`       | `float`     | File search storage (GB-days)  |
+| `vector_store_gb_days`      | `float`     | Vector store storage (GB-days) |
+
+The returned `CostBreakdown` object contains per-category costs (`input_cost`, `output_cost`, `reasoning_cost`, …) and a `total_cost`, all in USD.
+
+## Generate pricing artifacts
+
+Re-generate the bundled pricing data from the LiteLLM catalog:
+
+```bash
+python -m tokenary.generator
+```