chatlab-cli 0.25.0 → 0.25.1

package/dist/{chunk-GNT5G74Z.mjs → chunk-PDQFSQO6.mjs}

@@ -5,48 +5,29 @@ const require = __createRequire(import.meta.url);
 const __filename = __fileURLToPath(import.meta.url);
 const __dirname = __pathDirname(__filename);
 import {
-  CHAT_DB_TABLES,
+  DataDirCompatibilityError,
   analyzeIncrementalImport,
   analyzeNewImport,
   generateIncrementalSessionIndex,
   generateSessionIndex,
   incrementalImport,
-  openBetterSqliteDatabase,
   streamingImport
-} from "./chunk-4OMV4VNH.mjs";
+} from "./chunk-7VG6QGQ2.mjs";
 
 // src/import/stream-import.ts
-import * as fs from "fs";
-import * as path from "path";
 import * as crypto from "crypto";
 function generateSessionId() {
   const ts = Date.now();
   const rand = crypto.randomBytes(4).toString("hex");
   return `chat_${ts}_${rand}`;
 }
-function resolveNativeBinding(dbManager) {
-  return dbManager.nativeBinding;
-}
 function buildStreamImportDeps(dbManager, onProgress) {
-  const nativeBinding = resolveNativeBinding(dbManager);
   return {
     openDatabase(sessionId) {
-      const dbPath = dbManager.getDbPath(sessionId);
-      const dir = path.dirname(dbPath);
-      if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
-      const db = openBetterSqliteDatabase(dbPath, { nativeBinding });
-      db.exec(CHAT_DB_TABLES);
-      return db;
+      return dbManager.openRawSessionDatabase(sessionId, { create: true, initializeChatTables: true });
     },
     deleteDatabase(sessionId) {
-      const dbPath = dbManager.getDbPath(sessionId);
-      for (const suffix of ["", "-wal", "-shm"]) {
-        try {
-          const p = dbPath + suffix;
-          if (fs.existsSync(p)) fs.unlinkSync(p);
-        } catch {
-        }
-      }
+      dbManager.deleteSessionDatabaseFiles(sessionId);
     },
     onProgress: onProgress ?? (() => {
     }),
@@ -59,6 +40,9 @@ function buildStreamImportDeps(dbManager, onProgress) {
     generateSessionId
   };
 }
+function deleteSessionDatabase(dbManager, sessionId) {
+  dbManager.deleteSessionDatabaseFiles(sessionId);
+}
 async function streamImport(dbManager, filePath, options) {
   const { formatId, chatIndex, onProgress } = options || {};
   const formatOptions = {};
@@ -101,17 +85,29 @@ async function streamImport(dbManager, filePath, options) {
   } : () => {
   };
   const deps = buildStreamImportDeps(dbManager, progressAdapter);
-  return streamingImport(filePath, deps, formatOptions);
+  const result = await streamingImport(filePath, deps, formatOptions);
+  if (!result.success || !result.sessionId) return result;
+  try {
+    dbManager.raiseCurrentChatDbCompatibilityGate();
+  } catch (error) {
+    deleteSessionDatabase(dbManager, result.sessionId);
+    return {
+      success: false,
+      error: error instanceof Error ? error.message : String(error),
+      diagnostics: result.diagnostics
+    };
+  }
+  return result;
 }
-function buildIncrementalDeps(dbManager, onProgress) {
-  const nativeBinding = resolveNativeBinding(dbManager);
+function buildIncrementalDeps(dbManager, onProgress, onCompatibilityError) {
   return {
     openDatabase(sessionId, readonly) {
-      const dbPath = dbManager.getDbPath(sessionId);
-      if (!fs.existsSync(dbPath)) {
-        throw new Error(`Session database not found: ${sessionId}`);
+      try {
+        return dbManager.openRawSessionDatabase(sessionId, { readonly: readonly ?? false });
+      } catch (error) {
+        if (error instanceof DataDirCompatibilityError) onCompatibilityError?.(error);
+        throw error;
       }
-      return openBetterSqliteDatabase(dbPath, { readonly: readonly ?? false, nativeBinding });
     },
     onProgress: onProgress ?? (() => {
     }),
@@ -125,10 +121,40 @@ function buildIncrementalDeps(dbManager, onProgress) {
 }
 async function incrementalImport2(dbManager, sessionId, filePath, options) {
   const { onProgress, ...importOpts } = options || {};
-  return incrementalImport(sessionId, filePath, buildIncrementalDeps(dbManager, onProgress), importOpts);
+  let compatibilityError = null;
+  const result = await incrementalImport(
+    sessionId,
+    filePath,
+    buildIncrementalDeps(dbManager, onProgress, (error) => {
+      compatibilityError = error;
+    }),
+    importOpts
+  );
+  if (compatibilityError) throw compatibilityError;
+  if (!result.success) return result;
+  try {
+    dbManager.raiseCurrentChatDbCompatibilityGate();
+  } catch (error) {
+    return {
+      ...result,
+      success: false,
+      newMessageCount: 0,
+      error: error instanceof Error ? error.message : String(error)
+    };
+  }
+  return result;
 }
 async function analyzeIncrementalImport2(dbManager, sessionId, filePath, onProgress) {
-  return analyzeIncrementalImport(sessionId, filePath, buildIncrementalDeps(dbManager, onProgress));
+  let compatibilityError = null;
+  const result = await analyzeIncrementalImport(
+    sessionId,
+    filePath,
+    buildIncrementalDeps(dbManager, onProgress, (error) => {
+      compatibilityError = error;
+    })
+  );
+  if (compatibilityError) throw compatibilityError;
+  return result;
 }
 async function analyzeNewImport2(filePath, onProgress) {
   return analyzeNewImport(filePath, onProgress ?? (() => {
@@ -141,4 +167,4 @@ export {
   analyzeIncrementalImport2 as analyzeIncrementalImport,
   analyzeNewImport2 as analyzeNewImport
 };
-//# sourceMappingURL=chunk-GNT5G74Z.mjs.map
+//# sourceMappingURL=chunk-PDQFSQO6.mjs.map

package/dist/chunk-PDQFSQO6.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/import/stream-import.ts"],"sourcesContent":["/**\n * Server/CLI streaming import — adapter for @openchatlab/node-runtime StreamingImporter.\n *\n * Replaces the old buffered-in-memory approach with the same high-performance\n * streaming pipeline used by Electron (batched transactions, deferred indexes,\n * nickname history, FTS, format fallback).\n */\n\nimport type { DatabaseManager } from '@openchatlab/node-runtime'\nimport {\n  DataDirCompatibilityError,\n  streamingImport,\n  analyzeNewImport as sharedAnalyzeNewImport,\n  analyzeIncrementalImport as sharedAnalyzeIncremental,\n  incrementalImport as sharedIncrementalImport,\n} from '@openchatlab/node-runtime'\nimport { generateSessionIndex, generateIncrementalSessionIndex } from '@openchatlab/core'\nimport type {\n  StreamImportResult,\n  StreamImportDeps,\n  ImportProgressCallback,\n  IncrementalImportResult,\n  IncrementalAnalyzeResult,\n  IncrementalImportDeps,\n  ImportOptions,\n  AnalyzeNewImportResult,\n} from '@openchatlab/node-runtime'\nimport {\n  detectFormat as parserDetectFormat,\n  detectAllFormats,\n  getFormatFeatureById,\n  getSupportedFormats as parserGetSupportedFormats,\n  scanMultiChatFile as parserScanMultiChatFile,\n  findEntryFileInDirectory,\n  type FormatFeature,\n  type MultiChatInfo,\n  type ParseProgress,\n} from '@openchatlab/parser'\nimport * as crypto from 'crypto'\n\n// ==================== Legacy progress interface (for SSE routes) ====================\n\nexport interface StreamImportProgress {\n  stage: 'detecting' | 'parsing' | 'saving' | 'indexing' | 'done' | 'error'\n  progress: number\n  message: string\n  bytesRead?: number\n  totalBytes?: number\n  messagesProcessed?: number\n}\n\nexport interface StreamImportOptions {\n  formatId?: string\n  chatIndex?: number\n  nativeBinding?: string\n  onProgress?: (progress: StreamImportProgress) => void\n}\n\nfunction generateSessionId(): string {\n  const ts = Date.now()\n  const rand = crypto.randomBytes(4).toString('hex')\n  return `chat_${ts}_${rand}`\n}\n\nfunction buildStreamImportDeps(dbManager: DatabaseManager, onProgress?: ImportProgressCallback): StreamImportDeps {\n  return {\n    openDatabase(sessionId: string) {\n      return dbManager.openRawSessionDatabase(sessionId, { create: true, initializeChatTables: true })\n    },\n    deleteDatabase(sessionId: string) {\n      dbManager.deleteSessionDatabaseFiles(sessionId)\n    },\n    onProgress: onProgress ?? (() => {}),\n    postImportHook(db) {\n      try {\n        generateSessionIndex(db)\n      } catch {\n        /* non-fatal: frontend can regenerate via generate-index route */\n      }\n    },\n    generateSessionId,\n  }\n}\n\nfunction deleteSessionDatabase(dbManager: DatabaseManager, sessionId: string): void {\n  dbManager.deleteSessionDatabaseFiles(sessionId)\n}\n\n/**\n * High-performance streaming import: parse a file and write to DB\n * with batched transactions, deferred indexes, and FTS.\n */\nexport async function streamImport(\n  dbManager: DatabaseManager,\n  filePath: string,\n  options?: StreamImportOptions\n): Promise<StreamImportResult> {\n  const { formatId, chatIndex, onProgress } = options || {}\n\n  const formatOptions: Record<string, unknown> = {}\n  if (formatId) formatOptions.formatId = formatId\n  if (chatIndex !== undefined) formatOptions.chatIndex = chatIndex\n\n  const progressAdapter: ImportProgressCallback = onProgress\n    ? (progress) => {\n        let stage: StreamImportProgress['stage'] = 'parsing'\n        let pct = 0\n        switch (progress.stage) {\n          case 'detecting':\n            stage = 'detecting'\n            pct = 5\n            break\n          case 'parsing':\n            stage = 'parsing'\n            pct = Math.min(Math.round(progress.percentage * 0.7), 70)\n            break\n          case 'importing':\n          case 'saving':\n            stage = 'saving'\n            pct = 80\n            break\n          case 'done':\n            stage = 'done'\n            pct = 100\n            break\n          case 'error':\n            stage = 'error'\n            pct = 0\n            break\n        }\n        onProgress({\n          stage,\n          progress: pct,\n          message: progress.message || '',\n          bytesRead: progress.bytesRead,\n          totalBytes: progress.totalBytes,\n          messagesProcessed: progress.messagesProcessed,\n        })\n      }\n    : () => {}\n\n  const deps = buildStreamImportDeps(dbManager, progressAdapter)\n  const result = await streamingImport(filePath, deps, formatOptions)\n  if (!result.success || !result.sessionId) return result\n\n  try {\n    dbManager.raiseCurrentChatDbCompatibilityGate()\n  } catch (error) {\n    deleteSessionDatabase(dbManager, result.sessionId)\n    return {\n      success: false,\n      error: error instanceof Error ? error.message : String(error),\n      diagnostics: result.diagnostics,\n    }\n  }\n\n  return result\n}\n\n// ==================== Incremental import ====================\n\nfunction buildIncrementalDeps(\n  dbManager: DatabaseManager,\n  onProgress?: ImportProgressCallback,\n  onCompatibilityError?: (error: DataDirCompatibilityError) => void\n): IncrementalImportDeps {\n  return {\n    openDatabase(sessionId: string, readonly?: boolean) {\n      try {\n        return dbManager.openRawSessionDatabase(sessionId, { readonly: readonly ?? false })\n      } catch (error) {\n        if (error instanceof DataDirCompatibilityError) onCompatibilityError?.(error)\n        throw error\n      }\n    },\n    onProgress: onProgress ?? (() => {}),\n    postImportHook(db) {\n      try {\n        generateIncrementalSessionIndex(db)\n      } catch {\n        /* non-fatal */\n      }\n    },\n  }\n}\n\nexport async function incrementalImport(\n  dbManager: DatabaseManager,\n  sessionId: string,\n  filePath: string,\n  options?: ImportOptions & { onProgress?: ImportProgressCallback }\n): Promise<IncrementalImportResult> {\n  const { onProgress, ...importOpts } = options || {}\n  let compatibilityError: DataDirCompatibilityError | null = null\n  const result = await sharedIncrementalImport(\n    sessionId,\n    filePath,\n    buildIncrementalDeps(dbManager, onProgress, (error) => {\n      compatibilityError = error\n    }),\n    importOpts\n  )\n  if (compatibilityError) throw compatibilityError\n  if (!result.success) return result\n\n  try {\n    dbManager.raiseCurrentChatDbCompatibilityGate()\n  } catch (error) {\n    return {\n      ...result,\n      success: false,\n      newMessageCount: 0,\n      error: error instanceof Error ? error.message : String(error),\n    }\n  }\n\n  return result\n}\n\nexport async function analyzeIncrementalImport(\n  dbManager: DatabaseManager,\n  sessionId: string,\n  filePath: string,\n  onProgress?: ImportProgressCallback\n): Promise<IncrementalAnalyzeResult> {\n  let compatibilityError: DataDirCompatibilityError | null = null\n  const result = await sharedAnalyzeIncremental(\n    sessionId,\n    filePath,\n    buildIncrementalDeps(dbManager, onProgress, (error) => {\n      compatibilityError = error\n    })\n  )\n  if (compatibilityError) throw compatibilityError\n  return result\n}\n\nexport async function analyzeNewImport(\n  filePath: string,\n  onProgress?: ImportProgressCallback\n): Promise<AnalyzeNewImportResult> {\n  return sharedAnalyzeNewImport(filePath, onProgress ?? (() => {}))\n}\n\n// ==================== Re-exports from parser ====================\n\nexport {\n  parserDetectFormat as detectFormat,\n  detectAllFormats,\n  getFormatFeatureById,\n  parserGetSupportedFormats as getSupportedFormats,\n  parserScanMultiChatFile as scanMultiChatFile,\n  findEntryFileInDirectory,\n}\nexport type { FormatFeature, MultiChatInfo, ParseProgress }\nexport type {\n  StreamImportResult,\n  IncrementalImportResult,\n  IncrementalAnalyzeResult,\n  AnalyzeNewImportResult,\n  ImportOptions,\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAsCA,YAAY,YAAY;AAoBxB,SAAS,oBAA4B;AACnC,QAAM,KAAK,KAAK,IAAI;AACpB,QAAM,OAAc,mBAAY,CAAC,EAAE,SAAS,KAAK;AACjD,SAAO,QAAQ,EAAE,IAAI,IAAI;AAC3B;AAEA,SAAS,sBAAsB,WAA4B,YAAuD;AAChH,SAAO;AAAA,IACL,aAAa,WAAmB;AAC9B,aAAO,UAAU,uBAAuB,WAAW,EAAE,QAAQ,MAAM,sBAAsB,KAAK,CAAC;AAAA,IACjG;AAAA,IACA,eAAe,WAAmB;AAChC,gBAAU,2BAA2B,SAAS;AAAA,IAChD;AAAA,IACA,YAAY,eAAe,MAAM;AAAA,IAAC;AAAA,IAClC,eAAe,IAAI;AACjB,UAAI;AACF,6BAAqB,EAAE;AAAA,MACzB,QAAQ;AAAA,MAER;AAAA,IACF;AAAA,IACA;AAAA,EACF;AACF;AAEA,SAAS,sBAAsB,WAA4B,WAAyB;AAClF,YAAU,2BAA2B,SAAS;AAChD;AAMA,eAAsB,aACpB,WACA,UACA,SAC6B;AAC7B,QAAM,EAAE,UAAU,WAAW,WAAW,IAAI,WAAW,CAAC;AAExD,QAAM,gBAAyC,CAAC;AAChD,MAAI,SAAU,eAAc,WAAW;AACvC,MAAI,cAAc,OAAW,eAAc,YAAY;AAEvD,QAAM,kBAA0C,aAC5C,CAAC,aAAa;AACZ,QAAI,QAAuC;AAC3C,QAAI,MAAM;AACV,YAAQ,SAAS,OAAO;AAAA,MACtB,KAAK;AACH,gBAAQ;AACR,cAAM;AACN;AAAA,MACF,KAAK;AACH,gBAAQ;AACR,cAAM,KAAK,IAAI,KAAK,MAAM,SAAS,aAAa,GAAG,GAAG,EAAE;AACxD;AAAA,MACF,KAAK;AAAA,MACL,KAAK;AACH,gBAAQ;AACR,cAAM;AACN;AAAA,MACF,KAAK;AACH,gBAAQ;AACR,cAAM;AACN;AAAA,MACF,KAAK;AACH,gBAAQ;AACR,cAAM;AACN;AAAA,IACJ;AACA,eAAW;AAAA,MACT;AAAA,MACA,UAAU;AAAA,MACV,SAAS,SAAS,WAAW;AAAA,MAC7B,WAAW,SAAS;AAAA,MACpB,YAAY,SAAS;AAAA,MACrB,mBAAmB,SAAS;AAAA,IAC9B,CAAC;AAAA,EACH,IACA,MAAM;AAAA,EAAC;AAEX,QAAM,OAAO,sBAAsB,WAAW,eAAe;AAC7D,QAAM,SAAS,MAAM,gBAAgB,UAAU,MAAM,aAAa;AAClE,MAAI,CAAC,OAAO,WAAW,CAAC,OAAO,UAAW,QAAO;AAEjD,MAAI;AACF,cAAU,oCAAoC;AAAA,EAChD,SAAS,OAAO;AACd,0BAAsB,WAAW,OAAO,SAAS;AACjD,WAAO;AAAA,MACL,SAAS;AAAA,MACT,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,MAC5D,aAAa,OAAO;AAAA,IACtB;AAAA,EACF;AAEA,SAAO;AACT;AAIA,SAAS,qBACP,WACA,YACA,sBACuB;AACvB,SAAO;AAAA,IACL,aAAa,WAAmB,UAAoB;AAClD,UAAI;AACF,eAAO,UAAU,uBAAuB,WAAW,EAAE,UAAU,YAAY,MAAM,CAAC;AAAA,MACpF,SAAS,OAAO;AACd,YAAI,iBAAiB,0BAA2B,wBAAuB,KAAK;AAC5E,cAAM;AAAA,MACR;AAAA,IACF;AAAA,IACA,YAAY,eAAe,MAAM;AAAA,IAAC;AAAA,IAClC,eAAe,IAAI;AACjB,UAAI;AACF,wCAAgC,EAAE;AAAA,MACpC,QAAQ;AAAA,MAER;AAAA,IACF;AAAA,EACF;AACF;AAEA,eAAsBA,mBACpB,WACA,WACA,UACA,SACkC;AAClC,QAAM,EAAE,YAAY,GAAG,WAAW,IAAI,WAAW,CAAC;AAClD,MAAI,qBAAuD;AAC3D,QAAM,SAAS,MAAM;AAAA,IACnB;AAAA,IACA;AAAA,IACA,qBAAqB,WAAW,YAAY,CAAC,UAAU;AACrD,2BAAqB;AAAA,IACvB,CAAC;AAAA,IACD;AAAA,EACF;AACA,MAAI,mBAAoB,OAAM;AAC9B,MAAI,CAAC,OAAO,QAAS,QAAO;AAE5B,MAAI;AACF,cAAU,oCAAoC;AAAA,EAChD,SAAS,OAAO;AACd,WAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS;AAAA,MACT,iBAAiB;AAAA,MACjB,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,IAC9D;AAAA,EACF;AAEA,SAAO;AACT;AAEA,eAAsBC,0BACpB,WACA,WACA,UACA,YACmC;AACnC,MAAI,qBAAuD;AAC3D,QAAM,SAAS,MAAM;AAAA,IACnB;AAAA,IACA;AAAA,IACA,qBAAqB,WAAW,YAAY,CAAC,UAAU;AACrD,2BAAqB;AAAA,IACvB,CAAC;AAAA,EACH;AACA,MAAI,mBAAoB,OAAM;AAC9B,SAAO;AACT;AAEA,eAAsBC,kBACpB,UACA,YACiC;AACjC,SAAO,iBAAuB,UAAU,eAAe,MAAM;AAAA,EAAC,EAAE;AAClE;","names":["incrementalImport","analyzeIncrementalImport","analyzeNewImport"]}

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