Sha256: 1792e33a9b5ae5c3fad57bd52c9dd0c6346423bb4774d36e42754dcaa2abd8df

Contents?: true

Size: 671 Bytes

Versions: 4

Compression:

Stored size: 671 Bytes

Contents

# lib/ai_client/chat.rb

class AiClient

  ######################################
  # OmniAI Params
  #   model:        @model    [String] optional
  #   format:       @format   [Symbol] optional :text or :json
  #   stream:       @stream   [Proc, nil] optional
  #   tools:        @tools    [Array<OmniAI::Tool>] optional
  #   temperature:  @temperature  [Float, nil] optional

  def chat(messages, **params)
    result = call_with_middlewares(:chat_without_middlewares, messages, **params)
    @last_response = result
    raw? ? result : content
  end


  def chat_without_middlewares(messages, **params)
    @client.chat(messages, model: @model, **params)
  end

end

Version data entries

4 entries across 4 versions & 1 rubygems

Version Path
ai_client-0.2.5 lib/ai_client/chat.rb
ai_client-0.2.4 lib/ai_client/chat.rb
ai_client-0.2.3 lib/ai_client/chat.rb
ai_client-0.2.2 lib/ai_client/chat.rb