22 lines
634 B
JavaScript
22 lines
634 B
JavaScript
const { Configuration, OpenAIApi } = require("openai");
|
|
const configuration = new Configuration({
|
|
apiKey: process.env["OPENAI_API_KEY"]
|
|
});
|
|
const openai = new OpenAIApi(configuration);
|
|
|
|
const model = process.env["OPENAI_MODEL_NAME"] ?? 'image-alpha-001'
|
|
const max_tokens = Number(process.env["OPENAI_MAX_TOKENS"] ?? 2000)
|
|
|
|
async function continueThread(messages) {
|
|
const response = await openai.generateImage({
|
|
prompt: messages,
|
|
model,
|
|
n: 1,
|
|
size: "512x512",
|
|
response_format: "b64_json",
|
|
});
|
|
return response.json?.data?.choices?.[0]?.message?.content
|
|
}
|
|
|
|
module.exports = { continueThread }
|