feat: support dalle api

This commit is contained in:
ikechan8370 2023-02-25 11:38:34 +08:00
parent 47976f2325
commit 8c9c098e25
5 changed files with 81 additions and 1 deletions

19
utils/dalle.js Normal file
View file

@ -0,0 +1,19 @@
import { Configuration, OpenAIApi } from 'openai'
import { Config } from './config.js'
export async function createImage (prompt, n = 1, size = '512x512') {
const configuration = new Configuration({
apiKey: Config.apiKey
})
const openai = new OpenAIApi(configuration)
if (Config.debug) {
logger.info({ prompt, n, size })
}
const response = await openai.createImage({
prompt,
n,
size,
response_format: 'b64_json'
})
return response.data.data?.map(pic => pic.b64_json)
}