From b3471c500871538e0c42c6c7465f86f90a531338 Mon Sep 17 00:00:00 2001 From: cyq <15000851237@163.com> Date: Mon, 1 Jun 2026 19:41:35 +0800 Subject: [PATCH] fix(tui): use effective model window in context inspector --- crates/tui/src/tui/context_inspector.rs | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/crates/tui/src/tui/context_inspector.rs b/crates/tui/src/tui/context_inspector.rs index f141a7f13..52d4d9fb3 100644 --- a/crates/tui/src/tui/context_inspector.rs +++ b/crates/tui/src/tui/context_inspector.rs @@ -133,7 +133,8 @@ pub fn build_context_inspector_text(app: &App) -> String { } fn context_usage(app: &App) -> (usize, u32, f64) { - let max = context_window_for_model(&app.model).unwrap_or(LEGACY_DEEPSEEK_CONTEXT_WINDOW_TOKENS); + let max = context_window_for_model(app.effective_model_for_budget()) + .unwrap_or(LEGACY_DEEPSEEK_CONTEXT_WINDOW_TOKENS); let estimated = estimate_input_tokens_conservative(&app.api_messages, app.system_prompt.as_ref()); let total_chars = estimate_message_chars(&app.api_messages); @@ -495,6 +496,18 @@ mod tests { assert!(text.contains("Context: critical"), "{text}"); } + #[test] + fn inspector_uses_effective_auto_model_context_window() { + let mut app = test_app(); + app.model = "auto".to_string(); + app.auto_model = true; + app.last_effective_model = Some("deepseek-v4-pro".to_string()); + + let text = build_context_inspector_text(&app); + assert!(text.contains("Model: auto"), "{text}"); + assert!(text.contains("/1000000 tokens"), "{text}"); + } + #[test] fn inspector_no_system_prompt_shows_section() { let app = test_app();