refactored the classes and functions

app/MindWork AI Studio/Tools/Services/ArbitraryFileDataSegment.cs

@@ -0,0 +1,3 @@
+namespace AIStudio.Tools.Services;
+
+public sealed record ArbitraryFileDataSegment(string Content, int TokenCount);

app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingFailure.cs

@@ -0,0 +1,3 @@
+namespace AIStudio.Tools.Services;
+
+public sealed record DataSourceEmbeddingFailure(string FilePath, string Reason);

app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingManifest.cs

@@ -0,0 +1,14 @@
+namespace AIStudio.Tools.Services;
+
+public sealed class DataSourceEmbeddingManifest
+{
+    public string EmbeddingProviderId { get; set; } = string.Empty;
+
+    public string EmbeddingSignature { get; set; } = string.Empty;
+
+    public string SourceHash { get; set; } = string.Empty;
+
+    public int VectorSize { get; set; }
+
+    public Dictionary<string, EmbeddedFileRecord> Files { get; init; } = new(StringComparer.OrdinalIgnoreCase);
+}

app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingModels.cs

@@ -1,97 +0,0 @@
-using AIStudio.Settings.DataModel;
-using AIStudio.Tools.PluginSystem;
-
-namespace AIStudio.Tools.Services;
-
-public sealed record DataSourceEmbeddingOverview(
-    bool IsVisible,
-    DataSourceEmbeddingState State,
-    int IndexedFiles,
-    int TotalFiles,
-    int FailedFiles);
-
-public enum DataSourceEmbeddingState
-{
-    IDLE,
-    QUEUED,
-    RUNNING,
-    COMPLETED,
-    FAILED,
-}
-
-public sealed record DataSourceEmbeddingStatus(
-    string DataSourceId,
-    string DataSourceName,
-    DataSourceType DataSourceType,
-    DataSourceEmbeddingState State,
-    int TotalFiles,
-    int IndexedFiles,
-    int FailedFiles,
-    string CurrentFile,
-    string LastError,
-    IReadOnlyList<DataSourceEmbeddingFailure> Failures)
-{
-    private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(DataSourceEmbeddingService).Namespace, nameof(DataSourceEmbeddingService));
-
-    public int ProgressPercent => this.TotalFiles <= 0 ? 0 : Math.Clamp((int)Math.Round(this.IndexedFiles * 100d / this.TotalFiles), 0, 100);
-
-    public string StateLabel => this.State switch
-    {
-        DataSourceEmbeddingState.QUEUED => TB("Queued"),
-        DataSourceEmbeddingState.RUNNING => TB("Running"),
-        DataSourceEmbeddingState.COMPLETED => TB("Completed"),
-        DataSourceEmbeddingState.FAILED => TB("Needs attention"),
-        _ => TB("Idle")
-    };
-
-    public int SortOrder => this.State switch
-    {
-        DataSourceEmbeddingState.RUNNING => 0,
-        DataSourceEmbeddingState.QUEUED => 1,
-        DataSourceEmbeddingState.FAILED => 2,
-        DataSourceEmbeddingState.COMPLETED => 3,
-        _ => 4,
-    };
-}
-
-public sealed record DataSourceEmbeddingFailure(
-    string FilePath,
-    string Reason);
-
-public sealed class FileEnumerationResult
-{
-    public List<FileInfo> Files { get; } = [];
-
-    public List<DataSourceEmbeddingFailure> Failures { get; } = [];
-
-    public int FailedFiles { get; set; }
-
-    public string LastError { get; set; } = string.Empty;
-
-    public void AddFailure(string filePath, string reason)
-    {
-        this.Failures.Add(new DataSourceEmbeddingFailure(filePath, reason));
-        this.FailedFiles = this.Failures.Count;
-        this.LastError = reason;
-    }
-}
-
-public sealed class DataSourceEmbeddingManifest
-{
-    public string EmbeddingProviderId { get; set; } = string.Empty;
-
-    public string EmbeddingSignature { get; set; } = string.Empty;
-
-    public string SourceHash { get; set; } = string.Empty;
-
-    public int VectorSize { get; set; }
-
-    public Dictionary<string, EmbeddedFileRecord> Files { get; init; } = new(StringComparer.OrdinalIgnoreCase);
-}
-
-public sealed record EmbeddedFileRecord(
-    string Fingerprint,
-    long FileSize,
-    DateTimeOffset LastWriteUtc,
-    DateTimeOffset EmbeddedAtUtc,
-    int ChunkCount);

app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingOverview.cs

@@ -0,0 +1,3 @@
+namespace AIStudio.Tools.Services;
+
+public sealed record DataSourceEmbeddingOverview(bool IsVisible, DataSourceEmbeddingState State, int IndexedFiles, int TotalFiles, int FailedFiles);

app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingService.cs

@@ -13,7 +13,9 @@ using AIStudio.Tools.PluginSystem;
 
 namespace AIStudio.Tools.Services;
 
-public sealed partial class DataSourceEmbeddingService(SettingsManager settingsManager, RustService rustService, DatabaseClientProvider databaseClientProvider, ILogger<DataSourceEmbeddingService> logger)
+public sealed partial class DataSourceEmbeddingService(
+    SettingsManager settingsManager, RustService rustService, DatabaseClientProvider databaseClientProvider,
+    ILogger<DataSourceEmbeddingService> logger)
     : BackgroundService
 {
     private const int VECTOR_STORE_OPTIMIZATION_CHUNK_THRESHOLD = 100_000;

app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingState.cs

@@ -0,0 +1,10 @@
+namespace AIStudio.Tools.Services;
+
+public enum DataSourceEmbeddingState
+{
+    IDLE,
+    QUEUED,
+    RUNNING,
+    COMPLETED,
+    FAILED,
+}

app/MindWork AI Studio/Tools/Services/DataSourceEmbeddingStatus.cs

@@ -0,0 +1,39 @@
+using AIStudio.Settings.DataModel;
+using AIStudio.Tools.PluginSystem;
+
+namespace AIStudio.Tools.Services;
+
+public sealed record DataSourceEmbeddingStatus(
+    string DataSourceId,
+    string DataSourceName,
+    DataSourceType DataSourceType,
+    DataSourceEmbeddingState State,
+    int TotalFiles,
+    int IndexedFiles,
+    int FailedFiles,
+    string CurrentFile,
+    string LastError,
+    IReadOnlyList<DataSourceEmbeddingFailure> Failures)
+{
+    private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(DataSourceEmbeddingService).Namespace, nameof(DataSourceEmbeddingService));
+
+    public int ProgressPercent => this.TotalFiles <= 0 ? 0 : Math.Clamp((int)Math.Round(this.IndexedFiles * 100d / this.TotalFiles), 0, 100);
+
+    public string StateLabel => this.State switch
+    {
+        DataSourceEmbeddingState.QUEUED => TB("Queued"),
+        DataSourceEmbeddingState.RUNNING => TB("Running"),
+        DataSourceEmbeddingState.COMPLETED => TB("Completed"),
+        DataSourceEmbeddingState.FAILED => TB("Needs attention"),
+        _ => TB("Idle")
+    };
+
+    public int SortOrder => this.State switch
+    {
+        DataSourceEmbeddingState.RUNNING => 0,
+        DataSourceEmbeddingState.QUEUED => 1,
+        DataSourceEmbeddingState.FAILED => 2,
+        DataSourceEmbeddingState.COMPLETED => 3,
+        _ => 4,
+    };
+}

app/MindWork AI Studio/Tools/Services/DataSourceLocalRetrievalService.cs

@@ -11,9 +11,7 @@ using AIStudio.Tools.Rust;
 namespace AIStudio.Tools.Services;
 
 public sealed class DataSourceLocalRetrievalService(
-    SettingsManager settingsManager,
-    RustService rustService,
-    DatabaseClientProvider databaseClientProvider,
+    SettingsManager settingsManager, RustService rustService, DatabaseClientProvider databaseClientProvider,
     ILogger<DataSourceLocalRetrievalService> logger)
 {
     private enum RetrievalChannel

app/MindWork AI Studio/Tools/Services/EmbeddedFileRecord.cs

@@ -0,0 +1,3 @@
+namespace AIStudio.Tools.Services;
+
+public sealed record EmbeddedFileRecord(string Fingerprint, long FileSize, DateTimeOffset LastWriteUtc, DateTimeOffset EmbeddedAtUtc, int ChunkCount);

app/MindWork AI Studio/Tools/Services/FileEnumerationResult.cs

@@ -0,0 +1,19 @@
+namespace AIStudio.Tools.Services;
+
+public sealed class FileEnumerationResult
+{
+    public List<FileInfo> Files { get; } = [];
+
+    public List<DataSourceEmbeddingFailure> Failures { get; } = [];
+
+    public int FailedFiles { get; set; }
+
+    public string LastError { get; set; } = string.Empty;
+
+    public void AddFailure(string filePath, string reason)
+    {
+        this.Failures.Add(new DataSourceEmbeddingFailure(filePath, reason));
+        this.FailedFiles = this.Failures.Count;
+        this.LastError = reason;
+    }
+}

app/MindWork AI Studio/Tools/Services/RustService.Retrieval.cs

@@ -7,8 +7,6 @@ using AIStudio.Tools.Rust;
 
 namespace AIStudio.Tools.Services;
 
-public sealed record ArbitraryFileDataSegment(string Content, int TokenCount);
-
 public sealed partial class RustService
 {
     public async Task<string> ReadArbitraryFileData(string path, int maxChunks, bool extractImages = false)