legion-tty 0.4.36 → 0.4.37

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2e331de1a9cb6b1a10576797303b5f1c691b639a74212c5cca6dc5d5fb161f7c
-  data.tar.gz: 01b5bfdefa8e991d01db3da00c1f0d7f51ff029d08c89e6a008a1abb607c8543
+  metadata.gz: f6aab5b9e38570a9072b676e7573506c5051693e503ff3861c1f7c042e836267
+  data.tar.gz: 412cd703adfd6117e87fb0a57820d8d01c60d2701dea9b2b0c6c30c5f0a44f06
 SHA512:
-  metadata.gz: f94def1314643beeda49eb00c7a9f70a4cacee8843c56551bf24b9edfa77dfcbf17fbca6f9c5f708574a14bdf3316e382a04d41e542e8d841eba21324cc0d248
-  data.tar.gz: fa6d3c970d1e33492185f3cb89e3dbc9f349c8f36431558f86f13bd1a48c96656a6b0adecee84bd2dc3713ee11c25a09282bdacc3cc7ffff6a3886e785b1839e
+  metadata.gz: e6164b922a3e05f202cc8ae224ce0a8ebc84cba655ee333bfc74dfaf66f3ab62aeff4dfde49e624d7d40f26e0cb377a3ba36a20b0f03e0806f55f08e9efd43da
+  data.tar.gz: 37b23570482bc80906620439d165594bd436d56c87da0f13612729e8e5d06f26ebf786881c88b708236efb9af3d55f71c407e46051680479bd1f6c95940f1d68

data/CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 
+## [0.4.37] - 2026-03-26
+
+### Fixed
+- Token tracking uses `model_id` from RubyLLM::Message response (was incorrectly checking non-existent `model` method)
+- Initialize TokenTracker with actual model from LLM chat session when available
+- Guard against blank `model_id` in `track_response_tokens` — a nil `model_id` no longer overwrites the previously-set model with an empty string
+
+### Added
+- Daemon path token tracking via `track_daemon_tokens` (reads `meta[:tokens_in]`/`meta[:tokens_out]` from daemon response)
+- Extracted `update_status_bar_tokens` helper to DRY up status bar updates after tracking
+
 ## [0.4.36] - 2026-03-26
 
 ### Changed

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

@@ -74,7 +74,10 @@ module Legion
           @message_stream = Components::MessageStream.new
           @status_bar = Components::StatusBar.new
           @llm_chat = app.respond_to?(:llm_chat) ? app.llm_chat : nil
-          @token_tracker = Components::TokenTracker.new(provider: detect_provider)
+          @token_tracker = Components::TokenTracker.new(
+            provider: detect_provider,
+            model: @llm_chat.respond_to?(:model) ? @llm_chat.model.to_s : nil
+          )
           @session_store = SessionStore.new
           @session_name = 'default'
           @plan_mode = false
@@ -251,6 +254,7 @@ module Legion
             parser = build_tool_call_parser
             parser.feed(result[:response])
             parser.flush
+            track_daemon_tokens(result)
           when :error
             err = result.dig(:error, :message) || 'Unknown error'
             @message_stream.append_streaming("\n[Daemon error: #{err}]")
@@ -602,12 +606,31 @@ module Legion
         def track_response_tokens(response)
           return unless response.respond_to?(:input_tokens)
 
-          model_id = response.respond_to?(:model) ? response.model.to_s : nil
+          raw_model = response.respond_to?(:model_id) ? response.model_id.to_s : nil
+          model_id = raw_model && !raw_model.empty? ? raw_model : nil
+          input_tokens = response.input_tokens.to_i
+          output_tokens = response.respond_to?(:output_tokens) ? response.output_tokens.to_i : 0
           @token_tracker.track(
-            input_tokens: response.input_tokens.to_i,
-            output_tokens: response.output_tokens.to_i,
+            input_tokens: input_tokens,
+            output_tokens: output_tokens,
             model: model_id
           )
+          update_status_bar_tokens
+        end
+
+        def track_daemon_tokens(result)
+          meta = result[:meta]
+          return unless meta.is_a?(Hash) && (meta[:tokens_in] || meta[:tokens_out])
+
+          @token_tracker.track(
+            input_tokens: meta[:tokens_in].to_i,
+            output_tokens: meta[:tokens_out].to_i,
+            model: meta[:model]&.to_s
+          )
+          update_status_bar_tokens
+        end
+
+        def update_status_bar_tokens
           @status_bar.update(
             tokens: @token_tracker.total_input_tokens + @token_tracker.total_output_tokens,
             cost: @token_tracker.total_cost

data/lib/legion/tty/version.rb

@@ -2,6 +2,6 @@
 
 module Legion
   module TTY
-    VERSION = '0.4.36'
+    VERSION = '0.4.37'
   end
 end

metadata

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