31 lines
957 B
Python
31 lines
957 B
Python
from google.genai import types
|
|
|
|
from .structures import OutputMessage
|
|
|
|
SAFETY_SETTINGS = [
|
|
types.SafetySetting(
|
|
category=category.value, threshold=types.HarmBlockThreshold.OFF.value
|
|
)
|
|
for category in types.HarmCategory
|
|
]
|
|
|
|
|
|
def generate_respond_config(prompt: str) -> types.GenerateContentConfig:
|
|
return
|
|
|
|
|
|
def generate_review_config(prompt: str) -> types.GenerateContentConfig:
|
|
return types.GenerateContentConfig(
|
|
system_instruction=prompt,
|
|
thinking_config=types.ThinkingConfig(thinking_budget=0),
|
|
response_mime_type="application/json",
|
|
response_schema=list[int],
|
|
# safety_settings=SAFETY_SETTINGS,
|
|
)
|
|
|
|
|
|
# MESSAGE_CONTENT_CONFIG = types.GenerateContentConfig(
|
|
# response_mime_type="application/json", # возможно можно скипнуть это если мы используем response_schema, надо проверить
|
|
# response_schema=list[OutputMessage],
|
|
# )
|