open_ai_bot 0.2.6 → 0.2.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bb11aa4fa04faff58bebc78dd9f5b7205bd7d096e0be1ea1260f44d62eaaf279
-  data.tar.gz: 79df08164999f67aa37cd1dff5d73d090d9665a38292923c43d9096805275e6c
+  metadata.gz: 94799cab343433c6d8ab68af8ab21136281696a228c1c39802e07b37ee76b32a
+  data.tar.gz: f3829f42e1407d317b24d3b5af80b75dd7b0d659dec0a350ae36de7fab019717
 SHA512:
-  metadata.gz: 4bb4fec967f3d6a1ed50040cac94db80f7fff03740b29ffc07038f1f4de184f95754ce4bc6e159395ee825c71fadeb2a17605b44addbf1b3ba66caa9014e744d
-  data.tar.gz: 8162ea6493434c0770e4e3aafa91e017fb58a92b4eaaa93fd6feb30f269f18bc223ef80029f44db572d356743f8af9a317956781d845b18b7bd556d7ebf9e9a1
+  metadata.gz: 7684048deff97e7bacacad34c1feb5cb2f1c1e14619b92ffc36d570a64af74847a05aa2e0b19210a2f9f94907f10c1c96de617cdec1d25b46dd9aeecd6b9f65d
+  data.tar.gz: e89357f8be7e0fe46d4e1ec314f429dee3e3f0d92cfbb35c3de03aa768b5e37307f5d0769114a2a10fe3da133f7f01f0c94d6cc8bbf42beb40e898ce35b5f760

data/lib/open_ai/chat_gpt.rb

@@ -7,8 +7,9 @@ module OpenAI
         @threads ||= {}
       end
 
-      def new_thread(chat_id)
-        new_thread = ChatThread.new(initial_messages)
+      def new_thread(chat_id, model = nil)
+        msgs = config.open_ai.whitelist.include?(chat_id) ? initial_messages :
+        new_thread = ChatThread.new(msgs, model)
         threads[chat_id] = new_thread
       end
 
@@ -56,12 +57,13 @@ module OpenAI
       end
     end
 
+
     def self.included(base)
       base.extend ClassMethods
     end
 
     def init_session
-      self.class.new_thread(@chat.id)
+      self.class.new_thread(@chat.id, model)
       send_message(session_restart_message)
     end
 
@@ -120,7 +122,7 @@ module OpenAI
 
       response = open_ai.chat(
         parameters: {
-          model: config.open_ai["chat_gpt_model"],
+          model: current_thread.model || config.open_ai["chat_gpt_model"],
           messages: current_thread.as_json
         }
       )

data/lib/open_ai/chat_thread.rb

@@ -2,12 +2,14 @@
 
 module OpenAI
   class ChatThread
-    def initialize(defaults = [])
+    def initialize(defaults = [], model = nil)
       @history ||= defaults
+      @model = model
       puts @history
     end
 
     attr_reader :history
+    attr_reader :model
 
     alias_method :messages, :history
 

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.2.6"
+  spec.version       = "0.2.8"
   spec.authors       = ["bulgakke"]
   spec.email         = ["vvp835@yandex.ru"]
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: open_ai_bot
 version: !ruby/object:Gem::Version
-  version: 0.2.6
+  version: 0.2.8
 platform: ruby
 authors:
 - bulgakke
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-09-24 00:00:00.000000000 Z
+date: 2023-10-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: down