Merge pull request #122 from mlverse/updates

Fixes bug that shows GPT 4o even if there is no token

Author: edgararuiz

DESCRIPTION

@@ -1,6 +1,6 @@
 Package: chattr
 Title: Interact with Large Language Models in 'RStudio'
-Version: 0.2.0
+Version: 0.2.0.9000
 Authors@R: c(
   person("Edgar", "Ruiz", , "[email protected]", role = c("aut", "cre")),
   person(given = "Posit Software, PBC", role = c("cph", "fnd"))

NEWS.md

@@ -1,3 +1,7 @@
+# chattr (dev)
+
+* Prevents OpenAI 4o from showing as an option if no token is found
+
 # chattr 0.2.0
 
 ## General

R/chattr-use.R

@@ -93,6 +93,7 @@ ch_get_ymls <- function(menu = TRUE) {
   if (!gpt_exists) {
     prep_files$gpt35 <- NULL
     prep_files$gpt4 <- NULL
+    prep_files$gpt4o <- NULL
   }
 
   if (!dbrx_exists) {

tests/testthat/_snaps/chattr-use.md

@@ -12,6 +12,13 @@
     Output
       [1] "OpenAI - Chat Completions - gpt-4 (gpt4) \n"
 
+---
+
+    Code
+      out$gpt4o
+    Output
+      [1] "OpenAI - Chat Completions - gpt-4o (gpt4o) \n"
+
 # Menu works
 
     Code

tests/testthat/test-chattr-use.R

@@ -6,10 +6,28 @@ test_that("Request submission works", {
       expect_equal(class(out), "list")
       expect_snapshot(out$gpt35)
       expect_snapshot(out$gpt4)
+      expect_snapshot(out$gpt4o)
     }
   )
 })
 
+test_that("Missing token prevents showing the option", {
+  withr::with_envvar(
+    new = c(
+      "OPENAI_API_KEY" = NA,
+      "DATABRICKS_TOKEN" = "test",
+      "DATABRICKS_HOST" = "test"
+      ),
+    {
+      out <- ch_get_ymls(menu = FALSE)
+      expect_null(out$gpt35)
+      expect_null(out$gpt4)
+      expect_null(out$gpt4o)
+    }
+  )
+})
+
+
 test_that("Menu works", {
   skip_on_cran()
   withr::with_envvar(