fix: use lazy load for unknown embedding dimensions and cache results

viacheslav-dobrynin · ilayaperumalg · commit a8b3982ef664 · 2025-11-05T12:04:11.000Z
Signed-off-by: Viacheslav Dobrynin &lt;w.v.dobr@gmail.com&gt;
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
@@ -259,7 +259,7 @@ private DefaultUsage getDefaultUsage(Integer totalTokens) {
 
 	@Override
 	public int dimensions() {
-		return KNOWN_EMBEDDING_DIMENSIONS.getOrDefault(this.defaultOptions.getModel(), super.dimensions());
+		return KNOWN_EMBEDDING_DIMENSIONS.computeIfAbsent(this.defaultOptions.getModel(), model -> super.dimensions());
 	}
 
 	/**

Original file line number	Diff line number	Diff line change
`@@ -259,7 +259,7 @@ private DefaultUsage getDefaultUsage(Integer totalTokens) {`
`259`	`259`
`260`	`260`	`@Override`
`261`	`261`	`public int dimensions() {`
`262`		`- return KNOWN_EMBEDDING_DIMENSIONS.getOrDefault(this.defaultOptions.getModel(), super.dimensions());`
	`262`	`+ return KNOWN_EMBEDDING_DIMENSIONS.computeIfAbsent(this.defaultOptions.getModel(), model -> super.dimensions());`
`263`	`263`	`}`
`264`	`264`
`265`	`265`	`/**`