tests/testthat/test_embedding.R

In openaistream: Streaming Capabilities for 'OpenAI API' Interactions

#test_embedding
test_that("embedding",{
  skip_on_cran()
  Sys.setenv(TEST_EX_COND = "")
  handle_openai<-openai$new(Sys.getenv("OPENAI_KEY"))
  if(Sys.getenv("USE_PROXY")=="TRUE"){
    handle_openai$set_proxy("10.0.108.36",7890)
  }
  embed<-handle_openai$embeddings$create(model = "text-embedding-ada-002", input = "who are you?",verbosity = 3)
  expect_type(embed$data$embedding[[1]],"double")

  embed<-handle_openai$embeddings$create(model = "text-embedding-ada-002", input = "who are you?",verbosity = 4)
  expect_true(!embed$success)
})