check()

R/chat.R

@@ -1,10 +1,10 @@
 #' Chat with the Mistral api
 #'
 #' @param messages Messages
-#' @param text some text
 #' @param model which model to use. See [models()] for more information about which models are available
 #' @param dry_run if TRUE the request is not performed
-#' @inheritParams httr2::req_perform
+#' @inheritParams rlang::args_dots_empty
+#' @inheritParams rlang::args_error_context
 #'
 #' @return A tibble with columns `role` and `content` with class `chat_tibble` or a request
 #'         if this is a `dry_run`
@@ -13,7 +13,9 @@
 #' chat("Top 5 R packages", dry_run = TRUE)
 #'
 #' @export
-chat <- function(messages, model = "mistral-tiny", dry_run = FALSE, error_call = current_env()) {
+chat <- function(messages, model = "mistral-tiny", dry_run = FALSE, ..., error_call = current_env()) {
+  check_dots_empty()
+
   req <- req_chat(messages, model = model, error_call = error_call, dry_run = dry_run)
   if (is_true(dry_run)) {
     return(req)
@@ -29,7 +31,9 @@ print.chat <- function(x, ...) {
   invisible(x)
 }
 
-req_chat <- function(messages, model = "mistral-tiny", stream = FALSE, dry_run = FALSE, error_call = caller_env()) {
+req_chat <- function(messages, model = "mistral-tiny", stream = FALSE, dry_run = FALSE, ..., error_call = caller_env()) {
+  check_dots_empty()
+
   if (!is_true(dry_run)) {
     check_model(model, error_call = error_call)
   }
@@ -50,8 +54,12 @@ req_chat <- function(messages, model = "mistral-tiny", stream = FALSE, dry_run =
 
 #' @export
 as.data.frame.chat_response <- function(x, ...) {
-  df_req <- map_dfr(resp$request$body$data$messages, as.data.frame)
-  df_resp <- as.data.frame(resp_body_json(resp)$choices[[1]]$message[c("role", "content")])
+  req_messages <- x$request$body$data$messages
+  df_req <- map_dfr(req_messages, as.data.frame)
+
+  df_resp <- as.data.frame(
+    resp_body_json(x)$choices[[1]]$message[c("role", "content")]
+  )
 
   rbind(df_req, df_resp)
 }

R/messages.R

@@ -1,24 +1,37 @@
+#' Convert object into a messages list
+#'
+#' @param messages object to convert to messages
+#' @param ... ignored
+#' @inheritParams rlang::args_error_context
+#'
+#' @examples
+#' as_messages("hello")
+#' as_messages(list("hello"))
+#' as_messages(list(assistant = "hello", user = "hello"))
+#'
 #' @export
-as_messages <- function(messages, ...) {
+as_messages <- function(messages, ..., error_call = current_env()) {
   UseMethod("as_messages")
 }
 
 #' @export
-as_messages.character <- function(x, ..., error_call = current_env()) {
-  check_scalar_string(x, error_call = error_call)
-  check_unnamed_string(x, error_call = error_call)
+as_messages.character <- function(messages, ..., error_call = current_env()) {
+  check_dots_empty(call = error_call)
+  check_scalar_string(messages, error_call = error_call)
+  check_unnamed_string(messages, error_call = error_call)
 
   list(
-    list(role = "user", content = x)
+    list(role = "user", content = messages)
   )
 }
 
 #' @export
-as_messages.list <- function(x, ..., error_call = caller_env()) {
+as_messages.list <- function(messages, ..., error_call = caller_env()) {
   check_dots_empty()
 
-  bits <- map2(x, names2(x), as_msg, error_call = error_call)
-  out <- list_flatten(bits)
+  out <- list_flatten(
+    map2(messages, names2(messages), as_msg, error_call = error_call)
+  )
   names(out) <- NULL
   out
 }