feat(mcp-server): support env var AGENT_TOOLS_ONLY (#145)

mcp/server/README.md

@@ -8,10 +8,10 @@ Let LLM-functions tools/agents be used through the Model Context Protocol.
 {
   "mcpServers": {
     "tools": {
-      "command": "node",
+      "command": "npx",
       "args": [
-        "<path-to-llm-functions>/mcp/server/index.js",
-        "<path-to-llm-functions>"
+        "mcp-llm-functions",
+        "<llm-functions-dir>"
       ]
     }
   }
@@ -26,11 +26,15 @@ Let LLM-functions tools/agents be used through the Model Context Protocol.
     "<agent-name>": {
       "command": "node",
       "args": [
-        "<path-to-llm-functions>/mcp/server/index.js",
-        "<path-to-llm-functions>",
+        "mcp-llm-functions",
+        "<llm-functions-dir>"
         "<agent-name>",
       ]
     }
   }
 }
 ```
+
+## Environment Variables
+
+- `AGENT_TOOLS_ONLY`: Set to `true` or `1` to ignore shared tools and display only agent tools.

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);

mcp/server/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-llm-functions",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "Let LLM-functions tools/agents be used through the Model Context Protocol",
   "license": "MIT",
   "author": "sigoden <[email protected]>",