feat(bot): developing integration with solaris

This commit is contained in:
h
2025-07-07 01:26:03 +03:00
parent 1b52c7fdca
commit fbccb91630
16 changed files with 145 additions and 21 deletions

View File

@@ -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,
)