temba-mcp 0.3.5 → 0.4.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "temba-mcp",
-  "version": "0.3.5",
+  "version": "0.4.1",
   "description": "MCP for Temba documentation",
   "author": "Bouwe (https://bouwe.io)",
   "scripts": {

package/src/cli.js

@@ -1,11 +1,35 @@
 #!/usr/bin/env node
 import { startMcpServer } from './mcp.js'
+import { searchDocs } from './searchDocs.js'
 
 const args = process.argv.slice(2)
 const isDebug = args.includes('--debug')
+const queryIndex = args.findIndex((a) => a === '-q' || a === '--query')
 
 if (isDebug) {
   console.error('✨ Temba Docs MCP running in DEBUG mode')
 }
 
+// Check for the testing flag
+if (queryIndex !== -1 && args[queryIndex + 1]) {
+  const query = args[queryIndex + 1]
+
+  try {
+    // 1. Fetch from the exact same remote location the MCP server uses
+    const response = await fetch('https://temba.bouwe.io/search_index.json')
+    if (!response.ok) throw new Error(`HTTP error! status: ${response.status}`)
+    const index = await response.json()
+
+    // 2. Execute logic
+    const results = searchDocs(query, index)
+
+    // 3. Output raw JSON (Agent-fidelity)
+    process.stdout.write(JSON.stringify(results, null, 2))
+    process.exit(0)
+  } catch (err) {
+    console.error('❌ Failed to fetch remote index for test:', err.message)
+    process.exit(1)
+  }
+}
+
 startMcpServer({ debug: isDebug }).catch(console.error)

package/src/searchDocs.js

@@ -1,16 +1,67 @@
+const stopWords = new Set([
+  'a',
+  'an',
+  'the',
+  'is',
+  'i',
+  'do',
+  'how',
+  'to',
+  'and',
+  'or',
+  'of',
+  'in',
+  'on',
+  'for',
+  'with',
+  'by',
+  'as',
+  'at',
+  'from',
+  'that',
+  'this',
+  'it',
+  'are',
+  'was',
+  'were',
+  'be',
+  'been',
+])
+
 export const searchDocs = (query, index) => {
   const lowerQuery = query?.toLowerCase().trim()
+  if (!lowerQuery) return []
+
+  const tokens = lowerQuery.split(/\s+/).filter((token) => !stopWords.has(token))
 
-  if (!lowerQuery) {
-    return []
-  }
+  // Early return if the user only searched for stop words
+  if (tokens.length === 0) return []
 
   return (
-    index?.filter(
-      (page) =>
-        page.content.toLowerCase().includes(lowerQuery) ||
-        page.title.toLowerCase().includes(lowerQuery) ||
-        (page.keywords && page.keywords.some((k) => k.toLowerCase().includes(lowerQuery))),
-    ) || []
+    index
+      .map((page) => {
+        const content = (page.content || '').toLowerCase()
+        const title = (page.title || '').toLowerCase()
+        const keywords = (page.keywords || []).map((k) => k.toLowerCase())
+
+        // Count how many tokens are found in this page
+        let score = 0
+        tokens.forEach((token) => {
+          if (
+            content.includes(token) ||
+            title.includes(token) ||
+            keywords.some((k) => k.includes(token))
+          ) {
+            score += 1
+          }
+        })
+
+        return { page, score }
+      })
+      // Filter out pages that didn't match at least one token
+      .filter((item) => item.score > 0)
+      // Sort by most matches first
+      .sort((a, b) => b.score - a.score)
+      .map((item) => item.page)
   )
 }