Function handler

packages/tools/src/definitions/get-message-context.ts:19–42

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

}

async function handler(params: Record<string, unknown>, context: ToolExecutionContext): Promise<ToolResult> {
  const { locale } = context
  const messageIds = params.message_ids as number[]
  const contextSize = (params.context_size as number) || 20

  const messages = await context.dataProvider!.getMessageContext(messageIds, contextSize)

  if (messages.length === 0) {
    const data = {
      error: t('noMessageContext', locale),
      messageIds,
    }
    return { content: JSON.stringify(data), data }
  }

  const data = {
    totalMessages: messages.length,
    contextSize,
    requestedMessageIds: messageIds,
    rawMessages: messages,
  }

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

export const getMessageContextTool: ToolDefinition = {
  name: 'get_message_context',