chatlab-cli 0.28.1 → 0.30.0

package/dist/{chunk-3QBVO4J7.mjs → chunk-5UBHNQFZ.mjs}

@@ -6,6 +6,7 @@ const __filename = __fileURLToPath(import.meta.url);
 const __dirname = __pathDirname(__filename);
 import {
   buildChartPayload,
+  computeResponseTimeStats,
   executeParameterizedSql,
   executeReadonlySql,
   getChatOverview,
@@ -18,15 +19,22 @@ import {
   getMembers,
   getMembersWithAliases,
   getMessageContext,
-  getRecentMessages,
+  getMonthlyActivity,
   getSearchMessageContext,
   getSegmentMessages,
   getSegmentSummaries,
   getSessionMeta,
   getSessionOverview,
   getWeekdayActivity,
+  queryMessages,
   searchMessagesByKeywords
-} from "./chunk-ZVDOFWL7.mjs";
+} from "./chunk-7BZLS2KW.mjs";
+
+// ../../packages/tools/src/utils/limits.ts
+function resolveMessageLimit(requestedLimit, defaultLimit, maxMessagesLimit) {
+  const requested = typeof requestedLimit === "number" && requestedLimit > 0 ? requestedLimit : defaultLimit;
+  return maxMessagesLimit && maxMessagesLimit > 0 ? Math.min(requested, maxMessagesLimit) : requested;
+}
 
 // ../../packages/tools/src/definitions/sql-query.ts
 var inputSchema = {
@@ -34,15 +42,31 @@ var inputSchema = {
   properties: {
     sql: {
       type: "string",
-      description: "\u8981\u6267\u884C\u7684 SELECT SQL \u67E5\u8BE2\u8BED\u53E5"
+      description: "\u8981\u6267\u884C\u7684 SELECT SQL \u67E5\u8BE2\u8BED\u53E5\u3002\u5EFA\u8BAE\u663E\u5F0F\u6DFB\u52A0 LIMIT\uFF0C\u5DE5\u5177\u8F93\u51FA\u4ECD\u4F1A\u88AB max_rows \u622A\u65AD\u3002"
+    },
+    max_rows: {
+      type: "number",
+      description: "\u6700\u591A\u8FD4\u56DE\u591A\u5C11\u884C\uFF0C\u9ED8\u8BA4 1000\uFF1B\u8FD8\u4F1A\u53D7\u8C03\u7528\u4E0A\u4E0B\u6587\u7684\u6D88\u606F\u4E0A\u9650\u9650\u5236\u3002"
     }
   },
   required: ["sql"]
 };
+function clampSqlResultRows(result, maxRows) {
+  if (!result || typeof result !== "object" || maxRows <= 0) return result;
+  const resultLike = result;
+  if (!Array.isArray(resultLike.rows) || resultLike.rows.length <= maxRows) return result;
+  return {
+    ...resultLike,
+    rows: resultLike.rows.slice(0, maxRows),
+    rowCount: maxRows,
+    truncated: true
+  };
+}
 async function handler(params, context) {
   const sql = params.sql;
+  const maxRows = resolveMessageLimit(params.max_rows, 1e3, context.maxMessagesLimit);
   try {
-    const result = await context.dataProvider.executeSql(sql);
+    const result = clampSqlResultRows(await context.dataProvider.executeSql(sql, { maxRows }), maxRows);
     return {
       content: JSON.stringify(result),
       data: result
@@ -66,7 +90,7 @@ async function schemaHandler(_params, context) {
 }
 var sqlQueryTool = {
   name: "execute_sql",
-  description: "\u5BF9\u804A\u5929\u6570\u636E\u5E93\u6267\u884C\u53EA\u8BFB SELECT \u67E5\u8BE2\u3002\u4F7F\u7528\u524D\u53EF\u5148\u8C03\u7528 get_schema \u67E5\u770B\u8868\u7ED3\u6784\u3002",
+  description: "\u5BF9\u804A\u5929\u6570\u636E\u5E93\u6267\u884C\u53EA\u8BFB SELECT \u67E5\u8BE2\u3002\u4F7F\u7528\u524D\u53EF\u5148\u8C03\u7528 get_schema \u67E5\u770B\u8868\u7ED3\u6784\u3002\u5EFA\u8BAE SQL \u663E\u5F0F\u6DFB\u52A0 LIMIT\uFF1B\u5DE5\u5177\u4F1A\u6309 max_rows \u622A\u65AD\u8F93\u51FA\uFF0C\u907F\u514D\u8FD4\u56DE\u8FC7\u5927\u7ED3\u679C\u96C6\u3002",
   inputSchema,
   handler,
   category: "analysis"
@@ -394,6 +418,32 @@ var timeParamProperties = {
   }
 };
 
+// ../../packages/tools/src/definitions/search-context.ts
+function trimMessagesPreservingHits(messages, hitIds, limit) {
+  if (limit <= 0) return [];
+  if (messages.length <= limit) return messages;
+  const requiredIds = new Set(hitIds);
+  if (requiredIds.size === 0) return messages.slice(0, limit);
+  const trimmed = messages.slice(0, limit);
+  const presentIds = new Set(trimmed.map((message) => message.id).filter((id) => id != null));
+  const missingHits = messages.filter(
+    (message) => message.id != null && requiredIds.has(message.id) && !presentIds.has(message.id)
+  );
+  for (const hit of missingHits) {
+    let replaceIndex = -1;
+    for (let i = trimmed.length - 1; i >= 0; i--) {
+      const id = trimmed[i].id;
+      if (id == null || !requiredIds.has(id)) {
+        replaceIndex = i;
+        break;
+      }
+    }
+    if (replaceIndex === -1) break;
+    trimmed[replaceIndex] = hit;
+  }
+  return trimmed.sort((a, b) => messages.indexOf(a) - messages.indexOf(b));
+}
+
 // ../../packages/tools/src/definitions/search-messages.ts
 var inputSchema6 = {
   type: "object",
@@ -408,7 +458,7 @@ var inputSchema6 = {
 async function handler6(params, context) {
   const { locale, timeFilter: contextTimeFilter, maxMessagesLimit } = context;
   const keywords = params.keywords;
-  const limit = Math.min(maxMessagesLimit || params.limit || 1e3, 5e4);
+  const limit = Math.min(resolveMessageLimit(params.limit, 1e3, maxMessagesLimit), 5e4);
   const effectiveTimeFilter = parseExtendedTimeParams(params, contextTimeFilter);
   const result = await context.dataProvider.searchMessages(keywords, {
     timeFilter: effectiveTimeFilter,
@@ -418,12 +468,13 @@ async function handler6(params, context) {
   const contextBefore = context.searchContextBefore ?? 2;
   const contextAfter = context.searchContextAfter ?? 2;
   let finalMessages = result.messages;
+  const hitIds = result.messages.map((m) => m.id).filter((id) => id != null);
   if ((contextBefore > 0 || contextAfter > 0) && result.messages.length > 0) {
-    const hitIds = result.messages.map((m) => m.id).filter((id) => id != null);
     if (hitIds.length > 0) {
       finalMessages = await context.dataProvider.getSearchMessageContext(hitIds, contextBefore, contextAfter);
     }
   }
+  finalMessages = trimMessagesPreservingHits(finalMessages, hitIds, limit);
   const data = {
     total: result.total,
     returned: finalMessages.length,
@@ -455,7 +506,7 @@ var inputSchema7 = {
 async function handler7(params, context) {
   const { locale, timeFilter: contextTimeFilter, maxMessagesLimit } = context;
   const keywords = params.keywords;
-  const limit = Math.min(maxMessagesLimit || params.limit || 1e3, 5e4);
+  const limit = Math.min(resolveMessageLimit(params.limit, 1e3, maxMessagesLimit), 5e4);
   const effectiveTimeFilter = parseExtendedTimeParams(params, contextTimeFilter);
   const result = await context.dataProvider.deepSearchMessages(keywords, {
     timeFilter: effectiveTimeFilter,
@@ -465,12 +516,13 @@ async function handler7(params, context) {
   const contextBefore = context.searchContextBefore ?? 2;
   const contextAfter = context.searchContextAfter ?? 2;
   let finalMessages = result.messages;
+  const hitIds = result.messages.map((m) => m.id).filter((id) => id != null);
   if ((contextBefore > 0 || contextAfter > 0) && result.messages.length > 0) {
-    const hitIds = result.messages.map((m) => m.id).filter((id) => id != null);
     if (hitIds.length > 0) {
       finalMessages = await context.dataProvider.getSearchMessageContext(hitIds, contextBefore, contextAfter);
     }
   }
+  finalMessages = trimMessagesPreservingHits(finalMessages, hitIds, limit);
   const data = {
     total: result.total,
     returned: finalMessages.length,
@@ -499,8 +551,16 @@ var inputSchema8 = {
 };
 async function handler8(params, context) {
   const { locale } = context;
-  const messageIds = params.message_ids;
-  const contextSize = params.context_size || 20;
+  const maxMessagesLimit = context.maxMessagesLimit;
+  const requestedMessageIds = params.message_ids;
+  const messageIds = maxMessagesLimit && maxMessagesLimit > 0 ? requestedMessageIds.slice(0, maxMessagesLimit) : requestedMessageIds;
+  const requestedContextSize = params.context_size;
+  let contextSize = typeof requestedContextSize === "number" && Number.isFinite(requestedContextSize) ? Math.max(0, Math.floor(requestedContextSize)) : 20;
+  if (maxMessagesLimit && maxMessagesLimit > 0 && messageIds.length > 0) {
+    const maxMessagesPerId = Math.max(1, Math.floor(maxMessagesLimit / messageIds.length));
+    const maxContextSize = Math.max(0, Math.floor((maxMessagesPerId - 1) / 2));
+    contextSize = Math.min(contextSize, maxContextSize);
+  }
   const messages = await context.dataProvider.getMessageContext(messageIds, contextSize);
   if (messages.length === 0) {
     const data2 = {
@@ -537,7 +597,7 @@ var inputSchema9 = {
 };
 async function handler9(params, context) {
   const { locale, maxMessagesLimit } = context;
-  const limit = maxMessagesLimit || params.limit || 1e3;
+  const limit = resolveMessageLimit(params.limit, 1e3, maxMessagesLimit);
   const result = await context.dataProvider.getSegmentMessages(params.segment_id, limit);
   if (!result) {
     const data2 = {
@@ -664,12 +724,14 @@ var inputSchema12 = {
       description: "\u7EDF\u8BA1\u7C7B\u578B\uFF1Ahourly\uFF08\u6309\u5C0F\u65F6\uFF09\u3001weekday\uFF08\u6309\u661F\u671F\uFF09\u3001daily\uFF08\u6309\u5929\uFF09",
       enum: ["hourly", "weekday", "daily"],
       default: "hourly"
-    }
+    },
+    ...timeParamProperties
   }
 };
 async function handler12(params, context) {
   const type = params.type || "hourly";
-  const data = await context.dataProvider.getTimeStats(type, { timeFilter: context.timeFilter });
+  const effectiveTimeFilter = parseExtendedTimeParams(params, context.timeFilter);
+  const data = await context.dataProvider.getTimeStats(type, { timeFilter: effectiveTimeFilter });
   return {
     content: JSON.stringify({ type, data }),
     data
@@ -697,7 +759,7 @@ var inputSchema13 = {
 };
 async function handler13(params, context) {
   const { locale, timeFilter: contextTimeFilter, maxMessagesLimit } = context;
-  const limit = maxMessagesLimit || params.limit || 100;
+  const limit = resolveMessageLimit(params.limit, 100, maxMessagesLimit);
   const effectiveTimeFilter = parseExtendedTimeParams(params, contextTimeFilter);
   const result = await context.dataProvider.getRecentMessages({ timeFilter: effectiveTimeFilter, limit });
   const data = {
@@ -783,7 +845,7 @@ var inputSchema15 = {
 };
 async function handler15(params, context) {
   const { locale, timeFilter: contextTimeFilter, maxMessagesLimit } = context;
-  const limit = maxMessagesLimit || params.limit || 100;
+  const limit = resolveMessageLimit(params.limit, 100, maxMessagesLimit);
   const effectiveTimeFilter = parseExtendedTimeParams(params, contextTimeFilter);
   const result = await context.dataProvider.getConversationBetween(
     params.member_id_1,
@@ -904,24 +966,10 @@ async function handler17(params, context) {
     const text = isZh ? "\u8BE5\u65F6\u95F4\u8303\u56F4\u5185\u6D88\u606F\u4E0D\u8DB3\uFF0C\u65E0\u6CD5\u5206\u6790\u54CD\u5E94\u65F6\u95F4" : "Not enough messages in this time range to analyze response time";
     return { content: text, data: null };
   }
-  const responseTimes = /* @__PURE__ */ new Map();
-  for (let i = 1; i < rows.length; i++) {
-    const prev = rows[i - 1];
-    const curr = rows[i];
-    if (curr.sender_id === prev.sender_id) continue;
-    const gap = curr.ts - prev.ts;
-    if (gap < 5 || gap > 1800) continue;
-    if (!responseTimes.has(curr.sender_id)) {
-      responseTimes.set(curr.sender_id, { name: curr.name, times: [] });
-    }
-    responseTimes.get(curr.sender_id).times.push(gap);
-  }
-  const stats = [...responseTimes.entries()].map(([id, { name, times }]) => {
-    times.sort((a, b) => a - b);
-    const avg = Math.round(times.reduce((s, t2) => s + t2, 0) / times.length);
-    const median = times[Math.floor(times.length / 2)];
-    return { id, name, avgSeconds: avg, medianSeconds: median, responseCount: times.length };
-  }).filter((s) => s.responseCount >= 3).sort((a, b) => a.medianSeconds - b.medianSeconds).slice(0, topN);
+  const stats = computeResponseTimeStats(
+    rows.map((r) => ({ senderId: r.sender_id, name: r.name, ts: r.ts })),
+    topN
+  );
   if (stats.length === 0) {
     const text = isZh ? "\u6CA1\u6709\u8DB3\u591F\u7684\u54CD\u5E94\u6570\u636E\u8FDB\u884C\u5206\u6790" : "Not enough response data for analysis";
     return { content: text, data: null };
@@ -1725,6 +1773,7 @@ var MCP_TOOL_REGISTRY = [
 ];
 
 // ../../packages/tools/src/providers/core-data-provider.ts
+var TOOL_MESSAGE_LIMIT_CAP = 5e4;
 var CoreDataProvider = class {
   constructor(db) {
     this.db = db;
@@ -1735,7 +1784,11 @@ var CoreDataProvider = class {
       startTs: options?.timeFilter?.startTs,
       endTs: options?.timeFilter?.endTs,
       senderId: options?.senderId,
-      limit: options?.limit ?? 50
+      limit: options?.limit ?? 50,
+      offset: options?.offset,
+      matchMode: options?.matchMode,
+      excludeKeywords: options?.excludeKeywords,
+      sort: options?.sort
     });
     return {
       messages: result.messages.map((m) => ({
@@ -1756,9 +1809,14 @@ var CoreDataProvider = class {
     return getSearchMessageContext(this.db, messageIds, contextBefore, contextAfter);
   }
   async getRecentMessages(options) {
-    const messages = getRecentMessages(this.db, { limit: options?.limit ?? 50 });
+    const result = queryMessages(this.db, {
+      limit: options?.limit ?? 50,
+      maxLimit: TOOL_MESSAGE_LIMIT_CAP,
+      startTs: options?.timeFilter?.startTs,
+      endTs: options?.timeFilter?.endTs
+    });
     return {
-      messages: messages.map((m) => ({
+      messages: result.messages.map((m) => ({
         id: m.id,
         senderId: m.senderId,
         senderName: m.senderName,
@@ -1766,7 +1824,7 @@ var CoreDataProvider = class {
         content: m.content,
         timestamp: m.timestamp
       })),
-      total: messages.length
+      total: result.total
     };
   }
   async getMessageContext(messageIds, contextSize) {
@@ -1797,6 +1855,8 @@ var CoreDataProvider = class {
         return getWeekdayActivity(this.db, filter);
       case "daily":
         return getDailyActivity(this.db, filter);
+      case "monthly":
+        return getMonthlyActivity(this.db, filter);
       case "hourly":
       default:
         return getHourlyActivity(this.db, filter);
@@ -1811,8 +1871,8 @@ var CoreDataProvider = class {
   async getConversationBetween(memberId1, memberId2, timeFilter, limit) {
     return getConversationBetween(this.db, memberId1, memberId2, timeFilter, limit);
   }
-  async executeSql(sql) {
-    return executeReadonlySql(this.db, sql);
+  async executeSql(sql, options) {
+    return executeReadonlySql(this.db, sql, options?.maxRows);
   }
   async executeParameterizedSql(query, params) {
     return executeParameterizedSql(this.db, query, params);
@@ -1828,4 +1888,4 @@ export {
   MCP_TOOL_REGISTRY,
   CoreDataProvider
 };
-//# sourceMappingURL=chunk-3QBVO4J7.mjs.map
+//# sourceMappingURL=chunk-5UBHNQFZ.mjs.map