Merge pull request #3314 from H0llyW00dzZ/text-moderation-azure

Feat ChatGPT LLM Api [Console Log] [Text Moderation] [Azure]
This commit is contained in:
DeanYao 2024-03-28 13:38:56 +08:00 committed by GitHub
commit 6b3daec23f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -228,19 +228,21 @@ export class ChatGPTApi implements LLMApi {
}
const text = msg.data;
try {
const json = JSON.parse(text) as {
choices: Array<{
delta: {
content: string;
};
}>;
};
const delta = json.choices[0]?.delta?.content;
const json = JSON.parse(text);
const choices = json.choices as Array<{ delta: { content: string } }>;
const delta = choices[0]?.delta?.content;
const textmoderation = json?.prompt_filter_results;
if (delta) {
remainText += delta;
}
if (textmoderation && textmoderation.length > 0 && ServiceProvider.Azure) {
const contentFilterResults = textmoderation[0]?.content_filter_results;
console.log(`[${ServiceProvider.Azure}] [Text Moderation] flagged categories result:`, contentFilterResults);
}
} catch (e) {
console.error("[Request] parse error", text);
console.error("[Request] parse error", text, msg);
}
},
onclose() {