|
|
@@ -59,6 +59,24 @@ export class GeminiProApi implements LLMApi {
|
|
|
topP: modelConfig.top_p,
|
|
|
// "topK": modelConfig.top_k,
|
|
|
},
|
|
|
+ safetySettings: [
|
|
|
+ {
|
|
|
+ category: "HARM_CATEGORY_HARASSMENT",
|
|
|
+ threshold: "BLOCK_ONLY_HIGH",
|
|
|
+ },
|
|
|
+ {
|
|
|
+ category: "HARM_CATEGORY_HATE_SPEECH",
|
|
|
+ threshold: "BLOCK_ONLY_HIGH",
|
|
|
+ },
|
|
|
+ {
|
|
|
+ category: "HARM_CATEGORY_SEXUALLY_EXPLICIT",
|
|
|
+ threshold: "BLOCK_ONLY_HIGH",
|
|
|
+ },
|
|
|
+ {
|
|
|
+ category: "HARM_CATEGORY_DANGEROUS_CONTENT",
|
|
|
+ threshold: "BLOCK_ONLY_HIGH",
|
|
|
+ },
|
|
|
+ ],
|
|
|
};
|
|
|
|
|
|
console.log("[Request] google payload: ", requestPayload);
|