authenticate(dry_run)

tadascience · Mar 9, 2024 · 796ee08 · 796ee08
1 parent 38081ca
commit 796ee08
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/R/authenticate.R b/R/authenticate.R
@@ -1,6 +1,6 @@
-authenticate <- function(request, error_call = caller_env()){
+authenticate <- function(request, dry_run = FALSE, error_call = caller_env()){
   key <- Sys.getenv("MISTRAL_API_KEY")
-  if (identical(key, "")) {
+  if (!is_true(dry_run) && identical(key, "")) {
     cli_abort(call = error_call, c(
       "Please set the {.code MISTRAL_API_KEY} environment variable",
       i = "Get an API key from {.url https://console.mistral.ai/api-keys/}",
@@ -9,4 +9,3 @@ authenticate <- function(request, error_call = caller_env()){
   }
   req_auth_bearer_token(request, key)
 }
-
diff --git a/R/chat.R b/R/chat.R
@@ -28,7 +28,7 @@ req_chat <- function(text = "What are the top 5 R packages ?", model = "mistral-
   }
   request(mistral_base_url) |>
     req_url_path_append("v1", "chat", "completions") |>
-    authenticate(error_call = error_call) |>
+    authenticate(error_call = error_call, dry_run = dry_run) |>
     req_body_json(
       list(
         model = model,

diff --git a/R/models.R b/R/models.R
@@ -11,7 +11,7 @@
 models <- function(error_call = caller_env(), dry_run = FALSE) {
   req <- request(mistral_base_url) |>
     req_url_path_append("v1", "models") |>
-    authenticate(error_call = call) |>
+    authenticate(error_call = call, dry_run = dry_run) |>
     req_cache(tempdir(),
               use_on_error = TRUE,
               max_age = 2 * 60 * 60) # 2 hours