Add entries variable to user message in AI chat (JabRef#12184)

* Fix AI settings * Add `entries` variable to user message template
JabRef · Nov 11, 2024 · 9d8c590 · 9d8c590
1 parent 71b238c
commit 9d8c590
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/main/java/org/jabref/logic/ai/chatting/AiChatLogic.java b/src/main/java/org/jabref/logic/ai/chatting/AiChatLogic.java
@@ -171,7 +171,7 @@ public AiMessage execute(UserMessage message) {
 
         chatMemory.messages().forEach(tempChatMemory::add);
 
-        tempChatMemory.add(new UserMessage(templatesService.makeChattingUserMessage(message.singleText(), excerpts)));
+        tempChatMemory.add(new UserMessage(templatesService.makeChattingUserMessage(entries, message.singleText(), excerpts)));
         chatMemory.add(message);
 
         AiMessage aiMessage = chatLanguageModel.generate(tempChatMemory.messages()).content();

diff --git a/src/main/java/org/jabref/logic/ai/templates/TemplatesService.java b/src/main/java/org/jabref/logic/ai/templates/TemplatesService.java
@@ -30,8 +30,9 @@ public String makeChattingSystemMessage(List<BibEntry> entries) {
         return makeTemplate(AiTemplate.CHATTING_SYSTEM_MESSAGE, context);
     }
 
-    public String makeChattingUserMessage(String message, List<PaperExcerpt> excerpts) {
+    public String makeChattingUserMessage(List<BibEntry> entries, String message, List<PaperExcerpt> excerpts) {
         VelocityContext context = new VelocityContext(baseContext);
+        context.put("entries", entries);
         context.put("message", message);
         context.put("excerpts", excerpts);