code-graph-context 2.6.1 → 2.7.0

package/dist/mcp/tools/swarm-post-task.tool.js

@@ -6,7 +6,7 @@ import { z } from 'zod';
 import { Neo4jService } from '../../storage/neo4j/neo4j.service.js';
 import { TOOL_NAMES, TOOL_METADATA } from '../constants.js';
 import { createErrorResponse, createSuccessResponse, resolveProjectIdOrError, debugLog } from '../utils.js';
-import { TASK_PRIORITIES, TASK_TYPES, generateTaskId, } from './swarm-constants.js';
+import { TASK_PRIORITIES, TASK_TYPES, generateTaskId } from './swarm-constants.js';
 /**
  * Query to get node IDs for a file path (fallback when no nodeIds provided)
  * Uses ENDS WITH for flexible matching (handles absolute vs relative paths)
@@ -135,10 +135,7 @@ export const createSwarmPostTaskTool = (server) => {
                 .default([])
                 .describe('Task IDs that must be completed before this task can start'),
             createdBy: z.string().describe('Agent ID or identifier of who created this task'),
-            metadata: z
-                .record(z.unknown())
-                .optional()
-                .describe('Additional metadata (context, acceptance criteria, etc.)'),
+            metadata: z.record(z.unknown()).optional().describe('Additional metadata (context, acceptance criteria, etc.)'),
         },
     }, async ({ projectId, swarmId, title, description, type = 'implement', priority = 'normal', targetNodeIds = [], targetFilePaths = [], dependencies = [], createdBy, metadata, }) => {
         const neo4jService = new Neo4jService();
@@ -162,10 +159,7 @@ export const createSwarmPostTaskTool = (server) => {
                         projectId: resolvedProjectId,
                     });
                     if (fileNodes.length > 0) {
-                        resolvedNodeIds = [
-                            ...resolvedNodeIds,
-                            ...fileNodes.map((n) => n.id).filter(Boolean),
-                        ];
+                        resolvedNodeIds = [...resolvedNodeIds, ...fileNodes.map((n) => n.id).filter(Boolean)];
                     }
                 }
             }
@@ -198,9 +192,7 @@ export const createSwarmPostTaskTool = (server) => {
                 });
                 if (depCheck.length > 0) {
                     dependencyStatus = {
-                        totalDeps: typeof depCheck[0].totalDeps === 'object'
-                            ? depCheck[0].totalDeps.toNumber()
-                            : depCheck[0].totalDeps,
+                        totalDeps: typeof depCheck[0].totalDeps === 'object' ? depCheck[0].totalDeps.toNumber() : depCheck[0].totalDeps,
                         incompleteDeps: typeof depCheck[0].incompleteDeps === 'object'
                             ? depCheck[0].incompleteDeps.toNumber()
                             : depCheck[0].incompleteDeps,
@@ -232,9 +224,7 @@ export const createSwarmPostTaskTool = (server) => {
                     targetFilePaths: task.targetFilePaths,
                     dependencies: task.dependencies,
                     createdBy: task.createdBy,
-                    createdAt: typeof task.createdAt === 'object'
-                        ? task.createdAt.toNumber()
-                        : task.createdAt,
+                    createdAt: typeof task.createdAt === 'object' ? task.createdAt.toNumber() : task.createdAt,
                 },
                 dependencyStatus: {
                     isBlocked,

package/dist/mcp/tools/swarm-sense.tool.js

@@ -21,6 +21,7 @@ const SENSE_PHEROMONES_QUERY = `
     AND ($agentIds IS NULL OR size($agentIds) = 0 OR p.agentId IN $agentIds)
     AND ($swarmId IS NULL OR p.swarmId = $swarmId)
     AND ($excludeAgentId IS NULL OR p.agentId <> $excludeAgentId)
+    AND ($sessionId IS NULL OR p.sessionId = $sessionId)
 
   // Calculate current intensity with exponential decay
   WITH p,
@@ -49,6 +50,7 @@ const SENSE_PHEROMONES_QUERY = `
     p.timestamp AS timestamp,
     p.data AS data,
     p.halfLife AS halfLifeMs,
+    p.sessionId AS sessionId,
     CASE WHEN target IS NOT NULL THEN labels(target)[0] ELSE null END AS targetType,
     CASE WHEN target IS NOT NULL THEN target.name ELSE null END AS targetName,
     CASE WHEN target IS NOT NULL THEN target.filePath ELSE null END AS targetFilePath
@@ -108,6 +110,10 @@ export const createSwarmSenseTool = (server) => {
                 .string()
                 .optional()
                 .describe('Exclude pheromones from this agent ID (useful for seeing what OTHER agents are doing)'),
+            sessionId: z
+                .string()
+                .optional()
+                .describe('Filter pheromones by session ID. Use to recover context after compaction or session restart.'),
             minIntensity: z
                 .number()
                 .min(0)
@@ -130,7 +136,7 @@ export const createSwarmSenseTool = (server) => {
                 .default(false)
                 .describe('Run cleanup of fully decayed pheromones (intensity < 0.01)'),
         },
-    }, async ({ projectId, types, nodeIds, agentIds, swarmId, excludeAgentId, minIntensity = 0.3, limit = 50, includeStats = false, cleanup = false, }) => {
+    }, async ({ projectId, types, nodeIds, agentIds, swarmId, excludeAgentId, sessionId, minIntensity = 0.3, limit = 50, includeStats = false, cleanup = false, }) => {
         const neo4jService = new Neo4jService();
         // Resolve project ID
         const projectResult = await resolveProjectIdOrError(projectId, neo4jService);
@@ -162,6 +168,7 @@ export const createSwarmSenseTool = (server) => {
                 agentIds: agentIds ?? null,
                 swarmId: swarmId ?? null,
                 excludeAgentId: excludeAgentId ?? null,
+                sessionId: sessionId ?? null,
                 minIntensity,
                 limit: Math.floor(limit),
             });
@@ -177,6 +184,7 @@ export const createSwarmSenseTool = (server) => {
                     originalIntensity: p.originalIntensity,
                     agentId: p.agentId,
                     swarmId: p.swarmId,
+                    sessionId: p.sessionId ?? null,
                     timestamp: ts,
                     age: ts ? `${Math.round((Date.now() - ts) / 1000)}s ago` : null,
                     data: p.data ? JSON.parse(p.data) : null,

package/dist/storage/neo4j/neo4j.service.js

@@ -138,7 +138,15 @@ export const QUERIES = {
   `,
     CREATE_EMBEDDED_VECTOR_INDEX: `
   CREATE VECTOR INDEX embedded_nodes_idx IF NOT EXISTS
-  FOR (n:Embedded) ON (n.embedding) 
+  FOR (n:Embedded) ON (n.embedding)
+  OPTIONS {indexConfig: {
+    \`vector.dimensions\`: 3072,
+    \`vector.similarity_function\`: 'cosine'
+  }}
+`,
+    CREATE_SESSION_NOTES_VECTOR_INDEX: `
+  CREATE VECTOR INDEX session_notes_idx IF NOT EXISTS
+  FOR (n:SessionNote) ON (n.embedding)
   OPTIONS {indexConfig: {
     \`vector.dimensions\`: 3072,
     \`vector.similarity_function\`: 'cosine'

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-graph-context",
-  "version": "2.6.1",
+  "version": "2.7.0",
   "description": "MCP server that builds code graphs to provide rich context to LLMs",
   "type": "module",
   "homepage": "https://github.com/drewdrewH/code-graph-context#readme",