omniai 1.2.0 → 1.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d6114f0ef4a95633c3391eb421c23e8df99413c876b838053ddb89bde4e7d6f0
-  data.tar.gz: fe2156f9db8802f44975cc1aab0a58f34cb568e1b34fe14e33bd34c52f4b6ff9
+  metadata.gz: 5b37e667b2b004a012bf7d7d76fd80c08aa8948d46b8670651daa3a426c591d2
+  data.tar.gz: ded01ccbbf34486159fd965953a9d5cec9a62da16404960e5b0c8bca8d5384ad
 SHA512:
-  metadata.gz: 2e3a7b49bd65210142978e79ccb9ed88cd19b085c48334b16939c1401217913b951d474515d2ac3fedd1984cc1c7e31e9fa2af206ee04da3765eb1cfb8e2b393
-  data.tar.gz: 7d966a598e0ec358785e25be08530f7f82898601030f703c78a728e31e7885841239d50c1c851ef3912060318df588050ddf7f35870f5f97254e9a23f412b045
+  metadata.gz: c6c26067f0b0b3c7ecefbdc00ed1746f0f597f729a41ff1e0a253bb723cf89ef1b82ccc2c411a8b43b93d5f54c83a9f3e58ab7074be324e9b3bf24e099e7ca2a
+  data.tar.gz: fe75154f85d6a68b1698c923548570695c724293462225bdb9d29db8ccf884ab2965bce3090a9f9c64851c5547ab4a655e59e7079c0923bff9ee18f46320517c

data/lib/omniai/chat/content/file.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    module Content
+      # A file that is either audio / image / video.
+      class File < Media
+        attr_accessor :io
+
+        # @param io [IO, Pathname, String]
+        # @param type [Symbol, String] :image, :video, :audio, "audio/flac", "image/jpeg", "video/mpeg", etc.
+        def initialize(io, type)
+          super(type)
+          @io = io
+        end
+
+        # @return [String]
+        def fetch!
+          case @io
+          when IO then @io.read
+          else ::File.binread(@io)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/omniai/chat/content/media.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    module Content
+      # An abstract class that represents audio / image / video and is used for both files and urls.
+      class Media
+        attr_accessor :type
+
+        # @param type [String] "audio/flac", "image/jpeg", "video/mpeg", etc.
+        def initialize(type)
+          @type = type
+        end
+
+        # @return [Boolean]
+        def text?
+          @type.match?(%r{^text/})
+        end
+
+        # @return [Boolean]
+        def audio?
+          @type.match?(%r{^audio/})
+        end
+
+        # @return [Boolean]
+        def image?
+          @type.match?(%r{^image/})
+        end
+
+        # @return [Boolean]
+        def video?
+          @type.match?(%r{^video/})
+        end
+
+        # @yield [io]
+        def fetch!(&)
+          raise NotImplementedError, "#{self.class}#fetch! undefined"
+        end
+
+        # e.g. "Hello" -> "SGVsbG8h"
+        #
+        # @return [String]
+        def data
+          Base64.strict_encode64(fetch!)
+        end
+
+        # e.g. "data:text/html;base64,..."
+        #
+        # @return [String]
+        def data_uri
+          "data:#{@type};base64,#{data}"
+        end
+      end
+    end
+  end
+end

data/lib/omniai/chat/content/text.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    module Content
+      # Just some text.
+      class Text
+        attr_accessor :text
+
+        # @param text [text]
+        def initialize(text)
+          @text = text
+        end
+      end
+    end
+  end
+end

data/lib/omniai/chat/content/url.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    module Content
+      # A url that is either audio / image / video.
+      class URL < Media
+        attr_accessor :url, :type
+
+        class HTTPError < OmniAI::HTTPError; end
+
+        # @param url [URI, String]
+        # @param type [Symbol, String] "audio/flac", "image/jpeg", "video/mpeg", etc.
+        def initialize(url, type)
+          super(type)
+          @url = url
+        end
+
+        # @raise [HTTPError]
+        #
+        # @return [String]
+        def fetch!
+          response = request!
+          String(response.body)
+        end
+
+        private
+
+        # @raise [HTTPError]
+        #
+        # @return [HTTP::Response]
+        def request!
+          response = HTTP.get(@url)
+          raise HTTPError, response.flush unless response.status.success?
+
+          response
+        end
+      end
+    end
+  end
+end

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = '1.2.0'
+  VERSION = '1.2.1'
 end

data/lib/omniai.rb

@@ -2,10 +2,12 @@
 
 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.setup
 
 module OmniAI

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.2.0
+  version: 1.2.1
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -69,7 +69,10 @@ files:
 - lib/omniai/chat/choice.rb
 - lib/omniai/chat/chunk.rb
 - lib/omniai/chat/completion.rb
-- lib/omniai/chat/content.rb
+- lib/omniai/chat/content/file.rb
+- lib/omniai/chat/content/media.rb
+- lib/omniai/chat/content/text.rb
+- lib/omniai/chat/content/url.rb
 - lib/omniai/chat/delta.rb
 - lib/omniai/chat/message.rb
 - lib/omniai/chat/stream.rb

data/lib/omniai/chat/content.rb

@@ -1,17 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    # A file used for analysis.
-    class Content
-      attr_accessor :type, :value
-
-      # @param value [String]
-      # @param type [Symbol] :image / :video / :audio / :text
-      def initialize(value, type: :text)
-        @value = value
-        @type = type
-      end
-    end
-  end
-end