axio-transport-nebius

Nebius AI Studio transport for axio.

Extends axio-transport-openai with Nebius-specific API endpoint and dynamic model discovery — list available models directly from the Studio API.

Features

• Nebius AI Studio endpoint pre-configured
• Dynamic model list — fetch available models at runtime from the Nebius API
• Full OpenAI compatibility — inherits streaming, retry, and tool-calling from axio-transport-openai
• Optional TUI settings screen — install with [tui] extra

Installation

pip install axio-transport-nebius

Usage

from axio import Agent
from axio.context import MemoryContextStore
from axio_transport_nebius import NebiusTransport

transport = NebiusTransport(
    api_key="your-nebius-api-key",
    model="meta-llama/Meta-Llama-3.1-70B-Instruct",
)

agent = Agent(system="You are a helpful assistant.", tools=[], transport=transport)

async def main() -> None:
    ctx = MemoryContextStore()
    result = await agent.run("Explain transformers in one paragraph.", ctx)
    print(result)

Discover available models

from axio_transport_nebius import NebiusTransport

async def list_models() -> None:
    transport = NebiusTransport(api_key="your-key", model="")
    models = await transport.list_models()
    for m in models:
        print(m)

Plugin registration

[project.entry-points."axio.transport"]
nebius = "axio_transport_nebius:NebiusTransport"

Part of the axio ecosystem

axio · axio-transport-openai · axio-transport-codex · axio-tui

License

MIT