const OpenAI = require('openai');
const { logger } = require('@librechat/data-schemas');

/**
 * Handles errors that may occur when making requests to OpenAI's API.
 * It checks the instance of the error and prints a specific warning message
 * to the console depending on the type of error encountered.
 * It then calls an optional error callback function with the error object.
 *
 * @param {Error} err - The error object thrown by OpenAI API.
 * @param {Function} errorCallback - A callback function that is called with the error object.
 * @param {string} [context='stream'] - A string providing context where the error occurred, defaults to 'stream'.
 */
async function handleOpenAIErrors(err, errorCallback, context = 'stream') {
  if (err instanceof OpenAI.APIError && err?.message?.includes('abort')) {
    logger.warn(`[OpenAIClient.chatCompletion][${context}] Aborted Message`);
  }
  if (err instanceof OpenAI.OpenAIError && err?.message?.includes('missing finish_reason')) {
    logger.warn(`[OpenAIClient.chatCompletion][${context}] Missing finish_reason`);
  } else if (err instanceof OpenAI.APIError) {
    logger.warn(`[OpenAIClient.chatCompletion][${context}] API error`);
  } else {
    logger.warn(`[OpenAIClient.chatCompletion][${context}] Unhandled error type`);
  }

  logger.error(err);

  if (errorCallback) {
    errorCallback(err);
  }
}

module.exports = handleOpenAIErrors;