Skip to content

Commit 3c57027

Browse files
committed
Fix VertexAiGeminiChatModelToolCallingIT#functionCallTestInferredOpenApiSchemaStream
- Improve user message for the consistent test result assertion Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
1 parent 07688d5 commit 3c57027

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/tool/VertexAiGeminiChatModelToolCallingIT.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,8 @@ public void functionCallTestInferredOpenApiSchema() {
141141
public void functionCallTestInferredOpenApiSchemaStream() {
142142

143143
UserMessage userMessage = new UserMessage(
144-
"What's the weather like in San Francisco, Paris and in Tokyo? Return the temperature in Celsius.");
144+
"What's the weather like in San Francisco, Paris and in Tokyo? Return the temperature in Celsius."
145+
+ "You must call the getCurrentWeather tool once for each city and return the temperatures in Celsius as plain numbers.");
145146

146147
List<Message> messages = new ArrayList<>(List.of(userMessage));
147148

0 commit comments

Comments
 (0)