Skip to content

Commit de35b9f

Browse files
committed
fixes
1 parent 916f97c commit de35b9f

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/main/java/io/twentysixty/ollama/hologram/chatbot/svc/OllamaService.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,8 @@ public class OllamaService {
6666

6767
public List<OllamaChatMessage> getMessagesFromHistory(UUID connectionId) {
6868

69-
List<History> histories = this.getHistory(connectionId);
69+
Session s = em.find(Session.class, connectionId);
70+
List<History> histories = this.getHistory(s);
7071

7172
List<OllamaChatMessage> messages = new ArrayList<OllamaChatMessage>(histories.size());
7273

@@ -101,9 +102,9 @@ public String getChatResponse(List<OllamaChatMessage> messages) throws OllamaBas
101102
return chatResult.getResponse();
102103
}
103104

104-
public List<History> getHistory(UUID connectionId) {
105+
public List<History> getHistory(Session session) {
105106
Query q = em.createNamedQuery("History.find");
106-
q.setParameter("connectionId", connectionId);
107+
q.setParameter("session", session);
107108
q.setMaxResults(maxHistorySize);
108109
return q.getResultList();
109110
}

0 commit comments

Comments
 (0)