adds function check_model_installed

NAMESPACE

@@ -2,6 +2,7 @@
 
 export(chat)
 export(chat_history)
+export(check_model_installed)
 export(copy_model)
 export(create_model)
 export(delete_model)

NEWS.md

@@ -1,3 +1,7 @@
+# rollama 0.1.0
+
+* adds function `check_model_installed`
+
 # rollama 0.0.3
 
 * add option to query several models at once

R/utils.r

@@ -4,3 +4,33 @@ screen_answer <- function(x, model = NULL) {
   # "{i}" instead of i stops glue from evaluating code inside the answer
   for (i in pars) cli::cli_text("{i}")
 }
+
+
+#' Check if one or several models are installed on the server
+#'
+#' @param model names of one or several models as character vector.
+#' @param auto_pull if FALSE, the default, asks before downloading models.
+#' @inheritParams query
+#'
+#' @return invisible TRUE/FALSE
+#' @export
+check_model_installed <- function(model, auto_pull = FALSE, server = NULL) {
+  models_df <- list_models(server = server)
+  model <- setdiff(model, models_df$name)
+  model_wo_vers <- gsub(":.*", "", models_df$name)
+  model <- setdiff(model, model_wo_vers)
+  if (length(model) > 0L && !auto_pull) {
+    cli::cli_alert_info("Model{?s} {model} not installed. Would you like to download {?it/them}?")
+    if (interactive()) auto_pull <- askYesNo("")
+    if (!auto_pull) {
+      cli::cli_alert_danger("Model{?s} {model} not installed.")
+      invisible(FALSE)
+    }
+  }
+  if (auto_pull) {
+    for (m in model) {
+      pull_model(m, server = server)
+    }
+  }
+  invisible(TRUE)
+}