diff --git a/app/MindWork AI Studio/Agents/AssistantAudit/AssistantAuditAgent.cs b/app/MindWork AI Studio/Agents/AssistantAudit/AssistantAuditAgent.cs index bc306978..e116a134 100644 --- a/app/MindWork AI Studio/Agents/AssistantAudit/AssistantAuditAgent.cs +++ b/app/MindWork AI Studio/Agents/AssistantAudit/AssistantAuditAgent.cs @@ -117,10 +117,14 @@ public sealed class AssistantAuditAgent(ILogger logger, ILo /// /// Resolves and stores the provider configuration used for assistant plugin audits. /// + /// The provider to use when no provider is configured for the audit agent. /// The configured provider, or when no audit provider is configured. - public AIStudio.Settings.Provider ResolveProvider() + public AIStudio.Settings.Provider ResolveProvider(AIStudio.Settings.Provider? fallbackProvider = null) { var provider = this.SettingsManager.GetPreselectedProvider(Tools.Components.AGENT_ASSISTANT_PLUGIN_AUDIT, null, true); + if (provider == AIStudio.Settings.Provider.NONE && fallbackProvider is not null) + provider = fallbackProvider; + this.ProviderSettings = provider; return provider; } @@ -130,12 +134,13 @@ public sealed class AssistantAuditAgent(ILogger logger, ILo /// /// The assistant plugin to audit. /// A cancellation token for prompt generation and the audit request. + /// The provider to use when no provider is configured for the audit agent. /// /// The parsed audit result, or an UNKNOWN result when no provider is configured or the model response cannot be used. /// - public async Task AuditAsync(PluginAssistants plugin, CancellationToken token = default) + public async Task AuditAsync(PluginAssistants plugin, CancellationToken token = default, AIStudio.Settings.Provider? fallbackProvider = null) { - var provider = this.ResolveProvider(); + var provider = this.ResolveProvider(fallbackProvider); if (provider == AIStudio.Settings.Provider.NONE) { await MessageBus.INSTANCE.SendError(new (Icons.Material.Filled.SettingsSuggest, string.Format(TB("No provider is configured for the Security Audit Agent."))));