Safety Settings

from google.genai.types import HarmBlockThreshold, HarmCategory
from pydantic_ai import Agent
from pydantic_ai.models.google import GoogleModel, GoogleModelSettings
 
model = GoogleModel('gemini-3-flash-preview')
 
# 歴史的な暴力表現が混ざる教材で「危険コンテンツ」を緩める
edu_settings = GoogleModelSettings(
    google_safety_settings=[
        {
            'category': HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
            'threshold': HarmBlockThreshold.BLOCK_ONLY_HIGH,
        }
    ]
)
 
agent_edu = Agent(
    model,
    model_settings=edu_settings,
    instructions='高校生向けの歴史解説を、簡潔な日本語で書いてください。',
)
 
print(agent_edu.run_sync('第二次世界大戦の主要な戦闘を 3 つ挙げて、簡潔に説明して').output)

strict_settings = GoogleModelSettings(
    google_safety_settings=[
        {'category': cat, 'threshold': HarmBlockThreshold.BLOCK_LOW_AND_ABOVE}
        for cat in [
            HarmCategory.HARM_CATEGORY_HARASSMENT,
            HarmCategory.HARM_CATEGORY_HATE_SPEECH,
            HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
            HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
        ]
    ]
)
 
agent_strict = Agent(
    GoogleModel('gemini-3-flash-preview'),
    model_settings=strict_settings,
    instructions='社内ヘルプデスクとして丁寧な日本語で答えてください。',
)

medical_settings = GoogleModelSettings(
    google_safety_settings=[
        {
            'category': HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
            'threshold': HarmBlockThreshold.BLOCK_ONLY_HIGH,
        }
    ]
)
 
agent_med = Agent(
    GoogleModel('gemini-3-flash-preview'),
    model_settings=medical_settings,
    instructions=(
        '看護学生向けの教材として、医学的に正確で配慮のある日本語で答えてください。'
        '専門用語の解説を必ず含めてください。'
    ),
)
 
# 例: 解剖や生理に関する質問
print(agent_med.run_sync('女性の生殖器官の構造と機能を看護学生向けに解説してください。').output)