Spaces:
Running
Running
Commit
·
fcf04d1
1
Parent(s):
5551656
updated safety settings
Browse files- src/manager/manager.py +19 -0
src/manager/manager.py
CHANGED
|
@@ -62,6 +62,24 @@ class GeminiManager:
|
|
| 62 |
self.system_prompt = f.read()
|
| 63 |
self.messages = []
|
| 64 |
self.set_modes(modes)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
|
| 66 |
def get_current_modes(self):
|
| 67 |
return [mode.name for mode in self.modes]
|
|
@@ -99,6 +117,7 @@ class GeminiManager:
|
|
| 99 |
system_instruction=self.system_prompt,
|
| 100 |
temperature=0.2,
|
| 101 |
tools=tools,
|
|
|
|
| 102 |
),
|
| 103 |
)
|
| 104 |
|
|
|
|
| 62 |
self.system_prompt = f.read()
|
| 63 |
self.messages = []
|
| 64 |
self.set_modes(modes)
|
| 65 |
+
self.safety_settings = [
|
| 66 |
+
{
|
| 67 |
+
"category": "HARM_CATEGORY_HARASSMENT",
|
| 68 |
+
"threshold": "BLOCK_NONE",
|
| 69 |
+
},
|
| 70 |
+
{
|
| 71 |
+
"category": "HARM_CATEGORY_HATE_SPEECH",
|
| 72 |
+
"threshold": "BLOCK_NONE",
|
| 73 |
+
},
|
| 74 |
+
{
|
| 75 |
+
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
|
| 76 |
+
"threshold": "BLOCK_NONE",
|
| 77 |
+
},
|
| 78 |
+
{
|
| 79 |
+
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
|
| 80 |
+
"threshold": "BLOCK_NONE",
|
| 81 |
+
},
|
| 82 |
+
]
|
| 83 |
|
| 84 |
def get_current_modes(self):
|
| 85 |
return [mode.name for mode in self.modes]
|
|
|
|
| 117 |
system_instruction=self.system_prompt,
|
| 118 |
temperature=0.2,
|
| 119 |
tools=tools,
|
| 120 |
+
safety_settings=self.safety_settings,
|
| 121 |
),
|
| 122 |
)
|
| 123 |
|