提高会话加载性能 (#63)

Richasy · Jul 17, 2024 · 166589b · 166589b
1 parent 4822bad
commit 166589b
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/src/Desktop/RodelAgent.UI/ViewModels/Components/ChatSessionViewModel/ChatSessionViewModel.cs b/src/Desktop/RodelAgent.UI/ViewModels/Components/ChatSessionViewModel/ChatSessionViewModel.cs
@@ -69,10 +69,18 @@ private void Initialize(ChatSession data)
         CheckChatEmpty();
         CheckLastMessageTime();
         CheckRegenerateButtonShown();
-        CalcTotalTokenCount();
         RequestFocusInput?.Invoke(this, EventArgs.Empty);
     }
 
+    /// <summary>
+    /// 在进入视图开始显示时执行.
+    /// </summary>
+    [RelayCommand]
+    private void EnterView()
+    {
+        CalcTotalTokenCount();
+    }
+
     [RelayCommand]
     private void NewSession()
     {
@@ -158,6 +166,12 @@ private void CheckRegenerateButtonShown()
     [RelayCommand]
     private void CalcTotalTokenCount()
     {
+        var pageVM = GlobalDependencies.ServiceProvider.GetService<ChatServicePageViewModel>();
+        if (pageVM.CurrentSession != this)
+        {
+            return;
+        }
+
         CalcBaseTokenCount();
         CalcUserInputTokenCount();
     }

diff --git a/...sktop/RodelAgent.UI/ViewModels/Pages/ChatServicePageViewModel/ChatServicePageViewModel.cs b/...sktop/RodelAgent.UI/ViewModels/Pages/ChatServicePageViewModel/ChatServicePageViewModel.cs
@@ -132,6 +132,16 @@ partial void OnGroupPanelTypeChanged(ChatGroupPanelType value)
         CheckGroupPanelType();
     }
 
+    partial void OnCurrentSessionChanged(ChatSessionViewModel value)
+    {
+        if (value is null)
+        {
+            return;
+        }
+
+        value.EnterViewCommand.Execute(default);
+    }
+
     partial void OnIsServiceColumnManualHideChanged(bool value)
         => SettingsToolkit.WriteLocalSetting(SettingNames.IsChatServicePageServiceColumnManualHide, value);