secondbrainos-mcp-server 1.0.8 → 1.0.9

package/README.md

@@ -41,6 +41,13 @@ The server uses `@samchon/openapi` library for robust OpenAPI handling, providin
 - Automatic request formatting based on OpenAPI specifications
 - Support for complex parameters and nested objects
 
+### Prompts (Workflows)
+The server exposes your Second Brain OS workflows as MCP prompts, making them available in Claude Desktop's attach menu:
+- Automatically discovers your workflows via the `runPromptChain` service
+- Each workflow appears as a selectable prompt with its name and description
+- Selecting a prompt fetches the full prompt chain (ordered instructions) and injects them into the conversation
+- Supports an optional `user_input` argument to provide additional context
+
 ### Better Error Handling
 - Detailed error messages for debugging
 - Proper handling of authentication failures, bad requests, and service errors
@@ -117,7 +124,8 @@ The server requires the following environment variables:
 1. **Schema Fetching**: On startup, the server fetches your personalized OpenAPI schema from Second Brain OS
 2. **Schema Conversion**: The `@samchon/openapi` library converts the schema to an optimized format for LLM function calling
 3. **MCP Tools**: Each API endpoint becomes an MCP tool that Claude can use
-4. **Function Execution**: When Claude calls a tool, the server executes the corresponding API call with proper authentication
+4. **MCP Prompts**: If the `runPromptChain` service is available, your workflows are exposed as selectable prompts in the client UI
+5. **Function Execution**: When Claude calls a tool, the server executes the corresponding API call with proper authentication
 
 ## Troubleshooting
 

package/bin/cli.js

@@ -158,7 +158,7 @@ async function main() {
     
     // Verify user with the cloud function
     try {
-      const verificationUrl = 'https://us-central1-second-brain-os.cloudfunctions.net/verify-user-data-public';
+      const verificationUrl = 'https://us-central1-second-brain-os.cloudfunctions.net/gcf-sbos-verifyuserdatapublic';
       const response = await axios.post(verificationUrl, {
         "x-user-api-key": `Bearer ${USER_ID}:${USER_SECRET}`
       });

package/build/index.js

@@ -1,6 +1,6 @@
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { ListResourcesRequestSchema, ReadResourceRequestSchema, ListToolsRequestSchema, CallToolRequestSchema, ErrorCode, McpError } from "@modelcontextprotocol/sdk/types.js";
+import { ListResourcesRequestSchema, ReadResourceRequestSchema, ListToolsRequestSchema, CallToolRequestSchema, ListPromptsRequestSchema, GetPromptRequestSchema, ErrorCode, McpError } from "@modelcontextprotocol/sdk/types.js";
 import { OpenApi, HttpLlm } from "@samchon/openapi";
 import axios from "axios";
 import dotenv from "dotenv";
@@ -30,13 +30,30 @@ class SecondBrainOSServer {
                 this.functionMap.set(operation.operationId, func);
             }
         });
+        // Initialize workflow name-to-ID map
+        this.workflowNameToId = new Map();
+        // Discover runPromptChain path from the OpenAPI schema
+        this.runPromptChainPath = null;
+        if (initialSchema.paths) {
+            for (const [path, pathItem] of Object.entries(initialSchema.paths)) {
+                for (const operation of Object.values(pathItem)) {
+                    if (operation.operationId === 'runPromptChain') {
+                        this.runPromptChainPath = path;
+                        break;
+                    }
+                }
+                if (this.runPromptChainPath)
+                    break;
+            }
+        }
         this.server = new Server({
             name: "secondbrainos-server",
             version: "1.0.0"
         }, {
             capabilities: {
                 resources: {},
-                tools: {}
+                tools: {},
+                prompts: {}
             }
         });
         this.setupHandlers();
@@ -164,6 +181,96 @@ class SecondBrainOSServer {
                 throw error;
             }
         });
+        // List available prompts (user's workflows)
+        this.server.setRequestHandler(ListPromptsRequestSchema, async () => {
+            if (!this.runPromptChainPath) {
+                return { prompts: [] };
+            }
+            try {
+                const data = await this.callRunPromptChain('', '');
+                const workflows = data.workflows || [];
+                // Populate name-to-ID map
+                this.workflowNameToId.clear();
+                for (const wf of workflows) {
+                    if (wf.name && wf.workflow_id) {
+                        this.workflowNameToId.set(wf.name, wf.workflow_id);
+                    }
+                }
+                return {
+                    prompts: workflows.map((wf) => ({
+                        name: wf.name || wf.workflow_id,
+                        description: wf.description || wf.name,
+                        arguments: [
+                            {
+                                name: "user_input",
+                                description: "Optional context or input to guide the workflow execution",
+                                required: false
+                            }
+                        ]
+                    }))
+                };
+            }
+            catch (error) {
+                console.error('Failed to fetch workflows for prompts/list:', error);
+                return { prompts: [] };
+            }
+        });
+        // Get a specific prompt (workflow's prompt chain)
+        this.server.setRequestHandler(GetPromptRequestSchema, async (request) => {
+            const promptName = request.params.name;
+            const workflowId = this.workflowNameToId.get(promptName) || promptName;
+            const userInput = request.params.arguments?.user_input;
+            // Fetch workflow details to get prompt IDs
+            const workflowData = await this.callRunPromptChain('workflow', workflowId);
+            const promptIds = workflowData.prompt_id || [];
+            if (promptIds.length === 0) {
+                throw new McpError(ErrorCode.InvalidRequest, `No prompts found for workflow: ${workflowId}`);
+            }
+            // Fetch each prompt's instructions
+            const promptResults = await Promise.all(promptIds.map((promptId) => this.callRunPromptChain('prompt', promptId)));
+            // Sort by order and build messages
+            const sortedPrompts = promptResults.sort((a, b) => (a.order || 0) - (b.order || 0));
+            const messages = [];
+            for (const prompt of sortedPrompts) {
+                messages.push({
+                    role: "user",
+                    content: {
+                        type: "text",
+                        text: prompt.instructions
+                    }
+                });
+            }
+            // If user provided input, append it as the final message
+            if (userInput) {
+                messages.push({
+                    role: "user",
+                    content: {
+                        type: "text",
+                        text: userInput
+                    }
+                });
+            }
+            return {
+                description: `Workflow prompt chain (${sortedPrompts.length} steps)`,
+                messages
+            };
+        });
+    }
+    async callRunPromptChain(entity, entityId) {
+        if (!this.runPromptChainPath) {
+            throw new McpError(ErrorCode.InternalError, 'runPromptChain service not available for this user');
+        }
+        const url = `${this.baseUrl}${this.runPromptChainPath}`;
+        const response = await axios.post(url, {
+            entity,
+            entity_id: entityId
+        }, {
+            headers: {
+                'Authorization': `Bearer ${this.userId}:${this.userSecret}`,
+                'Content-Type': 'application/json'
+            }
+        });
+        return response.data;
     }
     setupErrorHandling() {
         // Error handling is now built into HttpLlm.execute
@@ -179,7 +286,7 @@ class SecondBrainOSServer {
 async function fetchSchema() {
     const userId = process.env.USER_ID;
     try {
-        const response = await axios.post('https://us-central1-second-brain-os.cloudfunctions.net/generate-open-api-schema', { user_id: userId }, {
+        const response = await axios.post('https://schema.secondbrainos.com', { user_id: userId }, {
             headers: {
                 'Content-Type': 'application/json'
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "secondbrainos-mcp-server",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "Second Brain OS MCP Server for Claude Desktop",
   "type": "module",
   "main": "build/index.js",
@@ -9,7 +9,7 @@
     "secondbrainos-mcp": "bin/cli.js"
   },
   "files": [
-    "build/",
+    "build/index.js",
     "bin/",
     "LICENSE",
     "README.md"

package/build/test-mcp-conversion.js

@@ -1,135 +0,0 @@
-import { OpenApi, HttpLlm } from "@samchon/openapi";
-import axios from "axios";
-import dotenv from "dotenv";
-import fs from "fs/promises";
-import path from "path";
-import { fileURLToPath } from 'url';
-import { dirname } from 'path';
-// Get __dirname equivalent for ES modules
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = dirname(__filename);
-// Load environment variables
-dotenv.config();
-// Function to fetch the schema from the API
-async function fetchSchema() {
-    const userId = process.env.USER_ID;
-    if (!userId) {
-        throw new Error("USER_ID environment variable is required");
-    }
-    console.log("Fetching schema for user:", userId);
-    try {
-        const response = await axios.post('https://us-central1-second-brain-os.cloudfunctions.net/generate-open-api-schema', { user_id: userId }, {
-            headers: {
-                'Content-Type': 'application/json'
-            }
-        });
-        return response.data;
-    }
-    catch (error) {
-        console.error('Failed to fetch schema:', error);
-        throw new Error('Failed to fetch API schema');
-    }
-}
-async function testMCPConversion() {
-    try {
-        console.log("=== MCP Conversion Test ===\n");
-        // Step 1: Fetch the original schema
-        console.log("1. Fetching OpenAPI schema...");
-        const originalSchema = await fetchSchema();
-        console.log("✓ Schema fetched successfully");
-        console.log(`  - Title: ${originalSchema.info?.title}`);
-        console.log(`  - Version: ${originalSchema.info?.version}`);
-        console.log(`  - Number of paths: ${Object.keys(originalSchema.paths || {}).length}\n`);
-        // Save original schema for reference
-        await fs.writeFile(path.join(__dirname, '../output/original-schema.json'), JSON.stringify(originalSchema, null, 2));
-        console.log("✓ Original schema saved to output/original-schema.json\n");
-        // Step 2: Convert using OpenApi
-        console.log("2. Converting schema using @samchon/openapi...");
-        const document = OpenApi.convert(originalSchema);
-        console.log("✓ Schema converted to OpenApi format\n");
-        // Step 3: Create LLM application
-        console.log("3. Creating LLM application...");
-        const application = HttpLlm.application({
-            model: "chatgpt",
-            document
-        });
-        console.log("✓ LLM application created");
-        console.log(`  - Number of functions: ${application.functions.length}\n`);
-        // Step 4: Create MCP-compatible tools
-        console.log("4. Converting to MCP tool format...");
-        const mcpTools = [];
-        const functionMap = new Map();
-        application.functions.forEach(func => {
-            const operation = func.operation();
-            const operationId = operation.operationId || func.name;
-            functionMap.set(operationId, func);
-            const mcpTool = {
-                name: operationId,
-                description: func.description || operationId,
-                inputSchema: func.parameters
-            };
-            mcpTools.push(mcpTool);
-        });
-        console.log(`✓ Converted ${mcpTools.length} tools to MCP format\n`);
-        // Step 5: Save the results
-        console.log("5. Saving conversion results...");
-        // Create output directory if it doesn't exist
-        await fs.mkdir(path.join(__dirname, '../output'), { recursive: true });
-        // Save MCP tools
-        await fs.writeFile(path.join(__dirname, '../output/mcp-tools.json'), JSON.stringify(mcpTools, null, 2));
-        console.log("✓ MCP tools saved to output/mcp-tools.json");
-        // Save function details
-        const functionDetails = application.functions.map(func => {
-            const operation = func.operation();
-            // Extract method and path from the route property if available
-            let method;
-            let path;
-            // The operation might have these properties in different ways depending on the OpenAPI version
-            // Let's check the actual structure
-            const route = operation.route;
-            if (route) {
-                method = route.method;
-                path = route.path;
-            }
-            return {
-                name: func.name,
-                operationId: operation.operationId,
-                method,
-                path,
-                description: func.description,
-                parameters: func.parameters,
-                operation: {
-                    summary: operation.summary,
-                    description: operation.description,
-                    tags: operation.tags
-                }
-            };
-        });
-        await fs.writeFile(path.join(__dirname, '../output/function-details.json'), JSON.stringify(functionDetails, null, 2));
-        console.log("✓ Function details saved to output/function-details.json\n");
-        // Step 6: Display summary
-        console.log("=== Conversion Summary ===");
-        console.log(`Total tools converted: ${mcpTools.length}`);
-        console.log("\nSample tools:");
-        mcpTools.slice(0, 5).forEach(tool => {
-            console.log(`  - ${tool.name}: ${tool.description}`);
-        });
-        if (mcpTools.length > 5) {
-            console.log(`  ... and ${mcpTools.length - 5} more`);
-        }
-        console.log("\n✓ Test completed successfully!");
-        console.log("\nCheck the 'output' directory for:");
-        console.log("  - original-schema.json: The raw OpenAPI schema from the API");
-        console.log("  - mcp-tools.json: Tools in MCP format");
-        console.log("  - function-details.json: Detailed function information");
-    }
-    catch (error) {
-        console.error("\n❌ Test failed:", error);
-        if (error instanceof Error) {
-            console.error("Error details:", error.message);
-        }
-        process.exit(1);
-    }
-}
-// Run the test
-testMCPConversion().catch(console.error);