Mirror/ollama4j
1
0
Fork 1
mirror of https://github.com/amithkoujalgi/ollama4j.git synced 2025-09-16 03:39:05 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Increase model pull retry delay and attempts

Browse Source 
Raised the base delay for model pull retries from 1s to 3s and updated log output to show seconds instead of milliseconds. Also increased the number of retries for model pull in integration tests from 3 to 5 to improve robustness.
This commit is contained in:
amithkoujalgi 2025-08-30 17:04:02 +05:30
parent de379d73b2
commit c705432510
No known key found for this signature in database
GPG Key ID: E29A37746AF94B70
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main/java/io/github/ollama4j/OllamaAPI.java
View File

@ -429,7 +429,7 @@ public class OllamaAPI {
return;
}
int numberOfRetries = 0;
long baseDelayMillis = 1000L; // 1 second base delay
long baseDelayMillis = 3000L; // 1 second base delay
while (numberOfRetries < numberOfRetriesForModelPull) {
try {
this.doPullModel(modelName);
@ -450,8 +450,8 @@ public class OllamaAPI {
int attempt = currentRetry + 1;
if (attempt < maxRetries) {
long backoffMillis = baseDelayMillis * (1L << currentRetry);
logger.error("Failed to pull model {}, retrying in {} ms... (attempt {}/{})",
modelName, backoffMillis, attempt, maxRetries);
logger.error("Failed to pull model {}, retrying in {} s... (attempt {}/{})",
modelName, backoffMillis/1000, attempt, maxRetries);
try {
Thread.sleep(backoffMillis);
} catch (InterruptedException ie) {

2
src/test/java/io/github/ollama4j/integrationtests/OllamaAPIIntegrationTest.java
View File

@ -75,7 +75,7 @@ class OllamaAPIIntegrationTest {
}
api.setRequestTimeoutSeconds(120);
api.setVerbose(true);
api.setNumberOfRetriesForModelPull(3);
api.setNumberOfRetriesForModelPull(5);
}
@Test