@cogitator-ai/memory 0.1.0
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/LICENSE +21 -0
- package/README.md +89 -0
- package/dist/__tests__/context-builder.test.d.ts +2 -0
- package/dist/__tests__/context-builder.test.d.ts.map +1 -0
- package/dist/__tests__/context-builder.test.js +139 -0
- package/dist/__tests__/context-builder.test.js.map +1 -0
- package/dist/__tests__/memory-adapter.test.d.ts +2 -0
- package/dist/__tests__/memory-adapter.test.d.ts.map +1 -0
- package/dist/__tests__/memory-adapter.test.js +236 -0
- package/dist/__tests__/memory-adapter.test.js.map +1 -0
- package/dist/__tests__/token-counter.test.d.ts +2 -0
- package/dist/__tests__/token-counter.test.d.ts.map +1 -0
- package/dist/__tests__/token-counter.test.js +56 -0
- package/dist/__tests__/token-counter.test.js.map +1 -0
- package/dist/adapters/base.d.ts +22 -0
- package/dist/adapters/base.d.ts.map +1 -0
- package/dist/adapters/base.js +16 -0
- package/dist/adapters/base.js.map +1 -0
- package/dist/adapters/index.d.ts +9 -0
- package/dist/adapters/index.d.ts.map +1 -0
- package/dist/adapters/index.js +25 -0
- package/dist/adapters/index.js.map +1 -0
- package/dist/adapters/memory.d.ts +29 -0
- package/dist/adapters/memory.d.ts.map +1 -0
- package/dist/adapters/memory.js +130 -0
- package/dist/adapters/memory.js.map +1 -0
- package/dist/adapters/postgres.d.ts +43 -0
- package/dist/adapters/postgres.d.ts.map +1 -0
- package/dist/adapters/postgres.js +450 -0
- package/dist/adapters/postgres.js.map +1 -0
- package/dist/adapters/redis.d.ts +30 -0
- package/dist/adapters/redis.d.ts.map +1 -0
- package/dist/adapters/redis.js +186 -0
- package/dist/adapters/redis.js.map +1 -0
- package/dist/context-builder.d.ts +26 -0
- package/dist/context-builder.d.ts.map +1 -0
- package/dist/context-builder.js +154 -0
- package/dist/context-builder.js.map +1 -0
- package/dist/embedding/factory.d.ts +6 -0
- package/dist/embedding/factory.d.ts.map +1 -0
- package/dist/embedding/factory.js +18 -0
- package/dist/embedding/factory.js.map +1 -0
- package/dist/embedding/index.d.ts +7 -0
- package/dist/embedding/index.d.ts.map +1 -0
- package/dist/embedding/index.js +7 -0
- package/dist/embedding/index.js.map +1 -0
- package/dist/embedding/ollama.d.ts +19 -0
- package/dist/embedding/ollama.d.ts.map +1 -0
- package/dist/embedding/ollama.js +63 -0
- package/dist/embedding/ollama.js.map +1 -0
- package/dist/embedding/openai.d.ts +14 -0
- package/dist/embedding/openai.d.ts.map +1 -0
- package/dist/embedding/openai.js +56 -0
- package/dist/embedding/openai.js.map +1 -0
- package/dist/index.d.ts +14 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +13 -0
- package/dist/index.js.map +1 -0
- package/dist/schema.d.ts +169 -0
- package/dist/schema.d.ts.map +1 -0
- package/dist/schema.js +52 -0
- package/dist/schema.js.map +1 -0
- package/dist/token-counter.d.ts +24 -0
- package/dist/token-counter.d.ts.map +1 -0
- package/dist/token-counter.js +40 -0
- package/dist/token-counter.js.map +1 -0
- package/package.json +62 -0
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Base memory adapter - abstract class for all adapters
|
|
3
|
+
*/
|
|
4
|
+
import { nanoid } from 'nanoid';
|
|
5
|
+
export class BaseMemoryAdapter {
|
|
6
|
+
generateId(prefix) {
|
|
7
|
+
return `${prefix}_${nanoid(12)}`;
|
|
8
|
+
}
|
|
9
|
+
success(data) {
|
|
10
|
+
return { success: true, data };
|
|
11
|
+
}
|
|
12
|
+
failure(error) {
|
|
13
|
+
return { success: false, error };
|
|
14
|
+
}
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/adapters/base.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAUhC,MAAM,OAAgB,iBAAiB;IAmC3B,UAAU,CAAC,MAAc;QACjC,OAAO,GAAG,MAAM,IAAI,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;IACnC,CAAC;IAES,OAAO,CAAI,IAAO;QAC1B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACjC,CAAC;IAES,OAAO,CAAC,KAAa;QAC7B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IACnC,CAAC;CACF"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Memory adapter exports and factory
|
|
3
|
+
*/
|
|
4
|
+
import type { MemoryAdapter, InMemoryAdapterConfig, RedisAdapterConfig, PostgresAdapterConfig } from '@cogitator-ai/types';
|
|
5
|
+
export { BaseMemoryAdapter } from './base';
|
|
6
|
+
export { InMemoryAdapter } from './memory';
|
|
7
|
+
export type MemoryAdapterConfigUnion = InMemoryAdapterConfig | RedisAdapterConfig | PostgresAdapterConfig;
|
|
8
|
+
export declare function createMemoryAdapter(config: MemoryAdapterConfigUnion): Promise<MemoryAdapter>;
|
|
9
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/adapters/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EACV,aAAa,EACb,qBAAqB,EACrB,kBAAkB,EAClB,qBAAqB,EACtB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAE3C,MAAM,MAAM,wBAAwB,GAChC,qBAAqB,GACrB,kBAAkB,GAClB,qBAAqB,CAAC;AAE1B,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,wBAAwB,GAC/B,OAAO,CAAC,aAAa,CAAC,CAsBxB"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Memory adapter exports and factory
|
|
3
|
+
*/
|
|
4
|
+
import { InMemoryAdapter } from './memory';
|
|
5
|
+
export { BaseMemoryAdapter } from './base';
|
|
6
|
+
export { InMemoryAdapter } from './memory';
|
|
7
|
+
export async function createMemoryAdapter(config) {
|
|
8
|
+
switch (config.provider) {
|
|
9
|
+
case 'memory':
|
|
10
|
+
return new InMemoryAdapter(config);
|
|
11
|
+
case 'redis': {
|
|
12
|
+
const { RedisAdapter } = await import('./redis');
|
|
13
|
+
return new RedisAdapter(config);
|
|
14
|
+
}
|
|
15
|
+
case 'postgres': {
|
|
16
|
+
const { PostgresAdapter } = await import('./postgres');
|
|
17
|
+
return new PostgresAdapter(config);
|
|
18
|
+
}
|
|
19
|
+
default: {
|
|
20
|
+
const exhaustive = config;
|
|
21
|
+
throw new Error(`Unknown memory provider: ${exhaustive.provider}`);
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/adapters/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAQH,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAE3C,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAO3C,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,MAAgC;IAEhC,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,QAAQ;YACX,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;QAErC,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YACjD,OAAO,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QAED,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,CAAC;YACvD,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,CAAC,CAAC,CAAC;YACR,MAAM,UAAU,GAAU,MAAM,CAAC;YACjC,MAAM,IAAI,KAAK,CACb,4BAA6B,UAAuC,CAAC,QAAQ,EAAE,CAChF,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* In-memory adapter - default, zero dependencies
|
|
3
|
+
*/
|
|
4
|
+
import type { Thread, MemoryEntry, MemoryQueryOptions, MemoryResult, InMemoryAdapterConfig, MemoryProvider } from '@cogitator-ai/types';
|
|
5
|
+
import { BaseMemoryAdapter } from './base';
|
|
6
|
+
export declare class InMemoryAdapter extends BaseMemoryAdapter {
|
|
7
|
+
readonly provider: MemoryProvider;
|
|
8
|
+
private threads;
|
|
9
|
+
private entries;
|
|
10
|
+
private threadEntries;
|
|
11
|
+
private maxEntries;
|
|
12
|
+
constructor(config?: InMemoryAdapterConfig);
|
|
13
|
+
connect(): Promise<MemoryResult<void>>;
|
|
14
|
+
disconnect(): Promise<MemoryResult<void>>;
|
|
15
|
+
createThread(agentId: string, metadata?: Record<string, unknown>): Promise<MemoryResult<Thread>>;
|
|
16
|
+
getThread(threadId: string): Promise<MemoryResult<Thread | null>>;
|
|
17
|
+
updateThread(threadId: string, metadata: Record<string, unknown>): Promise<MemoryResult<Thread>>;
|
|
18
|
+
deleteThread(threadId: string): Promise<MemoryResult<void>>;
|
|
19
|
+
addEntry(entry: Omit<MemoryEntry, 'id' | 'createdAt'>): Promise<MemoryResult<MemoryEntry>>;
|
|
20
|
+
getEntries(options: MemoryQueryOptions): Promise<MemoryResult<MemoryEntry[]>>;
|
|
21
|
+
getEntry(entryId: string): Promise<MemoryResult<MemoryEntry | null>>;
|
|
22
|
+
deleteEntry(entryId: string): Promise<MemoryResult<void>>;
|
|
23
|
+
clearThread(threadId: string): Promise<MemoryResult<void>>;
|
|
24
|
+
get stats(): {
|
|
25
|
+
threads: number;
|
|
26
|
+
entries: number;
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=memory.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../src/adapters/memory.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EACV,MAAM,EACN,WAAW,EACX,kBAAkB,EAClB,YAAY,EACZ,qBAAqB,EACrB,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAE3C,qBAAa,eAAgB,SAAQ,iBAAiB;IACpD,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAY;IAE7C,OAAO,CAAC,OAAO,CAA6B;IAC5C,OAAO,CAAC,OAAO,CAAkC;IACjD,OAAO,CAAC,aAAa,CAA+B;IACpD,OAAO,CAAC,UAAU,CAAS;gBAEf,MAAM,GAAE,qBAA8C;IAK5D,OAAO,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAItC,UAAU,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAOzC,YAAY,CAChB,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,GACrC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAa1B,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAIjE,YAAY,CAChB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAChC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAU1B,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAU3D,QAAQ,CACZ,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,GAAG,WAAW,CAAC,GAC3C,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IAsB/B,UAAU,CACd,OAAO,EAAE,kBAAkB,GAC1B,OAAO,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC;IA6BjC,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC;IAIpE,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAazD,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAShE,IAAI,KAAK;;;MAKR;CACF"}
|
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* In-memory adapter - default, zero dependencies
|
|
3
|
+
*/
|
|
4
|
+
import { BaseMemoryAdapter } from './base';
|
|
5
|
+
export class InMemoryAdapter extends BaseMemoryAdapter {
|
|
6
|
+
provider = 'memory';
|
|
7
|
+
threads = new Map();
|
|
8
|
+
entries = new Map();
|
|
9
|
+
threadEntries = new Map();
|
|
10
|
+
maxEntries;
|
|
11
|
+
constructor(config = { provider: 'memory' }) {
|
|
12
|
+
super();
|
|
13
|
+
this.maxEntries = config.maxEntries ?? 10000;
|
|
14
|
+
}
|
|
15
|
+
async connect() {
|
|
16
|
+
return this.success(undefined);
|
|
17
|
+
}
|
|
18
|
+
async disconnect() {
|
|
19
|
+
this.threads.clear();
|
|
20
|
+
this.entries.clear();
|
|
21
|
+
this.threadEntries.clear();
|
|
22
|
+
return this.success(undefined);
|
|
23
|
+
}
|
|
24
|
+
async createThread(agentId, metadata = {}) {
|
|
25
|
+
const thread = {
|
|
26
|
+
id: this.generateId('thread'),
|
|
27
|
+
agentId,
|
|
28
|
+
metadata,
|
|
29
|
+
createdAt: new Date(),
|
|
30
|
+
updatedAt: new Date(),
|
|
31
|
+
};
|
|
32
|
+
this.threads.set(thread.id, thread);
|
|
33
|
+
this.threadEntries.set(thread.id, []);
|
|
34
|
+
return this.success(thread);
|
|
35
|
+
}
|
|
36
|
+
async getThread(threadId) {
|
|
37
|
+
return this.success(this.threads.get(threadId) ?? null);
|
|
38
|
+
}
|
|
39
|
+
async updateThread(threadId, metadata) {
|
|
40
|
+
const thread = this.threads.get(threadId);
|
|
41
|
+
if (!thread) {
|
|
42
|
+
return this.failure(`Thread not found: ${threadId}`);
|
|
43
|
+
}
|
|
44
|
+
thread.metadata = { ...thread.metadata, ...metadata };
|
|
45
|
+
thread.updatedAt = new Date();
|
|
46
|
+
return this.success(thread);
|
|
47
|
+
}
|
|
48
|
+
async deleteThread(threadId) {
|
|
49
|
+
const entryIds = this.threadEntries.get(threadId) ?? [];
|
|
50
|
+
for (const id of entryIds) {
|
|
51
|
+
this.entries.delete(id);
|
|
52
|
+
}
|
|
53
|
+
this.threads.delete(threadId);
|
|
54
|
+
this.threadEntries.delete(threadId);
|
|
55
|
+
return this.success(undefined);
|
|
56
|
+
}
|
|
57
|
+
async addEntry(entry) {
|
|
58
|
+
if (this.entries.size >= this.maxEntries) {
|
|
59
|
+
const oldestId = this.entries.keys().next().value;
|
|
60
|
+
if (oldestId) {
|
|
61
|
+
await this.deleteEntry(oldestId);
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
const full = {
|
|
65
|
+
...entry,
|
|
66
|
+
id: this.generateId('entry'),
|
|
67
|
+
createdAt: new Date(),
|
|
68
|
+
};
|
|
69
|
+
this.entries.set(full.id, full);
|
|
70
|
+
const threadList = this.threadEntries.get(entry.threadId) ?? [];
|
|
71
|
+
threadList.push(full.id);
|
|
72
|
+
this.threadEntries.set(entry.threadId, threadList);
|
|
73
|
+
return this.success(full);
|
|
74
|
+
}
|
|
75
|
+
async getEntries(options) {
|
|
76
|
+
const entryIds = this.threadEntries.get(options.threadId) ?? [];
|
|
77
|
+
let entries = entryIds
|
|
78
|
+
.map((id) => this.entries.get(id))
|
|
79
|
+
.filter((e) => e !== undefined);
|
|
80
|
+
if (options.before) {
|
|
81
|
+
entries = entries.filter((e) => e.createdAt < options.before);
|
|
82
|
+
}
|
|
83
|
+
if (options.after) {
|
|
84
|
+
entries = entries.filter((e) => e.createdAt > options.after);
|
|
85
|
+
}
|
|
86
|
+
if (!options.includeToolCalls) {
|
|
87
|
+
entries = entries.map((e) => ({
|
|
88
|
+
...e,
|
|
89
|
+
toolCalls: undefined,
|
|
90
|
+
toolResults: undefined,
|
|
91
|
+
}));
|
|
92
|
+
}
|
|
93
|
+
entries.sort((a, b) => a.createdAt.getTime() - b.createdAt.getTime());
|
|
94
|
+
if (options.limit && entries.length > options.limit) {
|
|
95
|
+
entries = entries.slice(-options.limit);
|
|
96
|
+
}
|
|
97
|
+
return this.success(entries);
|
|
98
|
+
}
|
|
99
|
+
async getEntry(entryId) {
|
|
100
|
+
return this.success(this.entries.get(entryId) ?? null);
|
|
101
|
+
}
|
|
102
|
+
async deleteEntry(entryId) {
|
|
103
|
+
const entry = this.entries.get(entryId);
|
|
104
|
+
if (entry) {
|
|
105
|
+
const threadList = this.threadEntries.get(entry.threadId);
|
|
106
|
+
if (threadList) {
|
|
107
|
+
const idx = threadList.indexOf(entryId);
|
|
108
|
+
if (idx !== -1)
|
|
109
|
+
threadList.splice(idx, 1);
|
|
110
|
+
}
|
|
111
|
+
this.entries.delete(entryId);
|
|
112
|
+
}
|
|
113
|
+
return this.success(undefined);
|
|
114
|
+
}
|
|
115
|
+
async clearThread(threadId) {
|
|
116
|
+
const entryIds = this.threadEntries.get(threadId) ?? [];
|
|
117
|
+
for (const id of entryIds) {
|
|
118
|
+
this.entries.delete(id);
|
|
119
|
+
}
|
|
120
|
+
this.threadEntries.set(threadId, []);
|
|
121
|
+
return this.success(undefined);
|
|
122
|
+
}
|
|
123
|
+
get stats() {
|
|
124
|
+
return {
|
|
125
|
+
threads: this.threads.size,
|
|
126
|
+
entries: this.entries.size,
|
|
127
|
+
};
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
//# sourceMappingURL=memory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../src/adapters/memory.ts"],"names":[],"mappings":"AAAA;;GAEG;AAUH,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAE3C,MAAM,OAAO,eAAgB,SAAQ,iBAAiB;IAC3C,QAAQ,GAAmB,QAAQ,CAAC;IAErC,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IACpC,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;IACzC,aAAa,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC5C,UAAU,CAAS;IAE3B,YAAY,SAAgC,EAAE,QAAQ,EAAE,QAAQ,EAAE;QAChE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO;QACX,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAe,EACf,WAAoC,EAAE;QAEtC,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YAC7B,OAAO;YACP,QAAQ;YACR,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACtC,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,QAAgB;QAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,QAAgB,EAChB,QAAiC;QAEjC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC,OAAO,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,MAAM,CAAC,QAAQ,GAAG,EAAE,GAAG,MAAM,CAAC,QAAQ,EAAE,GAAG,QAAQ,EAAE,CAAC;QACtD,MAAM,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,QAAgB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxD,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9B,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACpC,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,KAA4C;QAE5C,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YAClD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAgB;YACxB,GAAG,KAAK;YACR,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC5B,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAChC,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAChE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAEnD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAA2B;QAE3B,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAChE,IAAI,OAAO,GAAG,QAAQ;aACnB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aACjC,MAAM,CAAC,CAAC,CAAC,EAAoB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;QAEpD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,OAAO,CAAC,MAAO,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,OAAO,CAAC,KAAM,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAC9B,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5B,GAAG,CAAC;gBACJ,SAAS,EAAE,SAAS;gBACpB,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC;QAEtE,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;YACpD,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAe;QAC5B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAe;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC1D,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,GAAG,GAAG,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBACxC,IAAI,GAAG,KAAK,CAAC,CAAC;oBAAE,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAC5C,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxD,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,IAAI,KAAK;QACP,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YAC1B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;SAC3B,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Postgres adapter for long-term memory
|
|
3
|
+
*
|
|
4
|
+
* Supports:
|
|
5
|
+
* - Threads and entries (MemoryAdapter)
|
|
6
|
+
* - Facts (FactAdapter)
|
|
7
|
+
* - Embeddings with pgvector (EmbeddingAdapter)
|
|
8
|
+
*/
|
|
9
|
+
import type { Thread, MemoryEntry, MemoryQueryOptions, MemoryResult, PostgresAdapterConfig, MemoryProvider, Fact, Embedding, SemanticSearchOptions, FactAdapter, EmbeddingAdapter } from '@cogitator-ai/types';
|
|
10
|
+
import { BaseMemoryAdapter } from './base';
|
|
11
|
+
export declare class PostgresAdapter extends BaseMemoryAdapter implements FactAdapter, EmbeddingAdapter {
|
|
12
|
+
readonly provider: MemoryProvider;
|
|
13
|
+
private pool;
|
|
14
|
+
private config;
|
|
15
|
+
private schema;
|
|
16
|
+
private vectorDimensions;
|
|
17
|
+
constructor(config: PostgresAdapterConfig);
|
|
18
|
+
connect(): Promise<MemoryResult<void>>;
|
|
19
|
+
private initSchema;
|
|
20
|
+
disconnect(): Promise<MemoryResult<void>>;
|
|
21
|
+
createThread(agentId: string, metadata?: Record<string, unknown>): Promise<MemoryResult<Thread>>;
|
|
22
|
+
getThread(threadId: string): Promise<MemoryResult<Thread | null>>;
|
|
23
|
+
updateThread(threadId: string, metadata: Record<string, unknown>): Promise<MemoryResult<Thread>>;
|
|
24
|
+
deleteThread(threadId: string): Promise<MemoryResult<void>>;
|
|
25
|
+
addEntry(entry: Omit<MemoryEntry, 'id' | 'createdAt'>): Promise<MemoryResult<MemoryEntry>>;
|
|
26
|
+
getEntries(options: MemoryQueryOptions): Promise<MemoryResult<MemoryEntry[]>>;
|
|
27
|
+
getEntry(entryId: string): Promise<MemoryResult<MemoryEntry | null>>;
|
|
28
|
+
deleteEntry(entryId: string): Promise<MemoryResult<void>>;
|
|
29
|
+
clearThread(threadId: string): Promise<MemoryResult<void>>;
|
|
30
|
+
addFact(fact: Omit<Fact, 'id' | 'createdAt' | 'updatedAt'>): Promise<MemoryResult<Fact>>;
|
|
31
|
+
getFacts(agentId: string, category?: string): Promise<MemoryResult<Fact[]>>;
|
|
32
|
+
updateFact(factId: string, updates: Partial<Pick<Fact, 'content' | 'category' | 'confidence' | 'metadata' | 'expiresAt'>>): Promise<MemoryResult<Fact>>;
|
|
33
|
+
deleteFact(factId: string): Promise<MemoryResult<void>>;
|
|
34
|
+
searchFacts(agentId: string, query: string): Promise<MemoryResult<Fact[]>>;
|
|
35
|
+
addEmbedding(embedding: Omit<Embedding, 'id' | 'createdAt'>): Promise<MemoryResult<Embedding>>;
|
|
36
|
+
search(options: SemanticSearchOptions): Promise<MemoryResult<(Embedding & {
|
|
37
|
+
score: number;
|
|
38
|
+
})[]>>;
|
|
39
|
+
deleteEmbedding(embeddingId: string): Promise<MemoryResult<void>>;
|
|
40
|
+
deleteBySource(sourceId: string): Promise<MemoryResult<void>>;
|
|
41
|
+
setVectorDimensions(dimensions: number): void;
|
|
42
|
+
}
|
|
43
|
+
//# sourceMappingURL=postgres.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"postgres.d.ts","sourceRoot":"","sources":["../../src/adapters/postgres.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EACV,MAAM,EACN,WAAW,EACX,kBAAkB,EAClB,YAAY,EACZ,qBAAqB,EACrB,cAAc,EACd,IAAI,EACJ,SAAS,EACT,qBAAqB,EACrB,WAAW,EACX,gBAAgB,EACjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAQ3C,qBAAa,eACX,SAAQ,iBACR,YAAW,WAAW,EAAE,gBAAgB;IAExC,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAc;IAE/C,OAAO,CAAC,IAAI,CAAqB;IACjC,OAAO,CAAC,MAAM,CAAwB;IACtC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,gBAAgB,CAAO;gBAEnB,MAAM,EAAE,qBAAqB;IAMnC,OAAO,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAuB9B,UAAU;IA+ElB,UAAU,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAQzC,YAAY,CAChB,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,GACrC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAe1B,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAoBjE,YAAY,CAChB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAChC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAyB1B,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAQ3D,QAAQ,CACZ,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,GAAG,WAAW,CAAC,GAC3C,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IAyB/B,UAAU,CACd,OAAO,EAAE,kBAAkB,GAC1B,OAAO,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC;IAmDjC,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC;IAuBpE,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAQzD,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAS1D,OAAO,CACX,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GACjD,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IA0BxB,QAAQ,CACZ,OAAO,EAAE,MAAM,EACf,QAAQ,CAAC,EAAE,MAAM,GAChB,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;IAgC1B,UAAU,CACd,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,OAAO,CACd,IAAI,CAAC,IAAI,EAAE,SAAS,GAAG,UAAU,GAAG,YAAY,GAAG,UAAU,GAAG,WAAW,CAAC,CAC7E,GACA,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAsDxB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAQvD,WAAW,CACf,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;IA2B1B,YAAY,CAChB,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,CAAC,GAC7C,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IA0B7B,MAAM,CACV,OAAO,EAAE,qBAAqB,GAC7B,OAAO,CAAC,YAAY,CAAC,CAAC,SAAS,GAAG;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,CAAC,CAAC;IA6CrD,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IASjE,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IASnE,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;CAG9C"}
|