Minor fixes and test adjustment

mlverse · Dec 24, 2024 · a8eba33 · a8eba33
1 parent d5b145d
commit a8eba33
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/R/chattr-use.R b/R/chattr-use.R
@@ -97,8 +97,8 @@ ch_get_ymls <- function(menu = TRUE) {
   }
 
   if (!dbrx_exists) {
-    prep_files$`databricks-meta-llama-3-1-405b-instruct` <- NULL
-    prep_files$`databricks-meta-llama-3-3-70b-instruct` <- NULL
+    prep_files$`databricks-meta-llama-3-1-405b` <- NULL
+    prep_files$`databricks-meta-llama-3-3-70b` <- NULL
   }
 
   if (!llama_exists) {

diff --git a/tests/testthat/test-chattr-use.R b/tests/testthat/test-chattr-use.R
@@ -31,7 +31,11 @@ test_that("Missing token prevents showing the option", {
 test_that("Menu works", {
   skip_on_cran()
   withr::with_envvar(
-    new = c("OPENAI_API_KEY" = "test", "DATABRICKS_TOKEN" = NA),
+    new = c(
+      "OPENAI_API_KEY" = "test",
+      "DATABRICKS_HOST" = NA,
+      "DATABRICKS_TOKEN" = NA
+      ),
     {
       local_mocked_bindings(
         menu = function(...) {