feat: Added await support for IAsyncEnumerable methods.

src/libs/Ollama/OllamaApiClientExtensions.cs

@@ -1,4 +1,5 @@
-using System.Text;
+using System.Runtime.CompilerServices;
+using System.Text;
 
 namespace Ollama;
 
@@ -165,4 +166,17 @@ public static async Task<T> WaitAsync<T>(
 
 		return currentResponse;
 	}
+
+	/// <summary>
+	/// Waits for the enumerable to complete and combines the responses into a single response.
+	/// </summary>
+	/// <param name="enumerable"></param>
+	/// <returns></returns>
+	public static TaskAwaiter<T> GetAwaiter<T>(
+		this IAsyncEnumerable<T> enumerable) where T : new()
+	{
+		enumerable = enumerable ?? throw new ArgumentNullException(nameof(enumerable));
+
+		return enumerable.WaitAsync().GetAwaiter();
+	}
 }

src/libs/Ollama/generate.sh

@@ -10,7 +10,8 @@ oag generate openapi.yaml \
   --namespace Ollama \
   --clientClassName OllamaApiClient \
   --targetFramework net8.0 \
-  --output Generated
+  --output Generated \
+  --exclude-deprecated-operations
 #openapi-generator generate \
 #  -i openapi.yaml \
 #  -g csharp \

src/tests/Ollama.IntegrationTests/Tests.GetCompletionWithOptions.cs

@@ -20,7 +20,7 @@ public async Task GetCompletionWithOptions()
             {
                 Temperature = 0,
             },
-        }).WaitAsync();
+        });
         Console.WriteLine(response.Response);
     }
 }

src/tests/Ollama.IntegrationTests/Tests.PullModel.cs

@@ -12,7 +12,7 @@ public async Task PullModel()
             Console.WriteLine($"{response.Status?.Object}. Progress: {response.Completed}/{response.Total}");
         }
 
-        var response2 = await container.ApiClient.Models.PullModelAsync("all-minilm").WaitAsync();
+        var response2 = await container.ApiClient.Models.PullModelAsync("all-minilm");
         response2.EnsureSuccess();
 
         await container.ApiClient.Models.PullModelAndEnsureSuccessAsync("all-minilm");