Function handler

packages/tools/src/definitions/search-messages.ts:23–54

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

}

async function handler(params: Record<string, unknown>, context: ToolExecutionContext): Promise<ToolResult> {
  const { locale, timeFilter: contextTimeFilter, maxMessagesLimit } = context
  const keywords = params.keywords as string[]
  const limit = Math.min(maxMessagesLimit || (params.limit as number) || 1000, 50000)
  const effectiveTimeFilter = parseExtendedTimeParams(params as any, contextTimeFilter)

  const result = await context.dataProvider!.searchMessages(keywords, {
    timeFilter: effectiveTimeFilter,
    limit,
    senderId: params.sender_id as number | undefined,
  })

  const contextBefore = context.searchContextBefore ?? 2
  const contextAfter = context.searchContextAfter ?? 2
  let finalMessages = result.messages

  if ((contextBefore > 0 || contextAfter > 0) && result.messages.length > 0) {
    const hitIds = result.messages.map((m) => m.id).filter((id): id is number => id != null)
    if (hitIds.length > 0) {
      finalMessages = await context.dataProvider!.getSearchMessageContext(hitIds, contextBefore, contextAfter)
    }
  }

  const data = {
    total: result.total,
    returned: finalMessages.length,
    timeRange: formatTimeRange(effectiveTimeFilter, locale),
    rawMessages: finalMessages,
  }

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

export const searchMessagesTool: ToolDefinition = {
  name: 'search_messages',