feat: 新增会话相关function calling调用

electron/main/worker/query/index.ts

@@ -62,6 +62,8 @@ export {
   getSessionStats,
   updateSessionGapThreshold,
   getSessions,
+  searchSessions,
+  getSessionMessages,
   DEFAULT_SESSION_GAP_THRESHOLD,
 } from './session'
-export type { ChatSessionItem } from './session'
+export type { ChatSessionItem, SessionSearchResultItem, SessionMessagesResult } from './session'

electron/main/worker/query/session.ts

@@ -340,3 +340,241 @@ export function getSessions(sessionId: string): ChatSessionItem[] {
     db.close()
   }
 }
+
+// ==================== AI 工具专用查询函数 ====================
+
+/**
+ * 会话搜索结果项类型（用于 AI 工具）
+ */
+export interface SessionSearchResultItem {
+  id: number
+  startTs: number
+  endTs: number
+  messageCount: number
+  /** 是否为完整会话（消息数 <= 预览条数） */
+  isComplete: boolean
+  /** 预览消息列表 */
+  previewMessages: Array<{
+    id: number
+    senderName: string
+    content: string | null
+    timestamp: number
+  }>
+}
+
+/**
+ * 搜索会话（用于 AI 工具）
+ * 支持按关键词和时间范围筛选会话
+ *
+ * @param sessionId 数据库会话ID
+ * @param keywords 关键词列表（可选，OR 逻辑匹配）
+ * @param timeFilter 时间过滤器（可选）
+ * @param limit 返回数量限制，默认 20
+ * @param previewCount 预览消息数量，默认 5
+ * @returns 匹配的会话列表
+ */
+export function searchSessions(
+  sessionId: string,
+  keywords?: string[],
+  timeFilter?: { startTs: number; endTs: number },
+  limit: number = 20,
+  previewCount: number = 5
+): SessionSearchResultItem[] {
+  const db = openReadonlyDatabase(sessionId)
+  if (!db) {
+    return []
+  }
+
+  try {
+    // 1. 构建会话查询 SQL
+    let sessionSql = `
+      SELECT
+        cs.id,
+        cs.start_ts as startTs,
+        cs.end_ts as endTs,
+        cs.message_count as messageCount
+      FROM chat_session cs
+      WHERE 1=1
+    `
+    const params: unknown[] = []
+
+    // 时间范围过滤
+    if (timeFilter) {
+      sessionSql += ` AND cs.start_ts >= ? AND cs.end_ts <= ?`
+      params.push(timeFilter.startTs, timeFilter.endTs)
+    }
+
+    // 关键词过滤：只返回包含关键词的会话
+    if (keywords && keywords.length > 0) {
+      const keywordConditions = keywords.map(() => `m.content LIKE ?`).join(' OR ')
+      sessionSql += `
+        AND cs.id IN (
+          SELECT DISTINCT mc.session_id
+          FROM message_context mc
+          JOIN message m ON m.id = mc.message_id
+          WHERE (${keywordConditions})
+        )
+      `
+      for (const kw of keywords) {
+        params.push(`%${kw}%`)
+      }
+    }
+
+    sessionSql += ` ORDER BY cs.start_ts DESC LIMIT ?`
+    params.push(limit)
+
+    const sessions = db.prepare(sessionSql).all(...params) as Array<{
+      id: number
+      startTs: number
+      endTs: number
+      messageCount: number
+    }>
+
+    // 2. 为每个会话获取预览消息
+    const previewSql = `
+      SELECT
+        m.id,
+        COALESCE(mb.group_nickname, mb.account_name, mb.platform_id) as senderName,
+        m.content,
+        m.ts as timestamp
+      FROM message_context mc
+      JOIN message m ON m.id = mc.message_id
+      JOIN member mb ON mb.id = m.sender_id
+      WHERE mc.session_id = ?
+      ORDER BY m.ts ASC
+      LIMIT ?
+    `
+
+    const results: SessionSearchResultItem[] = []
+    for (const session of sessions) {
+      const previewMessages = db.prepare(previewSql).all(session.id, previewCount) as Array<{
+        id: number
+        senderName: string
+        content: string | null
+        timestamp: number
+      }>
+
+      results.push({
+        id: session.id,
+        startTs: session.startTs,
+        endTs: session.endTs,
+        messageCount: session.messageCount,
+        isComplete: session.messageCount <= previewCount,
+        previewMessages,
+      })
+    }
+
+    return results
+  } catch (error) {
+    console.error('searchSessions error:', error)
+    return []
+  } finally {
+    db.close()
+  }
+}
+
+/**
+ * 会话消息结果类型（用于 AI 工具）
+ */
+export interface SessionMessagesResult {
+  sessionId: number
+  startTs: number
+  endTs: number
+  messageCount: number
+  returnedCount: number
+  /** 参与者列表 */
+  participants: string[]
+  /** 消息列表 */
+  messages: Array<{
+    id: number
+    senderName: string
+    content: string | null
+    timestamp: number
+  }>
+}
+
+/**
+ * 获取会话的完整消息（用于 AI 工具）
+ *
+ * @param sessionId 数据库会话ID
+ * @param chatSessionId 会话索引中的会话ID
+ * @param limit 返回数量限制，默认 500
+ * @returns 会话的完整消息
+ */
+export function getSessionMessages(
+  sessionId: string,
+  chatSessionId: number,
+  limit: number = 500
+): SessionMessagesResult | null {
+  const db = openReadonlyDatabase(sessionId)
+  if (!db) {
+    return null
+  }
+
+  try {
+    // 1. 获取会话基本信息
+    const sessionSql = `
+      SELECT
+        id,
+        start_ts as startTs,
+        end_ts as endTs,
+        message_count as messageCount
+      FROM chat_session
+      WHERE id = ?
+    `
+    const session = db.prepare(sessionSql).get(chatSessionId) as
+      | {
+          id: number
+          startTs: number
+          endTs: number
+          messageCount: number
+        }
+      | undefined
+
+    if (!session) {
+      return null
+    }
+
+    // 2. 获取会话消息
+    const messagesSql = `
+      SELECT
+        m.id,
+        COALESCE(mb.group_nickname, mb.account_name, mb.platform_id) as senderName,
+        m.content,
+        m.ts as timestamp
+      FROM message_context mc
+      JOIN message m ON m.id = mc.message_id
+      JOIN member mb ON mb.id = m.sender_id
+      WHERE mc.session_id = ?
+      ORDER BY m.ts ASC
+      LIMIT ?
+    `
+    const messages = db.prepare(messagesSql).all(chatSessionId, limit) as Array<{
+      id: number
+      senderName: string
+      content: string | null
+      timestamp: number
+    }>
+
+    // 3. 统计参与者
+    const participantsSet = new Set<string>()
+    for (const msg of messages) {
+      participantsSet.add(msg.senderName)
+    }
+
+    return {
+      sessionId: session.id,
+      startTs: session.startTs,
+      endTs: session.endTs,
+      messageCount: session.messageCount,
+      returnedCount: messages.length,
+      participants: Array.from(participantsSet),
+      messages,
+    }
+  } catch (error) {
+    console.error('getSessionMessages error:', error)
+    return null
+  } finally {
+    db.close()
+  }
+}