gemini-design-mcp 3.8.0 → 3.9.0

package/build/index.js

@@ -11,7 +11,7 @@ import { generateVibesSchema, generateVibes } from "./tools/generate-vibes.js";
 // Create MCP server
 const server = new McpServer({
     name: "gemini-design-mcp",
-    version: "3.8.0",
+    version: "3.9.0",
 });
 // =============================================================================
 // TOOL 1: CREATE_FRONTEND
@@ -304,7 +304,7 @@ async function main() {
             res.json({
                 status: "ok",
                 service: "gemini-design-mcp",
-                version: "3.8.0",
+                version: "3.9.0",
                 transport: "sse",
                 timestamp: Date.now(),
             });
@@ -315,9 +315,12 @@ async function main() {
             const authHeader = req.headers.authorization;
             const apiKey = authHeader?.replace("Bearer ", "");
 
-            if (!apiKey || !apiKey.startsWith("gd_")) {
+            // Accept both platform keys (gd_) and Google API keys (AIza)
+            const isValidKey = apiKey && (apiKey.startsWith("gd_") || apiKey.startsWith("AIza"));
+            if (!isValidKey) {
                 return res.status(401).json({
-                    error: "Missing or invalid API key. Use Authorization: Bearer gd_xxx",
+                    error: "Missing or invalid API key. Use Authorization: Bearer <key>. " +
+                           "Accepted formats: gd_xxx (platform) or AIza... (Google API key)",
                 });
             }
 
@@ -354,7 +357,7 @@ async function main() {
 
         const port = parseInt(process.env.PORT || "3000", 10);
         app.listen(port, "0.0.0.0", () => {
-            console.log(`gemini-design-mcp v3.8.0 running on SSE transport`);
+            console.log(`gemini-design-mcp v3.9.0 running on SSE transport`);
             console.log(`  SSE endpoint: http://0.0.0.0:${port}/sse`);
             console.log(`  Messages endpoint: http://0.0.0.0:${port}/messages`);
             console.log(`  Health check: http://0.0.0.0:${port}/health`);
@@ -363,7 +366,7 @@ async function main() {
         // Stdio mode (default, for local IDE usage)
         const transport = new StdioServerTransport();
         await server.connect(transport);
-        console.error("gemini-design-mcp v3.8.0 running on stdio");
+        console.error("gemini-design-mcp v3.9.0 running on stdio");
     }
 }
 main().catch((error) => {

package/build/lib/gemini.js

@@ -2,71 +2,105 @@
 const PROXY_URL = "https://dashing-alpaca-785.convex.site/v1/generate";
 // Default model - Gemini 3 Flash Preview
 export const DEFAULT_MODEL = "gemini-3-flash-preview";
-// Get and validate API key
+// Import Google Generative AI SDK for BYOK (Bring Your Own Key)
+import { GoogleGenAI } from "@google/genai";
+// Get and validate API key - supports both platform keys (gd_) and Google API keys (AIza)
 function getApiKey() {
     const apiKey = process.env.API_KEY;
     if (!apiKey) {
         throw new Error("Missing API_KEY environment variable. " +
-            "Get your API key at https://gemini-design-mcp.com/dashboard/api-keys");
+            "Use either a platform key (gd_xxx) from https://gemini-design-mcp.com/dashboard/api-keys " +
+            "or a Google API key (AIza...) from https://aistudio.google.com/apikey");
     }
-    if (!apiKey.startsWith("gd_")) {
-        throw new Error("Invalid API key format. API keys must start with 'gd_'. " +
-            "Get your API key at https://gemini-design-mcp.com/dashboard/api-keys");
+    // Accept both platform keys (gd_) and Google API keys (AIza)
+    if (!apiKey.startsWith("gd_") && !apiKey.startsWith("AIza")) {
+        throw new Error("Invalid API key format. Use either:\n" +
+            "- Platform key starting with 'gd_' from https://gemini-design-mcp.com/dashboard/api-keys\n" +
+            "- Google API key starting with 'AIza' from https://aistudio.google.com/apikey");
     }
     return apiKey;
 }
+// Generate via platform proxy (for gd_ keys)
+async function generateViaProxy(apiKey, systemPrompt, userPrompt, model, thinkingMode, endpoint) {
+    const response = await fetch(PROXY_URL, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            "Authorization": `Bearer ${apiKey}`,
+        },
+        body: JSON.stringify({
+            model,
+            endpoint,
+            contents: [
+                {
+                    role: "user",
+                    parts: [{ text: userPrompt }],
+                },
+            ],
+            systemInstruction: {
+                parts: [{ text: systemPrompt }],
+            },
+            generationConfig: {
+                temperature: 1,
+            },
+            // Pass thinking mode for billing/logging purposes
+            thinkingMode,
+        }),
+    });
+    const result = await response.json();
+    if (!response.ok) {
+        const errorMsg = result.error || `API error: ${response.status}`;
+        // Check for credit/quota related errors
+        if (errorMsg.toLowerCase().includes('credit') ||
+            errorMsg.toLowerCase().includes('quota') ||
+            errorMsg.toLowerCase().includes('insufficient') ||
+            errorMsg.toLowerCase().includes('limit') ||
+            response.status === 402 ||
+            response.status === 429) {
+            throw new Error(`${errorMsg}\n\n Top up your credits here: https://gemini-design-mcp.com/settings/billing`);
+        }
+        throw new Error(errorMsg);
+    }
+    // Extract text from Gemini response format
+    const text = result.candidates?.[0]?.content?.parts?.[0]?.text;
+    if (!text) {
+        throw new Error("No content in response");
+    }
+    return text;
+}
+// Generate via direct Google API (for AIza keys - BYOK)
+async function generateViaDirect(apiKey, systemPrompt, userPrompt, model) {
+    const ai = new GoogleGenAI({ apiKey });
+    const response = await ai.models.generateContent({
+        model,
+        contents: userPrompt,
+        config: {
+            systemInstruction: systemPrompt,
+            temperature: 1,
+        },
+    });
+    const text = response.text;
+    if (!text) {
+        throw new Error("No content in response from Google API");
+    }
+    return text;
+}
 export async function generateWithGemini(systemPrompt, userPrompt, model = DEFAULT_MODEL, thinkingMode = "minimal", endpoint = "generate") {
     const apiKey = getApiKey();
+    const isGoogleKey = apiKey.startsWith("AIza");
     try {
-        const response = await fetch(PROXY_URL, {
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                "Authorization": `Bearer ${apiKey}`,
-            },
-            body: JSON.stringify({
-                model,
-                endpoint,
-                contents: [
-                    {
-                        role: "user",
-                        parts: [{ text: userPrompt }],
-                    },
-                ],
-                systemInstruction: {
-                    parts: [{ text: systemPrompt }],
-                },
-                generationConfig: {
-                    temperature: 1,
-                },
-                // Pass thinking mode for billing/logging purposes
-                thinkingMode,
-            }),
-        });
-        const result = await response.json();
-        if (!response.ok) {
-            const errorMsg = result.error || `API error: ${response.status}`;
-            // Check for credit/quota related errors
-            if (errorMsg.toLowerCase().includes('credit') ||
-                errorMsg.toLowerCase().includes('quota') ||
-                errorMsg.toLowerCase().includes('insufficient') ||
-                errorMsg.toLowerCase().includes('limit') ||
-                response.status === 402 ||
-                response.status === 429) {
-                throw new Error(`${errorMsg}\n\n💳 Top up your credits here: https://gemini-design-mcp.com/settings/billing`);
-            }
-            throw new Error(errorMsg);
-        }
-        // Extract text from Gemini response format
-        const text = result.candidates?.[0]?.content?.parts?.[0]?.text;
-        if (!text) {
-            throw new Error("No content in response");
+        if (isGoogleKey) {
+            // BYOK: Direct call to Google API
+            return await generateViaDirect(apiKey, systemPrompt, userPrompt, model);
+        } else {
+            // Platform key: Use proxy
+            return await generateViaProxy(apiKey, systemPrompt, userPrompt, model, thinkingMode, endpoint);
         }
-        return text;
     }
     catch (error) {
         const errorMessage = error instanceof Error ? error.message : "Unknown error";
-        console.error("Gemini Design API error:", errorMessage);
-        throw new Error(`Gemini Design API error: ${errorMessage}`);
+        const source = isGoogleKey ? "Google Gemini API" : "Gemini Design API";
+        console.error(`${source} error:`, errorMessage);
+        throw new Error(`${source} error: ${errorMessage}`);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gemini-design-mcp",
-  "version": "3.8.0",
+  "version": "3.9.0",
   "description": "MCP server that uses Gemini 3 Pro for frontend/design code generation",
   "main": "build/index.js",
   "bin": {