richat 0.3.5 → 0.3.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 556b89d96438fa55db8004f7b7c988cc9e6f4e7186b5fe55cb3791e5a7448fc9
-  data.tar.gz: e7a2ca5f62fa9707130e1d535e4d8ff0137f3f0bcd5ee402b62b64ed01142c01
+  metadata.gz: 2fb025384c114dec37924fed7ec5c9d9e9df8346d0b94bd831aacec551602d95
+  data.tar.gz: f60c3e845fcc33da304d2cee618219267fd0b4a7136f86afc3f62c094ed7e688
 SHA512:
-  metadata.gz: 8a8d20d4dc3d35a34dff1a3821e877404c883adbabb132ef696159048fec580dd1249953a3c6e4dccbc2348e30e6b118934707c72436d9d9a57aadd224f3038f
-  data.tar.gz: fbb27db3fa171b55d0e01c26bf78f5565169c0a1dc6aa0599c015d04233d0e08d1e306f8b6ff599ad7cc7dacdc09d275768c8016f29a251f85cac8a3b6d9ace9
+  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

@@ -6,6 +6,7 @@ module Richat
     SYS_CMD_CODE = 3
     SYS_CHAT_CODE = 4
     TOGGLE_CONTEXT_CODE = 5
+    RESET_CONTEXT_CODE = 6
 
     class << self
       attr_reader :prompt, :prompt_id
@@ -31,7 +32,9 @@ module Richat
               handle_choose_prompt(user_input.split(" ").last)
             end
           elsif user_input == "/context"
-            handle_toggle_context
+            TOGGLE_CONTEXT_CODE
+          elsif user_input == "/context-reset"
+            RESET_CONTEXT_CODE
           end
         end
       end
@@ -88,16 +91,13 @@ module Richat
         NEXT_CODE
       end
 
-      def handle_toggle_context
-        TOGGLE_CONTEXT_CODE
-      end
-
       def handle_help
         puts "Version #{VERSION}"
         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

data/lib/richat/config.rb

@@ -22,7 +22,8 @@ module Richat
         "show_welcome_info" => true,
         "shell_history_file" => "~/.richat/history.txt",
         "exit_keywords" => ["/exit", "q", "quit", "exit"],
-        "chat_context_indicator" => "💡"
+        "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
@@ -65,9 +66,14 @@ module Richat
               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 = ''
 
@@ -91,7 +97,7 @@ module Richat
             context_messages.pop
           end
 
-          puts
+          puts "\n\n"
         end
       rescue Interrupt
         if sys_cmd_mode
@@ -124,7 +130,7 @@ module Richat
       if sys_cmd_mode
         prompt_str += "\e[33m>>\e[0m \e[32m$ \e[0m"
       else
-        prompt_str += "\e[33m>> \e[0m"
+        prompt_str += "\e[33m>> \e[0m#{user_color}"
       end
 
       prompt_str

data/lib/richat/version.rb

@@ -1,3 +1,3 @@
 module Richat
-  VERSION = "0.3.5"
+  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.5
+  version: 0.3.6
 platform: ruby
 authors:
 - fzdp
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-13 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