open_ai_bot 0.2.6 → 0.2.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bb11aa4fa04faff58bebc78dd9f5b7205bd7d096e0be1ea1260f44d62eaaf279
-  data.tar.gz: 79df08164999f67aa37cd1dff5d73d090d9665a38292923c43d9096805275e6c
+  metadata.gz: e7d7c3932e590fd3886be9afca2d8a816a61cd897f0d74154d31fc68708bc480
+  data.tar.gz: 8473ae837951bab8a8a8b0617ba53445e2098a601d97166e36dc31c8f5287e0f
 SHA512:
-  metadata.gz: 4bb4fec967f3d6a1ed50040cac94db80f7fff03740b29ffc07038f1f4de184f95754ce4bc6e159395ee825c71fadeb2a17605b44addbf1b3ba66caa9014e744d
-  data.tar.gz: 8162ea6493434c0770e4e3aafa91e017fb58a92b4eaaa93fd6feb30f269f18bc223ef80029f44db572d356743f8af9a317956781d845b18b7bd556d7ebf9e9a1
+  metadata.gz: 1c8cf3ea9587a13bb85fcdfa42bd10c879aee53e117c036964e3e65d9f922cc60589e80373d6555672df5fd6a10fa3bd39958ae400a2e26d2002f435c0294bf0
+  data.tar.gz: 32eb2f96a3d0a773dd809485caa889c3c03f6438989cf8188c0d1a3f2555b69d9c3c129788cff72f87d1340bb414fcf0afbd689317d5148623029516e429b5e1

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_at.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.7"
   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.7
 platform: ruby
 authors:
 - bulgakke
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-09-24 00:00:00.000000000 Z
+date: 2023-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: down