omniai 2.8.2 → 2.8.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b6f5ce2ea16526510c60f251aae959edd1f9d509e17736bf04c34c20fc484f7c
-  data.tar.gz: 6713edfadd57f58f75faeae63e1b39059aa576fd65fb5e2a133e1033d8061b33
+  metadata.gz: 533dd085c62bb70b08d71d16025423b981d96cbffe7a1280072908f175800d28
+  data.tar.gz: c40684be74123a132c11ecc70a620ee49d4b6e949daf08f91c8f5a360d0ea56f
 SHA512:
-  metadata.gz: e428cf0d49f1f076942f301b6a2e66c774101f0f54bcaa6fae3fa305be6d8af8017153fea31bccaed73fe046b0c49883eace49bd50dbb5730f774c8615d907e7
-  data.tar.gz: fa4e525fc536d34cbd1e3fdc7883d6f655ed2f5fc032bf10d8e124ceed6bd8ec34a0925b73a6c5390c77e1836d1b27a7cce976f5aaab7a824929b316fd2921bd
+  metadata.gz: 355abe067bdc6a6156300ea305b441a14388239506565a753f12dfc7824fe2d1a7acee4772fe8556fec5dbee2a9a3a9e25506ddb9faa7d78d3838bf647d17424
+  data.tar.gz: 03d292f24d8345c4f4043f185b4c5e01ce09894afc7e9cd72e81f251cd9c394c8af2165f2c054bcb0cff60f50f75498efb90f4eb79e724e829b86561420819ff

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/mcp/server.rb

@@ -19,6 +19,8 @@ module OmniAI
 
       # @param transport [OmniAI::MCP::Transport]
       def run(transport: OmniAI::MCP::Transport::Stdio.new)
+        @logger&.info("#{self.class}#run: running")
+
         loop do
           message = transport.gets
           break if message.nil?
@@ -29,9 +31,9 @@ module OmniAI
 
           transport.puts(response) if response
         end
-      end
 
-    private
+        @logger&.info("#{self.class}#run: finished")
+      end
 
       # @param message [String]
       #
@@ -68,7 +70,9 @@ module OmniAI
             name: @name,
             version: @version,
           },
-          capabilities: {},
+          capabilities: {
+            tools: { listChanged: true },
+          },
         })
       end
 
@@ -84,7 +88,7 @@ module OmniAI
       #
       # @return [JRPC::Response]
       def process_tools_list(request:)
-        result = @tools.map do |tool|
+        tools = @tools.map do |tool|
           {
             name: tool.name,
             description: tool.description,
@@ -92,7 +96,7 @@ module OmniAI
           }
         end
 
-        JRPC::Response.new(id: request.id, result:)
+        JRPC::Response.new(id: request.id, result: { tools: })
       end
 
       # @param request [JRPC::Request]
@@ -104,14 +108,21 @@ module OmniAI
         name = request.params["name"]
         tool = @tools.find { |tool| tool.name.eql?(name) }
 
-        result =
+        text =
           begin
-            tool.call(request.params["input"])
+            tool.call(request.params["arguments"])
           rescue StandardError => e
             raise JRPC::Error.new(code: JRPC::Error::Code::INTERNAL_ERROR, message: e.message)
           end
 
-        JRPC::Response.new(id: request.id, result:)
+        JRPC::Response.new(id: request.id, result: {
+          content: [
+            {
+              type: "text",
+              text:,
+            },
+          ],
+        })
       end
     end
   end

data/lib/omniai/parse_error.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module OmniAI
+  # An error that wraps a JSON::ParserError when using schemas.
+  class ParseError < Error
+  end
+end

data/lib/omniai/schema/format.rb

@@ -62,9 +62,13 @@ module OmniAI
       #
       # @param text [String]
       #
+      # @raise OmniAI::ParseError
+      #
       # @return [Hash]
       def parse(text)
         schema.parse(JSON.parse(text))
+      rescue JSON::ParserError => e
+        raise OmniAI::ParseError, "Unable to parse JSON text=#{text.inspect} message=#{e.message.inspect}."
       end
     end
   end

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.4"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 2.8.2
+  version: 2.8.4
 platform: ruby
 authors:
 - Kevin Sylvestre
 bindir: exe
 cert_chain: []
-date: 2025-06-09 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: base64
@@ -136,6 +136,7 @@ files:
 - lib/omniai/mcp/server.rb
 - lib/omniai/mcp/transport/base.rb
 - lib/omniai/mcp/transport/stdio.rb
+- lib/omniai/parse_error.rb
 - lib/omniai/schema.rb
 - lib/omniai/schema/array.rb
 - lib/omniai/schema/format.rb
@@ -168,7 +169,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.3
+rubygems_version: 3.7.2
 specification_version: 4
 summary: A generalized framework for interacting with many AI services
 test_files: []