File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed
models/spring-ai-ollama/src/test/java/org/springframework/ai/ollama Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change 1
1
package org .springframework .ai .ollama ;
2
2
3
+ import java .time .Duration ;
4
+
3
5
import org .springframework .ai .ollama .api .OllamaApi ;
6
+ import org .springframework .ai .ollama .management .ModelManagementOptions ;
4
7
import org .springframework .ai .ollama .management .OllamaModelManager ;
5
8
import org .springframework .ai .ollama .management .PullModelStrategy ;
6
9
import org .springframework .util .StringUtils ;
@@ -53,7 +56,11 @@ public static OllamaApi buildOllamaApiWithModel(String model) {
53
56
}
54
57
55
58
public static void ensureModelIsPresent (OllamaApi ollamaApi , String model ) {
56
- var ollamaModelManager = new OllamaModelManager (ollamaApi );
59
+ var modelManagementOptions = ModelManagementOptions .builder ()
60
+ .withMaxRetries (2 )
61
+ .withTimeout (Duration .ofMinutes (10 ))
62
+ .build ();
63
+ var ollamaModelManager = new OllamaModelManager (ollamaApi , modelManagementOptions );
57
64
ollamaModelManager .pullModel (model , PullModelStrategy .WHEN_MISSING );
58
65
}
59
66
You can’t perform that action at this time.
0 commit comments