102 lines
2.4 KiB
TypeScript
102 lines
2.4 KiB
TypeScript
import type { NextApiRequest, NextApiResponse } from 'next';
|
|
import { jsonRes } from '@/service/response';
|
|
import { ChatItemType } from '@/types/chat';
|
|
import { connectToDatabase, Chat, Model } from '@/service/mongo';
|
|
import { authModel } from '@/service/utils/auth';
|
|
import { authUser } from '@/service/utils/auth';
|
|
import mongoose from 'mongoose';
|
|
|
|
type Props = {
|
|
newChatId?: string;
|
|
chatId?: string;
|
|
modelId: string;
|
|
prompts: [ChatItemType, ChatItemType];
|
|
};
|
|
|
|
/* 聊天内容存存储 */
|
|
export default async function handler(req: NextApiRequest, res: NextApiResponse) {
|
|
try {
|
|
const { chatId, modelId, prompts, newChatId } = req.body as Props;
|
|
|
|
if (!prompts) {
|
|
throw new Error('缺少参数');
|
|
}
|
|
|
|
const { userId } = await authUser({ req, authToken: true });
|
|
|
|
const nId = await saveChat({
|
|
chatId,
|
|
modelId,
|
|
prompts,
|
|
newChatId,
|
|
userId
|
|
});
|
|
|
|
jsonRes(res, {
|
|
data: nId
|
|
});
|
|
} catch (err) {
|
|
jsonRes(res, {
|
|
code: 500,
|
|
error: err
|
|
});
|
|
}
|
|
}
|
|
|
|
export async function saveChat({
|
|
chatId,
|
|
newChatId,
|
|
modelId,
|
|
prompts,
|
|
userId
|
|
}: Props & { userId: string }) {
|
|
await connectToDatabase();
|
|
const { model } = await authModel({ modelId, userId, authOwner: false });
|
|
|
|
const content = prompts.map((item) => ({
|
|
_id: item._id ? new mongoose.Types.ObjectId(item._id) : undefined,
|
|
obj: item.obj,
|
|
value: item.value,
|
|
systemPrompt: item.systemPrompt,
|
|
quote: item.quote || []
|
|
}));
|
|
|
|
const [id] = await Promise.all([
|
|
...(chatId // update chat
|
|
? [
|
|
Chat.findByIdAndUpdate(chatId, {
|
|
$push: {
|
|
content: {
|
|
$each: content
|
|
}
|
|
},
|
|
title: content[0].value.slice(0, 20),
|
|
latestChat: content[1].value,
|
|
updateTime: new Date()
|
|
}).then(() => '')
|
|
]
|
|
: [
|
|
Chat.create({
|
|
_id: newChatId ? new mongoose.Types.ObjectId(newChatId) : undefined,
|
|
userId,
|
|
modelId,
|
|
content,
|
|
title: content[0].value.slice(0, 20),
|
|
latestChat: content[1].value
|
|
}).then((res) => res._id)
|
|
]),
|
|
// update model
|
|
...(String(model.userId) === userId
|
|
? [
|
|
Model.findByIdAndUpdate(modelId, {
|
|
updateTime: new Date()
|
|
})
|
|
]
|
|
: [])
|
|
]);
|
|
|
|
return {
|
|
id
|
|
};
|
|
}
|