chore: avoid awaiting delay if its zero

Signed-off-by: Vincent Biret <[email protected]>
microsoft · Nov 20, 2024 · c9539c1 · c9539c1
1 parent f5facce
commit c9539c1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/dotnet/src/Experimental/Orchestration.Flow/Execution/FlowExecutor.cs b/dotnet/src/Experimental/Orchestration.Flow/Execution/FlowExecutor.cs
@@ -679,7 +679,10 @@ private async Task<FunctionResult> ExecuteStepAsync(FlowStep step, string sessio
                 string? actionResult;
                 try
                 {
-                    await Task.Delay(this._config.MinIterationTimeMs).ConfigureAwait(false);
+                    if (this._config.MinIterationTimeMs > 0)
+                    {
+                        await Task.Delay(this._config.MinIterationTimeMs).ConfigureAwait(false);
+                    }
                     actionResult = await this._reActEngine.InvokeActionAsync(actionStep, input, chatHistory, kernel, actionContextVariables).ConfigureAwait(false);
 
                     if (string.IsNullOrEmpty(actionResult))
@@ -776,8 +779,11 @@ private async Task<FunctionResult> ExecuteStepAsync(FlowStep step, string sessio
                 this._logger?.LogWarning("Action: No action to take");
             }
 
-            // continue to next iteration
-            await Task.Delay(this._config.MinIterationTimeMs).ConfigureAwait(false);
+            if (this._config.MinIterationTimeMs > 0)
+            {
+                // continue to next iteration
+                await Task.Delay(this._config.MinIterationTimeMs).ConfigureAwait(false);
+            }
         }
 
         throw new KernelException($"Failed to complete step {stepId} for session {sessionId}.");

diff --git a/dotnet/src/Planners/Planners.OpenAI/Stepwise/FunctionCallingStepwisePlanner.cs b/dotnet/src/Planners/Planners.OpenAI/Stepwise/FunctionCallingStepwisePlanner.cs
@@ -88,7 +88,7 @@ private async Task<FunctionCallingStepwisePlannerResult> ExecuteCoreAsync(
         for (int i = 0; i < this._options.MaxIterations; i++)
         {
             // sleep for a bit to avoid rate limiting
-            if (i > 0)
+            if (i > 0 && this._options.MinIterationTimeMs > 0)
             {
                 await Task.Delay(this._options.MinIterationTimeMs, cancellationToken).ConfigureAwait(false);
             }