diff --git a/src/libs/Ollama/PullModelResponseExtensions.cs b/src/libs/Ollama/PullModelResponseExtensions.cs index b4343b4..da20b0e 100644 --- a/src/libs/Ollama/PullModelResponseExtensions.cs +++ b/src/libs/Ollama/PullModelResponseExtensions.cs @@ -14,9 +14,14 @@ public static class PullModelResponseExtensions public static double GetPercent(this PullModelResponse response) { response = response ?? throw new ArgumentNullException(nameof(response)); + + if (response.Total == null || response.Completed == null) + { + return 0; + } - return response.Total == 0 + return response.Total.Value == 0 ? 100.0 - : response.Completed * 100.0 / response.Total; + : response.Completed.Value * 100.0 / response.Total.Value; } } \ No newline at end of file diff --git a/src/tests/Ollama.SnapshotTests/TestHelper.cs b/src/tests/Ollama.SnapshotTests/TestHelper.cs index 818c7f2..501a2a8 100755 --- a/src/tests/Ollama.SnapshotTests/TestHelper.cs +++ b/src/tests/Ollama.SnapshotTests/TestHelper.cs @@ -33,10 +33,12 @@ public static async Task CheckSourceAsync( await Task.WhenAll( verifier .Verify(diagnostics) + //.AutoVerify() .UseDirectory("Snapshots") .UseTextForParameters("Diagnostics"), verifier .Verify(driver) + //.AutoVerify() .UseDirectory("Snapshots")); } } \ No newline at end of file