@raviolelabs/engram-mcp 0.5.2 → 0.5.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +13 -1
- package/dist/cloud/pairing.d.ts.map +1 -1
- package/dist/cloud/pairing.js +45 -4
- package/dist/cloud/pairing.js.map +1 -1
- package/package.json +1 -2
- package/dist/mcp-server/tests/mcp-e2e.test.d.ts +0 -2
- package/dist/mcp-server/tests/mcp-e2e.test.d.ts.map +0 -1
- package/dist/mcp-server/tests/mcp-e2e.test.js +0 -157
- package/dist/mcp-server/tests/mcp-e2e.test.js.map +0 -1
- package/dist/memory/modules/_custom/tests/custom-types.test.d.ts +0 -2
- package/dist/memory/modules/_custom/tests/custom-types.test.d.ts.map +0 -1
- package/dist/memory/modules/_custom/tests/custom-types.test.js +0 -96
- package/dist/memory/modules/_custom/tests/custom-types.test.js.map +0 -1
- package/dist/memory/modules/audio/tests/audio.test.d.ts +0 -2
- package/dist/memory/modules/audio/tests/audio.test.d.ts.map +0 -1
- package/dist/memory/modules/audio/tests/audio.test.js +0 -66
- package/dist/memory/modules/audio/tests/audio.test.js.map +0 -1
- package/dist/memory/modules/audio/tests/transcriber.test.d.ts +0 -2
- package/dist/memory/modules/audio/tests/transcriber.test.d.ts.map +0 -1
- package/dist/memory/modules/audio/tests/transcriber.test.js +0 -27
- package/dist/memory/modules/audio/tests/transcriber.test.js.map +0 -1
- package/dist/memory/modules/conversations/tests/conversations.test.d.ts +0 -2
- package/dist/memory/modules/conversations/tests/conversations.test.d.ts.map +0 -1
- package/dist/memory/modules/conversations/tests/conversations.test.js +0 -75
- package/dist/memory/modules/conversations/tests/conversations.test.js.map +0 -1
- package/dist/memory/modules/drive/tests/drive.test.d.ts +0 -2
- package/dist/memory/modules/drive/tests/drive.test.d.ts.map +0 -1
- package/dist/memory/modules/drive/tests/drive.test.js +0 -66
- package/dist/memory/modules/drive/tests/drive.test.js.map +0 -1
- package/dist/memory/modules/notes/tests/notes.test.d.ts +0 -2
- package/dist/memory/modules/notes/tests/notes.test.d.ts.map +0 -1
- package/dist/memory/modules/notes/tests/notes.test.js +0 -64
- package/dist/memory/modules/notes/tests/notes.test.js.map +0 -1
- package/dist/memory/modules/notion/tests/notion.test.d.ts +0 -2
- package/dist/memory/modules/notion/tests/notion.test.d.ts.map +0 -1
- package/dist/memory/modules/notion/tests/notion.test.js +0 -53
- package/dist/memory/modules/notion/tests/notion.test.js.map +0 -1
- package/dist/memory/modules/obsidian/tests/obsidian.test.d.ts +0 -2
- package/dist/memory/modules/obsidian/tests/obsidian.test.d.ts.map +0 -1
- package/dist/memory/modules/obsidian/tests/obsidian.test.js +0 -72
- package/dist/memory/modules/obsidian/tests/obsidian.test.js.map +0 -1
- package/dist/memory/modules/obsidian/tests/vault-reader.test.d.ts +0 -2
- package/dist/memory/modules/obsidian/tests/vault-reader.test.d.ts.map +0 -1
- package/dist/memory/modules/obsidian/tests/vault-reader.test.js +0 -37
- package/dist/memory/modules/obsidian/tests/vault-reader.test.js.map +0 -1
- package/dist/memory/modules/youtube/tests/channel.test.d.ts +0 -2
- package/dist/memory/modules/youtube/tests/channel.test.d.ts.map +0 -1
- package/dist/memory/modules/youtube/tests/channel.test.js +0 -61
- package/dist/memory/modules/youtube/tests/channel.test.js.map +0 -1
- package/dist/memory/modules/youtube/tests/transcript-fetcher.test.d.ts +0 -2
- package/dist/memory/modules/youtube/tests/transcript-fetcher.test.d.ts.map +0 -1
- package/dist/memory/modules/youtube/tests/transcript-fetcher.test.js +0 -23
- package/dist/memory/modules/youtube/tests/transcript-fetcher.test.js.map +0 -1
- package/dist/memory/modules/youtube/tests/youtube.test.d.ts +0 -2
- package/dist/memory/modules/youtube/tests/youtube.test.d.ts.map +0 -1
- package/dist/memory/modules/youtube/tests/youtube.test.js +0 -61
- package/dist/memory/modules/youtube/tests/youtube.test.js.map +0 -1
- package/dist/sync/tests/apply.test.d.ts +0 -4
- package/dist/sync/tests/apply.test.d.ts.map +0 -1
- package/dist/sync/tests/apply.test.js +0 -117
- package/dist/sync/tests/apply.test.js.map +0 -1
- package/dist/sync/tests/ops-log.test.d.ts +0 -2
- package/dist/sync/tests/ops-log.test.d.ts.map +0 -1
- package/dist/sync/tests/ops-log.test.js +0 -105
- package/dist/sync/tests/ops-log.test.js.map +0 -1
- package/dist/sync/tests/two-device-sync.test.d.ts +0 -2
- package/dist/sync/tests/two-device-sync.test.d.ts.map +0 -1
- package/dist/sync/tests/two-device-sync.test.js +0 -250
- package/dist/sync/tests/two-device-sync.test.js.map +0 -1
- package/dist/tests/chunker.test.d.ts +0 -2
- package/dist/tests/chunker.test.d.ts.map +0 -1
- package/dist/tests/chunker.test.js +0 -24
- package/dist/tests/chunker.test.js.map +0 -1
- package/dist/tests/cloud-auth.test.d.ts +0 -2
- package/dist/tests/cloud-auth.test.d.ts.map +0 -1
- package/dist/tests/cloud-auth.test.js +0 -75
- package/dist/tests/cloud-auth.test.js.map +0 -1
- package/dist/tests/cloud-crypto.test.d.ts +0 -2
- package/dist/tests/cloud-crypto.test.d.ts.map +0 -1
- package/dist/tests/cloud-crypto.test.js +0 -58
- package/dist/tests/cloud-crypto.test.js.map +0 -1
- package/dist/tests/cloud-integration.test.d.ts +0 -2
- package/dist/tests/cloud-integration.test.d.ts.map +0 -1
- package/dist/tests/cloud-integration.test.js +0 -193
- package/dist/tests/cloud-integration.test.js.map +0 -1
- package/dist/tests/cloud-pairing.test.d.ts +0 -2
- package/dist/tests/cloud-pairing.test.d.ts.map +0 -1
- package/dist/tests/cloud-pairing.test.js +0 -86
- package/dist/tests/cloud-pairing.test.js.map +0 -1
- package/dist/tests/cloud-saves-integration.test.d.ts +0 -2
- package/dist/tests/cloud-saves-integration.test.d.ts.map +0 -1
- package/dist/tests/cloud-saves-integration.test.js +0 -92
- package/dist/tests/cloud-saves-integration.test.js.map +0 -1
- package/dist/tests/cloud-transit.test.d.ts +0 -2
- package/dist/tests/cloud-transit.test.d.ts.map +0 -1
- package/dist/tests/cloud-transit.test.js +0 -263
- package/dist/tests/cloud-transit.test.js.map +0 -1
- package/dist/tests/config.test.d.ts +0 -2
- package/dist/tests/config.test.d.ts.map +0 -1
- package/dist/tests/config.test.js +0 -25
- package/dist/tests/config.test.js.map +0 -1
- package/dist/tests/db.test.d.ts +0 -2
- package/dist/tests/db.test.d.ts.map +0 -1
- package/dist/tests/db.test.js +0 -73
- package/dist/tests/db.test.js.map +0 -1
- package/dist/tests/embeddings-providers.test.d.ts +0 -2
- package/dist/tests/embeddings-providers.test.d.ts.map +0 -1
- package/dist/tests/embeddings-providers.test.js +0 -62
- package/dist/tests/embeddings-providers.test.js.map +0 -1
- package/dist/tests/embeddings.test.d.ts +0 -2
- package/dist/tests/embeddings.test.d.ts.map +0 -1
- package/dist/tests/embeddings.test.js +0 -22
- package/dist/tests/embeddings.test.js.map +0 -1
- package/dist/tests/integrations-api.test.d.ts +0 -2
- package/dist/tests/integrations-api.test.d.ts.map +0 -1
- package/dist/tests/integrations-api.test.js +0 -129
- package/dist/tests/integrations-api.test.js.map +0 -1
- package/dist/tests/memory-store.test.d.ts +0 -2
- package/dist/tests/memory-store.test.d.ts.map +0 -1
- package/dist/tests/memory-store.test.js +0 -129
- package/dist/tests/memory-store.test.js.map +0 -1
- package/dist/tests/module-registry.test.d.ts +0 -2
- package/dist/tests/module-registry.test.d.ts.map +0 -1
- package/dist/tests/module-registry.test.js +0 -44
- package/dist/tests/module-registry.test.js.map +0 -1
- package/dist/tests/property-extractor.test.d.ts +0 -2
- package/dist/tests/property-extractor.test.d.ts.map +0 -1
- package/dist/tests/property-extractor.test.js +0 -24
- package/dist/tests/property-extractor.test.js.map +0 -1
- package/dist/tests/public-tools.test.d.ts +0 -2
- package/dist/tests/public-tools.test.d.ts.map +0 -1
- package/dist/tests/public-tools.test.js +0 -286
- package/dist/tests/public-tools.test.js.map +0 -1
- package/dist/tests/reindex.test.d.ts +0 -2
- package/dist/tests/reindex.test.d.ts.map +0 -1
- package/dist/tests/reindex.test.js +0 -58
- package/dist/tests/reindex.test.js.map +0 -1
- package/dist/tests/scope-encryption.test.d.ts +0 -2
- package/dist/tests/scope-encryption.test.d.ts.map +0 -1
- package/dist/tests/scope-encryption.test.js +0 -71
- package/dist/tests/scope-encryption.test.js.map +0 -1
- package/dist/tests/shamir.test.d.ts +0 -2
- package/dist/tests/shamir.test.d.ts.map +0 -1
- package/dist/tests/shamir.test.js +0 -57
- package/dist/tests/shamir.test.js.map +0 -1
- package/dist/tests/source-registry.test.d.ts +0 -2
- package/dist/tests/source-registry.test.d.ts.map +0 -1
- package/dist/tests/source-registry.test.js +0 -58
- package/dist/tests/source-registry.test.js.map +0 -1
- package/dist/tests/types.test.d.ts +0 -2
- package/dist/tests/types.test.d.ts.map +0 -1
- package/dist/tests/types.test.js +0 -26
- package/dist/tests/types.test.js.map +0 -1
- package/dist/tests/vector.test.d.ts +0 -2
- package/dist/tests/vector.test.d.ts.map +0 -1
- package/dist/tests/vector.test.js +0 -61
- package/dist/tests/vector.test.js.map +0 -1
- package/dist/tests/wikilinks.test.d.ts +0 -2
- package/dist/tests/wikilinks.test.d.ts.map +0 -1
- package/dist/tests/wikilinks.test.js +0 -20
- package/dist/tests/wikilinks.test.js.map +0 -1
- package/dist/webapp/tests/api.test.d.ts +0 -2
- package/dist/webapp/tests/api.test.d.ts.map +0 -1
- package/dist/webapp/tests/api.test.js +0 -125
- package/dist/webapp/tests/api.test.js.map +0 -1
- package/dist/webapp/tests/mcp-http.test.d.ts +0 -2
- package/dist/webapp/tests/mcp-http.test.d.ts.map +0 -1
- package/dist/webapp/tests/mcp-http.test.js +0 -47
- package/dist/webapp/tests/mcp-http.test.js.map +0 -1
|
@@ -1,129 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Smoke test for integrations config read/write helpers.
|
|
3
|
-
*
|
|
4
|
-
* Uses a temp dir to test PATCH → GET status → DELETE flow without a running
|
|
5
|
-
* HTTP server. We test the config file writing logic in isolation, verifying
|
|
6
|
-
* that credentials persist, permissions are 0o600, and deletion works.
|
|
7
|
-
*/
|
|
8
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
9
|
-
import fs from 'fs';
|
|
10
|
-
import os from 'os';
|
|
11
|
-
import path from 'path';
|
|
12
|
-
// -- Inline the helpers from integrations.ts (avoids starting a full Express app) --
|
|
13
|
-
function getConfigPath(configDir) {
|
|
14
|
-
return path.join(configDir, 'config.json');
|
|
15
|
-
}
|
|
16
|
-
function readRawConfig(configDir) {
|
|
17
|
-
const p = getConfigPath(configDir);
|
|
18
|
-
if (!fs.existsSync(p))
|
|
19
|
-
return {};
|
|
20
|
-
try {
|
|
21
|
-
return JSON.parse(fs.readFileSync(p, 'utf-8'));
|
|
22
|
-
}
|
|
23
|
-
catch {
|
|
24
|
-
return {};
|
|
25
|
-
}
|
|
26
|
-
}
|
|
27
|
-
function writeConfig(configDir, updated) {
|
|
28
|
-
if (!fs.existsSync(configDir)) {
|
|
29
|
-
fs.mkdirSync(configDir, { recursive: true, mode: 0o700 });
|
|
30
|
-
}
|
|
31
|
-
const configPath = getConfigPath(configDir);
|
|
32
|
-
const tmpPath = configPath + '.tmp.' + process.pid;
|
|
33
|
-
fs.writeFileSync(tmpPath, JSON.stringify(updated, null, 2), { encoding: 'utf-8', mode: 0o600 });
|
|
34
|
-
fs.renameSync(tmpPath, configPath);
|
|
35
|
-
try {
|
|
36
|
-
fs.chmodSync(configPath, 0o600);
|
|
37
|
-
}
|
|
38
|
-
catch {
|
|
39
|
-
// non-fatal
|
|
40
|
-
}
|
|
41
|
-
}
|
|
42
|
-
function getStatus(config) {
|
|
43
|
-
const driveConf = config.drive;
|
|
44
|
-
const notionConf = config.notion;
|
|
45
|
-
return {
|
|
46
|
-
drive: {
|
|
47
|
-
configured: !!(driveConf?.clientId && driveConf?.clientSecret),
|
|
48
|
-
},
|
|
49
|
-
notion: {
|
|
50
|
-
configured: !!(notionConf?.clientId && notionConf?.clientSecret),
|
|
51
|
-
},
|
|
52
|
-
};
|
|
53
|
-
}
|
|
54
|
-
// -----------------------------------------------------------------------
|
|
55
|
-
describe('integrations config helpers', () => {
|
|
56
|
-
let tmpDir;
|
|
57
|
-
beforeEach(() => {
|
|
58
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-integrations-test-'));
|
|
59
|
-
});
|
|
60
|
-
afterEach(() => {
|
|
61
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
62
|
-
});
|
|
63
|
-
it('returns not-configured when no config exists', () => {
|
|
64
|
-
const raw = readRawConfig(tmpDir);
|
|
65
|
-
const status = getStatus(raw);
|
|
66
|
-
expect(status.drive.configured).toBe(false);
|
|
67
|
-
expect(status.notion.configured).toBe(false);
|
|
68
|
-
});
|
|
69
|
-
it('PATCH drive: writes credentials and returns configured=true', () => {
|
|
70
|
-
const raw = readRawConfig(tmpDir);
|
|
71
|
-
raw.drive = { clientId: 'test-client-id', clientSecret: 'test-secret', redirectPort: 7777 };
|
|
72
|
-
writeConfig(tmpDir, raw);
|
|
73
|
-
const after = readRawConfig(tmpDir);
|
|
74
|
-
const status = getStatus(after);
|
|
75
|
-
expect(status.drive.configured).toBe(true);
|
|
76
|
-
expect(status.notion.configured).toBe(false);
|
|
77
|
-
const drive = after.drive;
|
|
78
|
-
expect(drive.clientId).toBe('test-client-id');
|
|
79
|
-
expect(drive.clientSecret).toBe('test-secret');
|
|
80
|
-
});
|
|
81
|
-
it('does not clobber other config keys when patching drive', () => {
|
|
82
|
-
// Simulate existing embeddings config
|
|
83
|
-
const initial = {
|
|
84
|
-
dataDir: '~/.engram',
|
|
85
|
-
embeddings: { provider: 'ollama', model: 'nomic-embed-text', dimensions: 768 },
|
|
86
|
-
};
|
|
87
|
-
writeConfig(tmpDir, initial);
|
|
88
|
-
const raw = readRawConfig(tmpDir);
|
|
89
|
-
raw.drive = { clientId: 'cid', clientSecret: 'csecret', redirectPort: 7777 };
|
|
90
|
-
writeConfig(tmpDir, raw);
|
|
91
|
-
const after = readRawConfig(tmpDir);
|
|
92
|
-
const embeddings = after.embeddings;
|
|
93
|
-
expect(embeddings.provider).toBe('ollama'); // not clobbered
|
|
94
|
-
expect(getStatus(after).drive.configured).toBe(true);
|
|
95
|
-
});
|
|
96
|
-
it('DELETE drive: wipes credentials', () => {
|
|
97
|
-
const raw = readRawConfig(tmpDir);
|
|
98
|
-
raw.drive = { clientId: 'cid', clientSecret: 'csecret', redirectPort: 7777 };
|
|
99
|
-
writeConfig(tmpDir, raw);
|
|
100
|
-
const toDelete = readRawConfig(tmpDir);
|
|
101
|
-
delete toDelete.drive;
|
|
102
|
-
writeConfig(tmpDir, toDelete);
|
|
103
|
-
const after = readRawConfig(tmpDir);
|
|
104
|
-
expect(getStatus(after).drive.configured).toBe(false);
|
|
105
|
-
expect(after.drive).toBeUndefined();
|
|
106
|
-
});
|
|
107
|
-
it('config.json has mode 0o600 after write', () => {
|
|
108
|
-
writeConfig(tmpDir, { test: true });
|
|
109
|
-
const configPath = getConfigPath(tmpDir);
|
|
110
|
-
const stat = fs.statSync(configPath);
|
|
111
|
-
// Mode bits: file type (0o100000) + permissions (0o600)
|
|
112
|
-
const perms = stat.mode & 0o777;
|
|
113
|
-
expect(perms).toBe(0o600);
|
|
114
|
-
});
|
|
115
|
-
it('PATCH notion: writes notion credentials', () => {
|
|
116
|
-
const raw = readRawConfig(tmpDir);
|
|
117
|
-
raw.notion = { clientId: 'notion-cid', clientSecret: 'notion-secret', redirectPort: 7777 };
|
|
118
|
-
writeConfig(tmpDir, raw);
|
|
119
|
-
const after = readRawConfig(tmpDir);
|
|
120
|
-
expect(getStatus(after).notion.configured).toBe(true);
|
|
121
|
-
});
|
|
122
|
-
it('atomic write: tmp file does not persist on success', () => {
|
|
123
|
-
writeConfig(tmpDir, { ok: true });
|
|
124
|
-
const files = fs.readdirSync(tmpDir);
|
|
125
|
-
const tmpFiles = files.filter((f) => f.includes('.tmp.'));
|
|
126
|
-
expect(tmpFiles).toHaveLength(0);
|
|
127
|
-
});
|
|
128
|
-
});
|
|
129
|
-
//# sourceMappingURL=integrations-api.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"integrations-api.test.js","sourceRoot":"","sources":["../../src/tests/integrations-api.test.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,qFAAqF;AAErF,SAAS,aAAa,CAAC,SAAiB;IACtC,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;AAC7C,CAAC;AAED,SAAS,aAAa,CAAC,SAAiB;IACtC,MAAM,CAAC,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IACnC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;QAAE,OAAO,EAAE,CAAC;IACjC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,EAAE,OAAO,CAAC,CAA4B,CAAC;IAC5E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,SAAiB,EAAE,OAAgC;IACtE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,MAAM,UAAU,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;IAC5C,MAAM,OAAO,GAAG,UAAU,GAAG,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC;IACnD,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAChG,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnC,IAAI,CAAC;QACH,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;IAAC,MAAM,CAAC;QACP,YAAY;IACd,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,MAA+B;IAChD,MAAM,SAAS,GAAG,MAAM,CAAC,KAAiE,CAAC;IAC3F,MAAM,UAAU,GAAG,MAAM,CAAC,MAAkE,CAAC;IAC7F,OAAO;QACL,KAAK,EAAE;YACL,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,QAAQ,IAAI,SAAS,EAAE,YAAY,CAAC;SAC/D;QACD,MAAM,EAAE;YACN,UAAU,EAAE,CAAC,CAAC,CAAC,UAAU,EAAE,QAAQ,IAAI,UAAU,EAAE,YAAY,CAAC;SACjE;KACF,CAAC;AACJ,CAAC;AAED,0EAA0E;AAE1E,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,GAAG,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QAClC,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,GAAG,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QAClC,GAAG,CAAC,KAAK,GAAG,EAAE,QAAQ,EAAE,gBAAgB,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QAC5F,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAEzB,MAAM,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE7C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAmD,CAAC;QACxE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,sCAAsC;QACtC,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,WAAW;YACpB,UAAU,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,kBAAkB,EAAE,UAAU,EAAE,GAAG,EAAE;SACpD,CAAC;QAC7B,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAE7B,MAAM,GAAG,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QAClC,GAAG,CAAC,KAAK,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QAC7E,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAEzB,MAAM,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,KAAK,CAAC,UAAkC,CAAC;QAC5D,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,gBAAgB;QAC5D,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,GAAG,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QAClC,GAAG,CAAC,KAAK,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QAC7E,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAEzB,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,QAAQ,CAAC,KAAK,CAAC;QACtB,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAE9B,MAAM,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,WAAW,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACpC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACrC,wDAAwD;QACxD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC;QAChC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,GAAG,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QAClC,GAAG,CAAC,MAAM,GAAG,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QAC3F,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAEzB,MAAM,KAAK,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,WAAW,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAClC,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"memory-store.test.d.ts","sourceRoot":"","sources":["../../src/tests/memory-store.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,129 +0,0 @@
|
|
|
1
|
-
// src/tests/memory-store.test.ts
|
|
2
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
3
|
-
import fs from 'fs';
|
|
4
|
-
import os from 'os';
|
|
5
|
-
import path from 'path';
|
|
6
|
-
import { ulid } from 'ulid';
|
|
7
|
-
import { initDb, closeDb } from '../db/index.js';
|
|
8
|
-
import { initVectorStore } from '../vector/store.js';
|
|
9
|
-
import { MemoryStore } from '../memory/core/store.js';
|
|
10
|
-
const embeddingsConfig = {
|
|
11
|
-
provider: 'ollama',
|
|
12
|
-
baseUrl: 'http://localhost:11434',
|
|
13
|
-
model: 'nomic-embed-text',
|
|
14
|
-
dimensions: 768,
|
|
15
|
-
};
|
|
16
|
-
function buildItem(overrides = {}) {
|
|
17
|
-
const now = new Date().toISOString();
|
|
18
|
-
return {
|
|
19
|
-
id: ulid(),
|
|
20
|
-
type: 'notes',
|
|
21
|
-
source_id: 'local:test',
|
|
22
|
-
content: 'Test content',
|
|
23
|
-
content_hash: 'h1',
|
|
24
|
-
properties: { created_at: now, ingested_at: now },
|
|
25
|
-
wikilinks: [],
|
|
26
|
-
related_ids: [],
|
|
27
|
-
embedding_model: 'nomic-embed-text:v1.5',
|
|
28
|
-
...overrides,
|
|
29
|
-
};
|
|
30
|
-
}
|
|
31
|
-
describe('MemoryStore', () => {
|
|
32
|
-
let tmpDir;
|
|
33
|
-
let store;
|
|
34
|
-
beforeEach(() => {
|
|
35
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-store-'));
|
|
36
|
-
initDb(tmpDir);
|
|
37
|
-
initVectorStore(tmpDir);
|
|
38
|
-
store = new MemoryStore({ embeddings: embeddingsConfig });
|
|
39
|
-
});
|
|
40
|
-
afterEach(() => {
|
|
41
|
-
closeDb();
|
|
42
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
43
|
-
});
|
|
44
|
-
it('inserts and retrieves a memory by id', async () => {
|
|
45
|
-
const item = buildItem({ content: 'I love TypeScript' });
|
|
46
|
-
await store.insert(item);
|
|
47
|
-
const fetched = store.getById(item.id);
|
|
48
|
-
expect(fetched?.id).toBe(item.id);
|
|
49
|
-
expect(fetched?.content).toBe('I love TypeScript');
|
|
50
|
-
});
|
|
51
|
-
it('searches semantically and returns hits', async () => {
|
|
52
|
-
const a = buildItem({ content: 'I love TypeScript', source_id: 'local:a' });
|
|
53
|
-
const b = buildItem({ content: 'The weather is nice', source_id: 'local:b' });
|
|
54
|
-
await store.insert(a);
|
|
55
|
-
await store.insert(b);
|
|
56
|
-
const hits = await store.search('notes', 'coding in TS', 5);
|
|
57
|
-
expect(hits.length).toBeGreaterThan(0);
|
|
58
|
-
expect(hits[0].memory.id).toBe(a.id);
|
|
59
|
-
});
|
|
60
|
-
it('deletes a memory and removes it from results', async () => {
|
|
61
|
-
const item = buildItem({ content: 'delete me' });
|
|
62
|
-
await store.insert(item);
|
|
63
|
-
await store.delete(item.id);
|
|
64
|
-
expect(store.getById(item.id)).toBeUndefined();
|
|
65
|
-
const hits = await store.search('notes', 'delete me', 5);
|
|
66
|
-
expect(hits.find((h) => h.memory.id === item.id)).toBeUndefined();
|
|
67
|
-
});
|
|
68
|
-
it('findRelated returns items sharing a wikilink', async () => {
|
|
69
|
-
const a = buildItem({
|
|
70
|
-
content: 'Working on [[ProjectAlpha]] integration',
|
|
71
|
-
source_id: 'local:a',
|
|
72
|
-
});
|
|
73
|
-
const b = buildItem({
|
|
74
|
-
content: 'Status update for [[ProjectAlpha]]: stuck on auth',
|
|
75
|
-
source_id: 'local:b',
|
|
76
|
-
});
|
|
77
|
-
const c = buildItem({ content: 'Unrelated content about weather', source_id: 'local:c' });
|
|
78
|
-
await store.insert(a);
|
|
79
|
-
await store.insert(b);
|
|
80
|
-
await store.insert(c);
|
|
81
|
-
// Set title 'ProjectAlpha' on b to make it the wikilink target
|
|
82
|
-
store.setProperties(b.id, { title: 'ProjectAlpha' });
|
|
83
|
-
const related = await store.findRelated(b.id, 5);
|
|
84
|
-
expect(related.some((r) => r.memory.id === a.id)).toBe(true);
|
|
85
|
-
expect(related.some((r) => r.memory.id === c.id)).toBe(false);
|
|
86
|
-
});
|
|
87
|
-
it('setProperties updates and persists', () => {
|
|
88
|
-
const item = buildItem({ content: 'X' });
|
|
89
|
-
return store.insert(item).then(() => {
|
|
90
|
-
const ok = store.setProperties(item.id, { title: 'New Title', tags: ['x', 'y'] });
|
|
91
|
-
expect(ok).toBe(true);
|
|
92
|
-
const fetched = store.getById(item.id);
|
|
93
|
-
expect(fetched?.properties.title).toBe('New Title');
|
|
94
|
-
expect(fetched?.properties.tags).toEqual(['x', 'y']);
|
|
95
|
-
});
|
|
96
|
-
});
|
|
97
|
-
});
|
|
98
|
-
describe('MemoryStore with property extraction', () => {
|
|
99
|
-
let tmpDir;
|
|
100
|
-
let store;
|
|
101
|
-
beforeEach(() => {
|
|
102
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-store-extract-'));
|
|
103
|
-
initDb(tmpDir);
|
|
104
|
-
initVectorStore(tmpDir);
|
|
105
|
-
store = new MemoryStore({
|
|
106
|
-
embeddings: embeddingsConfig,
|
|
107
|
-
propertyExtraction: {
|
|
108
|
-
enabled: true,
|
|
109
|
-
baseUrl: 'http://localhost:11434',
|
|
110
|
-
model: 'llama3.2:3b',
|
|
111
|
-
maxTokens: 300,
|
|
112
|
-
},
|
|
113
|
-
});
|
|
114
|
-
});
|
|
115
|
-
afterEach(() => {
|
|
116
|
-
closeDb();
|
|
117
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
118
|
-
});
|
|
119
|
-
it('fills missing properties (title, tags) automatically', async () => {
|
|
120
|
-
const item = buildItem({
|
|
121
|
-
content: 'Met with Alice about the new Polymarket integration; deadline April 15.',
|
|
122
|
-
});
|
|
123
|
-
await store.insert(item);
|
|
124
|
-
const fetched = store.getById(item.id);
|
|
125
|
-
expect(fetched?.properties.title).toBeTruthy();
|
|
126
|
-
expect(fetched?.properties.tags?.length ?? 0).toBeGreaterThan(0);
|
|
127
|
-
}, 60_000);
|
|
128
|
-
});
|
|
129
|
-
//# sourceMappingURL=memory-store.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"memory-store.test.js","sourceRoot":"","sources":["../../src/tests/memory-store.test.ts"],"names":[],"mappings":"AAAA,iCAAiC;AACjC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAGtD,MAAM,gBAAgB,GAAG;IACvB,QAAQ,EAAE,QAAiB;IAC3B,OAAO,EAAE,wBAAwB;IACjC,KAAK,EAAE,kBAAkB;IACzB,UAAU,EAAE,GAAG;CAChB,CAAC;AAEF,SAAS,SAAS,CAAC,YAAiC,EAAE;IACpD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrC,OAAO;QACL,EAAE,EAAE,IAAI,EAAE;QACV,IAAI,EAAE,OAAO;QACb,SAAS,EAAE,YAAY;QACvB,OAAO,EAAE,cAAc;QACvB,YAAY,EAAE,IAAI;QAClB,UAAU,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE;QACjD,SAAS,EAAE,EAAE;QACb,WAAW,EAAE,EAAE;QACf,eAAe,EAAE,uBAAuB;QACxC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,MAAc,CAAC;IACnB,IAAI,KAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,eAAe,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,CAAC;QACf,eAAe,CAAC,MAAM,CAAC,CAAC;QACxB,KAAK,GAAG,IAAI,WAAW,CAAC,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,EAAE,CAAC;QACV,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC;QACzD,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,CAAC,GAAG,SAAS,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QAC5E,MAAM,CAAC,GAAG,SAAS,CAAC,EAAE,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9E,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAEtB,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QAC5D,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QACjD,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE5B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QACzD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,CAAC,GAAG,SAAS,CAAC;YAClB,OAAO,EAAE,yCAAyC;YAClD,SAAS,EAAE,SAAS;SACrB,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,SAAS,CAAC;YAClB,OAAO,EAAE,mDAAmD;YAC5D,SAAS,EAAE,SAAS;SACrB,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,SAAS,CAAC,EAAE,OAAO,EAAE,iCAAiC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;QAC1F,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAEtB,+DAA+D;QAC/D,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;QAErD,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QACzC,OAAO,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE;YAClC,MAAM,EAAE,GAAG,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YAClF,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACpD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,IAAI,MAAc,CAAC;IACnB,IAAI,KAAkB,CAAC;IAEvB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAC;QACzE,MAAM,CAAC,MAAM,CAAC,CAAC;QACf,eAAe,CAAC,MAAM,CAAC,CAAC;QACxB,KAAK,GAAG,IAAI,WAAW,CAAC;YACtB,UAAU,EAAE,gBAAgB;YAC5B,kBAAkB,EAAE;gBAClB,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,wBAAwB;gBACjC,KAAK,EAAE,aAAa;gBACpB,SAAS,EAAE,GAAG;aACf;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,EAAE,CAAC;QACV,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,OAAO,EAAE,yEAAyE;SACnF,CAAC,CAAC;QACH,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAC;QAC/C,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACnE,CAAC,EAAE,MAAM,CAAC,CAAC;AACb,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"module-registry.test.d.ts","sourceRoot":"","sources":["../../src/tests/module-registry.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,44 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, beforeEach } from 'vitest';
|
|
2
|
-
import { moduleRegistry } from '../memory/core/module-registry.js';
|
|
3
|
-
const fakeModule = (id) => ({
|
|
4
|
-
id,
|
|
5
|
-
displayName: `Fake ${id}`,
|
|
6
|
-
isCustom: false,
|
|
7
|
-
onBoot: async () => { },
|
|
8
|
-
onShutdown: async () => { },
|
|
9
|
-
ingest: async () => [],
|
|
10
|
-
tools: [
|
|
11
|
-
{
|
|
12
|
-
name: `search_${id}`,
|
|
13
|
-
description: `Search ${id}`,
|
|
14
|
-
inputSchema: { type: 'object', properties: {} },
|
|
15
|
-
handler: async () => ({ ok: true }),
|
|
16
|
-
},
|
|
17
|
-
],
|
|
18
|
-
});
|
|
19
|
-
describe('ModuleRegistry', () => {
|
|
20
|
-
beforeEach(() => {
|
|
21
|
-
// Reset registry between tests
|
|
22
|
-
for (const m of moduleRegistry.list()) {
|
|
23
|
-
moduleRegistry['modules'].delete(m.id);
|
|
24
|
-
}
|
|
25
|
-
moduleRegistry['booted'] = false;
|
|
26
|
-
});
|
|
27
|
-
it('registers a module and lists it', () => {
|
|
28
|
-
moduleRegistry.register(fakeModule('test'));
|
|
29
|
-
expect(moduleRegistry.list().map((m) => m.id)).toEqual(['test']);
|
|
30
|
-
});
|
|
31
|
-
it('throws on duplicate registration', () => {
|
|
32
|
-
moduleRegistry.register(fakeModule('dup'));
|
|
33
|
-
expect(() => moduleRegistry.register(fakeModule('dup'))).toThrow(/already registered/);
|
|
34
|
-
});
|
|
35
|
-
it('collects tools from all modules', () => {
|
|
36
|
-
moduleRegistry.register(fakeModule('a'));
|
|
37
|
-
moduleRegistry.register(fakeModule('b'));
|
|
38
|
-
expect(moduleRegistry
|
|
39
|
-
.collectTools()
|
|
40
|
-
.map((t) => t.name)
|
|
41
|
-
.sort()).toEqual(['search_a', 'search_b']);
|
|
42
|
-
});
|
|
43
|
-
});
|
|
44
|
-
//# sourceMappingURL=module-registry.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"module-registry.test.js","sourceRoot":"","sources":["../../src/tests/module-registry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAC;AAGnE,MAAM,UAAU,GAAG,CAAC,EAAU,EAAgB,EAAE,CAAC,CAAC;IAChD,EAAE;IACF,WAAW,EAAE,QAAQ,EAAE,EAAE;IACzB,QAAQ,EAAE,KAAK;IACf,MAAM,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;IACtB,UAAU,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;IAC1B,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;IACtB,KAAK,EAAE;QACL;YACE,IAAI,EAAE,UAAU,EAAE,EAAE;YACpB,WAAW,EAAE,UAAU,EAAE,EAAE;YAC3B,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;YAC/C,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;SACpC;KACF;CACF,CAAC,CAAC;AAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,UAAU,CAAC,GAAG,EAAE;QACd,+BAA+B;QAC/B,KAAK,MAAM,CAAC,IAAI,cAAc,CAAC,IAAI,EAAE,EAAE,CAAC;YACtC,cAAc,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;QACD,cAAc,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;IACzF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACzC,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACzC,MAAM,CACJ,cAAc;aACX,YAAY,EAAE;aACd,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,EAAE,CACV,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"property-extractor.test.d.ts","sourceRoot":"","sources":["../../src/tests/property-extractor.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,24 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect } from 'vitest';
|
|
2
|
-
import { extractProperties } from '../memory/core/property-extractor.js';
|
|
3
|
-
const config = {
|
|
4
|
-
enabled: true,
|
|
5
|
-
baseUrl: 'http://localhost:11434',
|
|
6
|
-
model: 'llama3.2:3b',
|
|
7
|
-
maxTokens: 300,
|
|
8
|
-
};
|
|
9
|
-
describe('property extractor (real Ollama)', () => {
|
|
10
|
-
it('extracts a title and tags from a short note', async () => {
|
|
11
|
-
const result = await extractProperties('The team agreed on a quarterly all-hands for the launch of PolymarketPro on June 1st. Action: confirm venue.', config);
|
|
12
|
-
expect(result.title).toBeTruthy();
|
|
13
|
-
expect(result.tags).toBeInstanceOf(Array);
|
|
14
|
-
expect(result.tags.length).toBeGreaterThan(0);
|
|
15
|
-
// Note: action_required is only extracted when the smart 5-field prompt is loaded
|
|
16
|
-
// (src/private/prompts/extraction-system.ts). OSS basic prompt returns only title + tags.
|
|
17
|
-
// See src/private/tests/property-extractor-smart.test.ts for the full assertion.
|
|
18
|
-
}, 30_000);
|
|
19
|
-
it('returns empty result when disabled', async () => {
|
|
20
|
-
const result = await extractProperties('whatever', { ...config, enabled: false });
|
|
21
|
-
expect(result).toEqual({});
|
|
22
|
-
});
|
|
23
|
-
});
|
|
24
|
-
//# sourceMappingURL=property-extractor.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"property-extractor.test.js","sourceRoot":"","sources":["../../src/tests/property-extractor.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AAEzE,MAAM,MAAM,GAAG;IACb,OAAO,EAAE,IAAI;IACb,OAAO,EAAE,wBAAwB;IACjC,KAAK,EAAE,aAAa;IACpB,SAAS,EAAE,GAAG;CACf,CAAC;AAEF,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,MAAM,GAAG,MAAM,iBAAiB,CACpC,8GAA8G,EAC9G,MAAM,CACP,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,IAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC/C,kFAAkF;QAClF,0FAA0F;QAC1F,iFAAiF;IACnF,CAAC,EAAE,MAAM,CAAC,CAAC;IAEX,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,UAAU,EAAE,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QAClF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"public-tools.test.d.ts","sourceRoot":"","sources":["../../src/tests/public-tools.test.ts"],"names":[],"mappings":""}
|