omniai 1.6.4 → 1.6.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a3c40ccbf59a89846731fb7e3b1a47c4d2ecc944ea2cb45e56b4911bdc301c13
-  data.tar.gz: fdcd923f1ccb0db66630e48d0230c406ac1d72eb8602bee2872e4d02d10ce081
+  metadata.gz: 5d9011a55d57b9dd0923f4d7edd70001c3f9ee5b8211e98d59f9d001687af044
+  data.tar.gz: 204e2f0ec0a44f65cba11cb49819e7f5c4e933a92d6bc1b48c4d630dee180ecd
 SHA512:
-  metadata.gz: d1cdb9d2b81d286a593c0f2becff6249c4e33e8143b4d158f3d48204dc4ca0fc6e627804722a52db69121bd5b223ae7d407eeac912d4830d7093eddfae089304
-  data.tar.gz: 1b0f10832178feb1304da5d12abe1bd9613df52ffa8eb2edf1678668cdb3425487e90323774ca8e092b95002f83001eb4b560bc0c8536bc133222842b67986ff
+  metadata.gz: 724827923255adade8f19bd9b43806d1af116d9c4dd6105713f0ba0b16c3d6cd5e5cbafdc2d324912a669523e565ba1ce190ed26035cdc1645a4505f2c6991a9
+  data.tar.gz: 5944ef3f285359662fc6105fd2a3aba7a44a279c5062ec0c1bdf194ddecd6842a33b1cd9b20cd4f5b36641b67cd8bc046e677260170d06dd2d5e3b37db37f356

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/url.rb

@@ -36,13 +36,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 +58,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.5'
 end

metadata

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