register_model()

Register a custom model profile.

Usage

Source

register_model(profile)

Use this to add profiles for models not included in the built-in registry (e.g., fine-tuned models or new releases).

Parameters

profile: ModelProfile

The model profile to register.

Examples

import talk_box as tb

tb.register_model(tb.ModelProfile(
    provider="openai",
    model="ft:gpt-4o:my-org:custom:id",
    display_name="My Fine-tuned GPT-4o",
    context_window=128_000,
    supports_tools=True,
    supports_vision=True,
    supports_structured_output=True,
    supports_streaming=True,
    cost_tier=tb.CostTier.HIGH,
))