feat(bot): developing integration with solaris
This commit is contained in:
@@ -3,7 +3,9 @@ from google.genai import types
|
||||
from .structures import OutputMessage
|
||||
|
||||
SAFETY_SETTINGS = [
|
||||
types.SafetySetting(category=category, threshold=types.HarmBlockThreshold.OFF)
|
||||
types.SafetySetting(
|
||||
category=category.value, threshold=types.HarmBlockThreshold.OFF.value
|
||||
)
|
||||
for category in types.HarmCategory
|
||||
]
|
||||
|
||||
@@ -18,7 +20,7 @@ def generate_review_config(prompt: str) -> types.GenerateContentConfig:
|
||||
thinking_config=types.ThinkingConfig(thinking_budget=0),
|
||||
response_mime_type="application/json",
|
||||
response_schema=list[int],
|
||||
safety_settings=SAFETY_SETTINGS,
|
||||
# safety_settings=SAFETY_SETTINGS,
|
||||
)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user