Adding ResourceDownloader functions

Author: dkincaid

NAMESPACE

@@ -227,6 +227,7 @@ export(nlp_chunk_filterer)
 export(nlp_chunker)
 export(nlp_classifier_dl)
 export(nlp_classifier_dl_pretrained)
+export(nlp_clear_cache)
 export(nlp_conll_read_dataset)
 export(nlp_conllu_read_dataset)
 export(nlp_context_spell_checker)
@@ -301,6 +302,9 @@ export(nlp_sentiment_dl_pretrained)
 export(nlp_set_input_cols)
 export(nlp_set_output_col)
 export(nlp_set_param)
+export(nlp_show_available_annotators)
+export(nlp_show_public_models)
+export(nlp_show_public_pipelines)
 export(nlp_spark_annotation)
 export(nlp_stemmer)
 export(nlp_stop_words_cleaner)

R/resource_downloader.R

@@ -0,0 +1,47 @@
+#' SparkNLP ResourceDownloader functions
+#' 
+#' ResourceDownloader provides functions to easily look for pretrained models & pipelines 
+#' inside Spark NLP. You can filter models or pipelines via language, version,
+#' or the name of the annotator
+#' 
+#' @param sc a spark_connect object
+#' @param lang language to restrict the results to
+#' @param version Spark NLP version to restrict results to
+#' 
+#' @return a markdown table containing the models or pipelines filtered by the provided arguments
+#' 
+#' @name nlp_resource_downloader
+#' @aliases ResourceDownloader 
+NULL
+
+#' @rdname nlp_resource_downloader
+#' @export
+nlp_show_public_pipelines <- function(sc, lang = NULL, version = NULL) {
+  result <- sparklyr::invoke_static(sc, "com.johnsnowlabs.nlp.pretrained.PythonResourceDownloader", "showPublicPipelines", lang, version)
+  return(result)
+}
+
+#' @param annotator name of annotator to restrict results
+#' @rdname nlp_resource_downloader
+#' @export
+nlp_show_public_models <- function(sc, annotator = NULL, lang = NULL, version = NULL) {
+  result <- sparklyr::invoke_static(sc, "com.johnsnowlabs.nlp.pretrained.PythonResourceDownloader", "showPublicModels", annotator, lang, version)
+  return(result)
+}
+
+#' @param name name of object to clear
+#' @param language language to clear
+#' @param remote_loc remote_loc of models to clear
+#' @rdname nlp_resource_downloader
+#' @export
+nlp_clear_cache <- function(sc, name = NULL, language = NULL, remote_loc = NULL) {
+  result <- sparklyr::invoke_static(sc, "com.johnsnowlabs.nlp.pretrained.PythonResourceDownloader", "clearCache", name, language, remote_loc)
+  return(result)
+}
+
+#' @rdname nlp_resource_downloader
+#' @export
+nlp_show_available_annotators <- function(sc) {
+  result <- sparklyr::invoke_static(sc, "com.johnsnowlabs.nlp.pretrained.PythonResourceDownloader", "showAvailableAnnotators")
+  return(result)
+}