@rigstate/mcp 0.5.1 → 0.5.3

package/dist/index.js

@@ -974,19 +974,30 @@ architecture rules, decisions, and constraints.`,
   }
 });
 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.");
+  const openRouterKey = process.env.OPENROUTER_API_KEY;
+  const googleKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+  if (!openRouterKey && !googleKey) {
+    console.warn("Neither OPENROUTER_API_KEY nor GOOGLE_GENERATIVE_AI_API_KEY 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;
+    if (openRouterKey) {
+      const { createOpenRouter } = await import("@openrouter/ai-sdk-provider");
+      const openrouter = createOpenRouter({ apiKey: openRouterKey });
+      const { embedding } = await embed({
+        model: openrouter.embedding("google/text-embedding-004"),
+        value: query.replace(/\n/g, " ")
+      });
+      return embedding;
+    } else {
+      const { google } = await import("@ai-sdk/google");
+      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;
@@ -999,12 +1010,14 @@ async function queryBrain(supabase, userId, projectId, query, limit = 8, thresho
   }
   const embedding = await generateQueryEmbedding(query);
   let memories = [];
+  console.log(`Searching brain for "${query}" (limit: ${limit}, threshold: ${threshold})`);
+  console.log(`Embedding present: ${!!embedding}`);
   const { data: searchResults, error: searchError } = await supabase.rpc("hybrid_search_memories", {
     p_project_id: projectId,
     p_query: query,
-    p_embedding: embedding,
+    p_embedding: embedding || null,
     p_limit: limit,
-    p_similarity_threshold: threshold || 0.2
+    p_similarity_threshold: threshold || 0.1
   });
   if (searchError) {
     console.error("Hybrid search error:", searchError);
@@ -1020,6 +1033,7 @@ async function queryBrain(supabase, userId, projectId, query, limit = 8, thresho
       }));
     }
   } else if (searchResults) {
+    console.log(`Found ${searchResults.length} results from RPC`);
     memories = searchResults.map((m) => ({
       id: m.id,
       content: m.content,