omniai 1.9.2 → 1.9.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4f2dff570036129e4ae7b41a113f644a19a6dd1c75e71852721c5f7fc5fdd068
-  data.tar.gz: d48375ab7e70758ad7fc931d0ac70f739807e10cc35bc22161ddcb8a758fc0b7
+  metadata.gz: eca1c768ff6bacdf521418a896e115253ad01e80848bb3a2416772b18e17cc84
+  data.tar.gz: 78a171dc3b7ccf17765166836f12daca5cc14baaa77396f680682b7c9ceb4a4b
 SHA512:
-  metadata.gz: 537c4e16730332602dccfc9fb3011f99739fdfec133362b362fded6e2fb54ec66c7070872426f331f6160dd3fdd3e82c61b7bfb326c8c2a5399fe662eb51acfc
-  data.tar.gz: 79b2adcacba078cbd8aa72330a9d8f6a1e1895e5824cccbf89bab50a86a789fdd83f49d23a70f39785912273ce8826cbbc6eba40eee2d7d671ff52a34b1bb171
+  metadata.gz: 865e34d906b82247bbd5a664610e96fdc256b40d9701b5cd2d1cb5166d1334dda233272117d7065a7dabd2e25e77a5e454ea2cb8af028f63b8998dfc575cd67c
+  data.tar.gz: 57951e245c5b6a29e5d147b0c91c6a1decb37bca7af7aabe9580a51016d332343f981d4bab733fef5350c6a40024633e0792e1ecceefc245dbe8f781d5bcf1a4

data/Gemfile

@@ -1,18 +1,19 @@
 # frozen_string_literal: true
 
-source 'https://rubygems.org'
+source "https://rubygems.org"
 
 gemspec
 
-gem 'factory_bot'
-gem 'logger'
-gem 'rake'
-gem 'rspec'
-gem 'rspec_junit_formatter'
-gem 'rubocop'
-gem 'rubocop-factory_bot'
-gem 'rubocop-rake'
-gem 'rubocop-rspec'
-gem 'simplecov'
-gem 'webmock'
-gem 'yard'
+gem "factory_bot"
+gem "irb"
+gem "rake"
+gem "rspec"
+gem "rspec_junit_formatter"
+gem "rubocop"
+gem "rubocop-basic"
+gem "rubocop-factory_bot"
+gem "rubocop-rake"
+gem "rubocop-rspec"
+gem "simplecov"
+gem "webmock"
+gem "yard"

data/README.md

@@ -9,6 +9,7 @@
 OmniAI standardizes the APIs of various AI / LLM companies such as Anthropic, Google, Mistral and OpenAI for the generation of text, the conversion of text-to-speech, the conversion of speech-to-text, the generation of embeddings, and more. It offers a unified API regardless of the provider and task.
 
 - [OmniAI::Anthropic](https://github.com/ksylvest/omniai-anthropic)
+- [OmniAI::DeepSeek](https://github.com/ksylvest/omniai-deepseek)
 - [OmniAI::Google](https://github.com/ksylvest/omniai-google)
 - [OmniAI::Mistral](https://github.com/ksylvest/omniai-mistral)
 - [OmniAI::OpenAI](https://github.com/ksylvest/omniai-openai)
@@ -167,12 +168,17 @@ search('Who do you call to fix a toilet?')
 
 ## Installation
 
+The main `omniai` gem is installed with:
+
 ```sh
 gem install omniai
 ```
 
+Specific provider gems are installed with:
+
 ```sh
 gem install omniai-anthropic
+gem install omniai-deepseek
 gem install omniai-mistral
 gem install omniai-google
 gem install omniai-openai
@@ -192,6 +198,14 @@ require 'omniai/anthropic'
 client = OmniAI::Anthropic::Client.new
 ```
 
+#### [OmniAI::DeepSeek](https://github.com/ksylvest/omniai-deepseek)
+
+```ruby
+require 'omniai/deepseek'
+
+client = OmniAI::DeepSeek::Client.new
+```
+
 #### [OmniAI::Google](https://github.com/ksylvest/omniai-google)
 
 ```ruby

data/bin/console

@@ -1,8 +1,8 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-require 'bundler/setup'
-require 'omniai'
+require "bundler/setup"
+require "omniai"
 
-require 'irb'
+require "irb"
 IRB.start(__FILE__)

data/exe/omniai

@@ -1,7 +1,7 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-require 'omniai'
+require "omniai"
 
 cli = OmniAI::CLI.new
 cli.parse

data/lib/omniai/chat/choice.rb

@@ -31,8 +31,8 @@ module OmniAI
         deserialize = context&.deserializer(:choice)
         return deserialize.call(data, context:) if deserialize
 
-        index = data['index']
-        message = Message.deserialize(data['message'] || data['delta'], context:)
+        index = data["index"]
+        message = Message.deserialize(data["message"] || data["delta"], context:)
 
         new(message:, index:)
       end

data/lib/omniai/chat/content.rb

@@ -32,10 +32,10 @@ module OmniAI
 
         return data if data.is_a?(String)
 
-        raise ArgumentError, "untyped data=#{data.inspect}" unless data.key?('type')
+        raise ArgumentError, "untyped data=#{data.inspect}" unless data.key?("type")
 
-        case data['type']
-        when 'text' then Text.deserialize(data, context:)
+        case data["type"]
+        when "text" then Text.deserialize(data, context:)
         when /(.*)_url/ then URL.deserialize(data, context:)
         else raise ArgumentError, "unknown type=#{data['type'].inspect}"
         end

data/lib/omniai/chat/file.rb

@@ -44,7 +44,7 @@ module OmniAI
       def filename
         case @io
         when Tempfile, File, String then ::File.basename(@io)
-        else 'DATA'
+        else "DATA"
         end
       end
     end

data/lib/omniai/chat/function.rb

@@ -30,11 +30,11 @@ module OmniAI
         deserialize = context&.deserializer(:function)
         return deserialize.call(data, context:) if deserialize
 
-        name = data['name']
+        name = data["name"]
         arguments = begin
-          JSON.parse(data['arguments']) if data['arguments']
+          JSON.parse(data["arguments"]) if data["arguments"]
         rescue JSON::ParserError
-          data['arguments']
+          data["arguments"]
         end
 
         new(name:, arguments:)

data/lib/omniai/chat/message.rb

@@ -30,7 +30,7 @@ module OmniAI
       #
       # @return [Message]
       def self.build(content = nil, role: Role::USER, &block)
-        raise ArgumentError, 'content or block is required' if content.nil? && block.nil?
+        raise ArgumentError, "content or block is required" if content.nil? && block.nil?
 
         Builder.build(role:) do |builder|
           builder.text(content) if content
@@ -81,9 +81,9 @@ module OmniAI
         deserialize = context&.deserializer(:message)
         return deserialize.call(data, context:) if deserialize
 
-        role = data['role']
-        content = Content.deserialize(data['content'], context:)
-        tool_call_list = data['tool_calls']&.map { |subdata| ToolCall.deserialize(subdata, context:) }
+        role = data["role"]
+        content = Content.deserialize(data["content"], context:)
+        tool_call_list = data["tool_calls"]&.map { |subdata| ToolCall.deserialize(subdata, context:) }
 
         new(content:, role:, tool_call_list:)
       end

data/lib/omniai/chat/payload.rb

@@ -28,8 +28,8 @@ module OmniAI
         deserialize = context&.deserializer(:payload)
         return deserialize.call(data, context:) if deserialize
 
-        choices = data['choices'].map { |choice_data| Choice.deserialize(choice_data, context:) }
-        usage = Usage.deserialize(data['usage'], context:) if data['usage']
+        choices = data["choices"].map { |choice_data| Choice.deserialize(choice_data, context:) }
+        usage = Usage.deserialize(data["usage"], context:) if data["usage"]
 
         new(choices:, usage:)
       end

data/lib/omniai/chat/stream.rb

@@ -23,7 +23,7 @@ module OmniAI
         end
       end
 
-      protected
+    protected
 
       # @param type [String]
       # @param data [String]
@@ -47,7 +47,7 @@ module OmniAI
       def process!(type, data, id, &block)
         log(type, data, id)
 
-        return if data.eql?('[DONE]')
+        return if data.eql?("[DONE]")
 
         block.call(Payload.deserialize(JSON.parse(data), context: @context))
       end

data/lib/omniai/chat/text.rb

@@ -28,7 +28,7 @@ module OmniAI
         deserialize = context&.deserializer(:text)
         return deserialize.call(data, context:) if deserialize
 
-        new(data['text'])
+        new(data["text"])
       end
 
       # @param context [Context] optional
@@ -38,7 +38,7 @@ module OmniAI
         serializer = context&.serializer(:text)
         return serializer.call(self, context:) if serializer
 
-        { type: 'text', text: @text }
+        { type: "text", text: @text }
       end
     end
   end

data/lib/omniai/chat/tool_call.rb

@@ -30,8 +30,8 @@ module OmniAI
         deserialize = context&.deserializer(:tool_call)
         return deserialize.call(data, context:) if deserialize
 
-        id = data['id']
-        function = Function.deserialize(data['function'], context:)
+        id = data["id"]
+        function = Function.deserialize(data["function"], context:)
 
         new(id:, function:)
       end
@@ -45,7 +45,7 @@ module OmniAI
 
         {
           id: @id,
-          type: 'function',
+          type: "function",
           function: @function.serialize(context:),
         }
       end

data/lib/omniai/chat/tool_call_message.rb

@@ -31,9 +31,9 @@ module OmniAI
         deserialize = context&.deserializer(:tool_message)
         return deserialize.call(data, context:) if deserialize
 
-        role = data['role']
-        content = JSON.parse(data['content'])
-        tool_call_id = data['tool_call_id']
+        role = data["role"]
+        content = JSON.parse(data["content"])
+        tool_call_id = data["tool_call_id"]
 
         new(role:, content:, tool_call_id:)
       end

data/lib/omniai/chat/tool_call_result.rb

@@ -41,8 +41,8 @@ module OmniAI
         deserialize = context&.deserializer(:tool_call_result)
         return deserialize.call(data, context:) if deserialize
 
-        content = JSON.parse(data['content'])
-        tool_call_id = data['tool_call_id']
+        content = JSON.parse(data["content"])
+        tool_call_id = data["tool_call_id"]
 
         new(content:, tool_call_id:)
       end

data/lib/omniai/chat/url.rb

@@ -31,8 +31,8 @@ module OmniAI
         deserialize = context&.deserializer(:url)
         return deserialize.call(data, context:) if deserialize
 
-        type = /(?<type>\w+)_url/.match(data['type'])[:type]
-        uri = data["#{type}_url"]['url']
+        type = /(?<type>\w+)_url/.match(data["type"])[:type]
+        uri = data["#{type}_url"]["url"]
 
         new(uri, type)
       end
@@ -68,7 +68,7 @@ module OmniAI
         ::File.basename(@uri)
       end
 
-      protected
+    protected
 
       # @raise [FetchError]
       #

data/lib/omniai/chat/usage.rb

@@ -35,9 +35,9 @@ module OmniAI
         deserialize = context&.deserializer(:usage)
         return deserialize.call(data, context:) if deserialize
 
-        input_tokens = data['input_tokens']
-        output_tokens = data['output_tokens']
-        total_tokens = data['total_tokens']
+        input_tokens = data["input_tokens"]
+        output_tokens = data["output_tokens"]
+        total_tokens = data["total_tokens"]
 
         new(input_tokens:, output_tokens:, total_tokens:)
       end

data/lib/omniai/chat.rb

@@ -25,7 +25,7 @@ module OmniAI
   #
   #   client.chat(messages, model: "...", temperature: 0.0, format: :text)
   class Chat
-    JSON_PROMPT = 'Respond with valid JSON. Do not include any non-JSON in the response.'
+    JSON_PROMPT = "Respond with valid JSON. Do not include any non-JSON in the response."
 
     # An error raised for tool-call issues.
     class ToolCallError < Error
@@ -45,10 +45,10 @@ module OmniAI
     end
 
     module Role
-      ASSISTANT = 'assistant'
-      USER = 'user'
-      SYSTEM = 'system'
-      TOOL = 'tool'
+      ASSISTANT = "assistant"
+      USER = "user"
+      SYSTEM = "system"
+      TOOL = "tool"
     end
 
     module Format
@@ -72,7 +72,7 @@ module OmniAI
     #
     # @return [OmniAi::Chat]
     def initialize(prompt = nil, client:, model:, temperature: nil, stream: nil, tools: nil, format: nil, &block)
-      raise ArgumentError, 'prompt or block is required' if !prompt && !block
+      raise ArgumentError, "prompt or block is required" if !prompt && !block
 
       @prompt = prompt ? Prompt.parse(prompt) : Prompt.new
       block&.call(@prompt)
@@ -94,7 +94,7 @@ module OmniAI
       parse!(response:)
     end
 
-    protected
+  protected
 
     # Override  to provide an context for serializers / deserializes for a provider.
     #

data/lib/omniai/cli/base_handler.rb

@@ -8,14 +8,14 @@ module OmniAI
       # @param stdout [IO] an optional stream for stdout
       # @param provider [String] an optional provider (defaults to 'openai')
       # @param argv [Array<String>]
-      def self.handle!(argv:, stdin: $stdin, stdout: $stdout, provider: 'openai')
+      def self.handle!(argv:, stdin: $stdin, stdout: $stdout, provider: "openai")
         new(stdin:, stdout:, provider:).handle!(argv:)
       end
 
       # @param stdin [IO] an optional stream for stdin
       # @param stdout [IO] an optional stream for stdout
       # @param provider [String] an optional provider (defaults to 'openai')
-      def initialize(stdin: $stdin, stdout: $stdout, provider: 'openai')
+      def initialize(stdin: $stdin, stdout: $stdout, provider: "openai")
         @stdin = stdin
         @stdout = stdout
         @provider = provider
@@ -27,7 +27,7 @@ module OmniAI
         raise NotImplementedError, "#{self.class}#handle! undefined"
       end
 
-      private
+    private
 
       # @return [OmniAI::Client]
       def client

data/lib/omniai/cli/chat_handler.rb

@@ -11,17 +11,17 @@ module OmniAI
         if argv.empty?
           listen!
         else
-          chat(prompt: argv.join(' '))
+          chat(prompt: argv.join(" "))
         end
       end
 
-      private
+    private
 
       def listen!
         @stdout.puts('Type "exit" or "quit" to leave.')
 
         loop do
-          @stdout.print('# ')
+          @stdout.print("# ")
           @stdout.flush
           prompt = @stdin.gets&.chomp
 
@@ -43,17 +43,17 @@ module OmniAI
         OptionParser.new do |options|
           options.banner = 'usage: omniai chat [options] "<prompt>"'
 
-          options.on('-h', '--help', 'help') do
+          options.on("-h", "--help", "help") do
             @stdout.puts(options)
             exit
           end
 
-          options.on('-p', '--provider=PROVIDER', 'provider') { |provider| @provider = provider }
-          options.on('-m', '--model=MODEL', 'model') { |model| @args[:model] = model }
-          options.on('-t', '--temperature=TEMPERATURE', Float, 'temperature') do |temperature|
+          options.on("-p", "--provider=PROVIDER", "provider") { |provider| @provider = provider }
+          options.on("-m", "--model=MODEL", "model") { |model| @args[:model] = model }
+          options.on("-t", "--temperature=TEMPERATURE", Float, "temperature") do |temperature|
             @args[:temperature] = temperature
           end
-          options.on('-f', '--format=FORMAT', 'format') { |format| @args[:format] = format.intern }
+          options.on("-f", "--format=FORMAT", "format") { |format| @args[:format] = format.intern }
         end
       end
     end

data/lib/omniai/cli/embed_handler.rb

@@ -11,17 +11,17 @@ module OmniAI
         if argv.empty?
           listen!
         else
-          embed(input: argv.join(' '))
+          embed(input: argv.join(" "))
         end
       end
 
-      private
+    private
 
       def listen!
         @stdout.puts('Type "exit" or "quit" to leave.')
 
         loop do
-          @stdout.print('# ')
+          @stdout.print("# ")
           @stdout.flush
           input = @stdin.gets&.chomp
 
@@ -44,13 +44,13 @@ module OmniAI
         OptionParser.new do |options|
           options.banner = 'usage: omniai embed [options] "<prompt>"'
 
-          options.on('-h', '--help', 'help') do
+          options.on("-h", "--help", "help") do
             @stdout.puts(options)
             exit
           end
 
-          options.on('-p', '--provider=PROVIDER', 'provider') { |provider| @provider = provider }
-          options.on('-m', '--model=MODEL', 'model') { |model| @args[:model] = model }
+          options.on("-p", "--provider=PROVIDER", "provider") { |provider| @provider = provider }
+          options.on("-m", "--model=MODEL", "model") { |model| @args[:model] = model }
         end
       end
     end

data/lib/omniai/cli.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'optparse'
+require "optparse"
 
 module OmniAI
   # Used when interacting with the suite from the command line interface (CLI).
@@ -15,7 +15,7 @@ module OmniAI
     # @param stdin [IO] a stream
     # @param stdout [IO] a stream
     # @param provider [String] a provider
-    def initialize(stdin: $stdin, stdout: $stdout, provider: 'openai')
+    def initialize(stdin: $stdin, stdout: $stdout, provider: "openai")
       @stdin = stdin
       @stdout = stdout
       @provider = provider
@@ -30,32 +30,32 @@ module OmniAI
 
       handler =
         case command
-        when 'chat' then ChatHandler
-        when 'embed' then EmbedHandler
+        when "chat" then ChatHandler
+        when "embed" then EmbedHandler
         else raise Error, "unsupported command=#{command.inspect}"
         end
 
       handler.handle!(stdin: @stdin, stdout: @stdout, provider: @provider, argv:)
     end
 
-    private
+  private
 
     # @return [OptionParser]
     def parser
       OptionParser.new do |options|
-        options.banner = 'usage: omniai [options] <command> [<args>]'
+        options.banner = "usage: omniai [options] <command> [<args>]"
 
-        options.on('-h', '--help', 'help') do
+        options.on("-h", "--help", "help") do
           @stdout.puts(options)
           exit
         end
 
-        options.on('-v', '--version', 'version') do
+        options.on("-v", "--version", "version") do
           @stdout.puts(VERSION)
           exit
         end
 
-        options.on('-p', '--provider=PROVIDER', 'provider (default="openai")') do |provider|
+        options.on("-p", "--provider=PROVIDER", 'provider (default="openai")') do |provider|
           @provider = provider
         end
 

data/lib/omniai/client.rb

@@ -33,7 +33,7 @@ module OmniAI
     # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
     # @return [Class<OmniAI::Client>]
     def self.anthropic
-      require 'omniai/anthropic' unless defined?(OmniAI::Anthropic::Client)
+      require "omniai/anthropic" unless defined?(OmniAI::Anthropic::Client)
       OmniAI::Anthropic::Client
     rescue LoadError
       raise Error, "requires 'omniai-anthropic': `gem install omniai-anthropic`"
@@ -44,7 +44,7 @@ module OmniAI
     # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
     # @return [Class<OmniAI::Client>]
     def self.google
-      require 'omniai/google' unless defined?(OmniAI::Google::Client)
+      require "omniai/google" unless defined?(OmniAI::Google::Client)
       OmniAI::Google::Client
     rescue LoadError
       raise Error, "requires 'omniai-google': `gem install omniai-google`"
@@ -55,7 +55,7 @@ module OmniAI
     # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
     # @return [Class<OmniAI::Client>]
     def self.mistral
-      require 'omniai/mistral' unless defined?(OmniAI::Mistral::Client)
+      require "omniai/mistral" unless defined?(OmniAI::Mistral::Client)
       OmniAI::Mistral::Client
     rescue LoadError
       raise Error, "requires 'omniai-mistral': `gem install omniai-mistral`"
@@ -66,7 +66,7 @@ module OmniAI
     # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
     # @return [Class<OmniAI::Client>]
     def self.openai
-      require 'omniai/openai' unless defined?(OmniAI::OpenAI::Client)
+      require "omniai/openai" unless defined?(OmniAI::OpenAI::Client)
       OmniAI::OpenAI::Client
     rescue LoadError
       raise Error, "requires 'omniai-openai': `gem install omniai-openai`"
@@ -80,10 +80,10 @@ module OmniAI
     def self.find(provider:, **)
       klass =
         case provider
-        when :anthropic, 'anthropic' then anthropic
-        when :google, 'google' then google
-        when :mistral, 'mistral' then mistral
-        when :openai, 'openai' then openai
+        when :anthropic, "anthropic" then anthropic
+        when :google, "google" then google
+        when :mistral, "mistral" then mistral
+        when :openai, "openai" then openai
         else raise Error, "unknown provider=#{provider.inspect}"
         end
 

data/lib/omniai/embed/response.rb

@@ -27,8 +27,8 @@ module OmniAI
           if deserializer
             deserializer.call(@data, context: @context)
           else
-            prompt_tokens = @data.dig('usage', 'prompt_tokens')
-            total_tokens = @data.dig('usage', 'total_tokens')
+            prompt_tokens = @data.dig("usage", "prompt_tokens")
+            total_tokens = @data.dig("usage", "total_tokens")
 
             Usage.new(prompt_tokens:, total_tokens:)
           end
@@ -50,7 +50,7 @@ module OmniAI
           if deserializer
             deserializer.call(@data, context: @context)
           else
-            @data['data'].map { |embedding| embedding['embedding'] }
+            @data["data"].map { |embedding| embedding["embedding"] }
           end
         end
       end

data/lib/omniai/embed.rb

@@ -51,7 +51,7 @@ module OmniAI
       parse!(response:)
     end
 
-    protected
+  protected
 
     # @param response [HTTP::Response]
     # @return [Response]

data/lib/omniai/files.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module OmniAI
+  # A manager for files.
+  class Files
+    # @param client [OmniAI::Client]
+    def initialize(client:)
+      @client = client
+    end
+  end
+end

data/lib/omniai/speak.rb

@@ -26,12 +26,12 @@ module OmniAI
   #   client.transcribe(File.open("..."), model: "...", format: :json)
   class Speak
     module Format
-      AAC = 'aac'
-      FLAC = 'flac'
-      MP3 = 'mp3'
-      OPUS = 'opus'
-      PCM = 'pcm'
-      WAV = 'wav'
+      AAC = "aac"
+      FLAC = "flac"
+      MP3 = "mp3"
+      OPUS = "opus"
+      PCM = "pcm"
+      WAV = "wav"
     end
 
     # @raise [HTTPError]
@@ -94,7 +94,7 @@ module OmniAI
       end
     end
 
-    protected
+  protected
 
     # @param response [HTTP::Response]
     #

data/lib/omniai/tool/array.rb

@@ -18,7 +18,7 @@ module OmniAI
     #      max_items: 5,
     #    })
     class Array
-      TYPE = 'array'
+      TYPE = "array"
 
       # @!attribute [rw] items
       #   @return [OmniAI::Tool::Object, OmniAI::Tool::Array, OmniAI::Tool::Property]

data/lib/omniai/tool/object.rb

@@ -13,7 +13,7 @@ module OmniAI
     #      required: %i[name]
     #    })
     class Object
-      TYPE = 'object'
+      TYPE = "object"
 
       # @!attribute [rw] properties
       #   @return [Hash]

data/lib/omniai/tool/property.rb

@@ -13,10 +13,10 @@ module OmniAI
     #   OmniAI::Tool::Property.boolean(description: '...')
     class Property
       module Type
-        BOOLEAN = 'boolean'
-        INTEGER = 'integer'
-        STRING = 'string'
-        NUMBER = 'number'
+        BOOLEAN = "boolean"
+        INTEGER = "integer"
+        STRING = "string"
+        NUMBER = "number"
       end
 
       # @return [String]

data/lib/omniai/tool.rb

@@ -70,7 +70,7 @@ module OmniAI
       return serialize.call(self, context:) if serialize
 
       {
-        type: 'function',
+        type: "function",
         function: {
           name: @name,
           description: @description,

data/lib/omniai/transcribe.rb

@@ -26,70 +26,70 @@ module OmniAI
   #   client.transcribe(File.open("..."), model: "...", format: :json)
   class Transcribe
     module Language
-      AFRIKAANS = 'af'
-      ARABIC = 'ar'
-      ARMENIAN = 'hy'
-      AZERBAIJANI = 'az'
-      BELARUSIAN = 'be'
-      BOSNIAN = 'bs'
-      BULGARIAN = 'bg'
-      CATALAN = 'ca'
-      CHINESE = 'zh'
-      CROATIAN = 'hr'
-      CZECH = 'cs'
-      DANISH = 'da'
-      DUTCH = 'nl'
-      ENGLISH = 'en'
-      ESTONIAN = 'et'
-      FINNISH = 'fi'
-      FRENCH = 'fr'
-      GALICIAN = 'gl'
-      GERMAN = 'de'
-      GREEK = 'el'
-      HEBREW = 'he'
-      HINDI = 'hi'
-      HUNGARIAN = 'hu'
-      ICELANDIC = 'is'
-      INDONESIAN = 'id'
-      ITALIAN = 'it'
-      JAPANESE = 'ja'
-      KANNADA = 'kn'
-      KAZAKH = 'kk'
-      KOREAN = 'ko'
-      LATVIAN = 'lv'
-      LITHUANIAN = 'lt'
-      MACEDONIAN = 'mk'
-      MALAY = 'ms'
-      MARATHI = 'mr'
-      MAORI = 'mi'
-      NEPALI = 'ne'
-      NORWEGIAN = 'no'
-      PERSIAN = 'fa'
-      POLISH = 'pl'
-      PORTUGUESE = 'pt'
-      ROMANIAN = 'ro'
-      RUSSIAN = 'ru'
-      SERBIAN = 'sr'
-      SLOVAK = 'sk'
-      SLOVENIAN = 'sl'
-      SPANISH = 'es'
-      SWAHILI = 'sw'
-      SWEDISH = 'sv'
-      TAGALOG = 'tl'
-      TAMIL = 'ta'
-      THAI = 'th'
-      TURKISH = 'tr'
-      UKRAINIAN = 'uk'
-      URDU = 'ur'
-      VIETNAMESE = 'vi'
-      WELSH = 'cy'
+      AFRIKAANS = "af"
+      ARABIC = "ar"
+      ARMENIAN = "hy"
+      AZERBAIJANI = "az"
+      BELARUSIAN = "be"
+      BOSNIAN = "bs"
+      BULGARIAN = "bg"
+      CATALAN = "ca"
+      CHINESE = "zh"
+      CROATIAN = "hr"
+      CZECH = "cs"
+      DANISH = "da"
+      DUTCH = "nl"
+      ENGLISH = "en"
+      ESTONIAN = "et"
+      FINNISH = "fi"
+      FRENCH = "fr"
+      GALICIAN = "gl"
+      GERMAN = "de"
+      GREEK = "el"
+      HEBREW = "he"
+      HINDI = "hi"
+      HUNGARIAN = "hu"
+      ICELANDIC = "is"
+      INDONESIAN = "id"
+      ITALIAN = "it"
+      JAPANESE = "ja"
+      KANNADA = "kn"
+      KAZAKH = "kk"
+      KOREAN = "ko"
+      LATVIAN = "lv"
+      LITHUANIAN = "lt"
+      MACEDONIAN = "mk"
+      MALAY = "ms"
+      MARATHI = "mr"
+      MAORI = "mi"
+      NEPALI = "ne"
+      NORWEGIAN = "no"
+      PERSIAN = "fa"
+      POLISH = "pl"
+      PORTUGUESE = "pt"
+      ROMANIAN = "ro"
+      RUSSIAN = "ru"
+      SERBIAN = "sr"
+      SLOVAK = "sk"
+      SLOVENIAN = "sl"
+      SPANISH = "es"
+      SWAHILI = "sw"
+      SWEDISH = "sv"
+      TAGALOG = "tl"
+      TAMIL = "ta"
+      THAI = "th"
+      TURKISH = "tr"
+      UKRAINIAN = "uk"
+      URDU = "ur"
+      VIETNAMESE = "vi"
+      WELSH = "cy"
     end
 
     module Format
-      JSON = 'json'
-      TEXT = 'text'
-      VTT = 'vtt'
-      SRT = 'srt'
+      JSON = "json"
+      TEXT = "text"
+      VTT = "vtt"
+      SRT = "srt"
     end
 
     def self.process!(...)
@@ -120,11 +120,11 @@ module OmniAI
 
       raise HTTPError, response.flush unless response.status.ok?
 
-      text = @format.nil? || @format.eql?(Format::JSON) ? response.parse['text'] : String(response.body)
+      text = @format.nil? || @format.eql?(Format::JSON) ? response.parse["text"] : String(response.body)
       Transcription.new(text:, model: @model, format: @format)
     end
 
-    protected
+  protected
 
     # @return [Hash]
     def payload

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = '1.9.2'
+  VERSION = "1.9.4"
 end

data/lib/omniai.rb

@@ -1,15 +1,15 @@
 # frozen_string_literal: true
 
-require 'logger'
-require 'event_stream_parser'
-require 'http'
-require 'uri'
-require 'zeitwerk'
+require "logger"
+require "event_stream_parser"
+require "http"
+require "uri"
+require "zeitwerk"
 
 loader = Zeitwerk::Loader.for_gem
-loader.inflector.inflect 'omniai' => 'OmniAI'
-loader.inflector.inflect 'url' => 'URL'
-loader.inflector.inflect 'cli' => 'CLI'
+loader.inflector.inflect "omniai" => "OmniAI"
+loader.inflector.inflect "url" => "URL"
+loader.inflector.inflect "cli" => "CLI"
 loader.setup
 
 module OmniAI

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.9.2
+  version: 1.9.4
 platform: ruby
 authors:
 - Kevin Sylvestre
-autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-11-05 00:00:00.000000000 Z
+date: 2025-01-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -38,6 +37,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: logger
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: zeitwerk
   requirement: !ruby/object:Gem::Requirement
@@ -95,6 +108,7 @@ files:
 - lib/omniai/embed.rb
 - lib/omniai/embed/response.rb
 - lib/omniai/embed/usage.rb
+- lib/omniai/files.rb
 - lib/omniai/instrumentation.rb
 - lib/omniai/speak.rb
 - lib/omniai/tool.rb
@@ -112,7 +126,6 @@ metadata:
   homepage_uri: https://github.com/ksylvest/omniai
   changelog_uri: https://github.com/ksylvest/omniai/releases
   rubygems_mfa_required: 'true'
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -127,8 +140,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.18
-signing_key: 
+rubygems_version: 3.6.2
 specification_version: 4
 summary: A generalized framework for interacting with many AI services
 test_files: []