brief-mcp 1.2.0 → 1.4.0

package/dist/index.js

@@ -41,28 +41,35 @@ class SupabaseClient {
             'Content-Type': 'application/json'
         };
     }
-    async search(question) {
-        const response = await axios.post(`${EDGE_FUNCTIONS_URL}/search-context`, { question, match_count: 5, match_threshold: 0.3 }, { headers: this.headers });
+    async search(question, orgRef) {
+        const response = await axios.post(`${EDGE_FUNCTIONS_URL}/search-context`, { question, match_count: 5, match_threshold: 0.3, org_ref: orgRef }, { headers: this.headers });
         return response.data;
     }
-    async ingestDocument(content, title, orgId) {
-        // First get user's org_id from a simple call
+    async ingestDocument(content, title, orgRef) {
         const response = await axios.post(`${EDGE_FUNCTIONS_URL}/ingest-context`, {
             content,
             title: title || 'Untitled',
-            org_id: orgId,
+            org_ref: orgRef,
             source_type: 'mcp'
         }, { headers: this.headers });
         return response.data;
     }
-    async listDocuments(limit = 20) {
-        const response = await axios.get(`${EDGE_FUNCTIONS_URL}/list-documents?limit=${limit}`, { headers: this.headers });
+    async listDocuments(limit = 20, orgRef) {
+        let url = `${EDGE_FUNCTIONS_URL}/list-documents?limit=${limit}`;
+        if (orgRef) {
+            url += `&org_ref=${encodeURIComponent(orgRef)}`;
+        }
+        const response = await axios.get(url, { headers: this.headers });
         return response.data;
     }
     async deleteDocument(documentId) {
         const response = await axios.post(`${EDGE_FUNCTIONS_URL}/delete-document`, { document_id: documentId }, { headers: this.headers });
         return response.data;
     }
+    async listOrganizations() {
+        const response = await axios.get(`${EDGE_FUNCTIONS_URL}/list-organizations`, { headers: this.headers });
+        return response.data;
+    }
 }
 // ============================================================================
 // MCP Server
@@ -83,13 +90,17 @@ async function startMCPServer() {
         tools: [
             {
                 name: 'search-context',
-                description: 'Search your organization\'s knowledge base and get AI-powered answers. 8 models available: Mistral Small/Nemo, Llama 70B, Qwen 72B/Coder, DeepSeek V3, GLM, GPT-4o Mini.',
+                description: 'Search a knowledge base and get AI-powered answers. IMPORTANT: Call list-organizations first to get org_ref values, then specify which organization to search. If no org_ref is provided, uses your default organization.',
                 inputSchema: {
                     type: 'object',
                     properties: {
                         question: {
                             type: 'string',
                             description: 'Your question or search query'
+                        },
+                        org_ref: {
+                            type: 'string',
+                            description: 'Organization slug or ref to search in (from list-organizations). Required to search a specific org.'
                         }
                     },
                     required: ['question']
@@ -97,7 +108,7 @@ async function startMCPServer() {
             },
             {
                 name: 'add-knowledge',
-                description: 'Add new content to your organization\'s knowledge base. Content is automatically chunked and vectorized.',
+                description: 'Add new content to a knowledge base. IMPORTANT: Call list-organizations first to get org_ref values if you want to add to a specific organization.',
                 inputSchema: {
                     type: 'object',
                     properties: {
@@ -108,6 +119,10 @@ async function startMCPServer() {
                         title: {
                             type: 'string',
                             description: 'Title for the document'
+                        },
+                        org_ref: {
+                            type: 'string',
+                            description: 'Organization slug or ref to add content to (from list-organizations)'
                         }
                     },
                     required: ['content']
@@ -115,13 +130,17 @@ async function startMCPServer() {
             },
             {
                 name: 'list-knowledge',
-                description: 'List documents in your organization\'s knowledge base',
+                description: 'List documents in a knowledge base. IMPORTANT: Call list-organizations first to get org_ref values, then specify which organization to list. If no org_ref is provided, uses your default organization.',
                 inputSchema: {
                     type: 'object',
                     properties: {
                         limit: {
                             type: 'number',
                             description: 'Maximum number of documents to return (default: 20)'
+                        },
+                        org_ref: {
+                            type: 'string',
+                            description: 'Organization slug or ref to list documents from (from list-organizations). Required to list a specific org.'
                         }
                     }
                 }
@@ -139,6 +158,14 @@ async function startMCPServer() {
                     },
                     required: ['id']
                 }
+            },
+            {
+                name: 'list-organizations',
+                description: 'List all organizations you belong to, including your role and API key for each',
+                inputSchema: {
+                    type: 'object',
+                    properties: {}
+                }
             }
         ]
     }));
@@ -148,11 +175,11 @@ async function startMCPServer() {
         try {
             switch (name) {
                 case 'search-context': {
-                    const { question } = toolArgs;
+                    const { question, org_ref } = toolArgs;
                     if (!question || typeof question !== 'string') {
                         throw new Error('Question is required');
                     }
-                    const result = await supabase.search(question);
+                    const result = await supabase.search(question, org_ref);
                     let responseText = result.answer;
                     if (result.sources && result.sources.length > 0) {
                         responseText += '\n\n---\n**Sources:**';
@@ -166,30 +193,26 @@ async function startMCPServer() {
                     };
                 }
                 case 'add-knowledge': {
-                    const { content, title } = toolArgs;
+                    const { content, title, org_ref } = toolArgs;
                     if (!content || typeof content !== 'string') {
                         throw new Error('Content is required');
                     }
-                    // Get org_id from list-documents if not cached
-                    if (!cachedOrgId) {
-                        const listResult = await supabase.listDocuments(1);
-                        cachedOrgId = listResult.org_id;
-                    }
-                    const result = await supabase.ingestDocument(content, title, cachedOrgId);
+                    const result = await supabase.ingestDocument(content, title, org_ref);
+                    cachedOrgId = result.org_id; // Cache for future use
                     return {
                         content: [{
                                 type: 'text',
-                                text: `Document added!\nID: ${result.documentId}\nChunks created: ${result.chunks}`
+                                text: `Document added!\nID: ${result.documentId}\nChunks created: ${result.chunks}${org_ref ? `\nOrganization: ${org_ref}` : ''}`
                             }]
                     };
                 }
                 case 'list-knowledge': {
-                    const { limit } = toolArgs;
-                    const result = await supabase.listDocuments(limit || 20);
+                    const { limit, org_ref } = toolArgs;
+                    const result = await supabase.listDocuments(limit || 20, org_ref);
                     cachedOrgId = result.org_id; // Cache for future add-knowledge calls
                     if (result.documents.length === 0) {
                         return {
-                            content: [{ type: 'text', text: 'No documents in knowledge base.' }]
+                            content: [{ type: 'text', text: `No documents in knowledge base${org_ref ? ` (org: ${org_ref})` : ''}.` }]
                         };
                     }
                     let text = `**${result.documents.length} document(s):**\n\n`;
@@ -216,6 +239,24 @@ async function startMCPServer() {
                             }]
                     };
                 }
+                case 'list-organizations': {
+                    const result = await supabase.listOrganizations();
+                    if (result.organizations.length === 0) {
+                        return {
+                            content: [{ type: 'text', text: 'You are not a member of any organization.' }]
+                        };
+                    }
+                    let text = `**${result.total_count} organization(s):**\n\n`;
+                    result.organizations.forEach((org, i) => {
+                        const defaultMark = org.is_default ? ' ⭐ (default)' : '';
+                        text += `${i + 1}. **${org.name}**${defaultMark}\n`;
+                        text += `   Slug: \`${org.slug}\` | Role: ${org.role} | Members: ${org.member_count}\n`;
+                        text += `   API Key: \`${org.api_key}\`\n\n`;
+                    });
+                    return {
+                        content: [{ type: 'text', text }]
+                    };
+                }
                 default:
                     throw new Error(`Unknown tool: ${name}`);
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "brief-mcp",
-  "version": "1.2.0",
+  "version": "1.4.0",
   "description": "Brief MCP - Context-first development with RAG-powered knowledge base",
   "type": "module",
   "main": "dist/index.js",