@rigstate/mcp 0.4.5 → 0.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@rigstate/mcp",
-    "version": "0.4.5",
+    "version": "0.5.0",
     "description": "Rigstate MCP Server - Model Context Protocol for AI Editors",
     "type": "module",
     "main": "./dist/index.js",
@@ -15,9 +15,12 @@
         "test": "node dist/index.js --help"
     },
     "dependencies": {
+        "@ai-sdk/google": "^3.0.13",
+        "@google/generative-ai": "^0.24.1",
         "@modelcontextprotocol/sdk": "^1.0.0",
-        "@supabase/supabase-js": "^2.39.0",
         "@rigstate/rules-engine": "0.1.0",
+        "@supabase/supabase-js": "^2.39.0",
+        "ai": "^6.0.5",
         "dotenv": "^17.2.3",
         "uuid": "^9.0.0",
         "zod": "^3.22.4"

package/roadmap.json

@@ -1,6 +1,6 @@
 {
   "project": "Rigstate Core",
-  "last_synced": "2026-01-22T20:06:06.152Z",
+  "last_synced": "2026-01-23T10:05:02.143Z",
   "roadmap": [
     {
       "id": "e503e35a-d480-457b-a771-63cf0297b48a",
@@ -47,7 +47,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "8f2f2655-f846-4561-bde8-b4618fa8ae28",
@@ -89,7 +92,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "8b31ac01-cf36-4668-98bd-3b6f83fd0042",
@@ -131,7 +137,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "9c28bc40-043c-4c68-9235-f47b71ca5d86",
@@ -173,7 +182,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "f3735dc7-ade5-4eee-bfd5-bb75c63fd399",
@@ -215,7 +227,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "31ae7181-be64-4d59-80f5-7a9d880f5741",
@@ -257,7 +272,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "aa6a6ff4-8028-4e91-8af0-0fa53c05dbb0",
@@ -299,7 +317,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "a8e185f3-f9c5-4534-96f5-fd5d50bc84fb",
@@ -341,7 +362,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "e1c1d0b4-f636-4f4c-99a4-736c575935fc",
@@ -383,7 +407,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "5fca8601-8fb3-4bf9-bfb2-7bc24feda26f",
@@ -425,7 +452,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "a8165a27-1f1d-45d5-87f4-468b2cfce98d",
@@ -467,7 +497,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "41baefd7-baf3-44d3-9aad-8faf64fb7d6a",
@@ -509,7 +542,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "de44fe4e-fc90-442f-87ec-8432c69324cd",
@@ -551,7 +587,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "569aad50-0096-40c5-8e3a-41a65fb7e30f",
@@ -593,7 +632,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "2cf3faac-067d-4a6c-99dc-938cf8bfd1b8",
@@ -635,7 +677,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "a0500716-55ec-4e66-99b0-4f51f59243f1",
@@ -677,7 +722,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "4d51757b-2c33-4e14-9cf4-0d4dc638c6e9",
@@ -719,7 +767,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "8eb5f854-30e0-41d4-bb88-27d854e3c60e",
@@ -761,7 +812,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "3784e7a0-629c-419d-9327-1743444466dc",
@@ -803,7 +857,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "d95fba4b-170c-4713-a80d-dc4d255a281c",
@@ -845,7 +902,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "19688111-5f4c-4a4b-a826-249643adeb80",
@@ -887,7 +947,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "6c2d2efe-70a9-44fc-8b53-53d24142adb7",
@@ -929,7 +992,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "cdfe0a5c-3287-453c-87b5-afa4138d90b7",
@@ -971,7 +1037,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "7a452451-23d0-4a00-ae94-36eae05adac0",
@@ -1013,7 +1082,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "05d106da-37a5-46f0-bca1-f19746aa5e0f",
@@ -1055,7 +1127,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "6dfd911a-ec75-4a73-a91f-f2ea0943f6c0",
@@ -1100,7 +1175,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "94c8d12a-1aed-4745-89dc-e95c8ff9be1a",
@@ -1145,7 +1223,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "bde519dc-b6ff-443c-b39b-a05f3d85b965",
@@ -1190,7 +1271,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "0b8da658-fc25-423a-a556-bddfb50f6fca",
@@ -1235,7 +1319,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "2ff8f7ef-0351-4b54-aaf2-ad0bf984af4b",
@@ -1277,7 +1364,10 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
     },
     {
       "id": "7f70a61d-be0a-40f3-821f-d0c0740bfce1",
@@ -1319,7 +1409,100 @@
       "metadata": null,
       "started_at": null,
       "description": null,
-      "updated_at": "2026-01-21T18:40:49.607549+00:00"
+      "updated_at": "2026-01-21T18:40:49.607549+00:00",
+      "source_cycle_id": null,
+      "archived_at": null,
+      "archived_reason": null
+    },
+    {
+      "id": "2cb8d1a2-a2c0-4bdd-b720-7add6f97d658",
+      "project_id": "bb9f8445-39fd-438c-8ab6-8057f5514395",
+      "step_number": 1025,
+      "title": "[SOVEREIGN] Autonomous REFAC: File Split",
+      "status": "ACTIVE",
+      "prompt_content": "Decompose apps/web/src/components/admin/global-nexus-sovereign.tsx into smaller, modular components to satisfy ARC-001.",
+      "verification_criteria": null,
+      "type": "CODE",
+      "created_at": "2026-01-22T23:08:30.360978+00:00",
+      "sprint_id": 0,
+      "sprint_focus": "Foundation",
+      "sprint_description": null,
+      "estimated_minutes": 20,
+      "verification_type": null,
+      "verification_path": null,
+      "verification_match": null,
+      "summary": null,
+      "sprint_number": 1,
+      "escalated_by": null,
+      "escalated_at": "2026-01-22T23:08:30.360978+00:00",
+      "visibility": "private",
+      "priority": "MEDIUM",
+      "is_legacy": false,
+      "completed_at": null,
+      "uid": "T-03feaa",
+      "severity": null,
+      "category": "ARCHITECTURE",
+      "origin_id": null,
+      "architectural_brief": "Frank has identified structural debt in apps/web/src/components/admin/global-nexus-sovereign.tsx. \n            Governance: SOVEREIGN. Worker dispatched to apply fixes.",
+      "icon": null,
+      "tags": null,
+      "context_summary": null,
+      "origin_feature_id": null,
+      "feature_id": null,
+      "checklist": [],
+      "origin_type": "AUTONOMOUS",
+      "metadata": null,
+      "started_at": null,
+      "description": null,
+      "updated_at": "2026-01-22T23:08:30.360978+00:00",
+      "source_cycle_id": "1d2da87d-3b90-4a69-930e-d60c2380427c",
+      "archived_at": null,
+      "archived_reason": null
+    },
+    {
+      "id": "622cbd96-ea02-4180-ba17-74985a1dadc5",
+      "project_id": "bb9f8445-39fd-438c-8ab6-8057f5514395",
+      "step_number": 1026,
+      "title": "[SOVEREIGN] Autonomous SECURITY_SCAN: Security Patch",
+      "status": "ACTIVE",
+      "prompt_content": "Apply missing security validators (Zod/RLS) identified in latest pulse.",
+      "verification_criteria": null,
+      "type": "CODE",
+      "created_at": "2026-01-23T00:58:55.548607+00:00",
+      "sprint_id": 0,
+      "sprint_focus": "Foundation",
+      "sprint_description": null,
+      "estimated_minutes": 20,
+      "verification_type": null,
+      "verification_path": null,
+      "verification_match": null,
+      "summary": null,
+      "sprint_number": 1,
+      "escalated_by": null,
+      "escalated_at": "2026-01-23T00:58:55.548607+00:00",
+      "visibility": "private",
+      "priority": "HIGH",
+      "is_legacy": false,
+      "completed_at": null,
+      "uid": "T-089a43",
+      "severity": null,
+      "category": "SECURITY",
+      "origin_id": null,
+      "architectural_brief": "Frank has identified security gaps in Project Core. \n            Governance: SOVEREIGN. Worker dispatched to apply fixes.",
+      "icon": null,
+      "tags": null,
+      "context_summary": null,
+      "origin_feature_id": null,
+      "feature_id": null,
+      "checklist": [],
+      "origin_type": "AUTONOMOUS",
+      "metadata": null,
+      "started_at": null,
+      "description": null,
+      "updated_at": "2026-01-23T00:58:55.548607+00:00",
+      "source_cycle_id": "80b163aa-4b22-450c-bfd5-e4198ec5fca5",
+      "archived_at": null,
+      "archived_reason": null
     }
   ]
 }

package/src/lib/supabase.ts

@@ -96,7 +96,8 @@ export async function authenticateApiKey(apiKey: string): Promise<{
         .then();
 
     // Create a user-scoped client for subsequent operations
-    const userSupabase = createSupabaseClient(SUPABASE_URL, SUPABASE_ANON_KEY);
+    // Use the most privileged key available (SERVICE_KEY or ANON_KEY)
+    const userSupabase = createSupabaseClient(SUPABASE_URL, clientKey);
 
     return {
         success: true,

package/src/tools/query-brain.ts

@@ -34,15 +34,27 @@ architecture rules, decisions, and constraints.`,
     }
 });
 
-// Simple embedding generation using text normalization
-async function generateQueryEmbedding(query: string): Promise<number[]> {
-    // For MCP server, we'll use the match_memories_text RPC if available,
-    // or fall back to keyword-based search.
-    // This is a placeholder - the actual embedding should be done via the Supabase edge function
-    // or passed through to the main Rigstate API.
-
-    // For now, return null to trigger keyword fallback
-    return [];
+// Simple embedding generation using Google Gemini
+async function generateQueryEmbedding(query: string): Promise<number[] | null> {
+    const apiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    if (!apiKey) {
+        console.warn('GOOGLE_GENERATIVE_AI_API_KEY not found, skipping vector search.');
+        return null;
+    }
+
+    try {
+        const { google } = await import('@ai-sdk/google');
+        const { embed } = await import('ai');
+
+        const { embedding } = await embed({
+            model: google.embedding('text-embedding-004'),
+            value: query.replace(/\n/g, ' '),
+        });
+        return embedding;
+    } catch (error) {
+        console.error('Failed to generate embedding for search:', error);
+        return null;
+    }
 }
 
 export async function queryBrain(
@@ -67,45 +79,24 @@ export async function queryBrain(
 
     // Try semantic search first using the match_memories RPC
     // This requires the embedding to be generated, so we'll try
-    // a text-based search as fallback
+    // Generate embedding if possible for semantic search
+    const embedding = await generateQueryEmbedding(query);
     let memories: MemoryRecord[] = [];
 
-    // Attempt keyword-based search using ilike on content
-    // This is a simpler approach that works without embeddings
-    const searchTerms = query.toLowerCase().split(/\s+/).filter(t => t.length > 2);
-
-    if (searchTerms.length > 0) {
-        // Build OR condition for fuzzy matching
-        const orConditions = searchTerms.map(term => `content.ilike.%${term}%`).join(',');
-
-        const { data: keywordMatches, error: searchError } = await supabase
-            .from('project_memories')
-            .select('id, content, category, tags, importance, created_at')
-            .eq('project_id', projectId)
-            .eq('is_active', true)
-            .or(orConditions)
-            .order('importance', { ascending: false, nullsFirst: false })
-            .limit(limit);
-
-        if (searchError) {
-            console.error('Search error:', searchError);
-        }
-
-        if (keywordMatches) {
-            memories = keywordMatches.map(m => ({
-                id: m.id,
-                content: m.content,
-                category: m.category || 'general',
-                tags: m.tags || [],
-                netVotes: m.importance || 0,
-                createdAt: m.created_at
-            }));
-        }
-    }
-
-    // If no keyword matches, fetch recent memories
-    if (memories.length === 0) {
-        const { data: recentMemories, error: recentError } = await supabase
+    // Use the hybrid search RPC
+    const { data: searchResults, error: searchError } = await supabase
+        .rpc('hybrid_search_memories', {
+            p_project_id: projectId,
+            p_query: query,
+            p_embedding: embedding,
+            p_limit: limit,
+            p_similarity_threshold: threshold || 0.2
+        });
+
+    if (searchError) {
+        console.error('Hybrid search error:', searchError);
+        // Fallback to basic recent fetch if RPC fails
+        const { data: recentMemories } = await supabase
             .from('project_memories')
             .select('id, content, category, tags, importance, created_at')
             .eq('project_id', projectId)
@@ -113,10 +104,6 @@ export async function queryBrain(
             .order('created_at', { ascending: false })
             .limit(limit);
 
-        if (recentError) {
-            console.error('Recent error:', recentError);
-        }
-
         if (recentMemories) {
             memories = recentMemories.map(m => ({
                 id: m.id,
@@ -127,17 +114,30 @@ export async function queryBrain(
                 createdAt: m.created_at
             }));
         }
+    } else if (searchResults) {
+        memories = searchResults.map((m: any) => ({
+            id: m.id,
+            content: m.content,
+            category: m.category,
+            tags: m.tags,
+            netVotes: m.importance,
+            createdAt: m.created_at
+        }));
     }
 
     // Format memories into a readable context block
     const contextLines = memories.map((m) => {
         const voteIndicator = m.netVotes && m.netVotes < 0 ? ` [⚠️ POORLY RATED: ${m.netVotes}]` : '';
-        const tagStr = m.tags.length > 0 ? ` [${m.tags.join(', ')}]` : '';
-        return `- [${m.category.toUpperCase()}]${tagStr}${voteIndicator}: ${m.content}`;
+        const tagStr = m.tags && m.tags.length > 0 ? ` [${m.tags.join(', ')}]` : '';
+        const category = m.category ? m.category.toUpperCase() : 'GENERAL';
+        return `- [${category}]${tagStr}${voteIndicator}: ${m.content}`;
     });
 
+    const searchType = embedding ? 'TRIPLE-HYBRID (Vector + FTS + Fuzzy)' : 'HYBRID (FTS + Fuzzy)';
+
     const formatted = memories.length > 0
         ? `=== PROJECT BRAIN: RELEVANT MEMORIES ===
+Search Mode: ${searchType}
 Query: "${query}"
 Found ${memories.length} relevant memories: