File size: 790 Bytes
40a9423
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import { ChatMessage } from "../types.js";
import { LRUCache } from "lru-cache";

export type SessionState = {
  messages: ChatMessage[];
  summary: string;
  lastAccess: number;
};

const sessions = new LRUCache<string, SessionState>({
  max: 5000,
  ttl: 1000 * 60 * 60 * 6
});

export const memoryStore = {
  get(sessionId: string): SessionState {
    const existing = sessions.get(sessionId);
    if (existing) {
      existing.lastAccess = Date.now();
      return existing;
    }
    const created: SessionState = {
      messages: [],
      summary: "",
      lastAccess: Date.now()
    };
    sessions.set(sessionId, created);
    return created;
  },
  set(sessionId: string, state: SessionState) {
    state.lastAccess = Date.now();
    sessions.set(sessionId, state);
  }
};