tokenfactory-pi

Nebius Token Factory provider extension for pi coding agent.

Fetches the current model catalog from the Token Factory API on startup and registers all tool-capable models. No changes to pi-mono required.

Prerequisites

# Install pi coding agent globally (required)
npm install -g @mariozechner/pi-coding-agent

Installation

# Install the extension using pi's built-in package manager
pi install npm:tokenfactory-pi

Setup

# Get an API key from https://tokenfactory.nebius.com/
export NEBIUS_API_KEY=your-key-here

Usage

# List available models to verify installation
pi --list-models | grep nebius

Once running in interactive mode, use /nebius-models to list all available models.

Development

For local development:

# Build the TypeScript
npm run build

# Test locally from the project directory
cd path/to/tokenfactory-pi
pi -e . --provider nebius

How it works

On startup the extension:

1. Reads NEBIUS_API_KEY from environment (no-op if missing)
2. Fetches GET /v1/models?verbose=true from the Token Factory API
3. Filters for models with tools support and ->text output modality
4. Registers them as the nebius provider via pi.registerProvider()

All models use the openai-completions API with compat: { supportsDeveloperRole: false, maxTokensField: "max_tokens" }.