omniai 2.3.0 → 2.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c51b267abaf795fb5fcef9ec71af335e1ac5ecb70339781692e70f6956eddbb2
-  data.tar.gz: aa06c0f668458bce2bbf396c3e512179950b6916b601b1df8d90b1766726f154
+  metadata.gz: 8b888dff5046dce34b72c4e1e37cad2ed2ce0a9a5259b1a998cc85567dc3f202
+  data.tar.gz: 14d100fd069b47cc0cd6a40794a0582639baf07384050d0e47a574a0062e01b9
 SHA512:
-  metadata.gz: abb95079f33a91d0fa842e8f6ec6980cadbee8669f8aa2ead86f9413ccc2f7edf760f22adee9c7b24d94bd8b1b10f985fa247c310a23f8519f8c19de73c3d775
-  data.tar.gz: e3307314435c7fa559fca55266c1ca9155ed9c7fef9bc0dee16443ba1fe7b056771ec7e49fc461f2504ef0ee443a6fd7c2975b5433a0b308fc694bf80f6c7908
+  metadata.gz: 3a11e8b75566c699dd7f3c5bf2ae97a004bfb4ff0479e671dd7665db09970627666cacca21d72d11016ab4b783256e96799d66ea65f063492531a5a0320c025a
+  data.tar.gz: cc7e0e511676995ce72b092fd9238ba507f78bb371bc2eb6c1cff0d76ac27c3c28672f1ccd66da4360c4ff1294b6732301744bb4db06c97d51ac65580401dd60

data/lib/omniai/chat/file.rb

@@ -9,7 +9,7 @@ module OmniAI
       attr_accessor :io
 
       # @param io [IO, Pathname, String]
-      # @param type [Symbol, String] :image, :video, :audio, "audio/flac", "image/jpeg", "video/mpeg", etc.
+      # @param type [Symbol, String] "audio/flac", "image/jpeg", "video/mpeg", :audio, :image, :video, :document, etc.
       def initialize(io, type)
         super(type)
         @io = io

data/lib/omniai/chat/media.rb

@@ -9,7 +9,7 @@ module OmniAI
       # @return [Symbol, String]
       attr_accessor :type
 
-      # @param type [String] "audio/flac", "image/jpeg", "video/mpeg", etc.
+      # @param type [String] "audio/flac", "image/jpeg", "video/mpeg", "application/pdf", etc.
       def initialize(type)
         super()
         @type = type
@@ -40,12 +40,18 @@ module OmniAI
         @type.match?(%r{^video/})
       end
 
-      # @return [:video, :audio, :image, :text]
+      # @return [Boolean]
+      def document?
+        @type.match?(%r{^application/pdf$})
+      end
+
+      # @return [:text, :audio, :image, :video :document]
       def kind
         if text? then :text
         elsif audio? then :audio
         elsif image? then :image
         elsif video? then :video
+        elsif document? then :document
         else
           raise(TypeError, "unsupported type=#{@type}")
         end

data/lib/omniai/chat/url.rb

@@ -10,7 +10,7 @@ module OmniAI
       class FetchError < HTTPError; end
 
       # @param uri [URI, String] "https://example.com/cat.jpg"
-      # @param type [Symbol, String] "audio/flac", "image/jpeg", "video/mpeg", :audi, :image, :video, etc.
+      # @param type [Symbol, String] "audio/flac", "image/jpeg", "video/mpeg", :audio, :image, :video, :document, etc.
       def initialize(uri, type = nil)
         super(type)
         @uri = uri

data/lib/omniai/chat.rb

@@ -114,7 +114,7 @@ module OmniAI
       !@stream.nil?
     end
 
-    # Override  to provide an context for serializers / deserializes for a provider.
+    # Override  to provide a context (serializers / deserializes) for a provider.
     #
     # @return [Context, nil]
     def context

data/lib/omniai/embed.rb

@@ -53,10 +53,17 @@ module OmniAI
 
   protected
 
+    # Override  to provide a context (serializers / deserializes) for a provider.
+    #
+    # @return [Context, nil]
+    def context
+      nil
+    end
+
     # @param response [HTTP::Response]
     # @return [Response]
     def parse!(response:)
-      Response.new(data: response.parse)
+      Response.new(data: response.parse, context:)
     end
 
     # @return [HTTP::Response]

data/lib/omniai/version.rb

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

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 2.3.0
+  version: 2.4.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 bindir: exe
 cert_chain: []
-date: 2025-04-03 00:00:00.000000000 Z
+date: 2025-04-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser