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
1 changed files with 11 additions and 9 deletions

View File

@ -228,19 +228,21 @@ export class ChatGPTApi implements LLMApi {
} }
const text = msg.data; const text = msg.data;
try { try {
const json = JSON.parse(text) as { const json = JSON.parse(text);
choices: Array<{ const choices = json.choices as Array<{ delta: { content: string } }>;
delta: { const delta = choices[0]?.delta?.content;
content: string; const textmoderation = json?.prompt_filter_results;
};
}>;
};
const delta = json.choices[0]?.delta?.content;
if (delta) { if (delta) {
remainText += 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) { } catch (e) {
console.error("[Request] parse error", text); console.error("[Request] parse error", text, msg);
} }
}, },
onclose() { onclose() {