@rigstate/mcp 0.4.6 → 0.5.1

package/dist/index.js

@@ -676,7 +676,7 @@ var QueryBrainInputSchema = z4.object({
   projectId: z4.string().uuid("Invalid project ID"),
   query: z4.string().min(1, "Query is required"),
   limit: z4.number().min(1).max(20).optional().default(8),
-  threshold: z4.number().min(0).max(1).optional().default(0.5)
+  threshold: z4.number().min(0).max(1).optional().default(0.1)
 });
 var GetProjectContextInputSchema = z4.object({
   projectId: z4.string().uuid("Invalid project ID")
@@ -973,35 +973,42 @@ architecture rules, decisions, and constraints.`,
     return { content: [{ type: "text", text: result.formatted }] };
   }
 });
+async function generateQueryEmbedding(query) {
+  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;
+  }
+}
 async function queryBrain(supabase, userId, projectId, query, limit = 8, threshold = 0.5) {
   const { data: project, error: projectError } = await supabase.from("projects").select("id").eq("id", projectId).eq("owner_id", userId).single();
   if (projectError || !project) {
     throw new Error("Project not found or access denied");
   }
+  const embedding = await generateQueryEmbedding(query);
   let memories = [];
-  const searchTerms = query.toLowerCase().split(/\s+/).filter((t) => t.length > 2);
-  if (searchTerms.length > 0) {
-    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 (memories.length === 0) {
-    const { data: recentMemories, error: recentError } = await supabase.from("project_memories").select("id, content, category, tags, importance, created_at").eq("project_id", projectId).eq("is_active", true).order("created_at", { ascending: false }).limit(limit);
-    if (recentError) {
-      console.error("Recent error:", recentError);
-    }
+  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);
+    const { data: recentMemories } = await supabase.from("project_memories").select("id, content, category, tags, importance, created_at").eq("project_id", projectId).eq("is_active", true).order("created_at", { ascending: false }).limit(limit);
     if (recentMemories) {
       memories = recentMemories.map((m) => ({
         id: m.id,
@@ -1012,13 +1019,25 @@ async function queryBrain(supabase, userId, projectId, query, limit = 8, thresho
         createdAt: m.created_at
       }));
     }
+  } else if (searchResults) {
+    memories = searchResults.map((m) => ({
+      id: m.id,
+      content: m.content,
+      category: m.category,
+      tags: m.tags,
+      netVotes: m.importance,
+      createdAt: m.created_at
+    }));
   }
   const contextLines = memories.map((m) => {
     const voteIndicator = m.netVotes && m.netVotes < 0 ? ` [\u26A0\uFE0F 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: