MythEclipse
47bac6ff8f
- Updated .env.example to include Groq API configuration. - Enhanced config schema in config.ts to support Groq API keys and settings. - Implemented Groq moderation API call in indonesianTextNormalizer.ts as a fallback for badword detection. - Removed the Indonesian slang normalization function and related tests to streamline moderation logic. - Updated tests to reflect changes in moderation strategy, focusing on emoji normalization and badword detection.
105 lines
3.6 KiB
TypeScript
105 lines
3.6 KiB
TypeScript
import { afterAll, afterEach, describe, expect, it } from "vitest";
|
|
import { config } from "../../src/config";
|
|
import {
|
|
buildModerationTextEvidence,
|
|
detectIndonesianBadwords,
|
|
formatModerationTextEvidenceForPrompt,
|
|
normalizeDiscordCustomEmoji,
|
|
} from "../../src/moderation/indonesianTextNormalizer";
|
|
|
|
const originalNemotronKey = config.NVIDIA_NEMOTRON_API_KEY;
|
|
const originalPrimaryAiKey = config.AI_LLM_API_KEY;
|
|
const originalGroqKey = config.GROQ_API_KEY;
|
|
|
|
function disableRemoteModeration(): void {
|
|
config.NVIDIA_NEMOTRON_API_KEY = undefined;
|
|
config.AI_LLM_API_KEY = undefined;
|
|
config.GROQ_API_KEY = undefined;
|
|
}
|
|
|
|
disableRemoteModeration();
|
|
|
|
afterEach(() => {
|
|
disableRemoteModeration();
|
|
});
|
|
|
|
afterAll(() => {
|
|
config.NVIDIA_NEMOTRON_API_KEY = originalNemotronKey;
|
|
config.AI_LLM_API_KEY = originalPrimaryAiKey;
|
|
config.GROQ_API_KEY = originalGroqKey;
|
|
});
|
|
|
|
describe("normalizeDiscordCustomEmoji", () => {
|
|
it("replaces static custom emoji", () => {
|
|
const result = normalizeDiscordCustomEmoji(
|
|
"Bersiaplah woy <:hadeh:1217434294281048185>",
|
|
);
|
|
expect(result.text).toBe("Bersiaplah woy [emoji:hadeh]");
|
|
expect(result.emojiNames).toContain("hadeh");
|
|
});
|
|
|
|
it("replaces animated custom emoji", () => {
|
|
const result = normalizeDiscordCustomEmoji("cek <a:speen:1234567890> dulu");
|
|
expect(result.text).toBe("cek [emoji:speen] dulu");
|
|
expect(result.emojiNames).toContain("speen");
|
|
});
|
|
|
|
it("handles text without emoji", () => {
|
|
const result = normalizeDiscordCustomEmoji("halo semua");
|
|
expect(result.text).toBe("halo semua");
|
|
expect(result.emojiNames).toHaveLength(0);
|
|
});
|
|
});
|
|
|
|
describe("detectIndonesianBadwords", () => {
|
|
it("detects known badword via local fallback", async () => {
|
|
const badwords = await detectIndonesianBadwords("kontol banget");
|
|
expect(badwords).toContain("kontol");
|
|
});
|
|
|
|
it("returns empty array for safe slang", async () => {
|
|
const badwords = await detectIndonesianBadwords("woy hadeh gua");
|
|
expect(badwords).toHaveLength(0);
|
|
});
|
|
});
|
|
|
|
describe("buildModerationTextEvidence", () => {
|
|
it("produces correct evidence for woy with emoji", async () => {
|
|
const evidence = await buildModerationTextEvidence(
|
|
"Bersiaplah woy <:hadeh:1217434294281048185>",
|
|
);
|
|
expect(evidence.normalized).toContain("[emoji:hadeh]");
|
|
// Slang normalization removed; only emoji normalization and badword detection remain
|
|
expect(evidence.notes.some((n) => n.includes("emoji:hadeh"))).toBe(true);
|
|
});
|
|
|
|
it("detects badword when present", async () => {
|
|
const evidence = await buildModerationTextEvidence("anjing loe kontol");
|
|
expect(evidence.hasBadwords).toBe(true);
|
|
expect(evidence.notes.some((n) => n.includes("badword detected"))).toBe(
|
|
true,
|
|
);
|
|
});
|
|
});
|
|
|
|
describe("formatModerationTextEvidenceForPrompt", () => {
|
|
it("returns prompt evidence for emoji", async () => {
|
|
const formatted = await formatModerationTextEvidenceForPrompt(
|
|
"Bersiaplah woy <:hadeh:1217434294281048185>",
|
|
);
|
|
expect(formatted).toContain("[normalized_text:");
|
|
expect(formatted).toContain("[emoji:hadeh]");
|
|
expect(formatted).toContain("[normalization_notes:");
|
|
// The NVIDIA API may or may not detect badwords for this input
|
|
expect(formatted).toMatch(
|
|
/no Indonesian badword detected|Indonesian badword detected/,
|
|
);
|
|
});
|
|
|
|
it("includes normalized text even for clean input", async () => {
|
|
const formatted = await formatModerationTextEvidenceForPrompt("Halo semua");
|
|
expect(formatted).toContain("[normalized_text:");
|
|
expect(formatted).toContain("no Indonesian badword detected");
|
|
});
|
|
});
|