@hazeljs/rag 0.2.0-beta.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +379 -0
- package/dist/agentic/agentic-rag.service.d.ts +49 -0
- package/dist/agentic/agentic-rag.service.d.ts.map +1 -0
- package/dist/agentic/agentic-rag.service.js +149 -0
- package/dist/agentic/agentic-rag.service.js.map +1 -0
- package/dist/agentic/decorators/active-learning.decorator.d.ts +19 -0
- package/dist/agentic/decorators/active-learning.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/active-learning.decorator.js +98 -0
- package/dist/agentic/decorators/active-learning.decorator.js.map +1 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.d.ts +16 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.js +114 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.js.map +1 -0
- package/dist/agentic/decorators/cached.decorator.d.ts +18 -0
- package/dist/agentic/decorators/cached.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/cached.decorator.js +93 -0
- package/dist/agentic/decorators/cached.decorator.js.map +1 -0
- package/dist/agentic/decorators/context-aware.decorator.d.ts +16 -0
- package/dist/agentic/decorators/context-aware.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/context-aware.decorator.js +169 -0
- package/dist/agentic/decorators/context-aware.decorator.js.map +1 -0
- package/dist/agentic/decorators/corrective-rag.decorator.d.ts +15 -0
- package/dist/agentic/decorators/corrective-rag.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/corrective-rag.decorator.js +146 -0
- package/dist/agentic/decorators/corrective-rag.decorator.js.map +1 -0
- package/dist/agentic/decorators/hyde.decorator.d.ts +14 -0
- package/dist/agentic/decorators/hyde.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/hyde.decorator.js +90 -0
- package/dist/agentic/decorators/hyde.decorator.js.map +1 -0
- package/dist/agentic/decorators/index.d.ts +16 -0
- package/dist/agentic/decorators/index.d.ts.map +1 -0
- package/dist/agentic/decorators/index.js +32 -0
- package/dist/agentic/decorators/index.js.map +1 -0
- package/dist/agentic/decorators/multi-hop.decorator.d.ts +14 -0
- package/dist/agentic/decorators/multi-hop.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/multi-hop.decorator.js +120 -0
- package/dist/agentic/decorators/multi-hop.decorator.js.map +1 -0
- package/dist/agentic/decorators/query-planner.decorator.d.ts +19 -0
- package/dist/agentic/decorators/query-planner.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/query-planner.decorator.js +227 -0
- package/dist/agentic/decorators/query-planner.decorator.js.map +1 -0
- package/dist/agentic/decorators/query-rewriter.decorator.d.ts +15 -0
- package/dist/agentic/decorators/query-rewriter.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/query-rewriter.decorator.js +140 -0
- package/dist/agentic/decorators/query-rewriter.decorator.js.map +1 -0
- package/dist/agentic/decorators/self-reflective.decorator.d.ts +18 -0
- package/dist/agentic/decorators/self-reflective.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/self-reflective.decorator.js +210 -0
- package/dist/agentic/decorators/self-reflective.decorator.js.map +1 -0
- package/dist/agentic/decorators/source-verification.decorator.d.ts +15 -0
- package/dist/agentic/decorators/source-verification.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/source-verification.decorator.js +121 -0
- package/dist/agentic/decorators/source-verification.decorator.js.map +1 -0
- package/dist/agentic/index.d.ts +9 -0
- package/dist/agentic/index.d.ts.map +1 -0
- package/dist/agentic/index.js +25 -0
- package/dist/agentic/index.js.map +1 -0
- package/dist/agentic/types.d.ts +209 -0
- package/dist/agentic/types.d.ts.map +1 -0
- package/dist/agentic/types.js +7 -0
- package/dist/agentic/types.js.map +1 -0
- package/dist/decorators/embeddable.decorator.d.ts +31 -0
- package/dist/decorators/embeddable.decorator.d.ts.map +1 -0
- package/dist/decorators/embeddable.decorator.js +44 -0
- package/dist/decorators/embeddable.decorator.js.map +1 -0
- package/dist/decorators/rag.decorator.d.ts +58 -0
- package/dist/decorators/rag.decorator.d.ts.map +1 -0
- package/dist/decorators/rag.decorator.js +78 -0
- package/dist/decorators/rag.decorator.js.map +1 -0
- package/dist/decorators/semantic-search.decorator.d.ts +69 -0
- package/dist/decorators/semantic-search.decorator.d.ts.map +1 -0
- package/dist/decorators/semantic-search.decorator.js +116 -0
- package/dist/decorators/semantic-search.decorator.js.map +1 -0
- package/dist/embeddings/cohere-embeddings.d.ts +33 -0
- package/dist/embeddings/cohere-embeddings.d.ts.map +1 -0
- package/dist/embeddings/cohere-embeddings.js +91 -0
- package/dist/embeddings/cohere-embeddings.js.map +1 -0
- package/dist/embeddings/openai-embeddings.d.ts +21 -0
- package/dist/embeddings/openai-embeddings.d.ts.map +1 -0
- package/dist/embeddings/openai-embeddings.js +53 -0
- package/dist/embeddings/openai-embeddings.js.map +1 -0
- package/dist/index.d.ts +27 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +54 -0
- package/dist/index.js.map +1 -0
- package/dist/memory/index.d.ts +11 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +31 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/memory/memory-manager.d.ts +96 -0
- package/dist/memory/memory-manager.d.ts.map +1 -0
- package/dist/memory/memory-manager.js +369 -0
- package/dist/memory/memory-manager.js.map +1 -0
- package/dist/memory/memory-store.interface.d.ts +73 -0
- package/dist/memory/memory-store.interface.d.ts.map +1 -0
- package/dist/memory/memory-store.interface.js +6 -0
- package/dist/memory/memory-store.interface.js.map +1 -0
- package/dist/memory/stores/buffer-memory.d.ts +47 -0
- package/dist/memory/stores/buffer-memory.d.ts.map +1 -0
- package/dist/memory/stores/buffer-memory.js +280 -0
- package/dist/memory/stores/buffer-memory.js.map +1 -0
- package/dist/memory/stores/hybrid-memory.d.ts +49 -0
- package/dist/memory/stores/hybrid-memory.d.ts.map +1 -0
- package/dist/memory/stores/hybrid-memory.js +194 -0
- package/dist/memory/stores/hybrid-memory.js.map +1 -0
- package/dist/memory/stores/vector-memory.d.ts +48 -0
- package/dist/memory/stores/vector-memory.d.ts.map +1 -0
- package/dist/memory/stores/vector-memory.js +312 -0
- package/dist/memory/stores/vector-memory.js.map +1 -0
- package/dist/memory/types.d.ts +119 -0
- package/dist/memory/types.d.ts.map +1 -0
- package/dist/memory/types.js +18 -0
- package/dist/memory/types.js.map +1 -0
- package/dist/rag-pipeline-with-memory.d.ts +68 -0
- package/dist/rag-pipeline-with-memory.d.ts.map +1 -0
- package/dist/rag-pipeline-with-memory.js +186 -0
- package/dist/rag-pipeline-with-memory.js.map +1 -0
- package/dist/rag-pipeline.d.ts +59 -0
- package/dist/rag-pipeline.d.ts.map +1 -0
- package/dist/rag-pipeline.js +181 -0
- package/dist/rag-pipeline.js.map +1 -0
- package/dist/rag.module.d.ts +26 -0
- package/dist/rag.module.d.ts.map +1 -0
- package/dist/rag.module.js +40 -0
- package/dist/rag.module.js.map +1 -0
- package/dist/rag.service.d.ts +95 -0
- package/dist/rag.service.d.ts.map +1 -0
- package/dist/rag.service.js +174 -0
- package/dist/rag.service.js.map +1 -0
- package/dist/retrieval/bm25.d.ts +57 -0
- package/dist/retrieval/bm25.d.ts.map +1 -0
- package/dist/retrieval/bm25.js +106 -0
- package/dist/retrieval/bm25.js.map +1 -0
- package/dist/retrieval/hybrid-search.d.ts +48 -0
- package/dist/retrieval/hybrid-search.d.ts.map +1 -0
- package/dist/retrieval/hybrid-search.js +123 -0
- package/dist/retrieval/hybrid-search.js.map +1 -0
- package/dist/retrieval/multi-query.d.ts +38 -0
- package/dist/retrieval/multi-query.d.ts.map +1 -0
- package/dist/retrieval/multi-query.js +135 -0
- package/dist/retrieval/multi-query.js.map +1 -0
- package/dist/text-splitters/recursive-text-splitter.d.ts +21 -0
- package/dist/text-splitters/recursive-text-splitter.d.ts.map +1 -0
- package/dist/text-splitters/recursive-text-splitter.js +95 -0
- package/dist/text-splitters/recursive-text-splitter.js.map +1 -0
- package/dist/types/index.d.ts +144 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +16 -0
- package/dist/types/index.js.map +1 -0
- package/dist/utils/similarity.d.ts +16 -0
- package/dist/utils/similarity.d.ts.map +1 -0
- package/dist/utils/similarity.js +58 -0
- package/dist/utils/similarity.js.map +1 -0
- package/dist/vector-stores/chroma.store.d.ts +42 -0
- package/dist/vector-stores/chroma.store.d.ts.map +1 -0
- package/dist/vector-stores/chroma.store.js +242 -0
- package/dist/vector-stores/chroma.store.js.map +1 -0
- package/dist/vector-stores/memory-vector-store.d.ts +20 -0
- package/dist/vector-stores/memory-vector-store.d.ts.map +1 -0
- package/dist/vector-stores/memory-vector-store.js +94 -0
- package/dist/vector-stores/memory-vector-store.js.map +1 -0
- package/dist/vector-stores/pinecone.store.d.ts +34 -0
- package/dist/vector-stores/pinecone.store.d.ts.map +1 -0
- package/dist/vector-stores/pinecone.store.js +146 -0
- package/dist/vector-stores/pinecone.store.js.map +1 -0
- package/dist/vector-stores/qdrant.store.d.ts +31 -0
- package/dist/vector-stores/qdrant.store.d.ts.map +1 -0
- package/dist/vector-stores/qdrant.store.js +173 -0
- package/dist/vector-stores/qdrant.store.js.map +1 -0
- package/dist/vector-stores/weaviate.store.d.ts +37 -0
- package/dist/vector-stores/weaviate.store.d.ts.map +1 -0
- package/dist/vector-stores/weaviate.store.js +226 -0
- package/dist/vector-stores/weaviate.store.js.map +1 -0
- package/package.json +87 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-manager.d.ts","sourceRoot":"","sources":["../../src/memory/memory-manager.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,OAAO,EACL,MAAM,EAEN,YAAY,EACZ,OAAO,EACP,MAAM,EACN,WAAW,EACX,mBAAmB,EACpB,MAAM,SAAS,CAAC;AAGjB;;;GAGG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,KAAK,CAAc;IAC3B,OAAO,CAAC,MAAM,CAAe;gBAEjB,KAAK,EAAE,WAAW,EAAE,MAAM,GAAE,YAAiB;IAcnD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAMjC;;OAEG;IACG,UAAU,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAuBvF;;OAEG;IACG,sBAAsB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAenF;;OAEG;IACG,qBAAqB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAQ/D;;OAEG;IACG,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAczD;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBpE;;OAEG;IACG,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAgBrD;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IA8BzE;;OAEG;IACG,cAAc,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAoB3D;;OAEG;IACG,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAelF;;OAEG;IACG,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAUlF;;OAEG;IACG,UAAU,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY/D;;OAEG;IACG,UAAU,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAwC/E;;OAEG;IACG,UAAU,CAAC,GAAG,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAiBlE;;OAEG;IACG,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAcpD;;OAEG;IACG,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAItF;;OAEG;IACG,WAAW,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAIxD;;OAEG;IACG,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,SAAS,EAAE,WAAW,CAAC;IAM1E;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAa3B;;OAEG;YACW,iBAAiB;IAuB/B;;OAEG;YACW,kBAAkB;CAiBjC"}
|
|
@@ -0,0 +1,369 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Memory Manager - High-level interface for managing different types of memory
|
|
4
|
+
*/
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.MemoryManager = void 0;
|
|
7
|
+
const types_1 = require("./types");
|
|
8
|
+
const crypto_1 = require("crypto");
|
|
9
|
+
/**
|
|
10
|
+
* Central manager for all memory operations
|
|
11
|
+
* Provides high-level APIs for conversation, entity, fact, and working memory
|
|
12
|
+
*/
|
|
13
|
+
class MemoryManager {
|
|
14
|
+
constructor(store, config = {}) {
|
|
15
|
+
this.store = store;
|
|
16
|
+
this.config = {
|
|
17
|
+
maxConversationLength: config.maxConversationLength || 20,
|
|
18
|
+
summarizeAfter: config.summarizeAfter || 50,
|
|
19
|
+
entityExtraction: config.entityExtraction ?? true,
|
|
20
|
+
importanceScoring: config.importanceScoring ?? true,
|
|
21
|
+
memoryDecay: config.memoryDecay ?? false,
|
|
22
|
+
decayRate: config.decayRate || 0.1,
|
|
23
|
+
maxWorkingMemorySize: config.maxWorkingMemorySize || 10,
|
|
24
|
+
...config,
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
async initialize() {
|
|
28
|
+
await this.store.initialize();
|
|
29
|
+
}
|
|
30
|
+
// ==================== Conversation Memory ====================
|
|
31
|
+
/**
|
|
32
|
+
* Add a message to conversation history
|
|
33
|
+
*/
|
|
34
|
+
async addMessage(message, sessionId, userId) {
|
|
35
|
+
const memory = {
|
|
36
|
+
id: (0, crypto_1.randomUUID)(),
|
|
37
|
+
type: types_1.MemoryType.CONVERSATION,
|
|
38
|
+
content: `${message.role}: ${message.content}`,
|
|
39
|
+
metadata: {
|
|
40
|
+
timestamp: message.timestamp || new Date(),
|
|
41
|
+
sessionId,
|
|
42
|
+
userId,
|
|
43
|
+
role: message.role,
|
|
44
|
+
importance: this.calculateImportance(message.content),
|
|
45
|
+
...message.metadata,
|
|
46
|
+
},
|
|
47
|
+
};
|
|
48
|
+
const id = await this.store.save(memory);
|
|
49
|
+
// Check if we need to summarize
|
|
50
|
+
await this.checkAndSummarize(sessionId);
|
|
51
|
+
return id;
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* Get conversation history for a session
|
|
55
|
+
*/
|
|
56
|
+
async getConversationHistory(sessionId, limit) {
|
|
57
|
+
const memories = await this.store.retrieve({
|
|
58
|
+
sessionId,
|
|
59
|
+
types: [types_1.MemoryType.CONVERSATION],
|
|
60
|
+
limit: limit || this.config.maxConversationLength,
|
|
61
|
+
});
|
|
62
|
+
return memories.map((m) => ({
|
|
63
|
+
role: m.metadata.role,
|
|
64
|
+
content: m.content.replace(/^(user|assistant|system): /, ''),
|
|
65
|
+
timestamp: m.metadata.timestamp,
|
|
66
|
+
metadata: m.metadata,
|
|
67
|
+
}));
|
|
68
|
+
}
|
|
69
|
+
/**
|
|
70
|
+
* Summarize conversation history
|
|
71
|
+
*/
|
|
72
|
+
async summarizeConversation(sessionId) {
|
|
73
|
+
return this.store.summarize({
|
|
74
|
+
sessionId,
|
|
75
|
+
maxLength: 1000,
|
|
76
|
+
strategy: 'extractive',
|
|
77
|
+
});
|
|
78
|
+
}
|
|
79
|
+
/**
|
|
80
|
+
* Clear conversation history for a session
|
|
81
|
+
*/
|
|
82
|
+
async clearConversation(sessionId) {
|
|
83
|
+
const memories = await this.store.retrieve({
|
|
84
|
+
sessionId,
|
|
85
|
+
types: [types_1.MemoryType.CONVERSATION],
|
|
86
|
+
});
|
|
87
|
+
const ids = memories.map((m) => m.id);
|
|
88
|
+
if (ids.length > 0) {
|
|
89
|
+
await this.store.deleteBatch(ids);
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
// ==================== Entity Memory ====================
|
|
93
|
+
/**
|
|
94
|
+
* Track an entity mentioned in conversation
|
|
95
|
+
*/
|
|
96
|
+
async trackEntity(entity, sessionId) {
|
|
97
|
+
const memory = {
|
|
98
|
+
id: (0, crypto_1.randomUUID)(),
|
|
99
|
+
type: types_1.MemoryType.ENTITY,
|
|
100
|
+
content: JSON.stringify(entity),
|
|
101
|
+
metadata: {
|
|
102
|
+
timestamp: new Date(),
|
|
103
|
+
sessionId,
|
|
104
|
+
entityName: entity.name,
|
|
105
|
+
entityType: entity.type,
|
|
106
|
+
entities: [entity.name],
|
|
107
|
+
importance: 0.8,
|
|
108
|
+
},
|
|
109
|
+
};
|
|
110
|
+
await this.store.save(memory);
|
|
111
|
+
}
|
|
112
|
+
/**
|
|
113
|
+
* Get entity by name
|
|
114
|
+
*/
|
|
115
|
+
async getEntity(name) {
|
|
116
|
+
const memories = await this.store.retrieve({
|
|
117
|
+
entities: [name],
|
|
118
|
+
types: [types_1.MemoryType.ENTITY],
|
|
119
|
+
limit: 1,
|
|
120
|
+
});
|
|
121
|
+
if (memories.length === 0)
|
|
122
|
+
return null;
|
|
123
|
+
try {
|
|
124
|
+
return JSON.parse(memories[0].content);
|
|
125
|
+
}
|
|
126
|
+
catch {
|
|
127
|
+
return null;
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
/**
|
|
131
|
+
* Update entity information
|
|
132
|
+
*/
|
|
133
|
+
async updateEntity(name, updates) {
|
|
134
|
+
const existing = await this.getEntity(name);
|
|
135
|
+
if (!existing) {
|
|
136
|
+
throw new Error(`Entity ${name} not found`);
|
|
137
|
+
}
|
|
138
|
+
const updated = {
|
|
139
|
+
...existing,
|
|
140
|
+
...updates,
|
|
141
|
+
lastSeen: new Date(),
|
|
142
|
+
mentions: existing.mentions + 1,
|
|
143
|
+
};
|
|
144
|
+
const memories = await this.store.retrieve({
|
|
145
|
+
entities: [name],
|
|
146
|
+
types: [types_1.MemoryType.ENTITY],
|
|
147
|
+
limit: 1,
|
|
148
|
+
});
|
|
149
|
+
if (memories.length > 0) {
|
|
150
|
+
await this.store.update(memories[0].id, {
|
|
151
|
+
content: JSON.stringify(updated),
|
|
152
|
+
metadata: {
|
|
153
|
+
...memories[0].metadata,
|
|
154
|
+
timestamp: new Date(),
|
|
155
|
+
},
|
|
156
|
+
});
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
/**
|
|
160
|
+
* Get all tracked entities
|
|
161
|
+
*/
|
|
162
|
+
async getAllEntities(sessionId) {
|
|
163
|
+
const memories = await this.store.retrieve({
|
|
164
|
+
sessionId,
|
|
165
|
+
types: [types_1.MemoryType.ENTITY],
|
|
166
|
+
limit: 1000,
|
|
167
|
+
});
|
|
168
|
+
return memories
|
|
169
|
+
.map((m) => {
|
|
170
|
+
try {
|
|
171
|
+
return JSON.parse(m.content);
|
|
172
|
+
}
|
|
173
|
+
catch {
|
|
174
|
+
return null;
|
|
175
|
+
}
|
|
176
|
+
})
|
|
177
|
+
.filter((e) => e !== null);
|
|
178
|
+
}
|
|
179
|
+
// ==================== Semantic Memory (Facts) ====================
|
|
180
|
+
/**
|
|
181
|
+
* Store a fact or piece of knowledge
|
|
182
|
+
*/
|
|
183
|
+
async storeFact(fact, metadata) {
|
|
184
|
+
const memory = {
|
|
185
|
+
id: (0, crypto_1.randomUUID)(),
|
|
186
|
+
type: types_1.MemoryType.FACT,
|
|
187
|
+
content: fact,
|
|
188
|
+
metadata: {
|
|
189
|
+
timestamp: new Date(),
|
|
190
|
+
importance: this.calculateImportance(fact),
|
|
191
|
+
...metadata,
|
|
192
|
+
},
|
|
193
|
+
};
|
|
194
|
+
return this.store.save(memory);
|
|
195
|
+
}
|
|
196
|
+
/**
|
|
197
|
+
* Recall facts related to a query
|
|
198
|
+
*/
|
|
199
|
+
async recallFacts(query, options) {
|
|
200
|
+
const memories = await this.store.search(query, {
|
|
201
|
+
types: [types_1.MemoryType.FACT],
|
|
202
|
+
topK: options?.topK || 5,
|
|
203
|
+
...options,
|
|
204
|
+
});
|
|
205
|
+
return memories.map((m) => m.content);
|
|
206
|
+
}
|
|
207
|
+
/**
|
|
208
|
+
* Update a fact
|
|
209
|
+
*/
|
|
210
|
+
async updateFact(id, newContent) {
|
|
211
|
+
await this.store.update(id, {
|
|
212
|
+
content: newContent,
|
|
213
|
+
metadata: {
|
|
214
|
+
timestamp: new Date(),
|
|
215
|
+
updated: true,
|
|
216
|
+
},
|
|
217
|
+
});
|
|
218
|
+
}
|
|
219
|
+
// ==================== Working Memory ====================
|
|
220
|
+
/**
|
|
221
|
+
* Set a value in working memory (temporary context)
|
|
222
|
+
*/
|
|
223
|
+
async setContext(key, value, sessionId) {
|
|
224
|
+
// Check if context already exists
|
|
225
|
+
const existing = await this.store.retrieve({
|
|
226
|
+
sessionId,
|
|
227
|
+
types: [types_1.MemoryType.WORKING],
|
|
228
|
+
limit: 1000,
|
|
229
|
+
});
|
|
230
|
+
const existingContext = existing.find((m) => m.metadata.contextKey === key);
|
|
231
|
+
if (existingContext) {
|
|
232
|
+
// Update existing
|
|
233
|
+
await this.store.update(existingContext.id, {
|
|
234
|
+
content: JSON.stringify(value),
|
|
235
|
+
metadata: {
|
|
236
|
+
...existingContext.metadata,
|
|
237
|
+
timestamp: new Date(),
|
|
238
|
+
},
|
|
239
|
+
});
|
|
240
|
+
}
|
|
241
|
+
else {
|
|
242
|
+
// Create new
|
|
243
|
+
const memory = {
|
|
244
|
+
id: (0, crypto_1.randomUUID)(),
|
|
245
|
+
type: types_1.MemoryType.WORKING,
|
|
246
|
+
content: JSON.stringify(value),
|
|
247
|
+
metadata: {
|
|
248
|
+
timestamp: new Date(),
|
|
249
|
+
sessionId,
|
|
250
|
+
contextKey: key,
|
|
251
|
+
importance: 0.5,
|
|
252
|
+
},
|
|
253
|
+
};
|
|
254
|
+
await this.store.save(memory);
|
|
255
|
+
}
|
|
256
|
+
// Enforce max working memory size
|
|
257
|
+
await this.pruneWorkingMemory(sessionId);
|
|
258
|
+
}
|
|
259
|
+
/**
|
|
260
|
+
* Get a value from working memory
|
|
261
|
+
*/
|
|
262
|
+
async getContext(key, sessionId) {
|
|
263
|
+
const memories = await this.store.retrieve({
|
|
264
|
+
sessionId,
|
|
265
|
+
types: [types_1.MemoryType.WORKING],
|
|
266
|
+
limit: 1000,
|
|
267
|
+
});
|
|
268
|
+
const memory = memories.find((m) => m.metadata.contextKey === key);
|
|
269
|
+
if (!memory)
|
|
270
|
+
return null;
|
|
271
|
+
try {
|
|
272
|
+
return JSON.parse(memory.content);
|
|
273
|
+
}
|
|
274
|
+
catch {
|
|
275
|
+
return memory.content;
|
|
276
|
+
}
|
|
277
|
+
}
|
|
278
|
+
/**
|
|
279
|
+
* Clear working memory for a session
|
|
280
|
+
*/
|
|
281
|
+
async clearContext(sessionId) {
|
|
282
|
+
const memories = await this.store.retrieve({
|
|
283
|
+
sessionId,
|
|
284
|
+
types: [types_1.MemoryType.WORKING],
|
|
285
|
+
});
|
|
286
|
+
const ids = memories.map((m) => m.id);
|
|
287
|
+
if (ids.length > 0) {
|
|
288
|
+
await this.store.deleteBatch(ids);
|
|
289
|
+
}
|
|
290
|
+
}
|
|
291
|
+
// ==================== Memory Retrieval ====================
|
|
292
|
+
/**
|
|
293
|
+
* Get relevant memories for a query
|
|
294
|
+
*/
|
|
295
|
+
async relevantMemories(query, options) {
|
|
296
|
+
return this.store.search(query, options);
|
|
297
|
+
}
|
|
298
|
+
/**
|
|
299
|
+
* Get memories by query parameters
|
|
300
|
+
*/
|
|
301
|
+
async getMemories(query) {
|
|
302
|
+
return this.store.retrieve(query);
|
|
303
|
+
}
|
|
304
|
+
/**
|
|
305
|
+
* Get memory statistics
|
|
306
|
+
*/
|
|
307
|
+
async getStats(sessionId) {
|
|
308
|
+
return this.store.getStats(sessionId);
|
|
309
|
+
}
|
|
310
|
+
// ==================== Private Helper Methods ====================
|
|
311
|
+
/**
|
|
312
|
+
* Calculate importance score for content
|
|
313
|
+
*/
|
|
314
|
+
calculateImportance(content) {
|
|
315
|
+
if (!this.config.importanceScoring)
|
|
316
|
+
return 0.5;
|
|
317
|
+
// Simple heuristic: longer content and questions are more important
|
|
318
|
+
let score = 0.5;
|
|
319
|
+
if (content.length > 100)
|
|
320
|
+
score += 0.2;
|
|
321
|
+
if (content.includes('?'))
|
|
322
|
+
score += 0.1;
|
|
323
|
+
if (content.includes('important') || content.includes('remember'))
|
|
324
|
+
score += 0.2;
|
|
325
|
+
return Math.min(score, 1.0);
|
|
326
|
+
}
|
|
327
|
+
/**
|
|
328
|
+
* Check if conversation needs summarization
|
|
329
|
+
*/
|
|
330
|
+
async checkAndSummarize(sessionId) {
|
|
331
|
+
if (!this.config.summarizeAfter)
|
|
332
|
+
return;
|
|
333
|
+
const memories = await this.store.retrieve({
|
|
334
|
+
sessionId,
|
|
335
|
+
types: [types_1.MemoryType.CONVERSATION],
|
|
336
|
+
limit: 1000,
|
|
337
|
+
});
|
|
338
|
+
if (memories.length >= this.config.summarizeAfter) {
|
|
339
|
+
// Summarize oldest messages
|
|
340
|
+
const toSummarize = memories.slice(this.config.maxConversationLength);
|
|
341
|
+
if (toSummarize.length > 0) {
|
|
342
|
+
await this.store.consolidate(toSummarize);
|
|
343
|
+
// Delete original messages
|
|
344
|
+
const ids = toSummarize.map((m) => m.id);
|
|
345
|
+
await this.store.deleteBatch(ids);
|
|
346
|
+
}
|
|
347
|
+
}
|
|
348
|
+
}
|
|
349
|
+
/**
|
|
350
|
+
* Prune working memory to max size
|
|
351
|
+
*/
|
|
352
|
+
async pruneWorkingMemory(sessionId) {
|
|
353
|
+
const memories = await this.store.retrieve({
|
|
354
|
+
sessionId,
|
|
355
|
+
types: [types_1.MemoryType.WORKING],
|
|
356
|
+
limit: 1000,
|
|
357
|
+
});
|
|
358
|
+
if (memories.length > this.config.maxWorkingMemorySize) {
|
|
359
|
+
// Remove oldest
|
|
360
|
+
const toRemove = memories
|
|
361
|
+
.sort((a, b) => a.metadata.timestamp.getTime() - b.metadata.timestamp.getTime())
|
|
362
|
+
.slice(0, memories.length - this.config.maxWorkingMemorySize);
|
|
363
|
+
const ids = toRemove.map((m) => m.id);
|
|
364
|
+
await this.store.deleteBatch(ids);
|
|
365
|
+
}
|
|
366
|
+
}
|
|
367
|
+
}
|
|
368
|
+
exports.MemoryManager = MemoryManager;
|
|
369
|
+
//# sourceMappingURL=memory-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-manager.js","sourceRoot":"","sources":["../../src/memory/memory-manager.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAGH,mCAQiB;AACjB,mCAAoC;AAEpC;;;GAGG;AACH,MAAa,aAAa;IAIxB,YAAY,KAAkB,EAAE,SAAuB,EAAE;QACvD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG;YACZ,qBAAqB,EAAE,MAAM,CAAC,qBAAqB,IAAI,EAAE;YACzD,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,EAAE;YAC3C,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,IAAI;YACjD,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,IAAI;YACnD,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,KAAK;YACxC,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,GAAG;YAClC,oBAAoB,EAAE,MAAM,CAAC,oBAAoB,IAAI,EAAE;YACvD,GAAG,MAAM;SACV,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;IAChC,CAAC;IAED,gEAAgE;IAEhE;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,OAAgB,EAAE,SAAiB,EAAE,MAAe;QACnE,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,IAAA,mBAAU,GAAE;YAChB,IAAI,EAAE,kBAAU,CAAC,YAAY;YAC7B,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,OAAO,EAAE;YAC9C,QAAQ,EAAE;gBACR,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE;gBAC1C,SAAS;gBACT,MAAM;gBACN,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,UAAU,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,OAAO,CAAC;gBACrD,GAAG,OAAO,CAAC,QAAQ;aACpB;SACF,CAAC;QAEF,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEzC,gCAAgC;QAChC,MAAM,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAExC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,sBAAsB,CAAC,SAAiB,EAAE,KAAc;QAC5D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,YAAY,CAAC;YAChC,KAAK,EAAE,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,qBAAqB;SAClD,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1B,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAuC;YACxD,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,4BAA4B,EAAE,EAAE,CAAC;YAC5D,SAAS,EAAE,CAAC,CAAC,QAAQ,CAAC,SAAS;YAC/B,QAAQ,EAAE,CAAC,CAAC,QAAQ;SACrB,CAAC,CAAC,CAAC;IACN,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,qBAAqB,CAAC,SAAiB;QAC3C,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;YAC1B,SAAS;YACT,SAAS,EAAE,IAAI;YACf,QAAQ,EAAE,YAAY;SACvB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QACvC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,YAAY,CAAC;SACjC,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,0DAA0D;IAE1D;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,SAAkB;QAClD,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,IAAA,mBAAU,GAAE;YAChB,IAAI,EAAE,kBAAU,CAAC,MAAM;YACvB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;YAC/B,QAAQ,EAAE;gBACR,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,SAAS;gBACT,UAAU,EAAE,MAAM,CAAC,IAAI;gBACvB,UAAU,EAAE,MAAM,CAAC,IAAI;gBACvB,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;gBACvB,UAAU,EAAE,GAAG;aAChB;SACF,CAAC;QAEF,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,IAAY;QAC1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,QAAQ,EAAE,CAAC,IAAI,CAAC;YAChB,KAAK,EAAE,CAAC,kBAAU,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAEvC,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAW,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,IAAY,EAAE,OAAwB;QACvD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,YAAY,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,OAAO,GAAW;YACtB,GAAG,QAAQ;YACX,GAAG,OAAO;YACV,QAAQ,EAAE,IAAI,IAAI,EAAE;YACpB,QAAQ,EAAE,QAAQ,CAAC,QAAQ,GAAG,CAAC;SAChC,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,QAAQ,EAAE,CAAC,IAAI,CAAC;YAChB,KAAK,EAAE,CAAC,kBAAU,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE;gBACtC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAChC,QAAQ,EAAE;oBACR,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ;oBACvB,SAAS,EAAE,IAAI,IAAI,EAAE;iBACtB;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,SAAkB;QACrC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,MAAM,CAAC;YAC1B,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,OAAO,QAAQ;aACZ,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAW,CAAC;YACzC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED,oEAAoE;IAEpE;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,QAAkC;QAC9D,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,IAAA,mBAAU,GAAE;YAChB,IAAI,EAAE,kBAAU,CAAC,IAAI;YACrB,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE;gBACR,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,UAAU,EAAE,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC1C,GAAG,QAAQ;aACZ;SACF,CAAC;QAEF,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,KAAa,EAAE,OAA6B;QAC5D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE;YAC9C,KAAK,EAAE,CAAC,kBAAU,CAAC,IAAI,CAAC;YACxB,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,CAAC;YACxB,GAAG,OAAO;SACX,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,EAAU,EAAE,UAAkB;QAC7C,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE;YAC1B,OAAO,EAAE,UAAU;YACnB,QAAQ,EAAE;gBACR,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,OAAO,EAAE,IAAI;aACd;SACF,CAAC,CAAC;IACL,CAAC;IAED,2DAA2D;IAE3D;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,KAAc,EAAE,SAAiB;QAC7D,kCAAkC;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,OAAO,CAAC;YAC3B,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,MAAM,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,KAAK,GAAG,CAAC,CAAC;QAE5E,IAAI,eAAe,EAAE,CAAC;YACpB,kBAAkB;YAClB,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE,EAAE;gBAC1C,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;gBAC9B,QAAQ,EAAE;oBACR,GAAG,eAAe,CAAC,QAAQ;oBAC3B,SAAS,EAAE,IAAI,IAAI,EAAE;iBACtB;aACF,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,aAAa;YACb,MAAM,MAAM,GAAW;gBACrB,EAAE,EAAE,IAAA,mBAAU,GAAE;gBAChB,IAAI,EAAE,kBAAU,CAAC,OAAO;gBACxB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;gBAC9B,QAAQ,EAAE;oBACR,SAAS,EAAE,IAAI,IAAI,EAAE;oBACrB,SAAS;oBACT,UAAU,EAAE,GAAG;oBACf,UAAU,EAAE,GAAG;iBAChB;aACF,CAAC;YAEF,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,kCAAkC;QAClC,MAAM,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,SAAiB;QAC7C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,OAAO,CAAC;YAC3B,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,KAAK,GAAG,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAEzB,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,MAAM,CAAC,OAAO,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,SAAiB;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,OAAO,CAAC;SAC5B,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,6DAA6D;IAE7D;;OAEG;IACH,KAAK,CAAC,gBAAgB,CAAC,KAAa,EAAE,OAA4B;QAChE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,KAAkB;QAClC,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,SAAkB;QAC/B,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAED,mEAAmE;IAEnE;;OAEG;IACK,mBAAmB,CAAC,OAAe;QACzC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB;YAAE,OAAO,GAAG,CAAC;QAE/C,oEAAoE;QACpE,IAAI,KAAK,GAAG,GAAG,CAAC;QAEhB,IAAI,OAAO,CAAC,MAAM,GAAG,GAAG;YAAE,KAAK,IAAI,GAAG,CAAC;QACvC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,KAAK,IAAI,GAAG,CAAC;QACxC,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;YAAE,KAAK,IAAI,GAAG,CAAC;QAEhF,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QAC/C,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc;YAAE,OAAO;QAExC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,YAAY,CAAC;YAChC,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAClD,4BAA4B;YAC5B,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAsB,CAAC,CAAC;YAEvE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;gBAE1C,2BAA2B;gBAC3B,MAAM,GAAG,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzC,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAAC,SAAiB;QAChD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACzC,SAAS;YACT,KAAK,EAAE,CAAC,kBAAU,CAAC,OAAO,CAAC;YAC3B,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAqB,EAAE,CAAC;YACxD,gBAAgB;YAChB,MAAM,QAAQ,GAAG,QAAQ;iBACtB,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;iBAC/E,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAqB,CAAC,CAAC;YAEjE,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACtC,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;CACF;AAnZD,sCAmZC"}
|
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Memory store interface
|
|
3
|
+
*/
|
|
4
|
+
import { Memory, MemoryQuery, MemorySearchOptions, SummarizeOptions, MemoryStats } from './types';
|
|
5
|
+
/**
|
|
6
|
+
* Interface for memory storage backends
|
|
7
|
+
*/
|
|
8
|
+
export interface MemoryStore {
|
|
9
|
+
/**
|
|
10
|
+
* Initialize the memory store
|
|
11
|
+
*/
|
|
12
|
+
initialize(): Promise<void>;
|
|
13
|
+
/**
|
|
14
|
+
* Save a memory
|
|
15
|
+
*/
|
|
16
|
+
save(memory: Memory): Promise<string>;
|
|
17
|
+
/**
|
|
18
|
+
* Save multiple memories
|
|
19
|
+
*/
|
|
20
|
+
saveBatch(memories: Memory[]): Promise<string[]>;
|
|
21
|
+
/**
|
|
22
|
+
* Retrieve memories by query
|
|
23
|
+
*/
|
|
24
|
+
retrieve(query: MemoryQuery): Promise<Memory[]>;
|
|
25
|
+
/**
|
|
26
|
+
* Search memories semantically
|
|
27
|
+
*/
|
|
28
|
+
search(query: string, options: MemorySearchOptions): Promise<Memory[]>;
|
|
29
|
+
/**
|
|
30
|
+
* Get a specific memory by ID
|
|
31
|
+
*/
|
|
32
|
+
get(id: string): Promise<Memory | null>;
|
|
33
|
+
/**
|
|
34
|
+
* Update a memory
|
|
35
|
+
*/
|
|
36
|
+
update(id: string, updates: Partial<Memory>): Promise<void>;
|
|
37
|
+
/**
|
|
38
|
+
* Delete a memory
|
|
39
|
+
*/
|
|
40
|
+
delete(id: string): Promise<void>;
|
|
41
|
+
/**
|
|
42
|
+
* Delete multiple memories
|
|
43
|
+
*/
|
|
44
|
+
deleteBatch(ids: string[]): Promise<void>;
|
|
45
|
+
/**
|
|
46
|
+
* Clear all memories for a session
|
|
47
|
+
*/
|
|
48
|
+
clearSession(sessionId: string): Promise<void>;
|
|
49
|
+
/**
|
|
50
|
+
* Clear all memories
|
|
51
|
+
*/
|
|
52
|
+
clear(): Promise<void>;
|
|
53
|
+
/**
|
|
54
|
+
* Summarize memories
|
|
55
|
+
*/
|
|
56
|
+
summarize(options: SummarizeOptions): Promise<string>;
|
|
57
|
+
/**
|
|
58
|
+
* Consolidate similar memories
|
|
59
|
+
*/
|
|
60
|
+
consolidate(memories: Memory[]): Promise<Memory>;
|
|
61
|
+
/**
|
|
62
|
+
* Get memory statistics
|
|
63
|
+
*/
|
|
64
|
+
getStats(sessionId?: string): Promise<MemoryStats>;
|
|
65
|
+
/**
|
|
66
|
+
* Prune old or low-importance memories
|
|
67
|
+
*/
|
|
68
|
+
prune(options?: {
|
|
69
|
+
olderThan?: Date;
|
|
70
|
+
minImportance?: number;
|
|
71
|
+
}): Promise<number>;
|
|
72
|
+
}
|
|
73
|
+
//# sourceMappingURL=memory-store.interface.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-store.interface.d.ts","sourceRoot":"","sources":["../../src/memory/memory-store.interface.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAElG;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B;;OAEG;IACH,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAE5B;;OAEG;IACH,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAEtC;;OAEG;IACH,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAEjD;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAEhD;;OAEG;IACH,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAEvE;;OAEG;IACH,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAExC;;OAEG;IACH,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE5D;;OAEG;IACH,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAElC;;OAEG;IACH,WAAW,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE1C;;OAEG;IACH,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE/C;;OAEG;IACH,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAEvB;;OAEG;IACH,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAEtD;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAEjD;;OAEG;IACH,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAEnD;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,IAAI,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAChF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-store.interface.js","sourceRoot":"","sources":["../../src/memory/memory-store.interface.ts"],"names":[],"mappings":";AAAA;;GAEG"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Buffer Memory Store - Simple FIFO buffer for recent memories
|
|
3
|
+
*/
|
|
4
|
+
import { MemoryStore } from '../memory-store.interface';
|
|
5
|
+
import { Memory, MemoryQuery, MemorySearchOptions, SummarizeOptions, MemoryStats } from '../types';
|
|
6
|
+
export interface BufferMemoryConfig {
|
|
7
|
+
maxSize?: number;
|
|
8
|
+
ttl?: number;
|
|
9
|
+
}
|
|
10
|
+
/**
|
|
11
|
+
* In-memory buffer that keeps the most recent memories
|
|
12
|
+
* Uses FIFO (First In, First Out) strategy
|
|
13
|
+
*/
|
|
14
|
+
export declare class BufferMemory implements MemoryStore {
|
|
15
|
+
private memories;
|
|
16
|
+
private sessionBuffers;
|
|
17
|
+
private maxSize;
|
|
18
|
+
private ttl?;
|
|
19
|
+
constructor(config?: BufferMemoryConfig);
|
|
20
|
+
initialize(): Promise<void>;
|
|
21
|
+
save(memory: Memory): Promise<string>;
|
|
22
|
+
saveBatch(memories: Memory[]): Promise<string[]>;
|
|
23
|
+
retrieve(query: MemoryQuery): Promise<Memory[]>;
|
|
24
|
+
search(query: string, options: MemorySearchOptions): Promise<Memory[]>;
|
|
25
|
+
get(id: string): Promise<Memory | null>;
|
|
26
|
+
update(id: string, updates: Partial<Memory>): Promise<void>;
|
|
27
|
+
delete(id: string): Promise<void>;
|
|
28
|
+
deleteBatch(ids: string[]): Promise<void>;
|
|
29
|
+
clearSession(sessionId: string): Promise<void>;
|
|
30
|
+
clear(): Promise<void>;
|
|
31
|
+
summarize(options: SummarizeOptions): Promise<string>;
|
|
32
|
+
consolidate(memories: Memory[]): Promise<Memory>;
|
|
33
|
+
getStats(sessionId?: string): Promise<MemoryStats>;
|
|
34
|
+
prune(options?: {
|
|
35
|
+
olderThan?: Date;
|
|
36
|
+
minImportance?: number;
|
|
37
|
+
}): Promise<number>;
|
|
38
|
+
/**
|
|
39
|
+
* Clean expired memories
|
|
40
|
+
*/
|
|
41
|
+
private cleanExpired;
|
|
42
|
+
/**
|
|
43
|
+
* Calculate simple text similarity score
|
|
44
|
+
*/
|
|
45
|
+
private calculateTextScore;
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=buffer-memory.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"buffer-memory.d.ts","sourceRoot":"","sources":["../../../src/memory/stores/buffer-memory.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EACL,MAAM,EACN,WAAW,EACX,mBAAmB,EACnB,gBAAgB,EAChB,WAAW,EAEZ,MAAM,UAAU,CAAC;AAGlB,MAAM,WAAW,kBAAkB;IACjC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED;;;GAGG;AACH,qBAAa,YAAa,YAAW,WAAW;IAC9C,OAAO,CAAC,QAAQ,CAAkC;IAClD,OAAO,CAAC,cAAc,CAAoC;IAC1D,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,GAAG,CAAC,CAAS;gBAET,MAAM,GAAE,kBAAuB;IAKrC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAkCrC,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAShD,QAAQ,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAmD/C,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAwBtE,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAKvC,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAkB3D,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAajC,WAAW,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAMzC,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAU9C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAKtB,SAAS,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAC;IAerD,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAqBhD,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAsClD,KAAK,CAAC,OAAO,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,IAAI,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IA8BpF;;OAEG;IACH,OAAO,CAAC,YAAY;IAiBpB;;OAEG;IACH,OAAO,CAAC,kBAAkB;CAa3B"}
|