Merge pull request #136 from ccoder64/main

add safetySettings, set block_none
babaohuang · Jan 30, 2024 · 1d044a3 · 1d044a3
2 parents e5fff5a + 3d3447c
commit 1d044a3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/utils/openAI.ts b/src/utils/openAI.ts
@@ -18,6 +18,12 @@ export const startChatAndSendMessageStream = async(history: ChatMessage[], newMe
     generationConfig: {
       maxOutputTokens: 8000,
     },
+    safetySettings: [
+      {category: 'HARM_CATEGORY_HARASSMENT', threshold: 'BLOCK_NONE'},
+      {category: 'HARM_CATEGORY_HATE_SPEECH', threshold: 'BLOCK_NONE'},
+      {category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', threshold: 'BLOCK_NONE'},
+      {category: 'HARM_CATEGORY_DANGEROUS_CONTENT', threshold: 'BLOCK_NONE'}
+      ],
   })
 
   // Use sendMessageStream for streaming responses