helloparthshah commited on
Commit
fcf04d1
·
1 Parent(s): 5551656

updated safety settings

Browse files
Files changed (1) hide show
  1. 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