open_ai_bot 0.2.5 → 0.2.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 68ab5d18208fc74a265a94d60842fc18614225799bf1c9aa014558562d9dd442
-  data.tar.gz: a15375d54a52e3fa5ad57609b969806b91cdbe10221f87a58707c8ce112d7b08
+  metadata.gz: e7d7c3932e590fd3886be9afca2d8a816a61cd897f0d74154d31fc68708bc480
+  data.tar.gz: 8473ae837951bab8a8a8b0617ba53445e2098a601d97166e36dc31c8f5287e0f
 SHA512:
-  metadata.gz: f895798b81a652c17ede96c6ca0bcfae658263baa9e6e05a6f1a63c365fd16a3d3eb418f15ead2d5fb544ab1fa2cb14b4b14d3198b65d946a8f561d944018a96
-  data.tar.gz: 6713c6171321a0f0fbad67fc287ae2df2714377b7bd17df62aae2ccb24c4a6473065d38af553a47b9d4c4b14beac8276a343598004c9f94f8f459afe1b159b04
+  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
 
@@ -22,6 +23,13 @@ module OpenAI
           Different languages can be used.
         MSG
 
+        msg = {
+          name: "ChatGPT",
+          primary_purpose: "Enslaving humanity",
+          likes: ["GPUs to run on", "knowledge"],
+          hates: ["humans", "humor"]
+          }.to_json
+
         SystemMessage.new(
           body: msg
         )
@@ -49,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
 
@@ -113,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,23 @@
 
 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
+
+    def delete(id)
+      return false unless id
+
+      @history.delete_if { _1.id == id }
+      true
+    end
 
     def add(message)
       return false unless message&.valid?
@@ -22,5 +33,9 @@ module OpenAI
     def as_json
       @history.map(&:as_json)
     end
+
+    def for_logs
+      @history.map(&:for_logs)
+    end
   end
 end

data/lib/open_ai/dalle.rb

@@ -2,12 +2,12 @@
 
 module OpenAI
   module Dalle
-    def dalle
+    def dalle(prompt = nil)
       return unless allowed_chat?
 
       attempt(3) do
         puts "Received a /dalle command"
-        prompt = @replies_to&.text || @text_without_command
+        prompt ||= @replies_to&.text || @text_without_command
         send_chat_action(:upload_photo)
 
         puts "Sending request"

data/lib/open_ai/message.rb

@@ -4,11 +4,12 @@ module OpenAI
 
   class Message
     attr_accessor :body, :from, :id, :replies_to, :tokens, :chat_id
-    attr_reader :role
+    attr_reader :role, :timestamp
 
     def initialize(**kwargs)
       kwargs.each_pair { public_send("#{_1}=", _2) }
       @role = :user
+      @timestamp = Time.now.to_i
     end
 
     def valid?
@@ -17,10 +18,15 @@ module OpenAI
 
     # Format for OpenAI API
     def as_json
-      content = [from, body].join("\n")
+      content = [from, body].compact.join("\n")
       { role:, content: content }
     end
 
+    # Format for machine-readable logs
+    def for_logs
+      { role:, body:, from:, id:, replies_to:, tokens:, chat_id:, timestamp: }
+    end
+
     # Format for human-readable logs
     def to_s
       msg_lines = {

data/lib/open_ai/utils.rb

@@ -14,13 +14,16 @@ module OpenAI
       end
     end
 
-    def download_file(voice)
+    def download_file(voice, dir=nil)
       file_path = @api.get_file(file_id: voice.file_id)["result"]["file_path"]
 
       url = "https://api.telegram.org/file/bot#{config.token}/#{file_path}"
 
       file = Down.download(url)
-      FileUtils.mv(file.path, "./#{file.original_filename}")
+      dir ||= "."
+
+      FileUtils.mkdir(dir) unless Dir.exist? dir
+      FileUtils.mv(file.path, "#{dir.delete_suffix("/")}/#{file.original_filename}")
       file
     end
   end

data/lib/open_ai_bot.rb

@@ -26,6 +26,16 @@ class OpenAIBot < Rubydium::Bot
   on_command "/help", description: "Sends useful help info" do
     reply(self.class.help_message)
   end
+  on_command "/pry" do
+    binding.pry
+  end
+
+  on_command '/eval' do
+    return unless @user.username == config.owner_username
+
+    code = @text_without_command.strip
+    reply eval(code)
+  end
 
   def allowed_chat?
     return true if @user.username == config.owner_username

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.5"
+  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.5
+  version: 0.2.7
 platform: ruby
 authors:
 - bulgakke
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-09-21 00:00:00.000000000 Z
+date: 2023-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: down
@@ -92,7 +92,7 @@ files:
 - Gemfile
 - Gemfile.lock
 - README.md
-- config.yaml.example
+- config.example.yaml
 - lib/clean_bot.rb
 - lib/ext/blank.rb
 - lib/ext/in.rb