omniai 1.6.4 → 1.6.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a3c40ccbf59a89846731fb7e3b1a47c4d2ecc944ea2cb45e56b4911bdc301c13
-  data.tar.gz: fdcd923f1ccb0db66630e48d0230c406ac1d72eb8602bee2872e4d02d10ce081
+  metadata.gz: 0bc92ec17e5ca87590eca2181453b6a3e54c2d15bb6dda0c80c604a443e59660
+  data.tar.gz: e4eb4ad22437711fbac2c538edc644861107985662f24a236b317d229802f7b7
 SHA512:
-  metadata.gz: d1cdb9d2b81d286a593c0f2becff6249c4e33e8143b4d158f3d48204dc4ca0fc6e627804722a52db69121bd5b223ae7d407eeac912d4830d7093eddfae089304
-  data.tar.gz: 1b0f10832178feb1304da5d12abe1bd9613df52ffa8eb2edf1678668cdb3425487e90323774ca8e092b95002f83001eb4b560bc0c8536bc133222842b67986ff
+  metadata.gz: 75b7b695353ec0899b786ee057beaedcd8ab21ba15efca6f3a21c4e22730dc73b7676fbdfece11f624a1dce77638ad2d7cc4a6ff054c2180edb8f423b20ca6ee
+  data.tar.gz: 9c7ab97db7e4d8085774c9fd76803b8abfe53905b6d9b6d141caed311d019347d3f0e880e957597b529de8e479d1b69415213e82ca1daba34bf00d7049eddbae

data/lib/omniai/chat/content.rb

@@ -4,6 +4,14 @@ module OmniAI
   class Chat
     # A placeholder for parts of a message. Any subclass must implement the serializable interface.
     class Content
+      # @return [String]
+      def self.summarize(content)
+        return content.map { |entry| summarize(entry) }.join("\n\n") if content.is_a?(Array)
+        return content if content.is_a?(String)
+
+        content.summarize
+      end
+
       # @param context [Context] optional
       #
       # @return [String]

data/lib/omniai/chat/file.rb

@@ -29,10 +29,23 @@ module OmniAI
       # @param context [Context]
       # @return [Hash]
       def serialize(context: nil)
-        serializer = context&.serializers&.[](:file)
-        return serializer.call(self, context:) if serializer
+        if text?
+          content = fetch!
+          Text.new("<file>#{filename}: #{content}</file>").serialize(context:)
+        else
+          serializer = context&.serializers&.[](:file)
+          return serializer.call(self, context:) if serializer
 
-        { type: "#{kind}_url", "#{kind}_url": { url: data_uri } }
+          { type: "#{kind}_url", "#{kind}_url": { url: data_uri } }
+        end
+      end
+
+      # @return [String]
+      def filename
+        case @io
+        when Tempfile, File, String then ::File.basename(@io)
+        else 'DATA'
+        end
       end
     end
   end

data/lib/omniai/chat/media.rb

@@ -15,6 +15,11 @@ module OmniAI
         @type = type
       end
 
+      # @return [String]
+      def summarize
+        "[#{filename}]"
+      end
+
       # @return [Boolean]
       def text?
         @type.match?(%r{^text/})

data/lib/omniai/chat/message.rb

@@ -31,6 +31,14 @@ module OmniAI
         "#<#{self.class} role=#{@role.inspect} content=#{@content.inspect}>"
       end
 
+      # @return [String]
+      def summarize
+        <<~TEXT
+          #{@role}
+          #{Content.summarize(@content)}
+        TEXT
+      end
+
       # Usage:
       #
       #   Message.deserialize({ role: :user, content: 'Hello!' }) # => #<Message ...>

data/lib/omniai/chat/prompt.rb

@@ -62,6 +62,11 @@ module OmniAI
         "#<#{self.class.name} messages=#{@messages.inspect}>"
       end
 
+      # @return [String]
+      def summarize
+        @messages.map(&:summarize).join("\n\n")
+      end
+
       # Usage:
       #
       #   prompt.serialize # => [{ content: "What is the capital of Canada?", role: :user }]

data/lib/omniai/chat/text.rb

@@ -18,6 +18,11 @@ module OmniAI
         "#<#{self.class} text=#{@text.inspect}>"
       end
 
+      # @return [String]
+      def summarize
+        @text
+      end
+
       # @param data [Hash]
       def self.deserialize(data, context: nil)
         deserialize = context&.deserializers&.[](:text)

data/lib/omniai/chat/url.rb

@@ -16,6 +16,11 @@ module OmniAI
         @uri = uri
       end
 
+      # @return [String]
+      def summarize
+        "[#{filename}]"
+      end
+
       # @return [String]
       def inspect
         "#<#{self.class} uri=#{@uri.inspect}>"
@@ -36,13 +41,18 @@ module OmniAI
       #
       # @return [Hash]
       def serialize(context: nil)
-        serializer = context&.serializers&.[](:url)
-        return serializer.call(self, context:) if serializer
+        if text?
+          content = fetch!
+          Text.new("<file>#{filename}: #{content}</file>").serialize(context:)
+        else
+          serializer = context&.serializers&.[](:url)
+          return serializer.call(self, context:) if serializer
 
-        {
-          type: "#{kind}_url",
-          "#{kind}_url": { url: @uri },
-        }
+          {
+            type: "#{kind}_url",
+            "#{kind}_url": { url: @uri },
+          }
+        end
       end
 
       # @raise [FetchError]
@@ -53,6 +63,11 @@ module OmniAI
         String(response.body)
       end
 
+      # @return [String]
+      def filename
+        ::File.basename(@uri)
+      end
+
       protected
 
       # @raise [FetchError]

data/lib/omniai/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.6.4
+  version: 1.6.6
 platform: ruby
 authors:
 - Kevin Sylvestre