34 lines
917 B
JavaScript
34 lines
917 B
JavaScript
|
import OpenAI from "openai"
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Generate an image using OpenAI's DALL-E
|
||
|
|
* @param agent SolanaAgentKit instance
|
||
|
|
* @param prompt Text description of the image to generate
|
||
|
|
* @param size Image size ('256x256', '512x512', or '1024x1024') (default: '1024x1024')
|
||
|
|
* @param n Number of images to generate (default: 1)
|
||
|
|
* @returns Object containing the generated image URLs
|
||
|
|
*/
|
||
|
|
export async function create_image(agent, prompt, size = "1024x1024", n = 1) {
|
||
|
|
try {
|
||
|
|
if (!agent.config.OPENAI_API_KEY) {
|
||
|
|
throw new Error("OpenAI API key not found in agent configuration")
|
||
|
|
}
|
||
|
|
|
||
|
|
const openai = new OpenAI({
|
||
|
|
apiKey: agent.config.OPENAI_API_KEY
|
||
|
|
})
|
||
|
|
|
||
|
|
const response = await openai.images.generate({
|
||
|
|
prompt,
|
||
|
|
n,
|
||
|
|
size
|
||
|
|
})
|
||
|
|
|
||
|
|
return {
|
||
|
|
images: response.data.map(img => img.url)
|
||
|
|
}
|
||
|
|
} catch (error) {
|
||
|
|
throw new Error(`Image generation failed: ${error.message}`)
|
||
|
|
}
|
||
|
|
}
|