Replace Qdrant with Qdrant Edge (#783)

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

@@ -4,6 +4,7 @@ using AIStudio.Components;
 using AIStudio.Dialogs;
 using AIStudio.Settings.DataModel;
 using AIStudio.Tools.Databases;
+using AIStudio.Tools.Databases.VectorStore;
 using AIStudio.Tools.Metadata;
 using AIStudio.Tools.PluginSystem;
 using AIStudio.Tools.Rust;
@@ -35,7 +36,7 @@ public partial class Information : MSGComponentBase
     private static readonly MetaDataAttribute META_DATA = ASSEMBLY.GetCustomAttribute<MetaDataAttribute>()!;
     private static readonly MetaDataArchitectureAttribute META_DATA_ARCH = ASSEMBLY.GetCustomAttribute<MetaDataArchitectureAttribute>()!;
     private static readonly MetaDataLibrariesAttribute META_DATA_LIBRARIES = ASSEMBLY.GetCustomAttribute<MetaDataLibrariesAttribute>()!;
-    private static readonly MetaDataDatabasesAttribute META_DATA_DATABASES = ASSEMBLY.GetCustomAttribute<MetaDataDatabasesAttribute>()!;
+    private static readonly MetaDataVectorStoreAttribute META_DATA_VECTOR_STORE = ASSEMBLY.GetCustomAttribute<MetaDataVectorStoreAttribute>()!;
     
     private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(Information).Namespace, nameof(Information));
 
@@ -77,18 +78,18 @@ public partial class Information : MSGComponentBase
     
     private string VersionPdfium => $"{T("Used PDFium version")}: v{META_DATA_LIBRARIES.PdfiumVersion}";
     
-    private string VersionDatabase
+    private string VersionVectorStore
     {
         get
         {
-            if (this.databaseClient is null)
-                return $"{T("Database")}: {T("checking availability")}";
+            if (this.vectorStore is null)
+                return $"{T("Vector store")}: {T("checking availability")}";
 
-            return this.databaseClient.Status switch
+            return this.vectorStore.Status switch
             {
-                DatabaseClientStatus.AVAILABLE => $"{T("Database version")}: {this.databaseClient.Name} v{META_DATA_DATABASES.DatabaseVersion}",
-                DatabaseClientStatus.STARTING => $"{T("Database")}: {this.databaseClient.Name} - {T("starting")}",
-                _ => $"{T("Database")}: {this.databaseClient.Name} - {T("not available")}"
+                DatabaseClientStatus.AVAILABLE => $"{T("Vector store version")}: {this.vectorStore.Name} v{META_DATA_VECTOR_STORE.VectorStoreVersion}",
+                DatabaseClientStatus.STARTING => $"{T("Vector store")}: {this.vectorStore.Name} - {T("starting")}",
+                _ => $"{T("Vector store")}: {this.vectorStore.Name} - {T("not available")}"
             };
         }
     }
@@ -100,10 +101,9 @@ public partial class Information : MSGComponentBase
     
     private bool showEnterpriseConfigDetails;
 
+    private bool showVectorStoreDetails;
     private bool showExternalHttpCustomRootCertificateDetails;
 
-    private bool showDatabaseDetails;
-
     private List<IAvailablePlugin> configPlugins = PluginFactory.AvailablePlugins
         .Where(x => x.Type is PluginType.CONFIGURATION)
         .OfType<IAvailablePlugin>()
@@ -112,14 +112,13 @@ public partial class Information : MSGComponentBase
     private List<EnterpriseEnvironment> enterpriseEnvironments = EnterpriseEnvironmentService.CURRENT_ENVIRONMENTS.ToList();
 
     private List<MandatoryInfoPanelData> mandatoryInfoPanels = [];
-    
-    private sealed record DatabaseDisplayInfo(string Label, string Value);
 
     private sealed record MandatoryInfoPanelData(string HeaderText, string PluginName, DataMandatoryInfo Info, DataMandatoryInfoAcceptance? Acceptance);
-
-    private readonly List<DatabaseDisplayInfo> databaseDisplayInfo = new();
-    private DatabaseClient? databaseClient;
-    private CancellationTokenSource? databaseRefreshCancellationTokenSource;
+    
+    private sealed record VectorStoreDisplayInfo(string Label, string Value);
+    private readonly List<VectorStoreDisplayInfo> vectorStoreDisplayInfo = new();
+    private DatabaseClient? vectorStore;
+    private CancellationTokenSource? vectorStoreRefreshCancellationTokenSource;
 
     private bool HasAnyActiveEnvironment => this.enterpriseEnvironments.Any(e => e.IsActive);
     
@@ -166,9 +165,9 @@ public partial class Information : MSGComponentBase
         this.runtimeInfo = await this.RustService.GetRuntimeInfo();
         this.logPaths = await this.RustService.GetLogPaths();
         
-        await this.RefreshDatabaseInfo(CancellationToken.None);
-        if (this.databaseClient?.Status is DatabaseClientStatus.STARTING)
-            this.StartShortDatabaseRefreshLoop();
+        await this.RefreshVectorStoreInfo(CancellationToken.None);
+        if (this.vectorStore?.Status is DatabaseClientStatus.STARTING)
+            this.StartShortVectorStoreRefreshLoop();
         
         // Determine the Pandoc version may take some time, so we start it here
         // without waiting for the result:
@@ -272,22 +271,22 @@ public partial class Information : MSGComponentBase
         this.showExternalHttpCustomRootCertificateDetails = !this.showExternalHttpCustomRootCertificateDetails;
     }
     
-    private void ToggleDatabaseDetails()
+    private void ToggleVectorStoreDetails()
     {
-        this.showDatabaseDetails = !this.showDatabaseDetails;
+        this.showVectorStoreDetails = !this.showVectorStoreDetails;
     }
 
-    private async Task RefreshDatabaseInfo(CancellationToken cancellationToken)
+    private async Task RefreshVectorStoreInfo(CancellationToken cancellationToken)
     {
         var refreshedClient = await this.DatabaseClientProvider.RefreshClientAsync(DatabaseRole.VECTOR_STORE, cancellationToken);
-        this.databaseClient = refreshedClient;
-        this.databaseDisplayInfo.Clear();
+        this.vectorStore = refreshedClient;
+        this.vectorStoreDisplayInfo.Clear();
 
         try
         {
             await foreach (var (label, value) in refreshedClient.GetDisplayInfo().WithCancellation(cancellationToken))
             {
-                this.databaseDisplayInfo.Add(new DatabaseDisplayInfo(label, value));
+                this.vectorStoreDisplayInfo.Add(new VectorStoreDisplayInfo(label, value));
             }
         }
         catch (OperationCanceledException)
@@ -296,20 +295,20 @@ public partial class Information : MSGComponentBase
         }
         catch (Exception e)
         {
-            this.databaseClient = new NoDatabaseClient(refreshedClient.Name, e.Message, DatabaseClientStatus.STARTING);
-            await foreach (var (label, value) in this.databaseClient.GetDisplayInfo().WithCancellation(cancellationToken))
+            this.vectorStore = new NoVectorStoreClient(refreshedClient.Name, e.Message, DatabaseClientStatus.STARTING);
+            await foreach (var (label, value) in this.vectorStore.GetDisplayInfo().WithCancellation(cancellationToken))
             {
-                this.databaseDisplayInfo.Add(new DatabaseDisplayInfo(label, value));
+                this.vectorStoreDisplayInfo.Add(new VectorStoreDisplayInfo(label, value));
             }
         }
     }
 
-    private void StartShortDatabaseRefreshLoop()
+    private void StartShortVectorStoreRefreshLoop()
     {
-        this.databaseRefreshCancellationTokenSource?.Cancel();
-        this.databaseRefreshCancellationTokenSource?.Dispose();
-        this.databaseRefreshCancellationTokenSource = new CancellationTokenSource();
-        var cancellationToken = this.databaseRefreshCancellationTokenSource.Token;
+        this.vectorStoreRefreshCancellationTokenSource?.Cancel();
+        this.vectorStoreRefreshCancellationTokenSource?.Dispose();
+        this.vectorStoreRefreshCancellationTokenSource = new CancellationTokenSource();
+        var cancellationToken = this.vectorStoreRefreshCancellationTokenSource.Token;
 
         _ = Task.Run(async () =>
         {
@@ -321,11 +320,11 @@ public partial class Information : MSGComponentBase
                     await Task.Delay(TimeSpan.FromSeconds(1), cancellationToken);
                     await this.InvokeAsync(async () =>
                     {
-                        await this.RefreshDatabaseInfo(cancellationToken);
+                        await this.RefreshVectorStoreInfo(cancellationToken);
                         this.StateHasChanged();
                     });
 
-                    if (this.databaseClient?.Status is not DatabaseClientStatus.STARTING)
+                    if (this.vectorStore?.Status is not DatabaseClientStatus.STARTING)
                         return;
                 }
                 catch (OperationCanceledException)
@@ -475,8 +474,8 @@ public partial class Information : MSGComponentBase
 
     protected override void DisposeResources()
     {
-        this.databaseRefreshCancellationTokenSource?.Cancel();
-        this.databaseRefreshCancellationTokenSource?.Dispose();
+        this.vectorStoreRefreshCancellationTokenSource?.Cancel();
+        this.vectorStoreRefreshCancellationTokenSource?.Dispose();
         base.DisposeResources();
     }