Prepared release v0.9.30 (#298)

README.md

@@ -19,9 +19,9 @@ Things we are currently working on:
   - [ ] App: Implement the process to vectorize one local file using embeddings
   - [ ] Runtime: Integration of the vector database [LanceDB](https://github.com/lancedb/lancedb)
   - [ ] App: Implement the continuous process of vectorizing data
-  - [ ] App: Define a common retrieval context interface for the integration of RAG processes in chats
-  - [ ] App: Define a common augmentation interface for the integration of RAG processes in chats
-  - [ ] App: Integrate data sources in chats
+  - [x] ~~App: Define a common retrieval context interface for the integration of RAG processes in chats (PR [#281](https://github.com/MindWorkAI/AI-Studio/pull/281), [#284](https://github.com/MindWorkAI/AI-Studio/pull/284), [#286](https://github.com/MindWorkAI/AI-Studio/pull/286), [#287](https://github.com/MindWorkAI/AI-Studio/pull/287))~~
+  - [x] ~~App: Define a common augmentation interface for the integration of RAG processes in chats (PR [#288](https://github.com/MindWorkAI/AI-Studio/pull/288), [#289](https://github.com/MindWorkAI/AI-Studio/pull/289))~~
+  - [x] ~~App: Integrate data sources in chats (PR [#282](https://github.com/MindWorkAI/AI-Studio/pull/282))~~
 
 
 - Since September 2024: Experiments have been started on how we can work on long texts with AI Studio. Let's say you want to write a fantasy novel or create a complex project proposal and use LLM for support. The initial experiments were promising, but not yet satisfactory. We are testing further approaches until a satisfactory solution is found. The current state of our experiment is available as an experimental preview feature through your app configuration. Related PR: ~~[#167](https://github.com/MindWorkAI/AI-Studio/pull/167), [#226](https://github.com/MindWorkAI/AI-Studio/pull/226)~~.

app/MindWork AI Studio.sln

@@ -2,6 +2,8 @@
 Microsoft Visual Studio Solution File, Format Version 12.00
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "MindWork AI Studio", "MindWork AI Studio\MindWork AI Studio.csproj", "{059FDFCC-7D0B-474E-9F20-B9C437DF1CDD}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "SourceCodeRules", "SourceCodeRules\SourceCodeRules\SourceCodeRules.csproj", "{0976C1CB-D499-4C86-8ADA-B7A7A4DE0BF8}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -12,6 +14,10 @@ Global
 		{059FDFCC-7D0B-474E-9F20-B9C437DF1CDD}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{059FDFCC-7D0B-474E-9F20-B9C437DF1CDD}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{059FDFCC-7D0B-474E-9F20-B9C437DF1CDD}.Release|Any CPU.Build.0 = Release|Any CPU
+		{0976C1CB-D499-4C86-8ADA-B7A7A4DE0BF8}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{0976C1CB-D499-4C86-8ADA-B7A7A4DE0BF8}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{0976C1CB-D499-4C86-8ADA-B7A7A4DE0BF8}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{0976C1CB-D499-4C86-8ADA-B7A7A4DE0BF8}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 	GlobalSection(NestedProjects) = preSolution
 	EndGlobalSection

app/MindWork AI Studio.sln.DotSettings

@@ -5,6 +5,10 @@
 	<s:String x:Key="/Default/CodeStyle/Naming/CSharpNaming/Abbreviations/=LLM/@EntryIndexedValue">LLM</s:String>
 	<s:String x:Key="/Default/CodeStyle/Naming/CSharpNaming/Abbreviations/=LM/@EntryIndexedValue">LM</s:String>
 	<s:String x:Key="/Default/CodeStyle/Naming/CSharpNaming/Abbreviations/=MSG/@EntryIndexedValue">MSG</s:String>
+	<s:String x:Key="/Default/CodeStyle/Naming/CSharpNaming/Abbreviations/=RAG/@EntryIndexedValue">RAG</s:String>
+	<s:String x:Key="/Default/CodeStyle/Naming/CSharpNaming/Abbreviations/=UI/@EntryIndexedValue">UI</s:String>
+	<s:Boolean x:Key="/Default/UserDictionary/Words/=agentic/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=groq/@EntryIndexedValue">True</s:Boolean>
+	<s:Boolean x:Key="/Default/UserDictionary/Words/=mwais/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=ollama/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/UserDictionary/Words/=tauri_0027s/@EntryIndexedValue">True</s:Boolean></wpf:ResourceDictionary>

app/MindWork AI Studio/Agents/AgentBase.cs

@@ -1,19 +1,37 @@
+using System.Text.Json;
+
 using AIStudio.Chat;
 using AIStudio.Provider;
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 // ReSharper disable MemberCanBePrivate.Global
 
 namespace AIStudio.Agents;
 
-public abstract class AgentBase(ILogger<AgentBase> logger, SettingsManager settingsManager, ThreadSafeRandom rng) : IAgent
+public abstract class AgentBase(ILogger<AgentBase> logger, SettingsManager settingsManager, DataSourceService dataSourceService, ThreadSafeRandom rng) : IAgent
 {
+    protected static readonly ContentBlock EMPTY_BLOCK = new()
+    {
+        Content = null,
+        ContentType = ContentType.NONE,
+        Role = ChatRole.AGENT,
+        Time = DateTimeOffset.UtcNow,
+    };
+    
+    protected static readonly JsonSerializerOptions JSON_SERIALIZER_OPTIONS = new()
+    {
+        PropertyNamingPolicy = JsonNamingPolicy.SnakeCaseLower,
+    };
+    
+    protected DataSourceService DataSourceService { get; init; } = dataSourceService;
+    
     protected SettingsManager SettingsManager { get; init; } = settingsManager;
 
     protected ThreadSafeRandom RNG { get; init; } = rng;
     
     protected ILogger<AgentBase> Logger { get; init; } = logger;
-
+    
     /// <summary>
     /// Represents the type or category of this agent.
     /// </summary>
@@ -60,24 +78,30 @@ public abstract class AgentBase(ILogger<AgentBase> logger, SettingsManager setti
         Blocks = [],
     };
 
-    protected DateTimeOffset AddUserRequest(ChatThread thread, string request)
+    protected UserRequest AddUserRequest(ChatThread thread, string request)
     {
         var time = DateTimeOffset.Now;
+        var lastUserPrompt = new ContentText
+        {
+            Text = request,
+        };
+        
         thread.Blocks.Add(new ContentBlock
         {
             Time = time,
             ContentType = ContentType.TEXT,
             Role = ChatRole.USER,
-            Content = new ContentText
-            {
-                Text = request,
-            },
+            Content = lastUserPrompt,
         });
-        
-        return time;
+
+        return new()
+        {
+            Time = time,
+            UserPrompt = lastUserPrompt,
+        };
     }
     
-    protected async Task AddAIResponseAsync(ChatThread thread, DateTimeOffset time)
+    protected async Task AddAIResponseAsync(ChatThread thread, IContent lastUserPrompt, DateTimeOffset time)
     {
         if(this.ProviderSettings is null)
             return;
@@ -103,6 +127,6 @@ public abstract class AgentBase(ILogger<AgentBase> logger, SettingsManager setti
         // Use the selected provider to get the AI response.
         // By awaiting this line, we wait for the entire
         // content to be streamed.
-        await aiText.CreateFromProviderAsync(providerSettings.CreateProvider(this.Logger), this.SettingsManager, providerSettings.Model, thread);
+        await aiText.CreateFromProviderAsync(providerSettings.CreateProvider(this.Logger), providerSettings.Model, lastUserPrompt, thread);
     }
 }

app/MindWork AI Studio/Agents/AgentDataSourceSelection.cs

@@ -0,0 +1,398 @@
+using System.Text;
+using System.Text.Json;
+
+using AIStudio.Chat;
+using AIStudio.Provider;
+using AIStudio.Settings;
+using AIStudio.Settings.DataModel;
+using AIStudio.Tools.ERIClient;
+using AIStudio.Tools.Services;
+
+namespace AIStudio.Agents;
+
+public sealed class AgentDataSourceSelection (ILogger<AgentDataSourceSelection> logger, ILogger<AgentBase> baseLogger, SettingsManager settingsManager, DataSourceService dataSourceService, ThreadSafeRandom rng) : AgentBase(baseLogger, settingsManager, dataSourceService, rng)
+{
+    private readonly List<ContentBlock> answers = new();
+    
+    #region Overrides of AgentBase
+
+    /// <inheritdoc />
+    protected override Type Type => Type.SYSTEM;
+    
+    /// <inheritdoc />
+    public override string Id => "Data Source Selection";
+
+    /// <inheritdoc />
+    protected override string JobDescription =>
+        """
+        You receive a system and a user prompt, as well as a list of possible data sources as input.
+        Your task is to select the appropriate data sources for the given task. You may choose none,
+        one, or multiple sources, depending on what best fits the system and user prompt. You need
+        to estimate and assess which source, based on its description, might be helpful in
+        processing the prompts.
+        
+        Your response is a JSON list in the following format:
+        
+        ```
+        [
+          {"id": "The data source ID", "reason": "Why did you choose this source?", "confidence": 0.87},
+          {"id": "The data source ID", "reason": "Why did you choose this source?", "confidence": 0.54}
+        ]
+        ```
+        
+        You express your confidence as a floating-point number between 0.0 (maximum uncertainty) and
+        1.0 (you are absolutely certain that this source is needed).
+        
+        The JSON schema is:
+        
+        ```
+        {
+          "$schema": "http://json-schema.org/draft-04/schema#",
+          "type": "array", 
+          "items": [
+            {
+              "type": "object",
+              "properties": {
+                "id": {
+                  "type": "string"
+                },
+                "reason": {
+                  "type": "string"
+                },
+                "confidence": {
+                  "type": "number"
+                }
+              },
+              "required": [
+                "id",
+                "reason",
+                "confidence"
+              ]
+            }
+          ]
+        }
+        ```
+        
+        When no data source is needed, you return an empty JSON list `[]`. You do not ask any
+        follow-up questions. You do not address the user. Your response consists solely of
+        the JSON list.
+        """;
+    
+    /// <inheritdoc />
+    protected override string SystemPrompt(string availableDataSources) => $"""
+                                                                      {this.JobDescription}
+                                                                      
+                                                                      {availableDataSources}
+                                                                      """;
+
+    /// <inheritdoc />
+    public override Settings.Provider? ProviderSettings { get; set; }
+    
+    /// <summary>
+    /// The data source selection agent does not work with context. Use
+    /// the process input method instead.
+    /// </summary>
+    /// <returns>The chat thread without any changes.</returns>
+    public override Task<ChatThread> ProcessContext(ChatThread chatThread, IDictionary<string, string> additionalData) => Task.FromResult(chatThread);
+
+    /// <inheritdoc />
+    public override async Task<ContentBlock> ProcessInput(ContentBlock input, IDictionary<string, string> additionalData)
+    {
+        if (input.Content is not ContentText text)
+            return EMPTY_BLOCK;
+        
+        if(text.InitialRemoteWait || text.IsStreaming)
+            return EMPTY_BLOCK;
+        
+        if(string.IsNullOrWhiteSpace(text.Text))
+            return EMPTY_BLOCK;
+        
+        if(!additionalData.TryGetValue("availableDataSources", out var availableDataSources) || string.IsNullOrWhiteSpace(availableDataSources))
+            return EMPTY_BLOCK;
+        
+        var thread = this.CreateChatThread(this.SystemPrompt(availableDataSources));
+        var userRequest = this.AddUserRequest(thread, text.Text);
+        await this.AddAIResponseAsync(thread, userRequest.UserPrompt, userRequest.Time);
+        
+        var answer = thread.Blocks[^1];
+        
+        this.answers.Add(answer);
+        return answer;
+    }
+
+    // <inheritdoc />
+    public override Task<bool> MadeDecision(ContentBlock input) => Task.FromResult(true);
+
+    // <inheritdoc />
+    public override IReadOnlyCollection<ContentBlock> GetContext() => [];
+
+    // <inheritdoc />
+    public override IReadOnlyCollection<ContentBlock> GetAnswers() => this.answers;
+
+    #endregion
+
+    public async Task<List<SelectedDataSource>> PerformSelectionAsync(IProvider provider, IContent lastPrompt, ChatThread chatThread, AllowedSelectedDataSources dataSources, CancellationToken token = default)
+    {
+        logger.LogInformation("The AI should select the appropriate data sources.");
+
+        //
+        // 1. Which LLM provider should the agent use?
+        //
+
+        // We start with the provider currently selected by the user:
+        var agentProvider = this.SettingsManager.GetPreselectedProvider(Tools.Components.AGENT_DATA_SOURCE_SELECTION, provider.Id, true);
+
+        // Assign the provider settings to the agent:
+        logger.LogInformation($"The agent for the data source selection uses the provider '{agentProvider.InstanceName}' ({agentProvider.UsedLLMProvider.ToName()}, confidence={agentProvider.UsedLLMProvider.GetConfidence(this.SettingsManager).Level.GetName()}).");
+        this.ProviderSettings = agentProvider;
+
+        //
+        // 2. Prepare the current system and user prompts as input for the agent: 
+        //
+        var lastPromptContent = lastPrompt switch
+        {
+            ContentText text => text.Text,
+
+            // Image prompts may be empty, e.g., when the image is too large:
+            ContentImage image => await image.AsBase64(token),
+
+            // Other content types are not supported yet:
+            _ => string.Empty,
+        };
+
+        if (string.IsNullOrWhiteSpace(lastPromptContent))
+        {
+            logger.LogWarning("The last prompt is empty. The AI cannot select data sources.");
+            return [];
+        }
+
+        //
+        // 3. Prepare the allowed data sources as input for the agent:
+        //
+        var additionalData = new Dictionary<string, string>();
+        logger.LogInformation("Preparing the list of allowed data sources for the agent to choose from.");
+        
+        // Notice: We do not dispose the Rust service here. The Rust service is a singleton
+        // and will be disposed when the application shuts down:
+        var rustService = Program.SERVICE_PROVIDER.GetService<RustService>()!;
+        
+        var sb = new StringBuilder();
+        sb.AppendLine("The following data sources are available for selection:");
+        foreach (var ds in dataSources.AllowedDataSources)
+        {
+            switch (ds)
+            {
+                case DataSourceLocalDirectory localDirectory:
+                    sb.AppendLine($"- Id={ds.Id}, name='{localDirectory.Name}', type=local directory, path='{localDirectory.Path}'");
+                    break;
+
+                case DataSourceLocalFile localFile:
+                    sb.AppendLine($"- Id={ds.Id}, name='{localFile.Name}', type=local file, path='{localFile.FilePath}'");
+                    break;
+
+                case IERIDataSource eriDataSource:
+                    var eriServerDescription = string.Empty;
+
+                    try
+                    {
+                        //
+                        // Call the ERI server to get the server description:
+                        //
+                        using var eriClient = ERIClientFactory.Get(eriDataSource.Version, eriDataSource)!;
+                        var authResponse = await eriClient.AuthenticateAsync(eriDataSource, rustService, token);
+                        if (authResponse.Successful)
+                        {
+                            var serverDescriptionResponse = await eriClient.GetDataSourceInfoAsync(token);
+                            if (serverDescriptionResponse.Successful)
+                            {
+                                eriServerDescription = serverDescriptionResponse.Data.Description;
+
+                                // Remove all line breaks from the description:
+                                eriServerDescription = eriServerDescription.Replace("\n", " ").Replace("\r", " ");
+                            }
+                            else
+                                logger.LogWarning($"Was not able to retrieve the server description from the ERI data source '{eriDataSource.Name}'. Message: {serverDescriptionResponse.Message}");
+                        }
+                        else
+                            logger.LogWarning($"Was not able to authenticate with the ERI data source '{eriDataSource.Name}'. Message: {authResponse.Message}");
+                    }
+                    catch (Exception e)
+                    {
+                        logger.LogWarning($"The ERI data source '{eriDataSource.Name}' is not available. Thus, we cannot retrieve the server description. Error: {e.Message}");
+                    }
+
+                    //
+                    // Append the ERI data source to the list. Use the server description if available:
+                    //
+                    if (string.IsNullOrWhiteSpace(eriServerDescription))
+                        sb.AppendLine($"- Id={ds.Id}, name='{eriDataSource.Name}', type=external data source");
+                    else
+                        sb.AppendLine($"- Id={ds.Id}, name='{eriDataSource.Name}', type=external data source, description='{eriServerDescription}'");
+
+                    break;
+            }
+        }
+
+        logger.LogInformation("Prepared the list of allowed data sources for the agent.");
+        additionalData.Add("availableDataSources", sb.ToString());
+
+        //
+        // 4. Let the agent select the data sources:
+        //
+        var prompt = $"""
+                      The system prompt is:
+
+                      ```
+                      {chatThread.SystemPrompt}
+                      ```
+
+                      The user prompt is:
+
+                      ```
+                      {lastPromptContent}
+                      ```
+                      """;
+
+        // Call the agent:
+        var aiResponse = await this.ProcessInput(new ContentBlock
+        {
+            Time = DateTimeOffset.UtcNow,
+            ContentType = ContentType.TEXT,
+            Role = ChatRole.USER,
+            Content = new ContentText
+            {
+                Text = prompt,
+            },
+        }, additionalData);
+
+        if(aiResponse.Content is null)
+        {
+            logger.LogWarning("The agent did not return a response.");
+            return [];
+        }
+
+        switch (aiResponse)
+        {
+            
+            //
+            // 5. Parse the agent response:
+            //
+            case { ContentType: ContentType.TEXT, Content: ContentText textContent }:
+            {
+                //
+                // What we expect is a JSON list of SelectedDataSource objects:
+                //
+                var selectedDataSourcesJson = textContent.Text;
+                
+                //
+                // We know how bad LLM may be in generating JSON without surrounding text.
+                // Thus, we expect the worst and try to extract the JSON list from the text:
+                //
+                var json = ExtractJson(selectedDataSourcesJson);
+                
+                try
+                {
+                    var aiSelectedDataSources = JsonSerializer.Deserialize<List<SelectedDataSource>>(json, JSON_SERIALIZER_OPTIONS);
+                    return aiSelectedDataSources ?? [];
+                }
+                catch
+                {
+                    logger.LogWarning("The agent answered with an invalid or unexpected JSON format.");
+                    return [];
+                }
+            }
+            
+            case { ContentType: ContentType.TEXT }:
+                logger.LogWarning("The agent answered with an unexpected inner content type.");
+                return [];
+            
+            case { ContentType: ContentType.NONE }:
+                logger.LogWarning("The agent did not return a response.");
+                return [];
+            
+            default:
+                logger.LogWarning($"The agent answered with an unexpected content type '{aiResponse.ContentType}'.");
+                return [];
+        }
+    }
+    
+    /// <summary>
+    /// Extracts the JSON list from the given text. The text may contain additional
+    /// information around the JSON list. The method tries to extract the JSON list
+    /// from the text.
+    /// </summary>
+    /// <remarks>
+    /// Algorithm: The method searches for the first line that contains only a '[' character.
+    /// Then, it searches for the first line that contains only a ']' character. The method
+    /// returns the text between these two lines (including the brackets). When the method
+    /// cannot find the JSON list, it returns an empty string.
+    /// <br/><br/>
+    /// This overload is using strings instead of spans. We can use this overload in any
+    /// async method. Thus, it is a wrapper around the span-based method. Yes, we are losing
+    /// the memory efficiency of the span-based method, but we still gain the performance
+    /// of the span-based method: the entire search algorithm is span-based.
+    /// </remarks>
+    /// <param name="text">The text that may contain the JSON list.</param>
+    /// <returns>The extracted JSON list.</returns>
+    private static string ExtractJson(string text) => ExtractJson(text.AsSpan()).ToString();
+
+    /// <summary>
+    /// Extracts the JSON list from the given text. The text may contain additional
+    /// information around the JSON list. The method tries to extract the JSON list
+    /// from the text.
+    /// </summary>
+    /// <remarks>
+    /// Algorithm: The method searches for the first line that contains only a '[' character.
+    /// Then, it searches for the first line that contains only a ']' character. The method
+    /// returns the text between these two lines (including the brackets). When the method
+    /// cannot find the JSON list, it returns an empty string.
+    /// </remarks>
+    /// <param name="text">The text that may contain the JSON list.</param>
+    /// <returns>The extracted JSON list.</returns>
+    private static ReadOnlySpan<char> ExtractJson(ReadOnlySpan<char> text)
+    {
+        var startIndex = -1;
+        var endIndex = -1;
+        var foundStart = false;
+        var foundEnd = false;
+        var lineStart = 0;
+        
+        for (var i = 0; i <= text.Length; i++)
+        {
+            // Handle the end of the line or the end of the text:
+            if (i == text.Length || text[i] == '\n')
+            {
+                if (IsCharacterAloneInLine(text, lineStart, i, '[') && !foundStart)
+                {
+                    startIndex = lineStart;
+                    foundStart = true;
+                }
+                else if (IsCharacterAloneInLine(text, lineStart, i, ']') && foundStart && !foundEnd)
+                {
+                    endIndex = i;
+                    foundEnd = true;
+                    break;
+                }
+                
+                lineStart = i + 1;
+            }
+        }
+        
+        if (foundStart && foundEnd)
+        {
+            // Adjust endIndex for slicing, ensuring it's within bounds:
+            return text.Slice(startIndex, Math.Min(text.Length, endIndex + 1) - startIndex);
+        }
+        
+        return ReadOnlySpan<char>.Empty;
+    }
+    
+    private static bool IsCharacterAloneInLine(ReadOnlySpan<char> text, int lineStart, int lineEnd, char character)
+    {
+        for (var i = lineStart; i < lineEnd; i++)
+            if (!char.IsWhiteSpace(text[i]) && text[i] != character)
+                return false;
+
+        return true;
+    }
+}

app/MindWork AI Studio/Agents/AgentRetrievalContextValidation.cs

@@ -0,0 +1,388 @@
+using System.Text.Json;
+
+using AIStudio.Chat;
+using AIStudio.Provider;
+using AIStudio.Settings;
+using AIStudio.Tools.RAG;
+using AIStudio.Tools.Services;
+
+namespace AIStudio.Agents;
+
+public sealed class AgentRetrievalContextValidation (ILogger<AgentRetrievalContextValidation> logger, ILogger<AgentBase> baseLogger, SettingsManager settingsManager, DataSourceService dataSourceService, ThreadSafeRandom rng) : AgentBase(baseLogger, settingsManager, dataSourceService, rng)
+{
+    #region Overrides of AgentBase
+
+    /// <inheritdoc />
+    protected override Type Type => Type.WORKER;
+    
+    /// <inheritdoc />
+    public override string Id => "Retrieval Context Validation";
+
+    /// <inheritdoc />
+    protected override string JobDescription =>
+        """
+        You receive a system and user prompt as well as a retrieval context as input. Your task is to decide whether this
+        retrieval context is helpful in processing the prompts or not. You respond with the decision (true or false),
+        your reasoning, and your confidence in this decision.
+
+        Your response is only one JSON object in the following format:
+
+        ```
+        {"decision": true, "reason": "Why did you choose this source?", "confidence": 0.87}
+        ```
+
+        You express your confidence as a floating-point number between 0.0 (maximum uncertainty) and
+        1.0 (you are absolutely certain that this retrieval context is needed).
+
+        The JSON schema is:
+
+        ```
+        {
+          "$schema": "http://json-schema.org/draft-04/schema#",
+          "type": "object",
+          "properties": {
+            "decision": {
+              "type": "boolean"
+            },
+            "reason": {
+              "type": "string"
+            },
+            "confidence": {
+              "type": "number"
+            }
+          },
+          "required": [
+            "decision",
+            "reason",
+            "confidence"
+          ]
+        }
+        ```
+
+        You do not ask any follow-up questions. You do not address the user. Your response consists solely of
+        that one JSON object.
+        """;
+    
+    /// <inheritdoc />
+    protected override string SystemPrompt(string retrievalContext) => $"""
+                                                                        {this.JobDescription}
+
+                                                                        {retrievalContext}
+                                                                        """;
+
+    /// <inheritdoc />
+    public override Settings.Provider? ProviderSettings { get; set; }
+    
+    /// <summary>
+    /// The retrieval context validation agent does not work with context. Use
+    /// the process input method instead.
+    /// </summary>
+    /// <returns>The chat thread without any changes.</returns>
+    public override Task<ChatThread> ProcessContext(ChatThread chatThread, IDictionary<string, string> additionalData) => Task.FromResult(chatThread);
+
+    /// <inheritdoc />
+    public override async Task<ContentBlock> ProcessInput(ContentBlock input, IDictionary<string, string> additionalData)
+    {
+        if (input.Content is not ContentText text)
+            return EMPTY_BLOCK;
+          
+        if(text.InitialRemoteWait || text.IsStreaming)
+            return EMPTY_BLOCK;
+          
+        if(string.IsNullOrWhiteSpace(text.Text))
+            return EMPTY_BLOCK;
+          
+        if(!additionalData.TryGetValue("retrievalContext", out var retrievalContext) || string.IsNullOrWhiteSpace(retrievalContext))
+            return EMPTY_BLOCK;
+          
+        var thread = this.CreateChatThread(this.SystemPrompt(retrievalContext));
+        var userRequest = this.AddUserRequest(thread, text.Text);
+        await this.AddAIResponseAsync(thread, userRequest.UserPrompt, userRequest.Time);
+          
+        return thread.Blocks[^1];
+    }
+
+    /// <inheritdoc />
+    public override Task<bool> MadeDecision(ContentBlock input) => Task.FromResult(true);
+
+    /// <summary>
+    /// We do not provide any context. This agent will process many retrieval contexts.
+    /// This would block a huge amount of memory.
+    /// </summary>
+    /// <returns>An empty list.</returns>
+    public override IReadOnlyCollection<ContentBlock> GetContext() => [];
+
+    /// <summary>
+    /// We do not provide any answers. This agent will process many retrieval contexts.
+    /// This would block a huge amount of memory.
+    /// </summary>
+    /// <returns>An empty list.</returns>
+    public override IReadOnlyCollection<ContentBlock> GetAnswers() => [];
+
+    #endregion
+
+    /// <summary>
+    /// Sets the LLM provider for the agent.
+    /// </summary>
+    /// <remarks>
+    /// When you have to call the validation in parallel for many retrieval contexts,
+    /// you can set the provider once and then call the validation method in parallel.
+    /// </remarks>
+    /// <param name="provider">The current LLM provider. When the user doesn't preselect an agent provider, the agent uses this provider.</param>
+    public void SetLLMProvider(IProvider provider)
+    {
+        // We start with the provider currently selected by the user:
+        var agentProvider = this.SettingsManager.GetPreselectedProvider(Tools.Components.AGENT_RETRIEVAL_CONTEXT_VALIDATION, provider.Id, true);
+        
+        // Assign the provider settings to the agent:
+        logger.LogInformation($"The agent for the retrieval context validation uses the provider '{agentProvider.InstanceName}' ({agentProvider.UsedLLMProvider.ToName()}, confidence={agentProvider.UsedLLMProvider.GetConfidence(this.SettingsManager).Level.GetName()}).");
+        this.ProviderSettings = agentProvider;
+    }
+    
+    /// <summary>
+    /// Validate all retrieval contexts against the last user and the system prompt.
+    /// </summary>
+    /// <param name="lastPrompt">The last user prompt.</param>
+    /// <param name="chatThread">The chat thread.</param>
+    /// <param name="retrievalContexts">All retrieval contexts to validate.</param>
+    /// <param name="token">The cancellation token.</param>
+    /// <returns>The validation results.</returns>
+    public async Task<IReadOnlyList<RetrievalContextValidationResult>> ValidateRetrievalContextsAsync(IContent lastPrompt, ChatThread chatThread, IReadOnlyList<IRetrievalContext> retrievalContexts, CancellationToken token = default)
+    {
+        // Check if the retrieval context validation is enabled:
+        if (!this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.EnableRetrievalContextValidation)
+            return [];
+        
+        logger.LogInformation($"Validating {retrievalContexts.Count:###,###,###,###} retrieval contexts.");
+        
+        // Prepare the list of validation tasks:
+        var validationTasks = new List<Task<RetrievalContextValidationResult>>(retrievalContexts.Count);
+        
+        // Read the number of parallel validations:
+        var numParallelValidations = 3;
+        if(this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions)
+            numParallelValidations = this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.NumParallelValidations;
+        
+        numParallelValidations = Math.Max(1, numParallelValidations);
+        
+        // Use a semaphore to limit the number of parallel validations:
+        using var semaphore = new SemaphoreSlim(numParallelValidations);
+        foreach (var retrievalContext in retrievalContexts)
+        {
+            // Wait for an available slot in the semaphore:
+            await semaphore.WaitAsync(token);
+            
+            // Start the next validation task:
+            validationTasks.Add(this.ValidateRetrievalContextAsync(lastPrompt, chatThread, retrievalContext, token, semaphore));
+        }
+
+        // Wait for all validation tasks to complete:
+        return await Task.WhenAll(validationTasks);
+    }
+
+    /// <summary>
+    /// Validates the retrieval context against the last user and the system prompt.
+    /// </summary>
+    /// <remarks>
+    /// Probably, you have a lot of retrieval contexts to validate. In this case, you
+    /// can call this method in parallel for each retrieval context. You might use
+    /// the ValidateRetrievalContextsAsync method to validate all retrieval contexts.
+    /// </remarks>
+    /// <param name="lastPrompt">The last user prompt.</param>
+    /// <param name="chatThread">The chat thread.</param>
+    /// <param name="retrievalContext">The retrieval context to validate.</param>
+    /// <param name="token">The cancellation token.</param>
+    /// <param name="semaphore">The optional semaphore to limit the number of parallel validations.</param>
+    /// <returns>The validation result.</returns>
+    public async Task<RetrievalContextValidationResult> ValidateRetrievalContextAsync(IContent lastPrompt, ChatThread chatThread, IRetrievalContext retrievalContext, CancellationToken token = default, SemaphoreSlim? semaphore = null)
+    {
+        try
+        {
+            //
+            // Check if the validation was canceled. This could happen when the user
+            // canceled the validation process or when the validation process took
+            // too long:
+            //
+            if(token.IsCancellationRequested)
+                return new(false, "The validation was canceled.", 1.0f, retrievalContext);
+            
+            //
+            // 1. Prepare the current system and user prompts as input for the agent: 
+            //
+            var lastPromptContent = lastPrompt switch
+            {
+                ContentText text => text.Text,
+
+                // Image prompts may be empty, e.g., when the image is too large:
+                ContentImage image => await image.AsBase64(token),
+
+                // Other content types are not supported yet:
+                _ => string.Empty,
+            };
+
+            if (string.IsNullOrWhiteSpace(lastPromptContent))
+            {
+                logger.LogWarning("The last prompt is empty. The AI cannot validate the retrieval context.");
+                return new(false, "The last prompt was empty.", 1.0f, retrievalContext);
+            }
+
+            //
+            // 2. Prepare the retrieval context for the agent:
+            //
+            var additionalData = new Dictionary<string, string>();
+            var markdownRetrievalContext = await retrievalContext.AsMarkdown(token: token);
+            additionalData.Add("retrievalContext", markdownRetrievalContext);
+
+            //
+            // 3. Let the agent validate the retrieval context:
+            //
+            var prompt = $"""
+                          The system prompt is:
+
+                          ```
+                          {chatThread.SystemPrompt}
+                          ```
+
+                          The user prompt is:
+
+                          ```
+                          {lastPromptContent}
+                          ```
+                          """;
+
+            // Call the agent:
+            var aiResponse = await this.ProcessInput(new ContentBlock
+            {
+                Time = DateTimeOffset.UtcNow,
+                ContentType = ContentType.TEXT,
+                Role = ChatRole.USER,
+                Content = new ContentText
+                {
+                    Text = prompt,
+                },
+            }, additionalData);
+
+            if (aiResponse.Content is null)
+            {
+                logger.LogWarning("The agent did not return a response.");
+                return new(false, "The agent did not return a response.", 1.0f, retrievalContext);
+            }
+
+            switch (aiResponse)
+            {
+
+                //
+                // 4. Parse the agent response:
+                //
+                case { ContentType: ContentType.TEXT, Content: ContentText textContent }:
+                {
+                    //
+                    // What we expect is one JSON object:
+                    //
+                    var validationJson = textContent.Text;
+
+                    //
+                    // We know how bad LLM may be in generating JSON without surrounding text.
+                    // Thus, we expect the worst and try to extract the JSON list from the text:
+                    //
+                    var json = ExtractJson(validationJson);
+
+                    try
+                    {
+                        var result = JsonSerializer.Deserialize<RetrievalContextValidationResult>(json, JSON_SERIALIZER_OPTIONS);
+                        return result with { RetrievalContext = retrievalContext };
+                    }
+                    catch
+                    {
+                        logger.LogWarning("The agent answered with an invalid or unexpected JSON format.");
+                        return new(false, "The agent answered with an invalid or unexpected JSON format.", 1.0f, retrievalContext);
+                    }
+                }
+
+                case { ContentType: ContentType.TEXT }:
+                    logger.LogWarning("The agent answered with an unexpected inner content type.");
+                    return new(false, "The agent answered with an unexpected inner content type.", 1.0f, retrievalContext);
+
+                case { ContentType: ContentType.NONE }:
+                    logger.LogWarning("The agent did not return a response.");
+                    return new(false, "The agent did not return a response.", 1.0f, retrievalContext);
+
+                default:
+                    logger.LogWarning($"The agent answered with an unexpected content type '{aiResponse.ContentType}'.");
+                    return new(false, $"The agent answered with an unexpected content type '{aiResponse.ContentType}'.", 1.0f, retrievalContext);
+            }
+        }
+        finally
+        {
+            // Release the semaphore slot:
+            semaphore?.Release();
+        }
+    }
+    
+    // A wrapper around the span version, because we need to call this method from an async context.
+    private static string ExtractJson(string text) => ExtractJson(text.AsSpan()).ToString();
+    
+    private static ReadOnlySpan<char> ExtractJson(ReadOnlySpan<char> input)
+    {
+        //
+        // 1. Expect the best case ;-)
+        //
+        if (CheckJsonObjectStart(input))
+            return ExtractJsonPart(input);
+
+        //
+        // 2. Okay, we have some garbage before the
+        // JSON object. We expected that...
+        //
+        for (var index = 0; index < input.Length; index++)
+        {
+            if (input[index] is '{' && CheckJsonObjectStart(input[index..]))
+                return ExtractJsonPart(input[index..]);
+        }
+
+        return [];
+    }
+    
+    private static bool CheckJsonObjectStart(ReadOnlySpan<char> area)
+    {
+        char[] expectedSymbols = ['{', '"', 'd'];
+        var symbolIndex = 0;
+
+        foreach (var c in area)
+        {
+            if (symbolIndex >= expectedSymbols.Length)
+                return true;
+
+            if (char.IsWhiteSpace(c))
+                continue;
+
+            if (c == expectedSymbols[symbolIndex++])
+                continue;
+
+            return false;
+        }
+
+        return true;
+    }
+
+    private static ReadOnlySpan<char> ExtractJsonPart(ReadOnlySpan<char> input)
+    {
+        var insideString = false;
+        for (var index = 0; index < input.Length; index++)
+        {
+            if (input[index] is '"')
+            {
+                insideString = !insideString;
+                continue;
+            }
+
+            if (insideString)
+                continue;
+
+            if (input[index] is '}')
+                return input[..++index];
+        }
+
+        return [];
+    }
+}

app/MindWork AI Studio/Agents/AgentTextContentCleaner.cs

@@ -1,18 +1,11 @@
 using AIStudio.Chat;
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 namespace AIStudio.Agents;
 
-public sealed class AgentTextContentCleaner(ILogger<AgentBase> logger, SettingsManager settingsManager, ThreadSafeRandom rng) : AgentBase(logger, settingsManager, rng)
+public sealed class AgentTextContentCleaner(ILogger<AgentBase> logger, SettingsManager settingsManager, DataSourceService dataSourceService, ThreadSafeRandom rng) : AgentBase(logger, settingsManager, dataSourceService, rng)
 {
-    private static readonly ContentBlock EMPTY_BLOCK = new()
-    {
-        Content = null,
-        ContentType = ContentType.NONE,
-        Role = ChatRole.AGENT,
-        Time = DateTimeOffset.UtcNow,
-    };
-    
     private readonly List<ContentBlock> context = new();
     private readonly List<ContentBlock> answers = new();
     
@@ -72,8 +65,8 @@ public sealed class AgentTextContentCleaner(ILogger<AgentBase> logger, SettingsM
             return EMPTY_BLOCK;
         
         var thread = this.CreateChatThread(this.SystemPrompt(sourceURL));
-        var time = this.AddUserRequest(thread, text.Text);
-        await this.AddAIResponseAsync(thread, time);
+        var userRequest = this.AddUserRequest(thread, text.Text);
+        await this.AddAIResponseAsync(thread, userRequest.UserPrompt, userRequest.Time);
         
         var answer = thread.Blocks[^1];
         this.answers.Add(answer);

app/MindWork AI Studio/Agents/RetrievalContextValidationResult.cs

@@ -0,0 +1,12 @@
+using AIStudio.Tools.RAG;
+
+namespace AIStudio.Agents;
+
+/// <summary>
+/// Represents the result of a retrieval context validation.
+/// </summary>
+/// <param name="Decision">Whether the retrieval context is useful or not.</param>
+/// <param name="Reason">The reason for the decision.</param>
+/// <param name="Confidence">The confidence of the decision.</param>
+/// <param name="RetrievalContext">The retrieval context that was validated.</param>
+public readonly record struct RetrievalContextValidationResult(bool Decision, string Reason, float Confidence, IRetrievalContext? RetrievalContext) : IConfidence;

app/MindWork AI Studio/Agents/SelectedDataSource.cs

@@ -0,0 +1,9 @@
+namespace AIStudio.Agents;
+
+/// <summary>
+/// Represents a selected data source, chosen by the agent.
+/// </summary>
+/// <param name="Id">The data source ID.</param>
+/// <param name="Reason">The reason for selecting the data source.</param>
+/// <param name="Confidence">The confidence of the agent in the selection.</param>
+public readonly record struct SelectedDataSource(string Id, string Reason, float Confidence) : IConfidence;

app/MindWork AI Studio/Agents/UserRequest.cs

@@ -0,0 +1,19 @@
+using AIStudio.Chat;
+
+namespace AIStudio.Agents;
+
+/// <summary>
+/// The created user request.
+/// </summary>
+public sealed class UserRequest
+{
+    /// <summary>
+    /// The time when the request was created.
+    /// </summary>
+    public required DateTimeOffset Time { get; init; }
+
+    /// <summary>
+    /// The user prompt.
+    /// </summary>
+    public required IContent UserPrompt { get; init; }
+}

app/MindWork AI Studio/Assistants/AssistantBase.razor

@@ -121,7 +121,7 @@
 
                 @if (this.SettingsManager.ConfigurationData.LLMProviders.ShowProviderConfidence)
                 {
-                    <ConfidenceInfo Mode="ConfidenceInfoMode.BUTTON" LLMProvider="@this.providerSettings.UsedLLMProvider"/>
+                    <ConfidenceInfo Mode="PopoverTriggerMode.BUTTON" LLMProvider="@this.providerSettings.UsedLLMProvider"/>
                 }
 
                 @if (this.AllowProfiles && this.ShowProfileSelection)

app/MindWork AI Studio/Assistants/AssistantBase.razor.cs

@@ -1,12 +1,12 @@
 using AIStudio.Chat;
 using AIStudio.Provider;
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 
 using MudBlazor.Utilities;
 
-using RustService = AIStudio.Tools.RustService;
 using Timer = System.Timers.Timer;
 
 namespace AIStudio.Assistants;
@@ -28,6 +28,9 @@ public abstract partial class AssistantBase : ComponentBase, IMessageBusReceiver
     [Inject]
     protected RustService RustService { get; init; } = null!;
     
+    [Inject]
+    protected DataSourceService DataSourceService { get; init; } = null!;
+    
     [Inject]
     protected NavigationManager NavigationManager { get; init; } = null!;
     
@@ -97,6 +100,7 @@ public abstract partial class AssistantBase : ComponentBase, IMessageBusReceiver
     protected bool inputIsValid;
     protected Profile currentProfile = Profile.NO_PROFILE;
     protected ChatThread? chatThread;
+    protected IContent? lastUserPrompt;
     
     private readonly Timer formChangeTimer = new(TimeSpan.FromSeconds(1.6));
     
@@ -242,16 +246,18 @@ public abstract partial class AssistantBase : ComponentBase, IMessageBusReceiver
     protected DateTimeOffset AddUserRequest(string request, bool hideContentFromUser = false)
     {
         var time = DateTimeOffset.Now;
+        this.lastUserPrompt = new ContentText
+        {
+            Text = request,
+        };
+        
         this.chatThread!.Blocks.Add(new ContentBlock
         {
             Time = time,
             ContentType = ContentType.TEXT,
             HideFromUser = hideContentFromUser,
             Role = ChatRole.USER,
-            Content = new ContentText
-            {
-                Text = request,
-            },
+            Content = this.lastUserPrompt,
         });
         
         return time;
@@ -287,7 +293,7 @@ public abstract partial class AssistantBase : ComponentBase, IMessageBusReceiver
         // Use the selected provider to get the AI response.
         // By awaiting this line, we wait for the entire
         // content to be streamed.
-        await aiText.CreateFromProviderAsync(this.providerSettings.CreateProvider(this.Logger), this.SettingsManager, this.providerSettings.Model, this.chatThread);
+        await aiText.CreateFromProviderAsync(this.providerSettings.CreateProvider(this.Logger), this.providerSettings.Model, this.lastUserPrompt, this.chatThread);
         
         this.isProcessing = false;
         this.StateHasChanged();

app/MindWork AI Studio/Assistants/ERI/AssistantERI.razor

@@ -3,12 +3,12 @@
 @using MudExtensions
 @inherits AssistantBaseCore
 
-<MudText Typo="Typo.body1" Class="mb-3">
+<MudJustifiedText Typo="Typo.body1" Class="mb-3">
     You can imagine it like this: Hypothetically, when Wikipedia implemented the ERI, it would vectorize
     all pages using an embedding method. All of Wikipedia’s data would remain with Wikipedia, including the
     vector database (decentralized approach). Then, any AI Studio user could add Wikipedia as a data source to
     significantly reduce the hallucination of the LLM in knowledge questions.
-</MudText>
+</MudJustifiedText>
 
 <MudText Typo="Typo.body1">
     <b>Related links:</b>
@@ -25,10 +25,10 @@
     ERI server presets
 </MudText>
 
-<MudText Typo="Typo.body1" Class="mb-3">
+<MudJustifiedText Typo="Typo.body1" Class="mb-3">
     Here you have the option to save different configurations for various ERI servers and switch between them. This is useful if
     you are responsible for multiple ERI servers.
-</MudText>
+</MudJustifiedText>
 
 @if(this.SettingsManager.ConfigurationData.ERI.ERIServers.Count is 0)
 {
@@ -59,26 +59,26 @@ else
 
 @if(this.AreServerPresetsBlocked)
 {
-    <MudText Typo="Typo.body1" Class="mb-3 mt-3">
+    <MudJustifiedText Typo="Typo.body1" Class="mb-3 mt-3">
         Hint: to allow this assistant to manage multiple presets, you must enable the preselection of values in the settings.
-    </MudText>
+    </MudJustifiedText>
 }
 
 <MudText Typo="Typo.h4" Class="mb-3 mt-6">
     Auto save
 </MudText>
 
-<MudText Typo="Typo.body1" Class="mb-3">
+<MudJustifiedText Typo="Typo.body1" Class="mb-3">
     The ERI specification will change over time. You probably want to keep your ERI server up to date. This means you might want to
     regenerate the code for your ERI server. To avoid having to make all inputs each time, all your inputs and decisions can be
     automatically saved. Would you like this?
-</MudText>
+</MudJustifiedText>
 
 @if(this.AreServerPresetsBlocked)
 {
-    <MudText Typo="Typo.body1" Class="mb-3">
+    <MudJustifiedText Typo="Typo.body1" Class="mb-3">
         Hint: to allow this assistant to automatically save your changes, you must enable the preselection of values in the settings.
-    </MudText>
+    </MudJustifiedText>
 }
 
 <MudTextSwitch Label="Should we automatically save any input made?" Disabled="@this.AreServerPresetsBlocked" @bind-Value="@this.autoSave" LabelOn="Yes, please save my inputs" LabelOff="No, I will enter everything again or configure it manually in the settings" />
@@ -204,18 +204,18 @@ else
     Embedding settings
 </MudText>
 
-<MudText Typo="Typo.body1" Class="mb-2">
+<MudJustifiedText Typo="Typo.body1" Class="mb-2">
     You will likely use one or more embedding methods to encode the meaning of your data into a typically high-dimensional vector
     space. In this case, you will use a vector database to store and search these vectors (called embeddings). However, you don't
     have to use embedding methods. When your retrieval method works without any embedding, you can ignore this section. An example: You
     store files on a file server, and your retrieval method works exclusively with file names in the file system, so you don't
     need embeddings.
-</MudText>
+</MudJustifiedText>
 
-<MudText Typo="Typo.body1" Class="mb-3">
+<MudJustifiedText Typo="Typo.body1" Class="mb-3">
     You can specify more than one embedding method. This can be useful when you want to use different embeddings for different queries
     or data types. For example, one embedding for texts, another for images, and a third for videos, etc.
-</MudText>
+</MudJustifiedText>
 
 @if (!this.IsNoneERIServerSelected)
 {
@@ -258,12 +258,12 @@ else
     Data retrieval settings
 </MudText>
 
-<MudText Typo="Typo.body1" Class="mb-2">
+<MudJustifiedText Typo="Typo.body1" Class="mb-2">
     For your ERI server, you need to retrieve data that matches a chat or prompt in some way. We call this the retrieval process.
     You must describe at least one such process. You may offer several retrieval processes from which users can choose. This allows
     you to test with beta users which process works better. Or you might generally want to give users the choice so they can select
     the process that best suits their circumstances.
-</MudText>
+</MudJustifiedText>
 
 @if (!this.IsNoneERIServerSelected)
 {
@@ -299,13 +299,13 @@ else
     Add Retrieval Process
 </MudButton>
 
-<MudText Typo="Typo.body1" Class="mb-1">
+<MudJustifiedText Typo="Typo.body1" Class="mb-1">
     You can integrate additional libraries. Perhaps you want to evaluate the prompts in advance using a machine learning method or analyze them with a text
     mining approach? Or maybe you want to preprocess images in the prompts? For such advanced scenarios, you can specify which libraries you want to use here.
     It's best to describe which library you want to integrate for which purpose. This way, the LLM that writes the ERI server for you can try to use these
     libraries effectively. This should result in less rework being necessary. If you don't know the necessary libraries, you can instead attempt to describe
     the intended use. The LLM can then attempt to choose suitable libraries. However, hallucinations can occur, and fictional libraries might be selected.
-</MudText>
+</MudJustifiedText>
 
 <MudTextField Disabled="@this.IsNoneERIServerSelected" T="string" @bind-Text="@this.additionalLibraries" Label="(Optional) Additional libraries" HelperText="Do you want to include additional libraries? Then name them and briefly describe what you want to achieve with them." Variant="Variant.Outlined" Margin="Margin.Normal" Lines="3" AutoGrow="@true" MaxLines="12" UserAttributes="@USER_INPUT_ATTRIBUTES" Class="mb-3"/>
 
@@ -313,17 +313,17 @@ else
     Provider selection for generation
 </MudText>
 
-<MudText Typo="Typo.body1" Class="mb-2">
+<MudJustifiedText Typo="Typo.body1" Class="mb-2">
     The task of writing the ERI server for you is very complex. Therefore, a very powerful LLM is needed to successfully accomplish this task.
     Small local models will probably not be sufficient. Instead, try using a large cloud-based or a large self-hosted model.
-</MudText>
+</MudJustifiedText>
 
-<MudText Typo="Typo.body1" Class="mb-2">
+<MudJustifiedText Typo="Typo.body1" Class="mb-2">
     <b>Important:</b> The LLM may need to generate many files. This reaches the request limit of most providers. Typically, only a certain number
     of requests can be made per minute, and only a maximum number of tokens can be generated per minute. AI Studio automatically considers this.
     <b>However, generating all the files takes a certain amount of time.</b> Local or self-hosted models may work without these limitations
     and can generate responses faster. AI Studio dynamically adapts its behavior and always tries to achieve the fastest possible data processing.
-</MudText>
+</MudJustifiedText>
 
 <ProviderSelection @bind-ProviderSettings="@this.providerSettings" ValidateProvider="@this.ValidatingProvider"/>
 
@@ -331,19 +331,19 @@ else
     Write code to file system
 </MudText>
 
-<MudText Typo="Typo.body1" Class="mb-2">
+<MudJustifiedText Typo="Typo.body1" Class="mb-2">
     AI Studio can save the generated code to the file system. You can select a base folder for this. AI Studio ensures that no files are created
     outside of this base folder. Furthermore, we recommend that you create a Git repository in this folder. This way, you can see what changes the
     AI has made in which files.
-</MudText>
+</MudJustifiedText>
 
-<MudText Typo="Typo.body1" Class="mb-2">
+<MudJustifiedText Typo="Typo.body1" Class="mb-2">
     When you rebuild / re-generate the ERI server code, AI Studio proceeds as follows: All files generated last time will be deleted. All
     other files you have created remain. Then, the AI generates the new files. <b>But beware:</b> It may happen that the AI generates a
     file this time that you manually created last time. In this case, your manually created file will then be overwritten. Therefore,
     you should always create a Git repository and commit or revert all changes before using this assistant. With a diff visualization,
     you can immediately see where the AI has made changes. It is best to use an IDE suitable for your selected language for this purpose.
-</MudText>
+</MudJustifiedText>
 
 <MudTextSwitch Label="Should we write the generated code to the file system?" Disabled="@this.IsNoneERIServerSelected" @bind-Value="@this.writeToFilesystem" LabelOn="Yes, please write or update all generated code to the file system" LabelOff="No, just show me the code" />
 <SelectDirectory Label="Base directory where to write the code" @bind-Directory="@this.baseDirectory" Disabled="@(this.IsNoneERIServerSelected || !this.writeToFilesystem)" DirectoryDialogTitle="Select the target directory for the ERI server" Validation="@this.ValidateDirectory" />

app/MindWork AI Studio/Chat/ChatRole.cs

@@ -12,6 +12,7 @@ public enum ChatRole
     USER,
     AI,
     AGENT,
+    RAG,
 }
 
 /// <summary>

app/MindWork AI Studio/Chat/ChatThread.cs

@@ -1,4 +1,7 @@
+using AIStudio.Components;
 using AIStudio.Settings;
+using AIStudio.Settings.DataModel;
+using AIStudio.Tools.ERIClient.DataModel;
 
 namespace AIStudio.Chat;
 
@@ -27,6 +30,16 @@ public sealed record ChatThread
     /// </summary>
     public string SelectedProfile { get; set; } = string.Empty;
 
+    /// <summary>
+    /// The data source options for this chat thread.
+    /// </summary>
+    public DataSourceOptions DataSourceOptions { get; set; } = new();
+
+    /// <summary>
+    /// The AI-selected data sources for this chat thread.
+    /// </summary>
+    public IReadOnlyList<DataSourceAgentSelected> AISelectedDataSources { get; set; } = [];
+
     /// <summary>
     /// The name of the chat thread. Usually generated by an AI model or manually edited by the user.
     /// </summary>
@@ -138,4 +151,46 @@ public sealed record ChatThread
         // Remove the block from the chat thread:
         this.Blocks.Remove(block);
     }
+
+    /// <summary>
+    /// Transforms this chat thread to an ERI chat thread.
+    /// </summary>
+    /// <param name="token">The cancellation token.</param>
+    /// <returns>The ERI chat thread.</returns>
+    public async Task<Tools.ERIClient.DataModel.ChatThread> ToERIChatThread(CancellationToken token = default)
+    {
+        //
+        // Transform the content blocks:
+        //
+        var contentBlocks = new List<Tools.ERIClient.DataModel.ContentBlock>(this.Blocks.Count);
+        foreach (var block in this.Blocks)
+        {
+            var (contentData, contentType) = block.Content switch
+            {
+                ContentImage image => (await image.AsBase64(token), Tools.ERIClient.DataModel.ContentType.IMAGE),
+                ContentText text => (text.Text, Tools.ERIClient.DataModel.ContentType.TEXT),
+                
+                _ => (string.Empty, Tools.ERIClient.DataModel.ContentType.UNKNOWN),
+            };
+            
+            contentBlocks.Add(new Tools.ERIClient.DataModel.ContentBlock
+            {
+                Role = block.Role switch
+                {
+                    ChatRole.AI => Role.AI,
+                    ChatRole.USER => Role.USER,
+                    ChatRole.AGENT => Role.AGENT,
+                    ChatRole.SYSTEM => Role.SYSTEM,
+                    ChatRole.NONE => Role.NONE,
+                    
+                    _ => Role.UNKNOW,
+                },
+                
+                Content = contentData,
+                Type = contentType,
+            });
+        }
+        
+        return new Tools.ERIClient.DataModel.ChatThread { ContentBlocks = contentBlocks };
+    }
 }

app/MindWork AI Studio/Chat/ContentBlockComponent.razor

@@ -78,9 +78,9 @@
                     break;
 
                 case ContentType.IMAGE:
-                    if (this.Content is ContentImage imageContent)
+                    if (this.Content is ContentImage { SourceType: ContentImageSource.URL or ContentImageSource.LOCAL_PATH } imageContent)
                     {
-                        <MudImage Src="@imageContent.URL"/>
+                        <MudImage Src="@imageContent.Source"/>
                     }
 
                     break;

app/MindWork AI Studio/Chat/ContentBlockComponent.razor.cs

@@ -1,9 +1,8 @@
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 
-using RustService = AIStudio.Tools.RustService;
-
 namespace AIStudio.Chat;
 
 /// <summary>

app/MindWork AI Studio/Chat/ContentImage.cs

@@ -1,14 +1,13 @@
 using System.Text.Json.Serialization;
 
 using AIStudio.Provider;
-using AIStudio.Settings;
 
 namespace AIStudio.Chat;
 
 /// <summary>
 /// Represents an image inside the chat.
 /// </summary>
-public sealed class ContentImage : IContent
+public sealed class ContentImage : IContent, IImageSource
 {
     #region Implementation of IContent
 
@@ -29,7 +28,7 @@ public sealed class ContentImage : IContent
     public Func<Task> StreamingEvent { get; set; } = () => Task.CompletedTask;
 
     /// <inheritdoc />
-    public Task CreateFromProviderAsync(IProvider provider, SettingsManager settings, Model chatModel, ChatThread chatChatThread, CancellationToken token = default)
+    public Task CreateFromProviderAsync(IProvider provider, Model chatModel, IContent? lastPrompt, ChatThread? chatChatThread, CancellationToken token = default)
     {
         throw new NotImplementedException();
     }
@@ -37,12 +36,15 @@ public sealed class ContentImage : IContent
     #endregion
 
     /// <summary>
-    /// The URL of the image.
+    /// The type of the image source.
     /// </summary>
-    public string URL { get; set; } = string.Empty;
+    /// <remarks>
+    /// Is the image source a URL, a local file path, a base64 string, etc.?
+    /// </remarks>
+    public required ContentImageSource SourceType { get; init; }
 
     /// <summary>
-    /// The local path of the image.
+    /// The image source.
     /// </summary>
-    public string LocalPath { get; set; } = string.Empty;
+    public required string Source { get; set; }
 }

app/MindWork AI Studio/Chat/ContentImageSource.cs

@@ -0,0 +1,8 @@
+namespace AIStudio.Chat;
+
+public enum ContentImageSource
+{
+    URL,
+    LOCAL_PATH,
+    BASE64,
+}

app/MindWork AI Studio/Chat/ContentText.cs

@@ -2,6 +2,7 @@ using System.Text.Json.Serialization;
 
 using AIStudio.Provider;
 using AIStudio.Settings;
+using AIStudio.Tools.RAG.RAGProcesses;
 
 namespace AIStudio.Chat;
 
@@ -35,10 +36,17 @@ public sealed class ContentText : IContent
     public Func<Task> StreamingEvent { get; set; } = () => Task.CompletedTask;
 
     /// <inheritdoc />
-    public async Task CreateFromProviderAsync(IProvider provider, SettingsManager settings, Model chatModel, ChatThread? chatThread, CancellationToken token = default)
+    public async Task CreateFromProviderAsync(IProvider provider, Model chatModel, IContent? lastPrompt, ChatThread? chatThread, CancellationToken token = default)
     {
         if(chatThread is null)
             return;
+        
+        // Call the RAG process. Right now, we only have one RAG process:
+        if (lastPrompt is not null)
+        {
+            var rag = new AISrcSelWithRetCtxVal();
+            chatThread = await rag.ProcessAsync(provider, lastPrompt, chatThread, token);
+        }
 
         // Store the last time we got a response. We use this later
         // to determine whether we should notify the UI about the
@@ -46,6 +54,9 @@ public sealed class ContentText : IContent
         // the user chose.
         var last = DateTimeOffset.Now;
 
+        // Get the settings manager:
+        var settings = Program.SERVICE_PROVIDER.GetService<SettingsManager>()!;
+        
         // Start another thread by using a task to uncouple
         // the UI thread from the AI processing:
         await Task.Run(async () =>

app/MindWork AI Studio/Chat/IContent.cs

@@ -1,7 +1,6 @@
 using System.Text.Json.Serialization;
 
 using AIStudio.Provider;
-using AIStudio.Settings;
 
 namespace AIStudio.Chat;
 
@@ -42,5 +41,16 @@ public interface IContent
     /// <summary>
     /// Uses the provider to create the content.
     /// </summary>
-    public Task CreateFromProviderAsync(IProvider provider, SettingsManager settings, Model chatModel, ChatThread chatChatThread, CancellationToken token = default);
+    public Task CreateFromProviderAsync(IProvider provider, Model chatModel, IContent? lastPrompt, ChatThread? chatChatThread, CancellationToken token = default);
+
+    /// <summary>
+    /// Returns the corresponding ERI content type.
+    /// </summary>
+    public Tools.ERIClient.DataModel.ContentType ToERIContentType => this switch
+    {
+        ContentText => Tools.ERIClient.DataModel.ContentType.TEXT,
+        ContentImage => Tools.ERIClient.DataModel.ContentType.IMAGE,
+        
+        _ => Tools.ERIClient.DataModel.ContentType.UNKNOWN,
+    };
 }

app/MindWork AI Studio/Chat/IImageSource.cs

@@ -0,0 +1,17 @@
+namespace AIStudio.Chat;
+
+public interface IImageSource
+{
+    /// <summary>
+    /// The type of the image source.
+    /// </summary>
+    /// <remarks>
+    /// Is the image source a URL, a local file path, a base64 string, etc.?
+    /// </remarks>
+    public ContentImageSource SourceType { get; init; }
+
+    /// <summary>
+    /// The image source.
+    /// </summary>
+    public string Source { get; set; }
+}

app/MindWork AI Studio/Chat/IImageSourceExtensions.cs

@@ -0,0 +1,63 @@
+namespace AIStudio.Chat;
+
+public static class IImageSourceExtensions
+{
+    /// <summary>
+    /// Read the image content as a base64 string.
+    /// </summary>
+    /// <remarks>
+    /// The images are directly converted to base64 strings. The maximum
+    /// size of the image is around 10 MB. If the image is larger, the method
+    /// returns an empty string.
+    /// 
+    /// As of now, this method does no sort of image processing. LLMs usually
+    /// do not work with arbitrary image sizes. In the future, we might have
+    /// to resize the images before sending them to the model.
+    /// </remarks>
+    /// <param name="image">The image source.</param>
+    /// <param name="token">The cancellation token.</param>
+    /// <returns>The image content as a base64 string; might be empty.</returns>
+    public static async Task<string> AsBase64(this IImageSource image, CancellationToken token = default)
+    {
+        switch (image.SourceType)
+        {
+            case ContentImageSource.BASE64:
+                return image.Source;
+            
+            case ContentImageSource.URL:
+            {
+                using var httpClient = new HttpClient();
+                using var response = await httpClient.GetAsync(image.Source, HttpCompletionOption.ResponseHeadersRead, token);
+                if(response.IsSuccessStatusCode)
+                {
+                    // Read the length of the content:
+                    var lengthBytes = response.Content.Headers.ContentLength;
+                    if(lengthBytes > 10_000_000)
+                        return string.Empty;
+                    
+                    var bytes = await response.Content.ReadAsByteArrayAsync(token);
+                    return Convert.ToBase64String(bytes);
+                }
+
+                return string.Empty;
+            }
+
+            case ContentImageSource.LOCAL_PATH:
+                if(File.Exists(image.Source))
+                {
+                    // Read the content length:
+                    var length = new FileInfo(image.Source).Length;
+                    if(length > 10_000_000)
+                        return string.Empty;
+                    
+                    var bytes = await File.ReadAllBytesAsync(image.Source, token);
+                    return Convert.ToBase64String(bytes);
+                }
+
+                return string.Empty;
+            
+            default:
+                return string.Empty;
+        }
+    }
+}

app/MindWork AI Studio/Components/Changelog.Logs.cs

@@ -13,6 +13,8 @@ public partial class Changelog
     
     public static readonly Log[] LOGS = 
     [
+        new (205, "v0.9.30, build 205 (2025-02-24 19:55 UTC)", "v0.9.30.md"),
+        new (204, "v0.9.29, build 204 (2025-02-24 13:48 UTC)", "v0.9.29.md"),
         new (203, "v0.9.28, build 203 (2025-02-09 16:33 UTC)", "v0.9.28.md"),
         new (202, "v0.9.27, build 202 (2025-01-21 18:24 UTC)", "v0.9.27.md"),
         new (201, "v0.9.26, build 201 (2025-01-13 19:11 UTC)", "v0.9.26.md"),

app/MindWork AI Studio/Components/ChatComponent.razor

@@ -97,7 +97,7 @@
 
             @if (this.SettingsManager.ConfigurationData.LLMProviders.ShowProviderConfidence)
             {
-                <ConfidenceInfo Mode="ConfidenceInfoMode.ICON" LLMProvider="@this.Provider.UsedLLMProvider"/>
+                <ConfidenceInfo Mode="PopoverTriggerMode.ICON" LLMProvider="@this.Provider.UsedLLMProvider"/>
             }
 
             @if (this.isStreaming && this.cancellationTokenSource is not null)
@@ -108,6 +108,11 @@
             }
 
             <ProfileSelection CurrentProfile="@this.currentProfile" CurrentProfileChanged="@this.ProfileWasChanged"/>
+            
+            @if (PreviewFeatures.PRE_RAG_2024.IsEnabled(this.SettingsManager))
+            {
+                <DataSourceSelection @ref="@this.dataSourceSelectionComponent" PopoverTriggerMode="PopoverTriggerMode.BUTTON" PopoverButtonClasses="ma-3" LLMProvider="@this.Provider" DataSourceOptions="@this.GetCurrentDataSourceOptions()" DataSourceOptionsChanged="@(async options => await this.SetCurrentDataSourceOptions(options))" DataSourcesAISelected="@this.GetAgentSelectedDataSources()"/>
+            }
         </MudToolBar>
     </FooterContent>
 </InnerScrolling>

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

@@ -3,6 +3,7 @@ using AIStudio.Dialogs;
 using AIStudio.Provider;
 using AIStudio.Settings;
 using AIStudio.Settings.DataModel;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 using Microsoft.AspNetCore.Components.Web;
@@ -40,9 +41,14 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
     [Inject]
     private IDialogService DialogService { get; init; } = null!;
     
+    [Inject]
+    private DataSourceService DataSourceService { get; init; } = null!;
+    
     private const Placement TOOLBAR_TOOLTIP_PLACEMENT = Placement.Top;
     private static readonly Dictionary<string, object?> USER_INPUT_ATTRIBUTES = new();
 
+    private DataSourceSelection? dataSourceSelectionComponent;
+    private DataSourceOptions earlyDataSourceOptions = new();
     private Profile currentProfile = Profile.NO_PROFILE;
     private bool hasUnsavedChanges;
     private bool mustScrollToBottomAfterRender;
@@ -66,21 +72,40 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
 
     protected override async Task OnInitializedAsync()
     {
+        // Apply the filters for the message bus:
         this.ApplyFilters([], [ Event.HAS_CHAT_UNSAVED_CHANGES, Event.RESET_CHAT_STATE, Event.CHAT_STREAMING_DONE, Event.WORKSPACE_LOADED_CHAT_CHANGED ]);
         
         // Configure the spellchecking for the user input:
         this.SettingsManager.InjectSpellchecking(USER_INPUT_ATTRIBUTES);
-        
+
+        // Get the preselected profile:
         this.currentProfile = this.SettingsManager.GetPreselectedProfile(Tools.Components.CHAT);
+        
+        //
+        // Check for deferred messages of the kind 'SEND_TO_CHAT',
+        // aka the user sends an assistant result to the chat:
+        //
         var deferredContent = MessageBus.INSTANCE.CheckDeferredMessages<ChatThread>(Event.SEND_TO_CHAT).FirstOrDefault();
         if (deferredContent is not null)
         {
+            //
+            // Yes, the user sent an assistant result to the chat.
+            //
+            
+            // Use chat thread sent by the user:
             this.ChatThread = deferredContent;
             this.Logger.LogInformation($"The chat '{this.ChatThread.Name}' with {this.ChatThread.Blocks.Count} messages was deferred and will be rendered now.");
             await this.ChatThreadChanged.InvokeAsync(this.ChatThread);
             
+            // We know already that the chat thread is not null,
+            // but we have to check it again for the nullability
+            // for the compiler:
             if (this.ChatThread is not null)
             {
+                //
+                // Check if the chat thread has a name. If not, we
+                // generate the name now:
+                //
                 if (string.IsNullOrWhiteSpace(this.ChatThread.Name))
                 {
                     var firstUserBlock = this.ChatThread.Blocks.FirstOrDefault(x => x.Role == ChatRole.USER);
@@ -94,12 +119,24 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                     }
                 }
 
+                //
+                // Check if the user wants to apply the standard chat data source options:
+                //
+                if (this.SettingsManager.ConfigurationData.Chat.SendToChatDataSourceBehavior is SendToChatDataSourceBehavior.APPLY_STANDARD_CHAT_DATA_SOURCE_OPTIONS)
+                    this.ChatThread.DataSourceOptions = this.SettingsManager.ConfigurationData.Chat.PreselectedDataSourceOptions.CreateCopy();
+
+                //
+                // Check if the user wants to store the chat automatically:
+                //
                 if (this.SettingsManager.ConfigurationData.Workspace.StorageBehavior is WorkspaceStorageBehavior.STORE_CHATS_AUTOMATICALLY)
                 {
                     this.autoSaveEnabled = true;
                     this.mustStoreChat = true;
                     
-                    // Ensure the workspace exists:
+                    //
+                    // When a standard workspace is used, we have to ensure
+                    // that the workspace is available:
+                    //
                     if(this.ChatThread.WorkspaceId == KnownWorkspaces.ERI_SERVER_WORKSPACE_ID)
                         await WorkspaceBehaviour.EnsureERIServerWorkspace();
                     
@@ -108,14 +145,38 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 }
             }
         }
+        else
+        {
+            //
+            // No, the user did not send an assistant result to the chat.
+            //
+            this.ApplyStandardDataSourceOptions();
+        }
         
+        //
+        // Check if the user wants to show the latest message after loading:
+        //
         if (this.SettingsManager.ConfigurationData.Chat.ShowLatestMessageAfterLoading)
         {
+            //
+            // We cannot scroll to the bottom right now because the
+            // chat component is not rendered yet. We have to wait for
+            // the rendering process to finish. Thus, we set a flag
+            // to scroll to the bottom after the rendering process.:
+            //
             this.mustScrollToBottomAfterRender = true;
             this.scrollRenderCountdown = 4;
             this.StateHasChanged();
         }
         
+        //
+        // Check if another component deferred the loading of a chat.
+        //
+        // This is used, e.g., for the bias-of-the-day component:
+        // when the bias for this day was already produced, the bias
+        // component sends a message to the chat component to load
+        // the chat with the bias:
+        //
         var deferredLoading = MessageBus.INSTANCE.CheckDeferredMessages<LoadChat>(Event.LOAD_CHAT).FirstOrDefault();
         if (deferredLoading != default)
         {
@@ -124,6 +185,11 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
             this.Logger.LogInformation($"The loading of the chat '{this.loadChat.ChatId}' was deferred and will be loaded now.");
         }
 
+        //
+        // When for whatever reason we have a chat thread, we have to
+        // ensure that the corresponding workspace id is set and the
+        // workspace name is loaded:
+        //
         if (this.ChatThread is not null)
         {
             this.currentWorkspaceId = this.ChatThread.WorkspaceId;
@@ -131,6 +197,7 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
             this.WorkspaceName(this.currentWorkspaceName);
         }
         
+        // Select the correct provider:
         await this.SelectProviderWhenLoadingChat();
         await base.OnInitializedAsync();
     }
@@ -204,6 +271,13 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
     
     private string UserInputClass => this.SettingsManager.ConfigurationData.LLMProviders.ShowProviderConfidence ? "confidence-border" : string.Empty;
     
+    private void ApplyStandardDataSourceOptions()
+    {
+        var chatDefaultOptions = this.SettingsManager.ConfigurationData.Chat.PreselectedDataSourceOptions.CreateCopy();
+        this.earlyDataSourceOptions = chatDefaultOptions;
+        this.dataSourceSelectionComponent?.ChangeOptionWithoutSaving(chatDefaultOptions);
+    }
+    
     private string ExtractThreadName(string firstUserInput)
     {
         // We select the first 10 words of the user input:
@@ -231,8 +305,43 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
         await this.ChatThreadChanged.InvokeAsync(this.ChatThread);
     }
 
+    private IReadOnlyList<DataSourceAgentSelected> GetAgentSelectedDataSources()
+    {
+        if (this.ChatThread is null)
+            return [];
+
+        return this.ChatThread.AISelectedDataSources;
+    }
+
+    private DataSourceOptions GetCurrentDataSourceOptions()
+    {
+        if (this.ChatThread is not null)
+            return this.ChatThread.DataSourceOptions;
+        
+        return this.earlyDataSourceOptions;
+    }
+    
+    private async Task SetCurrentDataSourceOptions(DataSourceOptions updatedOptions)
+    {
+        if (this.ChatThread is not null)
+        {
+            this.hasUnsavedChanges = true;
+            this.ChatThread.DataSourceOptions = updatedOptions;
+            if(this.SettingsManager.ConfigurationData.Workspace.StorageBehavior is WorkspaceStorageBehavior.STORE_CHATS_AUTOMATICALLY)
+            {
+                await this.SaveThread();
+                this.hasUnsavedChanges = false;
+            }
+        }
+        else
+            this.earlyDataSourceOptions = updatedOptions;
+    }
+
     private async Task InputKeyEvent(KeyboardEventArgs keyEvent)
     {
+        if(this.dataSourceSelectionComponent?.IsVisible ?? false)
+            this.dataSourceSelectionComponent.Hide();
+        
         this.hasUnsavedChanges = true;
         var key = keyEvent.Code.ToLowerInvariant();
         
@@ -276,6 +385,7 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 SystemPrompt = SystemPrompts.DEFAULT,
                 WorkspaceId = this.currentWorkspaceId,
                 ChatId = Guid.NewGuid(),
+                DataSourceOptions = this.earlyDataSourceOptions,
                 Name = this.ExtractThreadName(this.userInput),
                 Seed = this.RNG.Next(),
                 Blocks = [],
@@ -295,8 +405,14 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
         }
 
         var time = DateTimeOffset.Now;
+        IContent? lastUserPrompt;
         if (!reuseLastUserPrompt)
         {
+            lastUserPrompt = new ContentText
+            {
+                Text = this.userInput,
+            };
+
             //
             // Add the user message to the thread:
             //
@@ -305,10 +421,7 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 Time = time,
                 ContentType = ContentType.TEXT,
                 Role = ChatRole.USER,
-                Content = new ContentText
-                {
-                    Text = this.userInput,
-                },
+                Content = lastUserPrompt,
             });
 
             // Save the chat:
@@ -319,6 +432,8 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 this.StateHasChanged();
             }
         }
+        else
+            lastUserPrompt = this.ChatThread.Blocks.Last(x => x.Role is ChatRole.USER).Content;
 
         //
         // Add the AI response to the thread:
@@ -360,7 +475,7 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
             // Use the selected provider to get the AI response.
             // By awaiting this line, we wait for the entire
             // content to be streamed.
-            await aiText.CreateFromProviderAsync(this.Provider.CreateProvider(this.Logger), this.SettingsManager, this.Provider.Model, this.ChatThread, this.cancellationTokenSource.Token);
+            await aiText.CreateFromProviderAsync(this.Provider.CreateProvider(this.Logger), this.Provider.Model, lastUserPrompt, this.ChatThread, this.cancellationTokenSource.Token);
         }
         
         this.cancellationTokenSource = null;
@@ -374,6 +489,8 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
 
         // Disable the stream state:
         this.isStreaming = false;
+        
+        // Update the UI:
         this.StateHasChanged();
     }
     
@@ -407,6 +524,10 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
     
     private async Task StartNewChat(bool useSameWorkspace = false, bool deletePreviousChat = false)
     {
+        //
+        // Want the user to manage the chat storage manually? In that case, we have to ask the user
+        // about possible data loss:
+        //
         if (this.SettingsManager.ConfigurationData.Workspace.StorageBehavior is WorkspaceStorageBehavior.STORE_CHATS_MANUALLY && this.hasUnsavedChanges)
         {
             var dialogParameters = new DialogParameters
@@ -420,6 +541,9 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 return;
         }
 
+        //
+        // Delete the previous chat when desired and necessary:
+        //
         if (this.ChatThread is not null && deletePreviousChat)
         {
             string chatPath;
@@ -434,10 +558,16 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 await this.Workspaces.DeleteChat(chatPath, askForConfirmation: false, unloadChat: true);
         }
 
+        //
+        // Reset our state:
+        //
         this.isStreaming = false;
         this.hasUnsavedChanges = false;
         this.userInput = string.Empty;
         
+        //
+        // Reset the LLM provider considering the user's settings:
+        //
         switch (this.SettingsManager.ConfigurationData.Chat.AddChatProviderBehavior)
         {
             case AddChatProviderBehavior.ADDED_CHATS_USE_DEFAULT_PROVIDER:
@@ -456,8 +586,16 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 break;
         }
 
+        //
+        // Reset the chat thread or create a new one:
+        //
         if (!useSameWorkspace)
         {
+            //
+            // When the user wants to start a new chat outside the current workspace,
+            // we have to reset the workspace id and the workspace name. Also, we have
+            // to reset the chat thread:
+            //
             this.ChatThread = null;
             this.currentWorkspaceId = Guid.Empty;
             this.currentWorkspaceName = string.Empty;
@@ -465,6 +603,11 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
         }
         else
         {
+            //
+            // When the user wants to start a new chat in the same workspace, we have to
+            // reset the chat thread only. The workspace id and the workspace name remain
+            // the same:
+            //
             this.ChatThread = new()
             {
                 SelectedProvider = this.Provider.Id,
@@ -477,8 +620,11 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
                 Blocks = [],
             };
         }
-
-        this.userInput = string.Empty;
+        
+        // Now, we have to reset the data source options as well:
+        this.ApplyStandardDataSourceOptions();
+        
+        // Notify the parent component about the change:
         await this.ChatThreadChanged.InvokeAsync(this.ChatThread);
     }
     
@@ -538,12 +684,14 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
             this.currentWorkspaceId = this.ChatThread.WorkspaceId;
             this.currentWorkspaceName = await WorkspaceBehaviour.LoadWorkspaceName(this.ChatThread.WorkspaceId);
             this.WorkspaceName(this.currentWorkspaceName);
+            this.dataSourceSelectionComponent?.ChangeOptionWithoutSaving(this.ChatThread.DataSourceOptions, this.ChatThread.AISelectedDataSources);
         }
         else
         {
             this.currentWorkspaceId = Guid.Empty;
             this.currentWorkspaceName = string.Empty;
             this.WorkspaceName(this.currentWorkspaceName);
+            this.ApplyStandardDataSourceOptions();
         }
         
         await this.SelectProviderWhenLoadingChat();
@@ -567,6 +715,7 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
         this.WorkspaceName(this.currentWorkspaceName);
         
         this.ChatThread = null;
+        this.ApplyStandardDataSourceOptions();
         await this.ChatThreadChanged.InvokeAsync(this.ChatThread);
     }
     

app/MindWork AI Studio/Components/ConfidenceInfo.razor

@@ -1,7 +1,7 @@
 @using AIStudio.Provider
 <div class="d-flex">
     <MudTooltip Text="Shows and hides the confidence card with information about the selected LLM provider." Placement="Placement.Top">
-        @if (this.Mode is ConfidenceInfoMode.ICON)
+        @if (this.Mode is PopoverTriggerMode.ICON)
         {
             <MudIconButton Icon="@Icons.Material.Filled.Security" Class="confidence-icon" Style="@this.LLMProvider.GetConfidence(this.SettingsManager).SetColorStyle(this.SettingsManager)" OnClick="@(() => this.ToggleConfidence())"/>
         }
@@ -20,7 +20,7 @@
                     <MudText Typo="Typo.h5">Confidence Card</MudText>
                 </CardHeaderContent>
             </MudCardHeader>
-            <MudCardContent>
+            <MudCardContent Style="max-height: 50vh; overflow: auto;">
                 <MudText Typo="Typo.h6">Description</MudText>
                 <MudMarkdown Value="@this.currentConfidence.Description"/>
 

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

@@ -8,7 +8,7 @@ namespace AIStudio.Components;
 public partial class ConfidenceInfo : ComponentBase, IMessageBusReceiver, IDisposable
 {
     [Parameter]
-    public ConfidenceInfoMode Mode { get; set; } = ConfidenceInfoMode.BUTTON;
+    public PopoverTriggerMode Mode { get; set; } = PopoverTriggerMode.BUTTON;
     
     [Parameter]
     public LLMProviders LLMProvider { get; set; }

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

@@ -1,3 +1,5 @@
+using System.Diagnostics.CodeAnalysis;
+
 using AIStudio.Provider;
 using AIStudio.Settings;
 
@@ -47,6 +49,7 @@ public partial class ConfigurationProviderSelection : ComponentBase, IMessageBus
 
     #endregion
     
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
     private IEnumerable<ConfigurationSelectData<string>> FilteredData()
     {
         if(this.Component is not Tools.Components.NONE and not Tools.Components.APP_SETTINGS)

app/MindWork AI Studio/Components/DataSourceAgentSelected.cs

@@ -0,0 +1,25 @@
+using AIStudio.Agents;
+using AIStudio.Settings;
+
+namespace AIStudio.Components;
+
+/// <summary>
+/// A data structure to combine the data source and the underlying AI decision.
+/// </summary>
+public sealed class DataSourceAgentSelected
+{
+    /// <summary>
+    /// The data source.
+    /// </summary>
+    public required IDataSource DataSource { get; set; }
+
+    /// <summary>
+    /// The AI decision, which led to the selection of the data source.
+    /// </summary>
+    public required SelectedDataSource AIDecision { get; set; }
+
+    /// <summary>
+    /// Indicates whether the data source is part of the final selection for the RAG process.
+    /// </summary>
+    public bool Selected { get; set; }
+}

app/MindWork AI Studio/Components/DataSourceSelection.razor

@@ -0,0 +1,135 @@
+@using AIStudio.Settings
+
+@if (this.SelectionMode is DataSourceSelectionMode.SELECTION_MODE)
+{
+    <div class="d-flex">
+        <MudTooltip Text="Select the data sources you want to use here." Placement="Placement.Top">
+            @if (this.PopoverTriggerMode is PopoverTriggerMode.ICON)
+            {
+                <MudIconButton Icon="@Icons.Material.Filled.Source" Class="@this.PopoverButtonClasses" OnClick="@(() => this.ToggleDataSourceSelection())"/>
+            }
+            else
+            {
+                <MudButton Variant="Variant.Filled" StartIcon="@Icons.Material.Filled.Source" Class="@this.PopoverButtonClasses" OnClick="@(() => this.ToggleDataSourceSelection())">
+                    Select data sources
+                </MudButton>
+            }
+        </MudTooltip>
+
+        <MudPopover Open="@this.showDataSourceSelection" AnchorOrigin="Origin.TopLeft" TransformOrigin="Origin.BottomLeft" DropShadow="@true" Class="border-solid border-4 rounded-lg">
+            <MudCard>
+                <MudCardHeader>
+                    <CardHeaderContent>
+                        <PreviewPrototype/>
+                        <MudText Typo="Typo.h5">Data Source Selection</MudText>
+                    </CardHeaderContent>
+                </MudCardHeader>
+                <MudCardContent Style="min-width: 24em; max-height: 60vh; max-width: 45vw; overflow: auto;">
+                    @if (this.waitingForDataSources)
+                    {
+                        <MudSkeleton Width="30%" Height="42px;"/>
+                        <MudSkeleton Width="80%"/>
+                        <MudSkeleton Width="100%"/>
+                    }
+                    else if (this.showDataSourceSelection)
+                    {
+                        <MudTextSwitch Label="Are data sources enabled?" Value="@this.areDataSourcesEnabled" LabelOn="Yes, I want to use data sources." LabelOff="No, I don't want to use data sources." ValueChanged="@this.EnabledChanged"/>
+                        @if (this.areDataSourcesEnabled)
+                        {
+                            <MudTextSwitch Label="AI-based data source selection" Value="@this.aiBasedSourceSelection" LabelOn="Yes, let the AI decide which data sources are needed." LabelOff="No, I manually decide which data source to use." ValueChanged="@this.AutoModeChanged"/>
+                            
+                            @if (this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.EnableRetrievalContextValidation)
+                            {
+                                <MudTextSwitch Label="AI-based data validation" Value="@this.aiBasedValidation" LabelOn="Yes, let the AI validate & filter the retrieved data." LabelOff="No, use all data retrieved from the data sources." ValueChanged="@this.ValidationModeChanged"/>
+                            }
+
+                            @if (this.aiBasedSourceSelection is false || this.DataSourcesAISelected.Count == 0)
+                            {
+                                <MudField Label="Available Data Sources" Variant="Variant.Outlined" Class="mb-3" Disabled="@this.aiBasedSourceSelection">
+                                    <MudList T="IDataSource" SelectionMode="@this.GetListSelectionMode()" @bind-SelectedValues:get="@this.selectedDataSources" @bind-SelectedValues:set="@(x => this.SelectionChanged(x))" Style="max-height: 14em;">
+                                        @foreach (var source in this.availableDataSources)
+                                        {
+                                            <MudListItem Value="@source">
+                                                @source.Name
+                                            </MudListItem>
+                                        }
+                                    </MudList>
+                                </MudField>
+                            }
+                            else
+                            {
+                                <MudExpansionPanels MultiExpansion="@false" Class="mt-3" Style="max-height: 14em;">
+                                    <ExpansionPanel HeaderIcon="@Icons.Material.Filled.TouchApp" HeaderText="Available Data Sources">
+                                        <MudList T="IDataSource" SelectionMode="MudBlazor.SelectionMode.SingleSelection" SelectedValues="@this.selectedDataSources" Style="max-height: 14em;">
+                                            @foreach (var source in this.availableDataSources)
+                                            {
+                                                <MudListItem Value="@source">
+                                                    @source.Name
+                                                </MudListItem>
+                                            }
+                                        </MudList>
+                                    </ExpansionPanel>
+                                    <ExpansionPanel HeaderIcon="@Icons.Material.Filled.Filter" HeaderText="AI-Selected Data Sources">
+                                        <MudList T="DataSourceAgentSelected" SelectionMode="MudBlazor.SelectionMode.MultiSelection" ReadOnly="@true" SelectedValues="@this.GetSelectedDataSourcesWithAI()" Style="max-height: 14em;">
+                                            @foreach (var source in this.DataSourcesAISelected)
+                                            {
+                                                <MudListItem Value="@source">
+                                                    <ChildContent>
+                                                        <MudText Typo="Typo.body1">
+                                                            @source.DataSource.Name
+                                                        </MudText>
+                                                        
+                                                        <MudProgressLinear Color="Color.Info" Min="0" Max="1" Value="@source.AIDecision.Confidence"/>
+                                                        <MudJustifiedText Typo="Typo.body2">
+                                                            @this.GetAIReasoning(source)
+                                                        </MudJustifiedText>
+                                                    </ChildContent>
+                                                </MudListItem>
+                                            }
+                                        </MudList>
+                                    </ExpansionPanel>
+                                </MudExpansionPanels>
+                            }
+                        }
+                    }
+                </MudCardContent>
+                <MudCardActions>
+                    <MudButton Variant="Variant.Filled" OnClick="@(() => this.HideDataSourceSelection())">
+                        Close
+                    </MudButton>
+                </MudCardActions>
+            </MudCard>
+        </MudPopover>
+    </div>
+}
+else if (this.SelectionMode is DataSourceSelectionMode.CONFIGURATION_MODE)
+{
+    <MudPaper Class="pa-3 mb-8 mt-3 border-dashed border rounded-lg">
+        <PreviewPrototype/>
+        <MudText Typo="Typo.h5">Data Source Selection</MudText>
+        
+        @if (!string.IsNullOrWhiteSpace(this.ConfigurationHeaderMessage))
+        {
+            <MudText Typo="Typo.body1">
+                @this.ConfigurationHeaderMessage
+            </MudText>
+        }
+
+        <MudTextSwitch Label="Are data sources enabled?" Value="@this.areDataSourcesEnabled" LabelOn="Yes, I want to use data sources." LabelOff="No, I don't want to use data sources." ValueChanged="@this.EnabledChanged"/>
+        @if (this.areDataSourcesEnabled)
+        {
+            <MudTextSwitch Label="AI-based data source selection" Value="@this.aiBasedSourceSelection" LabelOn="Yes, let the AI decide which data sources are needed." LabelOff="No, I manually decide which data source to use." ValueChanged="@this.AutoModeChanged"/>
+            <MudTextSwitch Label="AI-based data validation" Value="@this.aiBasedValidation" LabelOn="Yes, let the AI validate & filter the retrieved data." LabelOff="No, use all data retrieved from the data sources." ValueChanged="@this.ValidationModeChanged"/>
+            <MudField Label="Available Data Sources" Variant="Variant.Outlined" Class="mb-3" Disabled="@this.aiBasedSourceSelection">
+                <MudList T="IDataSource" SelectionMode="@this.GetListSelectionMode()" @bind-SelectedValues:get="@this.selectedDataSources" @bind-SelectedValues:set="@(x => this.SelectionChanged(x))">
+                    @foreach (var source in this.availableDataSources)
+                    {
+                        <MudListItem Value="@source">
+                            @source.Name
+                        </MudListItem>
+                    }
+                </MudList>
+            </MudField>
+        }
+    </MudPaper>
+}

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

@@ -0,0 +1,278 @@
+using AIStudio.Settings;
+using AIStudio.Settings.DataModel;
+using AIStudio.Tools.Services;
+
+using Microsoft.AspNetCore.Components;
+
+namespace AIStudio.Components;
+
+public partial class DataSourceSelection : ComponentBase, IMessageBusReceiver, IDisposable
+{
+    [Parameter]
+    public DataSourceSelectionMode SelectionMode { get; set; } = DataSourceSelectionMode.SELECTION_MODE;
+    
+    [Parameter]
+    public PopoverTriggerMode PopoverTriggerMode { get; set; } = PopoverTriggerMode.BUTTON;
+    
+    [Parameter]
+    public string PopoverButtonClasses { get; set; } = string.Empty;
+    
+    [Parameter]
+    public required AIStudio.Settings.Provider LLMProvider { get; set; }
+
+    [Parameter]
+    public required DataSourceOptions DataSourceOptions { get; set; }
+    
+    [Parameter]
+    public EventCallback<DataSourceOptions> DataSourceOptionsChanged { get; set; }
+
+    [Parameter]
+    public IReadOnlyList<DataSourceAgentSelected> DataSourcesAISelected { get; set; } = [];
+    
+    [Parameter]
+    public string ConfigurationHeaderMessage { get; set; } = string.Empty;
+    
+    [Parameter]
+    public bool AutoSaveAppSettings { get; set; }
+    
+    [Inject]
+    private SettingsManager SettingsManager { get; init; } = null!;
+    
+    [Inject]
+    private MessageBus MessageBus { get; init; } = null!;
+    
+    [Inject]
+    private DataSourceService DataSourceService { get; init; } = null!;
+    
+    [Inject]
+    private ILogger<DataSourceSelection> Logger { get; init; } = null!;
+
+    private bool internalChange;
+    private bool showDataSourceSelection;
+    private bool waitingForDataSources = true;
+    private IReadOnlyList<IDataSource> availableDataSources = [];
+    private IReadOnlyCollection<IDataSource> selectedDataSources = [];
+    private bool aiBasedSourceSelection;
+    private bool aiBasedValidation;
+    private bool areDataSourcesEnabled;
+    
+    #region Overrides of ComponentBase
+
+    protected override async Task OnInitializedAsync()
+    {
+        this.MessageBus.RegisterComponent(this);
+        this.MessageBus.ApplyFilters(this, [], [ Event.COLOR_THEME_CHANGED, Event.RAG_AUTO_DATA_SOURCES_SELECTED ]);
+        
+        //
+        // Load the settings:
+        //
+        this.aiBasedSourceSelection = this.DataSourceOptions.AutomaticDataSourceSelection;
+        this.aiBasedValidation = this.DataSourceOptions.AutomaticValidation;
+        this.areDataSourcesEnabled = !this.DataSourceOptions.DisableDataSources;
+        this.waitingForDataSources = this.areDataSourcesEnabled;
+
+        //
+        // Preselect the data sources. Right now, we cannot filter
+        // the data sources. Later, when the component is shown, we
+        // will filter the data sources.
+        //
+        // Right before the preselection would be used to kick off the
+        // RAG process, we will filter the data sources as well.
+        //
+        var preselectedSources = new List<IDataSource>(this.DataSourceOptions.PreselectedDataSourceIds.Count);
+        foreach (var preselectedDataSourceId in this.DataSourceOptions.PreselectedDataSourceIds)
+        {
+            var dataSource = this.SettingsManager.ConfigurationData.DataSources.FirstOrDefault(ds => ds.Id == preselectedDataSourceId);
+            if (dataSource is not null)
+                preselectedSources.Add(dataSource);
+        }
+        
+        this.selectedDataSources = preselectedSources;
+        await base.OnInitializedAsync();
+    }
+
+    protected override async Task OnParametersSetAsync()
+    {
+        if (!this.internalChange)
+        {
+            this.aiBasedSourceSelection = this.DataSourceOptions.AutomaticDataSourceSelection;
+            this.aiBasedValidation = this.DataSourceOptions.AutomaticValidation;
+            this.areDataSourcesEnabled = !this.DataSourceOptions.DisableDataSources;
+        }
+
+        switch (this.SelectionMode)
+        {
+            //
+            // In selection mode, we have to load & filter the data sources
+            // when the component is shown:
+            //
+            case DataSourceSelectionMode.SELECTION_MODE:
+                
+                //
+                // For external changes, we have to reload & filter
+                // the data sources:
+                //
+                if (this.showDataSourceSelection && !this.internalChange)
+                    await this.LoadAndApplyFilters();
+                else
+                    this.internalChange = false;
+                
+                break;
+            
+            //
+            // In configuration mode, we have to load all data sources:
+            //
+            case DataSourceSelectionMode.CONFIGURATION_MODE:
+                this.availableDataSources = this.SettingsManager.ConfigurationData.DataSources;
+                break;
+        }
+        
+        await base.OnParametersSetAsync();
+    }
+
+    #endregion
+
+    private SelectionMode GetListSelectionMode() => this.aiBasedSourceSelection ? MudBlazor.SelectionMode.SingleSelection : MudBlazor.SelectionMode.MultiSelection;
+    
+    private IReadOnlyCollection<DataSourceAgentSelected> GetSelectedDataSourcesWithAI() => this.DataSourcesAISelected.Where(n => n.Selected).ToList();
+
+    private string GetAIReasoning(DataSourceAgentSelected source) => $"AI reasoning (confidence {source.AIDecision.Confidence:P0}): {source.AIDecision.Reason}";
+    
+    public void ChangeOptionWithoutSaving(DataSourceOptions options, IReadOnlyList<DataSourceAgentSelected>? aiSelectedDataSources = null)
+    {
+        this.DataSourceOptions = options;
+        this.DataSourcesAISelected = aiSelectedDataSources ?? [];
+        
+        this.aiBasedSourceSelection = this.DataSourceOptions.AutomaticDataSourceSelection;
+        this.aiBasedValidation = this.DataSourceOptions.AutomaticValidation;
+        this.areDataSourcesEnabled = !this.DataSourceOptions.DisableDataSources;
+        this.selectedDataSources = this.SettingsManager.ConfigurationData.DataSources.Where(ds => this.DataSourceOptions.PreselectedDataSourceIds.Contains(ds.Id)).ToList();
+        this.waitingForDataSources = false;
+
+        //
+        // Remark: We do not apply the filters here. This is done later
+        // when either the parameters are changed or just before the
+        // RAG process is started (outside of this component).
+        //
+        // In fact, when we apply the filters here, multiple calls
+        // to the filter method would be made. We would get conflicts.
+        //
+    }
+    
+    public bool IsVisible => this.showDataSourceSelection;
+
+    public void Hide()
+    {
+        this.showDataSourceSelection = false;
+        this.StateHasChanged();
+    }
+    
+    private async Task LoadAndApplyFilters()
+    {
+        if(this.DataSourceOptions.DisableDataSources)
+            return;
+        
+        this.waitingForDataSources = true;
+        this.StateHasChanged();
+            
+        // Load the data sources:
+        var sources = await this.DataSourceService.GetDataSources(this.LLMProvider, this.selectedDataSources);
+        this.availableDataSources = sources.AllowedDataSources;
+        this.selectedDataSources = sources.SelectedDataSources;
+        this.waitingForDataSources = false;
+        this.StateHasChanged();
+    }
+    
+    private async Task EnabledChanged(bool state)
+    {
+        this.areDataSourcesEnabled = state;
+        this.DataSourceOptions.DisableDataSources = !this.areDataSourcesEnabled;
+        
+        await this.LoadAndApplyFilters();
+        await this.OptionsChanged();
+        this.StateHasChanged();
+    }
+    
+    private async Task AutoModeChanged(bool state)
+    {
+        this.aiBasedSourceSelection = state;
+        this.DataSourceOptions.AutomaticDataSourceSelection = this.aiBasedSourceSelection;
+        
+        await this.OptionsChanged();
+    }
+    
+    private async Task ValidationModeChanged(bool state)
+    {
+        this.aiBasedValidation = state;
+        this.DataSourceOptions.AutomaticValidation = this.aiBasedValidation;
+        
+        await this.OptionsChanged();
+    }
+    
+    private async Task SelectionChanged(IReadOnlyCollection<IDataSource>? chosenDataSources)
+    {
+        this.selectedDataSources = chosenDataSources ?? [];
+        this.DataSourceOptions.PreselectedDataSourceIds = this.selectedDataSources.Select(ds => ds.Id).ToList();
+
+        await this.OptionsChanged();
+    }
+
+    private async Task OptionsChanged()
+    {
+        this.internalChange = true;
+        
+        await this.DataSourceOptionsChanged.InvokeAsync(this.DataSourceOptions);
+        
+        if(this.AutoSaveAppSettings)
+            await this.SettingsManager.StoreSettings();
+    }
+    
+    private async Task ToggleDataSourceSelection()
+    {
+        this.showDataSourceSelection = !this.showDataSourceSelection;
+        if (this.showDataSourceSelection)
+            await this.LoadAndApplyFilters();
+    }
+    
+    private void HideDataSourceSelection() => this.showDataSourceSelection = false;
+
+    #region Implementation of IMessageBusReceiver
+
+    public string ComponentName => nameof(ConfidenceInfo);
+    
+    public Task ProcessMessage<T>(ComponentBase? sendingComponent, Event triggeredEvent, T? data)
+    {
+        switch (triggeredEvent)
+        {
+            case Event.COLOR_THEME_CHANGED:
+                this.showDataSourceSelection = false;
+                this.StateHasChanged();
+                break;
+            
+            case Event.RAG_AUTO_DATA_SOURCES_SELECTED:
+                if(data is IReadOnlyList<DataSourceAgentSelected> aiSelectedDataSources)
+                    this.DataSourcesAISelected = aiSelectedDataSources;
+                    
+                this.StateHasChanged();
+                break;
+        }
+        
+        return Task.CompletedTask;
+    }
+
+    public Task<TResult?> ProcessMessageWithResult<TPayload, TResult>(ComponentBase? sendingComponent, Event triggeredEvent, TPayload? data)
+    {
+        return Task.FromResult<TResult?>(default);
+    }
+
+    #endregion
+    
+    #region Implementation of IDisposable
+
+    public void Dispose()
+    {
+        this.MessageBus.Unregister(this);
+    }
+
+    #endregion
+}

app/MindWork AI Studio/Components/DataSourceSelectionMode.cs

@@ -0,0 +1,23 @@
+namespace AIStudio.Components;
+
+public enum DataSourceSelectionMode
+{
+    /// <summary>
+    /// The user is selecting data sources for, e.g., the chat.
+    /// </summary>
+    /// <remarks>
+    /// In this case, we have to filter the data sources based on the
+    /// selected provider and check security requirements.
+    /// </remarks>
+    SELECTION_MODE,
+    
+    /// <summary>
+    /// The user is configuring the default data sources, e.g., for the chat.
+    /// </summary>
+    /// <remarks>
+    /// In this case, all data sources are available for selection.
+    /// They get filtered later based on the selected provider and
+    /// security requirements.
+    /// </remarks>
+    CONFIGURATION_MODE,
+}

app/MindWork AI Studio/Components/PopoverTriggerMode.cs

@@ -1,6 +1,6 @@
 namespace AIStudio.Components;
 
-public enum ConfidenceInfoMode
+public enum PopoverTriggerMode
 {
     BUTTON,
     ICON,

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

@@ -14,11 +14,14 @@ public partial class ProfileSelection : ComponentBase
 
     [Parameter]
     public string MarginLeft { get; set; } = "ml-3";
+
+    [Parameter]
+    public string MarginRight { get; set; } = string.Empty;
     
     [Inject]
     private SettingsManager SettingsManager { get; init; } = null!;
     
-    private string MarginClass => $"{this.MarginLeft}";
+    private string MarginClass => $"{this.MarginLeft} {this.MarginRight}";
     
     private async Task SelectionChanged(Profile profile)
     {

app/MindWork AI Studio/Components/ProviderSelection.razor

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

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

@@ -1,3 +1,5 @@
+using System.Diagnostics.CodeAnalysis;
+
 using AIStudio.Assistants;
 using AIStudio.Provider;
 using AIStudio.Settings;
@@ -29,6 +31,7 @@ public partial class ProviderSelection : ComponentBase
         await this.ProviderSettingsChanged.InvokeAsync(provider);
     }
     
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
     private IEnumerable<AIStudio.Settings.Provider> GetAvailableProviders()
     {
         var minimumLevel = this.SettingsManager.GetMinimumConfidenceLevel(this.AssistantBase?.Component ?? Tools.Components.NONE);

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

@@ -59,11 +59,7 @@ public partial class ReadWebContent : ComponentBase
         if(this.PreselectContentCleanerAgent)
             this.useContentCleanerAgent = true;
         
-        if (this.SettingsManager.ConfigurationData.TextContentCleaner.PreselectAgentOptions)
-            this.providerSettings = this.SettingsManager.ConfigurationData.Providers.FirstOrDefault(x => x.Id == this.SettingsManager.ConfigurationData.TextContentCleaner.PreselectedAgentProvider);
-        else
-            this.providerSettings = this.ProviderSettings;
-
+        this.ProviderSettings = this.SettingsManager.GetPreselectedProvider(Tools.Components.AGENT_TEXT_CONTENT_CLEANER, this.ProviderSettings.Id, true);
         await base.OnInitializedAsync();
     }
 

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

@@ -1,4 +1,5 @@
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 

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

@@ -1,4 +1,5 @@
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 

app/MindWork AI Studio/Components/Settings/SettingsPanelAgentDataSourceSelection.razor

@@ -0,0 +1,11 @@
+@inherits SettingsPanelBase
+
+<ExpansionPanel HeaderIcon="@Icons.Material.Filled.SelectAll" HeaderText="Agent: Data Source Selection 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 select the appropriate data sources for the current prompt.
+        </MudText>
+        <ConfigurationOption OptionDescription="Preselect data source selection options?" LabelOn="Options are preselected" LabelOff="No options are preselected" State="@(() => this.SettingsManager.ConfigurationData.AgentDataSourceSelection.PreselectAgentOptions)" StateUpdate="@(updatedState => this.SettingsManager.ConfigurationData.AgentDataSourceSelection.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.AgentDataSourceSelection.PreselectAgentOptions)" SelectedValue="@(() => this.SettingsManager.ConfigurationData.AgentDataSourceSelection.PreselectedAgentProvider)" SelectionUpdate="@(selectedValue => this.SettingsManager.ConfigurationData.AgentDataSourceSelection.PreselectedAgentProvider = selectedValue)"/>
+    </MudPaper>
+</ExpansionPanel>

app/MindWork AI Studio/Components/Settings/SettingsPanelAgentDataSourceSelection.razor.cs

@@ -0,0 +1,3 @@
+namespace AIStudio.Components.Settings;
+
+public partial class SettingsPanelAgentDataSourceSelection : SettingsPanelBase;

app/MindWork AI Studio/Components/Settings/SettingsPanelAgentRetrievalContextValidation.razor

@@ -0,0 +1,18 @@
+@inherits SettingsPanelBase
+
+<ExpansionPanel HeaderIcon="@Icons.Material.Filled.Assessment" HeaderText="Agent: Retrieval Context Validation Options">
+    <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."/>
+            <ConfigurationSlider T="int" OptionDescription="How many validation agents should work simultaneously?" Min="1" Max="100" Step="1" Unit="agents" Value="@(() => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.NumParallelValidations)" ValueUpdate="@(updatedValue => this.SettingsManager.ConfigurationData.AgentRetrievalContextValidation.NumParallelValidations = updatedValue)" OptionHelp="More active agents also mean that a corresponding number of requests are made simultaneously. Some providers limit the number of requests per minute. When you are unsure, choose a low setting between 1 to 6 agents."/>
+            <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>

app/MindWork AI Studio/Components/Settings/SettingsPanelAgentRetrievalContextValidation.razor.cs

@@ -0,0 +1,3 @@
+namespace AIStudio.Components.Settings;
+
+public partial class SettingsPanelAgentRetrievalContextValidation : SettingsPanelBase;

app/MindWork AI Studio/Components/Settings/SettingsPanelBase.cs

@@ -1,4 +1,5 @@
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 

app/MindWork AI Studio/Components/Settings/SettingsPanelChat.razor

@@ -1,4 +1,5 @@
 @using AIStudio.Settings
+@using AIStudio.Settings.DataModel
 @inherits SettingsPanelBase
 
 <ExpansionPanel HeaderIcon="@Icons.Material.Filled.Chat" HeaderText="Chat Options">
@@ -12,4 +13,10 @@
         <ConfigurationProviderSelection Component="Components.CHAT" Data="@this.AvailableLLMProvidersFunc()" Disabled="@(() => !this.SettingsManager.ConfigurationData.Chat.PreselectOptions)" SelectedValue="@(() => this.SettingsManager.ConfigurationData.Chat.PreselectedProvider)" SelectionUpdate="@(selectedValue => this.SettingsManager.ConfigurationData.Chat.PreselectedProvider = selectedValue)"/>
         <ConfigurationSelect OptionDescription="Preselect one of your profiles?" Disabled="@(() => !this.SettingsManager.ConfigurationData.Chat.PreselectOptions)" SelectedValue="@(() => this.SettingsManager.ConfigurationData.Chat.PreselectedProfile)" Data="@ConfigurationSelectDataFactory.GetProfilesData(this.SettingsManager.ConfigurationData.Profiles)" SelectionUpdate="@(selectedValue => this.SettingsManager.ConfigurationData.Chat.PreselectedProfile = selectedValue)" OptionHelp="Would you like to set one of your profiles as the default for chats?"/>
     </MudPaper>
+
+    @if (PreviewFeatures.PRE_RAG_2024.IsEnabled(this.SettingsManager))
+    {
+        <DataSourceSelection SelectionMode="DataSourceSelectionMode.CONFIGURATION_MODE" AutoSaveAppSettings="@true" @bind-DataSourceOptions="@this.SettingsManager.ConfigurationData.Chat.PreselectedDataSourceOptions" ConfigurationHeaderMessage="You can set default data sources and options for new chats. You can change these settings later for each individual chat."/>
+        <ConfigurationSelect OptionDescription="Apply default data source option when sending assistant results to chat" SelectedValue="@(() => this.SettingsManager.ConfigurationData.Chat.SendToChatDataSourceBehavior)" Data="@ConfigurationSelectDataFactory.GetSendToChatDataSourceBehaviorData()" SelectionUpdate="@(selectedValue => this.SettingsManager.ConfigurationData.Chat.SendToChatDataSourceBehavior = selectedValue)" OptionHelp="Do you want to apply the default data source options when sending assistant results to chat?"/>
+    }
 </ExpansionPanel>

app/MindWork AI Studio/Components/Settings/SettingsPanelProviders.razor.cs

@@ -1,3 +1,5 @@
+using System.Diagnostics.CodeAnalysis;
+
 using AIStudio.Dialogs;
 using AIStudio.Provider;
 using AIStudio.Settings;
@@ -26,6 +28,7 @@ public partial class SettingsPanelProviders : SettingsPanelBase
 
     #endregion
     
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
     private async Task AddLLMProvider()
     {
         var dialogParameters = new DialogParameters<ProviderDialog>
@@ -48,6 +51,7 @@ public partial class SettingsPanelProviders : SettingsPanelBase
         await this.MessageBus.SendMessage<bool>(this, Event.CONFIGURATION_CHANGED);
     }
 
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
     private async Task EditLLMProvider(AIStudio.Settings.Provider provider)
     {
         var dialogParameters = new DialogParameters<ProviderDialog>
@@ -82,6 +86,7 @@ public partial class SettingsPanelProviders : SettingsPanelBase
         await this.MessageBus.SendMessage<bool>(this, Event.CONFIGURATION_CHANGED);
     }
 
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
     private async Task DeleteLLMProvider(AIStudio.Settings.Provider provider)
     {
         var dialogParameters = new DialogParameters
@@ -112,6 +117,7 @@ public partial class SettingsPanelProviders : SettingsPanelBase
         return modelName.Length > MAX_LENGTH ? "[...] " + modelName[^Math.Min(MAX_LENGTH, modelName.Length)..] : modelName;
     }
     
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
     private async Task UpdateProviders()
     {
         this.AvailableLLMProviders.Clear();

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

@@ -1,4 +1,5 @@
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 

app/MindWork AI Studio/Components/TextInfoLines.razor

@@ -9,6 +9,7 @@
         Lines="3"
         MaxLines="@this.MaxLines"
         AutoGrow="@true"
+        Style="@this.GetColor()"
         UserAttributes="@USER_INPUT_ATTRIBUTES" />
 
     @if (this.ShowingCopyButton)

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

@@ -1,4 +1,5 @@
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 
@@ -20,6 +21,9 @@ public partial class TextInfoLines : ComponentBase
 
     [Parameter]
     public bool ShowingCopyButton { get; set; } = true;
+
+    [Parameter]
+    public TextColor Color { get; set; } = TextColor.DEFAULT;
     
     [Inject]
     private RustService RustService { get; init; } = null!;
@@ -47,4 +51,13 @@ public partial class TextInfoLines : ComponentBase
     private string ClipboardTooltip => $"Copy {this.ClipboardTooltipSubject} to the clipboard";
     
     private async Task CopyToClipboard(string content) => await this.RustService.CopyText2Clipboard(this.Snackbar, content);
+    
+    private string GetColor()
+    {
+        var htmlColorCode = this.Color.GetHTMLColor(this.SettingsManager);
+        if(string.IsNullOrWhiteSpace(htmlColorCode))
+            return string.Empty;
+        
+        return $"color: {htmlColorCode} !important;";
+    }
 }

app/MindWork AI Studio/Dialogs/DataSourceERI-V1InfoDialog.razor

@@ -1,4 +1,6 @@
+@using AIStudio.Settings.DataModel
 @using AIStudio.Tools.ERIClient.DataModel
+
 <MudDialog>
     <DialogContent>
 
@@ -25,7 +27,8 @@
         }
 
         <TextInfoLines Label="Server description" MaxLines="14" Value="@this.serverDescription" ClipboardTooltipSubject="the server description"/>
-        <TextInfoLines Label="Security requirements" MaxLines="3" Value="@this.securityRequirements.Explain()" ClipboardTooltipSubject="the security requirements"/>
+        <TextInfoLines Label="Security requirements of the data provider" MaxLines="3" Value="@this.securityRequirements.Explain()" ClipboardTooltipSubject="the security requirements of the data provider"/>
+        <TextInfoLines Label="Your security policy" MaxLines="3" Value="@this.DataSource.SecurityPolicy.ToInfoText()" Color="@this.DataSource.SecurityPolicy.GetColor()" ClipboardTooltipSubject="your security policy"/>
 
         <MudText Typo="Typo.h5" Class="mt-6">
             Retrieval information

app/MindWork AI Studio/Dialogs/DataSourceERI-V1InfoDialog.razor.cs

@@ -6,6 +6,7 @@ using AIStudio.Assistants.ERI;
 using AIStudio.Settings.DataModel;
 using AIStudio.Tools.ERIClient;
 using AIStudio.Tools.ERIClient.DataModel;
+using AIStudio.Tools.Services;
 
 using Microsoft.AspNetCore.Components;
 

app/MindWork AI Studio/Dialogs/DataSourceERI_V1Dialog.razor

@@ -1,3 +1,4 @@
+@using AIStudio.Settings.DataModel
 @using AIStudio.Tools.ERIClient.DataModel
 <MudDialog>
     <DialogContent>
@@ -108,6 +109,13 @@
                     InputType="InputType.Password"
                     UserAttributes="@SPELLCHECK_ATTRIBUTES"/>
             }
+
+            <MudSelect @bind-Value="@this.dataSecurityPolicy" Text="@this.dataSecurityPolicy.ToSelectionText()" Label="Your security policy" Class="mb-3" OpenIcon="@Icons.Material.Filled.ExpandMore" AdornmentColor="Color.Info" Adornment="Adornment.Start" Validation="@this.dataSourceValidation.ValidateSecurityPolicy">
+                @foreach (var policy in Enum.GetValues<DataSourceSecurity>())
+                {
+                    <MudSelectItem Value="@policy">@policy.ToSelectionText()</MudSelectItem>
+                }
+            </MudSelect>
             
         </MudForm>
         <Issues IssuesData="@this.dataIssues"/>

app/MindWork AI Studio/Dialogs/DataSourceERI_V1Dialog.razor.cs

@@ -3,6 +3,7 @@ using AIStudio.Settings;
 using AIStudio.Settings.DataModel;
 using AIStudio.Tools.ERIClient;
 using AIStudio.Tools.ERIClient.DataModel;
+using AIStudio.Tools.Services;
 using AIStudio.Tools.Validation;
 
 using Microsoft.AspNetCore.Components;
@@ -45,6 +46,8 @@ public partial class DataSourceERI_V1Dialog : ComponentBase, ISecretId
     private string dataSecretStorageIssue = string.Empty;
     private string dataEditingPreviousInstanceName = string.Empty;
     private List<AuthMethod> availableAuthMethods = [];
+    private DataSourceSecurity dataSecurityPolicy;
+    private SecurityRequirements dataSourceSecurityRequirements;
     private bool connectionTested;
     private bool connectionSuccessfulTested;
     
@@ -71,6 +74,7 @@ public partial class DataSourceERI_V1Dialog : ComponentBase, ISecretId
             GetTestedConnection = () => this.connectionTested,
             GetTestedConnectionResult = () => this.connectionSuccessfulTested,
             GetAvailableAuthMethods = () => this.availableAuthMethods,
+            GetSecurityRequirements = () => this.dataSourceSecurityRequirements,
         };
     }
     
@@ -95,6 +99,7 @@ public partial class DataSourceERI_V1Dialog : ComponentBase, ISecretId
             this.dataPort = this.DataSource.Port;
             this.dataAuthMethod = this.DataSource.AuthMethod;
             this.dataUsername = this.DataSource.Username;
+            this.dataSecurityPolicy = this.DataSource.SecurityPolicy;
 
             if (this.dataAuthMethod is AuthMethod.TOKEN or AuthMethod.USERNAME_PASSWORD)
             {
@@ -147,6 +152,7 @@ public partial class DataSourceERI_V1Dialog : ComponentBase, ISecretId
             AuthMethod = this.dataAuthMethod,
             Username = this.dataUsername,
             Type = DataSourceType.ERI_V1,
+            SecurityPolicy = this.dataSecurityPolicy,
         };
     }
     
@@ -196,6 +202,28 @@ public partial class DataSourceERI_V1Dialog : ComponentBase, ISecretId
 
             this.availableAuthMethods = authSchemes.Data!.Select(n => n.AuthMethod).ToList();
 
+            var loginResult = await client.AuthenticateAsync(this.DataSource, this.RustService, cts.Token);
+            if (!loginResult.Successful)
+            {
+                await this.form.Validate();
+                
+                Array.Resize(ref this.dataIssues, this.dataIssues.Length + 1);
+                this.dataIssues[^1] = loginResult.Message;
+                return;
+            }
+            
+            var securityRequirementsRequest = await client.GetSecurityRequirementsAsync(cts.Token);
+            if (!securityRequirementsRequest.Successful)
+            {
+                await this.form.Validate();
+                
+                Array.Resize(ref this.dataIssues, this.dataIssues.Length + 1);
+                this.dataIssues[^1] = securityRequirementsRequest.Message;
+                return;
+            }
+            
+            this.dataSourceSecurityRequirements = securityRequirementsRequest.Data;
+            
             this.connectionTested = true;
             this.connectionSuccessfulTested = true;
             this.Logger.LogInformation("Connection to the ERI v1 server was successful tested.");

app/MindWork AI Studio/Dialogs/DataSourceLocalDirectoryDialog.razor

@@ -1,3 +1,5 @@
+@using AIStudio.Settings.DataModel
+
 <MudDialog>
     <DialogContent>
         <MudForm @ref="@this.form" @bind-IsValid="@this.dataIsValid" @bind-Errors="@this.dataIssues">
@@ -60,6 +62,13 @@
                     </MudJustifiedText>
                 }
             }
+
+            <MudSelect @bind-Value="@this.dataSecurityPolicy" Text="@this.dataSecurityPolicy.ToSelectionText()" Label="Your security policy" Class="mb-3" OpenIcon="@Icons.Material.Filled.ExpandMore" AdornmentColor="Color.Info" Adornment="Adornment.Start" Validation="@this.dataSourceValidation.ValidateSecurityPolicy">
+                @foreach (var policy in Enum.GetValues<DataSourceSecurity>())
+                {
+                    <MudSelectItem Value="@policy">@policy.ToSelectionText()</MudSelectItem>
+                }
+            </MudSelect>
         </MudForm>
         <Issues IssuesData="@this.dataIssues"/>
     </DialogContent>

app/MindWork AI Studio/Dialogs/DataSourceLocalDirectoryDialog.razor.cs

@@ -42,6 +42,7 @@ public partial class DataSourceLocalDirectoryDialog : ComponentBase
     private bool dataUserAcknowledgedCloudEmbedding;
     private string dataEmbeddingId = string.Empty;
     private string dataPath = string.Empty;
+    private DataSourceSecurity dataSecurityPolicy;
     
     // We get the form reference from Blazor code to validate it manually:
     private MudForm form = null!;
@@ -75,6 +76,7 @@ public partial class DataSourceLocalDirectoryDialog : ComponentBase
             this.dataName = this.DataSource.Name;
             this.dataEmbeddingId = this.DataSource.EmbeddingId;
             this.dataPath = this.DataSource.Path;
+            this.dataSecurityPolicy = this.DataSource.SecurityPolicy;
         }
         
         await base.OnInitializedAsync();
@@ -102,6 +104,7 @@ public partial class DataSourceLocalDirectoryDialog : ComponentBase
         Type = DataSourceType.LOCAL_DIRECTORY,
         EmbeddingId = this.dataEmbeddingId,
         Path = this.dataPath,
+        SecurityPolicy = this.dataSecurityPolicy,
     };
 
     private async Task Store()

app/MindWork AI Studio/Dialogs/DataSourceLocalDirectoryInfoDialog.razor

@@ -1,3 +1,5 @@
+@using AIStudio.Settings.DataModel
+
 <MudDialog>
     <DialogContent>
         <TextInfoLine Icon="@Icons.Material.Filled.Tag" Label="Data source name" Value="@this.DataSource.Name" ClipboardTooltipSubject="the data source name"/>
@@ -31,6 +33,8 @@
             </MudJustifiedText>
         }
 
+        <TextInfoLines Label="Your security policy" MaxLines="3" Value="@this.DataSource.SecurityPolicy.ToInfoText()" Color="@this.DataSource.SecurityPolicy.GetColor()" ClipboardTooltipSubject="your security policy"/>
+
         <TextInfoLine Icon="@Icons.Material.Filled.SquareFoot" Label="Number of files" Value="@this.NumberFilesInDirectory" ClipboardTooltipSubject="the number of files in the directory"/>
         <TextInfoLines Label="Files list" MaxLines="14" Value="@this.directoryFiles.ToString()" ClipboardTooltipSubject="the files list"/>
         @if (this.directorySizeNumFiles > 100)

app/MindWork AI Studio/Dialogs/DataSourceLocalFileDialog.razor

@@ -1,3 +1,4 @@
+@using AIStudio.Settings.DataModel
 <MudDialog>
     <DialogContent>
         <MudForm @ref="@this.form" @bind-IsValid="@this.dataIsValid" @bind-Errors="@this.dataIssues">
@@ -59,6 +60,13 @@
                     </MudJustifiedText>
                 }
             }
+
+            <MudSelect @bind-Value="@this.dataSecurityPolicy" Text="@this.dataSecurityPolicy.ToSelectionText()" Label="Your security policy" Class="mb-3" OpenIcon="@Icons.Material.Filled.ExpandMore" AdornmentColor="Color.Info" Adornment="Adornment.Start" Validation="@this.dataSourceValidation.ValidateSecurityPolicy">
+                @foreach (var policy in Enum.GetValues<DataSourceSecurity>())
+                {
+                    <MudSelectItem Value="@policy">@policy.ToSelectionText()</MudSelectItem>
+                }
+            </MudSelect>
         </MudForm>
         <Issues IssuesData="@this.dataIssues"/>
     </DialogContent>

app/MindWork AI Studio/Dialogs/DataSourceLocalFileDialog.razor.cs

@@ -42,6 +42,7 @@ public partial class DataSourceLocalFileDialog : ComponentBase
     private bool dataUserAcknowledgedCloudEmbedding;
     private string dataEmbeddingId = string.Empty;
     private string dataFilePath = string.Empty;
+    private DataSourceSecurity dataSecurityPolicy;
     
     // We get the form reference from Blazor code to validate it manually:
     private MudForm form = null!;
@@ -75,6 +76,7 @@ public partial class DataSourceLocalFileDialog : ComponentBase
             this.dataName = this.DataSource.Name;
             this.dataEmbeddingId = this.DataSource.EmbeddingId;
             this.dataFilePath = this.DataSource.FilePath;
+            this.dataSecurityPolicy = this.DataSource.SecurityPolicy;
         }
         
         await base.OnInitializedAsync();
@@ -102,6 +104,7 @@ public partial class DataSourceLocalFileDialog : ComponentBase
         Type = DataSourceType.LOCAL_FILE,
         EmbeddingId = this.dataEmbeddingId,
         FilePath = this.dataFilePath,
+        SecurityPolicy = this.dataSecurityPolicy,
     };
     
     private async Task Store()

app/MindWork AI Studio/Dialogs/DataSourceLocalFileInfoDialog.razor

@@ -1,3 +1,5 @@
+@using AIStudio.Settings.DataModel
+
 <MudDialog>
     <DialogContent>
         <TextInfoLine Icon="@Icons.Material.Filled.Tag" Label="Data source name" Value="@this.DataSource.Name" ClipboardTooltipSubject="the data source name"/>
@@ -31,6 +33,7 @@
             </MudJustifiedText>
         }
 
+        <TextInfoLines Label="Your security policy" MaxLines="3" Value="@this.DataSource.SecurityPolicy.ToInfoText()" Color="@this.DataSource.SecurityPolicy.GetColor()" ClipboardTooltipSubject="your security policy"/>
         <TextInfoLine Icon="@Icons.Material.Filled.SquareFoot" Label="File size" Value="@this.FileSize" ClipboardTooltipSubject="the file size"/>
     </DialogContent>
     <DialogActions>

app/MindWork AI Studio/Dialogs/EmbeddingProviderDialog.razor.cs

@@ -1,5 +1,6 @@
 using AIStudio.Provider;
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 using AIStudio.Tools.Validation;
 
 using Microsoft.AspNetCore.Components;

app/MindWork AI Studio/Dialogs/ProviderDialog.razor.cs

@@ -1,11 +1,11 @@
 using AIStudio.Provider;
 using AIStudio.Settings;
+using AIStudio.Tools.Services;
 using AIStudio.Tools.Validation;
 
 using Microsoft.AspNetCore.Components;
 
 using Host = AIStudio.Provider.SelfHosted.Host;
-using RustService = AIStudio.Tools.RustService;
 
 namespace AIStudio.Dialogs;
 
@@ -137,7 +137,9 @@ public partial class ProviderDialog : ComponentBase, ISecretId
         this.SettingsManager.InjectSpellchecking(SPELLCHECK_ATTRIBUTES);
         
         // Load the used instance names:
+        #pragma warning disable MWAIS0001
         this.UsedInstanceNames = this.SettingsManager.ConfigurationData.Providers.Select(x => x.InstanceName.ToLowerInvariant()).ToList();
+        #pragma warning restore MWAIS0001
         
         // When editing, we need to load the data:
         if(this.IsEditing)

app/MindWork AI Studio/Layout/MainLayout.razor.cs

@@ -8,7 +8,6 @@ using Microsoft.AspNetCore.Components;
 using Microsoft.AspNetCore.Components.Routing;
 
 using DialogOptions = AIStudio.Dialogs.DialogOptions;
-using RustService = AIStudio.Tools.RustService;
 
 namespace AIStudio.Layout;
 

app/MindWork AI Studio/MindWork AI Studio.csproj

@@ -47,12 +47,16 @@
     <ItemGroup>
       <PackageReference Include="CodeBeam.MudBlazor.Extensions" Version="7.1.0" />
       <PackageReference Include="HtmlAgilityPack" Version="1.11.72" />
-      <PackageReference Include="Microsoft.Extensions.FileProviders.Embedded" Version="9.0.1" />
+      <PackageReference Include="Microsoft.Extensions.FileProviders.Embedded" Version="9.0.2" />
       <PackageReference Include="MudBlazor" Version="7.16.0" />
       <PackageReference Include="MudBlazor.Markdown" Version="7.14.0" />
       <PackageReference Include="ReverseMarkdown" Version="4.6.0" />
     </ItemGroup>
 
+    <ItemGroup>
+      <ProjectReference Include="..\SourceCodeRules\SourceCodeRules\SourceCodeRules.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false"/>
+    </ItemGroup>
+
     <!-- Read the meta data file -->
     <Target Name="ReadMetaData" BeforeTargets="BeforeBuild">
         <Error Text="The ../../metadata.txt file was not found!" Condition="!Exists('../../metadata.txt')" />

app/MindWork AI Studio/Pages/Settings.razor

@@ -1,5 +1,6 @@
 @attribute [Route(Routes.SETTINGS)]
 @using AIStudio.Components.Settings
+@using AIStudio.Settings.DataModel
 
 <div class="inner-scrolling-context">
     <MudText Typo="Typo.h3" Class="mb-12">Settings</MudText>
@@ -7,8 +8,12 @@
     <InnerScrolling>
         <MudExpansionPanels Class="mb-3" MultiExpansion="@false">
             <SettingsPanelProviders @bind-AvailableLLMProviders="@this.availableLLMProviders" />
-            <SettingsPanelEmbeddings AvailableLLMProvidersFunc="() => this.availableLLMProviders" @bind-AvailableEmbeddingProviders="@this.availableEmbeddingProviders" />
-            <SettingsPanelDataSources AvailableLLMProvidersFunc="() => this.availableLLMProviders" AvailableEmbeddingsFunc="() => this.availableEmbeddingProviders" @bind-AvailableDataSources="@this.availableDataSources" />
+            
+            @if (PreviewFeatures.PRE_RAG_2024.IsEnabled(this.SettingsManager))
+            {
+                <SettingsPanelEmbeddings AvailableLLMProvidersFunc="() => this.availableLLMProviders" @bind-AvailableEmbeddingProviders="@this.availableEmbeddingProviders"/>
+                <SettingsPanelDataSources AvailableLLMProvidersFunc="() => this.availableLLMProviders" AvailableEmbeddingsFunc="() => this.availableEmbeddingProviders" @bind-AvailableDataSources="@this.availableDataSources"/>
+            }
             <SettingsPanelProfiles AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelApp AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelChat AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
@@ -16,7 +21,12 @@
             <SettingsPanelIconFinder AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelTranslation AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelCoding AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
-            <SettingsPanelERIServer AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
+            
+            @if (PreviewFeatures.PRE_RAG_2024.IsEnabled(this.SettingsManager))
+            {
+                <SettingsPanelERIServer AvailableLLMProvidersFunc="() => this.availableLLMProviders"/>
+            }
+            
             <SettingsPanelTextSummarizer AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelAgenda AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelGrammarSpelling AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
@@ -27,6 +37,13 @@
             <SettingsPanelSynonyms AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelMyTasks AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
             <SettingsPanelAssistantBias AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
+            
+            @if (PreviewFeatures.PRE_RAG_2024.IsEnabled(this.SettingsManager))
+            {
+                <SettingsPanelAgentDataSourceSelection AvailableLLMProvidersFunc="() => this.availableLLMProviders"/>
+                <SettingsPanelAgentRetrievalContextValidation AvailableLLMProvidersFunc="() => this.availableLLMProviders"/>
+            }
+            
             <SettingsPanelAgentContentCleaner AvailableLLMProvidersFunc="() => this.availableLLMProviders" />
         </MudExpansionPanels>
     </InnerScrolling>

app/MindWork AI Studio/Pages/Settings.razor.cs

@@ -6,6 +6,9 @@ namespace AIStudio.Pages;
 
 public partial class Settings : ComponentBase, IMessageBusReceiver, IDisposable
 {
+    [Inject]
+    private SettingsManager SettingsManager { get; init; } = null!;
+    
     [Inject]
     private MessageBus MessageBus { get; init; } = null!;
     

app/MindWork AI Studio/Pages/Writer.razor.cs

@@ -106,17 +106,19 @@ public partial class Writer : MSGComponentBase, IAsyncDisposable
         };
         
         var time = DateTimeOffset.Now;
+        var lastUserPrompt = new ContentText
+        {
+            // We use the maximum 160 characters from the end of the text:
+            Text = this.userInput.Length > 160 ? this.userInput[^160..] : this.userInput,
+        };
+        
         this.chatThread.Blocks.Clear();
         this.chatThread.Blocks.Add(new ContentBlock
         {
             Time = time,
             ContentType = ContentType.TEXT,
             Role = ChatRole.USER,
-            Content = new ContentText
-            {
-                // We use the maximum 160 characters from the end of the text:
-                Text = this.userInput.Length > 160 ? this.userInput[^160..] : this.userInput,
-            },
+            Content = lastUserPrompt,
         });
         
         var aiText = new ContentText
@@ -137,7 +139,7 @@ public partial class Writer : MSGComponentBase, IAsyncDisposable
         this.isStreaming = true;
         this.StateHasChanged();
         
-        await aiText.CreateFromProviderAsync(this.providerSettings.CreateProvider(this.Logger), this.SettingsManager, this.providerSettings.Model, this.chatThread);
+        await aiText.CreateFromProviderAsync(this.providerSettings.CreateProvider(this.Logger), this.providerSettings.Model, lastUserPrompt, this.chatThread);
         this.suggestion = aiText.Text;
         
         this.isStreaming = false;

app/MindWork AI Studio/Program.cs

@@ -21,6 +21,7 @@ internal sealed class Program
     public static RustService RUST_SERVICE = null!;
     public static Encryption ENCRYPTION = null!;
     public static string API_TOKEN = null!;
+    public static IServiceProvider SERVICE_PROVIDER = null!;
     
     public static async Task Main(string[] args)
     {
@@ -115,7 +116,10 @@ internal sealed class Program
         builder.Services.AddMudMarkdownClipboardService<MarkdownClipboardService>();
         builder.Services.AddSingleton<SettingsManager>();
         builder.Services.AddSingleton<ThreadSafeRandom>();
+        builder.Services.AddSingleton<DataSourceService>();
         builder.Services.AddTransient<HTMLParser>();
+        builder.Services.AddTransient<AgentDataSourceSelection>();
+        builder.Services.AddTransient<AgentRetrievalContextValidation>();
         builder.Services.AddTransient<AgentTextContentCleaner>();
         builder.Services.AddHostedService<UpdateService>();
         builder.Services.AddHostedService<TemporaryChatService>();
@@ -143,12 +147,22 @@ internal sealed class Program
         // Execute the builder to get the app:
         var app = builder.Build();
 
+        // Get a program logger:
+        var programLogger = app.Services.GetRequiredService<ILogger<Program>>();
+        programLogger.LogInformation("Starting the AI Studio server.");
+        
+        // Store the service provider (DI). We need it later for some classes,
+        // which are not part of the request pipeline:
+        SERVICE_PROVIDER = app.Services;
+        
         // Initialize the encryption service:
+        programLogger.LogInformation("Initializing the encryption service.");
         var encryptionLogger = app.Services.GetRequiredService<ILogger<Encryption>>();
         var encryption = new Encryption(encryptionLogger, secretPassword, secretKeySalt);
         var encryptionInitializer = encryption.Initialize();
 
         // Set the logger for the Rust service:
+        programLogger.LogInformation("Initializing the Rust service.");
         var rustLogger = app.Services.GetRequiredService<ILogger<RustService>>();
         rust.SetLogger(rustLogger);
         rust.SetEncryptor(encryption);
@@ -156,6 +170,7 @@ internal sealed class Program
         RUST_SERVICE = rust;
         ENCRYPTION = encryption;
 
+        programLogger.LogInformation("Initialize internal file system.");
         app.Use(Redirect.HandlerContentAsync);
 
 #if DEBUG
@@ -175,9 +190,21 @@ internal sealed class Program
             .AddInteractiveServerRenderMode();
 
         var serverTask = app.RunAsync();
+        programLogger.LogInformation("Server was started successfully.");
 
         await encryptionInitializer;
         await rust.AppIsReady();
+        programLogger.LogInformation("The AI Studio server is ready.");
+        
+        TaskScheduler.UnobservedTaskException += (sender, taskArgs) =>
+        {
+            programLogger.LogError(taskArgs.Exception, $"Unobserved task exception by sender '{sender ?? "n/a"}'.");
+            taskArgs.SetObserved();
+        };
+        
         await serverTask;
+        
+        RUST_SERVICE.Dispose();
+        programLogger.LogInformation("The AI Studio server was stopped.");
     }
 }

app/MindWork AI Studio/Provider/Anthropic/ProviderAnthropic.cs

@@ -1,3 +1,4 @@
+using System.Net.Http.Headers;
 using System.Runtime.CompilerServices;
 using System.Text;
 using System.Text.Json;
@@ -37,6 +38,7 @@ public sealed class ProviderAnthropic(ILogger logger) : BaseProvider("https://ap
                     ChatRole.USER => "user",
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
+                    ChatRole.RAG => "assistant",
 
                     _ => "user",
                 },
@@ -86,21 +88,15 @@ public sealed class ProviderAnthropic(ILogger logger) : BaseProvider("https://ap
     /// <inheritdoc />
     public override Task<IEnumerable<Model>> GetTextModels(string? apiKeyProvisional = null, CancellationToken token = default)
     {
-        return Task.FromResult(new[]
+        var additionalModels = new[]
         {
-            new Model("claude-3-5-sonnet-latest", "Claude 3.5 Sonnet (latest)"),
-            new Model("claude-3-5-sonnet-20240620", "Claude 3.5 Sonnet (20. June 2024)"),
-            new Model("claude-3-5-sonnet-20241022", "Claude 3.5 Sonnet (22. October 2024)"),
-            
-            new Model("claude-3-5-haiku-latest", "Claude 3.5 Haiku (latest)"),
-            new Model("claude-3-5-heiku-20241022", "Claude 3.5 Haiku (22. October 2024)"),
-            
-            new Model("claude-3-opus-20240229", "Claude 3.0 Opus (29. February 2024)"),
-            new Model("claude-3-opus-latest", "Claude 3.0 Opus (latest)"),
-            
-            new Model("claude-3-sonnet-20240229", "Claude 3.0 Sonnet (29. February 2024)"),
-            new Model("claude-3-haiku-20240307", "Claude 3.0 Haiku (7. March 2024)"),
-        }.AsEnumerable());
+            new Model("claude-3-7-sonnet-latest", "Claude 3.7 Sonnet (Latest)"),
+            new Model("claude-3-5-sonnet-latest", "Claude 3.5 Sonnet (Latest)"),
+            new Model("claude-3-5-haiku-latest", "Claude 3.5 Haiku (Latest)"),
+            new Model("claude-3-opus-latest", "Claude 3 Opus (Latest)"),
+        };
+        
+        return this.LoadModels(token, apiKeyProvisional).ContinueWith(t => t.Result.Concat(additionalModels).OrderBy(x => x.Id).AsEnumerable(), token);
     }
 
     /// <inheritdoc />
@@ -116,4 +112,37 @@ public sealed class ProviderAnthropic(ILogger logger) : BaseProvider("https://ap
     }
 
     #endregion
+    
+    private async Task<IEnumerable<Model>> LoadModels(CancellationToken token, string? apiKeyProvisional = null)
+    {
+        var secretKey = apiKeyProvisional switch
+        {
+            not null => apiKeyProvisional,
+            _ => await RUST_SERVICE.GetAPIKey(this) switch
+            {
+                { Success: true } result => await result.Secret.Decrypt(ENCRYPTION),
+                _ => null,
+            }
+        };
+
+        if (secretKey is null)
+            return [];
+        
+        using var request = new HttpRequestMessage(HttpMethod.Get, "models?limit=100");
+        
+        // Set the authorization header:
+        request.Headers.Add("x-api-key", secretKey);
+
+        // Set the Anthropic version:
+        request.Headers.Add("anthropic-version", "2023-06-01");
+        
+        request.Headers.Authorization = new AuthenticationHeaderValue("Bearer", secretKey);
+
+        using var response = await this.httpClient.SendAsync(request, token);
+        if(!response.IsSuccessStatusCode)
+            return [];
+
+        var modelResponse = await response.Content.ReadFromJsonAsync<ModelsResponse>(JSON_SERIALIZER_OPTIONS, token);
+        return modelResponse.Data;
+    }
 }

app/MindWork AI Studio/Provider/BaseProvider.cs

@@ -4,8 +4,7 @@ using System.Text.Json;
 
 using AIStudio.Chat;
 using AIStudio.Settings;
-
-using RustService = AIStudio.Tools.RustService;
+using AIStudio.Tools.Services;
 
 namespace AIStudio.Provider;
 

app/MindWork AI Studio/Provider/Fireworks/ProviderFireworks.cs

@@ -49,6 +49,7 @@ public class ProviderFireworks(ILogger logger) : BaseProvider("https://api.firew
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
                     ChatRole.SYSTEM => "system",
+                    ChatRole.RAG => "assistant",
 
                     _ => "user",
                 },

app/MindWork AI Studio/Provider/Google/ProviderGoogle.cs

@@ -50,6 +50,7 @@ public class ProviderGoogle(ILogger logger) : BaseProvider("https://generativela
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
                     ChatRole.SYSTEM => "system",
+                    ChatRole.RAG => "assistant",
 
                     _ => "user",
                 },

app/MindWork AI Studio/Provider/Groq/ProviderGroq.cs

@@ -50,6 +50,7 @@ public class ProviderGroq(ILogger logger) : BaseProvider("https://api.groq.com/o
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
                     ChatRole.SYSTEM => "system",
+                    ChatRole.RAG => "assistant",
 
                     _ => "user",
                 },

app/MindWork AI Studio/Provider/Mistral/ProviderMistral.cs

@@ -48,6 +48,7 @@ public sealed class ProviderMistral(ILogger logger) : BaseProvider("https://api.
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
                     ChatRole.SYSTEM => "system",
+                    ChatRole.RAG => "assistant",
 
                     _ => "user",
                 },

app/MindWork AI Studio/Provider/OpenAI/ProviderOpenAI.cs

@@ -76,6 +76,7 @@ public sealed class ProviderOpenAI(ILogger logger) : BaseProvider("https://api.o
                     ChatRole.USER => "user",
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
+                    ChatRole.RAG => "assistant",
                     ChatRole.SYSTEM => systemPromptRole,
 
                     _ => "user",

app/MindWork AI Studio/Provider/SelfHosted/ProviderSelfHosted.cs

@@ -46,6 +46,7 @@ public sealed class ProviderSelfHosted(ILogger logger, Host host, string hostnam
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
                     ChatRole.SYSTEM => "system",
+                    ChatRole.RAG => "assistant",
 
                     _ => "user",
                 },

app/MindWork AI Studio/Provider/X/ProviderX.cs

@@ -50,6 +50,7 @@ public sealed class ProviderX(ILogger logger) : BaseProvider("https://api.x.ai/v
                     ChatRole.AI => "assistant",
                     ChatRole.AGENT => "assistant",
                     ChatRole.SYSTEM => "system",
+                    ChatRole.RAG => "assistant",
 
                     _ => "user",
                 },

app/MindWork AI Studio/Settings/ConfigurationSelectData.cs

@@ -94,6 +94,12 @@ public static class ConfigurationSelectDataFactory
             yield return new(source.GetPreviewDescription(), source);
     }
     
+    public static IEnumerable<ConfigurationSelectData<SendToChatDataSourceBehavior>> GetSendToChatDataSourceBehaviorData()
+    {
+        foreach (var behavior in Enum.GetValues<SendToChatDataSourceBehavior>())
+            yield return new(behavior.Description(), behavior);
+    }
+    
     public static IEnumerable<ConfigurationSelectData<NavBehavior>> GetNavBehaviorData()
     {
         yield return new("Navigation expands on mouse hover", NavBehavior.EXPAND_ON_HOVER);

app/MindWork AI Studio/Settings/DataModel/Data.cs

@@ -74,6 +74,10 @@ public sealed class Data
 
     public DataTextContentCleaner TextContentCleaner { get; init; } = new();
     
+    public DataAgentDataSourceSelection AgentDataSourceSelection { get; init; } = new();
+    
+    public DataAgentRetrievalContextValidation AgentRetrievalContextValidation { get; init; } = new();
+    
     public DataAgenda Agenda { get; init; } = new();
     
     public DataGrammarSpelling GrammarSpelling { get; init; } = new();

app/MindWork AI Studio/Settings/DataModel/DataAgentDataSourceSelection.cs

@@ -0,0 +1,14 @@
+namespace AIStudio.Settings.DataModel;
+
+public sealed class DataAgentDataSourceSelection
+{
+    /// <summary>
+    /// Preselect any data source selection options?
+    /// </summary>
+    public bool PreselectAgentOptions { get; set; }
+    
+    /// <summary>
+    /// Preselect a data source selection provider?
+    /// </summary>
+    public string PreselectedAgentProvider { get; set; } = string.Empty;
+}

app/MindWork AI Studio/Settings/DataModel/DataAgentRetrievalContextValidation.cs

@@ -0,0 +1,24 @@
+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>
+    public bool PreselectAgentOptions { get; set; }
+    
+    /// <summary>
+    /// Preselect a retrieval context validation provider?
+    /// </summary>
+    public string PreselectedAgentProvider { get; set; } = string.Empty;
+
+    /// <summary>
+    /// Configure how many parallel validations to run.
+    /// </summary>
+    public int NumParallelValidations { get; set; } = 3;
+}

app/MindWork AI Studio/Settings/DataModel/DataChat.cs

@@ -17,6 +17,11 @@ public sealed class DataChat
     /// </summary>
     public AddChatProviderBehavior AddChatProviderBehavior { get; set; } = AddChatProviderBehavior.ADDED_CHATS_USE_LATEST_PROVIDER;
 
+    /// <summary>
+    /// Defines the data source behavior when sending assistant results to a chat.
+    /// </summary>
+    public SendToChatDataSourceBehavior SendToChatDataSourceBehavior { get; set; } = SendToChatDataSourceBehavior.NO_DATA_SOURCES;
+
     /// <summary>
     /// Preselect any chat options?
     /// </summary>
@@ -31,6 +36,11 @@ public sealed class DataChat
     /// Preselect a profile?
     /// </summary>
     public string PreselectedProfile { get; set; } = string.Empty;
+    
+    /// <summary>
+    /// Should we preselect data sources options for a created chat?
+    /// </summary>
+    public DataSourceOptions PreselectedDataSourceOptions { get; set; } = new();
 
     /// <summary>
     /// Should we show the latest message after loading? When false, we show the first (aka oldest) message.

app/MindWork AI Studio/Settings/DataModel/DataSourceERI_V1.cs

@@ -1,6 +1,14 @@
 // ReSharper disable InconsistentNaming
 
+using AIStudio.Assistants.ERI;
+using AIStudio.Chat;
+using AIStudio.Tools.ERIClient;
 using AIStudio.Tools.ERIClient.DataModel;
+using AIStudio.Tools.RAG;
+using AIStudio.Tools.Services;
+
+using ChatThread = AIStudio.Chat.ChatThread;
+using ContentType = AIStudio.Tools.ERIClient.DataModel.ContentType;
 
 namespace AIStudio.Settings.DataModel;
 
@@ -36,4 +44,91 @@ public readonly record struct DataSourceERI_V1 : IERIDataSource
 
     /// <inheritdoc />
     public string Username { get; init; } = string.Empty;
+
+    /// <inheritdoc />
+    public DataSourceSecurity SecurityPolicy { get; init; } = DataSourceSecurity.NOT_SPECIFIED;
+    
+    /// <inheritdoc />
+    public ERIVersion Version { get; init; } = ERIVersion.V1;
+    
+    /// <inheritdoc />
+    public async Task<IReadOnlyList<IRetrievalContext>> RetrieveDataAsync(IContent lastPrompt, ChatThread thread, CancellationToken token = default)
+    {
+        // Important: Do not dispose the RustService here, as it is a singleton.
+        var rustService = Program.SERVICE_PROVIDER.GetRequiredService<RustService>();
+        var logger = Program.SERVICE_PROVIDER.GetRequiredService<ILogger<DataSourceERI_V1>>();
+        
+        using var eriClient = ERIClientFactory.Get(this.Version, this)!;
+        var authResponse = await eriClient.AuthenticateAsync(this, rustService, token);
+        if (authResponse.Successful)
+        {
+            var retrievalRequest = new RetrievalRequest
+            {
+                LatestUserPromptType = lastPrompt.ToERIContentType,
+                LatestUserPrompt = lastPrompt switch
+                {
+                    ContentText text => text.Text,
+                    ContentImage image => await image.AsBase64(token),
+                    _ => string.Empty
+                },
+                
+                Thread = await thread.ToERIChatThread(token),
+                MaxMatches = 10,
+                RetrievalProcessId = null, // The ERI server selects the retrieval process when multiple processes are available
+                Parameters = null, // The ERI server selects useful default parameters
+            };
+            
+            var retrievalResponse = await eriClient.ExecuteRetrievalAsync(retrievalRequest, token);
+            if(retrievalResponse is { Successful: true, Data: not null })
+            {
+                //
+                // Next, we have to transform the ERI context back to our generic retrieval context:
+                //
+                var genericRetrievalContexts = new List<IRetrievalContext>(retrievalResponse.Data.Count);
+                foreach (var eriContext in retrievalResponse.Data)
+                {
+                    switch (eriContext.Type)
+                    {
+                        case ContentType.TEXT:
+                            genericRetrievalContexts.Add(new RetrievalTextContext
+                            {
+                                Path = eriContext.Path ?? string.Empty,
+                                Type = eriContext.ToRetrievalContentType(),
+                                Links = eriContext.Links,
+                                Category = eriContext.Type.ToRetrievalContentCategory(),
+                                MatchedText = eriContext.MatchedContent,
+                                DataSourceName = eriContext.Name,
+                                SurroundingContent = eriContext.SurroundingContent,
+                            });
+                            break;
+                        
+                        case ContentType.IMAGE:
+                            genericRetrievalContexts.Add(new RetrievalImageContext
+                            {
+                                Path = eriContext.Path ?? string.Empty,
+                                Type = eriContext.ToRetrievalContentType(),
+                                Links = eriContext.Links,
+                                Source = eriContext.MatchedContent,
+                                Category = eriContext.Type.ToRetrievalContentCategory(),
+                                SourceType = ContentImageSource.BASE64,
+                                DataSourceName = eriContext.Name,
+                            });
+                            break;
+                        
+                        default:
+                            logger.LogWarning($"The ERI context type '{eriContext.Type}' is not supported yet.");
+                            break;
+                    }
+                }
+
+                return genericRetrievalContexts;
+            }
+
+            logger.LogWarning($"Was not able to retrieve data from the ERI data source '{this.Name}'. Message: {retrievalResponse.Message}");
+            return [];
+        }
+
+        logger.LogWarning($"Was not able to authenticate with the ERI data source '{this.Name}'. Message: {authResponse.Message}");
+        return [];
+    }
 }

app/MindWork AI Studio/Settings/DataModel/DataSourceLocalDirectory.cs

@@ -1,3 +1,6 @@
+using AIStudio.Chat;
+using AIStudio.Tools.RAG;
+
 namespace AIStudio.Settings.DataModel;
 
 /// <summary>
@@ -24,6 +27,16 @@ public readonly record struct DataSourceLocalDirectory : IInternalDataSource
     /// <inheritdoc />
     public string EmbeddingId { get; init; } = Guid.Empty.ToString();
     
+    /// <inheritdoc />
+    public DataSourceSecurity SecurityPolicy { get; init; } = DataSourceSecurity.NOT_SPECIFIED;
+    
+    /// <inheritdoc />
+    public Task<IReadOnlyList<IRetrievalContext>> RetrieveDataAsync(IContent lastPrompt, ChatThread thread, CancellationToken token = default)
+    {
+        IReadOnlyList<IRetrievalContext> retrievalContext = new List<IRetrievalContext>();
+        return Task.FromResult(retrievalContext);
+    }
+    
     /// <summary>
     /// The path to the directory.
     /// </summary>

app/MindWork AI Studio/Settings/DataModel/DataSourceLocalFile.cs

@@ -1,3 +1,6 @@
+using AIStudio.Chat;
+using AIStudio.Tools.RAG;
+
 namespace AIStudio.Settings.DataModel;
 
 /// <summary>
@@ -24,6 +27,16 @@ public readonly record struct DataSourceLocalFile : IInternalDataSource
     /// <inheritdoc />
     public string EmbeddingId { get; init; } = Guid.Empty.ToString();
     
+    /// <inheritdoc />
+    public DataSourceSecurity SecurityPolicy { get; init; } = DataSourceSecurity.NOT_SPECIFIED;
+    
+    /// <inheritdoc />
+    public Task<IReadOnlyList<IRetrievalContext>> RetrieveDataAsync(IContent lastPrompt, ChatThread thread, CancellationToken token = default)
+    {
+        IReadOnlyList<IRetrievalContext> retrievalContext = new List<IRetrievalContext>();
+        return Task.FromResult(retrievalContext);
+    }
+    
     /// <summary>
     /// The path to the file.
     /// </summary>

app/MindWork AI Studio/Settings/DataModel/DataSourceOptions.cs

@@ -0,0 +1,65 @@
+namespace AIStudio.Settings.DataModel;
+
+public sealed class DataSourceOptions
+{
+    /// <summary>
+    /// Whether data sources are disabled in this context.
+    /// </summary>
+    public bool DisableDataSources { get; set; } = true;
+    
+    /// <summary>
+    /// Whether the data sources should be selected automatically.
+    /// </summary>
+    /// <remarks>
+    /// When true, the appropriate data sources for the current prompt are
+    /// selected automatically. When false, the user has to select the
+    /// data sources manually.
+    ///
+    /// This setting does not affect the selection of the actual data
+    /// for augmentation.
+    /// </remarks>
+    public bool AutomaticDataSourceSelection { get; set; }
+    
+    /// <summary>
+    /// Whether the retrieved data should be validated for the current prompt.
+    /// </summary>
+    /// <remarks>
+    /// When true, the retrieved data is validated against the current prompt.
+    /// An AI will decide whether the data point is useful for answering the
+    /// prompt or not.
+    /// </remarks>
+    public bool AutomaticValidation { get; set; }
+
+    /// <summary>
+    /// The preselected data source IDs. When these data sources are available
+    /// for the selected provider, they are pre-selected.
+    /// </summary>
+    public List<string> PreselectedDataSourceIds { get; set; } = [];
+
+    /// <summary>
+    /// Returns true when data sources are enabled.
+    /// </summary>
+    /// <returns>True when data sources are enabled.</returns>
+    public bool IsEnabled()
+    {
+        if(this.DisableDataSources)
+            return false;
+        
+        if(this.AutomaticDataSourceSelection)
+            return true;
+        
+        return this.PreselectedDataSourceIds.Count > 0;
+    }
+
+    /// <summary>
+    /// Creates a copy of the current data source options.
+    /// </summary>
+    /// <returns>A copy of the current data source options.</returns>
+    public DataSourceOptions CreateCopy() => new()
+    {
+        DisableDataSources = this.DisableDataSources,
+        AutomaticDataSourceSelection = this.AutomaticDataSourceSelection,
+        AutomaticValidation = this.AutomaticValidation,
+        PreselectedDataSourceIds = [..this.PreselectedDataSourceIds],
+    };
+}

app/MindWork AI Studio/Settings/DataModel/DataSourceSecurity.cs

@@ -0,0 +1,19 @@
+namespace AIStudio.Settings.DataModel;
+
+public enum DataSourceSecurity
+{
+    /// <summary>
+    /// The security of the data source is not specified yet.
+    /// </summary>
+    NOT_SPECIFIED,
+    
+    /// <summary>
+    /// This data can be used with any LLM provider.
+    /// </summary>
+    ALLOW_ANY,
+    
+    /// <summary>
+    /// This data can only be used for self-hosted LLM providers.
+    /// </summary>
+    SELF_HOSTED,
+}

app/MindWork AI Studio/Settings/DataModel/DataSourceSecurityExtensions.cs

@@ -0,0 +1,32 @@
+namespace AIStudio.Settings.DataModel;
+
+public static class DataSourceSecurityExtensions
+{
+    public static string ToSelectionText(this DataSourceSecurity security) => security switch
+    {
+        DataSourceSecurity.NOT_SPECIFIED => "Please select a security policy",
+        
+        DataSourceSecurity.ALLOW_ANY => "This data source can be used with any LLM provider. Your data may be sent to a cloud-based provider.",
+        DataSourceSecurity.SELF_HOSTED => "This data source can only be used with a self-hosted LLM provider. Your data will not be sent to any cloud-based provider.",
+        
+        _ => "Unknown security policy"
+    };
+    
+    public static string ToInfoText(this DataSourceSecurity security) => security switch
+    {
+        DataSourceSecurity.NOT_SPECIFIED => "The security of the data source is not specified yet. You cannot use this data source until you specify a security policy.",
+        
+        DataSourceSecurity.ALLOW_ANY => "This data source can be used with any LLM provider. Your data may be sent to a cloud-based provider.",
+        DataSourceSecurity.SELF_HOSTED => "This data source can only be used with a self-hosted LLM provider. Your data will not be sent to any cloud-based provider.",
+        
+        _ => "Unknown security policy"
+    };
+    
+    public static TextColor GetColor(this DataSourceSecurity security) => security switch
+    {
+        DataSourceSecurity.ALLOW_ANY => TextColor.WARN,
+        DataSourceSecurity.SELF_HOSTED => TextColor.SUCCESS,
+        
+        _ => TextColor.ERROR
+    };
+}

app/MindWork AI Studio/Settings/DataModel/PreviewFeaturesExtensions.cs

@@ -1,6 +1,6 @@
 namespace AIStudio.Settings.DataModel;
 
-public static class PreviewFeatureExtensions
+public static class PreviewFeaturesExtensions
 {
     public static string GetPreviewDescription(this PreviewFeatures feature) => feature switch
     {

app/MindWork AI Studio/Settings/DataModel/SendToChatDataSourceBehavior.cs

@@ -0,0 +1,7 @@
+namespace AIStudio.Settings.DataModel;
+
+public enum SendToChatDataSourceBehavior
+{
+    NO_DATA_SOURCES,
+    APPLY_STANDARD_CHAT_DATA_SOURCE_OPTIONS,
+}

app/MindWork AI Studio/Settings/DataModel/SendToChatDataSourceBehaviorExtensions.cs

@@ -0,0 +1,12 @@
+namespace AIStudio.Settings.DataModel;
+
+public static class SendToChatDataSourceBehaviorExtensions
+{
+    public static string Description(this SendToChatDataSourceBehavior behavior) => behavior switch
+    {
+        SendToChatDataSourceBehavior.NO_DATA_SOURCES => "Use no data sources, when sending an assistant result to a chat",
+        SendToChatDataSourceBehavior.APPLY_STANDARD_CHAT_DATA_SOURCE_OPTIONS => "Apply standard chat data source options, when sending an assistant result to a chat",
+        
+        _ => "Unknown behavior",
+    };
+}

app/MindWork AI Studio/Settings/IDataSource.cs

@@ -1,6 +1,8 @@
 using System.Text.Json.Serialization;
 
+using AIStudio.Chat;
 using AIStudio.Settings.DataModel;
+using AIStudio.Tools.RAG;
 
 namespace AIStudio.Settings;
 
@@ -32,4 +34,18 @@ public interface IDataSource
     /// Which type of data source is this?
     /// </summary>
     public DataSourceType Type { get; init; }
+
+    /// <summary>
+    /// Which data security policy is applied to this data source?
+    /// </summary>
+    public DataSourceSecurity SecurityPolicy { get; init; }
+    
+    /// <summary>
+    /// Perform the data retrieval process.
+    /// </summary>
+    /// <param name="lastPrompt">The last prompt from the chat.</param>
+    /// <param name="thread">The chat thread.</param>
+    /// <param name="token">The cancellation token.</param>
+    /// <returns>The retrieved data context.</returns>
+    public Task<IReadOnlyList<IRetrievalContext>> RetrieveDataAsync(IContent lastPrompt, ChatThread thread, CancellationToken token = default);
 }

app/MindWork AI Studio/Settings/IERIDataSource.cs

@@ -1,3 +1,4 @@
+using AIStudio.Assistants.ERI;
 using AIStudio.Tools.ERIClient.DataModel;
 
 namespace AIStudio.Settings;
@@ -23,4 +24,9 @@ public interface IERIDataSource : IExternalDataSource
     /// The username to use for authentication, when the auth. method is USERNAME_PASSWORD.
     /// </summary>
     public string Username { get; init; }
+    
+    /// <summary>
+    /// The ERI specification to use.
+    /// </summary>
+    public ERIVersion Version { get; init; }
 }

app/MindWork AI Studio/Settings/SettingsManager.cs

@@ -1,3 +1,4 @@
+using System.Diagnostics.CodeAnalysis;
 using System.Text.Json;
 using System.Text.Json.Serialization;
 
@@ -141,7 +142,8 @@ public sealed class SettingsManager(ILogger<SettingsManager> logger)
         return minimumLevel;
     }
     
-    public Provider GetPreselectedProvider(Tools.Components component, string? chatProviderId = null)
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
+    public Provider GetPreselectedProvider(Tools.Components component, string? currentProviderId = null, bool usePreselectionBeforeCurrentProvider = false)
     {
         var minimumLevel = this.GetMinimumConfidenceLevel(component);
         
@@ -149,17 +151,43 @@ public sealed class SettingsManager(ILogger<SettingsManager> logger)
         if (this.ConfigurationData.Providers.Count == 1 && this.ConfigurationData.Providers[0].UsedLLMProvider.GetConfidence(this).Level >= minimumLevel)
             return this.ConfigurationData.Providers[0];
 
-        // When there is a chat provider, and it has a confidence level that is high enough, we return it:
-        if (chatProviderId is not null && !string.IsNullOrWhiteSpace(chatProviderId))
+        // Is there a current provider with a sufficiently high confidence level?
+        Provider currentProvider = default;
+        if (currentProviderId is not null && !string.IsNullOrWhiteSpace(currentProviderId))
         {
-            var chatProvider = this.ConfigurationData.Providers.FirstOrDefault(x => x.Id == chatProviderId);
-            if (chatProvider.UsedLLMProvider.GetConfidence(this).Level >= minimumLevel)
-                return chatProvider;
+            var currentProviderProbe = this.ConfigurationData.Providers.FirstOrDefault(x => x.Id == currentProviderId);
+            if (currentProviderProbe.UsedLLMProvider.GetConfidence(this).Level >= minimumLevel)
+                currentProvider = currentProviderProbe;
         }
         
-        // When there is a component-preselected provider, and it has a confidence level that is high enough, we return it:
-        var preselectedProvider = component.PreselectedProvider(this);
-        if(preselectedProvider != default && preselectedProvider.UsedLLMProvider.GetConfidence(this).Level >= minimumLevel)
+        // Is there a component-preselected provider with a sufficiently high confidence level?
+        Provider preselectedProvider = default;
+        var preselectedProviderProbe = component.PreselectedProvider(this);
+        if(preselectedProviderProbe != default && preselectedProviderProbe.UsedLLMProvider.GetConfidence(this).Level >= minimumLevel)
+            preselectedProvider = preselectedProviderProbe;
+
+        //
+        // Case: The preselected provider should be used before the current provider,
+        //       and the preselected provider is available and has a confidence level
+        //       that is high enough.
+        //
+        if(usePreselectionBeforeCurrentProvider && preselectedProvider != default)
+            return preselectedProvider;
+        
+        //
+        // Case: The current provider is available and has a confidence level that is
+        //       high enough.
+        //
+        if(currentProvider != default)
+            return currentProvider;
+        
+        //
+        // Case: The current provider should be used before the preselected provider,
+        //       but the current provider is not available or does not have a confidence
+        //       level that is high enough. The preselected provider is available and
+        //       has a confidence level that is high enough.
+        //
+        if(preselectedProvider != default)
             return preselectedProvider;
 
         // When there is an app-wide preselected provider, and it has a confidence level that is high enough, we return it:

app/MindWork AI Studio/Tools/AllowedSelectedDataSources.cs

@@ -0,0 +1,13 @@
+using AIStudio.Settings;
+
+namespace AIStudio.Tools;
+
+/// <summary>
+/// Contains both the allowed and selected data sources.
+/// </summary>
+/// <remarks>
+/// The selected data sources are a subset of the allowed data sources.
+/// </remarks>
+/// <param name="AllowedDataSources">The allowed data sources.</param>
+/// <param name="SelectedDataSources">The selected data sources, which are a subset of the allowed data sources.</param>
+public readonly record struct AllowedSelectedDataSources(IReadOnlyList<IDataSource> AllowedDataSources, IReadOnlyList<IDataSource> SelectedDataSources);

app/MindWork AI Studio/Tools/Components.cs

@@ -21,4 +21,8 @@ public enum Components
     
     CHAT,
     APP_SETTINGS,
+    
+    AGENT_TEXT_CONTENT_CLEANER,
+    AGENT_DATA_SOURCE_SELECTION,
+    AGENT_RETRIEVAL_CONTEXT_VALIDATION,
 }

app/MindWork AI Studio/Tools/ComponentsExtensions.cs

@@ -1,3 +1,5 @@
+using System.Diagnostics.CodeAnalysis;
+
 using AIStudio.Provider;
 using AIStudio.Settings;
 
@@ -10,6 +12,11 @@ public static class ComponentsExtensions
         Components.NONE => false,
         Components.ERI_ASSISTANT => false,
         Components.BIAS_DAY_ASSISTANT => false,
+        Components.APP_SETTINGS => false,
+        
+        Components.AGENT_TEXT_CONTENT_CLEANER => false,
+        Components.AGENT_DATA_SOURCE_SELECTION => false,
+        Components.AGENT_RETRIEVAL_CONTEXT_VALIDATION => false,
         
         _ => true,
     };
@@ -75,6 +82,7 @@ public static class ComponentsExtensions
         _ => default,
     };
 
+    [SuppressMessage("Usage", "MWAIS0001:Direct access to `Providers` is not allowed")]
     public static AIStudio.Settings.Provider PreselectedProvider(this Components component, SettingsManager settingsManager) => component switch
     {
         Components.GRAMMAR_SPELLING_ASSISTANT => settingsManager.ConfigurationData.GrammarSpelling.PreselectOptions ? settingsManager.ConfigurationData.Providers.FirstOrDefault(x => x.Id == settingsManager.ConfigurationData.GrammarSpelling.PreselectedProvider) : default,
@@ -94,6 +102,10 @@ public static class ComponentsExtensions
 
         Components.CHAT => settingsManager.ConfigurationData.Chat.PreselectOptions ? settingsManager.ConfigurationData.Providers.FirstOrDefault(x => x.Id == settingsManager.ConfigurationData.Chat.PreselectedProvider) : default,
         
+        Components.AGENT_TEXT_CONTENT_CLEANER => settingsManager.ConfigurationData.TextContentCleaner.PreselectAgentOptions ? settingsManager.ConfigurationData.Providers.FirstOrDefault(x => x.Id == settingsManager.ConfigurationData.TextContentCleaner.PreselectedAgentProvider) : default,
+        Components.AGENT_DATA_SOURCE_SELECTION => settingsManager.ConfigurationData.AgentDataSourceSelection.PreselectAgentOptions ? settingsManager.ConfigurationData.Providers.FirstOrDefault(x => x.Id == settingsManager.ConfigurationData.AgentDataSourceSelection.PreselectedAgentProvider) : default,
+        Components.AGENT_RETRIEVAL_CONTEXT_VALIDATION => settingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectAgentOptions ? settingsManager.ConfigurationData.Providers.FirstOrDefault(x => x.Id == settingsManager.ConfigurationData.AgentRetrievalContextValidation.PreselectedAgentProvider) : default,
+        
         _ => default,
     };
 

app/MindWork AI Studio/Tools/ERIClient/ERIClientV1.cs

@@ -3,6 +3,7 @@ using System.Text.Json;
 
 using AIStudio.Settings;
 using AIStudio.Tools.ERIClient.DataModel;
+using AIStudio.Tools.Services;
 
 namespace AIStudio.Tools.ERIClient;
 
@@ -12,332 +13,465 @@ public class ERIClientV1(string baseAddress) : ERIClientBase(baseAddress), IERIC
 
     public async Task<APIResponse<List<AuthScheme>>> GetAuthMethodsAsync(CancellationToken cancellationToken = default)
     {
-        using var response = await this.httpClient.GetAsync("/auth/methods", cancellationToken);
-        if(!response.IsSuccessStatusCode)
+        try
+        {
+            using var response = await this.httpClient.GetAsync("/auth/methods", cancellationToken);
+            if (!response.IsSuccessStatusCode)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = $"Failed to retrieve the authentication methods: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                };
+            }
+
+            var authMethods = await response.Content.ReadFromJsonAsync<List<AuthScheme>>(JSON_OPTIONS, cancellationToken);
+            if (authMethods is null)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = "Failed to retrieve the authentication methods: the ERI server did not return a valid response."
+                };
+            }
+
+            return new()
+            {
+                Successful = true,
+                Data = authMethods
+            };
+        }
+        catch (TaskCanceledException)
         {
             return new()
             {
                 Successful = false,
-                Message = $"Failed to retrieve the authentication methods: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                Message = "Failed to retrieve the authentication methods: the request was canceled either by the user or due to a timeout."
             };
         }
-
-        var authMethods = await response.Content.ReadFromJsonAsync<List<AuthScheme>>(JSON_OPTIONS, cancellationToken);
-        if(authMethods is null)
+        catch (Exception e)
         {
             return new()
             {
                 Successful = false,
-                Message = "Failed to retrieve the authentication methods: the ERI server did not return a valid response."
+                Message = $"Failed to retrieve the authentication methods due to an exception: {e.Message}"
             };
         }
-
-        return new()
-        {
-            Successful = true,
-            Data = authMethods
-        };
     }
 
     public async Task<APIResponse<AuthResponse>> AuthenticateAsync(IERIDataSource dataSource, RustService rustService, CancellationToken cancellationToken = default)
     {
-        var authMethod = dataSource.AuthMethod;
-        var username = dataSource.Username;
-        switch (dataSource.AuthMethod)
+        try
         {
-            case AuthMethod.NONE:
-                using (var request = new HttpRequestMessage(HttpMethod.Post, $"auth?authMethod={authMethod}"))
-                {
-                    using var noneAuthResponse = await this.httpClient.SendAsync(request, cancellationToken);
-                    if(!noneAuthResponse.IsSuccessStatusCode)
+            var authMethod = dataSource.AuthMethod;
+            var username = dataSource.Username;
+            switch (dataSource.AuthMethod)
+            {
+                case AuthMethod.NONE:
+                    using (var request = new HttpRequestMessage(HttpMethod.Post, $"auth?authMethod={authMethod}"))
                     {
+                        using var noneAuthResponse = await this.httpClient.SendAsync(request, cancellationToken);
+                        if(!noneAuthResponse.IsSuccessStatusCode)
+                        {
+                            return new()
+                            {
+                                Successful = false,
+                                Message = $"Failed to authenticate with the ERI server. Code: {noneAuthResponse.StatusCode}, Reason: {noneAuthResponse.ReasonPhrase}"
+                            };
+                        }
+                    
+                        var noneAuthResult = await noneAuthResponse.Content.ReadFromJsonAsync<AuthResponse>(JSON_OPTIONS, cancellationToken);
+                        if(noneAuthResult == default)
+                        {
+                            return new()
+                            {
+                                Successful = false,
+                                Message = "Failed to authenticate with the ERI server: the response was invalid."
+                            };
+                        }
+                    
+                        this.securityToken = noneAuthResult.Token ?? string.Empty;
                         return new()
                         {
-                            Successful = false,
-                            Message = $"Failed to authenticate with the ERI server. Code: {noneAuthResponse.StatusCode}, Reason: {noneAuthResponse.ReasonPhrase}"
+                            Successful = true,
+                            Data = noneAuthResult
                         };
                     }
-                    
-                    var noneAuthResult = await noneAuthResponse.Content.ReadFromJsonAsync<AuthResponse>(JSON_OPTIONS, cancellationToken);
-                    if(noneAuthResult == default)
-                    {
-                        return new()
-                        {
-                            Successful = false,
-                            Message = "Failed to authenticate with the ERI server: the response was invalid."
-                        };
-                    }
-                    
-                    this.securityToken = noneAuthResult.Token ?? string.Empty;
-                    return new()
-                    {
-                        Successful = true,
-                        Data = noneAuthResult
-                    };
-                }
             
-            case AuthMethod.USERNAME_PASSWORD:
-                var passwordResponse = await rustService.GetSecret(dataSource);
-                if (!passwordResponse.Success)
-                {
-                    return new()
-                    {
-                        Successful = false,
-                        Message = "Failed to retrieve the password."
-                    };
-                }
-
-                var password = await passwordResponse.Secret.Decrypt(Program.ENCRYPTION);
-                using (var request = new HttpRequestMessage(HttpMethod.Post, $"auth?authMethod={authMethod}"))
-                {
-                    // We must send both values inside the header. The username field is named 'user'.
-                    // The password field is named 'password'.
-                    request.Headers.Add("user", username);
-                    request.Headers.Add("password", password);
-
-                    using var usernamePasswordAuthResponse = await this.httpClient.SendAsync(request, cancellationToken);
-                    if(!usernamePasswordAuthResponse.IsSuccessStatusCode)
+                case AuthMethod.USERNAME_PASSWORD:
+                    var passwordResponse = await rustService.GetSecret(dataSource);
+                    if (!passwordResponse.Success)
                     {
                         return new()
                         {
                             Successful = false,
-                            Message = $"Failed to authenticate with the ERI server. Code: {usernamePasswordAuthResponse.StatusCode}, Reason: {usernamePasswordAuthResponse.ReasonPhrase}"
+                            Message = "Failed to retrieve the password."
                         };
                     }
-                    
-                    var usernamePasswordAuthResult = await usernamePasswordAuthResponse.Content.ReadFromJsonAsync<AuthResponse>(JSON_OPTIONS, cancellationToken);
-                    if(usernamePasswordAuthResult == default)
-                    {
-                        return new()
-                        {
-                            Successful = false,
-                            Message = "Failed to authenticate with the server: the response was invalid."
-                        };
-                    }
-                    
-                    this.securityToken = usernamePasswordAuthResult.Token ?? string.Empty;
-                    return new()
-                    {
-                        Successful = true,
-                        Data = usernamePasswordAuthResult
-                    };
-                }
 
-            case AuthMethod.TOKEN:
-                var tokenResponse = await rustService.GetSecret(dataSource);
-                if (!tokenResponse.Success)
-                {
-                    return new()
+                    var password = await passwordResponse.Secret.Decrypt(Program.ENCRYPTION);
+                    using (var request = new HttpRequestMessage(HttpMethod.Post, $"auth?authMethod={authMethod}"))
                     {
-                        Successful = false,
-                        Message = "Failed to retrieve the access token."
-                    };
-                }
+                        // We must send both values inside the header. The username field is named 'user'.
+                        // The password field is named 'password'.
+                        request.Headers.Add("user", username);
+                        request.Headers.Add("password", password);
 
-                var token = await tokenResponse.Secret.Decrypt(Program.ENCRYPTION);
-                using (var request = new HttpRequestMessage(HttpMethod.Post, $"auth?authMethod={authMethod}"))
-                {
-                    request.Headers.Add("Authorization", $"Bearer {token}");
+                        using var usernamePasswordAuthResponse = await this.httpClient.SendAsync(request, cancellationToken);
+                        if(!usernamePasswordAuthResponse.IsSuccessStatusCode)
+                        {
+                            return new()
+                            {
+                                Successful = false,
+                                Message = $"Failed to authenticate with the ERI server. Code: {usernamePasswordAuthResponse.StatusCode}, Reason: {usernamePasswordAuthResponse.ReasonPhrase}"
+                            };
+                        }
                     
-                    using var tokenAuthResponse = await this.httpClient.SendAsync(request, cancellationToken);
-                    if(!tokenAuthResponse.IsSuccessStatusCode)
+                        var usernamePasswordAuthResult = await usernamePasswordAuthResponse.Content.ReadFromJsonAsync<AuthResponse>(JSON_OPTIONS, cancellationToken);
+                        if(usernamePasswordAuthResult == default)
+                        {
+                            return new()
+                            {
+                                Successful = false,
+                                Message = "Failed to authenticate with the server: the response was invalid."
+                            };
+                        }
+                    
+                        this.securityToken = usernamePasswordAuthResult.Token ?? string.Empty;
+                        return new()
+                        {
+                            Successful = true,
+                            Data = usernamePasswordAuthResult
+                        };
+                    }
+
+                case AuthMethod.TOKEN:
+                    var tokenResponse = await rustService.GetSecret(dataSource);
+                    if (!tokenResponse.Success)
                     {
                         return new()
                         {
                             Successful = false,
-                            Message = $"Failed to authenticate with the ERI server. Code: {tokenAuthResponse.StatusCode}, Reason: {tokenAuthResponse.ReasonPhrase}"
+                            Message = "Failed to retrieve the access token."
                         };
                     }
-                    
-                    var tokenAuthResult = await tokenAuthResponse.Content.ReadFromJsonAsync<AuthResponse>(JSON_OPTIONS, cancellationToken);
-                    if(tokenAuthResult == default)
+
+                    var token = await tokenResponse.Secret.Decrypt(Program.ENCRYPTION);
+                    using (var request = new HttpRequestMessage(HttpMethod.Post, $"auth?authMethod={authMethod}"))
                     {
+                        request.Headers.Add("Authorization", $"Bearer {token}");
+                    
+                        using var tokenAuthResponse = await this.httpClient.SendAsync(request, cancellationToken);
+                        if(!tokenAuthResponse.IsSuccessStatusCode)
+                        {
+                            return new()
+                            {
+                                Successful = false,
+                                Message = $"Failed to authenticate with the ERI server. Code: {tokenAuthResponse.StatusCode}, Reason: {tokenAuthResponse.ReasonPhrase}"
+                            };
+                        }
+                    
+                        var tokenAuthResult = await tokenAuthResponse.Content.ReadFromJsonAsync<AuthResponse>(JSON_OPTIONS, cancellationToken);
+                        if(tokenAuthResult == default)
+                        {
+                            return new()
+                            {
+                                Successful = false,
+                                Message = "Failed to authenticate with the ERI server: the response was invalid."
+                            };
+                        }
+                    
+                        this.securityToken = tokenAuthResult.Token ?? string.Empty;
                         return new()
                         {
-                            Successful = false,
-                            Message = "Failed to authenticate with the ERI server: the response was invalid."
+                            Successful = true,
+                            Data = tokenAuthResult
                         };
                     }
-                    
-                    this.securityToken = tokenAuthResult.Token ?? string.Empty;
-                    return new()
-                    {
-                        Successful = true,
-                        Data = tokenAuthResult
-                    };
-                }
                 
-            default:
-                this.securityToken = string.Empty;
-                return new()
-                {
-                    Successful = false,
-                    Message = "The authentication method is not supported yet."
-                };
+                default:
+                    this.securityToken = string.Empty;
+                    return new()
+                    {
+                        Successful = false,
+                        Message = "The authentication method is not supported yet."
+                    };
+            }
+        }
+        catch(TaskCanceledException)
+        {
+            return new()
+            {
+                Successful = false,
+                Message = "Failed to authenticate with the ERI server: the request was canceled either by the user or due to a timeout."
+            };
+        }
+        catch (Exception e)
+        {
+            return new()
+            {
+                Successful = false,
+                Message = $"Failed to authenticate with the ERI server due to an exception: {e.Message}"
+            };
         }
     }
 
     public async Task<APIResponse<DataSourceInfo>> GetDataSourceInfoAsync(CancellationToken cancellationToken = default)
     {
-        using var request = new HttpRequestMessage(HttpMethod.Get, "/dataSource");
-        request.Headers.Add("token", this.securityToken);
+        try
+        {
+            using var request = new HttpRequestMessage(HttpMethod.Get, "/dataSource");
+            request.Headers.Add("token", this.securityToken);
         
-        using var response = await this.httpClient.SendAsync(request, cancellationToken);
-        if(!response.IsSuccessStatusCode)
+            using var response = await this.httpClient.SendAsync(request, cancellationToken);
+            if(!response.IsSuccessStatusCode)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = $"Failed to retrieve the data source information: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                };
+            }
+        
+            var dataSourceInfo = await response.Content.ReadFromJsonAsync<DataSourceInfo>(JSON_OPTIONS, cancellationToken);
+            if(dataSourceInfo == default)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = "Failed to retrieve the data source information: the ERI server did not return a valid response."
+                };
+            }
+        
+            return new()
+            {
+                Successful = true,
+                Data = dataSourceInfo
+            };
+        }
+        catch(TaskCanceledException)
         {
             return new()
             {
                 Successful = false,
-                Message = $"Failed to retrieve the data source information: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                Message = "Failed to retrieve the data source information: the request was canceled either by the user or due to a timeout."
             };
         }
-        
-        var dataSourceInfo = await response.Content.ReadFromJsonAsync<DataSourceInfo>(JSON_OPTIONS, cancellationToken);
-        if(dataSourceInfo == default)
+        catch (Exception e)
         {
             return new()
             {
                 Successful = false,
-                Message = "Failed to retrieve the data source information: the ERI server did not return a valid response."
+                Message = $"Failed to retrieve the data source information due to an exception: {e.Message}"
             };
         }
-        
-        return new()
-        {
-            Successful = true,
-            Data = dataSourceInfo
-        };
     }
 
     public async Task<APIResponse<List<EmbeddingInfo>>> GetEmbeddingInfoAsync(CancellationToken cancellationToken = default)
     {
-        using var request = new HttpRequestMessage(HttpMethod.Get, "/embedding/info");
-        request.Headers.Add("token", this.securityToken);
+        try
+        {
+            using var request = new HttpRequestMessage(HttpMethod.Get, "/embedding/info");
+            request.Headers.Add("token", this.securityToken);
         
-        using var response = await this.httpClient.SendAsync(request, cancellationToken);
-        if(!response.IsSuccessStatusCode)
+            using var response = await this.httpClient.SendAsync(request, cancellationToken);
+            if(!response.IsSuccessStatusCode)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = $"Failed to retrieve the embedding information: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                };
+            }
+        
+            var embeddingInfo = await response.Content.ReadFromJsonAsync<List<EmbeddingInfo>>(JSON_OPTIONS, cancellationToken);
+            if(embeddingInfo is null)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = "Failed to retrieve the embedding information: the ERI server did not return a valid response."
+                };
+            }
+        
+            return new()
+            {
+                Successful = true,
+                Data = embeddingInfo
+            };
+        }
+        catch(TaskCanceledException)
         {
             return new()
             {
                 Successful = false,
-                Message = $"Failed to retrieve the embedding information: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                Message = "Failed to retrieve the embedding information: the request was canceled either by the user or due to a timeout."
             };
         }
-        
-        var embeddingInfo = await response.Content.ReadFromJsonAsync<List<EmbeddingInfo>>(JSON_OPTIONS, cancellationToken);
-        if(embeddingInfo is null)
+        catch (Exception e)
         {
             return new()
             {
                 Successful = false,
-                Message = "Failed to retrieve the embedding information: the ERI server did not return a valid response."
+                Message = $"Failed to retrieve the embedding information due to an exception: {e.Message}"
             };
         }
-        
-        return new()
-        {
-            Successful = true,
-            Data = embeddingInfo
-        };
     }
 
     public async Task<APIResponse<List<RetrievalInfo>>> GetRetrievalInfoAsync(CancellationToken cancellationToken = default)
     {
-        using var request = new HttpRequestMessage(HttpMethod.Get, "/retrieval/info");
-        request.Headers.Add("token", this.securityToken);
+        try
+        {
+            using var request = new HttpRequestMessage(HttpMethod.Get, "/retrieval/info");
+            request.Headers.Add("token", this.securityToken);
         
-        using var response = await this.httpClient.SendAsync(request, cancellationToken);
-        if(!response.IsSuccessStatusCode)
+            using var response = await this.httpClient.SendAsync(request, cancellationToken);
+            if(!response.IsSuccessStatusCode)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = $"Failed to retrieve the retrieval information: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                };
+            }
+        
+            var retrievalInfo = await response.Content.ReadFromJsonAsync<List<RetrievalInfo>>(JSON_OPTIONS, cancellationToken);
+            if(retrievalInfo is null)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = "Failed to retrieve the retrieval information: the ERI server did not return a valid response."
+                };
+            }
+        
+            return new()
+            {
+                Successful = true,
+                Data = retrievalInfo
+            };
+        }
+        catch(TaskCanceledException)
         {
             return new()
             {
                 Successful = false,
-                Message = $"Failed to retrieve the retrieval information: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                Message = "Failed to retrieve the retrieval information: the request was canceled either by the user or due to a timeout."
             };
         }
-        
-        var retrievalInfo = await response.Content.ReadFromJsonAsync<List<RetrievalInfo>>(JSON_OPTIONS, cancellationToken);
-        if(retrievalInfo is null)
+        catch (Exception e)
         {
             return new()
             {
                 Successful = false,
-                Message = "Failed to retrieve the retrieval information: the ERI server did not return a valid response."
+                Message = $"Failed to retrieve the retrieval information due to an exception: {e.Message}"
             };
         }
-        
-        return new()
-        {
-            Successful = true,
-            Data = retrievalInfo
-        };
     }
 
     public async Task<APIResponse<List<Context>>> ExecuteRetrievalAsync(RetrievalRequest request, CancellationToken cancellationToken = default)
     {
-        using var requestMessage = new HttpRequestMessage(HttpMethod.Post, "/retrieval");
-        requestMessage.Headers.Add("token", this.securityToken);
+        try
+        {
+            using var requestMessage = new HttpRequestMessage(HttpMethod.Post, "/retrieval");
+            requestMessage.Headers.Add("token", this.securityToken);
 
-        using var content = new StringContent(JsonSerializer.Serialize(request, JSON_OPTIONS), Encoding.UTF8, "application/json");
-        requestMessage.Content = content;
+            using var content = new StringContent(JsonSerializer.Serialize(request, JSON_OPTIONS), Encoding.UTF8, "application/json");
+            requestMessage.Content = content;
         
-        using var response = await this.httpClient.SendAsync(requestMessage, cancellationToken);
-        if(!response.IsSuccessStatusCode)
+            using var response = await this.httpClient.SendAsync(requestMessage, cancellationToken);
+            if(!response.IsSuccessStatusCode)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = $"Failed to execute the retrieval request: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                };
+            }
+        
+            var contexts = await response.Content.ReadFromJsonAsync<List<Context>>(JSON_OPTIONS, cancellationToken);
+            if(contexts is null)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = "Failed to execute the retrieval request: the ERI server did not return a valid response."
+                };
+            }
+        
+            return new()
+            {
+                Successful = true,
+                Data = contexts
+            };
+        }
+        catch(TaskCanceledException)
         {
             return new()
             {
                 Successful = false,
-                Message = $"Failed to execute the retrieval request: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                Message = "Failed to execute the retrieval request: the request was canceled either by the user or due to a timeout."
             };
         }
-        
-        var contexts = await response.Content.ReadFromJsonAsync<List<Context>>(JSON_OPTIONS, cancellationToken);
-        if(contexts is null)
+        catch (Exception e)
         {
             return new()
             {
                 Successful = false,
-                Message = "Failed to execute the retrieval request: the ERI server did not return a valid response."
+                Message = $"Failed to execute the retrieval request due to an exception: {e.Message}"
             };
         }
-        
-        return new()
-        {
-            Successful = true,
-            Data = contexts
-        };
     }
 
     public async Task<APIResponse<SecurityRequirements>> GetSecurityRequirementsAsync(CancellationToken cancellationToken = default)
     {
-        using var request = new HttpRequestMessage(HttpMethod.Get, "/security/requirements");
-        request.Headers.Add("token", this.securityToken);
+        try
+        {
+            using var request = new HttpRequestMessage(HttpMethod.Get, "/security/requirements");
+            request.Headers.Add("token", this.securityToken);
         
-        using var response = await this.httpClient.SendAsync(request, cancellationToken);
-        if(!response.IsSuccessStatusCode)
+            using var response = await this.httpClient.SendAsync(request, cancellationToken);
+            if(!response.IsSuccessStatusCode)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = $"Failed to retrieve the security requirements: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                };
+            }
+        
+            var securityRequirements = await response.Content.ReadFromJsonAsync<SecurityRequirements>(JSON_OPTIONS, cancellationToken);
+            if(securityRequirements == default)
+            {
+                return new()
+                {
+                    Successful = false,
+                    Message = "Failed to retrieve the security requirements: the ERI server did not return a valid response."
+                };
+            }
+        
+            return new()
+            {
+                Successful = true,
+                Data = securityRequirements
+            };
+        }
+        catch(TaskCanceledException)
         {
             return new()
             {
                 Successful = false,
-                Message = $"Failed to retrieve the security requirements: there was an issue communicating with the ERI server. Code: {response.StatusCode}, Reason: {response.ReasonPhrase}"
+                Message = "Failed to retrieve the security requirements: the request was canceled either by the user or due to a timeout."
             };
         }
-        
-        var securityRequirements = await response.Content.ReadFromJsonAsync<SecurityRequirements>(JSON_OPTIONS, cancellationToken);
-        if(securityRequirements == default)
+        catch (Exception e)
         {
             return new()
             {
                 Successful = false,
-                Message = "Failed to retrieve the security requirements: the ERI server did not return a valid response."
+                Message = $"Failed to retrieve the security requirements due to an exception: {e.Message}"
             };
         }
-        
-        return new()
-        {
-            Successful = true,
-            Data = securityRequirements
-        };
     }
 
     #endregion

app/MindWork AI Studio/Tools/ERIClient/IERIClient.cs

@@ -1,5 +1,6 @@
 using AIStudio.Settings;
 using AIStudio.Tools.ERIClient.DataModel;
+using AIStudio.Tools.Services;
 
 namespace AIStudio.Tools.ERIClient;
 

app/MindWork AI Studio/Tools/Event.cs

@@ -23,6 +23,9 @@ public enum Event
     WORKSPACE_LOADED_CHAT_CHANGED,
     WORKSPACE_TOGGLE_OVERLAY,
     
+    // RAG events:
+    RAG_AUTO_DATA_SOURCES_SELECTED,
+    
     // Send events:
     SEND_TO_GRAMMAR_SPELLING_ASSISTANT,
     SEND_TO_ICON_FINDER_ASSISTANT,