feat: port holistic node refinement contract

tests/unit/tools/queryEdge.test.ts

@@ -0,0 +1,92 @@
+import { describe, expect, it, vi } from 'vitest';
+
+vi.mock('@/services/database/edges', () => ({
+  edgeService: {
+    getEdgeById: vi.fn(),
+    getNodeConnections: vi.fn(),
+    getEdges: vi.fn(),
+  },
+}));
+
+import { edgeService } from '@/services/database/edges';
+import { queryEdgeTool } from '@/tools/database/queryEdge';
+
+// Helper to call tool execute with proper AI SDK signature
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+const executeQueryEdge = async (params: any) => {
+  const result = await queryEdgeTool.execute!(params, { toolCallId: 'test', messages: [] });
+  return result as { success: boolean; data?: { edges: any[]; connections?: any[] } };
+};
+
+describe('queryEdgeTool', () => {
+  it('returns empty list when no edges for from_node_id', async () => {
+    vi.mocked(edgeService.getEdges).mockResolvedValueOnce([]);
+
+    const result = await executeQueryEdge({ filters: { from_node_id: 1, limit: 10 } });
+
+    expect(result.success).toBe(true);
+    expect(result.data?.edges).toHaveLength(0);
+  });
+
+  it('returns edges when found for from_node_id', async () => {
+    vi.mocked(edgeService.getEdges).mockResolvedValueOnce([
+      { id: 1, from_node_id: 1, to_node_id: 2, source: 'user', created_at: '' },
+    ]);
+
+    const result = await executeQueryEdge({ filters: { from_node_id: 1, limit: 10 } });
+
+    expect(result.success).toBe(true);
+    expect(result.data?.edges).toHaveLength(1);
+  });
+
+  it('returns edge by ID when edge_id provided', async () => {
+    vi.mocked(edgeService.getEdgeById).mockResolvedValueOnce({
+      id: 9,
+      from_node_id: 1,
+      to_node_id: 2,
+      source: 'user',
+      created_at: '',
+    });
+
+    const result = await executeQueryEdge({ filters: { edge_id: 9, limit: 10 } });
+
+    expect(result.success).toBe(true);
+    expect(result.data?.edges).toHaveLength(1);
+  });
+
+  it('returns compact connection payloads for node traversal', async () => {
+    vi.mocked(edgeService.getNodeConnections).mockResolvedValueOnce([
+      {
+        edge: {
+          id: 42,
+          from_node_id: 1,
+          to_node_id: 2,
+          source: 'user',
+          created_at: '',
+          context: {
+            type: 'related_to',
+            explanation: 'A'.repeat(300),
+            confidence: 0.8,
+          },
+        },
+        connected_node: {
+          id: 2,
+          title: 'Connected node',
+          description: 'B'.repeat(250),
+          notes: 'Hidden notes should not be returned',
+          chunk: 'Hidden chunk should not be returned',
+        },
+      },
+    ] as any);
+
+    const result = await executeQueryEdge({ filters: { node_id: 1, limit: 20 } });
+
+    expect(result.success).toBe(true);
+    expect(result.data?.edges).toHaveLength(1);
+    expect(result.data?.connections).toHaveLength(1);
+    expect(result.data?.connections?.[0].connected_node.notes).toBeUndefined();
+    expect(result.data?.connections?.[0].connected_node.chunk).toBeUndefined();
+    expect(result.data?.connections?.[0].connected_node.description.length).toBeLessThanOrEqual(140);
+    expect(result.data?.edges?.[0].context.explanation.length).toBeLessThanOrEqual(180);
+  });
+});