richat 0.3.3 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e4e97039d01522133cccb2e0577e36d1d7955b7f4a8f3dad30ca614f4133c57e
-  data.tar.gz: 958f913d87344bb797913d569952da0d211a91ebcc09dafa7c3f7da734a6d2b3
+  metadata.gz: 556b89d96438fa55db8004f7b7c988cc9e6f4e7186b5fe55cb3791e5a7448fc9
+  data.tar.gz: e7a2ca5f62fa9707130e1d535e4d8ff0137f3f0bcd5ee402b62b64ed01142c01
 SHA512:
-  metadata.gz: 31739062cb92784a44589feea8b2db3c43f0c9213ba748a66f46d10065e71a0d0aef07a5c7f30fd3a9f0f7849b714d5994b3dad887928789a0683b004402724e
-  data.tar.gz: c892747d24725cb3f793763317de4a522043fd0e4220463a4a07843636f78757f64c618e993898d35196e0b63345df7e15e0e8db73f0826243928d611723cd2c
+  metadata.gz: 8a8d20d4dc3d35a34dff1a3821e877404c883adbabb132ef696159048fec580dd1249953a3c6e4dccbc2348e30e6b118934707c72436d9d9a57aadd224f3038f
+  data.tar.gz: fbb27db3fa171b55d0e01c26bf78f5565169c0a1dc6aa0599c015d04233d0e08d1e306f8b6ff599ad7cc7dacdc09d275768c8016f29a251f85cac8a3b6d9ace9

data/lib/richat/command.rb

@@ -5,6 +5,7 @@ module Richat
     PROMPT_CHANGED_CODE = 2
     SYS_CMD_CODE = 3
     SYS_CHAT_CODE = 4
+    TOGGLE_CONTEXT_CODE = 5
 
     class << self
       attr_reader :prompt, :prompt_id
@@ -29,6 +30,8 @@ module Richat
             else
               handle_choose_prompt(user_input.split(" ").last)
             end
+          elsif user_input == "/context"
+            handle_toggle_context
           end
         end
       end
@@ -56,10 +59,11 @@ module Richat
       end
 
       def kill_process
-        return if @pid.nil? || !process_exist?
+        return if @pid.nil?
         if Gem.win_platform?
-          system("taskkill /F /PID #{@pid}")
+          system("taskkill /F /PID #{@pid}", err: File::NULL)
         else
+          return unless process_exist?
           begin
             Process.kill("TERM", @pid)
           rescue
@@ -84,11 +88,16 @@ 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/help\e[0m show help info"
         NEXT_CODE
       end
@@ -161,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,8 @@ 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" => "💡"
       },
       "sys_cmd" => {
         "activate_keywords" => [">", "!"],

data/lib/richat/shell.rb

@@ -36,7 +36,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,6 +58,13 @@ 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
             end
           end
 
@@ -108,12 +115,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"
       end
+
+      prompt_str
     end
   end
 end

data/lib/richat/version.rb

@@ -1,3 +1,3 @@
 module Richat
-  VERSION = "0.3.3"
+  VERSION = "0.3.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: richat
 version: !ruby/object:Gem::Version
-  version: 0.3.3
+  version: 0.3.5
 platform: ruby
 authors:
 - fzdp
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-11 00:00:00.000000000 Z
+date: 2023-04-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday