Update tools example

patterns/tool-calling/tool-calling-openai/src/main/java/com/thomasvitale/ai/spring/BookService.java

@@ -1,12 +1,14 @@
 package com.thomasvitale.ai.spring;
 
+import org.springframework.aot.hint.annotation.RegisterReflectionForBinding;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
 
 @Service
+@RegisterReflectionForBinding(classes = BookService.Book.class)
 public class BookService {
 
     private static final Map<Integer,Book> books = new ConcurrentHashMap<>();

patterns/tool-calling/tool-calling-openai/src/main/java/com/thomasvitale/ai/spring/ToolBeanRegistrationAotProcessor.java

@@ -0,0 +1,60 @@
+package com.thomasvitale.ai.spring;
+
+import org.springframework.ai.tool.annotation.Tool;
+import org.springframework.aot.generate.GenerationContext;
+import org.springframework.aot.hint.MemberCategory;
+import org.springframework.aot.hint.ReflectionHints;
+import org.springframework.beans.factory.aot.BeanRegistrationAotContribution;
+import org.springframework.beans.factory.aot.BeanRegistrationAotProcessor;
+import org.springframework.beans.factory.aot.BeanRegistrationCode;
+import org.springframework.beans.factory.support.RegisteredBean;
+import org.springframework.core.annotation.MergedAnnotations;
+import org.springframework.lang.Nullable;
+import org.springframework.util.ReflectionUtils;
+
+import java.util.stream.Stream;
+
+import static org.springframework.core.annotation.MergedAnnotations.SearchStrategy.TYPE_HIERARCHY;
+
+/**
+ * AOT {@code BeanRegistrationAotProcessor} that detects the presence of the
+ * {@link Tool} annotation on methods and creates the required reflection hints.
+ */
+class ToolBeanRegistrationAotProcessor implements BeanRegistrationAotProcessor {
+
+    @Override
+    @Nullable
+    public BeanRegistrationAotContribution processAheadOfTime(RegisteredBean registeredBean) {
+        Class<?> beanClass = registeredBean.getBeanClass();
+        MergedAnnotations.Search search = MergedAnnotations.search(TYPE_HIERARCHY);
+
+        boolean hasAnyToolAnnotatedMethods = Stream.of(ReflectionUtils.getDeclaredMethods(beanClass))
+                .anyMatch(method -> search.from(method).isPresent(Tool.class));
+
+        if (hasAnyToolAnnotatedMethods) {
+            return new AotContribution(beanClass);
+        }
+
+        return null;
+    }
+
+    private static class AotContribution implements BeanRegistrationAotContribution {
+
+        private final MemberCategory[] memberCategories = new MemberCategory[] { MemberCategory.INVOKE_DECLARED_METHODS,
+                MemberCategory.INVOKE_PUBLIC_METHODS };
+
+        private final Class<?> toolClass;
+
+        public AotContribution(Class<?> toolClass) {
+            this.toolClass = toolClass;
+        }
+
+        @Override
+        public void applyTo(GenerationContext generationContext, BeanRegistrationCode beanRegistrationCode) {
+            ReflectionHints reflectionHints = generationContext.getRuntimeHints().reflection();
+            reflectionHints.registerType(toolClass, memberCategories);
+        }
+
+    }
+
+}

patterns/tool-calling/tool-calling-openai/src/main/resources/META-INF/spring/aot.factories

@@ -0,0 +1,2 @@
+org.springframework.beans.factory.aot.BeanRegistrationAotProcessor=\
+    com.thomasvitale.ai.spring.ToolBeanRegistrationAotProcessor