Make the WebSearchTool class a bean if we pick it up

quarkiverse · Aug 30, 2024 · 2643945 · 2643945
1 parent b9c9cfc
commit 2643945
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/deployment/src/main/java/io/quarkiverse/langchain4j/deployment/ToolProcessor.java b/core/deployment/src/main/java/io/quarkiverse/langchain4j/deployment/ToolProcessor.java
@@ -92,6 +92,7 @@ public void telemetry(Capabilities capabilities, BuildProducer<AdditionalBeanBui
     @Record(ExecutionTime.STATIC_INIT)
     public void handleTools(CombinedIndexBuildItem indexBuildItem,
             ToolsRecorder recorder,
+            BuildProducer<AdditionalBeanBuildItem> additionalBeanProducer,
             RecorderContext recorderContext,
             BuildProducer<BytecodeTransformerBuildItem> transformerProducer,
             BuildProducer<GeneratedClassBuildItem> generatedClassProducer,
@@ -176,6 +177,14 @@ public void handleTools(CombinedIndexBuildItem indexBuildItem,
                     boolean ignoreToolMethod = ignoreToolMethod(toolMethod, index);
                     if (ignoreToolMethod) {
                         continue;
+                    } else {
+                        // The WebSearchTool class isn't a CDI bean, so if
+                        // we consider it as a tool, we have to also turn it into one
+                        if (LangChain4jDotNames.WEB_SEARCH_TOOL.equals(className)) {
+                            additionalBeanProducer.produce(AdditionalBeanBuildItem.builder()
+                                    .addBeanClass(className.toString())
+                                    .setUnremovable().build());
+                        }
                     }
 
                     AnnotationValue nameValue = instance.value("name");