open_ai_bot 0.2.4 → 0.2.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 867a815eac31b0a468c48e4318cdf2d235d575a7f950fb6dda6e084947b5b31f
-  data.tar.gz: 1d18f4c71a5725d06325f0e6725328af9acae9d9afa25dd1a91696b8e93d6e3f
+  metadata.gz: bb11aa4fa04faff58bebc78dd9f5b7205bd7d096e0be1ea1260f44d62eaaf279
+  data.tar.gz: 79df08164999f67aa37cd1dff5d73d090d9665a38292923c43d9096805275e6c
 SHA512:
-  metadata.gz: ccae7c2e54c883609af20f20c7b19dd3c59e14844d44964cfdd445a2357312c8aa3eebc6bfeb169f2d78fe62a92c49bcf8e7cf8958a810a47624b7a4c6e5b79c
-  data.tar.gz: '076387746b93f4d11cdf594727ac2b575cfe02b7b983b50e5928a938671937cda73d17760f5f246efd1059ce66597386f343a168640d29819c62ab1dcb6dd5c7'
+  metadata.gz: 4bb4fec967f3d6a1ed50040cac94db80f7fff03740b29ffc07038f1f4de184f95754ce4bc6e159395ee825c71fadeb2a17605b44addbf1b3ba66caa9014e744d
+  data.tar.gz: 8162ea6493434c0770e4e3aafa91e017fb58a92b4eaaa93fd6feb30f269f18bc223ef80029f44db572d356743f8af9a317956781d845b18b7bd556d7ebf9e9a1

data/lib/open_ai/chat_gpt.rb

@@ -22,6 +22,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
         )

data/lib/open_ai/chat_thread.rb

@@ -9,6 +9,15 @@ module OpenAI
 
     attr_reader :history
 
+    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?
       return false if @history.any? { message.id == _1.id}
@@ -22,5 +31,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

@@ -7,6 +7,10 @@ require_relative "open_ai/dalle"
 require_relative "open_ai/utils"
 require_relative "open_ai/whisper"
 
+require_relative "ext/blank"
+require_relative "ext/in"
+
+
 class OpenAIBot < Rubydium::Bot
   include OpenAI::ChatGPT
   include OpenAI::Dalle
@@ -22,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/main.rb

@@ -6,9 +6,6 @@ require "yaml"
 require "down"
 require "rubydium"
 
-require_relative "lib/ext/blank"
-require_relative "lib/ext/in"
-
 require_relative "lib/open_ai_bot"
 require_relative "lib/clean_bot"
 require_relative "lib/my_custom_bot"

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.4"
+  spec.version       = "0.2.6"
   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.4
+  version: 0.2.6
 platform: ruby
 authors:
 - bulgakke
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-09-21 00:00:00.000000000 Z
+date: 2023-09-24 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