@huyooo/ai-chat-storage 0.1.6 → 0.1.8

package/dist/{chunk-BRWTH4LQ.js → chunk-7AKL7W3Y.js}

@@ -28,9 +28,11 @@ async function initSchema(sql) {
       user_id TEXT,
       role TEXT NOT NULL,
       content TEXT NOT NULL,
-      thinking TEXT,
-      tool_calls TEXT,
-      search_results TEXT,
+      model TEXT,
+      mode TEXT,
+      web_search_enabled BOOLEAN,
+      thinking_enabled BOOLEAN,
+      steps TEXT,
       operation_ids TEXT,
       timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()
     )
@@ -75,6 +77,24 @@ async function initSchema(sql) {
       auto_delete_at TIMESTAMPTZ NOT NULL
     )
   `;
+  await sql`
+    CREATE TABLE IF NOT EXISTS user_settings (
+      id TEXT PRIMARY KEY,
+      app_id TEXT,
+      user_id TEXT,
+      key TEXT NOT NULL,
+      value TEXT NOT NULL,
+      created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+      updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+      UNIQUE(app_id, user_id, key)
+    )
+  `;
+  await sql`
+    CREATE INDEX IF NOT EXISTS idx_user_settings_tenant ON user_settings(app_id, user_id);
+  `;
+  await sql`
+    CREATE INDEX IF NOT EXISTS idx_user_settings_key ON user_settings(key);
+  `;
   try {
     await sql`
       CREATE TABLE IF NOT EXISTS embeddings (
@@ -146,13 +166,16 @@ function toSessionRecord(row) {
     title: row.title,
     model: row.model,
     mode: row.mode,
+    webSearchEnabled: row.web_search_enabled,
+    thinkingEnabled: row.thinking_enabled,
+    hidden: row.hidden,
     createdAt: new Date(row.created_at),
     updatedAt: new Date(row.updated_at)
   };
 }
 async function getSessions(sql, ctx) {
   const rows = await sql`
-    SELECT id, app_id, user_id, title, model, mode, created_at, updated_at
+    SELECT id, app_id, user_id, title, model, mode, web_search_enabled, thinking_enabled, hidden, created_at, updated_at
     FROM sessions
     WHERE (${ctx.appId || null}::TEXT IS NULL OR app_id = ${ctx.appId || null})
       AND (${ctx.userId || null}::TEXT IS NULL OR user_id = ${ctx.userId || null})
@@ -162,7 +185,7 @@ async function getSessions(sql, ctx) {
 }
 async function getSession(sql, id, ctx) {
   const rows = await sql`
-    SELECT id, app_id, user_id, title, model, mode, created_at, updated_at
+    SELECT id, app_id, user_id, title, model, mode, web_search_enabled, thinking_enabled, hidden, created_at, updated_at
     FROM sessions
     WHERE id = ${id}
       AND (${ctx.appId || null}::TEXT IS NULL OR app_id = ${ctx.appId || null})
@@ -172,9 +195,10 @@ async function getSession(sql, id, ctx) {
 }
 async function createSession(sql, input, ctx) {
   const now = /* @__PURE__ */ new Date();
+  const hidden = input.hidden ?? false;
   await sql`
-    INSERT INTO sessions (id, app_id, user_id, title, model, mode, created_at, updated_at)
-    VALUES (${input.id}, ${ctx.appId || null}, ${ctx.userId || null}, ${input.title}, ${input.model}, ${input.mode}, ${now}, ${now})
+    INSERT INTO sessions (id, app_id, user_id, title, model, mode, web_search_enabled, thinking_enabled, hidden, created_at, updated_at)
+    VALUES (${input.id}, ${ctx.appId || null}, ${ctx.userId || null}, ${input.title}, ${input.model}, ${input.mode}, ${input.webSearchEnabled}, ${input.thinkingEnabled}, ${hidden}, ${now}, ${now})
   `;
   return {
     id: input.id,
@@ -183,6 +207,9 @@ async function createSession(sql, input, ctx) {
     title: input.title,
     model: input.model,
     mode: input.mode,
+    webSearchEnabled: input.webSearchEnabled,
+    thinkingEnabled: input.thinkingEnabled,
+    hidden,
     createdAt: now,
     updatedAt: now
   };
@@ -192,9 +219,12 @@ async function updateSession(sql, id, data, ctx) {
   await sql`
     UPDATE sessions
     SET updated_at = ${now},
-        title = COALESCE(${data.title || null}, title),
-        model = COALESCE(${data.model || null}, model),
-        mode = COALESCE(${data.mode || null}, mode)
+        title = COALESCE(${data.title ?? null}, title),
+        model = COALESCE(${data.model ?? null}, model),
+        mode = COALESCE(${data.mode ?? null}, mode),
+        web_search_enabled = COALESCE(${data.webSearchEnabled ?? null}, web_search_enabled),
+        thinking_enabled = COALESCE(${data.thinkingEnabled ?? null}, thinking_enabled),
+        hidden = COALESCE(${data.hidden ?? null}, hidden)
     WHERE id = ${id}
       AND (${ctx.appId || null}::TEXT IS NULL OR app_id = ${ctx.appId || null})
       AND (${ctx.userId || null}::TEXT IS NULL OR user_id = ${ctx.userId || null})
@@ -218,17 +248,19 @@ function toMessageRecord(row) {
     userId: row.user_id,
     role: row.role,
     content: row.content,
-    thinking: row.thinking,
-    toolCalls: row.tool_calls,
-    searchResults: row.search_results,
+    model: row.model,
+    mode: row.mode,
+    webSearchEnabled: row.web_search_enabled,
+    thinkingEnabled: row.thinking_enabled,
+    steps: row.steps,
     operationIds: row.operation_ids,
     timestamp: new Date(row.timestamp)
   };
 }
 async function getMessages(sql, sessionId) {
   const rows = await sql`
-    SELECT id, session_id, app_id, user_id, role, content, thinking,
-           tool_calls, search_results, operation_ids, timestamp
+    SELECT id, session_id, app_id, user_id, role, content, model, mode,
+           web_search_enabled, thinking_enabled, steps, operation_ids, timestamp
     FROM messages
     WHERE session_id = ${sessionId}
     ORDER BY timestamp
@@ -237,8 +269,8 @@ async function getMessages(sql, sessionId) {
 }
 async function getMessage(sql, id, ctx) {
   const rows = await sql`
-    SELECT id, session_id, app_id, user_id, role, content, thinking,
-           tool_calls, search_results, operation_ids, timestamp
+    SELECT id, session_id, app_id, user_id, role, content, model, mode,
+           web_search_enabled, thinking_enabled, steps, operation_ids, timestamp
     FROM messages
     WHERE id = ${id}
       AND (${ctx.appId || null}::TEXT IS NULL OR app_id = ${ctx.appId || null})
@@ -249,8 +281,8 @@ async function getMessage(sql, id, ctx) {
 async function saveMessage(sql, input, ctx) {
   const now = /* @__PURE__ */ new Date();
   await sql`
-    INSERT INTO messages (id, session_id, app_id, user_id, role, content, thinking, tool_calls, search_results, operation_ids, timestamp)
-    VALUES (${input.id}, ${input.sessionId}, ${ctx.appId || null}, ${ctx.userId || null}, ${input.role}, ${input.content}, ${input.thinking}, ${input.toolCalls}, ${input.searchResults}, ${input.operationIds}, ${now})
+    INSERT INTO messages (id, session_id, app_id, user_id, role, content, model, mode, web_search_enabled, thinking_enabled, steps, operation_ids, timestamp)
+    VALUES (${input.id}, ${input.sessionId}, ${ctx.appId || null}, ${ctx.userId || null}, ${input.role}, ${input.content}, ${input.model}, ${input.mode}, ${input.webSearchEnabled}, ${input.thinkingEnabled}, ${input.steps}, ${input.operationIds}, ${now})
   `;
   await sql`UPDATE sessions SET updated_at = ${now} WHERE id = ${input.sessionId}`;
   return {
@@ -260,13 +292,50 @@ async function saveMessage(sql, input, ctx) {
     userId: ctx.userId || null,
     role: input.role,
     content: input.content,
-    thinking: input.thinking,
-    toolCalls: input.toolCalls,
-    searchResults: input.searchResults,
+    model: input.model,
+    mode: input.mode,
+    webSearchEnabled: input.webSearchEnabled,
+    thinkingEnabled: input.thinkingEnabled,
+    steps: input.steps,
     operationIds: input.operationIds,
     timestamp: now
   };
 }
+async function updateMessage(sql, id, data, ctx) {
+  const updates = [];
+  if (data.content !== void 0) {
+    updates.push("content");
+  }
+  if (data.steps !== void 0) {
+    updates.push("steps");
+  }
+  if (updates.length === 0) return;
+  if (data.content !== void 0 && data.steps !== void 0) {
+    await sql`
+      UPDATE messages
+      SET content = ${data.content}, steps = ${data.steps}
+      WHERE id = ${id}
+        AND (${ctx.appId || null}::TEXT IS NULL OR app_id = ${ctx.appId || null})
+        AND (${ctx.userId || null}::TEXT IS NULL OR user_id = ${ctx.userId || null})
+    `;
+  } else if (data.content !== void 0) {
+    await sql`
+      UPDATE messages
+      SET content = ${data.content}
+      WHERE id = ${id}
+        AND (${ctx.appId || null}::TEXT IS NULL OR app_id = ${ctx.appId || null})
+        AND (${ctx.userId || null}::TEXT IS NULL OR user_id = ${ctx.userId || null})
+    `;
+  } else if (data.steps !== void 0) {
+    await sql`
+      UPDATE messages
+      SET steps = ${data.steps}
+      WHERE id = ${id}
+        AND (${ctx.appId || null}::TEXT IS NULL OR app_id = ${ctx.appId || null})
+        AND (${ctx.userId || null}::TEXT IS NULL OR user_id = ${ctx.userId || null})
+    `;
+  }
+}
 async function deleteMessagesAfter(sql, sessionId, timestamp) {
   await sql`
     DELETE FROM messages
@@ -567,6 +636,10 @@ var PostgresAdapter = class {
     await this.ensureInitialized();
     return saveMessage(this.sql, input, ctx);
   }
+  async updateMessage(id, data, ctx) {
+    await this.ensureInitialized();
+    await updateMessage(this.sql, id, data, ctx);
+  }
   async deleteMessagesAfter(sessionId, timestamp, ctx) {
     await this.ensureInitialized();
     const session = await this.getSession(sessionId, ctx);
@@ -631,6 +704,50 @@ var PostgresAdapter = class {
     await this.ensureInitialized();
     return searchSimilar(this.sql, queryEmbedding, options, ctx);
   }
+  // ============ 用户设置 ============
+  async getUserSetting(key, ctx) {
+    await this.ensureInitialized();
+    const id = `${ctx.appId || "default"}_${ctx.userId || "default"}_${key}`;
+    const result = await this.sql`
+      SELECT value FROM user_settings
+      WHERE id = ${id}
+      LIMIT 1
+    `;
+    return result.length > 0 ? result[0].value : null;
+  }
+  async setUserSetting(key, value, ctx) {
+    await this.ensureInitialized();
+    const id = `${ctx.appId || "default"}_${ctx.userId || "default"}_${key}`;
+    await this.sql`
+      INSERT INTO user_settings (id, app_id, user_id, key, value, created_at, updated_at)
+      VALUES (${id}, ${ctx.appId || null}, ${ctx.userId || null}, ${key}, ${value}, NOW(), NOW())
+      ON CONFLICT (id) DO UPDATE SET
+        value = ${value},
+        updated_at = NOW()
+    `;
+  }
+  async getUserSettings(ctx) {
+    await this.ensureInitialized();
+    const results = await this.sql`
+      SELECT key, value FROM user_settings
+      WHERE app_id IS NOT DISTINCT FROM ${ctx.appId || null}
+        AND user_id IS NOT DISTINCT FROM ${ctx.userId || null}
+    `;
+    const settings = {};
+    for (const row of results) {
+      settings[row.key] = row.value;
+    }
+    return settings;
+  }
+  async deleteUserSetting(key, ctx) {
+    await this.ensureInitialized();
+    await this.sql`
+      DELETE FROM user_settings
+      WHERE key = ${key}
+        AND app_id IS NOT DISTINCT FROM ${ctx.appId || null}
+        AND user_id IS NOT DISTINCT FROM ${ctx.userId || null}
+    `;
+  }
   // ============ 生命周期 ============
   async close() {
     await this.sql.end();

package/dist/{chunk-GCKDCC3O.js → chunk-LZ2LBU6Q.js}

@@ -11,6 +11,10 @@ var sessions = sqliteTable("sessions", {
   title: text("title").notNull(),
   model: text("model").notNull(),
   mode: text("mode").notNull(),
+  webSearchEnabled: integer("web_search_enabled", { mode: "boolean" }).notNull().default(true),
+  thinkingEnabled: integer("thinking_enabled", { mode: "boolean" }).notNull().default(true),
+  /** 是否在 tab 栏隐藏（关闭但不删除） */
+  hidden: integer("hidden", { mode: "boolean" }).notNull().default(false),
   createdAt: integer("created_at", { mode: "timestamp" }).notNull(),
   updatedAt: integer("updated_at", { mode: "timestamp" }).notNull()
 }, (table) => [
@@ -24,9 +28,16 @@ var messages = sqliteTable("messages", {
   userId: text("user_id"),
   role: text("role").notNull(),
   content: text("content").notNull(),
-  thinking: text("thinking"),
-  toolCalls: text("tool_calls"),
-  searchResults: text("search_results"),
+  /** 生成此消息时使用的模型 */
+  model: text("model"),
+  /** 生成此消息时使用的模式 (ask/agent) */
+  mode: text("mode"),
+  /** 生成此消息时是否启用 web 搜索 */
+  webSearchEnabled: integer("web_search_enabled", { mode: "boolean" }),
+  /** 生成此消息时是否启用深度思考 */
+  thinkingEnabled: integer("thinking_enabled", { mode: "boolean" }),
+  /** 执行步骤列表 JSON */
+  steps: text("steps"),
   operationIds: text("operation_ids"),
   timestamp: integer("timestamp", { mode: "timestamp" }).notNull()
 }, (table) => [
@@ -105,6 +116,9 @@ function initSchema(sqlite) {
       title TEXT NOT NULL,
       model TEXT NOT NULL,
       mode TEXT NOT NULL,
+      web_search_enabled INTEGER NOT NULL DEFAULT 1,
+      thinking_enabled INTEGER NOT NULL DEFAULT 1,
+      hidden INTEGER NOT NULL DEFAULT 0,
       created_at INTEGER NOT NULL,
       updated_at INTEGER NOT NULL
     );
@@ -116,9 +130,11 @@ function initSchema(sqlite) {
       user_id TEXT,
       role TEXT NOT NULL,
       content TEXT NOT NULL,
-      thinking TEXT,
-      tool_calls TEXT,
-      search_results TEXT,
+      model TEXT,
+      mode TEXT,
+      web_search_enabled INTEGER,
+      thinking_enabled INTEGER,
+      steps TEXT,
       operation_ids TEXT,
       timestamp INTEGER NOT NULL
     );
@@ -173,6 +189,17 @@ function initSchema(sqlite) {
       created_at INTEGER NOT NULL
     );
 
+    CREATE TABLE IF NOT EXISTS user_settings (
+      id TEXT PRIMARY KEY,
+      app_id TEXT,
+      user_id TEXT,
+      key TEXT NOT NULL,
+      value TEXT NOT NULL,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL,
+      UNIQUE(app_id, user_id, key)
+    );
+
     -- \u7D22\u5F15
     CREATE INDEX IF NOT EXISTS idx_sessions_tenant ON sessions(app_id, user_id);
     CREATE INDEX IF NOT EXISTS idx_sessions_updated ON sessions(updated_at);
@@ -185,7 +212,41 @@ function initSchema(sqlite) {
     CREATE INDEX IF NOT EXISTS idx_trash_tenant ON trash(app_id, user_id);
     CREATE INDEX IF NOT EXISTS idx_trash_auto_delete ON trash(auto_delete_at);
     CREATE INDEX IF NOT EXISTS idx_embeddings_session ON embeddings(session_id);
+    CREATE INDEX IF NOT EXISTS idx_user_settings_tenant ON user_settings(app_id, user_id);
+    CREATE INDEX IF NOT EXISTS idx_user_settings_key ON user_settings(key);
   `);
+  try {
+    sqlite.exec(`ALTER TABLE sessions ADD COLUMN web_search_enabled INTEGER NOT NULL DEFAULT 1`);
+  } catch {
+  }
+  try {
+    sqlite.exec(`ALTER TABLE sessions ADD COLUMN thinking_enabled INTEGER NOT NULL DEFAULT 1`);
+  } catch {
+  }
+  try {
+    sqlite.exec(`ALTER TABLE messages ADD COLUMN model TEXT`);
+  } catch {
+  }
+  try {
+    sqlite.exec(`ALTER TABLE messages ADD COLUMN mode TEXT`);
+  } catch {
+  }
+  try {
+    sqlite.exec(`ALTER TABLE messages ADD COLUMN web_search_enabled INTEGER`);
+  } catch {
+  }
+  try {
+    sqlite.exec(`ALTER TABLE messages ADD COLUMN thinking_enabled INTEGER`);
+  } catch {
+  }
+  try {
+    sqlite.exec(`ALTER TABLE sessions ADD COLUMN hidden INTEGER NOT NULL DEFAULT 0`);
+  } catch {
+  }
+  try {
+    sqlite.exec(`ALTER TABLE messages ADD COLUMN steps TEXT`);
+  } catch {
+  }
 }
 function buildSessionsTenantCondition(ctx) {
   const conditions = [];
@@ -235,6 +296,9 @@ function toSessionRecord(row) {
     title: row.title,
     model: row.model,
     mode: row.mode,
+    webSearchEnabled: row.webSearchEnabled,
+    thinkingEnabled: row.thinkingEnabled,
+    hidden: row.hidden,
     createdAt: row.createdAt,
     updatedAt: row.updatedAt
   };
@@ -259,6 +323,9 @@ function createSession(db, input, ctx) {
     title: input.title,
     model: input.model,
     mode: input.mode,
+    webSearchEnabled: input.webSearchEnabled,
+    thinkingEnabled: input.thinkingEnabled,
+    hidden: input.hidden ?? false,
     createdAt: now,
     updatedAt: now
   };
@@ -269,6 +336,9 @@ function createSession(db, input, ctx) {
     title: record.title,
     model: record.model,
     mode: record.mode,
+    webSearchEnabled: record.webSearchEnabled,
+    thinkingEnabled: record.thinkingEnabled,
+    hidden: record.hidden,
     createdAt: record.createdAt,
     updatedAt: record.updatedAt
   }).run();
@@ -281,6 +351,9 @@ function updateSession(db, id, data, ctx) {
   if (data.title !== void 0) updateData.title = data.title;
   if (data.model !== void 0) updateData.model = data.model;
   if (data.mode !== void 0) updateData.mode = data.mode;
+  if (data.webSearchEnabled !== void 0) updateData.webSearchEnabled = data.webSearchEnabled;
+  if (data.thinkingEnabled !== void 0) updateData.thinkingEnabled = data.thinkingEnabled;
+  if (data.hidden !== void 0) updateData.hidden = data.hidden;
   db.update(sessions).set(updateData).where(condition).run();
 }
 function deleteSession(db, id, ctx) {
@@ -299,9 +372,11 @@ function toMessageRecord(row) {
     userId: row.userId,
     role: row.role,
     content: row.content,
-    thinking: row.thinking,
-    toolCalls: row.toolCalls,
-    searchResults: row.searchResults,
+    model: row.model,
+    mode: row.mode,
+    webSearchEnabled: row.webSearchEnabled,
+    thinkingEnabled: row.thinkingEnabled,
+    steps: row.steps,
     operationIds: row.operationIds,
     timestamp: row.timestamp
   };
@@ -325,9 +400,11 @@ function saveMessage(db, input, ctx) {
     userId: ctx.userId || null,
     role: input.role,
     content: input.content,
-    thinking: input.thinking,
-    toolCalls: input.toolCalls,
-    searchResults: input.searchResults,
+    model: input.model,
+    mode: input.mode,
+    webSearchEnabled: input.webSearchEnabled,
+    thinkingEnabled: input.thinkingEnabled,
+    steps: input.steps,
     operationIds: input.operationIds,
     timestamp: now
   };
@@ -338,15 +415,31 @@ function saveMessage(db, input, ctx) {
     userId: record.userId,
     role: record.role,
     content: record.content,
-    thinking: record.thinking,
-    toolCalls: record.toolCalls,
-    searchResults: record.searchResults,
+    model: record.model,
+    mode: record.mode,
+    webSearchEnabled: record.webSearchEnabled,
+    thinkingEnabled: record.thinkingEnabled,
+    steps: record.steps,
     operationIds: record.operationIds,
     timestamp: record.timestamp
   }).run();
   db.update(sessions).set({ updatedAt: now }).where(eq3(sessions.id, input.sessionId)).run();
   return record;
 }
+function updateMessage(db, id, data, ctx) {
+  const tenantCondition = buildMessagesTenantCondition(ctx);
+  const condition = tenantCondition ? and3(eq3(messages.id, id), tenantCondition) : eq3(messages.id, id);
+  const updateData = {};
+  if (data.content !== void 0) {
+    updateData.content = data.content;
+  }
+  if (data.steps !== void 0) {
+    updateData.steps = data.steps;
+  }
+  if (Object.keys(updateData).length > 0) {
+    db.update(messages).set(updateData).where(condition).run();
+  }
+}
 function deleteMessagesAfter(db, sessionId, timestamp) {
   db.delete(messages).where(and3(
     eq3(messages.sessionId, sessionId),
@@ -569,11 +662,60 @@ function deleteSessionEmbeddings(db, sessionId) {
   db.delete(embeddings).where(eq7(embeddings.sessionId, sessionId)).run();
 }
 
+// src/adapters/sqlite/user-settings.ts
+function getUserSetting(sqlite, key, ctx) {
+  const stmt = sqlite.prepare(`
+    SELECT value FROM user_settings
+    WHERE key = ? AND app_id IS ? AND user_id IS ?
+    LIMIT 1
+  `);
+  const result = stmt.get(key, ctx.appId || null, ctx.userId || null);
+  return result?.value || null;
+}
+function setUserSetting(sqlite, key, value, ctx) {
+  const now = Date.now();
+  const id = `${ctx.appId || "default"}_${ctx.userId || "default"}_${key}`;
+  const checkStmt = sqlite.prepare(`
+    SELECT created_at FROM user_settings WHERE id = ?
+  `);
+  const existing = checkStmt.get(id);
+  const createdAt = existing?.created_at || now;
+  const stmt = sqlite.prepare(`
+    INSERT OR REPLACE INTO user_settings 
+    (id, app_id, user_id, key, value, created_at, updated_at)
+    VALUES (?, ?, ?, ?, ?, ?, ?)
+  `);
+  stmt.run(id, ctx.appId || null, ctx.userId || null, key, value, createdAt, now);
+}
+function getUserSettings(sqlite, ctx) {
+  const stmt = sqlite.prepare(`
+    SELECT key, value FROM user_settings
+    WHERE app_id IS ? AND user_id IS ?
+  `);
+  const results = stmt.all(ctx.appId || null, ctx.userId || null);
+  const settings = {};
+  for (const row of results) {
+    settings[row.key] = row.value;
+  }
+  return settings;
+}
+function deleteUserSetting(sqlite, key, ctx) {
+  const stmt = sqlite.prepare(`
+    DELETE FROM user_settings
+    WHERE key = ? AND app_id IS ? AND user_id IS ?
+  `);
+  stmt.run(key, ctx.appId || null, ctx.userId || null);
+}
+
 // src/adapters/sqlite/index.ts
 var SqliteAdapter = class {
   sqlite;
   db;
   config;
+  // 暴露 sqlite 实例供 user-settings 使用
+  get sqliteInstance() {
+    return this.sqlite;
+  }
   constructor(dbPath, config = { type: "sqlite" }) {
     this.sqlite = new Database(dbPath);
     this.db = drizzle(this.sqlite);
@@ -611,6 +753,9 @@ var SqliteAdapter = class {
   async saveMessage(input, ctx) {
     return saveMessage(this.db, input, ctx);
   }
+  async updateMessage(id, data, ctx) {
+    updateMessage(this.db, id, data, ctx);
+  }
   async deleteMessagesAfter(sessionId, timestamp, ctx) {
     const session = await this.getSession(sessionId, ctx);
     if (!session) return;
@@ -661,6 +806,19 @@ var SqliteAdapter = class {
   async searchSimilar(queryEmbedding, options, ctx) {
     return searchSimilar(this.db, queryEmbedding, options, ctx);
   }
+  // ============ 用户设置 ============
+  async getUserSetting(key, ctx) {
+    return getUserSetting(this.sqlite, key, ctx);
+  }
+  async setUserSetting(key, value, ctx) {
+    return setUserSetting(this.sqlite, key, value, ctx);
+  }
+  async getUserSettings(ctx) {
+    return getUserSettings(this.sqlite, ctx);
+  }
+  async deleteUserSetting(key, ctx) {
+    return deleteUserSetting(this.sqlite, key, ctx);
+  }
   // ============ 生命周期 ============
   async close() {
     this.sqlite.close();

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 import * as drizzle_orm_sqlite_core from 'drizzle-orm/sqlite-core';
+import Database from 'better-sqlite3';
 
 /**
  * 存储层类型定义
@@ -16,11 +17,15 @@ interface SessionRecord {
     title: string;
     model: string;
     mode: 'agent' | 'plan' | 'ask';
+    webSearchEnabled: boolean;
+    thinkingEnabled: boolean;
+    /** 是否在 tab 栏隐藏（关闭但不删除） */
+    hidden: boolean;
     createdAt: Date;
     updatedAt: Date;
 }
 type CreateSessionInput = Omit<SessionRecord, 'createdAt' | 'updatedAt' | 'appId' | 'userId'>;
-type UpdateSessionInput = Partial<Pick<SessionRecord, 'title' | 'model' | 'mode'>>;
+type UpdateSessionInput = Partial<Pick<SessionRecord, 'title' | 'model' | 'mode' | 'webSearchEnabled' | 'thinkingEnabled' | 'hidden'>>;
 interface MessageRecord {
     id: string;
     sessionId: string;
@@ -28,13 +33,22 @@ interface MessageRecord {
     userId: string | null;
     role: 'user' | 'assistant';
     content: string;
-    thinking: string | null;
-    toolCalls: string | null;
-    searchResults: string | null;
+    /** 生成此消息时使用的模型 */
+    model: string | null;
+    /** 生成此消息时使用的模式 (ask/agent) */
+    mode: string | null;
+    /** 生成此消息时是否启用 web 搜索 */
+    webSearchEnabled: boolean | null;
+    /** 生成此消息时是否启用深度思考 */
+    thinkingEnabled: boolean | null;
+    /** 执行步骤列表 JSON */
+    steps: string | null;
     operationIds: string | null;
     timestamp: Date;
 }
 type CreateMessageInput = Omit<MessageRecord, 'timestamp' | 'appId' | 'userId'>;
+/** 更新消息输入（只更新 content 和 steps） */
+type UpdateMessageInput = Partial<Pick<MessageRecord, 'content' | 'steps'>>;
 type OperationStatus = 'pending' | 'confirmed' | 'executed' | 'reverted' | 'failed';
 interface OperationRecord {
     id: string;
@@ -106,6 +120,8 @@ interface StorageAdapter {
     getMessages(sessionId: string, ctx: StorageContext): Promise<MessageRecord[]>;
     getMessage(id: string, ctx: StorageContext): Promise<MessageRecord | null>;
     saveMessage(message: CreateMessageInput, ctx: StorageContext): Promise<MessageRecord>;
+    /** 更新消息（用于增量保存助手消息） */
+    updateMessage(id: string, data: UpdateMessageInput, ctx: StorageContext): Promise<void>;
     deleteMessagesAfter(sessionId: string, timestamp: Date, ctx: StorageContext): Promise<void>;
     getOperations(sessionId: string, ctx: StorageContext): Promise<OperationRecord[]>;
     getOperationsByMessage(messageId: string, ctx: StorageContext): Promise<OperationRecord[]>;
@@ -118,6 +134,10 @@ interface StorageAdapter {
     moveToTrash?(item: Omit<TrashRecord, 'deletedAt' | 'autoDeleteAt' | 'appId' | 'userId'>, ctx: StorageContext): Promise<TrashRecord>;
     restoreFromTrash?(id: string, ctx: StorageContext): Promise<TrashRecord>;
     emptyExpiredTrash?(): Promise<number>;
+    getUserSetting(key: string, ctx: StorageContext): Promise<string | null>;
+    setUserSetting(key: string, value: string, ctx: StorageContext): Promise<void>;
+    getUserSettings(ctx: StorageContext): Promise<Record<string, string>>;
+    deleteUserSetting(key: string, ctx: StorageContext): Promise<void>;
     saveEmbedding?(id: string, content: string, embedding: number[], metadata: {
         sessionId: string;
         messageId?: string;
@@ -257,6 +277,57 @@ declare const sessions: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
+        webSearchEnabled: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "web_search_enabled";
+            tableName: "sessions";
+            dataType: "boolean";
+            columnType: "SQLiteBoolean";
+            data: boolean;
+            driverParam: number;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        thinkingEnabled: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "thinking_enabled";
+            tableName: "sessions";
+            dataType: "boolean";
+            columnType: "SQLiteBoolean";
+            data: boolean;
+            driverParam: number;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        hidden: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "hidden";
+            tableName: "sessions";
+            dataType: "boolean";
+            columnType: "SQLiteBoolean";
+            data: boolean;
+            driverParam: number;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
         createdAt: drizzle_orm_sqlite_core.SQLiteColumn<{
             name: "created_at";
             tableName: "sessions";
@@ -413,8 +484,8 @@ declare const messages: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
-        thinking: drizzle_orm_sqlite_core.SQLiteColumn<{
-            name: "thinking";
+        model: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "model";
             tableName: "messages";
             dataType: "string";
             columnType: "SQLiteText";
@@ -432,8 +503,8 @@ declare const messages: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
-        toolCalls: drizzle_orm_sqlite_core.SQLiteColumn<{
-            name: "tool_calls";
+        mode: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "mode";
             tableName: "messages";
             dataType: "string";
             columnType: "SQLiteText";
@@ -451,8 +522,42 @@ declare const messages: drizzle_orm_sqlite_core.SQLiteTableWithColumns<{
         }, {}, {
             length: number | undefined;
         }>;
-        searchResults: drizzle_orm_sqlite_core.SQLiteColumn<{
-            name: "search_results";
+        webSearchEnabled: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "web_search_enabled";
+            tableName: "messages";
+            dataType: "boolean";
+            columnType: "SQLiteBoolean";
+            data: boolean;
+            driverParam: number;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        thinkingEnabled: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "thinking_enabled";
+            tableName: "messages";
+            dataType: "boolean";
+            columnType: "SQLiteBoolean";
+            data: boolean;
+            driverParam: number;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        steps: drizzle_orm_sqlite_core.SQLiteColumn<{
+            name: "steps";
             tableName: "messages";
             dataType: "string";
             columnType: "SQLiteText";
@@ -1259,6 +1364,7 @@ declare class SqliteAdapter implements StorageAdapter {
     private sqlite;
     private db;
     private config;
+    get sqliteInstance(): Database.Database;
     constructor(dbPath: string, config?: StorageConfig);
     getSessions(ctx: StorageContext): Promise<SessionRecord[]>;
     getSession(id: string, ctx: StorageContext): Promise<SessionRecord | null>;
@@ -1268,6 +1374,7 @@ declare class SqliteAdapter implements StorageAdapter {
     getMessages(sessionId: string, ctx: StorageContext): Promise<MessageRecord[]>;
     getMessage(id: string, ctx: StorageContext): Promise<MessageRecord | null>;
     saveMessage(input: CreateMessageInput, ctx: StorageContext): Promise<MessageRecord>;
+    updateMessage(id: string, data: UpdateMessageInput, ctx: StorageContext): Promise<void>;
     deleteMessagesAfter(sessionId: string, timestamp: Date, ctx: StorageContext): Promise<void>;
     getOperations(sessionId: string, ctx: StorageContext): Promise<OperationRecord[]>;
     getOperationsByMessage(messageId: string, ctx: StorageContext): Promise<OperationRecord[]>;
@@ -1286,6 +1393,10 @@ declare class SqliteAdapter implements StorageAdapter {
         contentType: EmbeddingRecord['contentType'];
     }, ctx: StorageContext): Promise<void>;
     searchSimilar(queryEmbedding: number[], options: VectorSearchOptions, ctx: StorageContext): Promise<VectorSearchResult[]>;
+    getUserSetting(key: string, ctx: StorageContext): Promise<string | null>;
+    setUserSetting(key: string, value: string, ctx: StorageContext): Promise<void>;
+    getUserSettings(ctx: StorageContext): Promise<Record<string, string>>;
+    deleteUserSetting(key: string, ctx: StorageContext): Promise<void>;
     close(): Promise<void>;
 }
 
@@ -1309,6 +1420,7 @@ declare class PostgresAdapter implements StorageAdapter {
     getMessages(sessionId: string, ctx: StorageContext): Promise<MessageRecord[]>;
     getMessage(id: string, ctx: StorageContext): Promise<MessageRecord | null>;
     saveMessage(input: CreateMessageInput, ctx: StorageContext): Promise<MessageRecord>;
+    updateMessage(id: string, data: UpdateMessageInput, ctx: StorageContext): Promise<void>;
     deleteMessagesAfter(sessionId: string, timestamp: Date, ctx: StorageContext): Promise<void>;
     getOperations(sessionId: string, ctx: StorageContext): Promise<OperationRecord[]>;
     getOperationsByMessage(messageId: string, ctx: StorageContext): Promise<OperationRecord[]>;
@@ -1327,6 +1439,10 @@ declare class PostgresAdapter implements StorageAdapter {
         contentType: EmbeddingRecord['contentType'];
     }, ctx: StorageContext): Promise<void>;
     searchSimilar(queryEmbedding: number[], options: VectorSearchOptions, ctx: StorageContext): Promise<VectorSearchResult[]>;
+    getUserSetting(key: string, ctx: StorageContext): Promise<string | null>;
+    setUserSetting(key: string, value: string, ctx: StorageContext): Promise<void>;
+    getUserSettings(ctx: StorageContext): Promise<Record<string, string>>;
+    deleteUserSetting(key: string, ctx: StorageContext): Promise<void>;
     close(): Promise<void>;
 }
 
@@ -1375,4 +1491,4 @@ declare function getDefaultBackupDir(): string;
  */
 declare function getDefaultTrashDir(): string;
 
-export { type BackupRecord, type CreateMessageInput, type CreateOperationInput, type CreateSessionInput, type EmbeddingRecord, type MessageRecord, type OperationRecord, type OperationStatus, PostgresAdapter, type SessionRecord, SqliteAdapter, type StorageAdapter, type StorageConfig, type StorageContext, type TrashRecord, type UpdateSessionInput, type VectorSearchOptions, type VectorSearchResult, backups, createStorage, embeddings, getDefaultBackupDir, getDefaultStoragePath, getDefaultTrashDir, messages, operations, sessions, trash };
+export { type BackupRecord, type CreateMessageInput, type CreateOperationInput, type CreateSessionInput, type EmbeddingRecord, type MessageRecord, type OperationRecord, type OperationStatus, PostgresAdapter, type SessionRecord, SqliteAdapter, type StorageAdapter, type StorageConfig, type StorageContext, type TrashRecord, type UpdateMessageInput, type UpdateSessionInput, type VectorSearchOptions, type VectorSearchResult, backups, createStorage, embeddings, getDefaultBackupDir, getDefaultStoragePath, getDefaultTrashDir, messages, operations, sessions, trash };

package/dist/index.js

@@ -6,10 +6,10 @@ import {
   operations,
   sessions,
   trash
-} from "./chunk-GCKDCC3O.js";
+} from "./chunk-LZ2LBU6Q.js";
 import {
   PostgresAdapter
-} from "./chunk-BRWTH4LQ.js";
+} from "./chunk-7AKL7W3Y.js";
 
 // src/index.ts
 import { mkdirSync, existsSync } from "fs";
@@ -24,14 +24,14 @@ async function createStorage(config) {
       if (!existsSync(dir)) {
         mkdirSync(dir, { recursive: true });
       }
-      const { SqliteAdapter: SqliteAdapter2 } = await import("./sqlite-AJRNISP2.js");
+      const { SqliteAdapter: SqliteAdapter2 } = await import("./sqlite-5SSRI3KS.js");
       return new SqliteAdapter2(config.sqlitePath, config);
     }
     case "postgres": {
       if (!config.postgresUrl) {
         throw new Error("PostgreSQL \u9700\u8981\u63D0\u4F9B postgresUrl \u914D\u7F6E");
       }
-      const { PostgresAdapter: PostgresAdapter2 } = await import("./postgres-PBPM2KDK.js");
+      const { PostgresAdapter: PostgresAdapter2 } = await import("./postgres-4AZE26G4.js");
       return new PostgresAdapter2(config.postgresUrl, config);
     }
     default:

package/dist/{postgres-PBPM2KDK.js → postgres-4AZE26G4.js}

@@ -1,6 +1,6 @@
 import {
   PostgresAdapter
-} from "./chunk-BRWTH4LQ.js";
+} from "./chunk-7AKL7W3Y.js";
 export {
   PostgresAdapter
 };

package/dist/{sqlite-AJRNISP2.js → sqlite-5SSRI3KS.js}

@@ -1,6 +1,6 @@
 import {
   SqliteAdapter
-} from "./chunk-GCKDCC3O.js";
+} from "./chunk-LZ2LBU6Q.js";
 export {
   SqliteAdapter
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@huyooo/ai-chat-storage",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "AI Chat 统一存储层 - SQLite",
   "type": "module",
   "main": "./dist/index.js",
@@ -8,8 +8,8 @@
   "types": "./dist/index.d.ts",
   "exports": {
     ".": {
-      "import": "./dist/index.js",
-      "types": "./dist/index.d.ts"
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
     }
   },
   "files": [