54 lines
1.4 KiB
TypeScript
54 lines
1.4 KiB
TypeScript
import consola from 'consola'
|
|
import type OpenAI from 'openai'
|
|
import { baseChatCompletionCreateParams, baseModel, config, openai } from './config'
|
|
|
|
// TODO: pass params
|
|
|
|
export async function getCompletion(msg: string) {
|
|
const chatCompletion = await openai.chat.completions.create({
|
|
...baseChatCompletionCreateParams,
|
|
messages: [{ role: 'user', content: msg }],
|
|
model: baseModel,
|
|
})
|
|
|
|
return chatCompletion.choices
|
|
}
|
|
|
|
/**
|
|
* 获取 ai 生成的菜谱信息
|
|
*/
|
|
export async function getAIRecipeInfo(zhFoods: string[]) {
|
|
/**
|
|
* 限制输入长度
|
|
*/
|
|
const promptFoods = zhFoods.join('、').slice(0, config.inputMaxLength)
|
|
|
|
// 尽可能少的 token
|
|
const tooltip = `
|
|
使用以下材料【${promptFoods}】做一道菜，请为这道菜起个名字，最好带有文化底蕴。
|
|
不要使用生僻字和标点符号。
|
|
并给出一个有趣的不超过100字的介绍。
|
|
格式类型：{
|
|
"名称": "",
|
|
"介绍": ""
|
|
}
|
|
直接给出可以被 JSON.parse 解析的字符串，不需要解释内容。`
|
|
|
|
const messages: OpenAI.ChatCompletionMessageParam[] = [
|
|
{
|
|
role: 'system',
|
|
content: tooltip,
|
|
},
|
|
]
|
|
|
|
const chatCompletion = await openai.chat.completions.create({
|
|
...baseChatCompletionCreateParams,
|
|
messages,
|
|
model: baseModel,
|
|
// stream: true
|
|
})
|
|
|
|
consola.debug(chatCompletion)
|
|
return chatCompletion.choices[0].message
|
|
}
|