Merge pull request #159 from quarkiverse/#158

Make sure ChatMemoryStore beans are unremovable
quarkiverse · Dec 14, 2023 · 0c8ab60 · 0c8ab60
2 parents b6a38ef + 868cdb3
commit 0c8ab60
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/core/deployment/src/main/java/io/quarkiverse/langchain4j/deployment/ChatMemoryProcessor.java b/core/deployment/src/main/java/io/quarkiverse/langchain4j/deployment/ChatMemoryProcessor.java
@@ -13,6 +13,7 @@
 import io.quarkiverse.langchain4j.runtime.aiservice.ChatMemoryConfig;
 import io.quarkus.arc.SyntheticCreationalContext;
 import io.quarkus.arc.deployment.SyntheticBeanBuildItem;
+import io.quarkus.arc.deployment.UnremovableBeanBuildItem;
 import io.quarkus.deployment.annotations.BuildProducer;
 import io.quarkus.deployment.annotations.BuildStep;
 import io.quarkus.deployment.annotations.ExecutionTime;
@@ -24,6 +25,7 @@ public class ChatMemoryProcessor {
     @Record(ExecutionTime.RUNTIME_INIT)
     void setupBeans(ChatMemoryBuildConfig buildConfig, ChatMemoryConfig runtimeConfig,
             ChatMemoryRecorder recorder,
+            BuildProducer<UnremovableBeanBuildItem> unremovableProducer,
             BuildProducer<SyntheticBeanBuildItem> syntheticBeanProducer) {
 
         Function<SyntheticCreationalContext<ChatMemoryProvider>, ChatMemoryProvider> fun;
@@ -47,5 +49,6 @@ void setupBeans(ChatMemoryBuildConfig buildConfig, ChatMemoryConfig runtimeConfi
         configurator.createWith(fun);
 
         syntheticBeanProducer.produce(configurator.done());
+        unremovableProducer.produce(UnremovableBeanBuildItem.beanTypes(ChatMemoryStore.class));
     }
 }
diff --git a/...in/java/io/quarkiverse/langchain4j/runtime/aiservice/InMemoryChatMemoryStoreProducer.java b/...in/java/io/quarkiverse/langchain4j/runtime/aiservice/InMemoryChatMemoryStoreProducer.java
@@ -6,7 +6,6 @@
 import dev.langchain4j.store.memory.chat.InMemoryChatMemoryStore;
 import io.quarkiverse.langchain4j.RegisterAiService;
 import io.quarkus.arc.DefaultBean;
-import io.quarkus.arc.Unremovable;
 
 /**
  * Creates the default {@link InMemoryChatMemoryStore} store to be used by classes annotated with {@link RegisterAiService}
@@ -16,7 +15,6 @@ public class InMemoryChatMemoryStoreProducer {
     @Produces
     @Singleton
     @DefaultBean
-    @Unremovable
     public InMemoryChatMemoryStore chatMemoryStore() {
         return new InMemoryChatMemoryStore();
     }