Refactor Ollama implementation and improve documentation

 - Enhance OllamaModelPuller with configurable retry timeout
 - Update test cases to use LLAMA3_1 instead of LLAMA3_2
 - Improve Ollama documentation with clearer prerequisites and model pulling instructions
 - Update Spring AI introduction page with new logo and diagram
 - Remove unnecessary main method from OllamaModelPuller

Author: tzolov

models/spring-ai-ollama/src/main/java/org/springframework/ai/ollama/api/OllamaModelPuller.java

@@ -35,8 +35,15 @@ public class OllamaModelPuller {
 
 	private OllamaApi ollamaApi;
 
+	private final long pullRetryTimeoutMs;
+
 	public OllamaModelPuller(OllamaApi ollamaApi) {
+		this(ollamaApi, 5000);
+	}
+
+	public OllamaModelPuller(OllamaApi ollamaApi, long retryTimeoutMs) {
 		this.ollamaApi = ollamaApi;
+		this.pullRetryTimeoutMs = retryTimeoutMs;
 	}
 
 	public boolean isModelAvailable(String modelName) {
@@ -56,35 +63,24 @@ public boolean deleteModel(String modelName) {
 		return this.ollamaApi.deleteModel(new DeleteModelRequest(modelName)).getStatusCode().equals(HttpStatus.OK);
 	}
 
-	public String pullModel(String modelName, boolean reTry) {
+	public String pullModel(String modelName, boolean enablePullRetry) {
 		String status = "";
 		do {
 			logger.info("Start Pulling model: {}", modelName);
 			var progress = this.ollamaApi.pullModel(new PullModelRequest(modelName));
 			status = progress.status();
 			logger.info("Pulling model: {} - Status: {}", modelName, status);
+
 			try {
-				Thread.sleep(5000);
+				Thread.sleep(this.pullRetryTimeoutMs);
 			}
 			catch (InterruptedException e) {
 				e.printStackTrace();
 			}
 		}
-		while (reTry && !status.equals("success"));
-		return status;
-	}
+		while (enablePullRetry && !status.equals("success"));
 
-	public static void main(String[] args) {
-
-		var utils = new OllamaModelPuller(new OllamaApi());
-
-		System.out.println(utils.isModelAvailable("orca-mini:latest"));
-
-		String model = "hf.co/bartowski/Llama-3.2-3B-Instruct-GGUF:Q8_0";
-
-		if (!utils.isModelAvailable(model)) {
-			utils.pullModel(model, true);
-		}
+		return status;
 	}
 
 }

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

@@ -10,7 +10,7 @@ public class BaseOllamaIT {
 	private static final Logger logger = LoggerFactory.getLogger(BaseOllamaIT.class);
 
 	// Toggle for running tests locally on native Ollama for a faster feedback loop.
-	private static final boolean useTestcontainers = false;
+	private static final boolean useTestcontainers = true;
 
 	public static final OllamaContainer ollamaContainer;
 

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

@@ -52,7 +52,7 @@ class OllamaChatModelFunctionCallingIT extends BaseOllamaIT {
 
 	private static final Logger logger = LoggerFactory.getLogger(OllamaChatModelFunctionCallingIT.class);
 
-	private static final String MODEL = OllamaModel.LLAMA3_2.getName();
+	private static final String MODEL = OllamaModel.LLAMA3_1.getName();
 
 	@Autowired
 	ChatModel chatModel;

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

@@ -46,7 +46,7 @@
 @DisabledIf("isDisabled")
 public class OllamaApiToolFunctionCallIT extends BaseOllamaIT {
 
-	private static final String MODEL = OllamaModel.LLAMA3_2.getName();
+	private static final String MODEL = OllamaModel.LLAMA3_1.getName();
 
 	private static final Logger logger = LoggerFactory.getLogger(OllamaApiToolFunctionCallIT.class);
 
@@ -64,13 +64,11 @@ public static void beforeAll() throws IOException, InterruptedException {
 	public void toolFunctionCall() {
 		// Step 1: send the conversation and available functions to the model
 		var message = Message.builder(Role.USER)
-			// .withContent("What's the weather like in San Francisco, Tokyo, and Paris?
-			// Perform multiple function calls for each location.")
-			.withContent("What's the weather like in San Francisco, Tokyo, and Paris?")
+			.withContent("What's the weather like in San Francisco, Tokyo, and Paris? Return temperature in Celsius.")
 			.build();
 
 		var functionTool = new OllamaApi.ChatRequest.Tool(new OllamaApi.ChatRequest.Tool.Function("getCurrentWeather",
-				"Get the weather in location. Return temperature in Celsius.", ModelOptionsUtils.jsonToMap("""
+				"Get the weather in location like city names.", ModelOptionsUtils.jsonToMap("""
 						{
 							"type": "object",
 							"properties": {