Create long running operation with OperationId passed in

solliancenet · Jul 26, 2024 · c2378a3 · c2378a3
1 parent 26ef028
commit c2378a3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/src/dotnet/State/Interfaces/IStateService.cs b/src/dotnet/State/Interfaces/IStateService.cs
@@ -46,8 +46,9 @@ public interface IStateService
         /// <summary>
         /// Creates a new long-running operation with default values.
         /// </summary>
+        /// <param name="operationId">The long-running operation identifier.</param>
         /// <returns></returns>
-        Task<LongRunningOperation> CreateLongRunningOperation();
+        Task<LongRunningOperation> CreateLongRunningOperation(string operationId);
 
         /// <summary>
         /// Inserts or updates the result of a long-running operation.

diff --git a/src/dotnet/State/Services/StateService.cs b/src/dotnet/State/Services/StateService.cs
@@ -47,13 +47,13 @@ public async Task<List<LongRunningOperationLogEntry>> GetLongRunningOperationLog
         }
 
         /// <inheritdoc/>
-        public async Task<LongRunningOperation> CreateLongRunningOperation()
+        public async Task<LongRunningOperation> CreateLongRunningOperation(string operationId)
         {
             logger.LogInformation("Creating long running operation.");
             var operation = new LongRunningOperation
             {
                 Status = OperationStatus.Pending,
-                OperationId = Guid.NewGuid().ToString()
+                OperationId = operationId
             };
             return await cosmosDbService.UpsertLongRunningOperation(operation);
         }

diff --git a/src/dotnet/StateAPI/Controllers/OperationsController.cs b/src/dotnet/StateAPI/Controllers/OperationsController.cs
@@ -60,13 +60,13 @@ public async Task<IActionResult> GetLongRunningOperationLogs(string instanceId,
         /// <summary>
         /// Inserts a long-running operation and creates a log entry.
         /// </summary>
+        /// <param name="operationId">The ID of the long running operation.</param>
         /// <param name="instanceId">The FoundationaLLM instance ID.</param>
         /// <returns></returns>
-        [HttpPost]
-        public async Task<IActionResult> CreateLongRunningOperation(string instanceId)
+        [HttpPost("{operationId}")]
+        public async Task<IActionResult> CreateLongRunningOperation(string instanceId, string operationId)
         {
-            var  createdLongRunningOperation = await stateService.CreateLongRunningOperation();
-
+            var createdLongRunningOperation = await stateService.CreateLongRunningOperation(operationId);            
             return new OkObjectResult(createdLongRunningOperation);
         }