class Multiwoven::Integrations::Source::AwsSagemakerModel::Client

def run_model(connection_config, payload)
  response = @client_runtime.invoke_endpoint(
    endpoint_name: connection_config[:endpoint_name],
    content_type: "application/json",
    body: payload
  )
  process_response(response)
rescue StandardError => e
  handle_exception(e, context: "AWS:SAGEMAKER MODEL:RUN_MODEL:EXCEPTION", type: "error")
end