Allow assistants to run in the background (#828)

app/MindWork AI Studio/Components/AssistantBlock.razor

@@ -7,7 +7,17 @@
         <MudCardHeader>
             <CardHeaderContent>
                 <MudStack AlignItems="AlignItems.Center" Row="@true">
-                    <MudIcon Icon="@this.Icon" Size="Size.Large" Color="Color.Primary"/>
+                    <MudElement HtmlTag="span" Style="position: relative; display: inline-flex; line-height: 1;">
+                        <MudIcon Icon="@this.Icon" Size="Size.Large" Color="Color.Primary"/>
+                        @if (this.AssistantSessionIndicator is { } indicator)
+                        {
+                            <MudTooltip Text="@indicator.Tooltip">
+                                <MudElement HtmlTag="span" Style="position: absolute; right: -0.45rem; bottom: -0.3rem; display: inline-flex; background-color: var(--mud-palette-surface); border-radius: 50%; padding: 1px;">
+                                    <MudIcon Icon="@indicator.Icon" Size="Size.Small" Color="@indicator.Color"/>
+                                </MudElement>
+                            </MudTooltip>
+                        }
+                    </MudElement>
                     <MudText Typo="Typo.h6">
                         @this.Name
                     </MudText>

app/MindWork AI Studio/Components/AssistantBlock.razor.cs

@@ -1,5 +1,6 @@
 using AIStudio.Dialogs.Settings;
 using AIStudio.Settings.DataModel;
+using AIStudio.Tools.AssistantSessions;
 using Microsoft.AspNetCore.Components;
 using DialogOptions = AIStudio.Dialogs.DialogOptions;
 
@@ -7,6 +8,14 @@ namespace AIStudio.Components;
 
 public partial class AssistantBlock<TSettings> : MSGComponentBase where TSettings : IComponent
 {
+    /// <summary>
+    /// Describes the assistant session indicator shown on top of the assistant icon.
+    /// </summary>
+    /// <param name="Icon">The icon that communicates the session status.</param>
+    /// <param name="Color">The color that communicates the session status.</param>
+    /// <param name="Tooltip">The tooltip text that explains the session status.</param>
+    private sealed record AssistantSessionIndicatorData(string Icon, Color Color, string Tooltip);
+
     [Parameter]
     public string Name { get; set; } = string.Empty;
 
@@ -31,6 +40,12 @@ public partial class AssistantBlock<TSettings> : MSGComponentBase where TSetting
     [Parameter]
     public Tools.Components Component { get; set; } = Tools.Components.NONE;
 
+    /// <summary>
+    /// Gets or sets the optional assistant session instance ID represented by this block.
+    /// </summary>
+    [Parameter]
+    public string AssistantSessionInstanceId { get; set; } = string.Empty;
+
     [Parameter]
     public PreviewFeatures RequiredPreviewFeature { get; set; } = PreviewFeatures.NONE;
 
@@ -39,6 +54,9 @@ public partial class AssistantBlock<TSettings> : MSGComponentBase where TSetting
 
     [Inject]
     private IDialogService DialogService { get; init; } = null!;
+
+    [Inject]
+    private AssistantSessionService AssistantSessionService { get; init; } = null!;
     
     private async Task OpenSettingsDialog()
     {
@@ -50,15 +68,50 @@ public partial class AssistantBlock<TSettings> : MSGComponentBase where TSetting
         await this.DialogService.ShowAsync<TSettings>(T("Open Settings"), dialogParameters, DialogOptions.FULLSCREEN);
     }
 
-    private string BorderColor => this.SettingsManager.IsDarkMode switch
+    private string BorderColor => this.AssistantSessionSnapshot?.IsActive is true ? this.ColorTheme.GetActivityIndicatorColor(this.SettingsManager) : this.SettingsManager.IsDarkMode switch
     {
-        true => this.ColorTheme.GetCurrentPalette(this.SettingsManager).GrayLight,
-        false => this.ColorTheme.GetCurrentPalette(this.SettingsManager).Primary.Value,
+        true => this.ColorTheme.GetCurrentPalette(this.SettingsManager).GrayDefault,
+        false => this.ColorTheme.GetCurrentPalette(this.SettingsManager).GrayDefault,
     };
 
-    private string BlockStyle => $"border-width: 2px; border-color: {this.BorderColor}; border-radius: 12px; border-style: solid; max-width: 20em;";
+    private string BlockStyle => $"border-width: 3px; border-color: {this.BorderColor}; border-radius: 12px; border-style: solid; max-width: 20em;";
 
     private bool IsVisible => this.SettingsManager.IsAssistantVisible(this.Component, assistantName: this.Name, requiredPreviewFeature: this.RequiredPreviewFeature);
 
     private bool HasSettingsPanel => typeof(TSettings) != typeof(NoSettingsPanel);
-}
+
+    /// <summary>
+    /// Gets the newest assistant session snapshot represented by this block.
+    /// </summary>
+    private AssistantSessionSnapshot? AssistantSessionSnapshot => string.IsNullOrWhiteSpace(this.AssistantSessionInstanceId)
+        ? this.AssistantSessionService.GetSnapshots().FirstOrDefault(snapshot => snapshot.Key.Component == this.Component)
+        : this.AssistantSessionService.GetSnapshots().FirstOrDefault(snapshot => snapshot.Key.InstanceId == this.AssistantSessionInstanceId);
+
+    /// <summary>
+    /// Gets the assistant session indicator shown on top of the assistant icon.
+    /// </summary>
+    private AssistantSessionIndicatorData? AssistantSessionIndicator => this.AssistantSessionSnapshot?.Status switch
+    {
+        AssistantSessionStatus.RUNNING or AssistantSessionStatus.CANCELING => new(Icons.Material.Filled.ChangeCircle, Color.Info, this.T("Assistant is still running.")),
+        AssistantSessionStatus.COMPLETED => new(Icons.Material.Filled.TaskAlt, Color.Success, this.T("The result is ready.")),
+        AssistantSessionStatus.FAILED => new(Icons.Material.Filled.Error, Color.Error, this.T("Assistant failed. Open it to review the result.")),
+        AssistantSessionStatus.CANCELED => new(Icons.Material.Filled.Cancel, Color.Warning, this.T("Assistant was canceled. Open it to review the result.")),
+        _ => null,
+    };
+
+    /// <summary>
+    /// Refreshes the block when assistant session activity changes.
+    /// </summary>
+    /// <typeparam name="T">The message payload type.</typeparam>
+    /// <param name="sendingComponent">The component that sent the message, if any.</param>
+    /// <param name="triggeredEvent">The event that was triggered.</param>
+    /// <param name="data">The message payload.</param>
+    /// <returns>A task that completes after the message was processed.</returns>
+    protected override Task ProcessIncomingMessage<T>(ComponentBase? sendingComponent, Event triggeredEvent, T? data) where T : default
+    {
+        if (triggeredEvent is Event.ASSISTANT_SESSION_CHANGED or Event.ASSISTANT_SESSION_FINISHED)
+            this.StateHasChanged();
+
+        return base.ProcessIncomingMessage(sendingComponent, triggeredEvent, data);
+    }
+}

app/MindWork AI Studio/Components/EnumSelection.razor

@@ -2,7 +2,7 @@
 @inherits EnumSelectionBase
 
 <MudStack Row="@true" Class="mb-3">
-    <MudSelect T="@T" Value="@this.Value" ValueChanged="@this.SelectionChanged" AdornmentIcon="@this.Icon" Adornment="Adornment.Start" Label="@this.Label" Variant="Variant.Outlined" Margin="Margin.Dense" Validation="@this.ValidateSelection">
+    <MudSelect T="@T" Value="@this.Value" ValueChanged="@this.SelectionChanged" AdornmentIcon="@this.Icon" Adornment="Adornment.Start" Label="@this.Label" Variant="Variant.Outlined" Margin="Margin.Dense" Validation="@this.ValidateSelection" Disabled="@this.Disabled">
         @foreach (var value in Enum.GetValues<T>())
         {
             <MudSelectItem Value="@value">
@@ -12,6 +12,6 @@
     </MudSelect>
     @if (this.AllowOther && this.Value.Equals(this.OtherValue))
     {
-        <MudTextField T="string" Text="@this.OtherInput" TextChanged="this.OtherValueChanged" Validation="@this.ValidateOther" Label="@this.LabelOther" Variant="Variant.Outlined" Margin="Margin.Dense" UserAttributes="@USER_INPUT_ATTRIBUTES" Immediate="@true"/>
+        <MudTextField T="string" Text="@this.OtherInput" TextChanged="this.OtherValueChanged" Validation="@this.ValidateOther" Label="@this.LabelOther" Variant="Variant.Outlined" Margin="Margin.Dense" UserAttributes="@USER_INPUT_ATTRIBUTES" Immediate="@true" Disabled="@this.Disabled"/>
     }
 </MudStack>

app/MindWork AI Studio/Components/EnumSelection.razor.cs

@@ -38,6 +38,12 @@ public partial class EnumSelection<T> : EnumSelectionBase where T : struct, Enum
     
     [Parameter]
     public string Icon { get; set; } = Icons.Material.Filled.ArrowDropDown;
+
+    /// <summary>
+    /// Gets or sets whether the selection controls are disabled.
+    /// </summary>
+    [Parameter]
+    public bool Disabled { get; set; }
     
     /// <summary>
     /// Gets or sets the custom name function for selecting the display name of an enum value.

app/MindWork AI Studio/Components/ProviderSelection.razor

@@ -1,6 +1,6 @@
 @using AIStudio.Settings
 @inherits MSGComponentBase
-<MudSelect T="Provider" Value="@this.ProviderSettings" ValueChanged="@this.SelectionChanged" Validation="@this.ValidateProvider" Adornment="Adornment.Start" AdornmentIcon="@Icons.Material.Filled.Apps" Margin="Margin.Dense" Label="@T("Provider")" Class="mb-3 rounded-lg" OuterClass="flex-grow-0" Variant="Variant.Outlined">
+<MudSelect T="Provider" Value="@this.ProviderSettings" ValueChanged="@this.SelectionChanged" Validation="@this.ValidateProvider" Adornment="Adornment.Start" AdornmentIcon="@Icons.Material.Filled.Apps" Margin="Margin.Dense" Label="@T("Provider")" Class="mb-3 rounded-lg" OuterClass="flex-grow-0" Variant="Variant.Outlined" Disabled="@this.Disabled">
     @foreach (var provider in this.GetAvailableProviders())
     {
         <MudSelectItem Value="@provider"/>

app/MindWork AI Studio/Components/ProviderSelection.razor.cs

@@ -20,6 +20,12 @@ public partial class ProviderSelection : MSGComponentBase
     [Parameter]
     public Func<AIStudio.Settings.Provider, string?> ValidateProvider { get; set; } = _ => null;
 
+    /// <summary>
+    /// Gets or sets whether provider selection is disabled.
+    /// </summary>
+    [Parameter]
+    public bool Disabled { get; set; }
+
     [Parameter]
     public ConfidenceLevel ExplicitMinimumConfidence { get; set; } = ConfidenceLevel.UNKNOWN;