Allow assistants to run in the background (#828)

app/MindWork AI Studio/Assistants/Dynamic/AssistantDynamic.razor.cs

@@ -1,6 +1,7 @@
 using System.Text;
 using AIStudio.Dialogs.Settings;
 using AIStudio.Settings;
+using AIStudio.Tools.AssistantSessions;
 using AIStudio.Tools.PluginSystem;
 using AIStudio.Tools.PluginSystem.Assistants;
 using AIStudio.Tools.PluginSystem.Assistants.DataModel;
@@ -27,6 +28,11 @@ public partial class AssistantDynamic : AssistantBaseCore<NoSettingsPanel>
     // Reuse chat-level provider filtering/preselection instead of NONE.
     protected override Tools.Components Component => Tools.Components.CHAT;
 
+    /// <summary>
+    /// Gets the plugin ID as the assistant session instance ID.
+    /// </summary>
+    protected override string AssistantSessionInstanceId => this.assistantPlugin is null ? base.AssistantSessionInstanceId : this.assistantPlugin.Id.ToString();
+
     private string title = string.Empty;
     private string description = string.Empty;
     private string systemPrompt = string.Empty;
@@ -44,6 +50,61 @@ public partial class AssistantDynamic : AssistantBaseCore<NoSettingsPanel>
     private string securityMessage = string.Empty;
     private bool isSecurityBlocked;
     private const string ASSISTANT_QUERY_KEY = "assistantId";
+    private static readonly AssistantSessionStateKey<string> TITLE_STATE_KEY = new(nameof(title));
+    private static readonly AssistantSessionStateKey<string> DESCRIPTION_STATE_KEY = new(nameof(description));
+    private static readonly AssistantSessionStateKey<string> SYSTEM_PROMPT_STATE_KEY = new(nameof(systemPrompt));
+    private static readonly AssistantSessionStateKey<bool> ALLOW_PROFILES_STATE_KEY = new(nameof(allowProfiles));
+    private static readonly AssistantSessionStateKey<string> SUBMIT_TEXT_STATE_KEY = new(nameof(submitText));
+    private static readonly AssistantSessionStateKey<bool> SHOW_FOOTER_PROFILE_SELECTION_STATE_KEY = new(nameof(showFooterProfileSelection));
+    private static readonly AssistantSessionStateKey<PluginAssistants?> ASSISTANT_PLUGIN_STATE_KEY = new(nameof(assistantPlugin));
+    private static readonly AssistantSessionStateKey<AssistantState> ASSISTANT_STATE_STATE_KEY = new(nameof(assistantState));
+    private static readonly AssistantSessionStateKey<Dictionary<string, string>> IMAGE_CACHE_STATE_KEY = new(nameof(imageCache));
+    private static readonly AssistantSessionStateKey<HashSet<string>> EXECUTING_BUTTON_ACTIONS_STATE_KEY = new(nameof(executingButtonActions));
+    private static readonly AssistantSessionStateKey<HashSet<string>> EXECUTING_SWITCH_ACTIONS_STATE_KEY = new(nameof(executingSwitchActions));
+    private static readonly AssistantSessionStateKey<string> PLUGIN_PATH_STATE_KEY = new(nameof(pluginPath));
+    private static readonly AssistantSessionStateKey<PluginAssistantAudit?> AUDIT_STATE_KEY = new(nameof(audit));
+    private static readonly AssistantSessionStateKey<string> SECURITY_MESSAGE_STATE_KEY = new(nameof(securityMessage));
+    private static readonly AssistantSessionStateKey<bool> IS_SECURITY_BLOCKED_STATE_KEY = new(nameof(isSecurityBlocked));
+
+    /// <inheritdoc />
+    protected override void CaptureCustomAssistantSessionState(AssistantSessionStateWriter state)
+    {
+        state.Set(TITLE_STATE_KEY, this.title);
+        state.Set(DESCRIPTION_STATE_KEY, this.description);
+        state.Set(SYSTEM_PROMPT_STATE_KEY, this.systemPrompt);
+        state.Set(ALLOW_PROFILES_STATE_KEY, this.allowProfiles);
+        state.Set(SUBMIT_TEXT_STATE_KEY, this.submitText);
+        state.Set(SHOW_FOOTER_PROFILE_SELECTION_STATE_KEY, this.showFooterProfileSelection);
+        state.Set(ASSISTANT_PLUGIN_STATE_KEY, this.assistantPlugin);
+        state.Set(ASSISTANT_STATE_STATE_KEY, this.assistantState.Clone());
+        state.SetDictionary(IMAGE_CACHE_STATE_KEY, this.imageCache);
+        state.SetHashSet(EXECUTING_BUTTON_ACTIONS_STATE_KEY, this.executingButtonActions);
+        state.SetHashSet(EXECUTING_SWITCH_ACTIONS_STATE_KEY, this.executingSwitchActions);
+        state.Set(PLUGIN_PATH_STATE_KEY, this.pluginPath);
+        state.Set(AUDIT_STATE_KEY, this.audit);
+        state.Set(SECURITY_MESSAGE_STATE_KEY, this.securityMessage);
+        state.Set(IS_SECURITY_BLOCKED_STATE_KEY, this.isSecurityBlocked);
+    }
+
+    /// <inheritdoc />
+    protected override void RestoreCustomAssistantSessionState(AssistantSessionStateReader state)
+    {
+        state.Restore(TITLE_STATE_KEY, value => this.title = value);
+        state.Restore(DESCRIPTION_STATE_KEY, value => this.description = value);
+        state.Restore(SYSTEM_PROMPT_STATE_KEY, value => this.systemPrompt = value);
+        state.Restore(ALLOW_PROFILES_STATE_KEY, value => this.allowProfiles = value);
+        state.Restore(SUBMIT_TEXT_STATE_KEY, value => this.submitText = value);
+        state.Restore(SHOW_FOOTER_PROFILE_SELECTION_STATE_KEY, value => this.showFooterProfileSelection = value);
+        state.Restore(ASSISTANT_PLUGIN_STATE_KEY, value => this.assistantPlugin = value);
+        state.Restore(ASSISTANT_STATE_STATE_KEY, value => this.assistantState.CopyFrom(value));
+        state.RestoreDictionary(IMAGE_CACHE_STATE_KEY, this.imageCache);
+        state.RestoreHashSet(EXECUTING_BUTTON_ACTIONS_STATE_KEY, this.executingButtonActions);
+        state.RestoreHashSet(EXECUTING_SWITCH_ACTIONS_STATE_KEY, this.executingSwitchActions);
+        state.Restore(PLUGIN_PATH_STATE_KEY, value => this.pluginPath = value);
+        state.Restore(AUDIT_STATE_KEY, value => this.audit = value);
+        state.Restore(SECURITY_MESSAGE_STATE_KEY, value => this.securityMessage = value);
+        state.Restore(IS_SECURITY_BLOCKED_STATE_KEY, value => this.isSecurityBlocked = value);
+    }
 
     #region Implementation of AssistantBase