28 lines
901 B
TypeScript
28 lines
901 B
TypeScript
|
export class ContentSafetyError extends Error {
|
||
|
|
public readonly isSafetyRejection = true;
|
||
|
|
|
||
|
|
constructor(message: string) {
|
||
|
|
super(message);
|
||
|
|
this.name = "ContentSafetyError";
|
||
|
|
Object.setPrototypeOf(this, ContentSafetyError.prototype);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export function isContentSafetyError(err: any): boolean {
|
||
|
|
if (!err) return false;
|
||
|
|
if (err.isSafetyRejection || err.name === "ContentSafetyError") return true;
|
||
|
|
|
||
|
|
const msg = String(err.message || err.error || err || "").toLowerCase();
|
||
|
|
return (
|
||
|
|
msg.includes("safety filter") ||
|
||
|
|
msg.includes("prohibited_content") ||
|
||
|
|
msg.includes("harm_category") ||
|
||
|
|
msg.includes("safe-for-work") ||
|
||
|
|
msg.includes("finishreason: safety") ||
|
||
|
|
msg.includes("finishreason: blocklist") ||
|
||
|
|
msg.includes("blockreason") ||
|
||
|
|
msg.includes("content safety") ||
|
||
|
|
msg.includes("safety policy") ||
|
||
|
|
msg.includes("violates safety")
|
||
|
|
);
|
||
|
|
}
|