omniai 2.8.2 → 2.8.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b6f5ce2ea16526510c60f251aae959edd1f9d509e17736bf04c34c20fc484f7c
-  data.tar.gz: 6713edfadd57f58f75faeae63e1b39059aa576fd65fb5e2a133e1033d8061b33
+  metadata.gz: 37bf696e25507734a39d98d3d17666a65e748fe0f3b63b3b5469270c91851e33
+  data.tar.gz: 71fe1fa42b566d2b3173fd58c08983d07956ebd05fdc30b9970c44c456cc3e94
 SHA512:
-  metadata.gz: e428cf0d49f1f076942f301b6a2e66c774101f0f54bcaa6fae3fa305be6d8af8017153fea31bccaed73fe046b0c49883eace49bd50dbb5730f774c8615d907e7
-  data.tar.gz: fa4e525fc536d34cbd1e3fdc7883d6f655ed2f5fc032bf10d8e124ceed6bd8ec34a0925b73a6c5390c77e1836d1b27a7cce976f5aaab7a824929b316fd2921bd
+  metadata.gz: c1b49be9e006dab6c28a63620ee853141059b9ff27774a62347f5283bdea58f79867c8cbcc41a3e08e1203f77e7f3464410f7b92b80708c97dfcf61484f46c27
+  data.tar.gz: ac59249678da47bf2207b2fa1d204d525b31a0c0bfabebebcf873c3910ab35433e51c60dceaf9d4fcedd6562953d8f997d3458e018edc4ed9ca5064cf23f40c4

data/lib/omniai/client.rb

@@ -207,7 +207,7 @@ module OmniAI
     # @param language [String, nil] optional
     # @param prompt [String, nil] optional
     # @param temperature [Float, nil] optional
-    # @param format [Symbol] :text, :srt, :vtt, or :json (default)
+    # @param format [String] 'text', 'srt', 'vtt', 'json' (default), or 'verbose_json'
     #
     # @return [OmniAI::Transcribe::Transcription]
     def transcribe(io, model:, language: nil, prompt: nil, temperature: nil, format: nil)

data/lib/omniai/transcribe/transcription.rb

@@ -16,18 +16,59 @@ module OmniAI
       #   @return [String]
       attr_accessor :format
 
+      # @!attribute [rw] duration
+      #   @return [Float, nil]
+      attr_accessor :duration
+
+      # @!attribute [rw] segments
+      #   @return [Array<Hash>, nil]
+      attr_accessor :segments
+
+      # @!attribute [rw] language
+      #   @return [String, nil]
+      attr_accessor :language
+
+      # @param data [Hash, String]
+      # @param model [String]
+      # @param format [String]
+      #
+      # @return [OmniAI::Transcribe::Transcription]
+      def self.parse(data:, model:, format:)
+        data = { "text" => data } if data.is_a?(String)
+
+        text = data["text"]
+        duration = data["duration"]
+        segments = data["segments"]
+        language = data["language"]
+
+        new(text:, model:, format:, duration:, segments:, language:)
+      end
+
       # @param text [String]
       # @param model [String]
       # @param format [String]
-      def initialize(text:, model:, format:)
+      # @param duration [Float, nil]
+      # @param segments [Array, nil]
+      # @param language [String, nil]
+      def initialize(text:, model:, format:, duration: nil, segments: nil, language: nil)
         @text = text
         @model = model
         @format = format
+        @duration = duration
+        @segments = segments
+        @language = language
       end
 
       # @return [String]
       def inspect
-        "#<#{self.class} text=#{text.inspect}>"
+        attrs = [
+          ("text=#{@text.inspect}" if @text),
+          ("model=#{@model.inspect}" if @model),
+          ("format=#{@format.inspect}" if @format),
+          ("duration=#{@duration.inspect}" if @duration),
+        ].compact.join(" ")
+
+        "#<#{self.class} #{attrs}>"
       end
     end
   end

data/lib/omniai/transcribe.rb

@@ -92,6 +92,7 @@ module OmniAI
       TEXT = "text"
       VTT = "vtt"
       SRT = "srt"
+      VERBOSE_JSON = "verbose_json"
     end
 
     def self.process!(...)
@@ -105,29 +106,40 @@ module OmniAI
     # @param prompt [String, nil] optional
     # @param temperature [Float, nil] optional
     # @param format [String, nil] optional
-    def initialize(io, client:, model:, language: nil, prompt: nil, temperature: nil, format: Format::JSON)
+    def initialize(io, client:, model:, language: nil, prompt: nil, temperature: nil, format: nil)
       @io = io
+      @client = client
       @model = model
       @language = language
       @prompt = prompt
       @temperature = temperature
-      @format = format
-      @client = client
+      @format = format || Format::JSON
     end
 
     # @raise [HTTPError]
+    #
     # @return [OmniAI::Transcribe::Transcription]
     def process!
       response = request!
-
       raise HTTPError, response.flush unless response.status.ok?
 
-      text = @format.nil? || @format.eql?(Format::JSON) ? response.parse["text"] : String(response.body)
-      Transcription.new(text:, model: @model, format: @format)
+      data = json? || verbose_json? ? response.parse : String(response.body)
+
+      Transcription.parse(model: @model, format: @format, data:)
     end
 
   protected
 
+    # @return [Boolean]
+    def json?
+      String(@format).eql?(Format::JSON)
+    end
+
+    # @return [Boolean]
+    def verbose_json?
+      String(@format).eql?(Format::VERBOSE_JSON)
+    end
+
     # @return [Hash]
     def payload
       {
@@ -136,9 +148,13 @@ module OmniAI
         language: @language,
         prompt: @prompt,
         temperature: @temperature,
+        response_format: @format,
+        timestamp_granularities: verbose_json? ? %w[segment] : nil,
       }.compact
     end
 
+  private
+
     # @return [String]
     def path
       raise NotImplementedError, "#{self.class.name}#path undefined"
@@ -148,7 +164,7 @@ module OmniAI
     def request!
       @client
         .connection
-        .accept(@format.eql?(Format::JSON) ? :json : :text)
+        .accept(json? || verbose_json? ? :json : :text)
         .post(path, form: payload)
     end
   end

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = "2.8.2"
+  VERSION = "2.8.3"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 2.8.2
+  version: 2.8.3
 platform: ruby
 authors:
 - Kevin Sylvestre
 bindir: exe
 cert_chain: []
-date: 2025-06-09 00:00:00.000000000 Z
+date: 2025-06-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: base64