UT for OpenAiChatOptions.fromOptions.webSearchOptions

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

Author: lambochen

models/spring-ai-openai/src/test/java/org/springframework/ai/openai/OpenAiChatOptionsTests.java

@@ -30,6 +30,7 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.springframework.ai.openai.api.OpenAiApi.ChatCompletionRequest.AudioParameters.Voice.ALLOY;
+import static org.springframework.ai.openai.api.OpenAiApi.ChatCompletionRequest.WebSearchOptions.SearchContextSize.MEDIUM;
 
 /**
  * Tests for {@link OpenAiChatOptions}.
@@ -259,4 +260,27 @@ void testDefaultValues() {
 		assertThat(options.getStopSequences()).isNull();
 	}
 
+	@Test
+	void testFromOptions_webSearchOptions() {
+		var chatOptions = OpenAiChatOptions.builder()
+				.webSearchOptions(new OpenAiApi.ChatCompletionRequest.WebSearchOptions(
+						MEDIUM,
+						new OpenAiApi.ChatCompletionRequest.WebSearchOptions.UserLocation(
+								"type",
+								new OpenAiApi.ChatCompletionRequest.WebSearchOptions.UserLocation.Approximate("beijing", "china", "region", "UTC+8")
+						)
+				))
+			.build();
+		var target = OpenAiChatOptions.fromOptions(chatOptions);
+		assertThat(target.getWebSearchOptions()).isNotNull();
+		assertThat(target.getWebSearchOptions().searchContextSize()).isEqualTo(MEDIUM);
+		assertThat(target.getWebSearchOptions().userLocation()).isNotNull();
+		assertThat(target.getWebSearchOptions().userLocation().type()).isEqualTo("type");
+		assertThat(target.getWebSearchOptions().userLocation().approximate()).isNotNull();
+		assertThat(target.getWebSearchOptions().userLocation().approximate().city()).isEqualTo("beijing");
+		assertThat(target.getWebSearchOptions().userLocation().approximate().country()).isEqualTo("china");
+		assertThat(target.getWebSearchOptions().userLocation().approximate().region()).isEqualTo("region");
+		assertThat(target.getWebSearchOptions().userLocation().approximate().timezone()).isEqualTo("UTC+8");
+	}
+
 }