Nebius Basic

Prompt a Nebius agent synchronously, asynchronously, and with streaming.

basic.py
"""
Nebius Basic
============

Cookbook example for `nebius/basic.py`.
"""

import asyncio

from agno.agent import Agent
from agno.models.nebius import Nebius

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------

agent = Agent(
    model=Nebius(),
    markdown=True,
)

# Print the response in the terminal

# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
    # --- Sync ---
    agent.print_response("write a two sentence horror story")

    # --- Sync + Streaming ---
    agent.print_response("write a two sentence horror story", stream=True)

    # --- Async ---
    asyncio.run(agent.aprint_response("write a two sentence horror story"))

    # --- Async + Streaming ---
    asyncio.run(agent.aprint_response("write a two sentence horror story", stream=True))

Run the Example

Set up your virtual environment

uv venv --python 3.12
source .venv/bin/activate

Install dependencies

uv pip install -U agno openai

Export your Nebius API key

export NEBIUS_API_KEY="your_nebius_api_key_here"

Select an enabled Token Factory model

Copy a model ID available to your account for https://api.tokenfactory.nebius.com/v1 from the Token Factory console. For knowledge and tool examples it must support function calling; for structured output it must support JSON Schema. Confirm the endpoint and model's access before running.

export NEBIUS_MODEL_ID="your-enabled-model-id"

Replace the placeholder with your copied model ID. Add from os import environ to the saved file, then set every model construction to Nebius(id=environ["NEBIUS_MODEL_ID"]). Keep any additional model options. The adapter's default ID does not establish which models your account can access. A dedicated deployment must instead use the model and endpoint supplied for that deployment, with an explicit base_url.

Run the example

Save the code above as basic.py, then run:

python basic.py

Full source: cookbook/90_models/nebius/basic.py