diff --git a/app/schemas/gemini_models.py b/app/schemas/gemini_models.py index 6fb9f0a..2addb13 100644 --- a/app/schemas/gemini_models.py +++ b/app/schemas/gemini_models.py @@ -4,7 +4,7 @@ from pydantic import BaseModel class SafetySetting(BaseModel): category: Optional[Literal[ - "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT"]] = None + "HARM_CATEGORY_HATE_SPEECH", "HARM_CATEGORY_DANGEROUS_CONTENT", "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_SEXUALLY_EXPLICIT", "HARM_CATEGORY_CIVIC_INTEGRITY"]] = None threshold: Optional[Literal[ "HARM_BLOCK_THRESHOLD_UNSPECIFIED", "BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE", "OFF"]] = None