@cogitator-ai/memory 0.4.2 → 0.5.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/dist/adapters/index.d.ts +2 -2
- package/dist/adapters/index.d.ts.map +1 -1
- package/dist/adapters/index.js +8 -0
- package/dist/adapters/index.js.map +1 -1
- package/dist/adapters/mongodb.d.ts +25 -0
- package/dist/adapters/mongodb.d.ts.map +1 -0
- package/dist/adapters/mongodb.js +240 -0
- package/dist/adapters/mongodb.js.map +1 -0
- package/dist/adapters/qdrant.d.ts +20 -0
- package/dist/adapters/qdrant.d.ts.map +1 -0
- package/dist/adapters/qdrant.js +149 -0
- package/dist/adapters/qdrant.js.map +1 -0
- package/dist/adapters/sqlite.d.ts +21 -0
- package/dist/adapters/sqlite.d.ts.map +1 -0
- package/dist/adapters/sqlite.js +250 -0
- package/dist/adapters/sqlite.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +3 -0
- package/dist/index.js.map +1 -1
- package/package.json +3 -3
package/dist/adapters/index.d.ts
CHANGED
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
/**
|
|
2
2
|
* Memory adapter exports and factory
|
|
3
3
|
*/
|
|
4
|
-
import type { MemoryAdapter, InMemoryAdapterConfig, RedisAdapterConfig, PostgresAdapterConfig } from '@cogitator-ai/types';
|
|
4
|
+
import type { MemoryAdapter, InMemoryAdapterConfig, RedisAdapterConfig, PostgresAdapterConfig, SQLiteAdapterConfig, MongoDBAdapterConfig } from '@cogitator-ai/types';
|
|
5
5
|
export { BaseMemoryAdapter } from './base';
|
|
6
6
|
export { InMemoryAdapter } from './memory';
|
|
7
|
-
export type MemoryAdapterConfigUnion = InMemoryAdapterConfig | RedisAdapterConfig | PostgresAdapterConfig;
|
|
7
|
+
export type MemoryAdapterConfigUnion = InMemoryAdapterConfig | RedisAdapterConfig | PostgresAdapterConfig | SQLiteAdapterConfig | MongoDBAdapterConfig;
|
|
8
8
|
export declare function createMemoryAdapter(config: MemoryAdapterConfigUnion): Promise<MemoryAdapter>;
|
|
9
9
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +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,
|
|
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,EACrB,mBAAmB,EACnB,oBAAoB,EACrB,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,GACrB,mBAAmB,GACnB,oBAAoB,CAAC;AAEzB,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,wBAAwB,GAC/B,OAAO,CAAC,aAAa,CAAC,CAgCxB"}
|
package/dist/adapters/index.js
CHANGED
|
@@ -16,6 +16,14 @@ export async function createMemoryAdapter(config) {
|
|
|
16
16
|
const { PostgresAdapter } = await import('./postgres');
|
|
17
17
|
return new PostgresAdapter(config);
|
|
18
18
|
}
|
|
19
|
+
case 'sqlite': {
|
|
20
|
+
const { SQLiteAdapter } = await import('./sqlite');
|
|
21
|
+
return new SQLiteAdapter(config);
|
|
22
|
+
}
|
|
23
|
+
case 'mongodb': {
|
|
24
|
+
const { MongoDBAdapter } = await import('./mongodb');
|
|
25
|
+
return new MongoDBAdapter(config);
|
|
26
|
+
}
|
|
19
27
|
default: {
|
|
20
28
|
const exhaustive = config;
|
|
21
29
|
throw new Error(`Unknown memory provider: ${exhaustive.provider}`);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/adapters/index.ts"],"names":[],"mappings":"AAAA;;GAEG;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/adapters/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAUH,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAE3C,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAS3C,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,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACnD,OAAO,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;QAED,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;YACrD,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;QACpC,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,25 @@
|
|
|
1
|
+
import type { Thread, MemoryEntry, MemoryQueryOptions, MemoryResult, MongoDBAdapterConfig, MemoryProvider } from '@cogitator-ai/types';
|
|
2
|
+
import { BaseMemoryAdapter } from './base';
|
|
3
|
+
export declare class MongoDBAdapter extends BaseMemoryAdapter {
|
|
4
|
+
readonly provider: MemoryProvider;
|
|
5
|
+
private client;
|
|
6
|
+
private db;
|
|
7
|
+
private uri;
|
|
8
|
+
private database;
|
|
9
|
+
private prefix;
|
|
10
|
+
constructor(config: MongoDBAdapterConfig);
|
|
11
|
+
private get threads();
|
|
12
|
+
private get entries();
|
|
13
|
+
connect(): Promise<MemoryResult<void>>;
|
|
14
|
+
disconnect(): Promise<MemoryResult<void>>;
|
|
15
|
+
createThread(agentId: string, metadata?: Record<string, unknown>, threadId?: string): 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
|
+
}
|
|
25
|
+
//# sourceMappingURL=mongodb.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mongodb.d.ts","sourceRoot":"","sources":["../../src/adapters/mongodb.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,MAAM,EACN,WAAW,EACX,kBAAkB,EAClB,YAAY,EACZ,oBAAoB,EACpB,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAsD3C,qBAAa,cAAe,SAAQ,iBAAiB;IACnD,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAa;IAE9C,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,EAAE,CAAmB;IAC7B,OAAO,CAAC,GAAG,CAAS;IACpB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,EAAE,oBAAoB;IAOxC,OAAO,KAAK,OAAO,GAGlB;IAED,OAAO,KAAK,OAAO,GAGlB;IAEK,OAAO,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IA0BtC,UAAU,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IASzC,YAAY,CAChB,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,EACtC,QAAQ,CAAC,EAAE,MAAM,GAChB,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAyB1B,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAmBjE,YAAY,CAChB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAChC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAwB1B,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAY3D,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IA0B1F,UAAU,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC;IAuC7E,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC;IAsBpE,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAWzD,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;CAUjE"}
|
|
@@ -0,0 +1,240 @@
|
|
|
1
|
+
import { BaseMemoryAdapter } from './base';
|
|
2
|
+
export class MongoDBAdapter extends BaseMemoryAdapter {
|
|
3
|
+
provider = 'mongodb';
|
|
4
|
+
client = null;
|
|
5
|
+
db = null;
|
|
6
|
+
uri;
|
|
7
|
+
database;
|
|
8
|
+
prefix;
|
|
9
|
+
constructor(config) {
|
|
10
|
+
super();
|
|
11
|
+
this.uri = config.uri;
|
|
12
|
+
this.database = config.database ?? 'cogitator';
|
|
13
|
+
this.prefix = config.collectionPrefix ?? 'memory_';
|
|
14
|
+
}
|
|
15
|
+
get threads() {
|
|
16
|
+
if (!this.db)
|
|
17
|
+
throw new Error('Not connected');
|
|
18
|
+
return this.db.collection(`${this.prefix}threads`);
|
|
19
|
+
}
|
|
20
|
+
get entries() {
|
|
21
|
+
if (!this.db)
|
|
22
|
+
throw new Error('Not connected');
|
|
23
|
+
return this.db.collection(`${this.prefix}entries`);
|
|
24
|
+
}
|
|
25
|
+
async connect() {
|
|
26
|
+
if (this.client)
|
|
27
|
+
return this.success(undefined);
|
|
28
|
+
let MongoClient;
|
|
29
|
+
try {
|
|
30
|
+
const mongodb = await import('mongodb');
|
|
31
|
+
MongoClient = mongodb.MongoClient;
|
|
32
|
+
}
|
|
33
|
+
catch {
|
|
34
|
+
return this.failure('mongodb not installed. Run: pnpm add mongodb');
|
|
35
|
+
}
|
|
36
|
+
try {
|
|
37
|
+
this.client = new MongoClient(this.uri);
|
|
38
|
+
await this.client.connect();
|
|
39
|
+
this.db = this.client.db(this.database);
|
|
40
|
+
await this.threads.createIndex({ agentId: 1 });
|
|
41
|
+
await this.entries.createIndex({ threadId: 1 });
|
|
42
|
+
await this.entries.createIndex({ createdAt: 1 });
|
|
43
|
+
return this.success(undefined);
|
|
44
|
+
}
|
|
45
|
+
catch (err) {
|
|
46
|
+
return this.failure(err.message);
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
async disconnect() {
|
|
50
|
+
if (this.client) {
|
|
51
|
+
await this.client.close();
|
|
52
|
+
this.client = null;
|
|
53
|
+
this.db = null;
|
|
54
|
+
}
|
|
55
|
+
return this.success(undefined);
|
|
56
|
+
}
|
|
57
|
+
async createThread(agentId, metadata = {}, threadId) {
|
|
58
|
+
if (!this.db)
|
|
59
|
+
return this.failure('Not connected');
|
|
60
|
+
const thread = {
|
|
61
|
+
id: threadId ?? this.generateId('thread'),
|
|
62
|
+
agentId,
|
|
63
|
+
metadata,
|
|
64
|
+
createdAt: new Date(),
|
|
65
|
+
updatedAt: new Date(),
|
|
66
|
+
};
|
|
67
|
+
try {
|
|
68
|
+
await this.threads.insertOne({
|
|
69
|
+
_id: thread.id,
|
|
70
|
+
agentId: thread.agentId,
|
|
71
|
+
metadata: thread.metadata,
|
|
72
|
+
createdAt: thread.createdAt,
|
|
73
|
+
updatedAt: thread.updatedAt,
|
|
74
|
+
});
|
|
75
|
+
return this.success(thread);
|
|
76
|
+
}
|
|
77
|
+
catch (err) {
|
|
78
|
+
return this.failure(err.message);
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
async getThread(threadId) {
|
|
82
|
+
if (!this.db)
|
|
83
|
+
return this.failure('Not connected');
|
|
84
|
+
try {
|
|
85
|
+
const doc = await this.threads.findOne({ _id: threadId });
|
|
86
|
+
if (!doc)
|
|
87
|
+
return this.success(null);
|
|
88
|
+
return this.success({
|
|
89
|
+
id: doc._id,
|
|
90
|
+
agentId: doc.agentId,
|
|
91
|
+
metadata: doc.metadata,
|
|
92
|
+
createdAt: doc.createdAt,
|
|
93
|
+
updatedAt: doc.updatedAt,
|
|
94
|
+
});
|
|
95
|
+
}
|
|
96
|
+
catch (err) {
|
|
97
|
+
return this.failure(err.message);
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
async updateThread(threadId, metadata) {
|
|
101
|
+
if (!this.db)
|
|
102
|
+
return this.failure('Not connected');
|
|
103
|
+
const existing = await this.getThread(threadId);
|
|
104
|
+
if (!existing.success)
|
|
105
|
+
return existing;
|
|
106
|
+
if (!existing.data)
|
|
107
|
+
return this.failure(`Thread not found: ${threadId}`);
|
|
108
|
+
const updated = {
|
|
109
|
+
...existing.data,
|
|
110
|
+
metadata: { ...existing.data.metadata, ...metadata },
|
|
111
|
+
updatedAt: new Date(),
|
|
112
|
+
};
|
|
113
|
+
try {
|
|
114
|
+
await this.threads.updateOne({ _id: threadId }, { $set: { metadata: updated.metadata, updatedAt: updated.updatedAt } });
|
|
115
|
+
return this.success(updated);
|
|
116
|
+
}
|
|
117
|
+
catch (err) {
|
|
118
|
+
return this.failure(err.message);
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
async deleteThread(threadId) {
|
|
122
|
+
if (!this.db)
|
|
123
|
+
return this.failure('Not connected');
|
|
124
|
+
try {
|
|
125
|
+
await this.entries.deleteMany({ threadId });
|
|
126
|
+
await this.threads.deleteOne({ _id: threadId });
|
|
127
|
+
return this.success(undefined);
|
|
128
|
+
}
|
|
129
|
+
catch (err) {
|
|
130
|
+
return this.failure(err.message);
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
async addEntry(entry) {
|
|
134
|
+
if (!this.db)
|
|
135
|
+
return this.failure('Not connected');
|
|
136
|
+
const full = {
|
|
137
|
+
...entry,
|
|
138
|
+
id: this.generateId('entry'),
|
|
139
|
+
createdAt: new Date(),
|
|
140
|
+
};
|
|
141
|
+
try {
|
|
142
|
+
await this.entries.insertOne({
|
|
143
|
+
_id: full.id,
|
|
144
|
+
threadId: full.threadId,
|
|
145
|
+
message: full.message,
|
|
146
|
+
toolCalls: full.toolCalls,
|
|
147
|
+
toolResults: full.toolResults,
|
|
148
|
+
tokenCount: full.tokenCount,
|
|
149
|
+
createdAt: full.createdAt,
|
|
150
|
+
metadata: full.metadata,
|
|
151
|
+
});
|
|
152
|
+
return this.success(full);
|
|
153
|
+
}
|
|
154
|
+
catch (err) {
|
|
155
|
+
return this.failure(err.message);
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
async getEntries(options) {
|
|
159
|
+
if (!this.db)
|
|
160
|
+
return this.failure('Not connected');
|
|
161
|
+
try {
|
|
162
|
+
const filter = { threadId: options.threadId };
|
|
163
|
+
if (options.before || options.after) {
|
|
164
|
+
filter.createdAt = {};
|
|
165
|
+
if (options.before)
|
|
166
|
+
filter.createdAt.$lt = options.before;
|
|
167
|
+
if (options.after)
|
|
168
|
+
filter.createdAt.$gt = options.after;
|
|
169
|
+
}
|
|
170
|
+
let cursor = this.entries.find(filter).sort({ createdAt: -1 });
|
|
171
|
+
if (options.limit)
|
|
172
|
+
cursor = cursor.limit(options.limit);
|
|
173
|
+
const docs = await cursor.toArray();
|
|
174
|
+
docs.reverse();
|
|
175
|
+
const entries = docs.map((doc) => ({
|
|
176
|
+
id: doc._id,
|
|
177
|
+
threadId: doc.threadId,
|
|
178
|
+
message: doc.message,
|
|
179
|
+
toolCalls: options.includeToolCalls
|
|
180
|
+
? doc.toolCalls
|
|
181
|
+
: undefined,
|
|
182
|
+
toolResults: options.includeToolCalls
|
|
183
|
+
? doc.toolResults
|
|
184
|
+
: undefined,
|
|
185
|
+
tokenCount: doc.tokenCount,
|
|
186
|
+
createdAt: doc.createdAt,
|
|
187
|
+
metadata: doc.metadata,
|
|
188
|
+
}));
|
|
189
|
+
return this.success(entries);
|
|
190
|
+
}
|
|
191
|
+
catch (err) {
|
|
192
|
+
return this.failure(err.message);
|
|
193
|
+
}
|
|
194
|
+
}
|
|
195
|
+
async getEntry(entryId) {
|
|
196
|
+
if (!this.db)
|
|
197
|
+
return this.failure('Not connected');
|
|
198
|
+
try {
|
|
199
|
+
const doc = await this.entries.findOne({ _id: entryId });
|
|
200
|
+
if (!doc)
|
|
201
|
+
return this.success(null);
|
|
202
|
+
return this.success({
|
|
203
|
+
id: doc._id,
|
|
204
|
+
threadId: doc.threadId,
|
|
205
|
+
message: doc.message,
|
|
206
|
+
toolCalls: doc.toolCalls,
|
|
207
|
+
toolResults: doc.toolResults,
|
|
208
|
+
tokenCount: doc.tokenCount,
|
|
209
|
+
createdAt: doc.createdAt,
|
|
210
|
+
metadata: doc.metadata,
|
|
211
|
+
});
|
|
212
|
+
}
|
|
213
|
+
catch (err) {
|
|
214
|
+
return this.failure(err.message);
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
async deleteEntry(entryId) {
|
|
218
|
+
if (!this.db)
|
|
219
|
+
return this.failure('Not connected');
|
|
220
|
+
try {
|
|
221
|
+
await this.entries.deleteOne({ _id: entryId });
|
|
222
|
+
return this.success(undefined);
|
|
223
|
+
}
|
|
224
|
+
catch (err) {
|
|
225
|
+
return this.failure(err.message);
|
|
226
|
+
}
|
|
227
|
+
}
|
|
228
|
+
async clearThread(threadId) {
|
|
229
|
+
if (!this.db)
|
|
230
|
+
return this.failure('Not connected');
|
|
231
|
+
try {
|
|
232
|
+
await this.entries.deleteMany({ threadId });
|
|
233
|
+
return this.success(undefined);
|
|
234
|
+
}
|
|
235
|
+
catch (err) {
|
|
236
|
+
return this.failure(err.message);
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
}
|
|
240
|
+
//# sourceMappingURL=mongodb.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mongodb.js","sourceRoot":"","sources":["../../src/adapters/mongodb.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAsD3C,MAAM,OAAO,cAAe,SAAQ,iBAAiB;IAC1C,QAAQ,GAAmB,SAAS,CAAC;IAEtC,MAAM,GAAuB,IAAI,CAAC;IAClC,EAAE,GAAc,IAAI,CAAC;IACrB,GAAG,CAAS;IACZ,QAAQ,CAAS;IACjB,MAAM,CAAS;IAEvB,YAAY,MAA4B;QACtC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,WAAW,CAAC;QAC/C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,gBAAgB,IAAI,SAAS,CAAC;IACrD,CAAC;IAED,IAAY,OAAO;QACjB,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,EAAE,CAAC,UAAU,CAAY,GAAG,IAAI,CAAC,MAAM,SAAS,CAAC,CAAC;IAChE,CAAC;IAED,IAAY,OAAO;QACjB,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QAC/C,OAAO,IAAI,CAAC,EAAE,CAAC,UAAU,CAAW,GAAG,IAAI,CAAC,MAAM,SAAS,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEhD,IAAI,WAA6C,CAAC;QAClD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YACxC,WAAW,GAAG,OAAO,CAAC,WAA0D,CAAC;QACnF,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC,OAAO,CAAC,8CAA8C,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACxC,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC5B,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAExC,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAChD,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;YAEjD,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;YACnB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAe,EACf,WAAoC,EAAE,EACtC,QAAiB;QAEjB,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,QAAQ,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YACzC,OAAO;YACP,QAAQ;YACR,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC;gBAC3B,GAAG,EAAE,MAAM,CAAC,EAAE;gBACd,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,QAAgB;QAC9B,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC1D,IAAI,CAAC,GAAG;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAEpC,OAAO,IAAI,CAAC,OAAO,CAAC;gBAClB,EAAE,EAAE,GAAG,CAAC,GAAG;gBACX,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,SAAS,EAAE,GAAG,CAAC,SAAS;aACzB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,QAAgB,EAChB,QAAiC;QAEjC,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,OAAO,QAAQ,CAAC;QACvC,IAAI,CAAC,QAAQ,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAW;YACtB,GAAG,QAAQ,CAAC,IAAI;YAChB,QAAQ,EAAE,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,QAAQ,EAAE;YACpD,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAC1B,EAAE,GAAG,EAAE,QAAQ,EAAE,EACjB,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CACvE,CAAC;YACF,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,QAAgB;QACjC,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC5C,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAA4C;QACzD,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,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;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC;gBAC3B,GAAG,EAAE,IAAI,CAAC,EAAE;gBACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAA2B;QAC1C,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,MAAM,GAA4B,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC;YAEvE,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBACpC,MAAM,CAAC,SAAS,GAAG,EAAE,CAAC;gBACtB,IAAI,OAAO,CAAC,MAAM;oBAAG,MAAM,CAAC,SAAkC,CAAC,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gBACpF,IAAI,OAAO,CAAC,KAAK;oBAAG,MAAM,CAAC,SAAkC,CAAC,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC;YACpF,CAAC;YAED,IAAI,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAC/D,IAAI,OAAO,CAAC,KAAK;gBAAE,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAExD,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,IAAI,CAAC,OAAO,EAAE,CAAC;YAEf,MAAM,OAAO,GAAkB,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAChD,EAAE,EAAE,GAAG,CAAC,GAAG;gBACX,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,OAAO,EAAE,GAAG,CAAC,OAAiC;gBAC9C,SAAS,EAAE,OAAO,CAAC,gBAAgB;oBACjC,CAAC,CAAE,GAAG,CAAC,SAAsC;oBAC7C,CAAC,CAAC,SAAS;gBACb,WAAW,EAAE,OAAO,CAAC,gBAAgB;oBACnC,CAAC,CAAE,GAAG,CAAC,WAA0C;oBACjD,CAAC,CAAC,SAAS;gBACb,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;aACvB,CAAC,CAAC,CAAC;YAEJ,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAe;QAC5B,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,GAAG;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAEpC,OAAO,IAAI,CAAC,OAAO,CAAC;gBAClB,EAAE,EAAE,GAAG,CAAC,GAAG;gBACX,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,OAAO,EAAE,GAAG,CAAC,OAAiC;gBAC9C,SAAS,EAAE,GAAG,CAAC,SAAqC;gBACpD,WAAW,EAAE,GAAG,CAAC,WAAyC;gBAC1D,UAAU,EAAE,GAAG,CAAC,UAAU;gBAC1B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;aACvB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAe;QAC/B,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;YAC/C,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB;QAChC,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC5C,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { Embedding, SemanticSearchOptions, MemoryResult, QdrantAdapterConfig, EmbeddingAdapter } from '@cogitator-ai/types';
|
|
2
|
+
export declare class QdrantAdapter implements EmbeddingAdapter {
|
|
3
|
+
private client;
|
|
4
|
+
private url;
|
|
5
|
+
private apiKey?;
|
|
6
|
+
private collection;
|
|
7
|
+
private dimensions;
|
|
8
|
+
constructor(config: QdrantAdapterConfig);
|
|
9
|
+
connect(): Promise<MemoryResult<void>>;
|
|
10
|
+
disconnect(): Promise<MemoryResult<void>>;
|
|
11
|
+
addEmbedding(embedding: Omit<Embedding, 'id' | 'createdAt'>): Promise<MemoryResult<Embedding>>;
|
|
12
|
+
search(options: SemanticSearchOptions): Promise<MemoryResult<(Embedding & {
|
|
13
|
+
score: number;
|
|
14
|
+
})[]>>;
|
|
15
|
+
deleteEmbedding(embeddingId: string): Promise<MemoryResult<void>>;
|
|
16
|
+
deleteBySource(sourceId: string): Promise<MemoryResult<void>>;
|
|
17
|
+
private success;
|
|
18
|
+
private failure;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=qdrant.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"qdrant.d.ts","sourceRoot":"","sources":["../../src/adapters/qdrant.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,SAAS,EACT,qBAAqB,EACrB,YAAY,EACZ,mBAAmB,EACnB,gBAAgB,EACjB,MAAM,qBAAqB,CAAC;AAqC7B,qBAAa,aAAc,YAAW,gBAAgB;IACpD,OAAO,CAAC,MAAM,CAA6B;IAC3C,OAAO,CAAC,GAAG,CAAS;IACpB,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,UAAU,CAAS;gBAEf,MAAM,EAAE,mBAAmB;IAOjC,OAAO,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAkCtC,UAAU,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAKzC,YAAY,CAChB,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,CAAC,GAC7C,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IA+B7B,MAAM,CACV,OAAO,EAAE,qBAAqB,GAC7B,OAAO,CAAC,YAAY,CAAC,CAAC,SAAS,GAAG;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,CAAC,CAAC;IAkDrD,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAajE,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAanE,OAAO,CAAC,OAAO;IAIf,OAAO,CAAC,OAAO;CAGhB"}
|
|
@@ -0,0 +1,149 @@
|
|
|
1
|
+
import { nanoid } from 'nanoid';
|
|
2
|
+
export class QdrantAdapter {
|
|
3
|
+
client = null;
|
|
4
|
+
url;
|
|
5
|
+
apiKey;
|
|
6
|
+
collection;
|
|
7
|
+
dimensions;
|
|
8
|
+
constructor(config) {
|
|
9
|
+
this.url = config.url ?? 'http://localhost:6333';
|
|
10
|
+
this.apiKey = config.apiKey;
|
|
11
|
+
this.collection = config.collection ?? 'cogitator';
|
|
12
|
+
this.dimensions = config.dimensions;
|
|
13
|
+
}
|
|
14
|
+
async connect() {
|
|
15
|
+
if (this.client)
|
|
16
|
+
return this.success(undefined);
|
|
17
|
+
let QdrantClient;
|
|
18
|
+
try {
|
|
19
|
+
const qdrant = await import('@qdrant/js-client-rest');
|
|
20
|
+
QdrantClient = qdrant.QdrantClient;
|
|
21
|
+
}
|
|
22
|
+
catch {
|
|
23
|
+
return this.failure('@qdrant/js-client-rest not installed. Run: pnpm add @qdrant/js-client-rest');
|
|
24
|
+
}
|
|
25
|
+
try {
|
|
26
|
+
this.client = new QdrantClient({ url: this.url, apiKey: this.apiKey });
|
|
27
|
+
const collections = await this.client.getCollections();
|
|
28
|
+
const exists = collections.collections.some((c) => c.name === this.collection);
|
|
29
|
+
if (!exists) {
|
|
30
|
+
await this.client.createCollection(this.collection, {
|
|
31
|
+
vectors: { size: this.dimensions, distance: 'Cosine' },
|
|
32
|
+
});
|
|
33
|
+
}
|
|
34
|
+
return this.success(undefined);
|
|
35
|
+
}
|
|
36
|
+
catch (err) {
|
|
37
|
+
return this.failure(err.message);
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
async disconnect() {
|
|
41
|
+
this.client = null;
|
|
42
|
+
return this.success(undefined);
|
|
43
|
+
}
|
|
44
|
+
async addEmbedding(embedding) {
|
|
45
|
+
if (!this.client)
|
|
46
|
+
return this.failure('Not connected');
|
|
47
|
+
const full = {
|
|
48
|
+
...embedding,
|
|
49
|
+
id: `emb_${nanoid(12)}`,
|
|
50
|
+
createdAt: new Date(),
|
|
51
|
+
};
|
|
52
|
+
try {
|
|
53
|
+
await this.client.upsert(this.collection, {
|
|
54
|
+
points: [
|
|
55
|
+
{
|
|
56
|
+
id: full.id,
|
|
57
|
+
vector: full.vector,
|
|
58
|
+
payload: {
|
|
59
|
+
sourceId: full.sourceId,
|
|
60
|
+
sourceType: full.sourceType,
|
|
61
|
+
content: full.content,
|
|
62
|
+
createdAt: full.createdAt.toISOString(),
|
|
63
|
+
...full.metadata,
|
|
64
|
+
},
|
|
65
|
+
},
|
|
66
|
+
],
|
|
67
|
+
});
|
|
68
|
+
return this.success(full);
|
|
69
|
+
}
|
|
70
|
+
catch (err) {
|
|
71
|
+
return this.failure(err.message);
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
async search(options) {
|
|
75
|
+
if (!this.client)
|
|
76
|
+
return this.failure('Not connected');
|
|
77
|
+
if (!options.vector) {
|
|
78
|
+
return this.failure('vector is required for Qdrant search');
|
|
79
|
+
}
|
|
80
|
+
try {
|
|
81
|
+
const filter = {};
|
|
82
|
+
if (options.filter) {
|
|
83
|
+
filter.must = [];
|
|
84
|
+
if (options.filter.sourceType) {
|
|
85
|
+
filter.must.push({ key: 'sourceType', match: { value: options.filter.sourceType } });
|
|
86
|
+
}
|
|
87
|
+
if (options.filter.threadId) {
|
|
88
|
+
filter.must.push({ key: 'threadId', match: { value: options.filter.threadId } });
|
|
89
|
+
}
|
|
90
|
+
if (options.filter.agentId) {
|
|
91
|
+
filter.must.push({ key: 'agentId', match: { value: options.filter.agentId } });
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
const results = await this.client.search(this.collection, {
|
|
95
|
+
vector: options.vector,
|
|
96
|
+
limit: options.limit ?? 10,
|
|
97
|
+
score_threshold: options.threshold,
|
|
98
|
+
filter: filter.must?.length ? filter : undefined,
|
|
99
|
+
});
|
|
100
|
+
const embeddings = results.map((r) => ({
|
|
101
|
+
id: r.id,
|
|
102
|
+
sourceId: r.payload.sourceId,
|
|
103
|
+
sourceType: r.payload.sourceType,
|
|
104
|
+
vector: [],
|
|
105
|
+
content: r.payload.content,
|
|
106
|
+
createdAt: new Date(r.payload.createdAt),
|
|
107
|
+
metadata: Object.fromEntries(Object.entries(r.payload).filter(([k]) => !['sourceId', 'sourceType', 'content', 'createdAt'].includes(k))),
|
|
108
|
+
score: r.score,
|
|
109
|
+
}));
|
|
110
|
+
return this.success(embeddings);
|
|
111
|
+
}
|
|
112
|
+
catch (err) {
|
|
113
|
+
return this.failure(err.message);
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
async deleteEmbedding(embeddingId) {
|
|
117
|
+
if (!this.client)
|
|
118
|
+
return this.failure('Not connected');
|
|
119
|
+
try {
|
|
120
|
+
await this.client.delete(this.collection, {
|
|
121
|
+
filter: { must: [{ key: 'id', match: { value: embeddingId } }] },
|
|
122
|
+
});
|
|
123
|
+
return this.success(undefined);
|
|
124
|
+
}
|
|
125
|
+
catch (err) {
|
|
126
|
+
return this.failure(err.message);
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
async deleteBySource(sourceId) {
|
|
130
|
+
if (!this.client)
|
|
131
|
+
return this.failure('Not connected');
|
|
132
|
+
try {
|
|
133
|
+
await this.client.delete(this.collection, {
|
|
134
|
+
filter: { must: [{ key: 'sourceId', match: { value: sourceId } }] },
|
|
135
|
+
});
|
|
136
|
+
return this.success(undefined);
|
|
137
|
+
}
|
|
138
|
+
catch (err) {
|
|
139
|
+
return this.failure(err.message);
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
success(data) {
|
|
143
|
+
return { success: true, data };
|
|
144
|
+
}
|
|
145
|
+
failure(error) {
|
|
146
|
+
return { success: false, error };
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
//# sourceMappingURL=qdrant.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"qdrant.js","sourceRoot":"","sources":["../../src/adapters/qdrant.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAoChC,MAAM,OAAO,aAAa;IAChB,MAAM,GAAwB,IAAI,CAAC;IACnC,GAAG,CAAS;IACZ,MAAM,CAAU;IAChB,UAAU,CAAS;IACnB,UAAU,CAAS;IAE3B,YAAY,MAA2B;QACrC,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,uBAAuB,CAAC;QACjD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,WAAW,CAAC;QACnD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEhD,IAAI,YAA6E,CAAC;QAClF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC;YACtD,YAAY,GAAG,MAAM,CAAC,YAGJ,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC,OAAO,CACjB,4EAA4E,CAC7E,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YAEvE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YACvD,MAAM,MAAM,GAAG,WAAW,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,UAAU,CAAC,CAAC;YAE/E,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,UAAU,EAAE;oBAClD,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE;iBACvD,CAAC,CAAC;YACL,CAAC;YAED,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,SAA8C;QAE9C,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEvD,MAAM,IAAI,GAAc;YACtB,GAAG,SAAS;YACZ,EAAE,EAAE,OAAO,MAAM,CAAC,EAAE,CAAC,EAAE;YACvB,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE;gBACxC,MAAM,EAAE;oBACN;wBACE,EAAE,EAAE,IAAI,CAAC,EAAE;wBACX,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,OAAO,EAAE;4BACP,QAAQ,EAAE,IAAI,CAAC,QAAQ;4BACvB,UAAU,EAAE,IAAI,CAAC,UAAU;4BAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;4BACrB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE;4BACvC,GAAG,IAAI,CAAC,QAAQ;yBACjB;qBACF;iBACF;aACF,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,OAA8B;QAE9B,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEvD,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC,OAAO,CAAC,sCAAsC,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAiE,EAAE,CAAC;YAChF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;gBACjB,IAAI,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;oBAC9B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBACvF,CAAC;gBACD,IAAI,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;oBAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBACnF,CAAC;gBACD,IAAI,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAC3B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;gBACjF,CAAC;YACH,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE;gBACxD,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE;gBAC1B,eAAe,EAAE,OAAO,CAAC,SAAS;gBAClC,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;aACjD,CAAC,CAAC;YAEH,MAAM,UAAU,GAAsC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACxE,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,QAAQ,EAAE,CAAC,CAAC,OAAO,CAAC,QAAkB;gBACtC,UAAU,EAAE,CAAC,CAAC,OAAO,CAAC,UAAqC;gBAC3D,MAAM,EAAE,EAAE;gBACV,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,OAAiB;gBACpC,SAAS,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,SAAmB,CAAC;gBAClD,QAAQ,EAAE,MAAM,CAAC,WAAW,CAC1B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAC9B,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,EAAE,YAAY,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CACzE,CACF;gBACD,KAAK,EAAE,CAAC,CAAC,KAAK;aACf,CAAC,CAAC,CAAC;YAEJ,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAClC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,WAAmB;QACvC,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEvD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE;gBACxC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,CAAC,EAAE;aACjE,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAgB;QACnC,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEvD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE;gBACxC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE;aACpE,CAAC,CAAC;YACH,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAEO,OAAO,CAAI,IAAO;QACxB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACjC,CAAC;IAEO,OAAO,CAAC,KAAa;QAC3B,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IACnC,CAAC;CACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import type { Thread, MemoryEntry, MemoryQueryOptions, MemoryResult, SQLiteAdapterConfig, MemoryProvider } from '@cogitator-ai/types';
|
|
2
|
+
import { BaseMemoryAdapter } from './base';
|
|
3
|
+
export declare class SQLiteAdapter extends BaseMemoryAdapter {
|
|
4
|
+
readonly provider: MemoryProvider;
|
|
5
|
+
private db;
|
|
6
|
+
private path;
|
|
7
|
+
private walMode;
|
|
8
|
+
constructor(config: SQLiteAdapterConfig);
|
|
9
|
+
connect(): Promise<MemoryResult<void>>;
|
|
10
|
+
disconnect(): Promise<MemoryResult<void>>;
|
|
11
|
+
createThread(agentId: string, metadata?: Record<string, unknown>, threadId?: string): Promise<MemoryResult<Thread>>;
|
|
12
|
+
getThread(threadId: string): Promise<MemoryResult<Thread | null>>;
|
|
13
|
+
updateThread(threadId: string, metadata: Record<string, unknown>): Promise<MemoryResult<Thread>>;
|
|
14
|
+
deleteThread(threadId: string): Promise<MemoryResult<void>>;
|
|
15
|
+
addEntry(entry: Omit<MemoryEntry, 'id' | 'createdAt'>): Promise<MemoryResult<MemoryEntry>>;
|
|
16
|
+
getEntries(options: MemoryQueryOptions): Promise<MemoryResult<MemoryEntry[]>>;
|
|
17
|
+
getEntry(entryId: string): Promise<MemoryResult<MemoryEntry | null>>;
|
|
18
|
+
deleteEntry(entryId: string): Promise<MemoryResult<void>>;
|
|
19
|
+
clearThread(threadId: string): Promise<MemoryResult<void>>;
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=sqlite.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.d.ts","sourceRoot":"","sources":["../../src/adapters/sqlite.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,MAAM,EACN,WAAW,EACX,kBAAkB,EAClB,YAAY,EACZ,mBAAmB,EACnB,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAe3C,qBAAa,aAAc,SAAQ,iBAAiB;IAClD,QAAQ,CAAC,QAAQ,EAAE,cAAc,CAAY;IAE7C,OAAO,CAAC,EAAE,CAAyB;IACnC,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,OAAO,CAAU;gBAEb,MAAM,EAAE,mBAAmB;IAMjC,OAAO,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAoDtC,UAAU,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAQzC,YAAY,CAChB,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,EACtC,QAAQ,CAAC,EAAE,MAAM,GAChB,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IA6B1B,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IA6BjE,YAAY,CAChB,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAChC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IAwB1B,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAY3D,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;IA8B1F,UAAU,CAAC,OAAO,EAAE,kBAAkB,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC;IAuD7E,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC;IAmCpE,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAWzD,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;CAUjE"}
|
|
@@ -0,0 +1,250 @@
|
|
|
1
|
+
import { BaseMemoryAdapter } from './base';
|
|
2
|
+
export class SQLiteAdapter extends BaseMemoryAdapter {
|
|
3
|
+
provider = 'sqlite';
|
|
4
|
+
db = null;
|
|
5
|
+
path;
|
|
6
|
+
walMode;
|
|
7
|
+
constructor(config) {
|
|
8
|
+
super();
|
|
9
|
+
this.path = config.path;
|
|
10
|
+
this.walMode = config.walMode ?? true;
|
|
11
|
+
}
|
|
12
|
+
async connect() {
|
|
13
|
+
if (this.db) {
|
|
14
|
+
return this.success(undefined);
|
|
15
|
+
}
|
|
16
|
+
let Database;
|
|
17
|
+
try {
|
|
18
|
+
const betterSqlite = await import('better-sqlite3');
|
|
19
|
+
Database = betterSqlite.default;
|
|
20
|
+
}
|
|
21
|
+
catch {
|
|
22
|
+
return this.failure('better-sqlite3 not installed. Run: pnpm add better-sqlite3');
|
|
23
|
+
}
|
|
24
|
+
try {
|
|
25
|
+
this.db = new Database(this.path);
|
|
26
|
+
if (this.walMode && this.path !== ':memory:') {
|
|
27
|
+
this.db.pragma('journal_mode = WAL');
|
|
28
|
+
}
|
|
29
|
+
this.db.exec(`
|
|
30
|
+
CREATE TABLE IF NOT EXISTS threads (
|
|
31
|
+
id TEXT PRIMARY KEY,
|
|
32
|
+
agent_id TEXT NOT NULL,
|
|
33
|
+
metadata TEXT NOT NULL DEFAULT '{}',
|
|
34
|
+
created_at TEXT NOT NULL,
|
|
35
|
+
updated_at TEXT NOT NULL
|
|
36
|
+
);
|
|
37
|
+
|
|
38
|
+
CREATE TABLE IF NOT EXISTS entries (
|
|
39
|
+
id TEXT PRIMARY KEY,
|
|
40
|
+
thread_id TEXT NOT NULL,
|
|
41
|
+
message TEXT NOT NULL,
|
|
42
|
+
tool_calls TEXT,
|
|
43
|
+
tool_results TEXT,
|
|
44
|
+
token_count INTEGER NOT NULL DEFAULT 0,
|
|
45
|
+
created_at TEXT NOT NULL,
|
|
46
|
+
metadata TEXT,
|
|
47
|
+
FOREIGN KEY (thread_id) REFERENCES threads(id) ON DELETE CASCADE
|
|
48
|
+
);
|
|
49
|
+
|
|
50
|
+
CREATE INDEX IF NOT EXISTS idx_entries_thread ON entries(thread_id);
|
|
51
|
+
CREATE INDEX IF NOT EXISTS idx_entries_created ON entries(created_at);
|
|
52
|
+
CREATE INDEX IF NOT EXISTS idx_threads_agent ON threads(agent_id);
|
|
53
|
+
`);
|
|
54
|
+
return this.success(undefined);
|
|
55
|
+
}
|
|
56
|
+
catch (err) {
|
|
57
|
+
return this.failure(err.message);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
async disconnect() {
|
|
61
|
+
if (this.db) {
|
|
62
|
+
this.db.close();
|
|
63
|
+
this.db = null;
|
|
64
|
+
}
|
|
65
|
+
return this.success(undefined);
|
|
66
|
+
}
|
|
67
|
+
async createThread(agentId, metadata = {}, threadId) {
|
|
68
|
+
if (!this.db)
|
|
69
|
+
return this.failure('Not connected');
|
|
70
|
+
const thread = {
|
|
71
|
+
id: threadId ?? this.generateId('thread'),
|
|
72
|
+
agentId,
|
|
73
|
+
metadata,
|
|
74
|
+
createdAt: new Date(),
|
|
75
|
+
updatedAt: new Date(),
|
|
76
|
+
};
|
|
77
|
+
try {
|
|
78
|
+
const stmt = this.db.prepare(`
|
|
79
|
+
INSERT INTO threads (id, agent_id, metadata, created_at, updated_at)
|
|
80
|
+
VALUES (?, ?, ?, ?, ?)
|
|
81
|
+
`);
|
|
82
|
+
stmt.run(thread.id, thread.agentId, JSON.stringify(thread.metadata), thread.createdAt.toISOString(), thread.updatedAt.toISOString());
|
|
83
|
+
return this.success(thread);
|
|
84
|
+
}
|
|
85
|
+
catch (err) {
|
|
86
|
+
return this.failure(err.message);
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
async getThread(threadId) {
|
|
90
|
+
if (!this.db)
|
|
91
|
+
return this.failure('Not connected');
|
|
92
|
+
try {
|
|
93
|
+
const stmt = this.db.prepare('SELECT * FROM threads WHERE id = ?');
|
|
94
|
+
const row = stmt.get(threadId);
|
|
95
|
+
if (!row)
|
|
96
|
+
return this.success(null);
|
|
97
|
+
return this.success({
|
|
98
|
+
id: row.id,
|
|
99
|
+
agentId: row.agent_id,
|
|
100
|
+
metadata: JSON.parse(row.metadata),
|
|
101
|
+
createdAt: new Date(row.created_at),
|
|
102
|
+
updatedAt: new Date(row.updated_at),
|
|
103
|
+
});
|
|
104
|
+
}
|
|
105
|
+
catch (err) {
|
|
106
|
+
return this.failure(err.message);
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
async updateThread(threadId, metadata) {
|
|
110
|
+
if (!this.db)
|
|
111
|
+
return this.failure('Not connected');
|
|
112
|
+
const existing = await this.getThread(threadId);
|
|
113
|
+
if (!existing.success)
|
|
114
|
+
return existing;
|
|
115
|
+
if (!existing.data)
|
|
116
|
+
return this.failure(`Thread not found: ${threadId}`);
|
|
117
|
+
const updated = {
|
|
118
|
+
...existing.data,
|
|
119
|
+
metadata: { ...existing.data.metadata, ...metadata },
|
|
120
|
+
updatedAt: new Date(),
|
|
121
|
+
};
|
|
122
|
+
try {
|
|
123
|
+
const stmt = this.db.prepare(`
|
|
124
|
+
UPDATE threads SET metadata = ?, updated_at = ? WHERE id = ?
|
|
125
|
+
`);
|
|
126
|
+
stmt.run(JSON.stringify(updated.metadata), updated.updatedAt.toISOString(), threadId);
|
|
127
|
+
return this.success(updated);
|
|
128
|
+
}
|
|
129
|
+
catch (err) {
|
|
130
|
+
return this.failure(err.message);
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
async deleteThread(threadId) {
|
|
134
|
+
if (!this.db)
|
|
135
|
+
return this.failure('Not connected');
|
|
136
|
+
try {
|
|
137
|
+
this.db.prepare('DELETE FROM entries WHERE thread_id = ?').run(threadId);
|
|
138
|
+
this.db.prepare('DELETE FROM threads WHERE id = ?').run(threadId);
|
|
139
|
+
return this.success(undefined);
|
|
140
|
+
}
|
|
141
|
+
catch (err) {
|
|
142
|
+
return this.failure(err.message);
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
async addEntry(entry) {
|
|
146
|
+
if (!this.db)
|
|
147
|
+
return this.failure('Not connected');
|
|
148
|
+
const full = {
|
|
149
|
+
...entry,
|
|
150
|
+
id: this.generateId('entry'),
|
|
151
|
+
createdAt: new Date(),
|
|
152
|
+
};
|
|
153
|
+
try {
|
|
154
|
+
const stmt = this.db.prepare(`
|
|
155
|
+
INSERT INTO entries (id, thread_id, message, tool_calls, tool_results, token_count, created_at, metadata)
|
|
156
|
+
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
|
157
|
+
`);
|
|
158
|
+
stmt.run(full.id, full.threadId, JSON.stringify(full.message), full.toolCalls ? JSON.stringify(full.toolCalls) : null, full.toolResults ? JSON.stringify(full.toolResults) : null, full.tokenCount, full.createdAt.toISOString(), full.metadata ? JSON.stringify(full.metadata) : null);
|
|
159
|
+
return this.success(full);
|
|
160
|
+
}
|
|
161
|
+
catch (err) {
|
|
162
|
+
return this.failure(err.message);
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
async getEntries(options) {
|
|
166
|
+
if (!this.db)
|
|
167
|
+
return this.failure('Not connected');
|
|
168
|
+
try {
|
|
169
|
+
const conditions = ['thread_id = ?'];
|
|
170
|
+
const params = [options.threadId];
|
|
171
|
+
if (options.before) {
|
|
172
|
+
conditions.push('created_at < ?');
|
|
173
|
+
params.push(options.before.toISOString());
|
|
174
|
+
}
|
|
175
|
+
if (options.after) {
|
|
176
|
+
conditions.push('created_at > ?');
|
|
177
|
+
params.push(options.after.toISOString());
|
|
178
|
+
}
|
|
179
|
+
let sql = `SELECT * FROM entries WHERE ${conditions.join(' AND ')} ORDER BY created_at ASC`;
|
|
180
|
+
if (options.limit) {
|
|
181
|
+
sql = `SELECT * FROM (
|
|
182
|
+
SELECT * FROM entries WHERE ${conditions.join(' AND ')} ORDER BY created_at DESC LIMIT ?
|
|
183
|
+
) ORDER BY created_at ASC`;
|
|
184
|
+
params.push(options.limit);
|
|
185
|
+
}
|
|
186
|
+
const stmt = this.db.prepare(sql);
|
|
187
|
+
const rows = stmt.all(...params);
|
|
188
|
+
const entries = rows.map((row) => ({
|
|
189
|
+
id: row.id,
|
|
190
|
+
threadId: row.thread_id,
|
|
191
|
+
message: JSON.parse(row.message),
|
|
192
|
+
toolCalls: options.includeToolCalls && row.tool_calls ? JSON.parse(row.tool_calls) : undefined,
|
|
193
|
+
toolResults: options.includeToolCalls && row.tool_results ? JSON.parse(row.tool_results) : undefined,
|
|
194
|
+
tokenCount: row.token_count,
|
|
195
|
+
createdAt: new Date(row.created_at),
|
|
196
|
+
metadata: row.metadata ? JSON.parse(row.metadata) : undefined,
|
|
197
|
+
}));
|
|
198
|
+
return this.success(entries);
|
|
199
|
+
}
|
|
200
|
+
catch (err) {
|
|
201
|
+
return this.failure(err.message);
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
async getEntry(entryId) {
|
|
205
|
+
if (!this.db)
|
|
206
|
+
return this.failure('Not connected');
|
|
207
|
+
try {
|
|
208
|
+
const stmt = this.db.prepare('SELECT * FROM entries WHERE id = ?');
|
|
209
|
+
const row = stmt.get(entryId);
|
|
210
|
+
if (!row)
|
|
211
|
+
return this.success(null);
|
|
212
|
+
return this.success({
|
|
213
|
+
id: row.id,
|
|
214
|
+
threadId: row.thread_id,
|
|
215
|
+
message: JSON.parse(row.message),
|
|
216
|
+
toolCalls: row.tool_calls ? JSON.parse(row.tool_calls) : undefined,
|
|
217
|
+
toolResults: row.tool_results ? JSON.parse(row.tool_results) : undefined,
|
|
218
|
+
tokenCount: row.token_count,
|
|
219
|
+
createdAt: new Date(row.created_at),
|
|
220
|
+
metadata: row.metadata ? JSON.parse(row.metadata) : undefined,
|
|
221
|
+
});
|
|
222
|
+
}
|
|
223
|
+
catch (err) {
|
|
224
|
+
return this.failure(err.message);
|
|
225
|
+
}
|
|
226
|
+
}
|
|
227
|
+
async deleteEntry(entryId) {
|
|
228
|
+
if (!this.db)
|
|
229
|
+
return this.failure('Not connected');
|
|
230
|
+
try {
|
|
231
|
+
this.db.prepare('DELETE FROM entries WHERE id = ?').run(entryId);
|
|
232
|
+
return this.success(undefined);
|
|
233
|
+
}
|
|
234
|
+
catch (err) {
|
|
235
|
+
return this.failure(err.message);
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
async clearThread(threadId) {
|
|
239
|
+
if (!this.db)
|
|
240
|
+
return this.failure('Not connected');
|
|
241
|
+
try {
|
|
242
|
+
this.db.prepare('DELETE FROM entries WHERE thread_id = ?').run(threadId);
|
|
243
|
+
return this.success(undefined);
|
|
244
|
+
}
|
|
245
|
+
catch (err) {
|
|
246
|
+
return this.failure(err.message);
|
|
247
|
+
}
|
|
248
|
+
}
|
|
249
|
+
}
|
|
250
|
+
//# sourceMappingURL=sqlite.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.js","sourceRoot":"","sources":["../../src/adapters/sqlite.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAe3C,MAAM,OAAO,aAAc,SAAQ,iBAAiB;IACzC,QAAQ,GAAmB,QAAQ,CAAC;IAErC,EAAE,GAAoB,IAAI,CAAC;IAC3B,IAAI,CAAS;IACb,OAAO,CAAU;IAEzB,YAAY,MAA2B;QACrC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAED,IAAI,QAAwC,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACpD,QAAQ,GAAG,YAAY,CAAC,OAAoD,CAAC;QAC/E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC,OAAO,CAAC,4DAA4D,CAAC,CAAC;QACpF,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAElC,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC7C,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;YACvC,CAAC;YAED,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;;;;;;OAwBZ,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAAe,EACf,WAAoC,EAAE,EACtC,QAAiB;QAEjB,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,QAAQ,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YACzC,OAAO;YACP,QAAQ;YACR,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;OAG5B,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CACN,MAAM,CAAC,EAAE,EACT,MAAM,CAAC,OAAO,EACd,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,EAC/B,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE,EAC9B,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE,CAC/B,CAAC;YACF,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,QAAgB;QAC9B,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC;YACnE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAQhB,CAAC;YAEd,IAAI,CAAC,GAAG;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAEpC,OAAO,IAAI,CAAC,OAAO,CAAC;gBAClB,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,QAAQ;gBACrB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC;gBAClC,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC;gBACnC,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC;aACpC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,QAAgB,EAChB,QAAiC;QAEjC,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,CAAC,OAAO;YAAE,OAAO,QAAQ,CAAC;QACvC,IAAI,CAAC,QAAQ,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;QAEzE,MAAM,OAAO,GAAW;YACtB,GAAG,QAAQ,CAAC,IAAI;YAChB,QAAQ,EAAE,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,QAAQ,EAAE;YACpD,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;OAE5B,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,QAAQ,CAAC,CAAC;YACtF,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,QAAgB;QACjC,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACzE,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAClE,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,KAA4C;QACzD,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,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;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;OAG5B,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CACN,IAAI,CAAC,EAAE,EACP,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAC5B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EACtD,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,EAC1D,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAC5B,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CACrD,CAAC;YACF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAA2B;QAC1C,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,CAAC,eAAe,CAAC,CAAC;YACrC,MAAM,MAAM,GAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAE7C,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBAClC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;YAC5C,CAAC;YACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBAClC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;YAC3C,CAAC;YAED,IAAI,GAAG,GAAG,+BAA+B,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,0BAA0B,CAAC;YAC5F,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,GAAG,GAAG;wCAC0B,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC;kCAC9B,CAAC;gBAC3B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAS7B,CAAC;YAEH,MAAM,OAAO,GAAkB,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBAChD,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,QAAQ,EAAE,GAAG,CAAC,SAAS;gBACvB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC;gBAChC,SAAS,EACP,OAAO,CAAC,gBAAgB,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS;gBACrF,WAAW,EACT,OAAO,CAAC,gBAAgB,IAAI,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;gBACzF,UAAU,EAAE,GAAG,CAAC,WAAW;gBAC3B,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC;gBACnC,QAAQ,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;aAC9D,CAAC,CAAC,CAAC;YAEJ,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAe;QAC5B,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAAC;YACnE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAWf,CAAC;YAEd,IAAI,CAAC,GAAG;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAEpC,OAAO,IAAI,CAAC,OAAO,CAAC;gBAClB,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,QAAQ,EAAE,GAAG,CAAC,SAAS;gBACvB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC;gBAChC,SAAS,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS;gBAClE,WAAW,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;gBACxE,UAAU,EAAE,GAAG,CAAC,WAAW;gBAC3B,SAAS,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC;gBACnC,QAAQ,EAAE,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;aAC9D,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAe;QAC/B,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACjE,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB;QAChC,IAAI,CAAC,IAAI,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEnD,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACzE,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,OAAO,CAAE,GAAa,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;CACF"}
|
package/dist/index.d.ts
CHANGED
|
@@ -6,6 +6,9 @@
|
|
|
6
6
|
export { BaseMemoryAdapter, InMemoryAdapter, createMemoryAdapter, type MemoryAdapterConfigUnion, } from './adapters/index';
|
|
7
7
|
export { RedisAdapter } from './adapters/redis';
|
|
8
8
|
export { PostgresAdapter } from './adapters/postgres';
|
|
9
|
+
export { SQLiteAdapter } from './adapters/sqlite';
|
|
10
|
+
export { MongoDBAdapter } from './adapters/mongodb';
|
|
11
|
+
export { QdrantAdapter } from './adapters/qdrant';
|
|
9
12
|
export { ContextBuilder, type ContextBuilderDeps, type BuildContextOptions, } from './context-builder';
|
|
10
13
|
export { countTokens, countMessageTokens, countMessagesTokens, truncateToTokens, } from './token-counter';
|
|
11
14
|
export { OpenAIEmbeddingService, OllamaEmbeddingService, GoogleEmbeddingService, createEmbeddingService, } from './embedding/index';
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,mBAAmB,EACnB,KAAK,wBAAwB,GAC9B,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,mBAAmB,EACnB,KAAK,wBAAwB,GAC9B,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,OAAO,EACL,cAAc,EACd,KAAK,kBAAkB,EACvB,KAAK,mBAAmB,GACzB,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EACL,WAAW,EACX,kBAAkB,EAClB,mBAAmB,EACnB,gBAAgB,GACjB,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EACL,oBAAoB,EACpB,oBAAoB,EACpB,iBAAiB,EACjB,oBAAoB,EACpB,yBAAyB,EACzB,qBAAqB,EACrB,0BAA0B,EAC1B,uBAAuB,EACvB,2BAA2B,EAC3B,2BAA2B,EAC3B,2BAA2B,EAC3B,4BAA4B,GAC7B,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,oBAAoB,EACpB,kBAAkB,EAClB,oBAAoB,EACpB,mBAAmB,EACnB,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,eAAe,EACf,eAAe,EACf,qBAAqB,EACrB,uBAAuB,EACvB,sBAAsB,EACtB,eAAe,EACf,eAAe,EACf,wBAAwB,EACxB,sBAAsB,EACtB,gCAAgC,EAChC,sBAAsB,EACtB,yBAAyB,EACzB,mBAAmB,EACnB,oCAAoC,EACpC,mCAAmC,EACnC,iCAAiC,EACjC,0BAA0B,GAC3B,MAAM,yBAAyB,CAAC;AAEjC,YAAY,EACV,0BAA0B,EAC1B,wBAAwB,EACxB,iBAAiB,EACjB,yBAAyB,GAC1B,MAAM,yBAAyB,CAAC;AAEjC,YAAY,EACV,UAAU,EACV,MAAM,EACN,WAAW,EACX,IAAI,EACJ,SAAS,EACT,cAAc,EACd,mBAAmB,EACnB,kBAAkB,EAClB,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,kBAAkB,EAClB,qBAAqB,EACrB,aAAa,EACb,WAAW,EACX,gBAAgB,EAChB,gBAAgB,EAChB,iBAAiB,EACjB,sBAAsB,EACtB,qBAAqB,EACrB,qBAAqB,EACrB,qBAAqB,EACrB,oBAAoB,EACpB,eAAe,EACf,YAAY,EACZ,YAAY,EACZ,YAAY,EACZ,SAAS,EACT,SAAS,EACT,SAAS,EACT,UAAU,EACV,YAAY,EACZ,mBAAmB,EACnB,SAAS,EACT,SAAS,EACT,gBAAgB,EAChB,eAAe,EACf,kBAAkB,EAClB,0BAA0B,EAC1B,UAAU,EACV,YAAY,EACZ,UAAU,EACV,eAAe,EACf,iBAAiB,EACjB,gBAAgB,EAChB,iBAAiB,EACjB,eAAe,EACf,eAAe,EACf,aAAa,EACb,gBAAgB,EAChB,mBAAmB,EACnB,YAAY,EACZ,gBAAgB,EAChB,oBAAoB,EACpB,8BAA8B,EAC9B,6BAA6B,EAC7B,2BAA2B,GAC5B,MAAM,qBAAqB,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -6,6 +6,9 @@
|
|
|
6
6
|
export { BaseMemoryAdapter, InMemoryAdapter, createMemoryAdapter, } from './adapters/index';
|
|
7
7
|
export { RedisAdapter } from './adapters/redis';
|
|
8
8
|
export { PostgresAdapter } from './adapters/postgres';
|
|
9
|
+
export { SQLiteAdapter } from './adapters/sqlite';
|
|
10
|
+
export { MongoDBAdapter } from './adapters/mongodb';
|
|
11
|
+
export { QdrantAdapter } from './adapters/qdrant';
|
|
9
12
|
export { ContextBuilder, } from './context-builder';
|
|
10
13
|
export { countTokens, countMessageTokens, countMessagesTokens, truncateToTokens, } from './token-counter';
|
|
11
14
|
export { OpenAIEmbeddingService, OllamaEmbeddingService, GoogleEmbeddingService, createEmbeddingService, } from './embedding/index';
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,mBAAmB,GAEpB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,mBAAmB,GAEpB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,OAAO,EACL,cAAc,GAGf,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EACL,WAAW,EACX,kBAAkB,EAClB,mBAAmB,EACnB,gBAAgB,GACjB,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EACL,sBAAsB,EACtB,sBAAsB,EACtB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EACL,oBAAoB,EACpB,oBAAoB,EACpB,iBAAiB,EACjB,oBAAoB,EACpB,yBAAyB,EACzB,qBAAqB,EACrB,0BAA0B,EAC1B,uBAAuB,EACvB,2BAA2B,EAC3B,2BAA2B,EAC3B,2BAA2B,EAC3B,4BAA4B,GAC7B,MAAM,UAAU,CAAC;AAElB,OAAO,EACL,oBAAoB,EACpB,kBAAkB,EAClB,oBAAoB,EACpB,mBAAmB,EACnB,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,eAAe,EACf,eAAe,EACf,qBAAqB,EACrB,uBAAuB,EACvB,sBAAsB,EACtB,eAAe,EACf,eAAe,EACf,wBAAwB,EACxB,sBAAsB,EACtB,gCAAgC,EAChC,sBAAsB,EACtB,yBAAyB,EACzB,mBAAmB,EACnB,oCAAoC,EACpC,mCAAmC,EACnC,iCAAiC,EACjC,0BAA0B,GAC3B,MAAM,yBAAyB,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@cogitator-ai/memory",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.5.0",
|
|
4
4
|
"description": "Memory adapters for Cogitator AI agents",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "./dist/index.js",
|
|
@@ -19,8 +19,8 @@
|
|
|
19
19
|
"@types/pg": "^8.10.9",
|
|
20
20
|
"nanoid": "^5.0.4",
|
|
21
21
|
"zod": "^3.22.4",
|
|
22
|
-
"@cogitator-ai/
|
|
23
|
-
"@cogitator-ai/
|
|
22
|
+
"@cogitator-ai/types": "0.9.0",
|
|
23
|
+
"@cogitator-ai/redis": "0.2.7"
|
|
24
24
|
},
|
|
25
25
|
"optionalDependencies": {
|
|
26
26
|
"ioredis": "^5.3.2",
|