R/embed.r

In hrbrmstr/rosette: Tools to Work with the 'Rosette' API

#' Rosette API text embedding service
#'
#' Provides the embedding for the input text. Returns a single vector of floating point
#' numbers for your input, known as a text embedding. Among other uses, a text embedding
#' enables you to calculate the similarity between two documents or two words.
#'
#' @param content either a URI or character vector of content to process
#' @param genre document genre (optional)
#' @param language Language: ISO 639-3 code (optional)
#' @export
#' @examples
#' ros_embedding("Cambridge, Massachusetts")
ros_embedding <- function(content, genre=NULL, language=NULL) {

  bdy <- list(genre=genre, language=language)

  if (grepl("^http[s]://", content)) {
    bdy$contentUri <- content
  } else {
    bdy$content <- content
  }

  res <- httr::POST(url = "https://api.rosette.com/rest/v1/text-embedding/",
                   encode="json",
                   httr::accept_json(),
                   body=bdy,
                   httr::add_headers(`X-Rosetteapi-Key` = rosette_api_key()))

  httr::stop_for_status(res)

  res <- httr::content(res, as="text", encoding="UTF-8")
  res <- jsonlite::fromJSON(res)

  res

}