chatgpt_assistant 0.1.3 → 0.1.5

data/lib/bots/helpers/validation_helper.rb

@@ -25,5 +25,14 @@ module ChatgptAssistant
     def discord_voice_bot_connected?
       user && evnt.user.voice_channel && evnt.voice && !chat.nil?
     end
+
+    def discord_next_action?
+      return true if evnt.channel.type != 1
+
+      %w[login register start help new_chat sl_chat ask list hist connect disconnect speak].each do |action|
+        return true if evnt.message.content.include?("#{discord_prefix}#{action}")
+      end
+      false
+    end
   end
 end

data/lib/bots/telegram_bot.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative "helpers/telegram_helper"
+require_relative "helpers/telegram_voice_helper"
 
 module ChatgptAssistant
   # This class is responsible for the telegram bot features
@@ -24,6 +25,7 @@ module ChatgptAssistant
     private
 
       include TelegramHelper
+      include TelegramVoiceHelper
 
       attr_accessor :msg, :visitor, :chat, :chat_id
 
@@ -47,14 +49,10 @@ module ChatgptAssistant
       end
 
       def start_event
-        register_visitor_action("start", visitor.id) unless visitor.tel_user
-        register_user_action("start", visitor.tel_user.id) if visitor.tel_user
         telegram_send_start_message
       end
 
       def help_event
-        register_visitor_action("help", visitor.id) unless visitor.tel_user
-        register_user_action("help", visitor.tel_user.id) if visitor.tel_user
         help_messages.each { |m| send_message m, msg.chat.id }
       end
 
@@ -63,7 +61,6 @@ module ChatgptAssistant
         return no_chat_selected_message unless user.current_chat
         return no_messages_founded_message if user.current_chat.messages.count.zero?
 
-        register_user_action("hist", user.id)
         telegram_user_history.each do |m|
           send_message m, msg.chat.id
         end
@@ -72,7 +69,6 @@ module ChatgptAssistant
       def list_event
         return unless valid_for_list_action?
 
-        register_user_action("list", user.id)
         send_message commom_messages[:chat_list], msg.chat.id
         chats_str = ""
         user.chats.each_with_index { |c, i| chats_str += "Chat #{i + 1} - #{c.title}\n" }
@@ -90,7 +86,6 @@ module ChatgptAssistant
       end
 
       def login_event
-        register_visitor_action("login", visitor.id) unless visitor.tel_user
         user_info = msg.text.split("/").last
         email, password = user_info.split(":")
         case telegram_user_auth(email, password, msg.chat.id)
@@ -104,7 +99,6 @@ module ChatgptAssistant
       end
 
       def register_event
-        register_visitor_action("register", visitor.id)
         user_info = msg.text.split("/").last
         email, password = user_info.split(":")
         registered_email = telegram_user_create visitor.id, email, password
@@ -114,14 +108,12 @@ module ChatgptAssistant
       def new_chat_event
         return not_logged_in_message unless user
 
-        register_user_action("new_chat", user.id)
         telegram_create_chat
       end
 
       def select_chat_event
         return not_logged_in_message unless user
 
-        register_user_action("select_chat", user.id)
         title = msg.text.split("/").last
         chat = user.chat_by_title(title)
         return chat_not_found_message unless chat
@@ -131,15 +123,11 @@ module ChatgptAssistant
       end
 
       def stop_event
-        register_event_action("stop", visitor.id) unless visitor.tel_user
-        register_user_action("stop", visitor.tel_user.id) if visitor.tel_user
         send_message commom_messages[:stop], msg.chat.id
         bot.api.leave_chat(chat_id: msg.chat.id)
       end
 
       def nil_event
-        register_event_action("nil", visitor.id) unless visitor.tel_user
-        register_user_action("nil", visitor.tel_user.id) if visitor.tel_user
         send_message error_messages[:nil], msg.chat.id
       end
 
@@ -147,7 +135,6 @@ module ChatgptAssistant
         return not_logged_in_message unless user
         return no_chat_selected_message if user.current_chat_id.nil?
 
-        register_user_action("audio", user.id)
         user_audio = transcribe_file(telegram_audio_url)
         message = Message.new(content: user_audio[:text], chat_id: user.current_chat_id, role: "user")
         if message.save

data/lib/chatgpt_assistant/audio_recognition.rb

@@ -29,9 +29,7 @@ module ChatgptAssistant
 
       def download_audio(audio_url)
         audio_conn = Faraday.new(url: audio_url)
-        File.open(dl_file_name, "wb") do |file|
-          file.write(audio_conn.get.body)
-        end
+        File.binwrite(dl_file_name, audio_conn.get.body)
         FFMPEG::Movie.new(dl_file_name).transcode(file_name)
         File.delete(dl_file_name)
       end
@@ -39,14 +37,14 @@ module ChatgptAssistant
       def header
         {
           "Content-Type": "multipart/form-data",
-          "Authorization": "Bearer #{openai_api_key}"
+          Authorization: "Bearer #{openai_api_key}"
         }
       end
 
       def payload
         {
-          "file": Faraday::UploadIO.new(file_name, "audio/mp3"),
-          "model": "whisper-1"
+          file: Faraday::UploadIO.new(file_name, "audio/mp3"),
+          model: "whisper-1"
         }
       end
 

data/lib/chatgpt_assistant/audio_synthesis.rb

@@ -61,9 +61,7 @@ module ChatgptAssistant
           engine: "neural"
         )
 
-        File.open("voice/aws-#{time}.mp3", "wb") do |file|
-          file.write(response.audio_stream.read)
-        end
+        File.binwrite("voice/aws-#{time}.mp3", response.audio_stream.read)
         "voice/aws-#{time}.mp3"
       end
 
@@ -86,9 +84,7 @@ module ChatgptAssistant
           voice: voice
         ).result
 
-        File.open("voice/ibm-#{time}.mp3", "wb") do |audio_file|
-          audio_file.write(audio)
-        end
+        File.binwrite("voice/ibm-#{time}.mp3", audio)
         "voice/ibm-#{time}.mp3"
       end
 

data/lib/chatgpt_assistant/chatter.rb

@@ -10,27 +10,36 @@ module ChatgptAssistant
       @openai_api_key = openai_api_key
     end
 
-    def chat(message, chat_id)
+    def chat(message, chat_id, error_message)
+      @error_message = error_message
       @chat_id = chat_id
       @message = message
       @response = request(message)
       @json = JSON.parse(response.body)
 
-      return error_log if response.status != 200
+      return no_response_error if json["choices"].empty?
+      return bot_offline_error if response.status != 200
 
       text = json["choices"][0]["message"]["content"]
 
       Message.create(content: text, role: "assistant", chat_id: chat_id)
       text
+    rescue StandardError => e
+      Error.create(message: e.message, backtrace: e.backtrace)
+      error_message
     end
 
     private
 
       attr_reader :openai_api_key, :response, :message
-      attr_accessor :chat_id, :json
+      attr_accessor :chat_id, :json, :error_message
 
-      def error_log
-        "Algo deu errado, tente novamente mais tarde." # TODO: use a DefaultMessage object
+      def no_response_error
+        "I'm sorry, I didn't understand you. Please, try again."
+      end
+
+      def bot_offline_error
+        "I'm sorry, I'm offline. Please, try again later."
       end
 
       def header
@@ -59,7 +68,9 @@ module ChatgptAssistant
                    end
         {
           model: "gpt-3.5-turbo",
-          messages: messages
+          messages: messages,
+          max_tokens: 1000,
+          temperature: 0.1
         }.to_json
       end
 

data/lib/chatgpt_assistant/config.rb

@@ -9,23 +9,23 @@ module ChatgptAssistant
   # This class is responsible for the configuration of the Chatgpt Assistant
   class Config
     def initialize
-      @env_type = ENV["ENV_TYPE"]
-      @language = ENV["LANGUAGE"]
-      @mode = ENV["MODE"]
-      @database_host = ENV["POSTGRES_HOST"]
-      @database_name = ENV["POSTGRES_DB"]
-      @database_username = ENV["POSTGRES_USER"]
-      @database_password = ENV["POSTGRES_PASSWORD"]
-      @openai_api_key = ENV["OPENAI_API_KEY"]
-      @telegram_token = ENV["TELEGRAM_TOKEN"]
-      @discord_token = ENV["DISCORD_TOKEN"]
-      @discord_client_id = ENV["DISCORD_CLIENT_ID"]
-      @ibm_api_key = ENV["IBM_API_KEY"]
-      @ibm_url = ENV["IBM_URL"]
-      @aws_access_key_id = ENV["AWS_ACCESS_KEY_ID"]
-      @aws_secret_access_key = ENV["AWS_SECRET_ACCESS_KEY"]
-      @aws_region = ENV["AWS_REGION"]
-      @discord_prefix = ENV["DISCORD_PREFIX"]
+      @env_type = ENV.fetch("ENV_TYPE", nil)
+      @language = ENV.fetch("LANGUAGE", nil)
+      @mode = ENV.fetch("MODE", nil)
+      @database_host = ENV.fetch("POSTGRES_HOST", nil)
+      @database_name = ENV.fetch("POSTGRES_DB", nil)
+      @database_username = ENV.fetch("POSTGRES_USER", nil)
+      @database_password = ENV.fetch("POSTGRES_PASSWORD", nil)
+      @openai_api_key = ENV.fetch("OPENAI_API_KEY", nil)
+      @telegram_token = ENV.fetch("TELEGRAM_TOKEN", nil)
+      @discord_token = ENV.fetch("DISCORD_TOKEN", nil)
+      @discord_client_id = ENV.fetch("DISCORD_CLIENT_ID", nil)
+      @ibm_api_key = ENV.fetch("IBM_API_KEY", nil)
+      @ibm_url = ENV.fetch("IBM_URL", nil)
+      @aws_access_key_id = ENV.fetch("AWS_ACCESS_KEY_ID", nil)
+      @aws_secret_access_key = ENV.fetch("AWS_SECRET_ACCESS_KEY", nil)
+      @aws_region = ENV.fetch("AWS_REGION", nil)
+      @discord_prefix = ENV.fetch("DISCORD_PREFIX", nil)
     end
 
     attr_reader :openai_api_key, :telegram_token, :discord_token, :ibm_api_key, :ibm_url,
@@ -48,9 +48,7 @@ module ChatgptAssistant
       db_connection
       ActiveRecord::Base.logger = Logger.new($stdout)
       VisitorMigration.new.migrate(:up)
-      VisitorActionsMigration.new.migrate(:up)
       UserMigration.new.migrate(:up)
-      UserActionsMigration.new.migrate(:up)
       ChatMigration.new.migrate(:up)
       MessageMigration.new.migrate(:up)
       ErrorMigration.new.migrate(:up)

data/lib/chatgpt_assistant/migrations.rb

@@ -13,22 +13,7 @@ class VisitorMigration < ActiveRecord::Migration[5.2]
       t.string :discord_id, limit: 100
       t.integer :platform, null: false, default: 0
       t.string :name, null: false
-      t.integer :current_user_id, null: false, default: 0
-      t.timestamps
-    end
-  end
-end
-
-# VisitorActions model
-class VisitorActionsMigration < ActiveRecord::Migration[5.2]
-  def change
-    return if ActiveRecord::Base.connection.table_exists? :visitor_actions
-
-    create_table :visitor_actions do |t|
-      t.references :visitor, null: false, foreign_key: true
-      t.text :action, null: false
-      t.text :description, null: false
-      t.integer :role, null: false, default: 0
+      t.integer :current_user_id, default: 0
       t.timestamps
     end
   end
@@ -56,21 +41,6 @@ class UserMigration < ActiveRecord::Migration[5.2]
   end
 end
 
-# UserActions model
-class UserActionsMigration < ActiveRecord::Migration[5.2]
-  def change
-    return if ActiveRecord::Base.connection.table_exists? :user_actions
-
-    create_table :user_actions do |t|
-      t.references :user, null: false, foreign_key: true
-      t.text :action, null: false, default: ""
-      t.text :description, null: false, default: ""
-      t.integer :role, null: false, default: 0
-      t.timestamps
-    end
-  end
-end
-
 # Chat model
 class ChatMigration < ActiveRecord::Migration[5.2]
   def change
@@ -80,6 +50,9 @@ class ChatMigration < ActiveRecord::Migration[5.2]
       t.references :user, null: false, foreign_key: true
       t.text :title, null: false, default: "", limit: 100
       t.integer :status, null: false, default: 0
+      t.integer :messages_count, null: false, default: 0
+      t.string :actor
+      t.text :prompt
       t.timestamps
     end
   end

data/lib/chatgpt_assistant/models.rb

@@ -7,8 +7,6 @@ class Visitor < ActiveRecord::Base
   has_many :visitor_actions
   # has_one :tel_user, foreign_key: "telegram_id", class_name: "User"
   # has_one :dis_user, foreign_key: "discord_id", class_name: "User"
-  validates :telegram_id, uniqueness: true
-  validates :discord_id, uniqueness: true
   validates :name, presence: true
   validates :platform, presence: true
   enum platform: { telegram: 0, discord: 1 }
@@ -22,15 +20,6 @@ class Visitor < ActiveRecord::Base
   end
 end
 
-# VisitorActions model
-class VisitorAction < ActiveRecord::Base
-  belongs_to :visitor
-  validates :action, presence: true
-  validates :description, presence: true
-  validates :role, presence: true
-  enum role: { user: 0, assistant: 1, warning: 2 }
-end
-
 # User model
 class User < ActiveRecord::Base
   attr_accessor :password
@@ -66,16 +55,6 @@ class User < ActiveRecord::Base
   end
 end
 
-# UserActions model
-class UserAction < ActiveRecord::Base
-  belongs_to :user
-  validates :user_id, presence: true
-  validates :action, presence: true
-  validates :description, presence: true
-  validates :role, presence: true
-  enum role: { user: 0, assistant: 1, warning: 2 }
-end
-
 # Chat model
 class Chat < ActiveRecord::Base
   validates :user_id, presence: true
@@ -84,6 +63,15 @@ class Chat < ActiveRecord::Base
 
   belongs_to :user
   has_many :messages
+
+  after_create :init_chat_if_actor_provided
+
+  def init_chat_if_actor_provided
+    return if actor.nil?
+
+    messages.create(content: prompt, role: "user")
+    messages.create(content: "Hello, I'm #{actor}. I will follow #{prompt}", role: "assistant")
+  end
 end
 
 # Message model

data/lib/chatgpt_assistant/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ChatgptAssistant
-  VERSION = "0.1.3"
+  VERSION = "0.1.5"
 end

data/lib/chatgpt_assistant.rb

@@ -10,6 +10,7 @@ require_relative "chatgpt_assistant/models"
 require_relative "bots/application_bot"
 require_relative "bots/telegram_bot"
 require_relative "bots/discord_bot"
+require "awesome_chatgpt_actors"
 require "streamio-ffmpeg"
 require "aws-sdk-polly"
 require "telegram/bot"