Be more defensive when removing chat memory ids

Fixes: #401 (cherry picked from commit 200bbdc)
quarkiverse · Mar 27, 2024 · 69a9633 · 69a9633
1 parent 3bd3a16
commit 69a9633
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...e/src/main/java/io/quarkiverse/langchain4j/runtime/aiservice/QuarkusAiServiceContext.java b/...e/src/main/java/io/quarkiverse/langchain4j/runtime/aiservice/QuarkusAiServiceContext.java
@@ -44,6 +44,9 @@ public void accept(Object memoryId, ChatMemory chatMemory) {
      * via {@link io.quarkiverse.langchain4j.ChatMemoryRemover}
      */
     public void removeChatMemoryIds(Object... ids) {
+        if (chatMemories == null) {
+            return;
+        }
         for (Object id : ids) {
             ChatMemory chatMemory = chatMemories.remove(id);
             if (chatMemory != null) {