omniai 1.0.4 → 1.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6d821530b3b40043754c26a175c2b5d9c70fe3f04bd62bf61c5506963e56dae3
-  data.tar.gz: 6f0cf3e3292b11492bd25526391ec3a396d5a9262050dcf4d5aba35726c2ed04
+  metadata.gz: 172ad18b8fbe97f2e8d23bf8624efea92591910f6d05eed5295789555f60e6d7
+  data.tar.gz: dc415cd3f63f293bc2d64cff39c3ee3c094f734d7a0fb330a1b25adf772f7016
 SHA512:
-  metadata.gz: 9f2d2fa897c94b741435df368ca0952fe00e4dafd0f199756008749a1267ac2d4ea325338b1da359c7c84944b76ec3a875d678039b9bd88ee819cacb470a6fdd
-  data.tar.gz: 698f4ec0159ac6a359f5bd93c02db3e4acd0cde6cfd42c17a30c341d6ad674720aa442040c0890cd3c600068c52717b011b28601335721fbdd7a55c1d74ae61d
+  metadata.gz: 1dc885877d697bf3c2053e8facbdc1a7329a8b3ae0804753e0e1daefcc9f37bcc7dd19de11c748fccdf4cc9be032cbbb1128558a8d437a125ab41ca44402d7ce
+  data.tar.gz: cb96cde1ff5203f08646217bfae1c0032a25f492cbb6b1565f34ea9a5a905c0626b6bfab5e651793cd853b34eaf0a95130ad8a41e2f8a77429c9faf0feb58cc5

data/lib/omniai/chat/completion.rb

@@ -35,15 +35,7 @@ module OmniAI
       def usage
         return unless @data['usage']
 
-        input_tokens = @data['usage']['input_tokens'] || @data['usage']['prompt_tokens']
-        output_tokens = @data['usage']['output_tokens'] || @data['usage']['completion_tokens']
-        total_tokens = @data['usage']['total_tokens'] || (input_tokens + output_tokens)
-
-        @usage ||= Usage.new(
-          input_tokens:,
-          output_tokens:,
-          total_tokens:
-        )
+        @usage ||= Usage.for(data: @data['usage'])
       end
 
       # @return [Array<OmniAI::Chat::Choice>]

data/lib/omniai/chat/usage.rb

@@ -6,6 +6,20 @@ module OmniAI
     class Usage
       attr_accessor :input_tokens, :output_tokens, :total_tokens
 
+      # @param data [Hash]
+      # @return [OmniAI::Chat::Usage]
+      def self.for(data:)
+        input_tokens = data['input_tokens'] || data['prompt_tokens']
+        output_tokens = data['output_tokens'] || data['completion_tokens']
+        total_tokens = data['total_tokens'] || (input_tokens + output_tokens)
+
+        new(
+          input_tokens:,
+          output_tokens:,
+          total_tokens:
+        )
+      end
+
       # @param input_tokens [Integer]
       # @param output_tokens [Integer]
       # @param total_tokens [Integer]

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.4'
+  VERSION = '1.0.6'
 end

metadata

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