From 8e8b2f736b8e635396e713e219c8fe8575a6f491 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 13 Dec 2024 06:01:53 +0800
Subject: feat(mcp): add `prefix` filed to  bridge server configuraiton (#148)

---
 mcp/bridge/index.js | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'mcp/bridge/index.js')

diff --git a/mcp/bridge/index.js b/mcp/bridge/index.js
index c978737..59896e3 100644
--- a/mcp/bridge/index.js
+++ b/mcp/bridge/index.js
@@ -29,8 +29,9 @@ try {
 async function startMcpServer(id, serverConfig) {
   console.log(`Starting ${id} server...`);
   const capabilities = { tools: {} };
+  const { prefix = true, ...rest } = serverConfig;
   const transport = new StdioClientTransport({
-    ...serverConfig,
+    ...rest,
   });
   const client = new Client(
     { name: id, version: "1.0.0" },
@@ -42,7 +43,7 @@ async function startMcpServer(id, serverConfig) {
     ({ name, description, inputSchema }) =>
     ({
       spec: {
-        name: `${normalizeToolName(`${id}_${name}`)}`,
+        name: `${formatToolName(id, name, prefix)}`,
         description,
         parameters: inputSchema,
       },
@@ -181,7 +182,8 @@ function arrayify(a) {
   return r
 }
 
-function normalizeToolName(name) {
+function formatToolName(serverName, toolName, prefix) {
+  const name = prefix ? `${serverName}_${toolName}` : toolName;
   return name.toLowerCase().replace(/-/g, "_");
 }
 
-- 
cgit v1.2.3