omniai 1.6.5 → 1.6.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5d9011a55d57b9dd0923f4d7edd70001c3f9ee5b8211e98d59f9d001687af044
-  data.tar.gz: 204e2f0ec0a44f65cba11cb49819e7f5c4e933a92d6bc1b48c4d630dee180ecd
+  metadata.gz: 0bc92ec17e5ca87590eca2181453b6a3e54c2d15bb6dda0c80c604a443e59660
+  data.tar.gz: e4eb4ad22437711fbac2c538edc644861107985662f24a236b317d229802f7b7
 SHA512:
-  metadata.gz: 724827923255adade8f19bd9b43806d1af116d9c4dd6105713f0ba0b16c3d6cd5e5cbafdc2d324912a669523e565ba1ce190ed26035cdc1645a4505f2c6991a9
-  data.tar.gz: 5944ef3f285359662fc6105fd2a3aba7a44a279c5062ec0c1bdf194ddecd6842a33b1cd9b20cd4f5b36641b67cd8bc046e677260170d06dd2d5e3b37db37f356
+  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/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}>"

data/lib/omniai/version.rb

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

metadata

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