omniai 1.6.3 → 1.6.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fe2124a214845732f67f1944d0d9538927a7fe034948f67485fb923d232ce3e7
-  data.tar.gz: 41849074c457cde063e0a14866c43370046b9a33c3e9354655ea5167201f5781
+  metadata.gz: 5d9011a55d57b9dd0923f4d7edd70001c3f9ee5b8211e98d59f9d001687af044
+  data.tar.gz: 204e2f0ec0a44f65cba11cb49819e7f5c4e933a92d6bc1b48c4d630dee180ecd
 SHA512:
-  metadata.gz: c11a71d77fd58667c3fb618284e0b5706c0e12d28ae6f09119123cc3d90f5410338a1fda3448c6b213d38af140d46a4baa3e509d55eaed6c6d338a6680715662
-  data.tar.gz: 56b16d80a3ec823d2d4a43e93553d904849a65e24e8aefebbf69b0cce5a2d8ba0427e7a039c4373c4e59875994f290559a013bef1ae634f6c706d8b5af26a0dd
+  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,13 +58,18 @@ module OmniAI
         String(response.body)
       end
 
+      # @return [String]
+      def filename
+        ::File.basename(@uri)
+      end
+
       protected
 
       # @raise [FetchError]
       #
       # @return [HTTP::Response]
       def request!
-        response = HTTP.get(@uri)
+        response = HTTP.follow.get(@uri)
         raise FetchError, response.flush unless response.status.success?
 
         response

data/lib/omniai/version.rb

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

metadata

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