feat: add generate cook & intro

This commit is contained in:
YunYouJun
2024-03-22 19:55:33 +08:00
parent a80d9163b8
commit 5efd7b0533
16 changed files with 3940 additions and 2578 deletions

53
packages/ai/src/api.ts Normal file
View File

@@ -0,0 +1,53 @@
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
}