From e819b7c313c574f4fe98637501df755d318a4b6a Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Sat, 2 Mar 2024 22:01:08 +0100 Subject: [PATCH] Add missing AOT hits for OpenAiAudioApi and OpenAiImageApi --- .../springframework/ai/openai/aot/OpenAiRuntimeHints.java | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/aot/OpenAiRuntimeHints.java b/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/aot/OpenAiRuntimeHints.java index f21a43b97..49f204efe 100644 --- a/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/aot/OpenAiRuntimeHints.java +++ b/models/spring-ai-openai/src/main/java/org/springframework/ai/openai/aot/OpenAiRuntimeHints.java @@ -16,6 +16,8 @@ package org.springframework.ai.openai.aot; import org.springframework.ai.openai.api.OpenAiApi; +import org.springframework.ai.openai.api.OpenAiAudioApi; +import org.springframework.ai.openai.api.OpenAiImageApi; import org.springframework.aot.hint.MemberCategory; import org.springframework.aot.hint.RuntimeHints; import org.springframework.aot.hint.RuntimeHintsRegistrar; @@ -39,6 +41,10 @@ public class OpenAiRuntimeHints implements RuntimeHintsRegistrar { var mcs = MemberCategory.values(); for (var tr : findJsonAnnotatedClassesInPackage(OpenAiApi.class)) hints.reflection().registerType(tr, mcs); + for (var tr : findJsonAnnotatedClassesInPackage(OpenAiAudioApi.class)) + hints.reflection().registerType(tr, mcs); + for (var tr : findJsonAnnotatedClassesInPackage(OpenAiImageApi.class)) + hints.reflection().registerType(tr, mcs); } }