@rigstate/mcp 0.5.2 → 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;