Add test for #1317

add integration test for #1317
spring-projects · Sep 10, 2024 · b5c7975 · b5c7975
1 parent 26c5674
commit b5c7975
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/...emini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java b/...emini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java
@@ -69,6 +69,19 @@ void roleTest() {
 		assertThat(response.getResult().getOutput().getContent()).containsAnyOf("Blackbeard", "Bartholomew");
 	}
 
+	@Test
+	void testMessageHistory() {
+		Prompt prompt = createPrompt(VertexAiGeminiChatOptions.builder().build());
+		ChatResponse response = chatModel.call(prompt);
+		assertThat(response.getResult().getOutput().getContent()).containsAnyOf("Blackbeard", "Bartholomew");
+
+		var promptWithMessageHistory = new Prompt(List.of(prompt.getInstructions().get(0),
+				response.getResult().getOutput(), new UserMessage("What was the answer from the previous question?")));
+		response = chatModel.call(promptWithMessageHistory);
+
+		assertThat(response.getResult().getOutput().getContent()).containsAnyOf("Blackbeard", "Bartholomew");
+	}
+
 	@Test
 	void googleSearchTool() {
 		Prompt prompt = createPrompt(VertexAiGeminiChatOptions.builder().withGoogleSearchRetrieval(true).build());