fix: Fixed GetPercent after update.

tryAGI · Sep 17, 2024 · 9ce89ae · 9ce89ae
1 parent ed1b5d2
commit 9ce89ae
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git 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;
 	}
 }
diff --git 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"));
     }
 }