Fix formatting

Signed-off-by: Filip Hrisafov <[email protected]>

Author: filiphr

models/spring-ai-anthropic/src/main/java/org/springframework/ai/anthropic/api/AnthropicApi.java

@@ -164,20 +164,17 @@ public ResponseEntity<ChatCompletionResponse> chatCompletionEntity(ChatCompletio
 		Assert.isTrue(!chatRequest.stream(), "Request must set the stream property to false.");
 		Assert.notNull(additionalHttpHeader, "The additional HTTP headers can not be null.");
 
+		// @formatter:off
 		return this.restClient.post()
 			.uri(this.completionsPath)
 			.headers(headers -> {
 				headers.addAll(additionalHttpHeader);
-				if (!headers.containsKey(HEADER_X_API_KEY)) {
-					String apiKeyValue = this.apiKey.getValue();
-					if (StringUtils.hasText(apiKeyValue)) {
-						headers.add(HEADER_X_API_KEY, apiKeyValue);
-					}
-				}
+				addDefaultHeadersIfMissing(headers);
 			})
 			.body(chatRequest)
 			.retrieve()
 			.toEntity(ChatCompletionResponse.class);
+		// @formatter:on
 	}
 
 	/**
@@ -208,17 +205,13 @@ public Flux<ChatCompletionResponse> chatCompletionStream(ChatCompletionRequest c
 
 		AtomicReference<ChatCompletionResponseBuilder> chatCompletionReference = new AtomicReference<>();
 
+		// @formatter:off
 		return this.webClient.post()
 			.uri(this.completionsPath)
 			.headers(headers -> {
 				headers.addAll(additionalHttpHeader);
-				if (!headers.containsKey(HEADER_X_API_KEY)) {
-					String apiKeyValue = this.apiKey.getValue();
-					if (StringUtils.hasText(apiKeyValue)) {
-						headers.add(HEADER_X_API_KEY, apiKeyValue);
-					}
-				}
-			})
+				addDefaultHeadersIfMissing(headers);
+			}) // @formatter:off
 			.body(Mono.just(chatRequest), ChatCompletionRequest.class)
 			.retrieve()
 			.bodyToFlux(String.class)
@@ -252,6 +245,15 @@ public Flux<ChatCompletionResponse> chatCompletionStream(ChatCompletionRequest c
 			.filter(chatCompletionResponse -> chatCompletionResponse.type() != null);
 	}
 
+	private void addDefaultHeadersIfMissing(HttpHeaders headers) {
+		if (!headers.containsKey(HEADER_X_API_KEY)) {
+			String apiKeyValue = this.apiKey.getValue();
+			if (StringUtils.hasText(apiKeyValue)) {
+				headers.add(HEADER_X_API_KEY, apiKeyValue);
+			}
+		}
+	}
+
 	/**
 	 * Check the <a href="https://docs.anthropic.com/claude/docs/models-overview">Models
 	 * overview</a> and <a href=

models/spring-ai-anthropic/src/test/java/org/springframework/ai/anthropic/api/AnthropicApiBuilderTests.java

@@ -196,12 +196,9 @@ void dynamicApiKeyRestClient() throws InterruptedException {
 
 		@Test
 		void dynamicApiKeyRestClientWithAdditionalApiKeyHeader() throws InterruptedException {
-			AnthropicApi api = AnthropicApi.builder()
-					.apiKey(() -> {
-						throw new AssertionFailedError("Should not be called, API key is provided in headers");
-					})
-				.baseUrl(mockWebServer.url("/").toString())
-				.build();
+			AnthropicApi api = AnthropicApi.builder().apiKey(() -> {
+				throw new AssertionFailedError("Should not be called, API key is provided in headers");
+			}).baseUrl(mockWebServer.url("/").toString()).build();
 
 			MockResponse mockResponse = new MockResponse().setResponseCode(200)
 				.addHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE)
@@ -231,7 +228,8 @@ void dynamicApiKeyRestClientWithAdditionalApiKeyHeader() throws InterruptedExcep
 				.build();
 			MultiValueMap<String, String> additionalHeaders = new LinkedMultiValueMap<>();
 			additionalHeaders.add("x-api-key", "additional-key");
-			ResponseEntity<AnthropicApi.ChatCompletionResponse> response = api.chatCompletionEntity(request, additionalHeaders);
+			ResponseEntity<AnthropicApi.ChatCompletionResponse> response = api.chatCompletionEntity(request,
+					additionalHeaders);
 			assertThat(response.getStatusCode()).isEqualTo(HttpStatus.OK);
 			RecordedRequest recordedRequest = mockWebServer.takeRequest();
 			assertThat(recordedRequest.getHeader(HttpHeaders.AUTHORIZATION)).isNull();
@@ -248,8 +246,7 @@ void dynamicApiKeyWebClient() throws InterruptedException {
 
 			MockResponse mockResponse = new MockResponse().setResponseCode(200)
 				.addHeader(HttpHeaders.CONTENT_TYPE, MediaType.TEXT_EVENT_STREAM_VALUE)
-				.setBody(
-						"""
+				.setBody("""
 						{
 							"type": "message_start",
 							"message": {
@@ -278,9 +275,7 @@ void dynamicApiKeyWebClient() throws InterruptedException {
 				.messages(List.of(chatCompletionMessage))
 				.stream(true)
 				.build();
-			api.chatCompletionStream(request)
-				.collectList()
-				.block();
+			api.chatCompletionStream(request).collectList().block();
 			RecordedRequest recordedRequest = mockWebServer.takeRequest();
 			assertThat(recordedRequest.getHeader(HttpHeaders.AUTHORIZATION)).isNull();
 			assertThat(recordedRequest.getHeader("x-api-key")).isEqualTo("key1");
@@ -301,26 +296,25 @@ void dynamicApiKeyWebClientWithAdditionalApiKey() throws InterruptedException {
 				.build();
 
 			MockResponse mockResponse = new MockResponse().setResponseCode(200)
-					.addHeader(HttpHeaders.CONTENT_TYPE, MediaType.TEXT_EVENT_STREAM_VALUE)
-					.setBody(
-							"""
-							{
-								"type": "message_start",
-								"message": {
-									"id": "msg_1nZdL29xx5MUA1yADyHTEsnR8uuvGzszyY",
-									"type": "message",
-									"role": "assistant",
-									"content": [],
-									"model": "claude-opus-4-20250514",
-									"stop_reason": null,
-									"stop_sequence": null,
-									"usage": {
-										"input_tokens": 25,
-										"output_tokens": 1
-									}
+				.addHeader(HttpHeaders.CONTENT_TYPE, MediaType.TEXT_EVENT_STREAM_VALUE)
+				.setBody("""
+						{
+							"type": "message_start",
+							"message": {
+								"id": "msg_1nZdL29xx5MUA1yADyHTEsnR8uuvGzszyY",
+								"type": "message",
+								"role": "assistant",
+								"content": [],
+								"model": "claude-opus-4-20250514",
+								"stop_reason": null,
+								"stop_sequence": null,
+								"usage": {
+									"input_tokens": 25,
+									"output_tokens": 1
 								}
 							}
-							""".replace("\n", ""));
+						}
+						""".replace("\n", ""));
 			mockWebServer.enqueue(mockResponse);
 
 			AnthropicApi.AnthropicMessage chatCompletionMessage = new AnthropicApi.AnthropicMessage(
@@ -334,9 +328,7 @@ void dynamicApiKeyWebClientWithAdditionalApiKey() throws InterruptedException {
 			MultiValueMap<String, String> additionalHeaders = new LinkedMultiValueMap<>();
 			additionalHeaders.add("x-api-key", "additional-key");
 
-			api.chatCompletionStream(request, additionalHeaders)
-				.collectList()
-				.block();
+			api.chatCompletionStream(request, additionalHeaders).collectList().block();
 			RecordedRequest recordedRequest = mockWebServer.takeRequest();
 			assertThat(recordedRequest.getHeader(HttpHeaders.AUTHORIZATION)).isNull();
 			assertThat(recordedRequest.getHeader("x-api-key")).isEqualTo("additional-key");