From e8ffb414c424a59523f94db5e8aba86b723b5b7e Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 12 Dec 2024 12:42:23 +0800
Subject: feat(mcp-server): support env var `AGENT_TOOLS_ONLY` (#145)

---
 mcp/server/index.js | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

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

diff --git a/mcp/server/index.js b/mcp/server/index.js
index d725568..3f4c5b6 100755
--- a/mcp/server/index.js
+++ b/mcp/server/index.js
@@ -33,6 +33,18 @@ try {
   console.error(`Failed to read functions at '${functionsJsonPath}'`);
   process.exit(1);
 }
+const agentToolsOnly = process.env["AGENT_TOOLS_ONLY"] === "true" || process.env["AGENT_TOOLS_ONLY"] === "1";
+functions = functions.filter(f => {
+  if (f.mcp) {
+    return false;
+  }
+  if (agentToolsOnly) {
+    return f.agent;
+  } else {
+    return true;
+  }
+});
+
 const env = Object.assign({}, process.env, {
   PATH: `${path.join(rootDir, "bin")}:${process.env.PATH}`
 });
@@ -114,7 +126,6 @@ function runCommand(command, args, env) {
 async function runServer() {
   const transport = new StdioServerTransport();
   await server.connect(transport);
-  console.error("LLM-Functions MCP Server running on stdio");
 }
 
 runServer().catch(console.error);
\ No newline at end of file
-- 
cgit v1.2.3