@andespindola/brainlink 0.1.0-beta.2 → 0.1.0-beta.20

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (55) hide show
  1. package/AGENTS.md +5 -5
  2. package/CHANGELOG.md +50 -2
  3. package/CONTRIBUTING.md +2 -2
  4. package/README.md +157 -20
  5. package/SECURITY.md +1 -1
  6. package/dist/application/add-note.js +62 -13
  7. package/dist/application/analyze-vault.js +95 -8
  8. package/dist/application/frontend/client-css.js +190 -99
  9. package/dist/application/frontend/client-html.js +57 -45
  10. package/dist/application/frontend/client-js.js +416 -85
  11. package/dist/application/get-graph-layout.js +22 -7
  12. package/dist/application/get-graph-node.js +12 -0
  13. package/dist/application/get-graph-summary.js +12 -0
  14. package/dist/application/get-graph.js +3 -3
  15. package/dist/application/index-vault.js +11 -4
  16. package/dist/application/list-agents.js +3 -3
  17. package/dist/application/list-links.js +5 -5
  18. package/dist/application/migrate-vault.js +91 -0
  19. package/dist/application/search-graph-node-ids.js +12 -0
  20. package/dist/application/search-knowledge.js +75 -5
  21. package/dist/application/server/routes.js +27 -1
  22. package/dist/benchmarks/large-vault.js +1 -1
  23. package/dist/cli/commands/agent-commands.js +412 -0
  24. package/dist/cli/commands/config-commands.js +167 -0
  25. package/dist/cli/commands/read-commands.js +25 -8
  26. package/dist/cli/commands/write-commands.js +173 -4
  27. package/dist/cli/main.js +4 -0
  28. package/dist/cli/runtime.js +5 -2
  29. package/dist/domain/context.js +53 -11
  30. package/dist/domain/embeddings.js +2 -1
  31. package/dist/domain/graph-layout.js +20 -14
  32. package/dist/domain/markdown.js +36 -4
  33. package/dist/domain/middle-out.js +18 -0
  34. package/dist/infrastructure/config.js +94 -8
  35. package/dist/infrastructure/file-index.js +294 -0
  36. package/dist/infrastructure/file-system-vault.js +15 -0
  37. package/dist/infrastructure/paths.js +9 -1
  38. package/dist/infrastructure/private-pack-codec.js +73 -0
  39. package/dist/infrastructure/search-packs.js +348 -0
  40. package/dist/infrastructure/session-state.js +172 -0
  41. package/dist/mcp/main.js +11 -3
  42. package/dist/mcp/server.js +17 -2
  43. package/dist/mcp/startup.js +35 -0
  44. package/dist/mcp/tools.js +571 -19
  45. package/docs/AGENT_USAGE.md +99 -15
  46. package/docs/ARCHITECTURE.md +37 -26
  47. package/docs/QUICKSTART.md +104 -0
  48. package/docs/RELEASE.md +3 -3
  49. package/package.json +1 -3
  50. package/dist/infrastructure/sqlite/document-writer.js +0 -51
  51. package/dist/infrastructure/sqlite/graph-reader.js +0 -120
  52. package/dist/infrastructure/sqlite/schema.js +0 -111
  53. package/dist/infrastructure/sqlite/search-reader.js +0 -156
  54. package/dist/infrastructure/sqlite/types.js +0 -1
  55. package/dist/infrastructure/sqlite-index.js +0 -25
@@ -1,11 +1,18 @@
1
1
  import { readFileSync } from 'node:fs';
2
+ import { mkdir, writeFile } from 'node:fs/promises';
3
+ import { dirname, relative, resolve } from 'node:path';
2
4
  import { addNote } from '../../application/add-note.js';
5
+ import { buildContextPackage } from '../../application/build-context.js';
3
6
  import { indexVault } from '../../application/index-vault.js';
7
+ import { migrateVaultContent, planVaultMigration, previewVaultMigration, shouldMigrateDefaultVault } from '../../application/migrate-vault.js';
4
8
  import { startServer } from '../../application/start-server.js';
5
9
  import { startVaultWatcher } from '../../application/watch-vault.js';
6
- import { doctorVault } from '../../application/analyze-vault.js';
10
+ import { doctorVault, getStats, validateVault } from '../../application/analyze-vault.js';
11
+ import { defaultBrainlinkConfig, sanitizeSearchMode } from '../../infrastructure/config.js';
7
12
  import { loadBrainlinkConfig } from '../../infrastructure/config.js';
8
13
  import { assertVaultAllowed, ensureVault } from '../../infrastructure/file-system-vault.js';
14
+ import { getBootstrapPolicy, getBootstrapSessionStatus, touchBootstrapSession } from '../../infrastructure/session-state.js';
15
+ import { installAgentIntegration } from './agent-commands.js';
9
16
  import { parsePositiveInteger, print, resolveOptions } from '../runtime.js';
10
17
  const resolveAddContent = (options) => {
11
18
  if (options.content != null && options.content.trim().length > 0) {
@@ -20,12 +27,77 @@ export const registerWriteCommands = (program) => {
20
27
  program
21
28
  .command('init')
22
29
  .argument('[vault]', 'vault directory')
30
+ .option('--migrate-from <vault>', 'copy existing vault content into the initialized vault')
31
+ .option('--no-migrate-existing', 'skip automatic migration from the default Brainlink vault into an empty custom vault')
23
32
  .option('--json', 'print machine-readable JSON')
24
33
  .description('initialize a Brainlink vault')
25
34
  .action(async (vault, options) => {
26
35
  const config = await loadBrainlinkConfig();
27
- const path = await ensureVault(assertVaultAllowed(vault ?? config.vault, config.allowedVaults));
28
- print(options.json, { path }, () => `Initialized Brainlink vault at ${path}`);
36
+ const targetVault = assertVaultAllowed(vault ?? config.vault, config.allowedVaults);
37
+ const path = await ensureVault(targetVault);
38
+ const explicitSource = options.migrateFrom ? assertVaultAllowed(options.migrateFrom, config.allowedVaults) : undefined;
39
+ const shouldAutoMigrate = explicitSource === undefined &&
40
+ options.migrateExisting !== false &&
41
+ (await shouldMigrateDefaultVault(defaultBrainlinkConfig.vault, targetVault));
42
+ const migration = explicitSource || shouldAutoMigrate ? await migrateVaultContent(explicitSource ?? defaultBrainlinkConfig.vault, targetVault) : undefined;
43
+ const index = migration && migration.copied + migration.conflicted > 0 ? await indexVault(targetVault) : undefined;
44
+ print(options.json, { path, ...(migration ? { migration } : {}), ...(index ? { index } : {}) }, () => {
45
+ const migrated = migration
46
+ ? ` Migrated ${migration.copied} files, preserved ${migration.conflicted} conflicts and kept ${migration.unchanged} unchanged files.`
47
+ : '';
48
+ return `Initialized Brainlink vault at ${path}.${migrated}`;
49
+ });
50
+ });
51
+ program
52
+ .command('migrate-vault')
53
+ .option('--from <vault>', 'source vault path')
54
+ .option('--to <vault>', 'target vault path')
55
+ .option('--dry-run', 'preview migration without writing files')
56
+ .option('--report <path>', 'write detailed per-file migration report to JSON file')
57
+ .option('--no-index', 'skip reindexing target vault after migration')
58
+ .option('--json', 'print machine-readable JSON')
59
+ .description('copy markdown memory from one vault to another with conflict preservation')
60
+ .action(async (options) => {
61
+ const config = await loadBrainlinkConfig();
62
+ const sourceVault = assertVaultAllowed(options.from ?? config.vault, config.allowedVaults);
63
+ const targetVault = assertVaultAllowed(options.to ?? defaultBrainlinkConfig.vault, config.allowedVaults);
64
+ const sourceRoot = await ensureVault(sourceVault);
65
+ const targetRoot = await ensureVault(targetVault);
66
+ const preview = await previewVaultMigration(sourceVault, targetVault);
67
+ const actions = await planVaultMigration(sourceRoot, targetRoot);
68
+ const reportEntries = actions.map((action) => ({
69
+ kind: action.kind,
70
+ sourcePath: action.sourcePath,
71
+ sourceRelativePath: relative(sourceRoot, action.sourcePath),
72
+ targetPath: action.targetPath,
73
+ targetRelativePath: relative(targetRoot, action.targetPath)
74
+ }));
75
+ const writeReport = async () => {
76
+ if (!options.report) {
77
+ return null;
78
+ }
79
+ const reportPath = resolve(options.report);
80
+ await mkdir(dirname(reportPath), { recursive: true });
81
+ await writeFile(reportPath, `${JSON.stringify({ source: sourceVault, target: targetVault, summary: preview, entries: reportEntries }, null, 2)}\n`, 'utf8');
82
+ return reportPath;
83
+ };
84
+ if (options.dryRun) {
85
+ const reportPath = await writeReport();
86
+ print(options.json, { dryRun: true, ...preview, entries: reportEntries, ...(reportPath ? { reportPath } : {}) }, () => `Dry run migration ${preview.source} -> ${preview.target}: copy=${preview.copied}, conflicts=${preview.conflicted}, unchanged=${preview.unchanged}${reportPath ? ` report=${reportPath}` : ''}`);
87
+ return;
88
+ }
89
+ const migration = await migrateVaultContent(sourceVault, targetVault);
90
+ const shouldIndex = options.index !== false && migration.copied + migration.conflicted > 0;
91
+ const index = shouldIndex ? await indexVault(targetVault) : undefined;
92
+ const reportPath = await writeReport();
93
+ print(options.json, { dryRun: false, ...migration, entries: reportEntries, ...(index ? { index } : {}), ...(reportPath ? { reportPath } : {}) }, () => {
94
+ const summary = `Migrated ${migration.copied} files, preserved ${migration.conflicted} conflicts and kept ${migration.unchanged} unchanged files.`;
95
+ const indexMessage = index
96
+ ? ` Indexed ${index.documentCount} documents, ${index.chunkCount} chunks and ${index.linkCount} links.`
97
+ : '';
98
+ const reportMessage = reportPath ? ` Report written to ${reportPath}.` : '';
99
+ return `${summary}${indexMessage}${reportMessage}`;
100
+ });
29
101
  });
30
102
  program
31
103
  .command('add')
@@ -66,7 +138,13 @@ export const registerWriteCommands = (program) => {
66
138
  .action(async (options) => {
67
139
  const resolved = await resolveOptions(options);
68
140
  const report = await doctorVault(resolved.vault);
69
- print(options.json, report, () => report.checks.map((check) => `${check.ok ? 'OK' : 'FAIL'} ${check.name}: ${check.message}`).join('\n'));
141
+ print(options.json, report, () => {
142
+ const checks = report.checks.map((check) => `${check.ok ? 'OK' : 'FAIL'} ${check.name}: ${check.message}`).join('\n');
143
+ const recommendations = report.recommendations && report.recommendations.length > 0
144
+ ? `\n\nRecommended next steps:\n${report.recommendations.map((step) => `- ${step}`).join('\n')}`
145
+ : '';
146
+ return `${checks}${recommendations}`;
147
+ });
70
148
  process.exitCode = report.ok ? 0 : 1;
71
149
  });
72
150
  program
@@ -117,4 +195,95 @@ export const registerWriteCommands = (program) => {
117
195
  });
118
196
  print(options.json, { url: server.url, watch: Boolean(options.watch), readonly: true }, () => `Brainlink graph server running at ${server.url}`);
119
197
  });
198
+ program
199
+ .command('quickstart')
200
+ .option('-v, --vault <vault>', 'vault directory')
201
+ .option('-a, --agent <agent>', 'agent memory namespace')
202
+ .option('--query <query>', 'optional task query to return immediate grounded context')
203
+ .option('--mode <mode>', 'search mode for context (fts|semantic|hybrid)')
204
+ .option('--limit <limit>', 'maximum context sections')
205
+ .option('--tokens <tokens>', 'maximum context token budget')
206
+ .option('--no-install-agent', 'skip agent MCP/plugin installation and upgrade automation')
207
+ .option('--mcp-only', 'when installing agent integration, only configure MCP section')
208
+ .option('--plugin-path <path>', 'custom source path for Brainlink plugin files')
209
+ .option('--allowed-vaults <paths>', 'comma separated vault allowlist to inject in MCP env')
210
+ .option('--brainlink-home <path>', 'BRAINLINK_HOME value to inject in MCP env')
211
+ .option('--json', 'print machine-readable JSON')
212
+ .description('run plug-and-play setup for vault, MCP integration and bootstrap readiness')
213
+ .action(async (options) => {
214
+ const resolved = await resolveOptions(options);
215
+ const limit = parsePositiveInteger(options.limit ?? String(resolved.defaults.defaultSearchLimit), resolved.defaults.defaultSearchLimit);
216
+ const tokens = parsePositiveInteger(options.tokens ?? String(resolved.defaults.defaultContextTokens), resolved.defaults.defaultContextTokens);
217
+ const mode = sanitizeSearchMode(options.mode, resolved.defaults.defaultSearchMode);
218
+ const index = await indexVault(resolved.vault);
219
+ const stats = await getStats(resolved.vault, resolved.agent);
220
+ const validation = await validateVault(resolved.vault, resolved.agent);
221
+ const doctor = await doctorVault(resolved.vault);
222
+ const session = await touchBootstrapSession(resolved.vault, resolved.agent);
223
+ const policy = await getBootstrapPolicy();
224
+ const bootstrapStatus = await getBootstrapSessionStatus(resolved.vault, resolved.agent);
225
+ const context = options.query
226
+ ? await buildContextPackage(resolved.vault, options.query, limit, tokens, resolved.agent, mode)
227
+ : null;
228
+ const agentIntegration = options.installAgent === false
229
+ ? null
230
+ : await installAgentIntegration({
231
+ mcpOnly: options.mcpOnly,
232
+ pluginPath: options.pluginPath,
233
+ allowedVaults: options.allowedVaults,
234
+ brainlinkHome: options.brainlinkHome,
235
+ selfTest: true
236
+ });
237
+ const nextActions = stats.documentCount === 0
238
+ ? [
239
+ {
240
+ priority: 'required',
241
+ command: `blink add "Architecture" --vault "${resolved.vault}" --content "Durable memory with [[Links]] and #tags."`,
242
+ reason: 'Seed your vault with at least one durable Markdown note.'
243
+ },
244
+ {
245
+ priority: 'required',
246
+ command: `blink index --vault "${resolved.vault}"`,
247
+ reason: 'Rebuild index after adding notes so retrieval can find new memory.'
248
+ }
249
+ ]
250
+ : options.query
251
+ ? [
252
+ {
253
+ priority: 'recommended',
254
+ command: `blink add "Task Update" --vault "${resolved.vault}" --agent "${resolved.agent ?? 'shared'}" --content "<durable memory>"`,
255
+ reason: 'Persist important findings as Markdown notes after using the returned context.'
256
+ }
257
+ ]
258
+ : [
259
+ {
260
+ priority: 'recommended',
261
+ command: `blink context "<task>" --vault "${resolved.vault}" --agent "${resolved.agent ?? 'shared'}" --mode ${mode}`,
262
+ reason: 'Retrieve grounded context for each task before responding.'
263
+ }
264
+ ];
265
+ print(options.json, {
266
+ vault: resolved.vault,
267
+ agent: resolved.agent ?? 'shared',
268
+ mode,
269
+ index,
270
+ stats,
271
+ validation,
272
+ doctor,
273
+ policy,
274
+ bootstrapStatus,
275
+ session,
276
+ context,
277
+ agentIntegration,
278
+ nextActions
279
+ }, () => [
280
+ `quickstart vault=${resolved.vault}`,
281
+ `agent=${resolved.agent ?? 'shared'}`,
282
+ `documents=${stats.documentCount}`,
283
+ `links=${stats.linkCount}`,
284
+ `bootstrapReady=${bootstrapStatus.ready}`,
285
+ ...(agentIntegration?.selfTest ? [`agentIntegrationSelfTest=${agentIntegration.selfTest.ok}`] : []),
286
+ ...(nextActions.length > 0 ? ['Next actions:', ...nextActions.map((step) => `- ${step.command}`)] : [])
287
+ ].join('\n'));
288
+ });
120
289
  };
package/dist/cli/main.js CHANGED
@@ -3,6 +3,8 @@ import { Command } from 'commander';
3
3
  import { readFileSync } from 'node:fs';
4
4
  import { basename, dirname, join } from 'node:path';
5
5
  import { fileURLToPath } from 'node:url';
6
+ import { registerAgentCommands } from './commands/agent-commands.js';
7
+ import { registerConfigCommands } from './commands/config-commands.js';
6
8
  import { registerReadCommands } from './commands/read-commands.js';
7
9
  import { registerWriteCommands } from './commands/write-commands.js';
8
10
  const readPackageVersion = () => {
@@ -21,6 +23,8 @@ program
21
23
  .version(readPackageVersion());
22
24
  registerWriteCommands(program);
23
25
  registerReadCommands(program);
26
+ registerConfigCommands(program);
27
+ registerAgentCommands(program);
24
28
  program.parseAsync().catch((error) => {
25
29
  const message = error instanceof Error ? error.message : String(error);
26
30
  console.error(message);
@@ -1,4 +1,4 @@
1
- import { loadBrainlinkConfig } from '../infrastructure/config.js';
1
+ import { loadBrainlinkConfig, resolveAgentRuntimeDefaults } from '../infrastructure/config.js';
2
2
  import { assertVaultAllowed } from '../infrastructure/file-system-vault.js';
3
3
  export const parsePositiveInteger = (value, fallback) => {
4
4
  const parsed = Number.parseInt(value, 10);
@@ -8,10 +8,13 @@ export const resolveOptions = async (options) => {
8
8
  const config = await loadBrainlinkConfig();
9
9
  const vault = options.vault ?? config.vault;
10
10
  const allowedVault = assertVaultAllowed(vault, config.allowedVaults);
11
+ const agent = options.agent ?? config.defaultAgent;
12
+ const defaults = resolveAgentRuntimeDefaults(config, agent);
11
13
  return {
12
14
  config,
13
15
  vault: allowedVault,
14
- agent: options.agent ?? config.defaultAgent
16
+ agent,
17
+ defaults
15
18
  };
16
19
  };
17
20
  export const print = (json, value, human) => {
@@ -1,13 +1,50 @@
1
+ import { middleOutIndices } from './middle-out.js';
2
+ const maxSectionsPerDocument = 3;
3
+ const byScore = (left, right) => right.score - left.score || left.title.localeCompare(right.title);
4
+ const byOrdinal = (left, right) => (left.chunkOrdinal ?? Number.MAX_SAFE_INTEGER) - (right.chunkOrdinal ?? Number.MAX_SAFE_INTEGER);
5
+ const middleOutDocumentResults = (results) => {
6
+ if (results.length <= 1) {
7
+ return results;
8
+ }
9
+ const sortedByOrdinal = [...results].sort(byOrdinal);
10
+ const pivotChunkId = [...results].sort(byScore)[0]?.chunkId;
11
+ const pivotIndex = sortedByOrdinal.findIndex((result) => result.chunkId === pivotChunkId);
12
+ if (pivotIndex < 0) {
13
+ return [...results].sort(byScore);
14
+ }
15
+ return middleOutIndices(sortedByOrdinal.length, pivotIndex).map((index) => sortedByOrdinal[index]);
16
+ };
1
17
  export const selectContextSections = (results, maxTokens) => {
2
- const selected = results.reduce((state, result) => {
3
- const tokenCost = Math.ceil(result.content.length / 4);
4
- if (state.usedTokens + tokenCost > maxTokens || state.seenDocuments.has(result.documentId)) {
5
- return state;
18
+ const grouped = results.reduce((state, result) => {
19
+ const current = state.get(result.documentId) ?? [];
20
+ state.set(result.documentId, [...current, result]);
21
+ return state;
22
+ }, new Map());
23
+ const documentOrder = Array.from(results.reduce((state, result) => {
24
+ if (!state.has(result.documentId)) {
25
+ state.set(result.documentId, result.score);
6
26
  }
7
- return {
8
- usedTokens: state.usedTokens + tokenCost,
9
- sections: [
10
- ...state.sections,
27
+ return state;
28
+ }, new Map()).entries())
29
+ .sort((left, right) => right[1] - left[1] || left[0].localeCompare(right[0]))
30
+ .map(([documentId]) => documentId);
31
+ const selected = documentOrder.reduce((state, documentId) => {
32
+ const ordered = middleOutDocumentResults(grouped.get(documentId) ?? []);
33
+ let usedTokens = state.usedTokens;
34
+ let sections = state.sections;
35
+ let seenChunks = state.seenChunks;
36
+ for (let index = 0; index < ordered.length && index < maxSectionsPerDocument; index += 1) {
37
+ const result = ordered[index];
38
+ if (seenChunks.has(result.chunkId)) {
39
+ continue;
40
+ }
41
+ const tokenCost = Math.ceil(result.content.length / 4);
42
+ if (usedTokens + tokenCost > maxTokens) {
43
+ break;
44
+ }
45
+ usedTokens += tokenCost;
46
+ sections = [
47
+ ...sections,
11
48
  {
12
49
  title: result.title,
13
50
  path: result.path,
@@ -16,13 +53,18 @@ export const selectContextSections = (results, maxTokens) => {
16
53
  searchMode: result.searchMode,
17
54
  tags: result.tags
18
55
  }
19
- ],
20
- seenDocuments: new Set([...state.seenDocuments, result.documentId])
56
+ ];
57
+ seenChunks = new Set([...seenChunks, result.chunkId]);
58
+ }
59
+ return {
60
+ usedTokens,
61
+ sections,
62
+ seenChunks
21
63
  };
22
64
  }, {
23
65
  usedTokens: 0,
24
66
  sections: [],
25
- seenDocuments: new Set()
67
+ seenChunks: new Set()
26
68
  });
27
69
  return selected.sections;
28
70
  };
@@ -58,7 +58,8 @@ const tokenize = (input) => input
58
58
  .match(tokenPattern)
59
59
  ?.map(normalizeToken)
60
60
  .filter((token) => token.length > 1 && !stopWords.has(token)) ?? [];
61
- const expandTokens = (tokens) => tokens.flatMap((token) => [token, ...(aliases[token] ?? [])]);
61
+ const getAliasesForToken = (token) => Object.hasOwn(aliases, token) ? aliases[token] ?? [] : [];
62
+ const expandTokens = (tokens) => tokens.flatMap((token) => [token, ...getAliasesForToken(token)]);
62
63
  const hash = (value) => Array.from(value).reduce((state, char) => Math.imul(state ^ char.charCodeAt(0), 16777619), 2166136261) >>> 0;
63
64
  const featureHash = (feature) => {
64
65
  const value = hash(feature);
@@ -45,20 +45,17 @@ const countDegrees = (edges) => edges.reduce((degrees, edge) => {
45
45
  ? incrementDegreeBy(incrementDegreeBy(degrees, edge.source, weight), edge.target, weight)
46
46
  : incrementDegreeBy(degrees, edge.source, weight);
47
47
  }, new Map());
48
- const uniqueIds = (ids) => Array.from(new Set(ids));
49
48
  const createAdjacency = (nodes, edges) => {
50
49
  const nodeIds = new Set(nodes.map((node) => node.id));
51
- const emptyAdjacency = new Map(nodes.map((node) => [node.id, []]));
52
- return edges.reduce((adjacency, edge) => {
50
+ const adjacency = new Map(nodes.map((node) => [node.id, new Set()]));
51
+ edges.forEach((edge) => {
53
52
  if (!edge.target || !nodeIds.has(edge.source) || !nodeIds.has(edge.target)) {
54
- return adjacency;
53
+ return;
55
54
  }
56
- return new Map([
57
- ...adjacency,
58
- [edge.source, uniqueIds([...(adjacency.get(edge.source) ?? []), edge.target])],
59
- [edge.target, uniqueIds([...(adjacency.get(edge.target) ?? []), edge.source])]
60
- ]);
61
- }, emptyAdjacency);
55
+ adjacency.get(edge.source)?.add(edge.target);
56
+ adjacency.get(edge.target)?.add(edge.source);
57
+ });
58
+ return new Map(Array.from(adjacency.entries(), ([id, neighbors]) => [id, Array.from(neighbors)]));
62
59
  };
63
60
  const byTitle = (left, right) => left.title.localeCompare(right.title);
64
61
  const byDegreeThenTitle = (degrees) => (left, right) => {
@@ -117,10 +114,19 @@ const assignSegments = (nodes, edges, degrees) => {
117
114
  }
118
115
  return new Map(nodes.map((node) => [node.id, assignments.get(node.id) ?? groupLabel(groupKey(node))]));
119
116
  };
120
- const groupNodesBySegment = (nodes, segments) => nodes.reduce((groups, node) => {
121
- const segment = segments.get(node.id) ?? groupLabel(groupKey(node));
122
- return new Map([...groups, [segment, [...(groups.get(segment) ?? []), node]]]);
123
- }, new Map());
117
+ const groupNodesBySegment = (nodes, segments) => {
118
+ const groups = new Map();
119
+ nodes.forEach((node) => {
120
+ const segment = segments.get(node.id) ?? groupLabel(groupKey(node));
121
+ const bucket = groups.get(segment);
122
+ if (bucket) {
123
+ bucket.push(node);
124
+ return;
125
+ }
126
+ groups.set(segment, [node]);
127
+ });
128
+ return new Map(groups);
129
+ };
124
130
  const segmentAngle = (segment, index, count) => segmentAngles[segment] ?? (Math.PI * 2 * index) / Math.max(count, 1) - Math.PI / 2;
125
131
  const createSegmentNodes = (segments, degrees, segmentCount) => ([segment, nodes], segmentIndex) => {
126
132
  const sortedNodes = [...nodes].sort(byDegreeThenTitle(degrees));
@@ -77,11 +77,13 @@ export const extractWikiLinkReferences = (content) => visibleMarkdownLines(conte
77
77
  }));
78
78
  });
79
79
  const priorityFromWeight = (weight) => weight >= 8 ? 'critical' : weight >= 4 ? 'high' : 'normal';
80
+ const normalizeAccumulatedWeight = (weight) => Math.max(1, Math.min(12, weight));
80
81
  export const extractWikiLinkWeights = (content) => {
81
82
  const weights = extractWikiLinkReferences(content).reduce((state, reference) => {
82
83
  const titleKey = reference.title.toLowerCase();
83
84
  const current = state.get(titleKey);
84
- const weight = (current?.weight ?? 0) + reference.weight;
85
+ const rawWeight = (current?.weight ?? 0) + reference.weight;
86
+ const weight = normalizeAccumulatedWeight(rawWeight);
85
87
  const explicitPriority = reference.priority
86
88
  ? maxPriority(current?.priority ?? reference.priority, reference.priority)
87
89
  : current?.priority;
@@ -116,10 +118,38 @@ const normalizeChunkContent = (content) => content
116
118
  .join('\n')
117
119
  .replace(/\n{3,}/g, '\n\n')
118
120
  .trim();
121
+ const splitLongParagraph = (paragraph, maxCharacters) => {
122
+ if (paragraph.length <= maxCharacters) {
123
+ return [paragraph];
124
+ }
125
+ const sentences = paragraph
126
+ .split(/(?<=[.!?])\s+/)
127
+ .map((sentence) => sentence.trim())
128
+ .filter(Boolean);
129
+ if (sentences.length <= 1) {
130
+ const chunks = [];
131
+ for (let index = 0; index < paragraph.length; index += maxCharacters) {
132
+ chunks.push(paragraph.slice(index, index + maxCharacters).trim());
133
+ }
134
+ return chunks.filter(Boolean);
135
+ }
136
+ return sentences.reduce((state, sentence) => {
137
+ const last = state.at(-1);
138
+ if (!last) {
139
+ return [sentence];
140
+ }
141
+ const merged = `${last} ${sentence}`;
142
+ if (merged.length <= maxCharacters) {
143
+ return [...state.slice(0, -1), merged];
144
+ }
145
+ return [...state, sentence];
146
+ }, []);
147
+ };
119
148
  export const splitIntoChunks = (documentId, content, maxCharacters = 1200) => {
120
149
  const paragraphs = normalizeChunkContent(stripFrontmatter(content))
121
150
  .split(/\n{2,}/)
122
- .filter(Boolean);
151
+ .filter(Boolean)
152
+ .flatMap((paragraph) => splitLongParagraph(paragraph, maxCharacters));
123
153
  const chunks = paragraphs.reduce((state, paragraph) => {
124
154
  const lastChunk = state.at(-1);
125
155
  if (!lastChunk) {
@@ -162,13 +192,15 @@ export const parseMarkdownDocument = (input) => {
162
192
  export const createIndexedDocument = (document, titleToDocumentId, maxChunkCharacters = 1200) => {
163
193
  const chunks = splitIntoChunks(document.id, document.content, maxChunkCharacters);
164
194
  const linkWeights = new Map(extractWikiLinkWeights(document.content).map((link) => [link.title.toLowerCase(), link]));
165
- const links = document.links.map((toTitle) => ({
195
+ const links = document.links
196
+ .map((toTitle) => ({
166
197
  fromDocumentId: document.id,
167
198
  toTitle,
168
199
  toDocumentId: titleToDocumentId.get(toTitle.toLowerCase()) ?? null,
169
200
  weight: linkWeights.get(toTitle.toLowerCase())?.weight ?? 1,
170
201
  priority: linkWeights.get(toTitle.toLowerCase())?.priority ?? 'normal'
171
- }));
202
+ }))
203
+ .filter((link) => link.toDocumentId !== document.id);
172
204
  return {
173
205
  document,
174
206
  chunks,
@@ -0,0 +1,18 @@
1
+ export const middleOutIndices = (size, pivotIndex) => {
2
+ if (!Number.isFinite(size) || size <= 0) {
3
+ return [];
4
+ }
5
+ const clampedPivot = Math.max(0, Math.min(Math.floor(pivotIndex), size - 1));
6
+ const indices = [clampedPivot];
7
+ for (let offset = 1; indices.length < size; offset += 1) {
8
+ const left = clampedPivot - offset;
9
+ const right = clampedPivot + offset;
10
+ if (left >= 0) {
11
+ indices.push(left);
12
+ }
13
+ if (right < size) {
14
+ indices.push(right);
15
+ }
16
+ }
17
+ return indices;
18
+ };
@@ -1,7 +1,8 @@
1
- import { readFile } from 'node:fs/promises';
2
- import { resolve } from 'node:path';
1
+ import { mkdir, readFile, writeFile } from 'node:fs/promises';
2
+ import { dirname, join, resolve } from 'node:path';
3
+ import { homedir } from 'node:os';
3
4
  import { sanitizeAgentId } from '../domain/agents.js';
4
- import { getDefaultVaultPath } from './paths.js';
5
+ import { getBrainlinkHomePath, getDefaultVaultPath } from './paths.js';
5
6
  export const defaultBrainlinkConfig = {
6
7
  vault: getDefaultVaultPath(),
7
8
  host: '127.0.0.1',
@@ -13,15 +14,61 @@ export const defaultBrainlinkConfig = {
13
14
  defaultContextTokens: 2000,
14
15
  embeddingProvider: 'local',
15
16
  defaultSearchMode: 'hybrid',
16
- chunkSize: 1200
17
+ chunkSize: 1200,
18
+ agentProfiles: {}
17
19
  };
18
20
  const configFilenames = ['brainlink.config.json', '.brainlink.json'];
21
+ const localConfigFilename = 'brainlink.config.json';
22
+ const globalConfigFilename = 'brainlink.config.json';
23
+ const globalConfigDirectoryMode = 0o700;
24
+ const globalConfigFileMode = 0o600;
25
+ const safeCwd = () => {
26
+ try {
27
+ return process.cwd();
28
+ }
29
+ catch {
30
+ return homedir();
31
+ }
32
+ };
19
33
  const isRecord = (value) => typeof value === 'object' && value !== null && !Array.isArray(value);
20
34
  const embeddingProviders = new Set(['none', 'local']);
21
35
  const searchModes = new Set(['fts', 'semantic', 'hybrid']);
22
36
  const sanitizeEmbeddingProvider = (value) => typeof value === 'string' && embeddingProviders.has(value) ? value : defaultBrainlinkConfig.embeddingProvider;
23
37
  export const sanitizeSearchMode = (value, fallback = defaultBrainlinkConfig.defaultSearchMode) => typeof value === 'string' && searchModes.has(value) ? value : fallback;
24
38
  const sanitizeAllowedVaults = (value) => Array.isArray(value) ? value.filter((item) => typeof item === 'string' && item.trim().length > 0) : [];
39
+ const sanitizePositiveNumber = (value) => typeof value === 'number' && Number.isFinite(value) && value > 0 ? value : undefined;
40
+ const sanitizeAgentProfile = (value) => {
41
+ if (!isRecord(value)) {
42
+ return null;
43
+ }
44
+ const defaultSearchLimit = sanitizePositiveNumber(value.defaultSearchLimit);
45
+ const defaultContextTokens = sanitizePositiveNumber(value.defaultContextTokens);
46
+ const defaultSearchMode = typeof value.defaultSearchMode === 'string' && searchModes.has(value.defaultSearchMode)
47
+ ? value.defaultSearchMode
48
+ : undefined;
49
+ const profile = {
50
+ ...(defaultSearchLimit ? { defaultSearchLimit } : {}),
51
+ ...(defaultContextTokens ? { defaultContextTokens } : {}),
52
+ ...(defaultSearchMode ? { defaultSearchMode } : {})
53
+ };
54
+ return Object.keys(profile).length > 0 ? profile : null;
55
+ };
56
+ const sanitizeAgentProfiles = (value) => {
57
+ if (!isRecord(value)) {
58
+ return {};
59
+ }
60
+ return Object.entries(value).reduce((state, [key, profile]) => {
61
+ const normalizedKey = key === '*' ? '*' : sanitizeAgentId(key);
62
+ const sanitizedProfile = sanitizeAgentProfile(profile);
63
+ if (!sanitizedProfile || normalizedKey.length === 0) {
64
+ return state;
65
+ }
66
+ return {
67
+ ...state,
68
+ [normalizedKey]: sanitizedProfile
69
+ };
70
+ }, {});
71
+ };
25
72
  const readAllowedVaultsFromEnv = () => (process.env.BRAINLINK_ALLOWED_VAULTS ?? '')
26
73
  .split(',')
27
74
  .map((value) => value.trim())
@@ -39,6 +86,34 @@ const readJsonConfig = async (path) => {
39
86
  throw error;
40
87
  }
41
88
  };
89
+ export const getGlobalConfigPath = () => join(getBrainlinkHomePath(), globalConfigFilename);
90
+ export const getLocalConfigPath = (cwd = safeCwd()) => resolve(cwd, localConfigFilename);
91
+ export const resolveConfigPath = (scope, cwd = safeCwd()) => scope === 'global' ? getGlobalConfigPath() : getLocalConfigPath(cwd);
92
+ export const loadRawConfig = async (scope, cwd = safeCwd()) => readJsonConfig(resolveConfigPath(scope, cwd));
93
+ export const loadLegacyLocalRawConfig = async (cwd = safeCwd()) => readJsonConfig(resolve(cwd, '.brainlink.json'));
94
+ export const detectVaultConfigSource = async (cwd = safeCwd()) => {
95
+ const [globalConfig, localConfig, legacyLocalConfig] = await Promise.all([
96
+ loadRawConfig('global', cwd),
97
+ loadRawConfig('local', cwd),
98
+ loadLegacyLocalRawConfig(cwd)
99
+ ]);
100
+ if (typeof legacyLocalConfig.vault === 'string' && legacyLocalConfig.vault.trim().length > 0) {
101
+ return 'local-legacy';
102
+ }
103
+ if (typeof localConfig.vault === 'string' && localConfig.vault.trim().length > 0) {
104
+ return 'local';
105
+ }
106
+ if (typeof globalConfig.vault === 'string' && globalConfig.vault.trim().length > 0) {
107
+ return 'global';
108
+ }
109
+ return 'default';
110
+ };
111
+ export const writeRawConfig = async (scope, value, cwd = safeCwd()) => {
112
+ const path = resolveConfigPath(scope, cwd);
113
+ await mkdir(dirname(path), { recursive: true, mode: globalConfigDirectoryMode });
114
+ await writeFile(path, `${JSON.stringify(value, null, 2)}\n`, { encoding: 'utf8', mode: globalConfigFileMode });
115
+ return path;
116
+ };
42
117
  const sanitizeConfig = (value) => ({
43
118
  ...defaultBrainlinkConfig,
44
119
  ...value,
@@ -56,11 +131,22 @@ const sanitizeConfig = (value) => ({
56
131
  allowedVaults: [...sanitizeAllowedVaults(value.allowedVaults), ...readAllowedVaultsFromEnv()],
57
132
  chunkSize: typeof value.chunkSize === 'number' && value.chunkSize > 0 ? value.chunkSize : defaultBrainlinkConfig.chunkSize,
58
133
  embeddingProvider: sanitizeEmbeddingProvider(value.embeddingProvider),
59
- defaultSearchMode: sanitizeSearchMode(value.defaultSearchMode)
134
+ defaultSearchMode: sanitizeSearchMode(value.defaultSearchMode),
135
+ agentProfiles: sanitizeAgentProfiles(value.agentProfiles)
60
136
  });
61
- export const loadBrainlinkConfig = async (cwd = process.cwd()) => {
62
- const configs = await Promise.all(configFilenames.map((filename) => readJsonConfig(resolve(cwd, filename))));
63
- const merged = configs.reduce((state, config) => ({
137
+ export const resolveAgentRuntimeDefaults = (config, agent) => {
138
+ const normalizedAgent = agent?.trim().length ? sanitizeAgentId(agent) : undefined;
139
+ const profile = (normalizedAgent ? config.agentProfiles[normalizedAgent] : undefined) ?? config.agentProfiles['*'];
140
+ return {
141
+ defaultSearchLimit: profile?.defaultSearchLimit ?? config.defaultSearchLimit,
142
+ defaultContextTokens: profile?.defaultContextTokens ?? config.defaultContextTokens,
143
+ defaultSearchMode: profile?.defaultSearchMode ?? config.defaultSearchMode
144
+ };
145
+ };
146
+ export const loadBrainlinkConfig = async (cwd = safeCwd()) => {
147
+ const globalConfig = await readJsonConfig(getGlobalConfigPath());
148
+ const localConfigs = await Promise.all(configFilenames.map((filename) => readJsonConfig(resolve(cwd, filename))));
149
+ const merged = [globalConfig, ...localConfigs].reduce((state, config) => ({
64
150
  ...state,
65
151
  ...config
66
152
  }), {});