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 |
|