sion123
088bdb9a8e
- 添加账号管理详情页（基本信息、提示词、CapCut、参考图标签页） - 重构资产页面，按项目组分开展示图片/视频 - 聊天界面支持深度思考内容折叠展示、复制、删除消息 - 设置页面支持Agent配置（Anthropic/OpenAI协议）和工具配置 - 后端支持OpenAI兼容协议流式输出和DeepSeek思考模式 - 添加对话置顶/删除功能、数据库迁移、资产清单API - 添加账号参考图上传/删除、技能配置持久化、连接测试API
39 lines
1.2 KiB
TypeScript
39 lines
1.2 KiB
TypeScript
import Database from 'better-sqlite3';
|
|
import path from 'path';
|
|
import fs from 'fs';
|
|
import { fileURLToPath } from 'url';
|
|
import { SCHEMA_SQL } from './schema';
|
|
|
|
const __filename = fileURLToPath(import.meta.url);
|
|
const __dirname = path.dirname(__filename);
|
|
const DB_PATH = path.resolve(__dirname, '..', '..', 'data', 'meitu-agent.db');
|
|
|
|
let db: Database.Database;
|
|
|
|
export function getDb(): Database.Database {
|
|
if (!db) {
|
|
const dir = path.dirname(DB_PATH);
|
|
if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
|
|
db = new Database(DB_PATH);
|
|
db.pragma('journal_mode = WAL');
|
|
db.pragma('foreign_keys = ON');
|
|
}
|
|
return db;
|
|
}
|
|
|
|
// Ensure a column exists on a table, add it with DEFAULT if missing
|
|
function ensureColumn(d: Database.Database, table: string, column: string, definition: string) {
|
|
const cols = d.prepare(`PRAGMA table_info('${table}')`).all() as { name: string }[];
|
|
if (!cols.some((c) => c.name === column)) {
|
|
d.exec(`ALTER TABLE ${table} ADD COLUMN ${column} ${definition}`);
|
|
}
|
|
}
|
|
|
|
export function initDb(): void {
|
|
const d = getDb();
|
|
d.exec(SCHEMA_SQL);
|
|
|
|
// Migrations for existing databases
|
|
ensureColumn(d, 'conversations', 'pinned', 'INTEGER NOT NULL DEFAULT 0');
|
|
}
|