omniai 1.2.2 → 1.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 239eb058ae53afd3f28fe00a18af282a856ca842a4f81bcdf22413a02d8d44fc
-  data.tar.gz: e74ac69ab6435fd9da3d1602c9cce5e052f53bebf609d8815868b58be0897884
+  metadata.gz: 2820bc43050fa4484a25e1a967b3d12ac71a3d5659e735f5904040990acf47b8
+  data.tar.gz: 8894714bf0c8055ca173176d8eeb078ce6f93a0481c19df3fd2dae911fbb264d
 SHA512:
-  metadata.gz: a6b18cae8b5aab610bf255860ccd514b794eda3021b2f9b3aa86020968e0f449b683b4db74291e71583698b61d7612a7e49b3ee0310c806a826f7b27c50c7d92
-  data.tar.gz: bd12b38f3c9a150fe09644aaf187aa05aa1063ee136196e11fcaa356f4fe5bdc2c93f6292a79025f301c6d35a1202186f6ad787787828dfd5af91f229a34b550
+  metadata.gz: b2658cb0bf197315f3679225a214871ffdbf670cabb65f3126b4a1e7ccd40b6f38810e9fea828b71eb1c9fae819f94e824dedc65ed69cc7531e496d9ae027896
+  data.tar.gz: 3f525900c8c319076549235713418e7cc7e87167151668900e3073530004b9bb1561469a9b02914dfac63eff4f35b67fbf9ca66171b5471cf24c665e280f3d23

data/lib/omniai/chat/chunk.rb

@@ -31,9 +31,9 @@ module OmniAI
         @data['model']
       end
 
-      # @return [Array<OmniAI::Chat::Choice>]
+      # @return [Array<OmniAI::Chat::DeltaChoice>]
       def choices
-        @choices ||= @data['choices'].map { |data| Choice.for(data:) }
+        @choices ||= @data['choices'].map { |data| DeltaChoice.for(data:) }
       end
 
       # @param index [Integer]

data/lib/omniai/chat/completion.rb

@@ -38,13 +38,13 @@ module OmniAI
         @usage ||= Usage.for(data: @data['usage'])
       end
 
-      # @return [Array<OmniAI::Chat::Choice>]
+      # @return [Array<OmniAI::Chat::MessageChoice>]
       def choices
-        @choices ||= @data['choices'].map { |data| Choice.for(data:) }
+        @choices ||= @data['choices'].map { |data| MessageChoice.for(data:) }
       end
 
       # @param index [Integer] optional - default is 0
-      # @return [OmniAI::Chat::Choice]
+      # @return [OmniAI::Chat::MessageChoice]
       def choice(index: 0)
         choices[index]
       end

data/lib/omniai/chat/delta_choice.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    # A delta choice returned by the API.
+    class DeltaChoice
+      attr_accessor :index, :delta
+
+      # @param data [Hash]
+      # @return [OmniAI::Chat::Choice]
+      def self.for(data:)
+        index = data['index']
+        delta = Delta.for(data: data['delta'])
+
+        new(index:, delta:)
+      end
+
+      # @param index [Integer]
+      # @param delta [Delta]
+      def initialize(index:, delta:)
+        @index = index
+        @delta = delta
+      end
+    end
+  end
+end

data/lib/omniai/chat/message_choice.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    # A choice returned by the API.
+    class MessageChoice
+      attr_accessor :index, :message
+
+      # @param data [Hash]
+      # @return [OmniAI::Chat::Choice]
+      def self.for(data:)
+        index = data['index']
+        message = Message.for(data: data['message'])
+
+        new(index:, message:)
+      end
+
+      # @param index [Integer]
+      # @param message [OmniAI::Chat::Message]
+      def initialize(index:, message:)
+        @index = index
+        @message = message
+      end
+    end
+  end
+end

data/lib/omniai/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.2.2
+  version: 1.2.3
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-20 00:00:00.000000000 Z
+date: 2024-06-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -66,7 +66,6 @@ files:
 - bin/setup
 - lib/omniai.rb
 - lib/omniai/chat.rb
-- lib/omniai/chat/choice.rb
 - lib/omniai/chat/chunk.rb
 - lib/omniai/chat/completion.rb
 - lib/omniai/chat/content/file.rb
@@ -74,7 +73,9 @@ files:
 - lib/omniai/chat/content/text.rb
 - lib/omniai/chat/content/url.rb
 - lib/omniai/chat/delta.rb
+- lib/omniai/chat/delta_choice.rb
 - lib/omniai/chat/message.rb
+- lib/omniai/chat/message_choice.rb
 - lib/omniai/chat/stream.rb
 - lib/omniai/chat/usage.rb
 - lib/omniai/client.rb

data/lib/omniai/chat/choice.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    # A choice returned by the API.
-    class Choice
-      attr_accessor :index, :delta, :message
-
-      # @param data [Hash]
-      # @return [OmniAI::Chat::Choice]
-      def self.for(data:)
-        index = data['index']
-        delta = Delta.for(data: data['delta']) if data['delta']
-        message = Message.for(data: data['message']) if data['message']
-
-        new(index:, delta:, message:)
-      end
-
-      # @param index [Integer]
-      # @param delta [OmniAI::Chat::Delta] optional
-      # @param message [OmniAI::Chat::Message] optional
-      def initialize(index:, delta:, message:)
-        @index = index
-        @delta = delta
-        @message = message
-      end
-    end
-  end
-end