From b5c7975facbb5cccc202ee3c04d5b2534f06c2a7 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Tue, 10 Sep 2024 11:23:17 +0200 Subject: [PATCH] Add test for #1317 add integration test for https://github.com/spring-projects/spring-ai/pull/1317 --- .../vertexai/gemini/VertexAiGeminiChatModelIT.java | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java b/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java index 786ee59cc..688291474 100644 --- a/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java +++ b/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java @@ -69,6 +69,19 @@ class VertexAiGeminiChatModelIT { 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());