“BeeRad”
28e570696c
- make source the canonical field across os routes, tools, ui, and mcp - align fresh-install schema, search, and embedding flows with source-first ingestion Generated with Claude Code
79 lines
2.6 KiB
TypeScript
79 lines
2.6 KiB
TypeScript
import { tool } from 'ai';
|
|
import { z } from 'zod';
|
|
import { nodeService } from '@/services/database/nodes';
|
|
import { formatNodeForChat } from '../infrastructure/nodeFormatter';
|
|
import type { Node } from '@/types/database';
|
|
|
|
export const queryDimensionNodesTool = tool({
|
|
description: 'Get nodes in a dimension, sorted by connection count.',
|
|
inputSchema: z.object({
|
|
dimension: z.string().describe('The dimension name to query nodes from'),
|
|
limit: z.number().optional().default(20).describe('Maximum number of nodes to return (default: 20)'),
|
|
offset: z.number().optional().default(0).describe('Number of nodes to skip for pagination'),
|
|
includeContent: z.boolean().optional().default(false).describe('Include truncated content preview (default: false)'),
|
|
}),
|
|
execute: async ({ dimension, limit = 20, offset = 0, includeContent = false }) => {
|
|
try {
|
|
// Query nodes with this dimension
|
|
const nodes = await nodeService.getNodes({
|
|
dimensions: [dimension],
|
|
limit,
|
|
offset,
|
|
sortBy: 'edges',
|
|
});
|
|
|
|
if (!nodes || nodes.length === 0) {
|
|
return {
|
|
success: true,
|
|
dimension,
|
|
nodes: [],
|
|
total: 0,
|
|
message: `No nodes found in dimension "${dimension}"`,
|
|
};
|
|
}
|
|
|
|
const formattedNodes = nodes.map((node: Node) => {
|
|
const formatted: Record<string, unknown> = {
|
|
id: node.id,
|
|
title: node.title,
|
|
label: formatNodeForChat({
|
|
id: node.id,
|
|
title: node.title,
|
|
dimensions: node.dimensions || [],
|
|
}),
|
|
edgeCount: node.edge_count || 0,
|
|
dimensions: node.dimensions || [],
|
|
created_at: node.created_at,
|
|
updated_at: node.updated_at,
|
|
event_date: node.event_date ?? null,
|
|
};
|
|
|
|
if (includeContent && (node.source || node.notes)) {
|
|
const previewSource = node.source || node.notes || '';
|
|
// Truncate to ~100 chars
|
|
formatted.sourcePreview = previewSource.length > 100
|
|
? previewSource.substring(0, 100) + '...'
|
|
: previewSource;
|
|
}
|
|
|
|
return formatted;
|
|
});
|
|
|
|
return {
|
|
success: true,
|
|
dimension,
|
|
nodes: formattedNodes,
|
|
total: nodes.length,
|
|
hasMore: nodes.length >= limit,
|
|
message: `Found ${nodes.length} nodes in dimension "${dimension}"`,
|
|
};
|
|
} catch (error) {
|
|
console.error('queryDimensionNodes error:', error);
|
|
return {
|
|
success: false,
|
|
error: error instanceof Error ? error.message : 'Failed to query dimension nodes',
|
|
};
|
|
}
|
|
},
|
|
});
|