更新
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* 迁移脚本:为 ai_models 表添加 frequency_penalty / presence_penalty 字段(已存在则跳过)
|
||||
* 用法: php database/migrate_add_penalty_fields.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,
|
||||
]);
|
||||
|
||||
$columns = [
|
||||
'frequency_penalty' => "ALTER TABLE ai_models ADD COLUMN frequency_penalty DECIMAL(3,2) DEFAULT 0.00 COMMENT '频率惩罚,抑制模型重复输出' AFTER support_image",
|
||||
'presence_penalty' => "ALTER TABLE ai_models ADD COLUMN presence_penalty DECIMAL(3,2) DEFAULT 0.00 COMMENT '存在惩罚,抑制模型重复话题' AFTER frequency_penalty",
|
||||
];
|
||||
|
||||
foreach ($columns as $name => $sql) {
|
||||
$exists = $pdo->query("SHOW COLUMNS FROM ai_models LIKE '{$name}'")->fetch();
|
||||
if ($exists) {
|
||||
echo "{$name} 字段已存在,跳过。\n";
|
||||
continue;
|
||||
}
|
||||
$pdo->exec($sql);
|
||||
echo "已添加 {$name} 字段。\n";
|
||||
}
|
||||
} catch (PDOException $e) {
|
||||
echo "Error: " . $e->getMessage() . "\n";
|
||||
exit(1);
|
||||
}
|
||||
Reference in New Issue
Block a user