stream(dry_run)

tadascience · Mar 9, 2024 · dcff55a · dcff55a
1 parent 45c460e
commit dcff55a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/R/stream.R b/R/stream.R
@@ -1,11 +1,16 @@
 #' stream
 #'
 #' @inheritParams chat
+#'
 #' @export
-stream <- function(text, model = "mistral-tiny", ..., error_call = current_env()) {
+stream <- function(text, model = "mistral-tiny", dry_run = FALSE, ..., error_call = current_env()) {
   check_model(model, error_call = error_call)
 
-  req <- req_chat(text, model, stream = TRUE, error_call = error_call)
+  req <- req_chat(text, model, stream = TRUE, error_call = error_call, dry_run = dry_run)
+  if (is_true(dry_run)) {
+    return(req)
+  }
+
   resp <- req_perform_stream(req,
     callback = stream_callback,
     round = "line",

diff --git a/man/stream.Rd b/man/stream.Rd