Co-Authored-by: Josh Long <54473+joshlong@users.noreply.github.com>

Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>

Author: ilayaperumalg

models/spring-ai-openai/src/main/java/org/springframework/ai/openai/aot/OpenAiRuntimeHints.java

@@ -16,6 +16,10 @@
 
 package org.springframework.ai.openai.aot;
 
+import java.util.Set;
+
+import org.springframework.ai.openai.api.OpenAiApi;
+import org.springframework.ai.openai.api.OpenAiEmbeddingDeserializer;
 import org.springframework.aot.hint.MemberCategory;
 import org.springframework.aot.hint.RuntimeHints;
 import org.springframework.aot.hint.RuntimeHintsRegistrar;
@@ -38,6 +42,11 @@ public class OpenAiRuntimeHints implements RuntimeHintsRegistrar {
 	public void registerHints(@NonNull RuntimeHints hints, @Nullable ClassLoader classLoader) {
 		var mcs = MemberCategory.values();
 
+		for (var c : Set.of(OpenAiApi.Embedding.class, OpenAiApi.EmbeddingList.class,
+				OpenAiEmbeddingDeserializer.class)) {
+			hints.reflection().registerType(c, MemberCategory.values());
+		}
+
 		for (var tr : findJsonAnnotatedClassesInPackage("org.springframework.ai.openai")) {
 			hints.reflection().registerType(tr, mcs);
 		}

models/spring-ai-openai/src/test/java/org/springframework/ai/openai/aot/OpenAiRuntimeHintsTests.java

@@ -25,6 +25,7 @@
 import org.springframework.ai.openai.OpenAiChatOptions;
 import org.springframework.ai.openai.api.OpenAiApi;
 import org.springframework.ai.openai.api.OpenAiAudioApi;
+import org.springframework.ai.openai.api.OpenAiEmbeddingDeserializer;
 import org.springframework.ai.openai.api.OpenAiImageApi;
 import org.springframework.aot.hint.MemberCategory;
 import org.springframework.aot.hint.RuntimeHints;
@@ -312,4 +313,17 @@ void verifyJsonAnnotatedClassesContainCriticalTypes() {
 		assertThat(containsImageApi).isTrue();
 	}
 
+	@Test
+	void verifyExplicitlyRegisteredEmbeddingClasses() {
+		this.openAiRuntimeHints.registerHints(this.runtimeHints, null);
+
+		Set<TypeReference> registeredTypes = new HashSet<>();
+		this.runtimeHints.reflection().typeHints().forEach(typeHint -> registeredTypes.add(typeHint.getType()));
+
+		// Verify the three classes explicitly registered in OpenAiRuntimeHints
+		assertThat(registeredTypes.contains(TypeReference.of(OpenAiApi.Embedding.class))).isTrue();
+		assertThat(registeredTypes.contains(TypeReference.of(OpenAiApi.EmbeddingList.class))).isTrue();
+		assertThat(registeredTypes.contains(TypeReference.of(OpenAiEmbeddingDeserializer.class))).isTrue();
+	}
+
 }