Image Generation Example
This example demonstrates how to use the ImageGenerator
utility within AgentiPy to create images using DALL·E. It leverages Agentipy's SolanaAgentKit
for authentication (by holding the OpenAI API key) and asynchronous handling of the image generation request.
Key Features of the ImageGenerator
Tool
ImageGenerator
ToolSecure Key Management: Automatically uses the
openai_api_key
stored in aSolanaAgentKit
instance.Async-First API: All operations use the OpenAI
AsyncOpenAI
client for non-blocking calls.Flexible Models & Sizes: Defaults to DALL·E 2 with support for 256×256, 512×512, and 1024×1024; easily switch to DALL·E 3 by adjusting the
model
parameter.Robust Error Handling: Throws descriptive errors if the API key is missing or if generation fails.
Code (gen_img.py
)
gen_img.py
)import os
import asyncio
from agentipy.tools.create_image import ImageGenerator
from agentipy.agent import SolanaAgentKit
async def main():
api_key = "sk-proj----"
#api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
raise ValueError("Please set the OPENAI_API_KEY environment variable.")
agent = SolanaAgentKit(openai_api_key=api_key)
prompt = "A cartoon green python snake wearing tiny glasses, typing Python code on a glowing laptop. The screen shows Solana blockchain code (e.g., 'async def transaction()') with Solana logo (purple S-shaped vortex) floating nearby. Snake’s tail forms a blockchain symbol (🔗) and a tiny SOL coin. Background: clean digital workspace with abstract crypto nodes. Style: friendly, minimalist 2D flat design with mint green and purple accents."
size = "1024x1024"
num_images = 1
try:
response = await ImageGenerator.create_image(agent, prompt, size=size, n=num_images)
images = response.get("images", [])
for i, url in enumerate(images, start=1):
print(f"Image {i}: {url}")
except Exception as err:
print(f"Failed to generate images: {err}")
if __name__ == "__main__":
asyncio.run(main())
Source Files
You can find the source files for this example (including readme.md
with more details on setup, configuration, and best practices) on GitHub: https://github.com/niceberginc/agentipy/blob/main/examples/Image_Generation/