references/lobe-chat
2
0
Fork 0
mirror of https://github.com/lobehub/lobe-chat.git synced 2026-06-16 04:25:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix: support Gemini 2.0 HarmBlockThreshold (#5422)

Browse Source 
* Update index.ts

* Update index.ts
This commit is contained in:
sxjeru
2025-01-14 22:52:54 +08:00
committed by GitHub
parent 3cffc7e5fc
commit 4d1a4cd800
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/libs/agent-runtime/google/index.ts
+5 -4
View File
@@ -36,6 +36,7 @@ enum HarmCategory {
enum HarmBlockThreshold {
BLOCK_NONE = 'BLOCK_NONE',
OFF = 'OFF', // https://discuss.ai.google.dev/t/59352
}
export class LobeGoogleAI implements LobeRuntimeAI {
@@ -70,19 +71,19 @@ export class LobeGoogleAI implements LobeRuntimeAI {
safetySettings: [
{
category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
threshold: HarmBlockThreshold.BLOCK_NONE,
threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
},
{
category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
threshold: HarmBlockThreshold.BLOCK_NONE,
threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
},
{
category: HarmCategory.HARM_CATEGORY_HARASSMENT,
threshold: HarmBlockThreshold.BLOCK_NONE,
threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
},
{
category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
threshold: HarmBlockThreshold.BLOCK_NONE,
threshold: model.includes('2.0') ? (HarmBlockThreshold.OFF as any) : HarmBlockThreshold.BLOCK_NONE,
},
],
},