"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.registerRecallFunction = registerRecallFunction;
var _types = require("../../common/types");
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0; you may not use this file except in compliance with the Elastic License
 * 2.0.
 */

function registerRecallFunction({
  service,
  registerFunction
}) {
  registerFunction({
    name: 'recall',
    contexts: ['core'],
    description: `Use this function to recall earlier learnings. Anything you will summarize can be retrieved again later via this function.
      
      Make sure the query covers the following aspects:
      - Anything you've inferred from the user's request, but is not mentioned in the user's request
      - The functions you think might be suitable for answering the user's request. If there are multiple functions that seem suitable, create multiple queries. Use the function name in the query.  

      DO NOT include the user's request. It will be added internally.
      
      The user asks: "can you visualise the average request duration for opbeans-go over the last 7 days?"
      You recall:
      - "APM service"
      - "lens function usage"
      - "get_apm_timeseries function usage"`,
    descriptionForUser: 'This function allows the assistant to recall previous learnings.',
    parameters: {
      type: 'object',
      additionalProperties: false,
      properties: {
        queries: {
          type: 'array',
          additionalItems: false,
          additionalProperties: false,
          items: {
            type: 'string',
            description: 'The query for the semantic search'
          }
        }
      },
      required: ['queries']
    }
  }, ({
    arguments: {
      queries
    },
    messages
  }, signal) => {
    var _userMessages;
    const userMessages = messages.filter(message => message.message.role === _types.MessageRole.User);
    const userPrompt = (_userMessages = userMessages[userMessages.length - 1]) === null || _userMessages === void 0 ? void 0 : _userMessages.message.content;
    const queriesWithUserPrompt = userPrompt ? [userPrompt, ...queries] : queries;
    return service.callApi('POST /internal/observability_ai_assistant/functions/recall', {
      params: {
        body: {
          queries: queriesWithUserPrompt
        }
      },
      signal
    }).then(response => ({
      content: response
    }));
  });
}