feat(ollama): Support Olama API key authentication

Signed-off-by: yupan <[email protected]>

Author: laterya

auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/main/java/org/springframework/ai/model/ollama/autoconfigure/OllamaApiAutoConfiguration.java

@@ -26,6 +26,7 @@
 import org.springframework.boot.restclient.autoconfigure.RestClientAutoConfiguration;
 import org.springframework.boot.webclient.autoconfigure.WebClientAutoConfiguration;
 import org.springframework.context.annotation.Bean;
+import org.springframework.util.StringUtils;
 import org.springframework.web.client.ResponseErrorHandler;
 import org.springframework.web.client.RestClient;
 import org.springframework.web.reactive.function.client.WebClient;
@@ -56,10 +57,21 @@ PropertiesOllamaConnectionDetails ollamaConnectionDetails(OllamaConnectionProper
 	public OllamaApi ollamaApi(OllamaConnectionDetails connectionDetails,
 			ObjectProvider<RestClient.Builder> restClientBuilderProvider,
 			ObjectProvider<WebClient.Builder> webClientBuilderProvider, ResponseErrorHandler responseErrorHandler) {
+		
+		RestClient.Builder restClientBuilder = restClientBuilderProvider.getIfAvailable(RestClient::builder);
+		WebClient.Builder webClientBuilder = webClientBuilderProvider.getIfAvailable(WebClient::builder);
+		
+		// Add Authorization header if API key is provided
+		if (StringUtils.hasText(connectionDetails.getApiKey())) {
+			String bearerToken = "Bearer " + connectionDetails.getApiKey();
+			restClientBuilder = restClientBuilder.defaultHeader("Authorization", bearerToken);
+			webClientBuilder = webClientBuilder.defaultHeader("Authorization", bearerToken);
+		}
+		
 		return OllamaApi.builder()
 			.baseUrl(connectionDetails.getBaseUrl())
-			.restClientBuilder(restClientBuilderProvider.getIfAvailable(RestClient::builder))
-			.webClientBuilder(webClientBuilderProvider.getIfAvailable(WebClient::builder))
+			.restClientBuilder(restClientBuilder)
+			.webClientBuilder(webClientBuilder)
 			.responseErrorHandler(responseErrorHandler)
 			.build();
 	}
@@ -77,6 +89,11 @@ public String getBaseUrl() {
 			return this.properties.getBaseUrl();
 		}
 
+		@Override
+		public String getApiKey() {
+			return this.properties.getApiKey();
+		}
+
 	}
 
 }

auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/main/java/org/springframework/ai/model/ollama/autoconfigure/OllamaConnectionDetails.java

@@ -27,4 +27,12 @@ public interface OllamaConnectionDetails extends ConnectionDetails {
 
 	String getBaseUrl();
 
+	/**
+	 * Returns the API key for authenticating requests.
+	 * @return the API key, or null if no authentication is required
+	 */
+	default String getApiKey() {
+		return null;
+	}
+
 }

auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/main/java/org/springframework/ai/model/ollama/autoconfigure/OllamaConnectionProperties.java

@@ -34,6 +34,12 @@ public class OllamaConnectionProperties {
 	 */
 	private String baseUrl = "http://localhost:11434";
 
+	/**
+	 * API key for authenticating requests to Ollama API server.
+	 * If provided, it will be sent as a Bearer token in the Authorization header.
+	 */
+	private String apiKey;
+
 	public String getBaseUrl() {
 		return this.baseUrl;
 	}
@@ -42,4 +48,12 @@ public void setBaseUrl(String baseUrl) {
 		this.baseUrl = baseUrl;
 	}
 
+	public String getApiKey() {
+		return this.apiKey;
+	}
+
+	public void setApiKey(String apiKey) {
+		this.apiKey = apiKey;
+	}
+
 }