@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.
Files changed (67) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +89 -0
  3. package/dist/__tests__/context-builder.test.d.ts +2 -0
  4. package/dist/__tests__/context-builder.test.d.ts.map +1 -0
  5. package/dist/__tests__/context-builder.test.js +139 -0
  6. package/dist/__tests__/context-builder.test.js.map +1 -0
  7. package/dist/__tests__/memory-adapter.test.d.ts +2 -0
  8. package/dist/__tests__/memory-adapter.test.d.ts.map +1 -0
  9. package/dist/__tests__/memory-adapter.test.js +236 -0
  10. package/dist/__tests__/memory-adapter.test.js.map +1 -0
  11. package/dist/__tests__/token-counter.test.d.ts +2 -0
  12. package/dist/__tests__/token-counter.test.d.ts.map +1 -0
  13. package/dist/__tests__/token-counter.test.js +56 -0
  14. package/dist/__tests__/token-counter.test.js.map +1 -0
  15. package/dist/adapters/base.d.ts +22 -0
  16. package/dist/adapters/base.d.ts.map +1 -0
  17. package/dist/adapters/base.js +16 -0
  18. package/dist/adapters/base.js.map +1 -0
  19. package/dist/adapters/index.d.ts +9 -0
  20. package/dist/adapters/index.d.ts.map +1 -0
  21. package/dist/adapters/index.js +25 -0
  22. package/dist/adapters/index.js.map +1 -0
  23. package/dist/adapters/memory.d.ts +29 -0
  24. package/dist/adapters/memory.d.ts.map +1 -0
  25. package/dist/adapters/memory.js +130 -0
  26. package/dist/adapters/memory.js.map +1 -0
  27. package/dist/adapters/postgres.d.ts +43 -0
  28. package/dist/adapters/postgres.d.ts.map +1 -0
  29. package/dist/adapters/postgres.js +450 -0
  30. package/dist/adapters/postgres.js.map +1 -0
  31. package/dist/adapters/redis.d.ts +30 -0
  32. package/dist/adapters/redis.d.ts.map +1 -0
  33. package/dist/adapters/redis.js +186 -0
  34. package/dist/adapters/redis.js.map +1 -0
  35. package/dist/context-builder.d.ts +26 -0
  36. package/dist/context-builder.d.ts.map +1 -0
  37. package/dist/context-builder.js +154 -0
  38. package/dist/context-builder.js.map +1 -0
  39. package/dist/embedding/factory.d.ts +6 -0
  40. package/dist/embedding/factory.d.ts.map +1 -0
  41. package/dist/embedding/factory.js +18 -0
  42. package/dist/embedding/factory.js.map +1 -0
  43. package/dist/embedding/index.d.ts +7 -0
  44. package/dist/embedding/index.d.ts.map +1 -0
  45. package/dist/embedding/index.js +7 -0
  46. package/dist/embedding/index.js.map +1 -0
  47. package/dist/embedding/ollama.d.ts +19 -0
  48. package/dist/embedding/ollama.d.ts.map +1 -0
  49. package/dist/embedding/ollama.js +63 -0
  50. package/dist/embedding/ollama.js.map +1 -0
  51. package/dist/embedding/openai.d.ts +14 -0
  52. package/dist/embedding/openai.d.ts.map +1 -0
  53. package/dist/embedding/openai.js +56 -0
  54. package/dist/embedding/openai.js.map +1 -0
  55. package/dist/index.d.ts +14 -0
  56. package/dist/index.d.ts.map +1 -0
  57. package/dist/index.js +13 -0
  58. package/dist/index.js.map +1 -0
  59. package/dist/schema.d.ts +169 -0
  60. package/dist/schema.d.ts.map +1 -0
  61. package/dist/schema.js +52 -0
  62. package/dist/schema.js.map +1 -0
  63. package/dist/token-counter.d.ts +24 -0
  64. package/dist/token-counter.d.ts.map +1 -0
  65. package/dist/token-counter.js +40 -0
  66. package/dist/token-counter.js.map +1 -0
  67. 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"}