Remove warnings

awaescher · May 10, 2024 · 4a3c57e · 4a3c57e
1 parent a79ac3c
commit 4a3c57e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/OllamaApiConsole/Demos/ModelManagerConsole.cs b/OllamaApiConsole/Demos/ModelManagerConsole.cs
@@ -128,9 +128,9 @@ private async Task PullModel()
 
 		await AnsiConsole.Progress().StartAsync(async context =>
 		{
-			ProgressTask task = null;
+			ProgressTask? task = null;
 			await Ollama.PullModel(pullModel, status => UpdateProgressTaskByStatus(context, ref task, status));
-			task.StopTask();
+			task?.StopTask();
 		});
 	}
 
@@ -140,7 +140,7 @@ private async Task PushModel()
 		await Ollama.PushModel("mattw/pygmalion:latest", status => AnsiConsole.MarkupLineInterpolated($"{status.Status}"));
 	}
 
-	private void UpdateProgressTaskByStatus(ProgressContext context, ref ProgressTask task, PullStatus status)
+	private void UpdateProgressTaskByStatus(ProgressContext context, ref ProgressTask? task, PullStatus status)
 	{
 		if (status.Status != task?.Description)
 		{

diff --git a/test/ChatTests.cs b/test/ChatTests.cs
@@ -15,13 +15,14 @@ public class SendMethod : ChatTests
 		[Test]
 		public async Task Sends_Assistant_Answer_To_Streamer()
 		{
-			ChatResponseStream answerFromAssistant = null;
+			ChatResponseStream? answerFromAssistant = null;
 
 			_ollama.DefineChatResponse("assistant", "hi!");
 
 			var chat = new Chat(_ollama, answer => answerFromAssistant = answer);
 			await chat.Send("henlo", CancellationToken.None);
 
+			answerFromAssistant.Should().NotBeNull();
 			answerFromAssistant.Message.Role.Should().Be(ChatRole.Assistant);
 			answerFromAssistant.Message.Content.Should().Be("hi!");
 		}

diff --git a/test/OllamaApiClientTests.cs b/test/OllamaApiClientTests.cs
@@ -14,7 +14,7 @@ namespace Tests;
 public class OllamaApiClientTests
 {
 	private OllamaApiClient _client;
-	private HttpResponseMessage _response;
+	private HttpResponseMessage? _response;
 
 	[OneTimeSetUp]
 	public void Setup()