Fix native hints

categolj · Feb 3, 2025 · 804bd19 · 804bd19
1 parent 6eef196
commit 804bd19
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/main/java/am/ik/translation/config/NativeConfig.java b/src/main/java/am/ik/translation/config/NativeConfig.java
@@ -13,6 +13,10 @@
 import am.ik.translation.github.CreatePullRequest;
 import am.ik.translation.github.CreatePullResponse;
 import am.ik.translation.github.IssueEvent;
+import am.ik.translation.openai.ChatCompletionRequest;
+import am.ik.translation.openai.ChatCompletionResponse;
+import am.ik.translation.openai.ChatMessage;
+import am.ik.translation.openai.ChatResponseChoice;
 import org.springframework.aot.hint.MemberCategory;
 import org.springframework.aot.hint.RuntimeHints;
 import org.springframework.aot.hint.RuntimeHintsRegistrar;
@@ -53,6 +57,18 @@ public void registerHints(RuntimeHints hints, ClassLoader classLoader) {
 				.registerType(CreatePullResponse.class, MemberCategory.INVOKE_PUBLIC_METHODS,
 						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS)
 				.registerType(IssueEvent.class, MemberCategory.INVOKE_PUBLIC_METHODS,
+						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS)
+				.registerType(IssueEvent.Issue.class, MemberCategory.INVOKE_PUBLIC_METHODS,
+						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS)
+				.registerType(IssueEvent.Repository.class, MemberCategory.INVOKE_PUBLIC_METHODS,
+						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS)
+				.registerType(ChatCompletionRequest.class, MemberCategory.INVOKE_PUBLIC_METHODS,
+						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS)
+				.registerType(ChatCompletionResponse.class, MemberCategory.INVOKE_PUBLIC_METHODS,
+						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS)
+				.registerType(ChatMessage.class, MemberCategory.INVOKE_PUBLIC_METHODS,
+						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS)
+				.registerType(ChatResponseChoice.class, MemberCategory.INVOKE_PUBLIC_METHODS,
 						MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS);
 		}