From a80401f2b54a978169ad9695b7898fe81f67eca6 Mon Sep 17 00:00:00 2001
From: Chris Turchin <645266+turchinc@users.noreply.github.com>
Date: Thu, 11 Jul 2024 17:34:01 +0000
Subject: [PATCH] withSystemInstruction returns a new model

---
 .../ai/vertexai/gemini/VertexAiGeminiChatModel.java             | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java
index 8cc1d732346..37d776b24cb 100644
--- a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java
+++ b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java
@@ -266,7 +266,7 @@ private GeminiRequest createGeminiRequest(Prompt prompt) {
 			.collect(Collectors.joining(System.lineSeparator()));
 
 		if (StringUtils.hasText(systemContext)) {
-			generativeModel.withSystemInstruction(ContentMaker.fromString(systemContext));
+			generativeModel = generativeModel.withSystemInstruction(ContentMaker.fromString(systemContext));
 		}
 
 		return new GeminiRequest(toGeminiContent(prompt), generativeModel);