fix: Remove Safety Block

pot-app · Dec 20, 2023 · bda4d8a · bda4d8a
1 parent 16d1632
commit bda4d8a
Show file tree

Hide file tree

Showing 4 changed files with 72 additions and 0 deletions.
diff --git a/src/services/translate/geminipro/index.jsx b/src/services/translate/geminipro/index.jsx
@@ -35,6 +35,24 @@ export async function translate(text, from, to, options = {}) {
     };
     let body = {
         contents: promptList,
+        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',
+            },
+        ],
     };
 
     if (stream) {

diff --git a/src/services/translate/geminipro_custom/index.jsx b/src/services/translate/geminipro_custom/index.jsx
@@ -35,6 +35,24 @@ export async function translate(text, from, to, options = {}) {
     };
     let body = {
         contents: promptList,
+        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',
+            },
+        ],
     };
 
     if (stream) {

diff --git a/src/services/translate/geminipro_polish/index.jsx b/src/services/translate/geminipro_polish/index.jsx
@@ -35,6 +35,24 @@ export async function translate(text, from, to, options = {}) {
     };
     let body = {
         contents: promptList,
+        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',
+            },
+        ],
     };
 
     if (stream) {

diff --git a/src/services/translate/geminipro_summary/index.jsx b/src/services/translate/geminipro_summary/index.jsx
@@ -35,6 +35,24 @@ export async function translate(text, from, to, options = {}) {
     };
     let body = {
         contents: promptList,
+        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',
+            },
+        ],
     };
 
     if (stream) {