legion-tty 0.4.13 → 0.4.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bc958211b7004e0392a9e96cf2a874d6c54d6f0c6f183d87d80f428e16d3b014
-  data.tar.gz: 13264f1761391724054983f250767d9308aab1a397d0dc1577292268a0f7e031
+  metadata.gz: bce8d6f696692fc8e9173eee8cb0afd87f37dc48b12325a149a7de1748f13447
+  data.tar.gz: d0e115df860ddd59c02e92b27d65a74a3a1413bafd25c7484f8dcc8c631832fb
 SHA512:
-  metadata.gz: fd92069e863ebb5bd1185068adf3f50323fb276a2ae0479a8253e0220c1e4bc22a76abb7b8eb73a4c4a22c84be9f22cddf6eea22d510d7772fd459d3d74a1ff9
-  data.tar.gz: afcce634192c2e16f133b0d237cd3720536dfce07e2cae17089f208867546d82d75a58bb352887da1d0d29c0e651d047b79596fea0428d35f152e48f17707069
+  metadata.gz: c6bc9b9eb07cf1838830fddd0bf626c4d9a0404db3330f7f6a658c5dd6425f9293d316a3a6f013858262f47ddb2875421c768a849a14213432028d66f56d94c7
+  data.tar.gz: 4eddcc425c258ec26d3c74e9ec3a9ce1ac470f8b25e985bed26db41f504e3825a61f21113587ccb70c5e7ce35789e06325affba7b704e0580687165687d4dbf5

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## [0.4.14] - 2026-03-19
+
+### Added
+- LLM response timing: tracks elapsed time per response, shows notification, includes avg in `/stats`
+- `/wc` command: word count statistics per role (user/assistant/system) with averages
+- `/import <path>` command: import session from any JSON file path with validation
+- `/mute` command: toggle system message display in chat (messages still tracked, just hidden)
+
 ## [0.4.13] - 2026-03-19
 
 ### Added

data/lib/legion/tty/components/message_stream.rb

@@ -5,12 +5,15 @@ require_relative '../theme'
 module Legion
   module TTY
     module Components
+      # rubocop:disable Metrics/ClassLength
       class MessageStream
         attr_reader :messages, :scroll_offset
+        attr_accessor :mute_system
 
         def initialize
           @messages = []
           @scroll_offset = 0
+          @mute_system = false
         end
 
         def add_message(role:, content:)
@@ -69,7 +72,11 @@ module Legion
         private
 
         def build_all_lines(width)
-          @messages.flat_map { |msg| render_message(msg, width) }
+          @messages.flat_map do |msg|
+            next [] if @mute_system && msg[:role] == :system
+
+            render_message(msg, width)
+          end
         end
 
         def render_message(msg, width)
@@ -131,6 +138,7 @@ module Legion
           panel.instance_variable_set(:@error, error) if error
         end
       end
+      # rubocop:enable Metrics/ClassLength
     end
   end
 end

data/lib/legion/tty/screens/chat/session_commands.rb

@@ -4,6 +4,7 @@ module Legion
   module TTY
     module Screens
       class Chat < Base
+        # rubocop:disable Metrics/ModuleLength
         module SessionCommands
           private
 
@@ -76,6 +77,42 @@ module Legion
             :handled
           end
 
+          def handle_import(input)
+            path = input.split(nil, 2)[1]
+            unless path
+              @message_stream.add_message(role: :system, content: 'Usage: /import <path>')
+              return :handled
+            end
+
+            load_import_file(File.expand_path(path))
+          rescue ::JSON::ParserError => e
+            @message_stream.add_message(role: :system, content: "Invalid JSON: #{e.message}")
+            :handled
+          end
+
+          def load_import_file(path)
+            unless File.exist?(path)
+              @message_stream.add_message(role: :system, content: "File not found: #{path}")
+              return :handled
+            end
+
+            data = ::JSON.parse(File.read(path), symbolize_names: true)
+            unless data.is_a?(Hash) && data[:messages].is_a?(Array)
+              @message_stream.add_message(role: :system, content: 'Invalid session file: missing messages array.')
+              return :handled
+            end
+
+            apply_imported_messages(data[:messages], path)
+          end
+
+          def apply_imported_messages(messages, path)
+            imported = messages.map { |m| { role: m[:role].to_sym, content: m[:content].to_s } }
+            @message_stream.messages.replace(imported)
+            @status_bar.notify(message: "Imported #{imported.size} messages", level: :success, ttl: 3)
+            @message_stream.add_message(role: :system, content: "Imported #{imported.size} messages from #{path}.")
+            :handled
+          end
+
           def auto_save_session
             return if @message_stream.messages.empty?
 
@@ -87,6 +124,7 @@ module Legion
             nil
           end
         end
+        # rubocop:enable Metrics/ModuleLength
       end
     end
   end

data/lib/legion/tty/screens/chat/ui_commands.rb

@@ -204,15 +204,62 @@ module Legion
             end
           end
 
+          def handle_wc
+            msgs = @message_stream.messages
+            by_role = word_counts_by_role(msgs)
+            total = by_role.values.sum
+            avg = (total.to_f / [msgs.size, 1].max).round
+            @message_stream.add_message(role: :system, content: build_wc_lines(by_role, total, avg).join("\n"))
+            :handled
+          end
+
+          def word_counts_by_role(msgs)
+            %i[user assistant system].to_h do |role|
+              words = msgs.select { |m| m[:role] == role }.sum { |m| m[:content].to_s.split.size }
+              [role, words]
+            end
+          end
+
+          def build_wc_lines(by_role, total, avg)
+            [
+              'Word count:',
+              "  Total: #{format_stat_number(total)}",
+              "  User: #{format_stat_number(by_role[:user])}",
+              "  Assistant: #{format_stat_number(by_role[:assistant])}",
+              "  System: #{format_stat_number(by_role[:system])}",
+              "  Avg words/message: #{avg}"
+            ]
+          end
+
+          def handle_mute
+            @muted_system = !@muted_system
+            @message_stream.mute_system = @muted_system
+            if @muted_system
+              @status_bar.notify(message: 'System messages hidden', level: :info, ttl: 3)
+            else
+              @status_bar.notify(message: 'System messages visible', level: :info, ttl: 3)
+            end
+            :handled
+          end
+
           def build_stats_lines
             msgs = @message_stream.messages
             counts = count_by_role(msgs)
             total_chars = msgs.sum { |m| m[:content].to_s.length }
             lines = stats_header_lines(msgs, counts, total_chars)
             lines << "  Tool calls: #{counts[:tool]}" if counts[:tool].positive?
+            append_response_time_stat(lines, msgs)
             lines
           end
 
+          def append_response_time_stat(lines, msgs)
+            timed = msgs.select { |m| m[:response_time] }
+            return unless timed.any?
+
+            avg_rt = timed.sum { |m| m[:response_time] }.to_f / timed.size
+            lines << "  Avg response time: #{avg_rt.round(2)}s (#{timed.size} responses)"
+          end
+
           def count_by_role(msgs)
             %i[user assistant system tool].to_h { |role| [role, msgs.count { |m| m[:role] == role }] }
           end

data/lib/legion/tty/screens/chat.rb

@@ -28,7 +28,8 @@ module Legion
         SLASH_COMMANDS = %w[/help /quit /clear /compact /copy /diff /model /session /cost /export /tools /dashboard
                             /hotkeys /save /load /sessions /system /delete /plan /palette /extensions /config
                             /theme /search /grep /stats /personality /undo /history /pin /pins /rename
-                            /context /alias /snippet /debug /uptime /time /bookmark /welcome /tips].freeze
+                            /context /alias /snippet /debug /uptime /time /bookmark /welcome /tips
+                            /wc /import /mute].freeze
 
         PERSONALITIES = {
           'default' => 'You are Legion, an async cognition engine and AI assistant. Be helpful and concise.',
@@ -58,6 +59,7 @@ module Legion
           @snippets = {}
           @debug_mode = false
           @session_start = Time.now
+          @muted_system = false
         end
 
         # rubocop:enable Metrics/AbcSize
@@ -202,15 +204,23 @@ module Legion
 
           @status_bar.update(thinking: true)
           render_screen
+          start_time = Time.now
           response = @llm_chat.ask(message) do |chunk|
             @status_bar.update(thinking: false)
             @message_stream.append_streaming(chunk.content) if chunk.content
             render_screen
           end
+          record_response_time(Time.now - start_time)
           @status_bar.update(thinking: false)
           track_response_tokens(response)
         end
 
+        def record_response_time(elapsed)
+          @last_response_time = elapsed
+          @message_stream.messages.last[:response_time] = elapsed if @message_stream.messages.last
+          @status_bar.notify(message: "Response: #{elapsed.round(1)}s", level: :info, ttl: 4)
+        end
+
         def daemon_available?
           !!(defined?(Legion::LLM::DaemonClient) && Legion::LLM::DaemonClient.available?)
         end
@@ -340,6 +350,9 @@ module Legion
           when '/bookmark' then handle_bookmark
           when '/welcome' then handle_welcome
           when '/tips' then handle_tips
+          when '/wc' then handle_wc
+          when '/import' then handle_import(input)
+          when '/mute' then handle_mute
           else :handled
           end
         end

data/lib/legion/tty/version.rb

@@ -2,6 +2,6 @@
 
 module Legion
   module TTY
-    VERSION = '0.4.13'
+    VERSION = '0.4.14'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: legion-tty
 version: !ruby/object:Gem::Version
-  version: 0.4.13
+  version: 0.4.14
 platform: ruby
 authors:
 - Esity