Class: NitroIntelligence::Client::Handlers::ChatHandler

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

Constant Summary collapse

ALLOWED_EXTRA_PARAMETERS = 
OpenAI::Models::Chat::CompletionCreateParams.fields.keys.uniq.freeze

Instance Method Summary collapse

Constructor Details

#initialize(client:) ⇒ ChatHandler

Returns a new instance of ChatHandler.



9
10
11
 
# File 'lib/nitro_intelligence/client/handlers/chat_handler.rb', line 9

def initialize(client:)
  @client = client
end

Instance Method Details

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



13
14
15
16
 
# File 'lib/nitro_intelligence/client/handlers/chat_handler.rb', line 13

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

#perform_request(parameters: {}) ⇒ Object 



18
19
20
 
# File 'lib/nitro_intelligence/client/handlers/chat_handler.rb', line 18

def perform_request(parameters: {})
  @client.chat.completions.create(**parameters.slice(*ALLOWED_EXTRA_PARAMETERS))
end

#validate_and_resolve!(parameters, message) ⇒ Object 



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/nitro_intelligence/client/handlers/chat_handler.rb', line 22

def validate_and_resolve!(parameters, message)
  if parameters[:messages].blank? && message.present?
    parameters[:messages] ||= [{ role: "user",
                                 content: message }]
  end

  default_parameters = {
    metadata: {},
    messages: [],
    model: NitroIntelligence.model_catalog.default_text_model&.name,
    extra_headers: { "Prefer" => "wait" },
  }

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