@andespindola/brainlink 0.1.0-beta.155 → 0.1.0-beta.156

package/dist/application/frontend/client-js.js

@@ -1,5 +1,5 @@
 export const createClientJs = () => `const canvas = document.getElementById('graph')
-const ctx2dFallback = canvas.getContext('2d')
+let ctx2dFallback = null
 const byId = (id) => document.getElementById(id)
 const elements = {
   search: byId('search'),
@@ -476,7 +476,11 @@ const setFocusedNodeIds = (ids) => {
 }
 
 const drawFallback = () => {
-  if (state.rendererMode !== 'fallback' || !ctx2dFallback) {
+  if (state.rendererMode !== 'fallback') {
+    return
+  }
+  ctx2dFallback = ctx2dFallback ?? canvas.getContext('2d')
+  if (!ctx2dFallback) {
     return
   }
   const width = state.viewport.width

package/dist/application/get-graph-contexts.js

@@ -1,19 +1,33 @@
 import { getGraphLayout } from './get-graph-layout.js';
 export const getGraphContexts = async (vaultPath, agentId) => {
     const { layout } = await getGraphLayout(vaultPath, { agentId });
-    const nodeIdsByContext = layout.nodes.reduce((contexts, node) => {
+    const nodeIdsByContext = new Map();
+    const contextByNodeId = new Map();
+    layout.nodes.forEach((node) => {
         const title = node.segment || node.group || 'root';
-        const nodeIds = contexts.get(title) ?? new Set();
+        const nodeIds = nodeIdsByContext.get(title) ?? new Set();
         nodeIds.add(node.id);
-        contexts.set(title, nodeIds);
-        return contexts;
-    }, new Map());
+        nodeIdsByContext.set(title, nodeIds);
+        contextByNodeId.set(node.id, title);
+    });
+    const edgeCountByContext = new Map();
+    layout.edges.forEach((edge) => {
+        if (!edge.target) {
+            return;
+        }
+        const sourceContext = contextByNodeId.get(edge.source);
+        const targetContext = contextByNodeId.get(edge.target);
+        if (!sourceContext || sourceContext !== targetContext) {
+            return;
+        }
+        edgeCountByContext.set(sourceContext, (edgeCountByContext.get(sourceContext) ?? 0) + 1);
+    });
     return Array.from(nodeIdsByContext.entries())
         .map(([title, nodeIds]) => ({
         id: title,
         title,
         nodeCount: nodeIds.size,
-        edgeCount: layout.edges.filter((edge) => nodeIds.has(edge.source) && Boolean(edge.target && nodeIds.has(edge.target))).length
+        edgeCount: edgeCountByContext.get(title) ?? 0
     }))
         .sort((left, right) => right.nodeCount - left.nodeCount || left.title.localeCompare(right.title));
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@andespindola/brainlink",
-  "version": "0.1.0-beta.155",
+  "version": "0.1.0-beta.156",
   "description": "Local-first knowledge memory for agents with Markdown, backlinks, indexing and context retrieval.",
   "type": "module",
   "license": "MIT",