From 9ce89aeb6bc48ae4348edb8a771163e3a1dfed8b Mon Sep 17 00:00:00 2001
From: HavenDV <havendv@gmail.com>
Date: Tue, 17 Sep 2024 17:52:40 +0400
Subject: [PATCH] fix: Fixed GetPercent after update.

---
 src/libs/Ollama/PullModelResponseExtensions.cs | 9 +++++++--
 src/tests/Ollama.SnapshotTests/TestHelper.cs   | 2 ++
 2 files changed, 9 insertions(+), 2 deletions(-)

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