Dalle

Prerequisites

You need to install the openai library.

pip install openai

Set the OPENAI_API_KEY environment variable.

export OPENAI_API_KEY=****

Example

The following agent will use DALL-E to generate an image based on a text prompt.

cookbook/tools/dalle_tools.py

from bitca.agent import Agent
from bitca.tools.dalle import Dalle

# Create an Agent with the DALL-E tool
agent = Agent(tools=[Dalle()], name="DALL-E Image Generator")

# Example 1: Generate a basic image with default settings
agent.print_response("Generate an image of a futuristic city with flying cars and tall skyscrapers", markdown=True)

# Example 2: Generate an image with custom settings
custom_dalle = Dalle(model="dall-e-3", size="1792x1024", quality="hd", style="natural")

agent_custom = Agent(
    tools=[custom_dalle],
    name="Custom DALL-E Generator",
    show_tool_calls=True,
)

agent_custom.print_response("Create a panoramic nature scene showing a peaceful mountain lake at sunset", markdown=True)

Toolkit Params

Parameter
Type
Default
Description

model

str

"dall-e-3"

The DALL-E model to use

n

int

1

Number of images to generate

size

str

"1024x1024"

Image size (256x256, 512x512, 1024x1024, 1792x1024, or 1024x1792)

quality

str

"standard"

Image quality (standard or hd)

style

str

"vivid"

Image style (vivid or natural)

api_key

str

None

The OpenAI API key for authentication

Toolkit Functions

Function
Description

generate_image

Generates an image based on a text prompt

Last updated