from anthropic import Anthropic

client = Anthropic()

# Без роли
response_no_role = client.messages.create(
    model="claude-opus-4-5",
    max_tokens=512,
    messages=[{"role": "user", "content": "Проверь эту SQL-схему на проблемы с производительностью"}]
)

# С ролью
response_with_role = client.messages.create(
    model="claude-opus-4-5",
    max_tokens=512,
    system="Ты — senior DBA с 15 годами опыта оптимизации PostgreSQL в высоконагруженных системах.",
    messages=[{"role": "user", "content": "Проверь эту SQL-схему на проблемы с производительностью"}]
)

# Разные роли — разные полезные перспективы на один код

roles = {
    "security": "Ты — специалист по информационной безопасности веб-приложений.",
    "performance": "Ты — инженер по производительности, оптимизировавший системы до 1M RPS.",
    "junior": "Ты — джун-разработчик, которому нужно понять этот код. Объясняй простыми словами.",
    "reviewer": "Ты — строгий code reviewer, который не пропускает даже мелкие проблемы.",
}

code = """
def get_user(user_id):
    query = f"SELECT * FROM users WHERE id = {user_id}"
    return db.execute(query).fetchone()
"""

for role_name, role_prompt in roles.items():
    response = client.messages.create(
        model="claude-opus-4-5",
        max_tokens=256,
        system=role_prompt,
        messages=[{"role": "user", "content": f"Посмотри на этот код:\n{code}"}]
    )
    print(f"\n[{role_name}]: {response.content[0].text[:200]}...")

# security найдёт SQL injection
# performance заметит SELECT *
# junior попросит объяснений
# reviewer найдёт всё сразу

# Это почти ничего не меняет
system_weak = "Ты — эксперт мирового класса с огромным опытом."

# Это работает — конкретная специализация и контекст
system_strong = """Ты — backend-разработчик, специализирующийся на Node.js и PostgreSQL.
Работаешь в финтех-компании с требованиями к надёжности 99.99%.
Твои ревью строгие: ты не пропускаешь код без обработки ошибок и транзакций."""

# Только роль — слабо
system = "Ты — эксперт по безопасности."

# Роль + что именно делать + как отвечать — сильно
system = """Ты — специалист по безопасности веб-приложений.

При ревью кода ищи:
- Инъекции (SQL, NoSQL, команды)
- Небезопасная работа с данными пользователя
- Отсутствие валидации и санитизации
- Проблемы с аутентификацией и авторизацией

Для каждой проблемы указывай: уровень критичности (HIGH/MEDIUM/LOW), строку, объяснение и исправление."""