Function handler

packages/tools/src/definitions/get-segment-messages.ts:17–51

(params: Record<string, unknown>, context: ToolExecutionContext)

}

async function handler(params: Record<string, unknown>, context: ToolExecutionContext): Promise<ToolResult> {
  const { locale, maxMessagesLimit } = context
  const limit = maxMessagesLimit || (params.limit as number) || 1000

  const result = await context.dataProvider!.getSegmentMessages(params.segment_id as number, limit)

  if (!result) {
    const data = {
      error: isChineseLocale(locale) ? '未找到指定的段落' : 'Segment not found',
      segmentId: params.segment_id,
    }
    return { content: JSON.stringify(data), data }
  }

  const localeStr = isChineseLocale(locale) ? 'zh-CN' : 'en-US'
  const startTime = new Date(result.startTs * 1000).toLocaleString(localeStr)
  const endTime = new Date(result.endTs * 1000).toLocaleString(localeStr)
  const rawMessages = result.messages.map((m) => ({
    id: m.id,
    senderName: m.senderName,
    content: m.content,
    timestamp: m.timestamp,
  }))

  const data = {
    segmentId: result.segmentId,
    time: `${startTime} ~ ${endTime}`,
    messageCount: result.messageCount,
    returnedCount: result.returnedCount,
    participants: result.participants,
    rawMessages,
  }

  return { content: JSON.stringify(data), data, rawMessages }
}

export const getSegmentMessagesTool: ToolDefinition = {
  name: 'get_segment_messages',