Replace Qdrant with Qdrant Edge (#783)

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

@@ -1,43 +1,53 @@
-using AIStudio.Tools.Rust;
-
 namespace AIStudio.Tools.Services;
 
 public sealed partial class RustService
 {
-    public async Task<QdrantInfo> GetQdrantInfo(CancellationToken cancellationToken = default)
+    public async Task<TDatabaseInfo> GetDatabaseInfo<TDatabaseInfo>(
+        string databaseName,
+        string infoPath,
+        Func<string, TDatabaseInfo> unavailableFactory,
+        CancellationToken cancellationToken = default)
     {
         try
         {
             using var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
             cts.CancelAfter(TimeSpan.FromSeconds(45));
-            
-            return await this.http.GetFromJsonAsync<QdrantInfo>("/system/qdrant/info", this.jsonRustSerializerOptions, cts.Token);
+
+            var databaseInfo = await this.http.GetFromJsonAsync<TDatabaseInfo>(infoPath, this.jsonRustSerializerOptions, cts.Token);
+            return databaseInfo ?? unavailableFactory("The database information response was empty.");
         }
         catch (OperationCanceledException) when (cancellationToken.IsCancellationRequested)
         {
             if(this.logger is not null)
-                this.logger.LogWarning("Fetching Qdrant info from Rust service was cancelled by caller.");
+                this.logger.LogWarning("Fetching {DatabaseName} info from Rust service was cancelled by caller.", databaseName);
             else
-                Console.WriteLine("Fetching Qdrant info from Rust service was cancelled by caller.");
-            
-            return new QdrantInfo
-            {
-                Status = QdrantStatus.UNAVAILABLE,
-                UnavailableReason = "Operation cancelled by caller."
-            };
+                Console.WriteLine($"Fetching {databaseName} info from Rust service was cancelled by caller.");
+
+            return unavailableFactory("Operation cancelled by caller.");
         }
         catch (Exception e)
         {
             if(this.logger is not null)
-                this.logger.LogError(e, "Error while fetching Qdrant info from Rust service.");
+                this.logger.LogError(e, "Error while fetching {DatabaseName} info from Rust service.", databaseName);
             else
-                Console.WriteLine($"Error while fetching Qdrant info from Rust service: '{e}'.");
-            
-            return new QdrantInfo
-            {
-                Status = QdrantStatus.UNAVAILABLE,
-                UnavailableReason = e.Message
-            };
+                Console.WriteLine($"Error while fetching {databaseName} info from Rust service: '{e}'.");
+
+            return unavailableFactory(e.Message);
         }
     }
-}
+
+    public async Task ExecuteDatabaseOperation<TRequest>(string databaseName, string path, TRequest request, CancellationToken cancellationToken = default)
+    {
+        using var cts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
+        cts.CancelAfter(TimeSpan.FromMinutes(5));
+
+        using var response = await this.http.PostAsJsonAsync(path, request, this.jsonRustSerializerOptions, cts.Token);
+        response.EnsureSuccessStatusCode();
+
+        var operation = await response.Content.ReadFromJsonAsync<DatabaseOperationResponse>(this.jsonRustSerializerOptions, cts.Token);
+        if (operation is not { Success: true })
+            throw new InvalidOperationException(operation?.Issue ?? $"The {databaseName} operation failed.");
+    }
+
+    private sealed record DatabaseOperationResponse(bool Success, string Issue);
+}