content censor

This commit is contained in:
archer
2023-08-05 13:26:48 +08:00
parent 8d3ad943be
commit 236e7d3c3f
5 changed files with 23 additions and 9 deletions

View File

@@ -15,6 +15,7 @@ import { getChatModel } from '@/service/utils/data';
import { countModelPrice } from '@/service/events/pushBill';
import { ChatModelItemType } from '@/types/model';
import { UserModelSchema } from '@/types/mongoSchema';
import { textCensor } from '@/service/api/plugins';
export type ChatProps = {
res: NextApiResponse;
@@ -63,6 +64,14 @@ export const dispatchChatCompletion = async (props: Record<string, any>): Promis
model: modelConstantsData
});
await textCensor({
text: `${systemPrompt}
${quotePrompt}
${limitPrompt}
${userChatInput}
`
});
const { messages, filterMessages } = getChatMessages({
model: modelConstantsData,
history,