R/stepfun_processor.R

In mLLMCelltype: Cell Type Annotation Using Large Language Models

#' StepFun API Processor
#' 
#' Concrete implementation of BaseAPIProcessor for StepFun models.
#' Handles StepFun-specific API calls, authentication, and response parsing.
#'
#' @importFrom R6 R6Class
#' @export
StepFunProcessor <- R6::R6Class("StepFunProcessor",
  inherit = BaseAPIProcessor,
  
  public = list(
    #' @description
    #' Initialize StepFun processor
    #
    initialize = function(base_url = NULL) {
      super$initialize("stepfun", base_url)
    },

    #' @description
    #' Get default StepFun API URL
    #
    get_default_api_url = function() {
      return("https://api.stepfun.com/v1/chat/completions")
    },
    
    #' @description
    #' Make API call to StepFun
    #
    #
    #
    #
    make_api_call = function(chunk_content, model, api_key) {
      # Prepare request body
      body <- list(
        model = model,
        messages = list(
          list(
            role = "user",
            content = chunk_content
          )
        )
      )
      
      self$logger$debug("Sending API request to StepFun",
                       list(model = model, provider = self$provider_name))
      
      # Make the API request
      response <- httr::POST(
        url = self$get_api_url(),
        httr::add_headers(
          "Authorization" = paste("Bearer", api_key),
          "Content-Type" = "application/json"
        ),
        body = jsonlite::toJSON(body, auto_unbox = TRUE),
        encode = "json"
      )
      
      # Check for HTTP errors
      if (httr::http_error(response)) {
        error_content <- tryCatch(
          httr::content(response, "parsed"),
          error = function(e) NULL
        )
        error_message <- if (is.list(error_content) && !is.null(error_content$error$message)) {
          error_content$error$message
        } else {
          sprintf("HTTP %d error", httr::status_code(response))
        }
        
        self$logger$error("StepFun API request failed",
                         list(error = error_message,
                              provider = self$provider_name,
                              model = model,
                              status_code = httr::status_code(response)))
        
        stop(sprintf("StepFun API request failed: %s", error_message))
      }
      
      return(response)
    },
    
    #' @description
    #' Extract response content from StepFun API response
    #
    #
    #
    extract_response_content = function(response, model) {
      self$logger$debug("Parsing StepFun API response",
                       list(provider = self$provider_name, model = model))
      
      # Parse the response
      content <- httr::content(response, "parsed")
      
      # Check if response has the expected structure
      if (is.null(content) || is.null(content$choices) || length(content$choices) == 0 ||
          is.null(content$choices[[1]]$message) || is.null(content$choices[[1]]$message$content)) {
        
        self$logger$error("Unexpected response format from StepFun API",
                         list(provider = self$provider_name,
                              model = model,
                              content_structure = names(content),
                              choices_available = !is.null(content$choices),
                              choices_count = if(!is.null(content$choices)) length(content$choices) else 0))
        
        stop("Unexpected response format from StepFun API")
      }
      
      # Extract the response content
      response_content <- content$choices[[1]]$message$content
      
      return(response_content)
    }
  )
)