chatlab-cli 0.24.1 → 0.25.1

package/dist/chunk-WYHGWTA4.mjs

@@ -0,0 +1,485 @@
+import { createRequire as __createRequire } from 'module';
+import { dirname as __pathDirname } from 'path';
+import { fileURLToPath as __fileURLToPath } from 'url';
+const require = __createRequire(import.meta.url);
+const __filename = __fileURLToPath(import.meta.url);
+const __dirname = __pathDirname(__filename);
+import {
+  AGENT_TOOL_REGISTRY,
+  CoreDataProvider
+} from "./chunk-JE3FOFF7.mjs";
+import {
+  AgentEventHandler,
+  AiLogger,
+  CHART_CAPABILITY_CORE_TOOLS,
+  DEFAULT_MAX_TOOL_ROUNDS,
+  SkillManager,
+  applyPreprocessingPipeline,
+  batchSegmentWithFrequency,
+  buildPiModel,
+  buildSkillMenuWithBuiltinChart,
+  buildSystemPrompt,
+  checkAndCompress,
+  createActivateSkillTool,
+  createAiTranslate,
+  createChartSchemaGateState,
+  createCompressionLlmAdapter,
+  formatAIError,
+  getAllowedBuiltinToolsForChartAutoSkill,
+  getChartCapabilitySkill,
+  getChatOverview,
+  getSkillConfigWithBuiltinChart,
+  initTokenizer,
+  normalizeBuiltinToolNames,
+  parseAssistantFile,
+  resolveApiKey,
+  resolveChartRuntimeForRequest,
+  runAgentCore,
+  wrapWithChartSchemaGate
+} from "./chunk-7VG6QGQ2.mjs";
+
+// src/ai/logger.ts
+var logger = null;
+function initServerAiLogger(logsDir) {
+  if (!logger) {
+    logger = new AiLogger(logsDir);
+  }
+  return logger;
+}
+function getServerAiLogger() {
+  return logger;
+}
+function closeServerAiLogger() {
+  if (logger) {
+    logger.close();
+    logger = null;
+  }
+}
+
+// src/ai/tool-adapter.ts
+var DEFAULT_MAX_TOOL_RESULT_TOKENS = 8e3;
+var TOOL_TRUNCATION_STRATEGY = {
+  search_messages: "keep_first",
+  deep_search_messages: "keep_first",
+  get_recent_messages: "keep_last",
+  get_message_context: "keep_last",
+  get_segment_messages: "keep_last",
+  get_conversation_between: "keep_last"
+};
+function convertJsonSchemaToParameters(schema) {
+  const properties = {};
+  for (const [key, prop] of Object.entries(schema.properties)) {
+    properties[key] = { ...prop };
+  }
+  return {
+    type: "object",
+    properties,
+    required: schema.required || []
+  };
+}
+function adaptToolsForAgent(tools, getContext, options) {
+  const tokenBudget = options?.maxToolResultTokens ?? DEFAULT_MAX_TOOL_RESULT_TOKENS;
+  const chartSchemaGateState = createChartSchemaGateState();
+  return tools.map(
+    (tool) => wrapWithChartSchemaGate(
+      {
+        name: tool.name,
+        label: tool.name,
+        description: tool.description,
+        parameters: convertJsonSchemaToParameters(tool.inputSchema),
+        async execute(_toolCallId, params) {
+          const toolParams = params && typeof params === "object" ? params : {};
+          const ctx = getContext();
+          const execCtx = {
+            db: ctx.db,
+            dataProvider: new CoreDataProvider(ctx.db),
+            sessionId: ctx.sessionId,
+            locale: ctx.locale,
+            segmentText: (texts, locale, options2) => batchSegmentWithFrequency(texts, locale, options2)
+          };
+          try {
+            const result = await tool.handler(toolParams, execCtx);
+            const chartDetails = result.chart || result.charts ? {
+              ...result.chart ? { chart: result.chart } : {},
+              ...result.charts ? { charts: result.charts } : {}
+            } : {};
+            if (result.rawMessages && result.rawMessages.length > 0) {
+              const pipelineResult = applyPreprocessingPipeline({
+                rawMessages: result.rawMessages,
+                locale: ctx.locale,
+                maxToolResultTokens: tokenBudget,
+                truncationStrategy: TOOL_TRUNCATION_STRATEGY[tool.name] ?? "keep_last",
+                extraDetails: result.data ?? {},
+                logger: getServerAiLogger() ?? void 0
+              });
+              return {
+                content: [{ type: "text", text: pipelineResult.text }],
+                details: Object.keys(chartDetails).length > 0 ? chartDetails : null
+              };
+            }
+            const baseDetails = typeof result.data === "object" && result.data !== null ? result.data : result.data === void 0 ? null : { value: result.data };
+            return {
+              content: [{ type: "text", text: result.content }],
+              details: Object.keys(chartDetails).length > 0 ? { ...baseDetails ?? {}, ...chartDetails } : baseDetails
+            };
+          } catch (error) {
+            const msg = error instanceof Error ? error.message : String(error);
+            return { content: [{ type: "text", text: `Error: ${msg}` }], details: null };
+          }
+        }
+      },
+      chartSchemaGateState
+    )
+  );
+}
+
+// src/ai/llm-config.ts
+import * as fs from "fs";
+import * as path from "path";
+function loadLlmConfig(aiDataDir) {
+  const configPath = path.join(aiDataDir, "llm-config.json");
+  if (!fs.existsSync(configPath)) {
+    return { configs: [], defaultAssistant: null, fastModel: null };
+  }
+  try {
+    const data = JSON.parse(fs.readFileSync(configPath, "utf-8"));
+    const configs = (data.configs || []).map((c) => {
+      const provider = c.provider || "";
+      const authProfile = c.authProfile;
+      const profileKey = resolveApiKey(provider, authProfile);
+      const rawKey = c.apiKey || "";
+      const fallbackKey = rawKey.startsWith("enc:") ? "" : rawKey;
+      return {
+        ...c,
+        apiKey: profileKey || fallbackKey
+      };
+    });
+    return {
+      configs,
+      defaultAssistant: data.defaultAssistant ?? null,
+      fastModel: data.fastModel ?? null
+    };
+  } catch {
+    return { configs: [], defaultAssistant: null, fastModel: null };
+  }
+}
+function resolveSlot(slot, configs) {
+  if (slot && configs.some((c) => c.id === slot.configId)) return slot;
+  const fallback = configs[0];
+  if (!fallback) return null;
+  return { configId: fallback.id, modelId: fallback.model };
+}
+function getDefaultAssistantConfig(aiDataDir) {
+  const store = loadLlmConfig(aiDataDir);
+  const slot = resolveSlot(store.defaultAssistant, store.configs);
+  if (!slot) return null;
+  const config = store.configs.find((c) => c.id === slot.configId);
+  if (!config) return null;
+  return { ...config, model: slot.modelId || config.model };
+}
+function buildPiModel2(config) {
+  return buildPiModel(config);
+}
+
+// src/ai/assistant-loader.ts
+import * as fs2 from "fs";
+import * as path2 from "path";
+function loadAssistantConfig(aiDataDir, assistantId) {
+  const filePath = path2.join(aiDataDir, "assistants", `${assistantId}.md`);
+  if (!fs2.existsSync(filePath)) return null;
+  try {
+    const content = fs2.readFileSync(filePath, "utf-8");
+    return parseAssistantFile(content, filePath);
+  } catch {
+    return null;
+  }
+}
+
+// src/ai/agent.ts
+async function runServerAgent(options) {
+  const {
+    userMessage,
+    aiChatId,
+    historyLeafMessageId,
+    chatType = "group",
+    locale = "zh-CN",
+    assistantSystemPrompt,
+    skillMenu,
+    skillDef,
+    compressionConfig,
+    tools = [],
+    aiDataDir,
+    aiChatManager,
+    onEvent,
+    abortSignal,
+    ownerInfo,
+    mentionedMembers,
+    dataSnapshot,
+    thinkingLevel
+  } = options;
+  const aiLogger = getServerAiLogger();
+  await initTokenizer();
+  const llmConfig = getDefaultAssistantConfig(aiDataDir);
+  if (!llmConfig) {
+    onEvent({ type: "error", error: { name: "ConfigError", message: "LLM service not configured" } });
+    onEvent({ type: "done", isFinished: true });
+    return;
+  }
+  const piModel = buildPiModel2(llmConfig);
+  const t = createAiTranslate(locale);
+  let skillCtx;
+  if (skillDef) {
+    skillCtx = { skillDef };
+  } else if (skillMenu) {
+    skillCtx = { skillMenu };
+  }
+  const systemPrompt = buildSystemPrompt({
+    t,
+    chatType,
+    assistantSystemPrompt,
+    ownerInfo,
+    locale,
+    skillCtx,
+    mentionedMembers,
+    dataSnapshot
+  });
+  const handler = new AgentEventHandler({
+    onChunk: onEvent,
+    context: {},
+    systemPrompt
+  });
+  if (compressionConfig?.enabled && historyLeafMessageId === void 0) {
+    const llmAdapter = createCompressionLlmAdapter({
+      piModel,
+      apiKey: llmConfig.apiKey,
+      onCompressing: () => handler.emitStatus("compressing", [])
+    });
+    const compressionResult = await checkAndCompress(
+      aiChatId,
+      compressionConfig,
+      systemPrompt,
+      llmAdapter,
+      aiChatManager,
+      aiLogger ?? void 0
+    );
+    if (compressionResult.compressed) {
+      onEvent({
+        type: "compression_done",
+        compressionResult: {
+          summaryContent: compressionResult.summaryContent ?? "",
+          tokensBefore: compressionResult.tokensBefore ?? 0,
+          tokensAfter: compressionResult.tokensAfter ?? 0,
+          timestamp: Date.now()
+        }
+      });
+    }
+  } else if (compressionConfig?.enabled && historyLeafMessageId !== void 0) {
+    aiLogger?.info?.("Compression", "Skipping compression for edited branch request", {
+      aiChatId,
+      historyLeafMessageId
+    });
+  }
+  if (abortSignal?.aborted) {
+    handler.emitStatus("aborted", [], { force: true });
+    onEvent({ type: "done", isFinished: true, usage: handler.cloneUsage() });
+    return;
+  }
+  let history = [];
+  try {
+    history = aiChatManager.getHistoryForAgent(aiChatId, void 0, historyLeafMessageId);
+  } catch {
+  }
+  handler.emitStatus("preparing", [], { pendingUserMessage: userMessage, force: true });
+  const steerMessage = t("ai.agent.answerWithoutTools");
+  let cachedMessages = [];
+  try {
+    const result = await runAgentCore({
+      piModel,
+      apiKey: llmConfig.apiKey,
+      systemPrompt,
+      tools,
+      history,
+      userMessage,
+      maxToolRounds: DEFAULT_MAX_TOOL_ROUNDS,
+      abortSignal,
+      steerMessage,
+      thinkingLevel,
+      onConvertToLlm: (filteredMessages) => {
+        cachedMessages = filteredMessages;
+      },
+      onEvent: (coreEvent) => handler.handleCoreEvent(coreEvent, cachedMessages),
+      onDebugContext: (messages) => {
+        try {
+          aiChatManager.setPendingDebugContext(aiChatId, JSON.stringify(messages, null, 2));
+        } catch {
+        }
+      }
+    });
+    if (result.error) {
+      const friendlyMessage = formatAIError(result.error);
+      onEvent({ type: "error", error: { name: "AgentError", message: friendlyMessage } });
+    }
+    handler.emitStatus("completed", cachedMessages, { force: true });
+    onEvent({ type: "done", isFinished: true, usage: result.usage });
+  } catch (error) {
+    const friendlyMessage = formatAIError(error);
+    aiLogger?.error("ServerAgent", "Agent execution error", { error: String(error) });
+    handler.emitStatus("error", cachedMessages, { force: true });
+    onEvent({ type: "error", error: { name: "AgentError", message: friendlyMessage } });
+    onEvent({ type: "done", isFinished: true, usage: handler.cloneUsage() });
+  }
+}
+
+// src/ai/agent-stream-runner.ts
+function getAiDir(dbManager) {
+  const pathProvider = dbManager["pathProvider"];
+  if (!pathProvider) {
+    throw Object.assign(new Error("PathProvider not available"), { statusCode: 500 });
+  }
+  return pathProvider.getAiDataDir();
+}
+var RAW_SQL_TOOL_NAMES = /* @__PURE__ */ new Set(["execute_sql"]);
+function getAvailableToolDefs(isChartCapability, allowedToolSet) {
+  if (isChartCapability) {
+    const chartCoreTools = new Set(CHART_CAPABILITY_CORE_TOOLS);
+    return AGENT_TOOL_REGISTRY.filter(
+      (tool) => chartCoreTools.has(tool.name) || tool.category === "analysis" && allowedToolSet?.has(tool.name) && !RAW_SQL_TOOL_NAMES.has(tool.name)
+    );
+  }
+  return allowedToolSet ? AGENT_TOOL_REGISTRY.filter((tool) => tool.category !== "analysis" || allowedToolSet.has(tool.name)) : AGENT_TOOL_REGISTRY;
+}
+function getAllowedToolSet(isChartCapability, allowedBuiltinTools) {
+  if (isChartCapability) {
+    return new Set(normalizeBuiltinToolNames(allowedBuiltinTools ?? []));
+  }
+  return allowedBuiltinTools === void 0 ? null : new Set(normalizeBuiltinToolNames(allowedBuiltinTools));
+}
+function createCliRunAgentStream(dbManager, aiChatManager) {
+  return async (params, onEvent, abortSignal) => {
+    const {
+      userMessage,
+      aiChatId,
+      historyLeafMessageId,
+      sessionId,
+      chatType,
+      locale,
+      assistantId,
+      skillId,
+      enableAutoSkill,
+      compressionConfig,
+      ownerInfo,
+      mentionedMembers,
+      thinkingLevel
+    } = params;
+    const aiDataDir = getAiDir(dbManager);
+    let assistantSystemPrompt;
+    let assistantAllowedTools;
+    if (assistantId) {
+      const assistantConfig = loadAssistantConfig(aiDataDir, assistantId);
+      if (assistantConfig?.systemPrompt) {
+        assistantSystemPrompt = assistantConfig.systemPrompt;
+      }
+      assistantAllowedTools = assistantConfig?.allowedBuiltinTools;
+    }
+    const llmConfig = getDefaultAssistantConfig(aiDataDir);
+    const maxToolResultPercent = compressionConfig?.maxToolResultPercent ?? 50;
+    const contextWindow = llmConfig ? buildPiModel2(llmConfig).contextWindow ?? 128e3 : 128e3;
+    const maxToolResultTokens = Math.floor(contextWindow * (maxToolResultPercent / 100));
+    const db = dbManager.open?.(sessionId);
+    const chartRuntime = resolveChartRuntimeForRequest({
+      skillId,
+      userMessage,
+      locale,
+      assistantAllowedTools
+    });
+    const isChartCapability = chartRuntime.isChartCapability;
+    const autoSkillAllowedTools = !skillId && enableAutoSkill ? getAllowedBuiltinToolsForChartAutoSkill(assistantAllowedTools) : assistantAllowedTools;
+    const allowedToolSet = getAllowedToolSet(
+      isChartCapability,
+      isChartCapability ? chartRuntime.allowedBuiltinTools : autoSkillAllowedTools
+    );
+    const availableToolDefs = getAvailableToolDefs(isChartCapability, allowedToolSet);
+    const agentTools = db ? adaptToolsForAgent(availableToolDefs, () => ({ db, sessionId, locale }), { maxToolResultTokens }) : [];
+    const skillMgr = new SkillManager(aiDataDir);
+    skillMgr.init();
+    const toolNames = agentTools.map((t) => t.name);
+    let resolvedSkillDef;
+    let resolvedSkillMenu;
+    if (isChartCapability) {
+      const def = chartRuntime.skillDef ?? getChartCapabilitySkill(locale ?? "zh-CN");
+      resolvedSkillDef = { name: def.name, prompt: def.prompt };
+    } else if (skillId) {
+      const def = skillMgr.getSkillConfig(skillId);
+      if (def) resolvedSkillDef = { name: def.name, prompt: def.prompt };
+    } else if (enableAutoSkill) {
+      const menu = buildSkillMenuWithBuiltinChart(
+        skillMgr.getSkillMenu(chatType ?? "group", toolNames),
+        locale,
+        toolNames
+      );
+      if (menu) resolvedSkillMenu = menu;
+    }
+    if (resolvedSkillMenu) {
+      const activateSkillTool = createActivateSkillTool({
+        chatType: chatType ?? "group",
+        allowedTools: toolNames,
+        coreToolNames: new Set(CHART_CAPABILITY_CORE_TOOLS),
+        locale,
+        getSkillConfig: (id) => getSkillConfigWithBuiltinChart(id, locale, (skillConfigId) => skillMgr.getSkillConfig(skillConfigId))
+      });
+      agentTools.push(activateSkillTool);
+    }
+    const resolvedCompression = compressionConfig?.enabled ? {
+      enabled: true,
+      tokenThresholdPercent: compressionConfig.tokenThresholdPercent ?? 75,
+      bufferSizePercent: compressionConfig.bufferSizePercent ?? 20,
+      maxToolResultPercent: compressionConfig.maxToolResultPercent
+    } : void 0;
+    let dataSnapshot;
+    if (db) {
+      try {
+        const overview = getChatOverview(db, 5);
+        if (overview) {
+          dataSnapshot = {
+            name: overview.name,
+            platform: overview.platform,
+            type: overview.type,
+            totalMessages: overview.totalMessages,
+            totalMembers: overview.totalMembers,
+            firstMessageTs: overview.firstMessageTs,
+            lastMessageTs: overview.lastMessageTs,
+            capturedAt: Math.floor(Date.now() / 1e3)
+          };
+        }
+      } catch {
+      }
+    }
+    await runServerAgent({
+      userMessage,
+      aiChatId,
+      historyLeafMessageId,
+      chatType,
+      locale,
+      assistantSystemPrompt,
+      skillMenu: resolvedSkillMenu,
+      skillDef: resolvedSkillDef,
+      compressionConfig: resolvedCompression,
+      tools: agentTools,
+      aiDataDir,
+      aiChatManager,
+      onEvent,
+      abortSignal,
+      ownerInfo,
+      mentionedMembers,
+      dataSnapshot,
+      thinkingLevel
+    });
+  };
+}
+
+export {
+  initServerAiLogger,
+  closeServerAiLogger,
+  createCliRunAgentStream
+};
+//# sourceMappingURL=chunk-WYHGWTA4.mjs.map

package/dist/chunk-WYHGWTA4.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ai/logger.ts","../src/ai/tool-adapter.ts","../src/ai/llm-config.ts","../src/ai/assistant-loader.ts","../src/ai/agent.ts","../src/ai/agent-stream-runner.ts"],"sourcesContent":["/**\n * Server 端 AI 日志\n *\n * 复用 @openchatlab/node-runtime 的 AiLogger，\n * 在 startHttpServer() 中通过 initServerAiLogger() 初始化。\n */\n\nimport { AiLogger } from '@openchatlab/node-runtime'\n\nlet logger: AiLogger | null = null\n\nexport function initServerAiLogger(logsDir: string): AiLogger {\n  if (!logger) {\n    logger = new AiLogger(logsDir)\n  }\n  return logger\n}\n\nexport function getServerAiLogger(): AiLogger | null {\n  return logger\n}\n\nexport function closeServerAiLogger(): void {\n  if (logger) {\n    logger.close()\n    logger = null\n  }\n}\n","/**\n * 工具适配层\n *\n * 将 @openchatlab/tools 的 ToolDefinition 适配为 @earendil-works/pi-agent-core 的 AgentTool 格式。\n * 消息类工具返回 rawMessages 时自动执行预处理管道（清洗、去噪、脱敏、截断、格式化）。\n */\n\nimport type { ToolDefinition, ToolExecutionContext } from '@openchatlab/tools'\nimport { CoreDataProvider } from '@openchatlab/tools'\nimport type { DatabaseAdapter } from '@openchatlab/core'\nimport {\n  applyPreprocessingPipeline,\n  batchSegmentWithFrequency,\n  type AgentTool,\n  type AgentToolResult,\n  type PreprocessableMessage,\n  type TruncationStrategy,\n  createChartSchemaGateState,\n  wrapWithChartSchemaGate,\n} from '@openchatlab/node-runtime'\nimport { getServerAiLogger } from './logger'\n\nconst DEFAULT_MAX_TOOL_RESULT_TOKENS = 8000\n\nconst TOOL_TRUNCATION_STRATEGY: Record<string, TruncationStrategy> = {\n  search_messages: 'keep_first',\n  deep_search_messages: 'keep_first',\n  get_recent_messages: 'keep_last',\n  get_message_context: 'keep_last',\n  get_segment_messages: 'keep_last',\n  get_conversation_between: 'keep_last',\n}\n\nexport interface ServerToolContext {\n  db: DatabaseAdapter\n  sessionId: string\n  locale?: string\n}\n\nfunction convertJsonSchemaToParameters(schema: ToolDefinition['inputSchema']) {\n  const properties: Record<string, unknown> = {}\n  for (const [key, prop] of Object.entries(schema.properties)) {\n    properties[key] = { ...prop }\n  }\n  return {\n    type: 'object' as const,\n    properties,\n    required: schema.required || [],\n  }\n}\n\nexport interface AdaptToolsOptions {\n  maxToolResultTokens?: number\n}\n\nexport function adaptToolsForAgent(\n  tools: ToolDefinition[],\n  getContext: () => ServerToolContext,\n  options?: AdaptToolsOptions\n): AgentTool<any, any>[] {\n  const tokenBudget = options?.maxToolResultTokens ?? DEFAULT_MAX_TOOL_RESULT_TOKENS\n  const chartSchemaGateState = createChartSchemaGateState()\n\n  return tools.map((tool) =>\n    wrapWithChartSchemaGate(\n      {\n        name: tool.name,\n        label: tool.name,\n        description: tool.description,\n        parameters: convertJsonSchemaToParameters(tool.inputSchema) as any,\n        async execute(_toolCallId: string, params: unknown): Promise<AgentToolResult<unknown>> {\n          const toolParams = (params && typeof params === 'object' ? params : {}) as Record<string, unknown>\n          const ctx = getContext()\n          const execCtx: ToolExecutionContext = {\n            db: ctx.db,\n            dataProvider: new CoreDataProvider(ctx.db),\n            sessionId: ctx.sessionId,\n            locale: ctx.locale,\n            segmentText: (texts, locale, options) => batchSegmentWithFrequency(texts, locale as any, options as any),\n          }\n          try {\n            const result = await tool.handler(toolParams, execCtx)\n            const chartDetails =\n              result.chart || result.charts\n                ? {\n                    ...(result.chart ? { chart: result.chart } : {}),\n                    ...(result.charts ? { charts: result.charts } : {}),\n                  }\n                : {}\n\n            if (result.rawMessages && result.rawMessages.length > 0) {\n              const pipelineResult = applyPreprocessingPipeline({\n                rawMessages: result.rawMessages as PreprocessableMessage[],\n                locale: ctx.locale,\n                maxToolResultTokens: tokenBudget,\n                truncationStrategy: TOOL_TRUNCATION_STRATEGY[tool.name] ?? 'keep_last',\n                extraDetails: (result.data ?? {}) as Record<string, unknown>,\n                logger: getServerAiLogger() ?? undefined,\n              })\n              return {\n                content: [{ type: 'text', text: pipelineResult.text }],\n                details: Object.keys(chartDetails).length > 0 ? chartDetails : null,\n              }\n            }\n\n            const baseDetails =\n              typeof result.data === 'object' && result.data !== null\n                ? (result.data as Record<string, unknown>)\n                : result.data === undefined\n                  ? null\n                  : { value: result.data }\n\n            return {\n              content: [{ type: 'text', text: result.content }],\n              details: Object.keys(chartDetails).length > 0 ? { ...(baseDetails ?? {}), ...chartDetails } : baseDetails,\n            }\n          } catch (error) {\n            const msg = error instanceof Error ? error.message : String(error)\n            return { content: [{ type: 'text', text: `Error: ${msg}` }], details: null }\n          }\n        },\n      },\n      chartSchemaGateState\n    )\n  )\n}\n","/**\n * 服务端 LLM 配置加载\n *\n * API Key 从 ~/.chatlab/auth-profiles.json 读取，\n * 匹配顺序：config.authProfile 精确匹配 > config.provider 兜底匹配。\n *\n * 模型/Provider 等配置仍从 llm-config.json 读取。\n */\n\nimport * as fs from 'fs'\nimport * as path from 'path'\nimport type { PiModel, PiApi } from '@openchatlab/node-runtime'\nimport { buildPiModel as buildPiModelCore, type PiModelConfig } from '@openchatlab/node-runtime'\nimport { resolveApiKey, writeAuthProfile } from '@openchatlab/config'\nimport { randomUUID } from 'crypto'\n\n// ==================== Types ====================\n\nexport interface AIServiceConfig {\n  id: string\n  name: string\n  provider: string\n  apiKey: string\n  model?: string\n  baseUrl?: string\n  maxTokens?: number\n  apiFormat?: string\n  customModels?: Array<{ id: string; name: string }>\n  authProfile?: string\n}\n\ninterface ModelSlot {\n  configId: string\n  modelId?: string\n}\n\ninterface AIConfigStore {\n  configs: AIServiceConfig[]\n  defaultAssistant: ModelSlot | null\n  fastModel: ModelSlot | null\n}\n\n// ==================== Config Loading ====================\n\nexport function loadLlmConfig(aiDataDir: string): AIConfigStore {\n  const configPath = path.join(aiDataDir, 'llm-config.json')\n  if (!fs.existsSync(configPath)) {\n    return { configs: [], defaultAssistant: null, fastModel: null }\n  }\n\n  try {\n    const data = JSON.parse(fs.readFileSync(configPath, 'utf-8'))\n\n    const configs = (data.configs || []).map((c: Record<string, unknown>) => {\n      const provider = (c.provider as string) || ''\n      const authProfile = c.authProfile as string | undefined\n      const profileKey = resolveApiKey(provider, authProfile)\n      const rawKey = (c.apiKey as string) || ''\n      const fallbackKey = rawKey.startsWith('enc:') ? '' : rawKey\n      return {\n        ...c,\n        apiKey: profileKey || fallbackKey,\n      }\n    })\n    return {\n      configs,\n      defaultAssistant: data.defaultAssistant ?? null,\n      fastModel: data.fastModel ?? null,\n    }\n  } catch {\n    return { configs: [], defaultAssistant: null, fastModel: null }\n  }\n}\n\nfunction resolveSlot(slot: ModelSlot | null | undefined, configs: AIServiceConfig[]): ModelSlot | null {\n  if (slot && configs.some((c) => c.id === slot.configId)) return slot\n  const fallback = configs[0]\n  if (!fallback) return null\n  return { configId: fallback.id, modelId: fallback.model }\n}\n\nexport function getDefaultAssistantConfig(aiDataDir: string): AIServiceConfig | null {\n  const store = loadLlmConfig(aiDataDir)\n  const slot = resolveSlot(store.defaultAssistant, store.configs)\n  if (!slot) return null\n  const config = store.configs.find((c) => c.id === slot.configId)\n  if (!config) return null\n  return { ...config, model: slot.modelId || config.model }\n}\n\n// ==================== Config Write Operations ====================\n\nconst MAX_CONFIG_COUNT = 99\n\nfunction saveLlmConfig(aiDataDir: string, store: AIConfigStore): void {\n  const configPath = path.join(aiDataDir, 'llm-config.json')\n  const toSave = {\n    ...store,\n    configs: store.configs.map((c) => {\n      const { apiKey: _k, ...rest } = c\n      return rest\n    }),\n    schemaVersion: 3,\n  }\n  fs.writeFileSync(configPath, JSON.stringify(toSave, null, 2), 'utf-8')\n}\n\nexport function addLlmConfig(\n  aiDataDir: string,\n  config: Omit<AIServiceConfig, 'id'>\n): { success: boolean; config?: AIServiceConfig; error?: string } {\n  const store = loadLlmConfig(aiDataDir)\n\n  if (store.configs.length >= MAX_CONFIG_COUNT) {\n    return { success: false, error: `Maximum ${MAX_CONFIG_COUNT} configs reached` }\n  }\n\n  const newConfig: AIServiceConfig = {\n    ...config,\n    id: randomUUID(),\n  }\n\n  const storeForSave = loadRawConfigStore(aiDataDir)\n  storeForSave.configs.push({\n    ...newConfig,\n    apiKey: '',\n  })\n\n  if (storeForSave.configs.length === 1) {\n    storeForSave.defaultAssistant = { configId: newConfig.id, modelId: newConfig.model || '' }\n  }\n\n  if (config.apiKey) {\n    const profileName = config.name?.toLowerCase().replace(/\\s+/g, '-') || config.provider\n    writeAuthProfile(profileName, {\n      type: 'api_key',\n      provider: config.provider,\n      key: config.apiKey,\n    })\n    ;(storeForSave.configs[storeForSave.configs.length - 1] as unknown as Record<string, unknown>).authProfile =\n      profileName\n  }\n\n  saveLlmConfig(aiDataDir, storeForSave)\n  return { success: true, config: { ...newConfig, apiKey: '' } }\n}\n\nexport function updateLlmConfig(\n  aiDataDir: string,\n  id: string,\n  updates: Partial<Omit<AIServiceConfig, 'id'>>\n): { success: boolean; error?: string } {\n  const storeForSave = loadRawConfigStore(aiDataDir)\n  const index = storeForSave.configs.findIndex((c) => c.id === id)\n\n  if (index === -1) {\n    return { success: false, error: 'Config not found' }\n  }\n\n  const { apiKey: newApiKey, ...restUpdates } = updates\n  const updated = {\n    ...storeForSave.configs[index],\n    ...restUpdates,\n  }\n  storeForSave.configs[index] = updated\n\n  if (newApiKey) {\n    const profileName = updated.name?.toLowerCase().replace(/\\s+/g, '-') || updated.provider\n    writeAuthProfile(profileName, {\n      type: 'api_key',\n      provider: updated.provider,\n      key: newApiKey,\n    })\n    ;(storeForSave.configs[index] as unknown as Record<string, unknown>).authProfile = profileName\n  }\n\n  saveLlmConfig(aiDataDir, storeForSave)\n  return { success: true }\n}\n\nexport function deleteLlmConfig(aiDataDir: string, id: string): { success: boolean; error?: string } {\n  const storeForSave = loadRawConfigStore(aiDataDir)\n  const index = storeForSave.configs.findIndex((c) => c.id === id)\n\n  if (index === -1) {\n    return { success: false, error: 'Config not found' }\n  }\n\n  storeForSave.configs.splice(index, 1)\n\n  const fallback = storeForSave.configs[0]\n  if (storeForSave.defaultAssistant?.configId === id) {\n    storeForSave.defaultAssistant = fallback ? { configId: fallback.id, modelId: fallback.model || '' } : null\n  }\n  if (storeForSave.fastModel?.configId === id) {\n    storeForSave.fastModel = fallback ? { configId: fallback.id, modelId: fallback.model || '' } : null\n  }\n\n  saveLlmConfig(aiDataDir, storeForSave)\n  return { success: true }\n}\n\nexport function setDefaultAssistantSlot(\n  aiDataDir: string,\n  configId: string,\n  modelId: string\n): { success: boolean; error?: string } {\n  const storeForSave = loadRawConfigStore(aiDataDir)\n  const config = storeForSave.configs.find((c) => c.id === configId)\n\n  if (!config) {\n    return { success: false, error: 'Config not found' }\n  }\n\n  storeForSave.defaultAssistant = { configId, modelId }\n  saveLlmConfig(aiDataDir, storeForSave)\n  return { success: true }\n}\n\nexport function setFastModelSlot(aiDataDir: string, slot: ModelSlot | null): { success: boolean; error?: string } {\n  const storeForSave = loadRawConfigStore(aiDataDir)\n\n  if (slot !== null) {\n    const config = storeForSave.configs.find((c) => c.id === slot.configId)\n    if (!config) {\n      return { success: false, error: 'Config not found' }\n    }\n  }\n\n  storeForSave.fastModel = slot\n  saveLlmConfig(aiDataDir, storeForSave)\n  return { success: true }\n}\n\nfunction loadRawConfigStore(aiDataDir: string): AIConfigStore {\n  const configPath = path.join(aiDataDir, 'llm-config.json')\n  if (!fs.existsSync(configPath)) {\n    return { configs: [], defaultAssistant: null, fastModel: null }\n  }\n  try {\n    const data = JSON.parse(fs.readFileSync(configPath, 'utf-8'))\n    return {\n      configs: data.configs || [],\n      defaultAssistant: data.defaultAssistant ?? null,\n      fastModel: data.fastModel ?? null,\n    }\n  } catch {\n    return { configs: [], defaultAssistant: null, fastModel: null }\n  }\n}\n\n// ==================== PiModel Builder ====================\n\nexport function buildPiModel(config: AIServiceConfig): PiModel<PiApi> {\n  return buildPiModelCore(config as PiModelConfig)\n}\n","/**\n * 助手配置加载器\n *\n * 从 ~/.chatlab/ai/assistants/*.md 加载助手系统提示词。\n */\n\nimport * as fs from 'fs'\nimport * as path from 'path'\nimport { parseAssistantFile } from '@openchatlab/node-runtime'\nimport type { AssistantConfig } from '@openchatlab/node-runtime'\n\nexport function loadAssistantConfig(aiDataDir: string, assistantId: string): AssistantConfig | null {\n  const filePath = path.join(aiDataDir, 'assistants', `${assistantId}.md`)\n  if (!fs.existsSync(filePath)) return null\n\n  try {\n    const content = fs.readFileSync(filePath, 'utf-8')\n    return parseAssistantFile(content, filePath)\n  } catch {\n    return null\n  }\n}\n","/**\n * 服务端 Agent\n *\n * 使用 @openchatlab/node-runtime 的 runAgentCore 编排对话流程，\n * 通过 AgentEventHandler 输出与 Electron 端一致的流式事件。\n */\n\nimport {\n  DEFAULT_MAX_TOOL_ROUNDS,\n  runAgentCore,\n  checkAndCompress,\n  buildSystemPrompt,\n  createAiTranslate,\n  createCompressionLlmAdapter,\n  AgentEventHandler,\n  formatAIError,\n  initTokenizer,\n  type AgentStreamChunk,\n  type PiMessage,\n  type SimpleHistoryMessage,\n  type AIChatManager,\n  type CompressionConfig,\n  type AgentTool,\n  type DataSnapshot,\n  type OwnerInfo,\n  type MentionedMember,\n} from '@openchatlab/node-runtime'\n\nimport { getDefaultAssistantConfig, buildPiModel } from './llm-config'\nimport { getServerAiLogger } from './logger'\n\nexport type { AgentStreamChunk }\n\nexport interface RunAgentOptions {\n  userMessage: string\n  aiChatId: string\n  historyLeafMessageId?: string | null\n  chatType?: 'group' | 'private'\n  locale?: string\n  assistantSystemPrompt?: string\n  skillMenu?: string | null\n  skillDef?: { name: string; prompt: string }\n  compressionConfig?: CompressionConfig\n  tools?: AgentTool[]\n  aiDataDir: string\n  aiChatManager: AIChatManager\n  onEvent: (event: AgentStreamChunk) => void\n  abortSignal?: AbortSignal\n  ownerInfo?: OwnerInfo\n  mentionedMembers?: MentionedMember[]\n  dataSnapshot?: DataSnapshot\n  thinkingLevel?: string\n}\n\nexport async function runServerAgent(options: RunAgentOptions): Promise<void> {\n  const {\n    userMessage,\n    aiChatId,\n    historyLeafMessageId,\n    chatType = 'group',\n    locale = 'zh-CN',\n    assistantSystemPrompt,\n    skillMenu,\n    skillDef,\n    compressionConfig,\n    tools = [],\n    aiDataDir,\n    aiChatManager,\n    onEvent,\n    abortSignal,\n    ownerInfo,\n    mentionedMembers,\n    dataSnapshot,\n    thinkingLevel,\n  } = options\n\n  const aiLogger = getServerAiLogger()\n\n  // 确保 tokenizer rank 表已加载（compression + agent 路径均依赖）\n  await initTokenizer()\n\n  const llmConfig = getDefaultAssistantConfig(aiDataDir)\n  if (!llmConfig) {\n    onEvent({ type: 'error', error: { name: 'ConfigError', message: 'LLM service not configured' } })\n    onEvent({ type: 'done', isFinished: true })\n    return\n  }\n\n  const piModel = buildPiModel(llmConfig)\n  const t = createAiTranslate(locale)\n\n  let skillCtx: { skillDef?: { name: string; prompt: string }; skillMenu?: string } | undefined\n  if (skillDef) {\n    skillCtx = { skillDef }\n  } else if (skillMenu) {\n    skillCtx = { skillMenu }\n  }\n\n  const systemPrompt = buildSystemPrompt({\n    t,\n    chatType,\n    assistantSystemPrompt,\n    ownerInfo,\n    locale,\n    skillCtx,\n    mentionedMembers,\n    dataSnapshot,\n  })\n\n  const handler = new AgentEventHandler({\n    onChunk: onEvent,\n    context: {},\n    systemPrompt,\n  })\n\n  if (compressionConfig?.enabled && historyLeafMessageId === undefined) {\n    const llmAdapter = createCompressionLlmAdapter({\n      piModel,\n      apiKey: llmConfig.apiKey,\n      onCompressing: () => handler.emitStatus('compressing', []),\n    })\n    const compressionResult = await checkAndCompress(\n      aiChatId,\n      compressionConfig,\n      systemPrompt,\n      llmAdapter,\n      aiChatManager,\n      aiLogger ?? undefined\n    )\n    if (compressionResult.compressed) {\n      onEvent({\n        type: 'compression_done',\n        compressionResult: {\n          summaryContent: compressionResult.summaryContent ?? '',\n          tokensBefore: compressionResult.tokensBefore ?? 0,\n          tokensAfter: compressionResult.tokensAfter ?? 0,\n          timestamp: Date.now(),\n        },\n      })\n    }\n  } else if (compressionConfig?.enabled && historyLeafMessageId !== undefined) {\n    aiLogger?.info?.('Compression', 'Skipping compression for edited branch request', {\n      aiChatId,\n      historyLeafMessageId,\n    })\n  }\n\n  if (abortSignal?.aborted) {\n    handler.emitStatus('aborted', [], { force: true })\n    onEvent({ type: 'done', isFinished: true, usage: handler.cloneUsage() })\n    return\n  }\n\n  let history: SimpleHistoryMessage[] = []\n  try {\n    history = aiChatManager.getHistoryForAgent(aiChatId, undefined, historyLeafMessageId)\n  } catch {\n    // empty history on failure\n  }\n\n  handler.emitStatus('preparing', [], { pendingUserMessage: userMessage, force: true })\n\n  const steerMessage = t('ai.agent.answerWithoutTools')\n  let cachedMessages: PiMessage[] = []\n\n  try {\n    const result = await runAgentCore({\n      piModel,\n      apiKey: llmConfig.apiKey,\n      systemPrompt,\n      tools,\n      history,\n      userMessage,\n      maxToolRounds: DEFAULT_MAX_TOOL_ROUNDS,\n      abortSignal,\n      steerMessage,\n      thinkingLevel: thinkingLevel as import('@openchatlab/core').ThinkingLevel | undefined,\n      onConvertToLlm: (filteredMessages) => {\n        cachedMessages = filteredMessages as PiMessage[]\n      },\n      onEvent: (coreEvent) => handler.handleCoreEvent(coreEvent, cachedMessages),\n      onDebugContext: (messages) => {\n        try {\n          aiChatManager.setPendingDebugContext(aiChatId, JSON.stringify(messages, null, 2))\n        } catch {\n          // silent\n        }\n      },\n    })\n\n    if (result.error) {\n      const friendlyMessage = formatAIError(result.error)\n      onEvent({ type: 'error', error: { name: 'AgentError', message: friendlyMessage } })\n    }\n\n    handler.emitStatus('completed', cachedMessages, { force: true })\n    onEvent({ type: 'done', isFinished: true, usage: result.usage })\n  } catch (error) {\n    const friendlyMessage = formatAIError(error)\n    aiLogger?.error('ServerAgent', 'Agent execution error', { error: String(error) })\n    handler.emitStatus('error', cachedMessages, { force: true })\n    onEvent({ type: 'error', error: { name: 'AgentError', message: friendlyMessage } })\n    onEvent({ type: 'done', isFinished: true, usage: handler.cloneUsage() })\n  }\n}\n","/**\n * CLI Agent stream runner — provides runAgentStream implementation\n * for the shared HTTP route context.\n */\n\nimport type { DatabaseManager, AIChatManager, AgentStreamChunk } from '@openchatlab/node-runtime'\nimport {\n  CHART_CAPABILITY_CORE_TOOLS,\n  SkillManager,\n  buildSkillMenuWithBuiltinChart,\n  createActivateSkillTool,\n  getAllowedBuiltinToolsForChartAutoSkill,\n  getChartCapabilitySkill,\n  getSkillConfigWithBuiltinChart,\n  resolveChartRuntimeForRequest,\n} from '@openchatlab/node-runtime'\nimport { getChatOverview, normalizeBuiltinToolNames } from '@openchatlab/core'\nimport type { DataSnapshot } from '@openchatlab/node-runtime'\nimport type { AgentStreamRequest } from '@openchatlab/http-routes'\nimport { AGENT_TOOL_REGISTRY } from '@openchatlab/tools'\nimport { adaptToolsForAgent } from './tool-adapter'\nimport { getDefaultAssistantConfig, buildPiModel } from './llm-config'\nimport { loadAssistantConfig } from './assistant-loader'\nimport { runServerAgent } from './agent'\n\nfunction getAiDir(dbManager: DatabaseManager): string {\n  const pathProvider = (dbManager as any)['pathProvider']\n  if (!pathProvider) {\n    throw Object.assign(new Error('PathProvider not available'), { statusCode: 500 })\n  }\n  return pathProvider.getAiDataDir()\n}\n\nconst RAW_SQL_TOOL_NAMES = new Set(['execute_sql'])\n\nexport function getAvailableToolDefs(isChartCapability: boolean, allowedToolSet: Set<string> | null) {\n  if (isChartCapability) {\n    const chartCoreTools = new Set<string>(CHART_CAPABILITY_CORE_TOOLS)\n    return AGENT_TOOL_REGISTRY.filter(\n      (tool) =>\n        chartCoreTools.has(tool.name) ||\n        (tool.category === 'analysis' && allowedToolSet?.has(tool.name) && !RAW_SQL_TOOL_NAMES.has(tool.name))\n    )\n  }\n\n  return allowedToolSet\n    ? AGENT_TOOL_REGISTRY.filter((tool) => tool.category !== 'analysis' || allowedToolSet.has(tool.name))\n    : AGENT_TOOL_REGISTRY\n}\n\n// 区分“未配置白名单”和“配置为空白名单”：前者无限制，后者禁用 analysis 工具。\nexport function getAllowedToolSet(\n  isChartCapability: boolean,\n  allowedBuiltinTools?: readonly string[]\n): Set<string> | null {\n  if (isChartCapability) {\n    return new Set(normalizeBuiltinToolNames(allowedBuiltinTools ?? []))\n  }\n  return allowedBuiltinTools === undefined ? null : new Set(normalizeBuiltinToolNames(allowedBuiltinTools))\n}\n\nexport function createCliRunAgentStream(\n  dbManager: DatabaseManager,\n  aiChatManager: AIChatManager\n): (params: AgentStreamRequest, onEvent: (chunk: AgentStreamChunk) => void, abortSignal: AbortSignal) => Promise<void> {\n  return async (params, onEvent, abortSignal) => {\n    const {\n      userMessage,\n      aiChatId,\n      historyLeafMessageId,\n      sessionId,\n      chatType,\n      locale,\n      assistantId,\n      skillId,\n      enableAutoSkill,\n      compressionConfig,\n      ownerInfo,\n      mentionedMembers,\n      thinkingLevel,\n    } = params\n\n    const aiDataDir = getAiDir(dbManager)\n\n    let assistantSystemPrompt: string | undefined\n    let assistantAllowedTools: string[] | undefined\n    if (assistantId) {\n      const assistantConfig = loadAssistantConfig(aiDataDir, assistantId)\n      if (assistantConfig?.systemPrompt) {\n        assistantSystemPrompt = assistantConfig.systemPrompt\n      }\n      assistantAllowedTools = assistantConfig?.allowedBuiltinTools\n    }\n\n    const llmConfig = getDefaultAssistantConfig(aiDataDir)\n    const maxToolResultPercent = compressionConfig?.maxToolResultPercent ?? 50\n    const contextWindow = llmConfig ? (buildPiModel(llmConfig).contextWindow ?? 128000) : 128000\n    const maxToolResultTokens = Math.floor(contextWindow * (maxToolResultPercent / 100))\n\n    const db = (dbManager as any).open?.(sessionId)\n    const chartRuntime = resolveChartRuntimeForRequest({\n      skillId,\n      userMessage,\n      locale,\n      assistantAllowedTools,\n    })\n    const isChartCapability = chartRuntime.isChartCapability\n    const autoSkillAllowedTools =\n      !skillId && enableAutoSkill\n        ? getAllowedBuiltinToolsForChartAutoSkill(assistantAllowedTools)\n        : assistantAllowedTools\n    const allowedToolSet = getAllowedToolSet(\n      isChartCapability,\n      isChartCapability ? chartRuntime.allowedBuiltinTools : autoSkillAllowedTools\n    )\n    const availableToolDefs = getAvailableToolDefs(isChartCapability, allowedToolSet)\n\n    const agentTools = db\n      ? adaptToolsForAgent(availableToolDefs, () => ({ db, sessionId, locale }), { maxToolResultTokens })\n      : []\n\n    const skillMgr = new SkillManager(aiDataDir)\n    skillMgr.init()\n    const toolNames = agentTools.map((t: { name: string }) => t.name)\n\n    let resolvedSkillDef: { name: string; prompt: string } | undefined\n    let resolvedSkillMenu: string | undefined\n    if (isChartCapability) {\n      const def = chartRuntime.skillDef ?? getChartCapabilitySkill(locale ?? 'zh-CN')\n      resolvedSkillDef = { name: def.name, prompt: def.prompt }\n    } else if (skillId) {\n      const def = skillMgr.getSkillConfig(skillId)\n      if (def) resolvedSkillDef = { name: def.name, prompt: def.prompt }\n    } else if (enableAutoSkill) {\n      const menu = buildSkillMenuWithBuiltinChart(\n        skillMgr.getSkillMenu(chatType ?? 'group', toolNames),\n        locale,\n        toolNames\n      )\n      if (menu) resolvedSkillMenu = menu\n    }\n\n    if (resolvedSkillMenu) {\n      const activateSkillTool = createActivateSkillTool({\n        chatType: chatType ?? 'group',\n        allowedTools: toolNames,\n        coreToolNames: new Set<string>(CHART_CAPABILITY_CORE_TOOLS),\n        locale,\n        getSkillConfig: (id) =>\n          getSkillConfigWithBuiltinChart(id, locale, (skillConfigId) => skillMgr.getSkillConfig(skillConfigId)),\n      })\n      agentTools.push(activateSkillTool as any)\n    }\n\n    const resolvedCompression = compressionConfig?.enabled\n      ? {\n          enabled: true as const,\n          tokenThresholdPercent: compressionConfig.tokenThresholdPercent ?? 75,\n          bufferSizePercent: compressionConfig.bufferSizePercent ?? 20,\n          maxToolResultPercent: compressionConfig.maxToolResultPercent,\n        }\n      : undefined\n\n    let dataSnapshot: DataSnapshot | undefined\n    if (db) {\n      try {\n        const overview = getChatOverview(db, 5)\n        if (overview) {\n          dataSnapshot = {\n            name: overview.name,\n            platform: overview.platform,\n            type: overview.type,\n            totalMessages: overview.totalMessages,\n            totalMembers: overview.totalMembers,\n            firstMessageTs: overview.firstMessageTs,\n            lastMessageTs: overview.lastMessageTs,\n            capturedAt: Math.floor(Date.now() / 1000),\n          }\n        }\n      } catch {\n        // non-fatal\n      }\n    }\n\n    await runServerAgent({\n      userMessage,\n      aiChatId,\n      historyLeafMessageId,\n      chatType,\n      locale,\n      assistantSystemPrompt,\n      skillMenu: resolvedSkillMenu,\n      skillDef: resolvedSkillDef,\n      compressionConfig: resolvedCompression,\n      tools: agentTools,\n      aiDataDir,\n      aiChatManager,\n      onEvent,\n      abortSignal,\n      ownerInfo,\n      mentionedMembers,\n      dataSnapshot,\n      thinkingLevel,\n    })\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AASA,IAAI,SAA0B;AAEvB,SAAS,mBAAmB,SAA2B;AAC5D,MAAI,CAAC,QAAQ;AACX,aAAS,IAAI,SAAS,OAAO;AAAA,EAC/B;AACA,SAAO;AACT;AAEO,SAAS,oBAAqC;AACnD,SAAO;AACT;AAEO,SAAS,sBAA4B;AAC1C,MAAI,QAAQ;AACV,WAAO,MAAM;AACb,aAAS;AAAA,EACX;AACF;;;ACLA,IAAM,iCAAiC;AAEvC,IAAM,2BAA+D;AAAA,EACnE,iBAAiB;AAAA,EACjB,sBAAsB;AAAA,EACtB,qBAAqB;AAAA,EACrB,qBAAqB;AAAA,EACrB,sBAAsB;AAAA,EACtB,0BAA0B;AAC5B;AAQA,SAAS,8BAA8B,QAAuC;AAC5E,QAAM,aAAsC,CAAC;AAC7C,aAAW,CAAC,KAAK,IAAI,KAAK,OAAO,QAAQ,OAAO,UAAU,GAAG;AAC3D,eAAW,GAAG,IAAI,EAAE,GAAG,KAAK;AAAA,EAC9B;AACA,SAAO;AAAA,IACL,MAAM;AAAA,IACN;AAAA,IACA,UAAU,OAAO,YAAY,CAAC;AAAA,EAChC;AACF;AAMO,SAAS,mBACd,OACA,YACA,SACuB;AACvB,QAAM,cAAc,SAAS,uBAAuB;AACpD,QAAM,uBAAuB,2BAA2B;AAExD,SAAO,MAAM;AAAA,IAAI,CAAC,SAChB;AAAA,MACE;AAAA,QACE,MAAM,KAAK;AAAA,QACX,OAAO,KAAK;AAAA,QACZ,aAAa,KAAK;AAAA,QAClB,YAAY,8BAA8B,KAAK,WAAW;AAAA,QAC1D,MAAM,QAAQ,aAAqB,QAAoD;AACrF,gBAAM,aAAc,UAAU,OAAO,WAAW,WAAW,SAAS,CAAC;AACrE,gBAAM,MAAM,WAAW;AACvB,gBAAM,UAAgC;AAAA,YACpC,IAAI,IAAI;AAAA,YACR,cAAc,IAAI,iBAAiB,IAAI,EAAE;AAAA,YACzC,WAAW,IAAI;AAAA,YACf,QAAQ,IAAI;AAAA,YACZ,aAAa,CAAC,OAAO,QAAQA,aAAY,0BAA0B,OAAO,QAAeA,QAAc;AAAA,UACzG;AACA,cAAI;AACF,kBAAM,SAAS,MAAM,KAAK,QAAQ,YAAY,OAAO;AACrD,kBAAM,eACJ,OAAO,SAAS,OAAO,SACnB;AAAA,cACE,GAAI,OAAO,QAAQ,EAAE,OAAO,OAAO,MAAM,IAAI,CAAC;AAAA,cAC9C,GAAI,OAAO,SAAS,EAAE,QAAQ,OAAO,OAAO,IAAI,CAAC;AAAA,YACnD,IACA,CAAC;AAEP,gBAAI,OAAO,eAAe,OAAO,YAAY,SAAS,GAAG;AACvD,oBAAM,iBAAiB,2BAA2B;AAAA,gBAChD,aAAa,OAAO;AAAA,gBACpB,QAAQ,IAAI;AAAA,gBACZ,qBAAqB;AAAA,gBACrB,oBAAoB,yBAAyB,KAAK,IAAI,KAAK;AAAA,gBAC3D,cAAe,OAAO,QAAQ,CAAC;AAAA,gBAC/B,QAAQ,kBAAkB,KAAK;AAAA,cACjC,CAAC;AACD,qBAAO;AAAA,gBACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,eAAe,KAAK,CAAC;AAAA,gBACrD,SAAS,OAAO,KAAK,YAAY,EAAE,SAAS,IAAI,eAAe;AAAA,cACjE;AAAA,YACF;AAEA,kBAAM,cACJ,OAAO,OAAO,SAAS,YAAY,OAAO,SAAS,OAC9C,OAAO,OACR,OAAO,SAAS,SACd,OACA,EAAE,OAAO,OAAO,KAAK;AAE7B,mBAAO;AAAA,cACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,OAAO,QAAQ,CAAC;AAAA,cAChD,SAAS,OAAO,KAAK,YAAY,EAAE,SAAS,IAAI,EAAE,GAAI,eAAe,CAAC,GAAI,GAAG,aAAa,IAAI;AAAA,YAChG;AAAA,UACF,SAAS,OAAO;AACd,kBAAM,MAAM,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AACjE,mBAAO,EAAE,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,UAAU,GAAG,GAAG,CAAC,GAAG,SAAS,KAAK;AAAA,UAC7E;AAAA,QACF;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;;;ACpHA,YAAY,QAAQ;AACpB,YAAY,UAAU;AAkCf,SAAS,cAAc,WAAkC;AAC9D,QAAM,aAAkB,UAAK,WAAW,iBAAiB;AACzD,MAAI,CAAI,cAAW,UAAU,GAAG;AAC9B,WAAO,EAAE,SAAS,CAAC,GAAG,kBAAkB,MAAM,WAAW,KAAK;AAAA,EAChE;AAEA,MAAI;AACF,UAAM,OAAO,KAAK,MAAS,gBAAa,YAAY,OAAO,CAAC;AAE5D,UAAM,WAAW,KAAK,WAAW,CAAC,GAAG,IAAI,CAAC,MAA+B;AACvE,YAAM,WAAY,EAAE,YAAuB;AAC3C,YAAM,cAAc,EAAE;AACtB,YAAM,aAAa,cAAc,UAAU,WAAW;AACtD,YAAM,SAAU,EAAE,UAAqB;AACvC,YAAM,cAAc,OAAO,WAAW,MAAM,IAAI,KAAK;AACrD,aAAO;AAAA,QACL,GAAG;AAAA,QACH,QAAQ,cAAc;AAAA,MACxB;AAAA,IACF,CAAC;AACD,WAAO;AAAA,MACL;AAAA,MACA,kBAAkB,KAAK,oBAAoB;AAAA,MAC3C,WAAW,KAAK,aAAa;AAAA,IAC/B;AAAA,EACF,QAAQ;AACN,WAAO,EAAE,SAAS,CAAC,GAAG,kBAAkB,MAAM,WAAW,KAAK;AAAA,EAChE;AACF;AAEA,SAAS,YAAY,MAAoC,SAA8C;AACrG,MAAI,QAAQ,QAAQ,KAAK,CAAC,MAAM,EAAE,OAAO,KAAK,QAAQ,EAAG,QAAO;AAChE,QAAM,WAAW,QAAQ,CAAC;AAC1B,MAAI,CAAC,SAAU,QAAO;AACtB,SAAO,EAAE,UAAU,SAAS,IAAI,SAAS,SAAS,MAAM;AAC1D;AAEO,SAAS,0BAA0B,WAA2C;AACnF,QAAM,QAAQ,cAAc,SAAS;AACrC,QAAM,OAAO,YAAY,MAAM,kBAAkB,MAAM,OAAO;AAC9D,MAAI,CAAC,KAAM,QAAO;AAClB,QAAM,SAAS,MAAM,QAAQ,KAAK,CAAC,MAAM,EAAE,OAAO,KAAK,QAAQ;AAC/D,MAAI,CAAC,OAAQ,QAAO;AACpB,SAAO,EAAE,GAAG,QAAQ,OAAO,KAAK,WAAW,OAAO,MAAM;AAC1D;AAqKO,SAASC,cAAa,QAAyC;AACpE,SAAO,aAAiB,MAAuB;AACjD;;;ACzPA,YAAYC,SAAQ;AACpB,YAAYC,WAAU;AAIf,SAAS,oBAAoB,WAAmB,aAA6C;AAClG,QAAM,WAAgB,WAAK,WAAW,cAAc,GAAG,WAAW,KAAK;AACvE,MAAI,CAAI,eAAW,QAAQ,EAAG,QAAO;AAErC,MAAI;AACF,UAAM,UAAa,iBAAa,UAAU,OAAO;AACjD,WAAO,mBAAmB,SAAS,QAAQ;AAAA,EAC7C,QAAQ;AACN,WAAO;AAAA,EACT;AACF;;;ACiCA,eAAsB,eAAe,SAAyC;AAC5E,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA,WAAW;AAAA,IACX,SAAS;AAAA,IACT;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,QAAQ,CAAC;AAAA,IACT;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,IAAI;AAEJ,QAAM,WAAW,kBAAkB;AAGnC,QAAM,cAAc;AAEpB,QAAM,YAAY,0BAA0B,SAAS;AACrD,MAAI,CAAC,WAAW;AACd,YAAQ,EAAE,MAAM,SAAS,OAAO,EAAE,MAAM,eAAe,SAAS,6BAA6B,EAAE,CAAC;AAChG,YAAQ,EAAE,MAAM,QAAQ,YAAY,KAAK,CAAC;AAC1C;AAAA,EACF;AAEA,QAAM,UAAUC,cAAa,SAAS;AACtC,QAAM,IAAI,kBAAkB,MAAM;AAElC,MAAI;AACJ,MAAI,UAAU;AACZ,eAAW,EAAE,SAAS;AAAA,EACxB,WAAW,WAAW;AACpB,eAAW,EAAE,UAAU;AAAA,EACzB;AAEA,QAAM,eAAe,kBAAkB;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAED,QAAM,UAAU,IAAI,kBAAkB;AAAA,IACpC,SAAS;AAAA,IACT,SAAS,CAAC;AAAA,IACV;AAAA,EACF,CAAC;AAED,MAAI,mBAAmB,WAAW,yBAAyB,QAAW;AACpE,UAAM,aAAa,4BAA4B;AAAA,MAC7C;AAAA,MACA,QAAQ,UAAU;AAAA,MAClB,eAAe,MAAM,QAAQ,WAAW,eAAe,CAAC,CAAC;AAAA,IAC3D,CAAC;AACD,UAAM,oBAAoB,MAAM;AAAA,MAC9B;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,YAAY;AAAA,IACd;AACA,QAAI,kBAAkB,YAAY;AAChC,cAAQ;AAAA,QACN,MAAM;AAAA,QACN,mBAAmB;AAAA,UACjB,gBAAgB,kBAAkB,kBAAkB;AAAA,UACpD,cAAc,kBAAkB,gBAAgB;AAAA,UAChD,aAAa,kBAAkB,eAAe;AAAA,UAC9C,WAAW,KAAK,IAAI;AAAA,QACtB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF,WAAW,mBAAmB,WAAW,yBAAyB,QAAW;AAC3E,cAAU,OAAO,eAAe,kDAAkD;AAAA,MAChF;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AAEA,MAAI,aAAa,SAAS;AACxB,YAAQ,WAAW,WAAW,CAAC,GAAG,EAAE,OAAO,KAAK,CAAC;AACjD,YAAQ,EAAE,MAAM,QAAQ,YAAY,MAAM,OAAO,QAAQ,WAAW,EAAE,CAAC;AACvE;AAAA,EACF;AAEA,MAAI,UAAkC,CAAC;AACvC,MAAI;AACF,cAAU,cAAc,mBAAmB,UAAU,QAAW,oBAAoB;AAAA,EACtF,QAAQ;AAAA,EAER;AAEA,UAAQ,WAAW,aAAa,CAAC,GAAG,EAAE,oBAAoB,aAAa,OAAO,KAAK,CAAC;AAEpF,QAAM,eAAe,EAAE,6BAA6B;AACpD,MAAI,iBAA8B,CAAC;AAEnC,MAAI;AACF,UAAM,SAAS,MAAM,aAAa;AAAA,MAChC;AAAA,MACA,QAAQ,UAAU;AAAA,MAClB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,eAAe;AAAA,MACf;AAAA,MACA;AAAA,MACA;AAAA,MACA,gBAAgB,CAAC,qBAAqB;AACpC,yBAAiB;AAAA,MACnB;AAAA,MACA,SAAS,CAAC,cAAc,QAAQ,gBAAgB,WAAW,cAAc;AAAA,MACzE,gBAAgB,CAAC,aAAa;AAC5B,YAAI;AACF,wBAAc,uBAAuB,UAAU,KAAK,UAAU,UAAU,MAAM,CAAC,CAAC;AAAA,QAClF,QAAQ;AAAA,QAER;AAAA,MACF;AAAA,IACF,CAAC;AAED,QAAI,OAAO,OAAO;AAChB,YAAM,kBAAkB,cAAc,OAAO,KAAK;AAClD,cAAQ,EAAE,MAAM,SAAS,OAAO,EAAE,MAAM,cAAc,SAAS,gBAAgB,EAAE,CAAC;AAAA,IACpF;AAEA,YAAQ,WAAW,aAAa,gBAAgB,EAAE,OAAO,KAAK,CAAC;AAC/D,YAAQ,EAAE,MAAM,QAAQ,YAAY,MAAM,OAAO,OAAO,MAAM,CAAC;AAAA,EACjE,SAAS,OAAO;AACd,UAAM,kBAAkB,cAAc,KAAK;AAC3C,cAAU,MAAM,eAAe,yBAAyB,EAAE,OAAO,OAAO,KAAK,EAAE,CAAC;AAChF,YAAQ,WAAW,SAAS,gBAAgB,EAAE,OAAO,KAAK,CAAC;AAC3D,YAAQ,EAAE,MAAM,SAAS,OAAO,EAAE,MAAM,cAAc,SAAS,gBAAgB,EAAE,CAAC;AAClF,YAAQ,EAAE,MAAM,QAAQ,YAAY,MAAM,OAAO,QAAQ,WAAW,EAAE,CAAC;AAAA,EACzE;AACF;;;ACnLA,SAAS,SAAS,WAAoC;AACpD,QAAM,eAAgB,UAAkB,cAAc;AACtD,MAAI,CAAC,cAAc;AACjB,UAAM,OAAO,OAAO,IAAI,MAAM,4BAA4B,GAAG,EAAE,YAAY,IAAI,CAAC;AAAA,EAClF;AACA,SAAO,aAAa,aAAa;AACnC;AAEA,IAAM,qBAAqB,oBAAI,IAAI,CAAC,aAAa,CAAC;AAE3C,SAAS,qBAAqB,mBAA4B,gBAAoC;AACnG,MAAI,mBAAmB;AACrB,UAAM,iBAAiB,IAAI,IAAY,2BAA2B;AAClE,WAAO,oBAAoB;AAAA,MACzB,CAAC,SACC,eAAe,IAAI,KAAK,IAAI,KAC3B,KAAK,aAAa,cAAc,gBAAgB,IAAI,KAAK,IAAI,KAAK,CAAC,mBAAmB,IAAI,KAAK,IAAI;AAAA,IACxG;AAAA,EACF;AAEA,SAAO,iBACH,oBAAoB,OAAO,CAAC,SAAS,KAAK,aAAa,cAAc,eAAe,IAAI,KAAK,IAAI,CAAC,IAClG;AACN;AAGO,SAAS,kBACd,mBACA,qBACoB;AACpB,MAAI,mBAAmB;AACrB,WAAO,IAAI,IAAI,0BAA0B,uBAAuB,CAAC,CAAC,CAAC;AAAA,EACrE;AACA,SAAO,wBAAwB,SAAY,OAAO,IAAI,IAAI,0BAA0B,mBAAmB,CAAC;AAC1G;AAEO,SAAS,wBACd,WACA,eACqH;AACrH,SAAO,OAAO,QAAQ,SAAS,gBAAgB;AAC7C,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAAI;AAEJ,UAAM,YAAY,SAAS,SAAS;AAEpC,QAAI;AACJ,QAAI;AACJ,QAAI,aAAa;AACf,YAAM,kBAAkB,oBAAoB,WAAW,WAAW;AAClE,UAAI,iBAAiB,cAAc;AACjC,gCAAwB,gBAAgB;AAAA,MAC1C;AACA,8BAAwB,iBAAiB;AAAA,IAC3C;AAEA,UAAM,YAAY,0BAA0B,SAAS;AACrD,UAAM,uBAAuB,mBAAmB,wBAAwB;AACxE,UAAM,gBAAgB,YAAaC,cAAa,SAAS,EAAE,iBAAiB,QAAU;AACtF,UAAM,sBAAsB,KAAK,MAAM,iBAAiB,uBAAuB,IAAI;AAEnF,UAAM,KAAM,UAAkB,OAAO,SAAS;AAC9C,UAAM,eAAe,8BAA8B;AAAA,MACjD;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AACD,UAAM,oBAAoB,aAAa;AACvC,UAAM,wBACJ,CAAC,WAAW,kBACR,wCAAwC,qBAAqB,IAC7D;AACN,UAAM,iBAAiB;AAAA,MACrB;AAAA,MACA,oBAAoB,aAAa,sBAAsB;AAAA,IACzD;AACA,UAAM,oBAAoB,qBAAqB,mBAAmB,cAAc;AAEhF,UAAM,aAAa,KACf,mBAAmB,mBAAmB,OAAO,EAAE,IAAI,WAAW,OAAO,IAAI,EAAE,oBAAoB,CAAC,IAChG,CAAC;AAEL,UAAM,WAAW,IAAI,aAAa,SAAS;AAC3C,aAAS,KAAK;AACd,UAAM,YAAY,WAAW,IAAI,CAAC,MAAwB,EAAE,IAAI;AAEhE,QAAI;AACJ,QAAI;AACJ,QAAI,mBAAmB;AACrB,YAAM,MAAM,aAAa,YAAY,wBAAwB,UAAU,OAAO;AAC9E,yBAAmB,EAAE,MAAM,IAAI,MAAM,QAAQ,IAAI,OAAO;AAAA,IAC1D,WAAW,SAAS;AAClB,YAAM,MAAM,SAAS,eAAe,OAAO;AAC3C,UAAI,IAAK,oBAAmB,EAAE,MAAM,IAAI,MAAM,QAAQ,IAAI,OAAO;AAAA,IACnE,WAAW,iBAAiB;AAC1B,YAAM,OAAO;AAAA,QACX,SAAS,aAAa,YAAY,SAAS,SAAS;AAAA,QACpD;AAAA,QACA;AAAA,MACF;AACA,UAAI,KAAM,qBAAoB;AAAA,IAChC;AAEA,QAAI,mBAAmB;AACrB,YAAM,oBAAoB,wBAAwB;AAAA,QAChD,UAAU,YAAY;AAAA,QACtB,cAAc;AAAA,QACd,eAAe,IAAI,IAAY,2BAA2B;AAAA,QAC1D;AAAA,QACA,gBAAgB,CAAC,OACf,+BAA+B,IAAI,QAAQ,CAAC,kBAAkB,SAAS,eAAe,aAAa,CAAC;AAAA,MACxG,CAAC;AACD,iBAAW,KAAK,iBAAwB;AAAA,IAC1C;AAEA,UAAM,sBAAsB,mBAAmB,UAC3C;AAAA,MACE,SAAS;AAAA,MACT,uBAAuB,kBAAkB,yBAAyB;AAAA,MAClE,mBAAmB,kBAAkB,qBAAqB;AAAA,MAC1D,sBAAsB,kBAAkB;AAAA,IAC1C,IACA;AAEJ,QAAI;AACJ,QAAI,IAAI;AACN,UAAI;AACF,cAAM,WAAW,gBAAgB,IAAI,CAAC;AACtC,YAAI,UAAU;AACZ,yBAAe;AAAA,YACb,MAAM,SAAS;AAAA,YACf,UAAU,SAAS;AAAA,YACnB,MAAM,SAAS;AAAA,YACf,eAAe,SAAS;AAAA,YACxB,cAAc,SAAS;AAAA,YACvB,gBAAgB,SAAS;AAAA,YACzB,eAAe,SAAS;AAAA,YACxB,YAAY,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAAA,UAC1C;AAAA,QACF;AAAA,MACF,QAAQ;AAAA,MAER;AAAA,IACF;AAEA,UAAM,eAAe;AAAA,MACnB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,WAAW;AAAA,MACX,UAAU;AAAA,MACV,mBAAmB;AAAA,MACnB,OAAO;AAAA,MACP;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACF;","names":["options","buildPiModel","fs","path","buildPiModel","buildPiModel"]}

package/dist/{chunk-CIIQ3S2X.mjs → chunk-YZYP5ST6.mjs}

@@ -4,10 +4,14 @@ import { fileURLToPath as __fileURLToPath } from 'url';
 const require = __createRequire(import.meta.url);
 const __filename = __fileURLToPath(import.meta.url);
 const __dirname = __pathDirname(__filename);
+import {
+  assertCliDataDirCompatible
+} from "./chunk-FDK6SPCG.mjs";
 import {
   getVersion
 } from "./chunk-5ADKYYNA.mjs";
 import {
+  AIChatManager,
   DEFAULT_API_PORT,
   DatabaseManager,
   NodePathProvider,
@@ -22,7 +26,7 @@ import {
   loadConfig,
   searchMessagesLike,
   verifyCliDataPath
-} from "./chunk-N5DGZQB4.mjs";
+} from "./chunk-7VG6QGQ2.mjs";
 import {
   resolveCliPath
 } from "./chunk-FJ35V3TE.mjs";
@@ -219,7 +223,7 @@ program.command("import <file>").description("Import a chat history file (14+ fo
     console.error(`File not found: ${file}`);
     process.exit(1);
   }
-  const { streamImport, detectFormat } = await import("./import-B745DRHD.mjs");
+  const { streamImport, detectFormat } = await import("./import-VS7L5JDX.mjs");
   const { dbManager } = initRuntime();
   const nativeBinding = resolveNativeBinding();
   const format = detectFormat(file);
@@ -261,7 +265,7 @@ Import error: ${err instanceof Error ? err.message : err}`);
   }
 });
 program.command("formats").description("List all supported chat history formats").action(async () => {
-  const { getSupportedFormats } = await import("./import-B745DRHD.mjs");
+  const { getSupportedFormats } = await import("./import-VS7L5JDX.mjs");
   const formats = getSupportedFormats();
   console.log(`${formats.length} supported format(s):
 `);
@@ -270,9 +274,33 @@ program.command("formats").description("List all supported chat history formats"
   }
 });
 program.command("mcp").description("Start MCP Server (stdio transport, for ClaudeCode / Cursor / AI agents)").action(async () => {
-  const { startCliMcpServer } = await import("./mcp-YWAVG5XW.mjs");
+  const { startCliMcpServer } = await import("./mcp-6VXDPUQK.mjs");
   await startCliMcpServer();
 });
+program.command("chat").description("Ask ChatLab AI about an imported chat session").option("--session-id <id>", "Source chat session ID").option("--ai-chat-id <id>", "Existing AI chat ID to continue").option("-q, --question <text>", "Question to ask").option("--json", "Output structured JSON").option("--no-stream", "Disable streaming output").option("--locale <locale>", "AI response locale", "zh-CN").action(async (options) => {
+  const { runChatCommand } = await import("./chat-command-5XMRLXZM.mjs");
+  const { dbManager, pathProvider } = initRuntime();
+  const aiChatManager = new AIChatManager(pathProvider.getAiDataDir(), { nativeBinding: resolveNativeBinding() });
+  try {
+    await runChatCommand(
+      {
+        sessionId: options.sessionId,
+        aiChatId: options.aiChatId,
+        question: options.question,
+        json: !!options.json,
+        stream: options.stream,
+        locale: options.locale
+      },
+      { dbManager, pathProvider, aiChatManager }
+    );
+  } catch (error) {
+    console.error(error instanceof Error ? error.message : String(error));
+    process.exitCode = 1;
+  } finally {
+    aiChatManager.close();
+    dbManager.closeAll();
+  }
+});
 program.command("start").description("Start ChatLab (HTTP API + Web UI)").option("--port <port>", "Server port", String(DEFAULT_API_PORT)).option("--host <host>", "Listen address", "127.0.0.1").option("--token <token>", "Custom Bearer Token (reads from config or auto-generates if omitted)").option("--headless", "API-only mode, do not serve the Web UI").option("--require-auth", "Require Bearer token for all routes including /_web/*").option("--no-open", "Do not auto-open the browser").option("--daemon", "Run as a resident system service (auto-start on login, macOS/Linux)").action(async (options) => {
   if (options.daemon) {
     const { serviceInstall } = await import("./service-KZTHBDQ5.mjs");
@@ -285,7 +313,7 @@ program.command("start").description("Start ChatLab (HTTP API + Web UI)").option
     });
     return;
   }
-  const { startHttpServer } = await import("./http-TPX74BLK.mjs");
+  const { startHttpServer } = await import("./http-C3CGXUBU.mjs");
   const port = parseInt(options.port, 10);
   let webRoot;
   if (!options.headless) {
@@ -332,7 +360,7 @@ Press Ctrl+C to stop.
     }
     const shutdown = async () => {
       console.log("\nShutting down...");
-      const { stopHttpServer } = await import("./http-TPX74BLK.mjs");
+      const { stopHttpServer } = await import("./http-C3CGXUBU.mjs");
       await stopHttpServer();
       process.exit(0);
     };
@@ -409,12 +437,13 @@ function initRuntime() {
   const userDataDir = config.data.user_data_dir || void 0;
   const pathProvider = new NodePathProvider(userDataDir);
   pathProvider.ensureAllDirs();
+  const runtime = assertCliDataDirCompatible(pathProvider, "cli");
   if (hasPendingElectronDataWarning() || !verifyCliDataPath(pathProvider.getDatabaseDir())) {
     printElectronDataError();
     process.exit(1);
   }
   const nativeBinding = resolveNativeBinding();
-  const dbManager = new DatabaseManager(pathProvider, { nativeBinding });
+  const dbManager = new DatabaseManager(pathProvider, { nativeBinding, runtime });
   return { config, pathProvider, dbManager };
 }
 function printElectronDataError() {
@@ -475,4 +504,4 @@ if (isDirectRun) {
 export {
   run
 };
-//# sourceMappingURL=chunk-CIIQ3S2X.mjs.map
+//# sourceMappingURL=chunk-YZYP5ST6.mjs.map