Example

The following agent can generate images using Nebius AI Studio:
from agno.agent import Agent
from agno.tools.models.nebius import NebiusTools

agent = Agent(
    instructions=[
        "You are an AI image generation assistant using Nebius AI Studio",
        "Create high-quality images based on user descriptions",
        "Provide detailed information about the generated images",
        "Help users refine their prompts for better results",
    ],
    tools=[NebiusTools()],
)

agent.print_response("Generate an image of a futuristic city with flying cars at sunset", stream=True)

Toolkit Params

ParameterTypeDefaultDescription
api_keyOptional[str]NoneNebius API key. Uses NEBIUS_API_KEY if not set.
base_urlstr"https://api.studio.nebius.com/v1"Nebius API base URL.
image_modelstr"black-forest-labs/flux-schnell"Default image generation model.
image_qualityOptional[str]"standard"Image quality setting.
image_sizeOptional[str]"1024x1024"Default image dimensions.
image_styleOptional[str]NoneImage style preference.
enable_generate_imageboolTrueEnable image generation functionality.

Toolkit Functions

FunctionDescription
generate_imageGenerate images from text descriptions using Nebius AI models.

Developer Resources