Move Chunk under Response namespace

lib/omniai/chat/response/chunk.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    module Response
+      # A chunk returned by the API.
+      class Chunk < Resource
+        # @return [String]
+        def inspect
+          "#<#{self.class.name} id=#{id.inspect} model=#{model.inspect} choices=#{choices.inspect}>"
+        end
+
+        # @return [String]
+        def id
+          @data['id']
+        end
+
+        # @return [Time]
+        def created
+          Time.at(@data['created']) if @data['created']
+        end
+
+        # @return [Time]
+        def updated
+          Time.at(@data['updated']) if @data['updated']
+        end
+
+        # @return [String]
+        def model
+          @data['model']
+        end
+
+        # @return [Array<OmniAI::Chat::Response::DeltaChoice>]
+        def choices
+          @choices ||= @data['choices'].map { |data| DeltaChoice.new(data:) }
+        end
+
+        # @param index [Integer]
+        # @return [OmniAI::Chat::Response::DeltaChoice]
+        def choice(index: 0)
+          choices[index]
+        end
+      end
+    end
+  end
+end

lib/omniai/chat/stream.rb

@@ -16,7 +16,7 @@ def stream!
           @parser.feed(chunk) do |_, data|
             next if data.eql?('[DONE]')
 
-            yield(Chunk.new(data: JSON.parse(data)))
+            yield(Response::Chunk.new(data: JSON.parse(data)))
           end
         end
       end

spec/omniai/chat/chunk_spec.rb → spec/omniai/chat/response/chunk_spec.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-RSpec.describe OmniAI::Chat::Chunk do
+RSpec.describe OmniAI::Chat::Response::Chunk do
   subject(:chunk) { described_class.new(data:) }
 
   let(:data) do
@@ -34,6 +34,6 @@
   end
 
   describe '#inspect' do
-    it { expect(chunk.inspect).to eq('#<OmniAI::Chat::Chunk id="fake_id" model="fake_model" choices=[]>') }
+    it { expect(chunk.inspect).to eq('#<OmniAI::Chat::Response::Chunk id="fake_id" model="fake_model" choices=[]>') }
   end
 end