R/genai.openai.utils.R

In GenAI: Generative Artificial Intelligence

#' @noRd
genai.openai.check = function(api, model, version, proxy, organization.id) {
  json.data = jsonlite::fromJSON("https://genai.gd.edu.kg/model.json")
  if (is.na(match(model, json.data$openai$model))) {
    stop(
      "Invalid value for model. Refer to 'available.models()' to view the supported models."
    )
  }
  if (is.na(match(version, json.data$openai$version))) {
    stop(
      "Invalid value for version. Refer to 'available.models()' to view the supported versions."
    )
  }
  if (!proxy %in% c(TRUE, FALSE)) {
    stop("Invalid value for proxy. It must be either TRUE or FALSE.")
  }
  if (!is.null(organization.id) && !is.character(organization.id)) {
    stop("Invalid value for organization.id. It must be either NULL (by default) or a character string.")
  }

  # Check connection
  api.url = paste0(
    "https://api.openai.com/",
    version,
    "/models"
  )
  if (proxy) {
    api.url = paste0(
      "https://api.genai.gd.edu.kg/openai/",
      version,
      "/models")
  }
  response = httr::GET(url = api.url,
                       httr::add_headers(
                         "Content-Type" = "application/json",
                         "Authorization" = paste("Bearer", api)
                       )
  )
  if (!is.null(organization.id) && is.character(organization.id)) {
    response = httr::GET(url = api.url,
                         httr::add_headers(
                           "Content-Type" = "application/json",
                           "Authorization" = paste("Bearer", api),
                           "OpenAI-Organization" = organization.id
                         )
    )
  }
  responseJSON = httr::content(response, "parsed")
  if (!is.null(responseJSON$error)) {
    stop(responseJSON$error$message)
  }
  if (response$status_code != 200) {
    stop(
      "Invalid parameter(s) detected. Please check the values for api, model, version, and proxy."
    )
  }
}

#' @noRd
genai.openai.config.check = function(config) {
  if (!is.list(config)) {
    stop("Invalid configuration. It must be a list.")
  }
  config.names = c(
    "frequency.penalty",
    "logit.bias",
    "logprobs",
    "top.logprobs",
    "max.tokens",
    "presence.penalty",
    "response.format",
    "seed",
    "stop",
    "temperature",
    "top.p",
    "tools",
    "tool.choice",
    "user"
  )
  wrong.config = setdiff(names(config), config.names)
  if (length(wrong.config) > 0) {
    stop("Invalid configuration(s) detected: ",
         paste0(wrong.config, collapse = ", "))
  }
  if (length(unique(names(config))) != length(names(config))) {
    stop("Invalid configurations. Duplicate parameters detected.")
  }
}

#' @noRd
genai.openai.img.config.check = function(config) {
  if (!is.list(config)) {
    stop("Invalid configuration. It must be a list.")
  }
  config.names = c(
    "quality",
    "size",
    "style",
    "user"
  )
  wrong.config = setdiff(names(config), config.names)
  if (length(wrong.config) > 0) {
    stop("Invalid configuration(s) detected: ",
         paste0(wrong.config, collapse = ", "))
  }
  if (length(unique(names(config))) != length(names(config))) {
    stop("Invalid configurations. Duplicate parameters detected.")
  }
}

#' @noRd
genai.openai.generation.config = function(requestBody, config) {
  config.names = c(
    frequency.penalty = "frequency_penalty",
    logit.bias = "logit_bias",
    logprobs = "logprobs",
    top.logprobs = "top_logprobs",
    max.tokens = "max_tokens",
    presence.penalty = "presence_penalty",
    response.format = "response_format",
    seed = "seed",
    stop = "stop",
    temperature = "temperature",
    top.p = "top_p",
    tools = "tools",
    tool.choice = "tool_choice",
    user = "user"
  )
  for (param_name in names(config)) {
    if (!is.null(config[[param_name]])) {
      requestBody[[config.names[param_name]]] = config[[param_name]]
    }
  }
  return(requestBody)
}

#' @noRd
genai.openai.img.generation.config = function(requestBody, config) {
  config.names = c(
    quality = "quality",
    size = "size",
    style = "style",
    user = "user"
  )
  for (param_name in names(config)) {
    if (!is.null(config[[param_name]])) {
      requestBody[[config.names[param_name]]] = config[[param_name]]
    }
  }
  return(requestBody)
}

#' @noRd
genai.openai.formated.confguration = function(request.body, prompt) {
  config.names = c(
    frequency.penalty = "frequency_penalty",
    logit.bias = "logit_bias",
    logprobs = "logprobs",
    top.logprobs = "top_logprobs",
    max.tokens = "max_tokens",
    presence.penalty = "presence_penalty",
    response.format = "response_format",
    seed = "seed",
    stop = "stop",
    temperature = "temperature",
    top.p = "top_p",
    tools = "tools",
    tool.choice = "tool_choice",
    user = "user"
  )
  intersect.param = intersect(names(request.body), config.names)
  if (length(intersect.param) > 0) {
    cat("=============================================================================\n")
    cat("   Generation Configuration\n")
    cat("-----------------------------------------------------------------------------\n")
    for (param in intersect.param) {
      if (is.list(request.body[[param]])) {
        cat("stop:",
            paste0(request.body[[param]],
                   collapse = ", "),
            "\n")
      }
      else {
        cat(paste0(param, ":"),
            request.body[[param]],
            "\n")
      }
    }
    cat("=============================================================================\n\n\n\n")
  }
  cat("=============================================================================\n")
  cat("   Prompt\n")
  cat("-----------------------------------------------------------------------------\n")
  cat(paste(strwrap(prompt, width = 76, exdent = 0), collapse = "\n"))
  cat("\n")
  cat("=============================================================================\n\n\n\n")
}

#' @noRd
genai.openai.img.formated.confguration = function(request.body, prompt) {
  config.names = c(
    quality = "quality",
    size = "size",
    style = "style",
    user = "user"
  )
  intersect.param = intersect(names(request.body), config.names)
  if (length(intersect.param) > 0) {
    cat("=============================================================================\n")
    cat("   Generation Configuration\n")
    cat("-----------------------------------------------------------------------------\n")
    for (param in intersect.param) {
      if (is.list(request.body[[param]])) {
        cat("stop:",
            paste0(request.body[[param]],
                   collapse = ", "),
            "\n")
      }
      else {
        cat(paste0(param, ":"),
            request.body[[param]],
            "\n")
      }
    }
    cat("=============================================================================\n\n\n\n")
  }
  cat("=============================================================================\n")
  cat("   Prompt\n")
  cat("-----------------------------------------------------------------------------\n")
  cat(paste(strwrap(prompt, width = 76, exdent = 0), collapse = "\n"))
  cat("\n")
  cat("=============================================================================\n\n\n\n")
}