Skip to content

Commit a291300

Browse files
authored
Pull model on initialization in Ollama test fixture (#371)
1 parent dc62da6 commit a291300

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

integration-tests/src/jvmMain/kotlin/ai/koog/integration/tests/OllamaTestFixture.kt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,12 @@ class OllamaTestFixture {
3535
waitForOllamaServer(baseUrl)
3636

3737
client = OllamaClient(baseUrl)
38+
39+
// Always pull the model to ensure it's available
40+
runBlocking {
41+
client.getModelOrNull(model.id, pullIfMissing = true)
42+
}
43+
3844
executor = SingleLLMPromptExecutor(client)
3945
}
4046

0 commit comments

Comments
 (0)