bla

Author: tzolov

spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/tool/FunctionCallbackWithPlainFunctionBeanIT.java

@@ -32,6 +32,8 @@
 import org.springframework.ai.chat.model.ChatResponse;
 import org.springframework.ai.chat.model.Generation;
 import org.springframework.ai.chat.prompt.Prompt;
+import org.springframework.ai.model.function.FunctionCallingOptions;
+import org.springframework.ai.model.function.FunctionCallingOptionsBuilder.PortableFunctionCallingOptions;
 import org.springframework.ai.openai.OpenAiChatModel;
 import org.springframework.ai.openai.OpenAiChatOptions;
 import org.springframework.ai.openai.api.OpenAiApi.ChatModel;
@@ -103,6 +105,31 @@ void functionCallWithPortableFunctionCallingOptions() {
 			});
 	}
 
+	@Test
+	void functionCallWithPortableFunctionCallingOptions2() {
+		contextRunner
+			.withPropertyValues("spring.ai.openai.chat.options.model=" + ChatModel.GPT_4_O_MINI.getName(),
+					"spring.ai.openai.chat.options.temperature=0.1")
+			.run(context -> {
+
+				OpenAiChatModel chatModel = context.getBean(OpenAiChatModel.class);
+
+				// Test weatherFunction
+				UserMessage userMessage = new UserMessage(
+						"What's the weather like in San Francisco, Tokyo, and Paris?");
+
+				PortableFunctionCallingOptions functionOptions = FunctionCallingOptions.builder()
+					.withFunction("weatherFunction")
+					.build();
+
+				ChatResponse response = chatModel.call(new Prompt(List.of(userMessage), functionOptions));
+
+				logger.info("Response: {}", response.getResult().getOutput().getContent());
+
+				assertThat(response.getResult().getOutput().getContent()).contains("30", "10", "15");
+			});
+	}
+
 	@Test
 	void streamFunctionCallTest() {
 		contextRunner