MindWorkAI/AI-Studio
MindWorkAI/AI-Studio
1
0
Fork 0
mirror of https://github.com/MindWorkAI/AI-Studio.git synced 2026-08-24 20:52:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

adding an optional fallback provider to the audit agent

Browse Source
This commit is contained in:
Nils Kruthoff 2026-07-20 11:24:41 +02:00
parent 647d3ea4ef
commit fec7e78e34
No known key found for this signature in database
GPG Key ID: A5C0151B4DDB172C
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app/MindWork AI Studio/Agents/AssistantAudit/AssistantAuditAgent.cs
View File

@ -117,10 +117,14 @@ public sealed class AssistantAuditAgent(ILogger<AssistantAuditAgent> logger, ILo
/// <summary> /// <summary>
/// Resolves and stores the provider configuration used for assistant plugin audits. /// Resolves and stores the provider configuration used for assistant plugin audits.
/// </summary> /// </summary>
/// <param name="fallbackProvider">The provider to use when no provider is configured for the audit agent.</param>
/// <returns>The configured provider, or <see cref="AIStudio.Settings.Provider.NONE"/> when no audit provider is configured.</returns> /// <returns>The configured provider, or <see cref="AIStudio.Settings.Provider.NONE"/> when no audit provider is configured.</returns>
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); 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; this.ProviderSettings = provider;
return provider; return provider;
} }
@ -130,12 +134,13 @@ public sealed class AssistantAuditAgent(ILogger<AssistantAuditAgent> logger, ILo
/// </summary> /// </summary>
/// <param name="plugin">The assistant plugin to audit.</param> /// <param name="plugin">The assistant plugin to audit.</param>
/// <param name="token">A cancellation token for prompt generation and the audit request.</param> /// <param name="token">A cancellation token for prompt generation and the audit request.</param>
/// <param name="fallbackProvider">The provider to use when no provider is configured for the audit agent.</param>
/// <returns> /// <returns>
/// The parsed audit result, or an <c>UNKNOWN</c> result when no provider is configured or the model response cannot be used. /// The parsed audit result, or an <c>UNKNOWN</c> result when no provider is configured or the model response cannot be used.
/// </returns> /// </returns>
public async Task<AssistantAuditResult> AuditAsync(PluginAssistants plugin, CancellationToken token = default) public async Task<AssistantAuditResult> 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) 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.")))); await MessageBus.INSTANCE.SendError(new (Icons.Material.Filled.SettingsSuggest, string.Format(TB("No provider is configured for the Security Audit Agent."))));