Class: OmniAI::Client

Inherits:
Object
  • Object
show all
Defined in:
lib/omniai/client.rb

Overview

An abstract class that must be subclassed (e.g. OmniAI::OpenAI::Client).

Usage:

class OmniAI::OpenAI::Client < OmniAI::Client
  def initialize(api_key: ENV.fetch('OPENAI_API_KEY'), logger: nil)
    super
  end

  # @return [HTTP::Client]
  def connection
    @connection ||= super.auth("Bearer: #{@api_key}")
  end
end

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(api_key: nil, logger: nil, host: nil, timeout: nil) ⇒ Client

Returns a new instance of Client.

Parameters:

  • api_key (String, nil) (defaults to: nil) —

    optional

  • host (String, nil) (defaults to: nil) —

    optional - supports for customzing the host of the client (e.g. ‘localhost:8080’)

  • logger (Logger, nil) (defaults to: nil) —

    optional

  • timeout (Integer, nil) (defaults to: nil) —

    optional



120
121
122
123
124
125
 
# File 'lib/omniai/client.rb', line 120

def initialize(api_key: nil, logger: nil, host: nil, timeout: nil)
  @api_key = api_key
  @host = host
  @logger = logger
  @timeout = timeout
end

Instance Attribute Details

#api_key ⇒ String?

Returns:

  • (String, nil) 


21
22
23
 
# File 'lib/omniai/client.rb', line 21

def api_key
  @api_key
end

#host ⇒ String?

Returns:

  • (String, nil) 


29
30
31
 
# File 'lib/omniai/client.rb', line 29

def host
  @host
end

#logger ⇒ Logger?

Returns:

  • (Logger, nil) 


25
26
27
 
# File 'lib/omniai/client.rb', line 25

def logger
  @logger
end

#timeout ⇒ Integer?

Returns:

  • (Integer, nil) 


33
34
35
 
# File 'lib/omniai/client.rb', line 33

def timeout
  @timeout
end

Class Method Details

.anthropic ⇒ Class<OmniAI::Client>

Initialize a client for Anthropic. This method requires the provider if it is undefined.

Returns:

Raises:

  • (OmniAI::Error) —

    if the provider is not defined and the gem is not installed



40
41
42
43
44
45
 
# File 'lib/omniai/client.rb', line 40

def self.anthropic
  require "omniai/anthropic" unless defined?(OmniAI::Anthropic::Client)
  OmniAI::Anthropic::Client
rescue ::LoadError
  raise Error, "requires 'omniai-anthropic': `gem install omniai-anthropic`"
end

.deepseek ⇒ Class<OmniAI::Client>

Initialize a client for DeepSeek. This method requires the provider if it is undefined.

Returns:

Raises:

  • (OmniAI::Error) —

    if the provider is not defined and the gem is not installed



52
53
54
55
56
57
 
# File 'lib/omniai/client.rb', line 52

def self.deepseek
  require "omniai/deepseek" unless defined?(OmniAI::DeepSeek::Client)
  OmniAI::DeepSeek::Client
rescue ::LoadError
  raise Error, "requires 'omniai-deepseek': `gem install omniai-deepseek`"
end

.find(provider:) ⇒ OmniAI::Client

Initialize a client by provider (e.g. ‘openai’). This method attempts to require the provider.

Parameters:

  • provider (String, Symbol) —

    required (e.g. ‘anthropic’, ‘deepsek’, ‘google’, ‘mistral’, ‘openai’, etc)

Returns:

Raises:

  • (OmniAI::Error) —

    if the provider is not defined and the gem is not installed



102
103
104
105
106
107
108
109
110
111
112
113
114
 
# File 'lib/omniai/client.rb', line 102

def self.find(provider:, **)
  klass =
    case provider
    when :anthropic, "anthropic" then anthropic
    when :deepseek, "deepseek" then deepseek
    when :google, "google" then google
    when :mistral, "mistral" then mistral
    when :openai, "openai" then openai
    else raise Error, "unknown provider=#{provider.inspect}"
    end

  klass.new(**)
end

.google ⇒ Class<OmniAI::Client>

Lookup the ‘OmniAI::Google::Client“. This method requires the provider if it is undefined.

Returns:

Raises:

  • (OmniAI::Error) —

    if the provider is not defined and the gem is not installed



64
65
66
67
68
69
 
# File 'lib/omniai/client.rb', line 64

def self.google
  require "omniai/google" unless defined?(OmniAI::Google::Client)
  OmniAI::Google::Client
rescue ::LoadError
  raise Error, "requires 'omniai-google': `gem install omniai-google`"
end

.mistral ⇒ Class<OmniAI::Client>

Initialize a client for Mistral. This method requires the provider if it is undefined.

Returns:

Raises:

  • (OmniAI::Error) —

    if the provider is not defined and the gem is not installed



76
77
78
79
80
81
 
# File 'lib/omniai/client.rb', line 76

def self.mistral
  require "omniai/mistral" unless defined?(OmniAI::Mistral::Client)
  OmniAI::Mistral::Client
rescue ::LoadError
  raise Error, "requires 'omniai-mistral': `gem install omniai-mistral`"
end

.openai ⇒ Class<OmniAI::Client>

Initialize a client for OpenAI. This method requires the provider if it is undefined.

Returns:

Raises:

  • (OmniAI::Error) —

    if the provider is not defined and the gem is not installed



88
89
90
91
92
93
 
# File 'lib/omniai/client.rb', line 88

def self.openai
  require "omniai/openai" unless defined?(OmniAI::OpenAI::Client)
  OmniAI::OpenAI::Client
rescue ::LoadError
  raise Error, "requires 'omniai-openai': `gem install omniai-openai`"
end

Instance Method Details

#chat(messages = nil, model:, temperature: nil, format: nil, stream: nil, tools: nil) {|prompt| ... } ⇒ OmniAI::Chat::Response

Parameters:

  • messages (String) (defaults to: nil) —

    optional

  • model (String) —

    optional

  • format (Symbol) (defaults to: nil) —

    optional :text or :json

  • temperature (Float, nil) (defaults to: nil) —

    optional

  • stream (Proc, nil) (defaults to: nil) —

    optional

  • tools (Array<OmniAI::Tool>) (defaults to: nil) —

    optional

Yields:

  • (prompt) —

    optional

Yield Parameters:

Returns:

Raises:



161
162
163
 
# File 'lib/omniai/client.rb', line 161

def chat(messages = nil, model:, temperature: nil, format: nil, stream: nil, tools: nil, &)
  raise NotImplementedError, "#{self.class.name}#chat undefined"
end

#connection ⇒ HTTP::Client

Returns:

  • (HTTP::Client) 


141
142
143
144
145
146
 
# File 'lib/omniai/client.rb', line 141

def connection
  http = HTTP.persistent(@host)
  http = http.use(instrumentation: { instrumenter: Instrumentation.new(logger: @logger) }) if @logger
  http = http.timeout(@timeout) if @timeout
  http
end

#embed(input, model:) ⇒ OmniAI::Embed::Embedding

Parameters:

  • input (String) —

    required

  • model (String) —

    required

Returns:

  • (OmniAI::Embed::Embedding)

Raises:



206
207
208
 
# File 'lib/omniai/client.rb', line 206

def embed(input, model:)
  raise NotImplementedError, "#{self.class.name}#embed undefined"
end

#inspect ⇒ String

Returns:

  • (String) 


128
129
130
131
132
133
 
# File 'lib/omniai/client.rb', line 128

def inspect
  props = []
  props << "api_key=#{masked_api_key.inspect}" if @api_key
  props << "host=#{@host.inspect}" if @host
  "#<#{self.class.name} #{props.join(' ')}>"
end

#masked_api_key ⇒ String

Returns:

  • (String) 


136
137
138
 
# File 'lib/omniai/client.rb', line 136

def masked_api_key
  "#{api_key[..2]}***" if api_key
end

#speak(input, model:, voice:, speed: nil, format: nil) {|output| ... } ⇒ Tempfile

Parameters:

  • input (String) —

    required

  • model (String) —

    required

  • voice (String) —

    required

  • speed (Float) (defaults to: nil) —

    optional

  • format (String) (defaults to: nil) —

    optional (default “aac”):

    • “aac”

    • “mp3”

    • “flac”

    • “opus”

    • “pcm”

    • “wav”

Yields:

  • (output) —

    optional

Returns:

  • (Tempfile)

Raises:



196
197
198
 
# File 'lib/omniai/client.rb', line 196

def speak(input, model:, voice:, speed: nil, format: nil, &stream)
  raise NotImplementedError, "#{self.class.name}#speak undefined"
end

#transcribe(io, model:, language: nil, prompt: nil, temperature: nil, format: nil) ⇒ OmniAI::Transcribe::Transcription

Parameters:

  • io (String, Pathname, IO) —

    required

  • model (String)
  • language (String, nil) (defaults to: nil) —

    optional

  • prompt (String, nil) (defaults to: nil) —

    optional

  • temperature (Float, nil) (defaults to: nil) —

    optional

  • format (Symbol) (defaults to: nil) —

    :text, :srt, :vtt, or :json (default)

Returns:

Raises:



175
176
177
 
# File 'lib/omniai/client.rb', line 175

def transcribe(io, model:, language: nil, prompt: nil, temperature: nil, format: nil)
  raise NotImplementedError, "#{self.class.name}#transcribe undefined"
end