From 129e0d0779806742ce9ea59d03241cd1d2620671 Mon Sep 17 00:00:00 2001
From: Thorsten Sommer <mail@tsommer.org>
Date: Tue, 18 Feb 2025 16:06:36 +0100
Subject: [PATCH] Added an option to enable or disable the context validation

---
 ...PanelAgentRetrievalContextValidation.razor | 22 +++++++++++--------
 .../DataAgentRetrievalContextValidation.cs    |  5 +++++
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/app/MindWork AI Studio/Components/Settings/SettingsPanelAgentRetrievalContextValidation.razor b/app/MindWork AI Studio/Components/Settings/SettingsPanelAgentRetrievalContextValidation.razor
index bc8f78c3..c7196447 100644
--- a/app/MindWork AI Studio/Components/Settings/SettingsPanelAgentRetrievalContextValidation.razor	
+++ b/app/MindWork AI Studio/Components/Settings/SettingsPanelAgentRetrievalContextValidation.razor	
@@ -1,13 +1,17 @@
 @inherits SettingsPanelBase
 
 <ExpansionPanel HeaderIcon="@Icons.Material.Filled.Assessment" HeaderText="Agent: Retrieval Context Validation Options">
-    <MudPaper Class="pa-3 mb-8 border-dashed border rounded-lg">
-        <MudText Typo="Typo.body1" Class="mb-3">
-            Use Case: this agent is used to validate any retrieval context of the retrieval process. Perhaps there are many of these
-            retrieval contexts and you want to validate them all. Therefore, you might want to use a cheap and fast LLM for this
-            job. When using a local or self-hosted LLM, look for a small (e.g. 3B) and fast model.
-        </MudText>
-        <ConfigurationOption OptionDescription="Preselect retrieval context validation options?" LabelOn="Options are preselected" LabelOff="No options are preselected" State="@(() => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions)" StateUpdate="@(updatedState => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions = updatedState)" OptionHelp="When enabled, you can preselect some agent options. This is might be useful when you prefer a LLM."/>
-        <ConfigurationProviderSelection Data="@this.AvailableLLMProvidersFunc()" Disabled="@(() => !this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions)" SelectedValue="@(() => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectedAgentProvider)" SelectionUpdate="@(selectedValue => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectedAgentProvider = selectedValue)"/>
-    </MudPaper>
+    <MudText Typo="Typo.body1" Class="mb-3">
+        Use Case: this agent is used to validate any retrieval context of any retrieval process. Perhaps there are many of these
+        retrieval contexts and you want to validate them all. Therefore, you might want to use a cheap and fast LLM for this
+        job. When using a local or self-hosted LLM, look for a small (e.g. 3B) and fast model.
+    </MudText>
+    <ConfigurationOption OptionDescription="Enable the retrieval context validation agent?" LabelOn="The validation agent is enabled" LabelOff="No validation is performed" State="@(() => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.EnableRetrievalContextValidation)" StateUpdate="@(updatedState => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.EnableRetrievalContextValidation = updatedState)" OptionHelp="When enabled, the retrieval context validation agent will check each retrieval context of any retrieval process, whether a context makes sense for the given prompt."/>
+    @if (this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.EnableRetrievalContextValidation)
+    {
+        <MudPaper Class="pa-3 mb-8 border-dashed border rounded-lg">
+            <ConfigurationOption OptionDescription="Preselect retrieval context validation options?" LabelOn="Options are preselected" LabelOff="No options are preselected" State="@(() => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions)" StateUpdate="@(updatedState => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions = updatedState)" OptionHelp="When enabled, you can preselect some agent options. This is might be useful when you prefer a LLM."/>
+            <ConfigurationProviderSelection Data="@this.AvailableLLMProvidersFunc()" Disabled="@(() => !this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions)" SelectedValue="@(() => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectedAgentProvider)" SelectionUpdate="@(selectedValue => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectedAgentProvider = selectedValue)"/>
+        </MudPaper>
+    }
 </ExpansionPanel>
\ No newline at end of file
diff --git a/app/MindWork AI Studio/Settings/DataModel/DataAgentRetrievalContextValidation.cs b/app/MindWork AI Studio/Settings/DataModel/DataAgentRetrievalContextValidation.cs
index 5cf3b186..4e5a545d 100644
--- a/app/MindWork AI Studio/Settings/DataModel/DataAgentRetrievalContextValidation.cs	
+++ b/app/MindWork AI Studio/Settings/DataModel/DataAgentRetrievalContextValidation.cs	
@@ -2,6 +2,11 @@ namespace AIStudio.Settings.DataModel;
 
 public sealed class DataAgentRetrievalContextValidation
 {
+    /// <summary>
+    /// Enable the retrieval context validation agent?
+    /// </summary>
+    public bool EnableRetrievalContextValidation { get; set; }
+    
     /// <summary>
     /// Preselect any retrieval context validation options?
     /// </summary>