R/util_services.R

In air: AI Assistant to Write and Understand R Code

validated_credentials_or_stop <- function() {
  key_res <- get_key()
  model_res <- get_model()

  if (result::is_failure(key_res) || result::is_failure(model_res)) {
    stop(paste("Please set your OpenAI key and model first.\n",
               "See package documentation for details."))
  }

  creds <- list(
    key = result::value(key_res),
    model = result::value(model_res)
  )

  class(creds) <- c("credentials", class(creds))
  creds
}

api_answer_or_stop <- function(input, creds, context, call_api = call_openai) {
  req <- ai_completion_request(input, creds$model, context)
  res <- call_api(req$endpoint, creds$key, req$json_body)

  if (result::is_failure(res)) {
    stop(result::value(res))
  }

  parse_response_message(result::value(res))
}