omniai-openai 1.3.2 → 1.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7cb12222b6c13b368f2f25d55c66ff57e98114830a32bdd54ee82947aaf38674
-  data.tar.gz: 7600584957b977be573321a25a995f720fde9e9112fdd3e1b8676ce3d75461e0
+  metadata.gz: c5f48cc11134ffb023a4d1797a637ee1a0a6ae0365da799df07c484f177a58cb
+  data.tar.gz: 971a2c0524376bd64cf4c1e04b82ab0373b20314d11a06eb0ebc7113f6b9515f
 SHA512:
-  metadata.gz: 75f8bdbfff09e789058e15bf962efcbbe0146b3e522e5efc701ca1303ef734b68f6d19094502c2e8fd89f5db213687b6c5001967e3dfd1b46eaab060b4c4e4c0
-  data.tar.gz: 5b206397432e32277bda4767bad41707699607cf93e9a396570cbe78ab668f42b5c34c75f99d26d93cccd042b9c08359a899a4f375fabae7fc6c55a587515550
+  metadata.gz: c3765f6900534a23d7dc6033f0dc73dc31845494b938e33d345374e4e76ba16e300083cf1c2bedfc82ccc727a29636327092ea86b05f265eae40881de43fe0e8
+  data.tar.gz: b37e93448e1b3314724af029fcdca1f472b6a4e55b027a864079881d5139211b4f8001c0b805479af24ff6f69e586fd919d78bf10843d7de046ca525805e75e3

data/lib/omniai/openai/thread/annotation.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    class Thread
+      # An OpenAI content w/ annotations.
+      class Annotation
+        # @!attribute [rw] data
+        #   @return [Hash, nil]
+        attr_accessor :data
+
+        # @param data [Hash] required
+        # @param client [OmniAI::OpenAI::Client] optional
+        def initialize(data:, client: Client.new)
+          @data = data
+          @client = client
+        end
+
+        # @return [String] "file_citation" or "file_path"
+        def type
+          @data['type']
+        end
+
+        # @return [String]
+        def text
+          @data['text']
+        end
+
+        # @return [Integer]
+        def start_index
+          @data['start_index']
+        end
+
+        # @return [Integer]
+        def end_index
+          @data['end_index']
+        end
+
+        # @return [Range<Integer>]
+        def range
+          start_index..end_index
+        end
+
+        # @return [String]
+        def file_id
+          @file_id ||= (@data['file_citation'] || @data['file_path'])['file_id']
+        end
+
+        # Present if type is "file_citation" or "file_path".
+        #
+        # @return [OmniAI::OpenAI::File, nil]
+        def file!
+          @file ||= @client.files.find(id: file_id)
+        end
+      end
+    end
+  end
+end

data/lib/omniai/openai/thread/attachment.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    class Thread
+      # An OpenAI attachment.
+      class Attachment
+        # @!attribute [rw] data
+        #   @return [Hash, nil]
+        attr_accessor :data
+
+        # @param data [Array]
+        # @param client [OmniAI::OpenAI::Client]
+        #
+        # @return [Array<OmniAI::OpenAI::Thread::Content>, String, nil]
+        def self.for(data:, client: Client.new)
+          return data unless data.is_a?(Enumerable)
+
+          data.map { |attachment| new(data: attachment, client:) }
+        end
+
+        # @param data [Hash]
+        # @param client [OmniAI::OpenAI::Client]
+        def initialize(data:, client: Client.new)
+          @data = data
+          @client = client
+        end
+
+        # @return [String] e.g. "text"
+        def file_id
+          @file_id ||= @data['file_id']
+        end
+
+        # @return [Array<Hash>]
+        def tools
+          @tools ||= @data['tools']
+        end
+
+        # @return [OmniAI::OpenAI::File]
+        def file!
+          @file ||= @client.files.find(id: file_id)
+        end
+      end
+    end
+  end
+end

data/lib/omniai/openai/thread/content.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    class Thread
+      # An OpenAI content w/ annotations.
+      class Content
+        module Type
+          TEXT = 'text'
+        end
+
+        # @param data [Array]
+        # @param client [OmniAI::OpenAI::Client]
+        #
+        # @return [Array<OmniAI::OpenAI::Thread::Content>, String, nil]
+        def self.for(data:, client: Client.new)
+          return data unless data.is_a?(Enumerable)
+
+          data.map { |attachment| new(data: attachment, client:) }
+        end
+
+        # @!attribute [rw] data
+        #   @return [Hash, nil]
+        attr_accessor :data
+
+        # @param data [Hash]
+        def initialize(data:, client:)
+          @data = data
+          @client = client
+        end
+
+        # @return [String] e.g. "text"
+        def type
+          @type ||= @data['type']
+        end
+
+        # @return [Boolean]
+        def text?
+          type.eql?(Type::TEXT)
+        end
+
+        # @return [OmniAI::OpenAI::Thread::Text]
+        def text
+          @text ||= Text.new(data: @data['text'], client: @client) if @data['text']
+        end
+      end
+    end
+  end
+end

data/lib/omniai/openai/thread/message.rb

@@ -168,8 +168,8 @@ module OmniAI
               thread_id: data['thread_id'],
               run_id: data['run_id'],
               role: data['role'],
-              content: data['content'],
-              attachments: data['attachments'],
+              content: Content.for(data: data['content'], client:),
+              attachments: Attachment.for(data: data['attachments'], client:),
               metadata: data['metadata']
             )
           end
@@ -181,9 +181,9 @@ module OmniAI
           @assistant_id = data['assistant_id']
           @thread_id =  data['thread_id']
           @run_id =  data['run_id']
-          @role =  data['role']
-          @content = data['content']
-          @attachments = data['attachments']
+          @role = data['role']
+          @content = Content.for(data: data['content'], client: @client)
+          @attachments = Attachment.for(data: data['content'], client: @client)
           @metadata =  data['metadata']
         end
 

data/lib/omniai/openai/thread/text.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    class Thread
+      # An OpenAI text w/ annotations.
+      class Text
+        # @!attribute [rw] data
+        #   @return [Hash, nil]
+        attr_accessor :data
+
+        # @param data [Hash]
+        # @param client [OmniAI::OpenAI::Client]
+        def initialize(data:, client:)
+          @data = data
+          @client = client
+        end
+
+        # @return [String]
+        def annotate!
+          text = value
+
+          annotations.each do |annotation|
+            file = annotation.file!
+            text = text.gsub(annotation.text, "[#{file.filename}:#{annotation.range}]")
+          end
+
+          text
+        end
+
+        # @return [String] e.g. "text"
+        def type
+          @data['type']
+        end
+
+        # @return [String]
+        def value
+          @data['value']
+        end
+
+        # @return [Array<OmniAI::OpenAI::Thread::Annotation>]
+        def annotations
+          @annotations ||= @data['annotations'].map { |data| Annotation.new(data:, client: @client) }
+        end
+      end
+    end
+  end
+end

data/lib/omniai/openai/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module OpenAI
-    VERSION = '1.3.2'
+    VERSION = '1.3.3'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-openai
 version: !ruby/object:Gem::Version
-  version: 1.3.2
+  version: 1.3.3
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-24 00:00:00.000000000 Z
+date: 2024-06-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -71,10 +71,14 @@ files:
 - lib/omniai/openai/files.rb
 - lib/omniai/openai/speak.rb
 - lib/omniai/openai/thread.rb
+- lib/omniai/openai/thread/annotation.rb
+- lib/omniai/openai/thread/attachment.rb
+- lib/omniai/openai/thread/content.rb
 - lib/omniai/openai/thread/message.rb
 - lib/omniai/openai/thread/messages.rb
 - lib/omniai/openai/thread/run.rb
 - lib/omniai/openai/thread/runs.rb
+- lib/omniai/openai/thread/text.rb
 - lib/omniai/openai/threads.rb
 - lib/omniai/openai/tool.rb
 - lib/omniai/openai/transcribe.rb