From b63fec6b59b48f83c32b08aa5da8c640fd7807ab Mon Sep 17 00:00:00 2001 From: Dang Zitou Date: Sat, 8 Aug 2026 01:31:50 +0800 Subject: [PATCH] Avoid shared map mutation in dimensions lookup Signed-off-by: Dang Zitou --- .../ai/google/genai/text/GoogleGenAiTextEmbeddingModel.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/spring-ai-google-genai-embedding/src/main/java/org/springframework/ai/google/genai/text/GoogleGenAiTextEmbeddingModel.java b/models/spring-ai-google-genai-embedding/src/main/java/org/springframework/ai/google/genai/text/GoogleGenAiTextEmbeddingModel.java index e826170775..cec21a8a90 100644 --- a/models/spring-ai-google-genai-embedding/src/main/java/org/springframework/ai/google/genai/text/GoogleGenAiTextEmbeddingModel.java +++ b/models/spring-ai-google-genai-embedding/src/main/java/org/springframework/ai/google/genai/text/GoogleGenAiTextEmbeddingModel.java @@ -276,7 +276,8 @@ private DefaultUsage getDefaultUsage(Integer totalTokens) { @Override public int dimensions() { - return KNOWN_EMBEDDING_DIMENSIONS.computeIfAbsent(this.options.getModel(), model -> super.dimensions()); + Integer dimensions = KNOWN_EMBEDDING_DIMENSIONS.get(this.options.getModel()); + return dimensions != null ? dimensions : super.dimensions(); } /**