richat 0.3.4 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7a7ca5a2cd65ccff6ccf2a79f9c0f93c05509bfe06118eed1f7b1f5a471cbd56
-  data.tar.gz: 9db3acdef234d0c0e9c826d4da10238095376fd161cc13cd3238d8515ac4d658
+  metadata.gz: 556b89d96438fa55db8004f7b7c988cc9e6f4e7186b5fe55cb3791e5a7448fc9
+  data.tar.gz: e7a2ca5f62fa9707130e1d535e4d8ff0137f3f0bcd5ee402b62b64ed01142c01
 SHA512:
-  metadata.gz: 923bc4c0bd06f5d947a2fd509f91c515fb1ba31368b6bbf6b5ba79283106716c19939c2cc3fba432de2e2a7fcd3fa4769793cc63e113f27ee7ce08c04652fac4
-  data.tar.gz: fc07dde0548be71271a9c9b2e9eadd4ece24f816a4ed7d12f3c247e003d7f9f60f4404134a85a609ef807c93ac8ed3d45e8ac0a8a4d2da2df133c58494e3b1fe
+  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
@@ -85,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
@@ -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,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.4"
+  VERSION = "0.3.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: richat
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  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