secondbrainos-mcp-server 1.2.5 → 1.3.0

package/build/index.js

@@ -5,7 +5,15 @@ import { OpenApi, HttpLlm } from "@samchon/openapi";
 import axios from "axios";
 import dotenv from "dotenv";
 import yaml from 'js-yaml';
+import fs from 'fs';
+import path from 'path';
 dotenv.config();
+function toSnakeCase(str) {
+    return str
+        .toLowerCase()
+        .replace(/[^a-z0-9]+/g, '_')
+        .replace(/^_+|_+$/g, '');
+}
 class SecondBrainOSServer {
     constructor(initialSchema) {
         this.originalSpec = initialSchema;
@@ -37,6 +45,7 @@ class SecondBrainOSServer {
         // Discover service paths from the OpenAPI schema
         this.runPromptChainPath = null;
         this.getAIAgentsSchemaPath = null;
+        this.generateUploadURLPath = null;
         if (initialSchema.paths) {
             for (const [path, pathItem] of Object.entries(initialSchema.paths)) {
                 for (const operation of Object.values(pathItem)) {
@@ -46,8 +55,11 @@ class SecondBrainOSServer {
                     if (operation.operationId === 'getAIAgentsSchema') {
                         this.getAIAgentsSchemaPath = path;
                     }
+                    if (operation.operationId === 'generateFileUploadGoogleCloudStorageURL') {
+                        this.generateUploadURLPath = path;
+                    }
                 }
-                if (this.runPromptChainPath && this.getAIAgentsSchemaPath)
+                if (this.runPromptChainPath && this.getAIAgentsSchemaPath && this.generateUploadURLPath)
                     break;
             }
         }
@@ -125,6 +137,8 @@ class SecondBrainOSServer {
                 throw new McpError(ErrorCode.MethodNotFound, `Unknown function: ${request.params.name}`);
             }
             try {
+                // Intercept file_path arguments: upload local files to GCS
+                const processedArgs = await this.interceptFilePathArgs((request.params.arguments || {}));
                 // Use HttpLlm.execute for better error handling and type safety
                 const result = await HttpLlm.execute({
                     connection: {
@@ -135,7 +149,7 @@ class SecondBrainOSServer {
                     },
                     application: this.application, // Type assertion to avoid generic issues
                     function: func,
-                    input: request.params.arguments || {}
+                    input: processedArgs
                 });
                 return {
                     content: [{
@@ -197,13 +211,15 @@ class SecondBrainOSServer {
                     this.workflowNameToId.clear();
                     for (const wf of workflows) {
                         if (wf.name && wf.workflow_id) {
-                            this.workflowNameToId.set(wf.name, wf.workflow_id);
+                            const snakeName = `skill_${toSnakeCase(wf.name)}`;
+                            this.workflowNameToId.set(snakeName, wf.workflow_id);
                         }
                     }
                     for (const wf of workflows) {
                         const skillName = wf.name || wf.workflow_id;
+                        const snakeName = `skill_${toSnakeCase(skillName)}`;
                         prompts.push({
-                            name: skillName,
+                            name: snakeName,
                             title: `[Skill] ${skillName}`,
                             description: wf.description || wf.name,
                             arguments: [
@@ -227,10 +243,10 @@ class SecondBrainOSServer {
                     this.agentNameToId.clear();
                     for (const agent of agents) {
                         if (agent.name && agent.id) {
-                            const agentPromptName = `[Agent] ${agent.name}`;
-                            this.agentNameToId.set(agentPromptName, agent.id);
+                            const snakeName = `agent_${toSnakeCase(agent.name)}`;
+                            this.agentNameToId.set(snakeName, agent.id);
                             prompts.push({
-                                name: agentPromptName,
+                                name: snakeName,
                                 title: `[Agent] ${agent.name}`,
                                 description: agent.description || agent.name,
                                 arguments: [
@@ -257,7 +273,7 @@ class SecondBrainOSServer {
                         .filter((id) => id && id.length > 0);
                     if (collectionIds.length > 0) {
                         prompts.push({
-                            name: "Knowledge Bases",
+                            name: "knowledge_bases",
                             title: "[Knowledge Bases]",
                             description: "Knowledge base collection IDs available to agents",
                             arguments: [
@@ -281,7 +297,7 @@ class SecondBrainOSServer {
             const promptName = request.params.name;
             const userInput = request.params.arguments?.user_input;
             // Check if this is the Knowledge Bases prompt
-            if (promptName === "Knowledge Bases") {
+            if (promptName === "knowledge_bases") {
                 const agents = await this.fetchAndEnrichAgents();
                 const collectionIds = agents
                     .map((a) => a.searchMyKnowledge_collection_id)
@@ -448,6 +464,71 @@ class SecondBrainOSServer {
         });
         return response.data;
     }
+    /**
+     * Scans tool arguments for file_path parameters containing local paths,
+     * uploads those files to GCS via signed URL, and replaces the values with GCS paths.
+     * - https:// paths are passed through
+     * - http:// paths are rejected
+     * - Everything else is treated as a local file path
+     */
+    async interceptFilePathArgs(args) {
+        const result = { ...args };
+        for (const [key, value] of Object.entries(result)) {
+            if (typeof value === 'object' && value !== null && !Array.isArray(value)) {
+                result[key] = await this.interceptFilePathArgs(value);
+                continue;
+            }
+            if (typeof value !== 'string' || !key.toLowerCase().includes('file_path')) {
+                continue;
+            }
+            const trimmed = value.trim();
+            // https:// — pass through
+            if (trimmed.startsWith('https://'))
+                continue;
+            // http:// — reject
+            if (trimmed.startsWith('http://')) {
+                throw new McpError(ErrorCode.InvalidRequest, `Insecure http:// URLs are not allowed for file_path. Use https:// or a local file path.`);
+            }
+            // Local file path — validate, upload via signed URL, replace
+            if (!this.generateUploadURLPath) {
+                throw new McpError(ErrorCode.InternalError, 'File upload service (generateFileUploadGoogleCloudStorageURL) is not available for this user. Cannot upload local files.');
+            }
+            const ext = path.extname(trimmed).toLowerCase();
+            if (!SecondBrainOSServer.ALLOWED_EXTENSIONS.includes(ext)) {
+                throw new McpError(ErrorCode.InvalidRequest, `Unsupported file extension "${ext}" for file_path. Only ${SecondBrainOSServer.ALLOWED_EXTENSIONS.join(', ')} are supported.`);
+            }
+            if (!fs.existsSync(trimmed)) {
+                throw new McpError(ErrorCode.InvalidRequest, `Local file not found: ${trimmed}`);
+            }
+            const fileContent = fs.readFileSync(trimmed, 'utf-8');
+            if (!fileContent.trim()) {
+                throw new McpError(ErrorCode.InvalidRequest, `File is empty: ${trimmed}`);
+            }
+            const fileName = path.basename(trimmed);
+            const fileSize = Buffer.byteLength(fileContent, 'utf-8');
+            // Step 1: Get signed upload URL from the GCF
+            const uploadUrlResponse = await axios.post(`${this.baseUrl}${this.generateUploadURLPath}`, {
+                file_name: fileName,
+                file_size: fileSize,
+                mime_type: 'text/plain'
+            }, {
+                headers: {
+                    'Authorization': `Bearer ${this.userId}:${this.userSecret}`,
+                    'Content-Type': 'application/json'
+                }
+            });
+            const { upload_url, gs_path } = uploadUrlResponse.data;
+            // Step 2: PUT file content directly to GCS via signed URL
+            await axios.put(upload_url, fileContent, {
+                headers: {
+                    'Content-Type': 'text/plain'
+                }
+            });
+            console.error(`Uploaded local file to ${gs_path}`);
+            result[key] = gs_path;
+        }
+        return result;
+    }
     async fetchAndEnrichAgents() {
         if (this.cachedAgents)
             return this.cachedAgents;
@@ -491,6 +572,7 @@ class SecondBrainOSServer {
         console.error("Second Brain OS MCP server running on stdio");
     }
 }
+SecondBrainOSServer.ALLOWED_EXTENSIONS = ['.txt', '.md'];
 // Function to fetch the schema from the API
 async function fetchSchema() {
     const userId = process.env.USER_ID;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "secondbrainos-mcp-server",
-  "version": "1.2.5",
+  "version": "1.3.0",
   "description": "Second Brain OS MCP Server for Claude Desktop",
   "type": "module",
   "main": "build/index.js",