@raviolelabs/engram-mcp 0.5.3 → 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/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,66 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import fs from 'fs';
|
|
3
|
-
import os from 'os';
|
|
4
|
-
import path from 'path';
|
|
5
|
-
import { initDb, closeDb, getDb } from '../../../../db/index.js';
|
|
6
|
-
import { initVectorStore } from '../../../../vector/store.js';
|
|
7
|
-
import { MemoryStore } from '../../../core/store.js';
|
|
8
|
-
import { sourceRegistry } from '../../../core/source-registry.js';
|
|
9
|
-
import { buildDriveItem } from '../ingest.js';
|
|
10
|
-
const embeddingsConfig = {
|
|
11
|
-
provider: 'ollama',
|
|
12
|
-
baseUrl: 'http://localhost:11434',
|
|
13
|
-
model: 'nomic-embed-text',
|
|
14
|
-
dimensions: 768,
|
|
15
|
-
};
|
|
16
|
-
describe('drive module integration (token + registry + ingest)', () => {
|
|
17
|
-
let tmpDir;
|
|
18
|
-
let store;
|
|
19
|
-
beforeEach(() => {
|
|
20
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-drive-'));
|
|
21
|
-
initDb(tmpDir);
|
|
22
|
-
initVectorStore(tmpDir);
|
|
23
|
-
store = new MemoryStore({ embeddings: embeddingsConfig });
|
|
24
|
-
});
|
|
25
|
-
afterEach(() => {
|
|
26
|
-
closeDb();
|
|
27
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
28
|
-
});
|
|
29
|
-
it('stores OAuth tokens via direct DB write and reads them back', () => {
|
|
30
|
-
getDb()
|
|
31
|
-
.prepare(`INSERT INTO oauth_tokens (provider, access_token, refresh_token, expires_at, extra_json, updated_at)
|
|
32
|
-
VALUES ('drive', 'fake', 'fakerefresh', ?, '{}', ?)`)
|
|
33
|
-
.run(Date.now() + 3_600_000, Date.now());
|
|
34
|
-
const row = getDb()
|
|
35
|
-
.prepare("SELECT access_token FROM oauth_tokens WHERE provider = 'drive'")
|
|
36
|
-
.get();
|
|
37
|
-
expect(row?.access_token).toBe('fake');
|
|
38
|
-
});
|
|
39
|
-
it('buildDriveItem + insert + search round-trips', async () => {
|
|
40
|
-
const item = buildDriveItem({
|
|
41
|
-
metadata: {
|
|
42
|
-
id: 'doc-123',
|
|
43
|
-
name: 'Quarterly Roadmap',
|
|
44
|
-
mimeType: 'application/vnd.google-apps.document',
|
|
45
|
-
modifiedTime: '2026-05-15T10:00:00Z',
|
|
46
|
-
},
|
|
47
|
-
content: 'The roadmap focuses on memory ingestion improvements and cloud transit hardening.',
|
|
48
|
-
embeddingModel: 'nomic-embed-text:v1.5',
|
|
49
|
-
});
|
|
50
|
-
await store.insert(item);
|
|
51
|
-
const hits = await store.search('drive', 'cloud transit', 5);
|
|
52
|
-
expect(hits.length).toBeGreaterThan(0);
|
|
53
|
-
expect(hits[0].memory.id).toBe(item.id);
|
|
54
|
-
});
|
|
55
|
-
it('watcher records sync timestamps via sourceRegistry', () => {
|
|
56
|
-
const id = sourceRegistry.add({
|
|
57
|
-
module_id: 'drive',
|
|
58
|
-
external_id: 'doc-X',
|
|
59
|
-
display_name: 'X',
|
|
60
|
-
});
|
|
61
|
-
sourceRegistry.recordSync(id, '2026-05-15T11:00:00Z');
|
|
62
|
-
const s = sourceRegistry.get(id);
|
|
63
|
-
expect(s?.last_modified_remote).toBe('2026-05-15T11:00:00Z');
|
|
64
|
-
});
|
|
65
|
-
});
|
|
66
|
-
//# sourceMappingURL=drive.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"drive.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/drive/tests/drive.test.ts"],"names":[],"mappings":"AAAA,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,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,MAAM,gBAAgB,GAAG;IACvB,QAAQ,EAAE,QAAiB;IAC3B,OAAO,EAAE,wBAAwB;IACjC,KAAK,EAAE,kBAAkB;IACzB,UAAU,EAAE,GAAG;CAChB,CAAC;AAEF,QAAQ,CAAC,sDAAsD,EAAE,GAAG,EAAE;IACpE,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,6DAA6D,EAAE,GAAG,EAAE;QACrE,KAAK,EAAE;aACJ,OAAO,CACN;6DACqD,CACtD;aACA,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,GAAG,GAAG,KAAK,EAAE;aAChB,OAAO,CAAC,gEAAgE,CAAC;aACzE,GAAG,EAA0C,CAAC;QACjD,MAAM,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,IAAI,GAAG,cAAc,CAAC;YAC1B,QAAQ,EAAE;gBACR,EAAE,EAAE,SAAS;gBACb,IAAI,EAAE,mBAAmB;gBACzB,QAAQ,EAAE,sCAAsC;gBAChD,YAAY,EAAE,sBAAsB;aACrC;YACD,OAAO,EAAE,mFAAmF;YAC5F,cAAc,EAAE,uBAAuB;SACxC,CAAC,CAAC;QACH,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;QAC7D,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,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,EAAE,GAAG,cAAc,CAAC,GAAG,CAAC;YAC5B,SAAS,EAAE,OAAO;YAClB,WAAW,EAAE,OAAO;YACpB,YAAY,EAAE,GAAG;SAClB,CAAC,CAAC;QACH,cAAc,CAAC,UAAU,CAAC,EAAE,EAAE,sBAAsB,CAAC,CAAC;QACtD,MAAM,CAAC,GAAG,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,MAAM,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"notes.test.d.ts","sourceRoot":"","sources":["../../../../../src/memory/modules/notes/tests/notes.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,64 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import fs from 'fs';
|
|
3
|
-
import os from 'os';
|
|
4
|
-
import path from 'path';
|
|
5
|
-
import { initDb, closeDb } from '../../../../db/index.js';
|
|
6
|
-
import { initVectorStore } from '../../../../vector/store.js';
|
|
7
|
-
import { MemoryStore } from '../../../core/store.js';
|
|
8
|
-
import { createNotesModule, buildNotesModuleTools } from '../module.js';
|
|
9
|
-
const config = {
|
|
10
|
-
dataDir: '~/.engram',
|
|
11
|
-
embeddings: {
|
|
12
|
-
provider: 'ollama',
|
|
13
|
-
baseUrl: 'http://localhost:11434',
|
|
14
|
-
model: 'nomic-embed-text',
|
|
15
|
-
dimensions: 768,
|
|
16
|
-
},
|
|
17
|
-
drive: undefined,
|
|
18
|
-
notion: undefined,
|
|
19
|
-
propertyExtraction: {
|
|
20
|
-
enabled: true,
|
|
21
|
-
baseUrl: 'http://localhost:11434',
|
|
22
|
-
model: 'llama3.2:3b',
|
|
23
|
-
maxTokens: 300,
|
|
24
|
-
},
|
|
25
|
-
whisper: { enabled: true, model: 'small.en', language: 'auto' },
|
|
26
|
-
youtube: { enabled: true, preferLanguage: 'en', fallbackToYtdlp: true },
|
|
27
|
-
modules: { notes: { enabled: true } },
|
|
28
|
-
mcp: { stdio: true, httpPort: 7777 },
|
|
29
|
-
};
|
|
30
|
-
describe('notes module', () => {
|
|
31
|
-
let tmpDir;
|
|
32
|
-
let store;
|
|
33
|
-
beforeEach(async () => {
|
|
34
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-notes-'));
|
|
35
|
-
initDb(tmpDir);
|
|
36
|
-
initVectorStore(tmpDir);
|
|
37
|
-
store = new MemoryStore({ embeddings: config.embeddings });
|
|
38
|
-
const mod = createNotesModule(config);
|
|
39
|
-
await mod.onBoot({ store });
|
|
40
|
-
});
|
|
41
|
-
afterEach(() => {
|
|
42
|
-
closeDb();
|
|
43
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
44
|
-
});
|
|
45
|
-
it('add_note tool inserts and search_notes retrieves it', async () => {
|
|
46
|
-
const tools = buildNotesModuleTools(store, config);
|
|
47
|
-
const addNote = tools.find((t) => t.name === 'add_note');
|
|
48
|
-
const searchNotes = tools.find((t) => t.name === 'search_notes');
|
|
49
|
-
await addNote.handler({ content: 'EngramMCP is a memory layer for AI agents' });
|
|
50
|
-
const results = (await searchNotes.handler({ query: 'AI memory' }));
|
|
51
|
-
expect(results.length).toBeGreaterThan(0);
|
|
52
|
-
expect(results[0].score).toBeGreaterThan(0);
|
|
53
|
-
});
|
|
54
|
-
it('extracts wikilinks from content', async () => {
|
|
55
|
-
const mod = createNotesModule(config);
|
|
56
|
-
await mod.onBoot({ store });
|
|
57
|
-
const items = await mod.ingest({
|
|
58
|
-
content: 'See [[ProjectX]] and [[Notes]].',
|
|
59
|
-
source_id: 'manual:test',
|
|
60
|
-
});
|
|
61
|
-
expect(items[0].wikilinks).toEqual(['ProjectX', 'Notes']);
|
|
62
|
-
});
|
|
63
|
-
});
|
|
64
|
-
//# sourceMappingURL=notes.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"notes.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/notes/tests/notes.test.ts"],"names":[],"mappings":"AAAA,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,MAAM,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AAGxE,MAAM,MAAM,GAAiB;IAC3B,OAAO,EAAE,WAAW;IACpB,UAAU,EAAE;QACV,QAAQ,EAAE,QAAiB;QAC3B,OAAO,EAAE,wBAAwB;QACjC,KAAK,EAAE,kBAAkB;QACzB,UAAU,EAAE,GAAG;KAChB;IACD,KAAK,EAAE,SAAS;IAChB,MAAM,EAAE,SAAS;IACjB,kBAAkB,EAAE;QAClB,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,wBAAwB;QACjC,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,GAAG;KACf;IACD,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE;IAC/D,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE;IACvE,OAAO,EAAE,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;IACrC,GAAG,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE;CACrC,CAAC;AAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,MAAc,CAAC;IACnB,IAAI,KAAkB,CAAC;IAEvB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,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,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;QAC3D,MAAM,GAAG,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC9B,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,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,KAAK,GAAG,qBAAqB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAE,CAAC;QAC1D,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAE,CAAC;QAElE,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,2CAA2C,EAAE,CAAC,CAAC;QAChF,MAAM,OAAO,GAAG,CAAC,MAAM,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAGhE,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,GAAG,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QAC5B,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC;YAC7B,OAAO,EAAE,iCAAiC;YAC1C,SAAS,EAAE,aAAa;SACzB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"notion.test.d.ts","sourceRoot":"","sources":["../../../../../src/memory/modules/notion/tests/notion.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,53 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import fs from 'fs';
|
|
3
|
-
import os from 'os';
|
|
4
|
-
import path from 'path';
|
|
5
|
-
import { initDb, closeDb, getDb } from '../../../../db/index.js';
|
|
6
|
-
import { initVectorStore } from '../../../../vector/store.js';
|
|
7
|
-
import { MemoryStore } from '../../../core/store.js';
|
|
8
|
-
import { isNotionConnected, getNotionToken } from '../oauth.js';
|
|
9
|
-
import { buildNotionItem } from '../ingest.js';
|
|
10
|
-
const embeddingsConfig = {
|
|
11
|
-
provider: 'ollama',
|
|
12
|
-
baseUrl: 'http://localhost:11434',
|
|
13
|
-
model: 'nomic-embed-text',
|
|
14
|
-
dimensions: 768,
|
|
15
|
-
};
|
|
16
|
-
describe('notion module integration (token + ingest)', () => {
|
|
17
|
-
let tmpDir;
|
|
18
|
-
let store;
|
|
19
|
-
beforeEach(() => {
|
|
20
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-notion-'));
|
|
21
|
-
initDb(tmpDir);
|
|
22
|
-
initVectorStore(tmpDir);
|
|
23
|
-
store = new MemoryStore({ embeddings: embeddingsConfig });
|
|
24
|
-
});
|
|
25
|
-
afterEach(() => {
|
|
26
|
-
closeDb();
|
|
27
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
28
|
-
});
|
|
29
|
-
it('stores OAuth tokens via direct DB write and reads them back', () => {
|
|
30
|
-
getDb()
|
|
31
|
-
.prepare(`INSERT INTO oauth_tokens (provider, access_token, refresh_token, expires_at, extra_json, updated_at)
|
|
32
|
-
VALUES ('notion', 'secret_xyz', NULL, NULL, '{"workspace_id":"w1","workspace_name":"MyOrg","bot_id":"b1"}', ?)`)
|
|
33
|
-
.run(Date.now());
|
|
34
|
-
expect(isNotionConnected()).toBe(true);
|
|
35
|
-
expect(getNotionToken()).toBe('secret_xyz');
|
|
36
|
-
});
|
|
37
|
-
it('buildNotionItem + insert + search', async () => {
|
|
38
|
-
const item = buildNotionItem({
|
|
39
|
-
metadata: {
|
|
40
|
-
id: 'page-1',
|
|
41
|
-
title: 'Engineering Standards',
|
|
42
|
-
last_edited_time: '2026-05-15T10:00:00Z',
|
|
43
|
-
url: 'https://www.notion.so/Engineering-Standards-page-1',
|
|
44
|
-
},
|
|
45
|
-
content: 'Use TypeScript strict mode. Use ULIDs. Avoid `any` type.',
|
|
46
|
-
embeddingModel: 'nomic-embed-text:v1.5',
|
|
47
|
-
});
|
|
48
|
-
await store.insert(item);
|
|
49
|
-
const hits = await store.search('notion', 'TypeScript guidelines', 5);
|
|
50
|
-
expect(hits[0].memory.id).toBe(item.id);
|
|
51
|
-
});
|
|
52
|
-
});
|
|
53
|
-
//# sourceMappingURL=notion.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"notion.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/notion/tests/notion.test.ts"],"names":[],"mappings":"AAAA,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,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/C,MAAM,gBAAgB,GAAG;IACvB,QAAQ,EAAE,QAAiB;IAC3B,OAAO,EAAE,wBAAwB;IACjC,KAAK,EAAE,kBAAkB;IACzB,UAAU,EAAE,GAAG;CAChB,CAAC;AAEF,QAAQ,CAAC,4CAA4C,EAAE,GAAG,EAAE;IAC1D,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,gBAAgB,CAAC,CAAC,CAAC;QAClE,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,6DAA6D,EAAE,GAAG,EAAE;QACrE,KAAK,EAAE;aACJ,OAAO,CACN;wHACgH,CACjH;aACA,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QACnB,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,IAAI,GAAG,eAAe,CAAC;YAC3B,QAAQ,EAAE;gBACR,EAAE,EAAE,QAAQ;gBACZ,KAAK,EAAE,uBAAuB;gBAC9B,gBAAgB,EAAE,sBAAsB;gBACxC,GAAG,EAAE,oDAAoD;aAC1D;YACD,OAAO,EAAE,0DAA0D;YACnE,cAAc,EAAE,uBAAuB;SACxC,CAAC,CAAC;QACH,MAAM,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,uBAAuB,EAAE,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"obsidian.test.d.ts","sourceRoot":"","sources":["../../../../../src/memory/modules/obsidian/tests/obsidian.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,72 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import fs from 'fs';
|
|
3
|
-
import os from 'os';
|
|
4
|
-
import path from 'path';
|
|
5
|
-
import { initDb, closeDb } from '../../../../db/index.js';
|
|
6
|
-
import { initVectorStore } from '../../../../vector/store.js';
|
|
7
|
-
import { MemoryStore } from '../../../core/store.js';
|
|
8
|
-
import { buildObsidianModuleTools, createObsidianModule } from '../module.js';
|
|
9
|
-
const config = {
|
|
10
|
-
dataDir: '~/.engram',
|
|
11
|
-
embeddings: {
|
|
12
|
-
provider: 'ollama',
|
|
13
|
-
baseUrl: 'http://localhost:11434',
|
|
14
|
-
model: 'nomic-embed-text',
|
|
15
|
-
dimensions: 768,
|
|
16
|
-
},
|
|
17
|
-
drive: undefined,
|
|
18
|
-
notion: undefined,
|
|
19
|
-
propertyExtraction: {
|
|
20
|
-
enabled: false,
|
|
21
|
-
baseUrl: 'http://localhost:11434',
|
|
22
|
-
model: 'llama3.2:3b',
|
|
23
|
-
maxTokens: 300,
|
|
24
|
-
},
|
|
25
|
-
whisper: { enabled: true, model: 'small.en', language: 'auto' },
|
|
26
|
-
youtube: { enabled: true, preferLanguage: 'en', fallbackToYtdlp: false },
|
|
27
|
-
modules: { obsidian: { enabled: true } },
|
|
28
|
-
mcp: { stdio: true, httpPort: 7777 },
|
|
29
|
-
};
|
|
30
|
-
describe('obsidian module e2e', () => {
|
|
31
|
-
let tmpData;
|
|
32
|
-
let vault;
|
|
33
|
-
let store;
|
|
34
|
-
beforeEach(async () => {
|
|
35
|
-
tmpData = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-obs-data-'));
|
|
36
|
-
vault = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-obs-vault-'));
|
|
37
|
-
initDb(tmpData);
|
|
38
|
-
initVectorStore(tmpData);
|
|
39
|
-
store = new MemoryStore({ embeddings: config.embeddings });
|
|
40
|
-
fs.writeFileSync(path.join(vault, 'engineering-standards.md'), `---\ntitle: Engineering Standards\ntags: [code, standards]\n---\n# Engineering Standards\n\nUse TypeScript strict mode and ULIDs.\n\nSee [[Code Review]] for review process.\n`);
|
|
41
|
-
fs.writeFileSync(path.join(vault, 'code-review.md'), `# Code Review\n\nEvery PR needs at least 1 reviewer.\n`);
|
|
42
|
-
const mod = createObsidianModule(config);
|
|
43
|
-
await mod.onBoot({ store });
|
|
44
|
-
});
|
|
45
|
-
afterEach(() => {
|
|
46
|
-
closeDb();
|
|
47
|
-
fs.rmSync(tmpData, { recursive: true, force: true });
|
|
48
|
-
fs.rmSync(vault, { recursive: true, force: true });
|
|
49
|
-
});
|
|
50
|
-
it('add_obsidian_vault indexes every markdown file', async () => {
|
|
51
|
-
const tools = buildObsidianModuleTools(store, config);
|
|
52
|
-
const result = (await tools
|
|
53
|
-
.find((t) => t.name === 'add_obsidian_vault')
|
|
54
|
-
.handler({
|
|
55
|
-
vault_path: vault,
|
|
56
|
-
}));
|
|
57
|
-
expect(result.files_ingested).toBe(2);
|
|
58
|
-
}, 30_000);
|
|
59
|
-
it('parses frontmatter title and tags', async () => {
|
|
60
|
-
const tools = buildObsidianModuleTools(store, config);
|
|
61
|
-
await tools.find((t) => t.name === 'add_obsidian_vault').handler({ vault_path: vault });
|
|
62
|
-
const types = store.listTypes();
|
|
63
|
-
expect(types).toContain('obsidian');
|
|
64
|
-
const hits = await store.search('obsidian', 'standards typescript ULID', 5);
|
|
65
|
-
expect(hits.length).toBeGreaterThan(0);
|
|
66
|
-
const top = hits[0].memory;
|
|
67
|
-
expect(top.properties.title).toBe('Engineering Standards');
|
|
68
|
-
expect(top.properties.tags).toContain('code');
|
|
69
|
-
expect(top.wikilinks).toContain('Code Review');
|
|
70
|
-
}, 30_000);
|
|
71
|
-
});
|
|
72
|
-
//# sourceMappingURL=obsidian.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"obsidian.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/obsidian/tests/obsidian.test.ts"],"names":[],"mappings":"AAAA,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,MAAM,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,wBAAwB,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AAG9E,MAAM,MAAM,GAAiB;IAC3B,OAAO,EAAE,WAAW;IACpB,UAAU,EAAE;QACV,QAAQ,EAAE,QAAiB;QAC3B,OAAO,EAAE,wBAAwB;QACjC,KAAK,EAAE,kBAAkB;QACzB,UAAU,EAAE,GAAG;KAChB;IACD,KAAK,EAAE,SAAS;IAChB,MAAM,EAAE,SAAS;IACjB,kBAAkB,EAAE;QAClB,OAAO,EAAE,KAAK;QACd,OAAO,EAAE,wBAAwB;QACjC,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,GAAG;KACf;IACD,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE;IAC/D,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE;IACxE,OAAO,EAAE,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;IACxC,GAAG,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE;CACrC,CAAC;AAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,OAAe,CAAC;IACpB,IAAI,KAAa,CAAC;IAClB,IAAI,KAAkB,CAAC;IAEvB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,kBAAkB,CAAC,CAAC,CAAC;QACrE,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QACpE,MAAM,CAAC,OAAO,CAAC,CAAC;QAChB,eAAe,CAAC,OAAO,CAAC,CAAC;QACzB,KAAK,GAAG,IAAI,WAAW,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;QAE3D,EAAE,CAAC,aAAa,CACd,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,0BAA0B,CAAC,EAC5C,gLAAgL,CACjL,CAAC;QACF,EAAE,CAAC,aAAa,CACd,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,gBAAgB,CAAC,EAClC,wDAAwD,CACzD,CAAC;QAEF,MAAM,GAAG,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,EAAE,CAAC;QACV,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,KAAK,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,CAAC,MAAM,KAAK;aACxB,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAAE;aAC7C,OAAO,CAAC;YACP,UAAU,EAAE,KAAK;SAClB,CAAC,CAA+B,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,EAAE,MAAM,CAAC,CAAC;IAEX,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,KAAK,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACtD,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAAE,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;QAEzF,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAEpC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,2BAA2B,EAAE,CAAC,CAAC,CAAC;QAC5E,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC3D,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;IACjD,CAAC,EAAE,MAAM,CAAC,CAAC;AACb,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"vault-reader.test.d.ts","sourceRoot":"","sources":["../../../../../src/memory/modules/obsidian/tests/vault-reader.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,37 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import fs from 'fs';
|
|
3
|
-
import os from 'os';
|
|
4
|
-
import path from 'path';
|
|
5
|
-
import { readVault } from '../vault-reader.js';
|
|
6
|
-
describe('vault-reader', () => {
|
|
7
|
-
let vault;
|
|
8
|
-
beforeEach(() => {
|
|
9
|
-
vault = fs.mkdtempSync(path.join(os.tmpdir(), 'obsidian-vault-'));
|
|
10
|
-
fs.writeFileSync(path.join(vault, 'note1.md'), '# First note\n[[note2]] reference');
|
|
11
|
-
fs.mkdirSync(path.join(vault, 'sub'));
|
|
12
|
-
fs.writeFileSync(path.join(vault, 'sub', 'note2.md'), '# Second\nlots of content');
|
|
13
|
-
// Files that should be ignored
|
|
14
|
-
fs.mkdirSync(path.join(vault, '.obsidian'));
|
|
15
|
-
fs.writeFileSync(path.join(vault, '.obsidian', 'workspace.json'), '{}');
|
|
16
|
-
fs.writeFileSync(path.join(vault, 'photo.png'), 'not markdown');
|
|
17
|
-
// .obsidianignore
|
|
18
|
-
fs.writeFileSync(path.join(vault, '.obsidianignore'), 'archive/*\nprivate/**\n');
|
|
19
|
-
fs.mkdirSync(path.join(vault, 'archive'));
|
|
20
|
-
fs.writeFileSync(path.join(vault, 'archive', 'old.md'), 'should not appear');
|
|
21
|
-
});
|
|
22
|
-
afterEach(() => {
|
|
23
|
-
fs.rmSync(vault, { recursive: true, force: true });
|
|
24
|
-
});
|
|
25
|
-
it('returns markdown files only, respecting .obsidianignore + .obsidian/', async () => {
|
|
26
|
-
const files = await readVault(vault);
|
|
27
|
-
const rels = files.map((f) => f.relativePath).sort();
|
|
28
|
-
expect(rels).toEqual(['note1.md', 'sub/note2.md']);
|
|
29
|
-
});
|
|
30
|
-
it('reads content correctly', async () => {
|
|
31
|
-
const files = await readVault(vault);
|
|
32
|
-
const note1 = files.find((f) => f.relativePath === 'note1.md');
|
|
33
|
-
expect(note1?.content).toContain('First note');
|
|
34
|
-
expect(note1?.content).toContain('[[note2]]');
|
|
35
|
-
});
|
|
36
|
-
});
|
|
37
|
-
//# sourceMappingURL=vault-reader.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"vault-reader.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/obsidian/tests/vault-reader.test.ts"],"names":[],"mappings":"AAAA,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,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAE/C,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,KAAa,CAAC;IAElB,UAAU,CAAC,GAAG,EAAE;QACd,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;QAClE,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,mCAAmC,CAAC,CAAC;QACpF,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;QACtC,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,EAAE,2BAA2B,CAAC,CAAC;QACnF,+BAA+B;QAC/B,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC;QAC5C,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,gBAAgB,CAAC,EAAE,IAAI,CAAC,CAAC;QACxE,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,EAAE,cAAc,CAAC,CAAC;QAChE,kBAAkB;QAClB,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,iBAAiB,CAAC,EAAE,yBAAyB,CAAC,CAAC;QACjF,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;QAC1C,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,mBAAmB,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACrC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACrC,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,KAAK,UAAU,CAAC,CAAC;QAC/D,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"channel.test.d.ts","sourceRoot":"","sources":["../../../../../src/memory/modules/youtube/tests/channel.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,61 +0,0 @@
|
|
|
1
|
-
// src/memory/modules/youtube/tests/channel.test.ts
|
|
2
|
-
import { describe, it, expect } from 'vitest';
|
|
3
|
-
import { resolveChannelId } from '../watcher.js';
|
|
4
|
-
const REAL_CHANNEL_ID = 'UCnUYZLuoy1rq1aVMwx4aTzw'; // Andrej Karpathy — stable, public
|
|
5
|
-
const REAL_CHANNEL_URL = 'https://www.youtube.com/@AndrejKarpathy';
|
|
6
|
-
describe('resolveChannelId', () => {
|
|
7
|
-
it('returns the ID unchanged when given a valid channel ID', async () => {
|
|
8
|
-
const id = await resolveChannelId(REAL_CHANNEL_ID);
|
|
9
|
-
expect(id).toBe(REAL_CHANNEL_ID);
|
|
10
|
-
});
|
|
11
|
-
it('resolves a channel URL to a channel ID', async () => {
|
|
12
|
-
let id;
|
|
13
|
-
try {
|
|
14
|
-
id = await resolveChannelId(REAL_CHANNEL_URL);
|
|
15
|
-
}
|
|
16
|
-
catch (e) {
|
|
17
|
-
const msg = e instanceof Error ? e.message : String(e);
|
|
18
|
-
// Network may be unavailable or YouTube may block — skip gracefully
|
|
19
|
-
if (msg.includes('HTTP') || msg.includes('fetch') || msg.includes('Cannot resolve')) {
|
|
20
|
-
console.warn(`Skipping resolveChannelId URL test — ${msg}`);
|
|
21
|
-
return;
|
|
22
|
-
}
|
|
23
|
-
throw e;
|
|
24
|
-
}
|
|
25
|
-
expect(id).toMatch(/^UC[A-Za-z0-9_-]{22}$/);
|
|
26
|
-
}, 20_000);
|
|
27
|
-
it('throws on unresolvable input', async () => {
|
|
28
|
-
await expect(resolveChannelId('not-a-channel')).rejects.toThrow(/Cannot resolve/);
|
|
29
|
-
});
|
|
30
|
-
});
|
|
31
|
-
describe('RSS feed fetch (real network)', () => {
|
|
32
|
-
it('fetches and parses the channel RSS feed', async () => {
|
|
33
|
-
const { XMLParser } = await import('fast-xml-parser');
|
|
34
|
-
const feedUrl = `https://www.youtube.com/feeds/videos.xml?channel_id=${REAL_CHANNEL_ID}`;
|
|
35
|
-
let res;
|
|
36
|
-
try {
|
|
37
|
-
res = await fetch(feedUrl, {
|
|
38
|
-
headers: {
|
|
39
|
-
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 Chrome/124 Safari/537.36',
|
|
40
|
-
'Accept-Language': 'en-US,en;q=0.9',
|
|
41
|
-
},
|
|
42
|
-
signal: AbortSignal.timeout(15_000),
|
|
43
|
-
});
|
|
44
|
-
}
|
|
45
|
-
catch (e) {
|
|
46
|
-
// Network unavailable in CI — skip gracefully
|
|
47
|
-
console.warn(`Skipping RSS test — network error: ${e instanceof Error ? e.message : String(e)}`);
|
|
48
|
-
return;
|
|
49
|
-
}
|
|
50
|
-
if (!res.ok) {
|
|
51
|
-
// YouTube rate-limit or temporary error — skip gracefully instead of failing
|
|
52
|
-
console.warn(`Skipping RSS test — HTTP ${res.status} from YouTube`);
|
|
53
|
-
return;
|
|
54
|
-
}
|
|
55
|
-
const xml = await res.text();
|
|
56
|
-
const parser = new XMLParser({ ignoreAttributes: false });
|
|
57
|
-
const parsed = parser.parse(xml);
|
|
58
|
-
expect(parsed.feed).toBeDefined();
|
|
59
|
-
}, 20_000);
|
|
60
|
-
});
|
|
61
|
-
//# sourceMappingURL=channel.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"channel.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/youtube/tests/channel.test.ts"],"names":[],"mappings":"AAAA,mDAAmD;AACnD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEjD,MAAM,eAAe,GAAG,0BAA0B,CAAC,CAAC,mCAAmC;AACvF,MAAM,gBAAgB,GAAG,yCAAyC,CAAC;AAEnE,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,GAAG,MAAM,gBAAgB,CAAC,eAAe,CAAC,CAAC;QACnD,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,IAAI,EAAU,CAAC;QACf,IAAI,CAAC;YACH,EAAE,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACvD,oEAAoE;YACpE,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBACpF,OAAO,CAAC,IAAI,CAAC,wCAAwC,GAAG,EAAE,CAAC,CAAC;gBAC5D,OAAO;YACT,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC;QACD,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAC9C,CAAC,EAAE,MAAM,CAAC,CAAC;IAEX,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,uDAAuD,eAAe,EAAE,CAAC;QACzF,IAAI,GAAa,CAAC;QAClB,IAAI,CAAC;YACH,GAAG,GAAG,MAAM,KAAK,CAAC,OAAO,EAAE;gBACzB,OAAO,EAAE;oBACP,YAAY,EACV,6FAA6F;oBAC/F,iBAAiB,EAAE,gBAAgB;iBACpC;gBACD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;aACpC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,8CAA8C;YAC9C,OAAO,CAAC,IAAI,CACV,sCAAsC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CACnF,CAAC;YACF,OAAO;QACT,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,6EAA6E;YAC7E,OAAO,CAAC,IAAI,CAAC,4BAA4B,GAAG,CAAC,MAAM,eAAe,CAAC,CAAC;YACpE,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAmC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;IACpC,CAAC,EAAE,MAAM,CAAC,CAAC;AACb,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"transcript-fetcher.test.d.ts","sourceRoot":"","sources":["../../../../../src/memory/modules/youtube/tests/transcript-fetcher.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,23 +0,0 @@
|
|
|
1
|
-
// src/memory/modules/youtube/tests/transcript-fetcher.test.ts
|
|
2
|
-
import { describe, it, expect } from 'vitest';
|
|
3
|
-
import { extractVideoId, fetchTranscript } from '../transcript-fetcher.js';
|
|
4
|
-
// Note: YouTube's timedtext API frequently returns empty XML for videos even when captions
|
|
5
|
-
// exist in captionTracks. We enable yt-dlp fallback for reliable transcript retrieval.
|
|
6
|
-
const config = { enabled: true, preferLanguage: 'en', fallbackToYtdlp: true };
|
|
7
|
-
// Stable target: TED Talk Sir Ken Robinson "Do schools kill creativity?"
|
|
8
|
-
// (long-standing video with reliable English captions)
|
|
9
|
-
const TEST_URL = 'https://www.youtube.com/watch?v=iG9CE55wbtY';
|
|
10
|
-
describe('youtube transcript fetcher', () => {
|
|
11
|
-
it('extracts video id from various URL shapes', () => {
|
|
12
|
-
expect(extractVideoId('https://www.youtube.com/watch?v=iG9CE55wbtY')).toBe('iG9CE55wbtY');
|
|
13
|
-
expect(extractVideoId('https://youtu.be/iG9CE55wbtY')).toBe('iG9CE55wbtY');
|
|
14
|
-
expect(extractVideoId('iG9CE55wbtY')).toBe('iG9CE55wbtY');
|
|
15
|
-
});
|
|
16
|
-
it('fetches an English transcript for a known video', async () => {
|
|
17
|
-
const result = await fetchTranscript(TEST_URL, config);
|
|
18
|
-
expect(result.video_id).toBe('iG9CE55wbtY');
|
|
19
|
-
expect(result.segments.length).toBeGreaterThan(50);
|
|
20
|
-
expect(result.full_text.toLowerCase()).toContain('creativity');
|
|
21
|
-
}, 60_000);
|
|
22
|
-
});
|
|
23
|
-
//# sourceMappingURL=transcript-fetcher.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"transcript-fetcher.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/youtube/tests/transcript-fetcher.test.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAE3E,2FAA2F;AAC3F,uFAAuF;AACvF,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;AAE9E,yEAAyE;AACzE,uDAAuD;AACvD,MAAM,QAAQ,GAAG,6CAA6C,CAAC;AAE/D,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,cAAc,CAAC,6CAA6C,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1F,MAAM,CAAC,cAAc,CAAC,8BAA8B,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3E,MAAM,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACjE,CAAC,EAAE,MAAM,CAAC,CAAC;AACb,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"youtube.test.d.ts","sourceRoot":"","sources":["../../../../../src/memory/modules/youtube/tests/youtube.test.ts"],"names":[],"mappings":""}
|
|
@@ -1,61 +0,0 @@
|
|
|
1
|
-
// src/memory/modules/youtube/tests/youtube.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 { initDb, closeDb } from '../../../../db/index.js';
|
|
7
|
-
import { initVectorStore } from '../../../../vector/store.js';
|
|
8
|
-
import { MemoryStore } from '../../../core/store.js';
|
|
9
|
-
import { buildYoutubeModuleTools } from '../module.js';
|
|
10
|
-
const config = {
|
|
11
|
-
dataDir: '~/.engram',
|
|
12
|
-
embeddings: {
|
|
13
|
-
provider: 'ollama',
|
|
14
|
-
baseUrl: 'http://localhost:11434',
|
|
15
|
-
model: 'nomic-embed-text',
|
|
16
|
-
dimensions: 768,
|
|
17
|
-
},
|
|
18
|
-
drive: undefined,
|
|
19
|
-
notion: undefined,
|
|
20
|
-
propertyExtraction: {
|
|
21
|
-
enabled: false,
|
|
22
|
-
baseUrl: 'http://localhost:11434',
|
|
23
|
-
model: 'llama3.2:3b',
|
|
24
|
-
maxTokens: 300,
|
|
25
|
-
},
|
|
26
|
-
whisper: { enabled: true, model: 'tiny.en', language: 'en' },
|
|
27
|
-
youtube: { enabled: true, preferLanguage: 'en', fallbackToYtdlp: true },
|
|
28
|
-
modules: { youtube: { enabled: true } },
|
|
29
|
-
mcp: { stdio: true, httpPort: 7777 },
|
|
30
|
-
};
|
|
31
|
-
describe('youtube module e2e (real video)', () => {
|
|
32
|
-
let tmpDir;
|
|
33
|
-
let store;
|
|
34
|
-
beforeEach(() => {
|
|
35
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'engram-yt-'));
|
|
36
|
-
initDb(tmpDir);
|
|
37
|
-
initVectorStore(tmpDir);
|
|
38
|
-
store = new MemoryStore({ embeddings: config.embeddings });
|
|
39
|
-
});
|
|
40
|
-
afterEach(() => {
|
|
41
|
-
closeDb();
|
|
42
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
43
|
-
});
|
|
44
|
-
it('add_youtube_url + search_youtube end-to-end', async () => {
|
|
45
|
-
const tools = buildYoutubeModuleTools(store, config);
|
|
46
|
-
const addRes = (await tools
|
|
47
|
-
.find((t) => t.name === 'add_youtube_url')
|
|
48
|
-
.handler({
|
|
49
|
-
url: 'https://www.youtube.com/watch?v=iG9CE55wbtY',
|
|
50
|
-
}));
|
|
51
|
-
expect(addRes.id).toBeDefined();
|
|
52
|
-
expect(addRes.segments).toBeGreaterThan(50);
|
|
53
|
-
const hits = (await tools
|
|
54
|
-
.find((t) => t.name === 'search_youtube')
|
|
55
|
-
.handler({
|
|
56
|
-
query: 'creativity in schools',
|
|
57
|
-
}));
|
|
58
|
-
expect(hits.some((h) => h.id === addRes.id)).toBe(true);
|
|
59
|
-
}, 120_000);
|
|
60
|
-
});
|
|
61
|
-
//# sourceMappingURL=youtube.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"youtube.test.js","sourceRoot":"","sources":["../../../../../src/memory/modules/youtube/tests/youtube.test.ts"],"names":[],"mappings":"AAAA,mDAAmD;AACnD,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,MAAM,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,uBAAuB,EAAE,MAAM,cAAc,CAAC;AAGvD,MAAM,MAAM,GAAiB;IAC3B,OAAO,EAAE,WAAW;IACpB,UAAU,EAAE;QACV,QAAQ,EAAE,QAAiB;QAC3B,OAAO,EAAE,wBAAwB;QACjC,KAAK,EAAE,kBAAkB;QACzB,UAAU,EAAE,GAAG;KAChB;IACD,KAAK,EAAE,SAAS;IAChB,MAAM,EAAE,SAAS;IACjB,kBAAkB,EAAE;QAClB,OAAO,EAAE,KAAK;QACd,OAAO,EAAE,wBAAwB;QACjC,KAAK,EAAE,aAAa;QACpB,SAAS,EAAE,GAAG;KACf;IACD,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE;IAC5D,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE;IACvE,OAAO,EAAE,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE;IACvC,GAAG,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE;CACrC,CAAC;AAEF,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,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,YAAY,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC;QACf,eAAe,CAAC,MAAM,CAAC,CAAC;QACxB,KAAK,GAAG,IAAI,WAAW,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;IAC7D,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,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,KAAK,GAAG,uBAAuB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,CAAC,MAAM,KAAK;aACxB,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,CAAE;aAC1C,OAAO,CAAC;YACP,GAAG,EAAE,6CAA6C;SACnD,CAAC,CAAoD,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAE5C,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK;aACtB,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAE;aACzC,OAAO,CAAC;YACP,KAAK,EAAE,uBAAuB;SAC/B,CAAC,CAAyC,CAAC;QAC9C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1D,CAAC,EAAE,OAAO,CAAC,CAAC;AACd,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"apply.test.d.ts","sourceRoot":"","sources":["../../../src/sync/tests/apply.test.ts"],"names":[],"mappings":"AAgBA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAoH1C,iBAAS,WAAW,CAClB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC/B,MAAM,CA4BR;AAGD,OAAO,EAAE,WAAW,EAAE,CAAC"}
|