omniai 1.0.5 → 1.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dddfc4d4ac08e936e1b2443e37ff93b4256428579aa1280b6ede769a01365f04
-  data.tar.gz: 235c5f109e0e3526d2abfa2cf7d6c3a3d1e00749da70beaf884a2a1bcb9ba8c6
+  metadata.gz: 3be83f1587fcdde2ec982dc7220e5c0abd9d9c756b3879b50cb8be2558dad0f7
+  data.tar.gz: 6b031c1deaf0d9ef144fdba86a25650671b4492e960fcb633f1558a72922d61e
 SHA512:
-  metadata.gz: fae66992c4f0a399ecbe795a39901e73caea1554861e23894fd005daab9b26e23cd6a110cca9cde9b9c9ce73af174037626b3505074bb4bd23477677a286b3ef
-  data.tar.gz: 5f604bcdc4ca4b748e523eff6cb4b383cd9211188673c768d2295a7f9e43c1f043ac9c8b14ff86b9dabb85a99b50a005ce40b920cdbcdf19f9f1ef05b4f33ca5
+  metadata.gz: a07f2c8b25842f73c34fb8ca80bbb3efb1554da4bae6ffaca140167ee4581a198b175252d687dad1ac9dd34c05d4dbd63bfce951ecc5c907119c175d7dd92597
+  data.tar.gz: ea95daacea2365aeb68253d1c7168cfbae8cb26323a2e60ffc8212b43ad0c24497f8e8679188dec6a838dc21daf10387e363ad7a55a21def55c22a48968763f0

data/lib/omniai/chat/choice.rb

@@ -4,28 +4,23 @@ module OmniAI
   class Chat
     # A choice returned by the API.
     class Choice
-      attr_accessor :data
+      attr_accessor :index, :delta, :message
 
       # @param data [Hash]
-      def initialize(data:)
-        @data = data
-      end
-
-      # @return [Integer]
-      def index
-        @data['index']
-      end
+      # @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']
 
-      # @return [OmniAI::Chat::Delta]
-      def delta
-        Delta.new(data: @data['delta']) if @data['delta']
+        new(index:, delta:, message:)
       end
 
-      # @return [OmniAI::Chat::Message]
-      def message
-        return unless @data['message']
-
-        @message ||= Message.for(role: @data['message'])
+      # @param data [Hash]
+      def initialize(index:, delta:, message:)
+        @index = index
+        @delta = delta
+        @message = message
       end
     end
   end

data/lib/omniai/chat/completion.rb

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

data/lib/omniai/chat/delta.rb

@@ -4,22 +4,27 @@ module OmniAI
   class Chat
     # A delta returned by the API.
     class Delta
-      attr_accessor :data
+      attr_accessor :role, :content
 
-      # @param content [Integer]
-      # @param role [String]
-      def initialize(data:)
-        @data = data
+      # @param data [Hash]
+      # @return [OmniAI::Chat::Message]
+      def self.for(data:)
+        content = data['content'] || data[:content]
+        role = data['role'] || data[:role]
+
+        new(content:, role: role || Role::USER)
       end
 
-      # @return [String, nil]
-      def role
-        @data['role']
+      # @param content [String]
+      # @param role [String]
+      def initialize(content:, role: nil)
+        @content = content
+        @role = role
       end
 
-      # @return [String, nil]
-      def content
-        @data['content']
+      # @return [String]
+      def inspect
+        "#<#{self.class.name} role=#{role.inspect} content=#{content.inspect}>"
       end
     end
   end

data/lib/omniai/chat.rb

@@ -55,7 +55,7 @@ module OmniAI
     # @raise [ExecutionError]
     def process!
       response = request!
-      raise HTTPError, response unless response.status.ok?
+      raise HTTPError, response.flush unless response.status.ok?
 
       parse!(response:)
     end

data/lib/omniai/version.rb

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

metadata

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