richat 0.3.4 → 0.3.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7a7ca5a2cd65ccff6ccf2a79f9c0f93c05509bfe06118eed1f7b1f5a471cbd56
-  data.tar.gz: 9db3acdef234d0c0e9c826d4da10238095376fd161cc13cd3238d8515ac4d658
+  metadata.gz: 2fb025384c114dec37924fed7ec5c9d9e9df8346d0b94bd831aacec551602d95
+  data.tar.gz: f60c3e845fcc33da304d2cee618219267fd0b4a7136f86afc3f62c094ed7e688
 SHA512:
-  metadata.gz: 923bc4c0bd06f5d947a2fd509f91c515fb1ba31368b6bbf6b5ba79283106716c19939c2cc3fba432de2e2a7fcd3fa4769793cc63e113f27ee7ce08c04652fac4
-  data.tar.gz: fc07dde0548be71271a9c9b2e9eadd4ece24f816a4ed7d12f3c247e003d7f9f60f4404134a85a609ef807c93ac8ed3d45e8ac0a8a4d2da2df133c58494e3b1fe
+  metadata.gz: 8d457a5fd0c5c1740b5b114aee779dbe2d3b77b7f780ac24743bf857b563147df43de933f5d673c8a35070e5e034a9ed074506679defbf10e664d7342d117104
+  data.tar.gz: 4ea395f9d4d57caa9b0c04f2fc87b901e6dbcc065e770eb50699a6bf7e1fd2ae492083d682b317b9ca95d059c94425b7d446566603aa39c0a48aadcc1e11fe6f

data/lib/richat/color_code.rb

@@ -0,0 +1,20 @@
+module Richat
+  class ColorCode
+    COLOR_CODE = {
+      black: "\e[30m",
+      red: "\e[31m",
+      green: "\e[32m",
+      yellow: "\e[33m",
+      blue: "\e[34m",
+      magenta: "\e[35m",
+      cyan: "\e[36m",
+      white: "\e[37m"
+    }
+
+    class << self
+      def get_code(color)
+        COLOR_CODE.fetch(color.to_sym, nil)
+      end
+    end
+  end
+end

data/lib/richat/command.rb

@@ -5,6 +5,8 @@ module Richat
     PROMPT_CHANGED_CODE = 2
     SYS_CMD_CODE = 3
     SYS_CHAT_CODE = 4
+    TOGGLE_CONTEXT_CODE = 5
+    RESET_CONTEXT_CODE = 6
 
     class << self
       attr_reader :prompt, :prompt_id
@@ -29,6 +31,10 @@ module Richat
             else
               handle_choose_prompt(user_input.split(" ").last)
             end
+          elsif user_input == "/context"
+            TOGGLE_CONTEXT_CODE
+          elsif user_input == "/context-reset"
+            RESET_CONTEXT_CODE
           end
         end
       end
@@ -90,6 +96,8 @@ module Richat
         puts "\e[32m/exit\e[0m exit Richat"
         puts "\e[32m/config\e[0m show configuration"
         puts "\e[32m/prompt\e[0m show prompt list"
+        puts "\e[32m/context\e[0m toggle chat context"
+        puts "\e[32m/context-reset\e[0m reset chat context"
         puts "\e[32m/help\e[0m show help info"
         NEXT_CODE
       end
@@ -162,6 +170,10 @@ module Richat
         puts "\e[31m#{message}\e[0m"
       end
 
+      def print_info(message)
+        puts "\e[32m#{message}\e[0m"
+      end
+
       def print_welcome
         puts "Richat is a command-line ChatGPT tool implemented in Ruby that supports highly customizable configuration, press \e[32m/help\e[0m to display help info. If you have any suggestions or questions, please feel free to provide feedback on https://github.com/fzdp/richat."
       end

data/lib/richat/config.rb

@@ -21,7 +21,9 @@ module Richat
         "enable_chat_context" => true,
         "show_welcome_info" => true,
         "shell_history_file" => "~/.richat/history.txt",
-        "exit_keywords" => ["/exit", "q", "quit", "exit"]
+        "exit_keywords" => ["/exit", "q", "quit", "exit"],
+        "chat_context_indicator" => "💡",
+        "user_content_color" => "green"
       },
       "sys_cmd" => {
         "activate_keywords" => [">", "!"],

data/lib/richat/shell.rb

@@ -2,7 +2,7 @@ require 'readline'
 
 module Richat
   class Shell
-    attr_reader :chat_client, :logger, :user_role, :ai_role, :system_role, :history_path
+    attr_reader :chat_client, :logger, :user_role, :ai_role, :system_role, :history_path, :user_color
 
     def initialize(options = {})
       @chat_client = options[:chat_client]
@@ -12,6 +12,7 @@ module Richat
       @system_role = Config.get("log", "system_role")
       @history_path = File.expand_path(Config.get("shell", "shell_history_file"))
       File.open(@history_path, 'w') {} unless File.exist?(@history_path)
+      @user_color = ColorCode.get_code(Config.get("shell", "user_content_color"))
     end
 
     def call
@@ -36,7 +37,7 @@ module Richat
       sys_cmd_mode = false
 
       begin
-        while (user_content = Readline.readline(shell_prompt(sys_cmd_mode), true))
+        while (user_content = Readline.readline(shell_prompt(sys_cmd_mode, enable_context_message), true))
           if user_content.empty?
             Readline::HISTORY&.pop
             next
@@ -58,9 +59,21 @@ module Richat
             elsif code == Command::SYS_CHAT_CODE
               sys_cmd_mode = false
               next
+            elsif code == Command::TOGGLE_CONTEXT_CODE
+              enable_context_message = !enable_context_message
+              unless enable_context_message
+                context_messages = Command.prompt_id ? [{ role: 'system', content: Command.prompt }] : []
+              end
+              Command.print_info("chat context mode " + (enable_context_message ? "enabled." : "disabled."))
+              next
+            elsif code == Command::RESET_CONTEXT_CODE
+              context_messages = Command.prompt_id ? [{ role: 'system', content: Command.prompt }] : []
+              Command.print_info("chat context reset.")
+              next
             end
           end
 
+          print "\e[0m"
           logger.call(role: user_role, content: user_content)
           response = ''
 
@@ -84,7 +97,7 @@ module Richat
             context_messages.pop
           end
 
-          puts
+          puts "\n\n"
         end
       rescue Interrupt
         if sys_cmd_mode
@@ -108,12 +121,19 @@ module Richat
 
     private
 
-    def shell_prompt(sys_cmd_mode)
+    def shell_prompt(sys_cmd_mode, chat_context_mode)
+      prompt_str = "\e[32m#{Command.prompt_id&.+" "}\e[0m"
+      if chat_context_mode
+        prompt_str += Config.get("shell", "chat_context_indicator")
+      end
+
       if sys_cmd_mode
-        "\e[32m#{Command.prompt_id&.+" "}\e[0m\e[33m>>\e[0m \e[32m$ \e[0m"
+        prompt_str += "\e[33m>>\e[0m \e[32m$ \e[0m"
       else
-        "\e[32m#{Command.prompt_id&.+" "}\e[0m\e[33m>> \e[0m"
+        prompt_str += "\e[33m>> \e[0m#{user_color}"
       end
+
+      prompt_str
     end
   end
 end

data/lib/richat/version.rb

@@ -1,3 +1,3 @@
 module Richat
-  VERSION = "0.3.4"
+  VERSION = "0.3.6"
 end

data/lib/richat.rb

@@ -1,5 +1,6 @@
 require "richat/version"
 
+require_relative 'richat/color_code'
 require_relative 'richat/utils'
 require_relative 'richat/config'
 require_relative 'richat/command'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: richat
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  version: 0.3.6
 platform: ruby
 authors:
 - fzdp
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-11 00:00:00.000000000 Z
+date: 2023-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -81,6 +81,7 @@ files:
 - lib/openai/chat.rb
 - lib/richat.rb
 - lib/richat/cli.rb
+- lib/richat/color_code.rb
 - lib/richat/command.rb
 - lib/richat/config.rb
 - lib/richat/logger.rb