open_ai_bot 0.3.2 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e31ed7bc34cf94ba2d339a1efa6b57a0d9fd972aec39961a98253cb5b9dcc4df
-  data.tar.gz: 2f92b54b9788efacd9bf50b66004c777a8eeb3bd934ffc65a2ea93a4a12c39e6
+  metadata.gz: 2aafb706ad49054fb0e35aab308ea6486ab89ea7e6740f081033b49a972b4dfb
+  data.tar.gz: 2b6b9c143e1465bc2e1f984d8ec611b234644b6f99bf3a033ca160a558871625
 SHA512:
-  metadata.gz: 38ff4646a9493b2771d62cce25d8a2ff0ab589bc2d300041fec4536e6e75065e33c79fe452994dd5d9a628be6646e4ed97bb97a7c28503811b3a3db1b8fefa9f
-  data.tar.gz: 872adfdeb95db9ccbdd8073ee736136a40d4517b1b4eddee9ddf5885638dfe286c705554dd708c654dd1e9b83c62682626fd63cf3d2e878fea536a4d79c08d67
+  metadata.gz: f486692224acc9d6897a426b37e56673779d2631bd2577e9a26ed0ff99715a1e8b14972649cfbbaab02db47233a878bd85f00b1a4cb0fb966a169d3a651a56f9
+  data.tar.gz: 6d0c11a1f76f4edf83ec4a011ded1a88bfc73d48ff7a7567c57525d519ec568fc458f5c262787422ac5f35fad97fe91e2f4d0b5c6564a7d5058ec08b3fd3458b

data/lib/open_ai/chat_gpt.rb

@@ -145,9 +145,8 @@ module OpenAI
     def get_tokens_info!(response)
       completion_tokens = response.dig("usage", "completion_tokens")
       prompt_tokens = response.dig("usage", "prompt_tokens")
-      vision_tokens = current_thread.claim_vision_tokens!
 
-      result = current_thread.model.request_cost(completion_tokens:, prompt_tokens:, vision_tokens:, current_thread:)
+      result = current_thread.model.request_cost(completion_tokens:, prompt_tokens:, current_thread:)
     end
 
     def send_chat_gpt_response(text, tokens_info)

data/lib/open_ai/chat_thread.rb

@@ -24,10 +24,6 @@ module OpenAI
       @history.map(&:cost).compact.sum
     end
 
-    def claim_vision_tokens!
-      @history.reject(&:vision_tokens_claimed?).map(&:claim_vision_tokens!).compact.sum
-    end
-
     def add(message)
       return false unless message&.valid?
       return false if @history.any? { message.id == _1.id}

data/lib/open_ai/message.rb

@@ -10,17 +10,6 @@ module OpenAI
       kwargs.each_pair { public_send("#{_1}=", _2) }
       @role = :user
       @timestamp = Time.now.to_i
-      @vision_tokens_claimed = !image
-    end
-
-    def vision_tokens_claimed?
-      @vision_tokens_claimed
-    end
-
-    def claim_vision_tokens!
-      # binding.pry
-      @vision_tokens_claimed = true
-      image&.tokens
     end
 
     def valid?

data/lib/open_ai/model.rb

@@ -6,19 +6,18 @@ module OpenAI
         max_context: 128_000,
         prompt_price: 0.005,
         completion_price: 0.015,
-        vision_price: 0.005
+        vision: true
       },
       "gpt-3.5-turbo": {
         max_context: 16385,
         prompt_price: 0.0005,
-        completion_price: 0.0015,
-        vision_price: 0
+        completion_price: 0.0015
       }
     }
 
-    attr_accessor :max_context, :prompt_price, :completion_price, :vision_price
+    attr_accessor :max_context, :prompt_price, :completion_price
 
-    [:max_context, :prompt_price, :completion_price, :vision_price].each do |attr|
+    [:max_context, :prompt_price, :completion_price].each do |attr|
       define_method(attr) do
         MODEL_INFO[@model][attr]
       end
@@ -37,21 +36,19 @@ module OpenAI
     end
 
     def has_vision?
-      MODEL_INFO[@model][:vision_price].positive?
+      MODEL_INFO[@model][:vision]
     end
 
-    def request_cost(prompt_tokens:, completion_tokens:, vision_tokens:, current_thread:)
+    def request_cost(prompt_tokens:, completion_tokens:, current_thread:)
       prompt_cost     = prompt_tokens * prompt_price / 1000
       completion_cost = completion_tokens * completion_price / 1000
-      vision_cost     = vision_tokens * vision_price / 1000
 
-      total = prompt_cost + completion_cost + vision_cost
+      total = prompt_cost + completion_cost
       thread_total = current_thread.total_cost
 
       info = "\n\n" + {
         prompt: "#{prompt_tokens} tokens (#{prompt_cost.round(5)}$)",
         completion: "#{completion_tokens} tokens (#{completion_cost.round(5)}$)",
-        vision: "#{vision_tokens} tokens (#{vision_cost.round(5)}$)",
         total: "#{total.round(5)}$",
         total_for_this_conversation: "#{(thread_total + total).round(5)}$",
         max_context: max_context

data/open_ai_bot.gemspec

@@ -8,7 +8,7 @@ require_relative "lib/ext/in"
 
 Gem::Specification.new do |spec|
   spec.name          = "open_ai_bot"
-  spec.version       = "0.3.2"
+  spec.version       = "0.3.3"
   spec.authors       = ["bulgakke"]
   spec.email         = ["vvp835@yandex.ru"]
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: open_ai_bot
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.3.3
 platform: ruby
 authors:
 - bulgakke