omniai 1.4.0 → 1.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8b3d66712c9a2fd460abf00c29b056b87ff8cdd08c8b38d69ece0755d8c76578
-  data.tar.gz: f0fdf7b20e3f3cc3c4e2bee6d21b68bd917d9267c9f1297dec91f2f8bccc71e9
+  metadata.gz: df4f1096ec0d74c8ab26e5cfc8db4902009d1bb398ed9913e801b31203d1ae66
+  data.tar.gz: 9cff5f3ecdc6a540bbbd3ca6501dbd07585a1f9dfe8127a09ee6421a51f79c2f
 SHA512:
-  metadata.gz: 98c6c3380181aaf7ad858968b998adab49f8cd200d94755ff6cf79423378dd401c4c538147aeb416b023c47fe7f1806a8742a9486429e90bcc487eb8944cba52
-  data.tar.gz: f251bb7436e9536275a285508f63964693a067378f55a72b3cf01fb9cb801f55ef96fd8a64c9b4e5790c6c400ee6adcee64f6dc5dc413e92b42d5f869afce45f
+  metadata.gz: db82f7cf3e8b43f55080c1ae1af2f6f9767dec7870bf23e2424eee8a272671b565f279829d9768fbb447eec8821fcd7da22058da8a4016b4de271a389a115c1c
+  data.tar.gz: 84362d233dbafa2dd6403b3d65056c7b836eb0b2ce420f4395c86188887f12ba7cb59caead7cfbc44d80bfa03ad2ef342b62f8235ace58afd0284e8e25b5c243

data/lib/omniai/chat.rb

@@ -89,7 +89,7 @@ module OmniAI
 
     # @param response [OmniAI::Chat::Completion]
     def complete!(response:)
-      Completion.new(data: response.parse)
+      self.class::Completion.new(data: response.parse)
     end
 
     # @param response [HTTP::Response]
@@ -97,21 +97,23 @@ module OmniAI
     def stream!(response:)
       raise Error, "#{self.class.name}#stream! unstreamable" unless @stream
 
-      Stream.new(response:).stream! do |chunk|
+      self.class::Stream.new(response:).stream! do |chunk|
         case @stream
-        when IO then @stream << chunk
+        when IO
+          @stream << chunk.choice.delta.content
+          @stream.flush
         else @stream.call(chunk)
         end
       end
 
-      @stream.flush if @stream.is_a?(IO)
+      @stream.puts if @stream.is_a?(IO)
     end
 
     # @return [Array<Hash>]
     def messages
       arrayify(@messages).map do |content|
         case content
-        when String then { role: OmniAI::Chat::Role::USER, content: }
+        when String then { role: Role::USER, content: }
         when Hash then content
         else raise Error, "Unsupported content=#{content.inspect}"
         end

data/lib/omniai/cli/chat_handler.rb

@@ -21,6 +21,7 @@ module OmniAI
 
         loop do
           @stdout.print('# ')
+          @stdout.flush
           prompt = @stdin.gets&.chomp
 
           break if prompt.nil? || prompt.match?(/\A(exit|quit)\z/i)

data/lib/omniai/version.rb

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

metadata

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