Class: NitroIntelligence::Client::Handlers::AudioTranscriptionHandler

Inherits:
BaseHandler
  • Object
show all
Defined in:
lib/nitro_intelligence/client/handlers/audio_transcription_handler.rb

Constant Summary collapse

ALLOWED_EXTRA_PARAMETERS = 
OpenAI::Models::Audio::TranscriptionCreateParams.fields.keys.uniq.freeze

Constants inherited from BaseHandler

BaseHandler::MODALITY_HEADER, BaseHandler::REQUESTED_MODEL_HEADER

Instance Method Summary collapse

Methods inherited from BaseHandler

#initialize

Constructor Details

This class inherits a constructor from NitroIntelligence::Client::Handlers::BaseHandler

Instance Method Details

#create(audio_file:, message: "", parameters: {}) ⇒ Object 



10
11
12
13
 
# File 'lib/nitro_intelligence/client/handlers/audio_transcription_handler.rb', line 10

def create(audio_file:, message: "", parameters: {})
  validate_and_resolve!(parameters)
  perform_request(audio_file:, message:, parameters:)
end

#perform_request(audio_file:, message: "", parameters: {}) ⇒ Object 



15
16
17
18
19
20
21
22
 
# File 'lib/nitro_intelligence/client/handlers/audio_transcription_handler.rb', line 15

def perform_request(audio_file:, message: "", parameters: {})
  add_request_headers(parameters, MODALITY_HEADER => "audio", REQUESTED_MODEL_HEADER => parameters[:model])
  @client.audio.transcriptions.create(
    prompt: message,
    file: audio_file,
    **parameters.slice(*ALLOWED_EXTRA_PARAMETERS)
  )
end

#validate_and_resolve!(parameters) ⇒ Object 



24
25
26
27
28
29
30
31
32
 
# File 'lib/nitro_intelligence/client/handlers/audio_transcription_handler.rb', line 24

def validate_and_resolve!(parameters)
  default_parameters = {
    metadata: {},
    model: NitroIntelligence.model_catalog.default_audio_transcription_model&.name,
  }

  parameters.replace(default_parameters.merge(parameters))
  Client.validate_model(parameters[:model])
end