open_ai_bot 0.3.4 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 70a3fabb10c860e1d7c45e83c74b38f7b393d4a1d31c86a73ccc4f827af6dcd8
-  data.tar.gz: 91436b52ead57fb638b4de1d74f573041fd541bae1352315632ae4414853fb28
+  metadata.gz: b2e5ee8b986bfaff8105b3ae2415e3e5dc0401ad836f4a7b3a3ae4f402c72f07
+  data.tar.gz: 781bef28b9a896f40e7c07933bc9ba180875fc242b992749cc6c5eff82c2cc85
 SHA512:
-  metadata.gz: '0908d995df133104dd4a591e7a9cff2c561f39753fc0570ac15eb3d9808ab644f7c04afef5682580c020bfca9d2e05e70573a4258427e9737bb18c2ef120ba60'
-  data.tar.gz: 6b54406776b00460dea2f1451313558b60bdb8fddb5dd843ec6ed64e6052a42affeae69bfe166be25ad7d641dd9d71980b25adaf0bb620462ce4b0e73d63d806
+  metadata.gz: c1f0c896c0d569751ceb554502918ecbd6521db9d734b3725a152686c343137b4b3f28c8062da9655eaa2a80fdb85b7320bc8bfea2573d90f5e0a1c522bb49c5
+  data.tar.gz: 5bf24970ad9796510f9de792978529d4413a6d39bd83277c1ab82da296cbb8cc61a2e534194182f3697350e1e8f76b70a5bab1c0a3c855c51ebe204eb5354e7c

data/lib/open_ai/chat_thread.rb

@@ -5,12 +5,11 @@ module OpenAI
     def initialize(defaults = [], model)
       @history ||= defaults
       @model = model.is_a?(Model) ? model : Model.new(model)
-      puts "Using #{@model}"
       puts @history
     end
 
     attr_reader :history
-    attr_reader :model
+    attr_accessor :model
 
     alias_method :messages, :history
 

data/lib/open_ai/model.rb

@@ -65,9 +65,14 @@ module OpenAI
       end
 
       @model = model
+      puts "Using #{@model}"
     end
 
     def to_s
+      @model.to_s
+    end
+
+    def to_sym
       @model
     end
 

data/lib/open_ai_bot.rb

@@ -19,6 +19,7 @@ class OpenAIBot < Rubydium::Bot
   include OpenAI::Whisper
 
   on_every_message :handle_gpt_command
+  on_every_message :handle_model_query
   on_every_message :transcribe
 
   on_command "/restart", :init_session, description: "Resets ChatGPT session"
@@ -36,6 +37,39 @@ class OpenAIBot < Rubydium::Bot
     safe_delete(@msg)
   end
 
+  on_command "/model" do
+    options = []
+    OpenAI::Model::MODEL_INFO.each do |model, info|
+      options << [
+        Telegram::Bot::Types::InlineKeyboardButton.new(
+          text: "#{model} - #{sprintf('%.2f', info[:output_price])}$",
+          callback_data: "/set #{model}"
+        )
+      ]
+    end
+    markup = Telegram::Bot::Types::InlineKeyboardMarkup.new(inline_keyboard: options)
+    reply("Select a model:", reply_markup: markup)
+  end
+
+  def handle_model_query
+    return unless @update.is_a? Telegram::Bot::Types::CallbackQuery
+    return unless @update.data.start_with? "/set "
+    return unless @user.username == config.owner_username
+
+    model = @update.data.delete_prefix("/set ").to_sym
+    return if OpenAI::Model::MODEL_INFO[model].nil?
+
+    text =
+      if current_thread.model.to_sym == model
+        "Already set to `#{model}`"
+      else
+        current_thread.model = OpenAI::Model.new(model)
+        "Was `#{current_thread.model.to_s}`, now `#{model}`"
+      end
+
+    reply(text, parse_mode: "Markdown")
+  end
+
   def allowed_chat?
     return true if @user.username == config.owner_username
     return true if config.open_ai["whitelist"].include?(@chat.id)

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.4"
+  spec.version       = "0.3.5"
   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.4
+  version: 0.3.5
 platform: ruby
 authors:
 - bulgakke