ChatLab/electron/main/ai/tools/definitions/search-sessions.ts

import { Type } from '@mariozechner/pi-ai'
import type { AgentTool } from '@mariozechner/pi-agent-core'
import type { ToolContext } from '../types'
import { timeParamProperties } from '../utils/schemas'
import * as workerManager from '../../../worker/workerManager'
import { parseExtendedTimeParams } from '../utils/time-params'
import { formatTimeRange, formatMessageCompact, isChineseLocale } from '../utils/format'

const schema = Type.Object({
  keywords: Type.Optional(Type.Array(Type.String(), { description: 'ai.tools.search_sessions.params.keywords' })),
  limit: Type.Optional(Type.Number({ description: 'ai.tools.search_sessions.params.limit' })),
  ...timeParamProperties,
})

/** 搜索聊天会话（对话段落）。会话是根据消息时间间隔自动切分的对话单元。适用于查找特定话题的讨论、了解某个时间段内发生了几次对话等场景。返回匹配的会话列表及每个会话的前5条消息预览。 */
export function createTool(context: ToolContext): AgentTool<typeof schema> {
  return {
    name: 'search_sessions',
    label: 'search_sessions',
    description: 'ai.tools.search_sessions.desc',
    parameters: schema,
    execute: async (_toolCallId, params) => {
      const { sessionId, timeFilter: contextTimeFilter, locale } = context
      const limit = params.limit || 20
      const effectiveTimeFilter = parseExtendedTimeParams(params, contextTimeFilter)

      const sessions = await workerManager.searchSessions(sessionId, params.keywords, effectiveTimeFilter, limit, 5)

      if (sessions.length === 0) {
        const data = {
          total: 0,
          message: isChineseLocale(locale) ? '未找到匹配的会话' : 'No matching sessions found',
        }
        return {
          content: [{ type: 'text', text: JSON.stringify(data) }],
          details: data,
        }
      }

      const localeStr = isChineseLocale(locale) ? 'zh-CN' : 'en-US'
      const msgSuffix = isChineseLocale(locale) ? '条消息' : ' messages'
      const completeLabel = isChineseLocale(locale) ? '完整会话' : 'complete'

      const data = {
        total: sessions.length,
        timeRange: formatTimeRange(effectiveTimeFilter, locale),
        sessions: sessions.map((s) => {
          const startTime = new Date(s.startTs * 1000).toLocaleString(localeStr)
          const endTime = new Date(s.endTs * 1000).toLocaleString(localeStr)
          const completeTag = s.isComplete ? ` [${completeLabel}]` : ''

          return {
            sessionId: s.id,
            time: `${startTime} ~ ${endTime}`,
            messageCount: `${s.messageCount}${msgSuffix}${completeTag}`,
            preview: s.previewMessages.map((m) => formatMessageCompact(m, locale)),
          }
        }),
      }

      return {
        content: [{ type: 'text', text: JSON.stringify(data) }],
        details: data,
      }
    },
  }
}