Skip to content

Commit d289eff

Browse files
committed
feat: Register Graph Tools in MCP Server Handler (#9654)
1 parent 8ce9d98 commit d289eff

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

ai/mcp/server/knowledge-base/services/toolService.mjs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ import {fileURLToPath} from 'url';
33
import DatabaseService from './DatabaseService.mjs';
44
import DatabaseLifecycleService from './DatabaseLifecycleService.mjs';
55
import DocumentService from './DocumentService.mjs';
6+
import GraphService from './GraphService.mjs';
67
import HealthService from './HealthService.mjs';
78
import QueryService from './QueryService.mjs';
89
import SearchService from './SearchService.mjs';
@@ -16,11 +17,14 @@ const serviceMapping = {
1617
ask_knowledge_base : SearchService .ask .bind(SearchService),
1718
get_class_hierarchy : QueryService .getClassHierarchy .bind(QueryService),
1819
get_document_by_id : DocumentService .getDocumentById .bind(DocumentService),
20+
get_neighbors : GraphService .getNeighbors .bind(GraphService),
21+
get_node : GraphService .getNode .bind(GraphService),
1922
healthcheck : HealthService .healthcheck .bind(HealthService),
2023
list_documents : DocumentService .listDocuments .bind(DocumentService),
2124
manage_database : DatabaseLifecycleService.manageDatabase .bind(DatabaseLifecycleService),
2225
manage_knowledge_base: DatabaseService .manageKnowledgeBase.bind(DatabaseService),
23-
query_documents : QueryService .queryDocuments .bind(QueryService)
26+
query_documents : QueryService .queryDocuments .bind(QueryService),
27+
search_nodes : GraphService .searchNodes .bind(GraphService)
2428
};
2529

2630
const toolService = Neo.create(ToolService, {

0 commit comments

Comments
 (0)