refactored the classes and functions

app/MindWork AI Studio/Tools/Databases/IndexStore/EmbeddingStateChunk.cs

@@ -0,0 +1,3 @@
+namespace AIStudio.Tools.Databases.IndexStore;
+
+public sealed record EmbeddingStateChunk(string ChunkId, string ParentFileId, int? PageNumber, int ChunkIndex, string ChunkText, DateTimeOffset EmbeddedAtUtc);

app/MindWork AI Studio/Tools/Databases/IndexStore/EmbeddingStateChunkEntity.cs

@@ -0,0 +1,20 @@
+namespace AIStudio.Tools.Databases.IndexStore;
+
+internal sealed class EmbeddingStateChunkEntity
+{
+    public int Id { get; set; }
+
+    public string ChunkId { get; set; } = string.Empty;
+
+    public string ParentFileId { get; set; } = string.Empty;
+
+    public int? PageNumber { get; set; }
+
+    public int ChunkIndex { get; set; }
+
+    public string ChunkText { get; set; } = string.Empty;
+
+    public DateTimeOffset EmbeddedAtUtc { get; set; }
+
+    public EmbeddingStateFileEntity? File { get; set; }
+}

app/MindWork AI Studio/Tools/Databases/IndexStore/EmbeddingStateDataSourceEntity.cs

@@ -0,0 +1,22 @@
+namespace AIStudio.Tools.Databases.IndexStore;
+
+internal sealed class EmbeddingStateDataSourceEntity
+{
+    public string DataSourceId { get; set; } = string.Empty;
+
+    public string DataSourceName { get; set; } = string.Empty;
+
+    public string DataSourceType { get; set; } = string.Empty;
+
+    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 DateTimeOffset UpdatedAtUtc { get; set; }
+
+    public List<EmbeddingStateFileEntity> Files { get; set; } = [];
+}

app/MindWork AI Studio/Tools/Databases/IndexStore/EmbeddingStateFile.cs

@@ -0,0 +1,16 @@
+namespace AIStudio.Tools.Databases.IndexStore;
+
+public sealed record EmbeddingStateFile(
+    string ParentFileId,
+    string AbsolutePath,
+    string FileName,
+    string RelativePath,
+    string FileType,
+    string Fingerprint,
+    long FileSize,
+    DateTimeOffset CreationUtc,
+    DateTimeOffset LastWriteUtc,
+    DateTimeOffset EmbeddedAtUtc,
+    int ChunkCount,
+    string ConfidenceLevel,
+    int ConfidenceLevelRank);

app/MindWork AI Studio/Tools/Databases/IndexStore/EmbeddingStateFileEntity.cs

@@ -0,0 +1,36 @@
+namespace AIStudio.Tools.Databases.IndexStore;
+
+internal sealed class EmbeddingStateFileEntity
+{
+    public string ParentFileId { get; set; } = string.Empty;
+
+    public string DataSourceId { get; set; } = string.Empty;
+
+    public string AbsolutePath { get; set; } = string.Empty;
+
+    public string FileName { get; set; } = string.Empty;
+
+    public string RelativePath { get; set; } = string.Empty;
+
+    public string FileType { get; set; } = string.Empty;
+
+    public string Fingerprint { get; set; } = string.Empty;
+
+    public long FileSize { get; set; }
+
+    public DateTimeOffset CreationUtc { get; set; }
+
+    public DateTimeOffset LastWriteUtc { get; set; }
+
+    public DateTimeOffset EmbeddedAtUtc { get; set; }
+
+    public int ChunkCount { get; set; }
+
+    public string ConfidenceLevel { get; set; } = string.Empty;
+
+    public int ConfidenceLevelRank { get; set; }
+
+    public EmbeddingStateDataSourceEntity? DataSource { get; set; }
+
+    public List<EmbeddingStateChunkEntity> Chunks { get; set; } = [];
+}

app/MindWork AI Studio/Tools/Databases/IndexStore/IndexStoreClient.cs

@@ -30,49 +30,3 @@ public abstract class IndexStoreClient(string name, string path) : DatabaseClien
 
     public abstract Task DeleteDataSourceAsync(string dataSourceId, CancellationToken token);
 }
-
-public sealed record EmbeddingStateFile(
-    string ParentFileId,
-    string AbsolutePath,
-    string FileName,
-    string RelativePath,
-    string FileType,
-    string Fingerprint,
-    long FileSize,
-    DateTimeOffset CreationUtc,
-    DateTimeOffset LastWriteUtc,
-    DateTimeOffset EmbeddedAtUtc,
-    int ChunkCount,
-    string ConfidenceLevel,
-    int ConfidenceLevelRank);
-
-public sealed record EmbeddingStateChunk(
-    string ChunkId,
-    string ParentFileId,
-    int? PageNumber,
-    int ChunkIndex,
-    string ChunkText,
-    DateTimeOffset EmbeddedAtUtc);
-
-public sealed record IndexStoreSearchResult(
-    string ChunkId,
-    string ParentFileId,
-    string DataSourceId,
-    string DataSourceName,
-    string DataSourceType,
-    string AbsolutePath,
-    string FileName,
-    string RelativePath,
-    string FileType,
-    int? PageNumber,
-    int ChunkIndex,
-    string ChunkText,
-    double Score,
-    string Fingerprint,
-    long FileSize,
-    DateTimeOffset CreationUtc,
-    DateTimeOffset LastWriteUtc,
-    DateTimeOffset EmbeddedAtUtc,
-    int ChunkCount,
-    string ConfidenceLevel,
-    int ConfidenceLevelRank);

app/MindWork AI Studio/Tools/Databases/IndexStore/IndexStoreDateTimeOffset.cs

@@ -0,0 +1,18 @@
+using System.Globalization;
+
+namespace AIStudio.Tools.Databases.IndexStore;
+
+internal static class IndexStoreDateTimeOffset
+{
+    public static string ToUtcText(DateTimeOffset dateTime)
+    {
+        return dateTime.ToUniversalTime().ToString("O", CultureInfo.InvariantCulture);
+    }
+
+    public static DateTimeOffset ParseUtc(string value)
+    {
+        return DateTimeOffset.TryParse(value, CultureInfo.InvariantCulture, DateTimeStyles.None, out var dateTime)
+            ? dateTime.ToUniversalTime()
+            : DateTimeOffset.UnixEpoch;
+    }
+}

app/MindWork AI Studio/Tools/Databases/IndexStore/IndexStoreDateTimeOffsetConverter.cs

@@ -0,0 +1,7 @@
+using Microsoft.EntityFrameworkCore.Storage.ValueConversion;
+
+namespace AIStudio.Tools.Databases.IndexStore;
+
+internal sealed class IndexStoreDateTimeOffsetConverter() : ValueConverter<DateTimeOffset, string>(
+    value => IndexStoreDateTimeOffset.ToUtcText(value),
+    value => IndexStoreDateTimeOffset.ParseUtc(value));

app/MindWork AI Studio/Tools/Databases/IndexStore/IndexStoreDbContext.cs

@@ -1,8 +1,5 @@
-using System.Globalization;
-
 using Microsoft.Data.Sqlite;
 using Microsoft.EntityFrameworkCore;
-using Microsoft.EntityFrameworkCore.Storage.ValueConversion;
 
 namespace AIStudio.Tools.Databases.IndexStore;
 
@@ -117,142 +114,3 @@ internal sealed class IndexStoreDbContext(DbContextOptions<IndexStoreDbContext>
         DefaultTimeout = 30,
     }.ToString();
 }
-
-internal sealed class EmbeddingStateDataSourceEntity
-{
-    public string DataSourceId { get; set; } = string.Empty;
-
-    public string DataSourceName { get; set; } = string.Empty;
-
-    public string DataSourceType { get; set; } = string.Empty;
-
-    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 DateTimeOffset UpdatedAtUtc { get; set; }
-
-    public List<EmbeddingStateFileEntity> Files { get; set; } = [];
-}
-
-internal sealed class EmbeddingStateFileEntity
-{
-    public string ParentFileId { get; set; } = string.Empty;
-
-    public string DataSourceId { get; set; } = string.Empty;
-
-    public string AbsolutePath { get; set; } = string.Empty;
-
-    public string FileName { get; set; } = string.Empty;
-
-    public string RelativePath { get; set; } = string.Empty;
-
-    public string FileType { get; set; } = string.Empty;
-
-    public string Fingerprint { get; set; } = string.Empty;
-
-    public long FileSize { get; set; }
-
-    public DateTimeOffset CreationUtc { get; set; }
-
-    public DateTimeOffset LastWriteUtc { get; set; }
-
-    public DateTimeOffset EmbeddedAtUtc { get; set; }
-
-    public int ChunkCount { get; set; }
-
-    public string ConfidenceLevel { get; set; } = string.Empty;
-
-    public int ConfidenceLevelRank { get; set; }
-
-    public EmbeddingStateDataSourceEntity? DataSource { get; set; }
-
-    public List<EmbeddingStateChunkEntity> Chunks { get; set; } = [];
-}
-
-internal sealed class EmbeddingStateChunkEntity
-{
-    public int Id { get; set; }
-
-    public string ChunkId { get; set; } = string.Empty;
-
-    public string ParentFileId { get; set; } = string.Empty;
-
-    public int? PageNumber { get; set; }
-
-    public int ChunkIndex { get; set; }
-
-    public string ChunkText { get; set; } = string.Empty;
-
-    public DateTimeOffset EmbeddedAtUtc { get; set; }
-
-    public EmbeddingStateFileEntity? File { get; set; }
-}
-
-internal sealed class IndexStoreSearchResultEntity
-{
-    public string ChunkId { get; set; } = string.Empty;
-
-    public string ParentFileId { get; set; } = string.Empty;
-
-    public string DataSourceId { get; set; } = string.Empty;
-
-    public string DataSourceName { get; set; } = string.Empty;
-
-    public string DataSourceType { get; set; } = string.Empty;
-
-    public string AbsolutePath { get; set; } = string.Empty;
-
-    public string FileName { get; set; } = string.Empty;
-
-    public string RelativePath { get; set; } = string.Empty;
-
-    public string FileType { get; set; } = string.Empty;
-
-    public int? PageNumber { get; set; }
-
-    public int ChunkIndex { get; set; }
-
-    public string ChunkText { get; set; } = string.Empty;
-
-    public double Score { get; set; }
-
-    public string Fingerprint { get; set; } = string.Empty;
-
-    public long FileSize { get; set; }
-
-    public DateTimeOffset CreationUtc { get; set; }
-
-    public DateTimeOffset LastWriteUtc { get; set; }
-
-    public DateTimeOffset EmbeddedAtUtc { get; set; }
-
-    public int ChunkCount { get; set; }
-
-    public string ConfidenceLevel { get; set; } = string.Empty;
-
-    public int ConfidenceLevelRank { get; set; }
-}
-
-internal sealed class IndexStoreDateTimeOffsetConverter() : ValueConverter<DateTimeOffset, string>(
-    value => IndexStoreDateTimeOffset.ToUtcText(value),
-    value => IndexStoreDateTimeOffset.ParseUtc(value));
-
-internal static class IndexStoreDateTimeOffset
-{
-    public static string ToUtcText(DateTimeOffset dateTime)
-    {
-        return dateTime.ToUniversalTime().ToString("O", CultureInfo.InvariantCulture);
-    }
-
-    public static DateTimeOffset ParseUtc(string value)
-    {
-        return DateTimeOffset.TryParse(value, CultureInfo.InvariantCulture, DateTimeStyles.None, out var dateTime)
-            ? dateTime.ToUniversalTime()
-            : DateTimeOffset.UnixEpoch;
-    }
-}

app/MindWork AI Studio/Tools/Databases/IndexStore/IndexStoreSearchResult.cs

@@ -0,0 +1,24 @@
+namespace AIStudio.Tools.Databases.IndexStore;
+
+public sealed record IndexStoreSearchResult(
+    string ChunkId,
+    string ParentFileId,
+    string DataSourceId,
+    string DataSourceName,
+    string DataSourceType,
+    string AbsolutePath,
+    string FileName,
+    string RelativePath,
+    string FileType,
+    int? PageNumber,
+    int ChunkIndex,
+    string ChunkText,
+    double Score,
+    string Fingerprint,
+    long FileSize,
+    DateTimeOffset CreationUtc,
+    DateTimeOffset LastWriteUtc,
+    DateTimeOffset EmbeddedAtUtc,
+    int ChunkCount,
+    string ConfidenceLevel,
+    int ConfidenceLevelRank);

app/MindWork AI Studio/Tools/Databases/IndexStore/IndexStoreSearchResultEntity.cs

@@ -0,0 +1,46 @@
+namespace AIStudio.Tools.Databases.IndexStore;
+
+internal sealed class IndexStoreSearchResultEntity
+{
+    public string ChunkId { get; set; } = string.Empty;
+
+    public string ParentFileId { get; set; } = string.Empty;
+
+    public string DataSourceId { get; set; } = string.Empty;
+
+    public string DataSourceName { get; set; } = string.Empty;
+
+    public string DataSourceType { get; set; } = string.Empty;
+
+    public string AbsolutePath { get; set; } = string.Empty;
+
+    public string FileName { get; set; } = string.Empty;
+
+    public string RelativePath { get; set; } = string.Empty;
+
+    public string FileType { get; set; } = string.Empty;
+
+    public int? PageNumber { get; set; }
+
+    public int ChunkIndex { get; set; }
+
+    public string ChunkText { get; set; } = string.Empty;
+
+    public double Score { get; set; }
+
+    public string Fingerprint { get; set; } = string.Empty;
+
+    public long FileSize { get; set; }
+
+    public DateTimeOffset CreationUtc { get; set; }
+
+    public DateTimeOffset LastWriteUtc { get; set; }
+
+    public DateTimeOffset EmbeddedAtUtc { get; set; }
+
+    public int ChunkCount { get; set; }
+
+    public string ConfidenceLevel { get; set; } = string.Empty;
+
+    public int ConfidenceLevelRank { get; set; }
+}

app/MindWork AI Studio/Tools/Databases/IndexStore/SqliteIndexStoreClientImplementation.cs

@@ -9,11 +9,7 @@ using Microsoft.EntityFrameworkCore;
 
 namespace AIStudio.Tools.Databases.IndexStore;
 
-public sealed class SqliteIndexStoreClientImplementation(
-    string name,
-    string databasePath,
-    string basePath,
-    string version) : IndexStoreClient(name, basePath)
+public sealed class SqliteIndexStoreClientImplementation(string name, string databasePath, string basePath, string version) : IndexStoreClient(name, basePath)
 {
     private const string DATABASE_NAME = "Local RAG Index";
     private const string DATABASE_FILENAME = "rag-index.sqlite3";