Skip to content

Commit 102c3de

Browse files
committed
chore: remove hardcoded tool filtering in server.ts
1 parent a92852b commit 102c3de

2 files changed

Lines changed: 8 additions & 14 deletions

File tree

mcp/dist/index.js

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -19569,9 +19569,8 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
1956919569
if (notebookClient) {
1957019570
try {
1957119571
const response = await notebookClient.listTools();
19572-
const filteredTools = response.tools.filter((t) => t.name !== "render_chart");
19573-
filteredTools.forEach((t) => toolOwnerMap.set(t.name, "notebook"));
19574-
aggregatedTools.push(...filteredTools);
19572+
response.tools.forEach((t) => toolOwnerMap.set(t.name, "notebook"));
19573+
aggregatedTools.push(...response.tools);
1957519574
} catch (e) {
1957619575
console.error("Error listing tools from notebook client:", e);
1957719576
}
@@ -19584,9 +19583,8 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
1958419583
if (vizClient) {
1958519584
try {
1958619585
const response = await vizClient.listTools();
19587-
const filteredTools = response.tools.filter((t) => t.name === "render_chart");
19588-
filteredTools.forEach((t) => toolOwnerMap.set(t.name, "viz"));
19589-
aggregatedTools.push(...filteredTools);
19586+
response.tools.forEach((t) => toolOwnerMap.set(t.name, "viz"));
19587+
aggregatedTools.push(...response.tools);
1959019588
} catch (e) {
1959119589
console.error("Error listing tools from viz client:", e);
1959219590
}

mcp/server.ts

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -239,10 +239,8 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
239239
if (notebookClient) {
240240
try {
241241
const response = await notebookClient.listTools();
242-
// Filter out visualization tools (e.g., render_chart)
243-
const filteredTools = response.tools.filter((t: any) => t.name !== 'render_chart');
244-
filteredTools.forEach((t: any) => toolOwnerMap.set(t.name, 'notebook'));
245-
aggregatedTools.push(...filteredTools);
242+
response.tools.forEach((t: any) => toolOwnerMap.set(t.name, 'notebook'));
243+
aggregatedTools.push(...response.tools);
246244
} catch (e) {
247245
console.error('Error listing tools from notebook client:', e);
248246
}
@@ -256,10 +254,8 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
256254
if (vizClient) {
257255
try {
258256
const response = await vizClient.listTools();
259-
// Only include visualization tools (e.g., render_chart)
260-
const filteredTools = response.tools.filter((t: any) => t.name === 'render_chart');
261-
filteredTools.forEach((t: any) => toolOwnerMap.set(t.name, 'viz'));
262-
aggregatedTools.push(...filteredTools);
257+
response.tools.forEach((t: any) => toolOwnerMap.set(t.name, 'viz'));
258+
aggregatedTools.push(...response.tools);
263259
} catch (e) {
264260
console.error('Error listing tools from viz client:', e);
265261
}

0 commit comments

Comments
 (0)