Skip to content

Commit cc73026

Browse files
committed
Fix race condition waiting for status of ollama model pull success status
1 parent a61437d commit cc73026

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

models/spring-ai-ollama/src/test/java/org/springframework/ai/ollama/api/OllamaApiModelsIT.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import java.io.IOException;
2020
import java.time.Duration;
2121

22+
import org.awaitility.Awaitility;
2223
import org.junit.jupiter.api.BeforeAll;
2324
import org.junit.jupiter.api.Test;
2425

@@ -89,6 +90,10 @@ public void pullModel() {
8990
.block();
9091

9192
assertThat(progressResponses).isNotNull();
93+
Awaitility.await().until(() -> {
94+
OllamaApi.ProgressResponse progressResponse = progressResponses.get(progressResponses.size() - 1);
95+
return progressResponse.status().equals("success");
96+
});
9297
assertThat(progressResponses.get(progressResponses.size() - 1))
9398
.isEqualTo(new OllamaApi.ProgressResponse("success", null, null, null));
9499

0 commit comments

Comments
 (0)