tests/testthat/test-as-json.R

In ellmer: Chat with Large Language Models

test_that("can convert basic types to json schema", {
  stub <- Provider("")

  expect_equal(
    as_json(stub, type_boolean("desc")),
    list(type = "boolean", description = "desc")
  )

  expect_equal(
    as_json(stub, type_enum("desc", letters[1:3])),
    list(type = "string", description = "desc", enum = as.list(letters[1:3]))
  )

  expect_equal(
    as_json(stub, type_array("a", type_boolean("b"))),
    list(
      type = "array",
      description = "a",
      items = list(type = "boolean", description = "b")
    )
  )
})

test_that("can convert an object to json schema", {
  stub <- Provider("")
  obj <- type_object(
    "a",
    integer = type_integer(),
    number = type_number(),
    string = type_string()
  )

  expect_equal(
    as_json(stub, obj),
    list(
      type = "object",
      description = "a",
       properties = list(
         integer = list(type = "integer", description = ""),
         number = list(type = "number", description = ""),
         string = list(type = "string", description = "")
       ),
       required = list("integer", "number", "string"),
       additionalProperties = FALSE
    )
  )
})