From 30918de6e46d6eef83571364a53b8072405d5b83 Mon Sep 17 00:00:00 2001
From: BeepBeepBopBop <116374330+BeepBeepBopBop@users.noreply.github.com>
Date: Sat, 28 Dec 2024 21:48:11 +0100
Subject: [PATCH] Update Chat.razor

---
 LM-Kit-Maestro/UI/Razor/Components/Chat.razor | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/LM-Kit-Maestro/UI/Razor/Components/Chat.razor b/LM-Kit-Maestro/UI/Razor/Components/Chat.razor
index 811cc43..4de7647 100644
--- a/LM-Kit-Maestro/UI/Razor/Components/Chat.razor
+++ b/LM-Kit-Maestro/UI/Razor/Components/Chat.razor
@@ -79,7 +79,7 @@
                 Tokens: @ViewModel.ConversationListViewModel.CurrentConversation.LMKitConversation.ContextUsedSpace /
                 @ViewModel.ConversationListViewModel.CurrentConversation.LMKitConversation.ContextSize
                 (@CalculateUsagePercentage(ViewModel.ConversationListViewModel.CurrentConversation.LMKitConversation.ContextUsedSpace,
-                                ViewModel.ConversationListViewModel.CurrentConversation.LMKitConversation.ContextSize)%)
+                                        ViewModel.ConversationListViewModel.CurrentConversation.LMKitConversation.ContextSize)%)
             </text>
         }
     </div>
@@ -128,7 +128,6 @@
         }
 
         ViewModel.ConversationListViewModel.PropertyChanged += OnConversationListViewModelPropertyChanged;
-
         ViewModel.ConversationListViewModel.CurrentConversation.LMKitConversation.PropertyChanged += OnLMKitConversationPropertyChanged;
 
         await ResizeHandler.RegisterPageResizeAsync(Resized);
@@ -260,6 +259,8 @@
         {
             Snackbar.Add($"Text generation failed unexpectedly:\n{e.Exception.Message}", Severity.Error);
         }
+
+        UpdateUIAsync();
     }
 
     private async void OnLatestAssistantResponseProgressed()