<?php

/**
 * 迁移脚本：为 ai_models 表添加 support_context 字段（已存在则跳过）
 * 用法: php database/migrate_add_support_context.php
 */

$host = '127.0.0.1';
$user = 'root';
$pass = 'root';
$port = 3306;
$dbName = 'ai_chat';

try {
    $pdo = new PDO("mysql:host={$host};port={$port};dbname={$dbName};charset=utf8mb4", $user, $pass, [
        PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
    ]);

    $exists = $pdo->query("SHOW COLUMNS FROM ai_models LIKE 'support_context'")->fetch();

    if ($exists) {
        echo "support_context 字段已存在，跳过。\n";
    } else {
        $pdo->exec("ALTER TABLE ai_models ADD COLUMN support_context TINYINT(1) DEFAULT 1 COMMENT '是否支持上下文（多轮对话）' AFTER enabled");
        echo "已添加 support_context 字段。\n";
    }
} catch (PDOException $e) {
    echo "Error: " . $e->getMessage() . "\n";
    exit(1);
}