Adds tests for OpenAI switching mechanism

R/chattr-defaults.R

@@ -106,9 +106,13 @@ chattr_defaults <- function(type = "default",
   )
 
   ret <- chattr_defaults_get(type)
-  ret$type <- type
+  as_ch_model(ret, type)
+}
+
+as_ch_model <- function(x, type) {
+  x$type <- type
 
-  provider <- tolower(ret$provider)
+  provider <- tolower(x$provider)
 
   sp_provider <- unlist(strsplit(provider, " - "))
   if (length(sp_provider) > 1) {
@@ -117,13 +121,12 @@ chattr_defaults <- function(type = "default",
     first_cl <- NULL
   }
 
-  class(ret) <- c(
+  class(x) <- c(
     paste0("ch_", prep_class_name(provider)),
     first_cl,
     "ch_model"
   )
-
-  ret
+  x
 }
 
 prep_class_name <- function(x) {

tests/testthat/_snaps/ch_context.md

@@ -20,7 +20,7 @@
     Code
       ch_context_data_files(file_types = "R")
     Output
-      [1] "Data files available: \n|- setup.R\n|- test-app_server.R\n|- test-app_ui.R\n|- test-app_utils.R\n|- test-backend-llamagpt.R\n|- test-backend-openai-core.R\n|- test-backend-openai.R\n|- test-ch-defaults-save.R\n|- test-ch-history.R\n|- test-ch-submit.R\n|- test-ch_context.R\n|- test-ch_defaults.R\n|- test-chattr-test.R\n|- test-chattr-use.R\n|- test-chattr.R\n|- test-ide.R"
+      [1] "Data files available: \n|- setup.R\n|- test-app_server.R\n|- test-app_ui.R\n|- test-app_utils.R\n|- test-backend-llamagpt.R\n|- test-backend-openai-core.R\n|- test-backend-openai-switch.R\n|- test-backend-openai.R\n|- test-ch-defaults-save.R\n|- test-ch-history.R\n|- test-ch-submit.R\n|- test-ch_context.R\n|- test-ch_defaults.R\n|- test-chattr-test.R\n|- test-chattr-use.R\n|- test-chattr.R\n|- test-ide.R"
 
 ---
 

tests/testthat/test-backend-openai-switch.R

@@ -0,0 +1,64 @@
+test_that("Using 'chat'", {
+  withr::with_envvar(
+    new = c("OPENAI_API_KEY" = "test"),
+    {
+      local_mocked_bindings(
+        req_perform = httr2::req_dry_run
+      )
+      defaults <- yaml::read_yaml(package_file("configs", "gpt35.yml"))
+      defaults <- as_ch_model(defaults$default, "chat")
+      expect_error(
+        openai_switch(
+          prompt = "test",
+          req_body = defaults$model_arguments,
+          defaults = defaults,
+          r_file_stream =  tempfile(),
+          r_file_complete = tempfile()
+        )
+      )
+    }
+  )
+})
+
+test_that("Using 'console'", {
+  withr::with_envvar(
+    new = c("OPENAI_API_KEY" = "test"),
+    {
+      local_mocked_bindings(
+        req_perform_stream = function(...) {
+          ch_env$stream$response <- "test"
+        }
+      )
+      defaults <- yaml::read_yaml(package_file("configs", "gpt35.yml"))
+      defaults <- as_ch_model(defaults$default, "console")
+      expect_silent(
+      openai_switch(
+          prompt = "test",
+          req_body = defaults$model_arguments,
+          defaults = defaults,
+          r_file_stream =  tempfile(),
+          r_file_complete = tempfile()
+        )
+      )
+    }
+  )
+})
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+