gemini-design-mcp 3.7.2 → 3.9.0

package/build/index.js

@@ -1,6 +1,9 @@
 #!/usr/bin/env node
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import express from "express";
+import cors from "cors";
 import { createFrontendSchema, createFrontend } from "./tools/create-frontend.js";
 import { modifyFrontendSchema, modifyFrontend } from "./tools/modify-frontend.js";
 import { snippetFrontendSchema, snippetFrontend } from "./tools/snippet-frontend.js";
@@ -8,7 +11,7 @@ import { generateVibesSchema, generateVibes } from "./tools/generate-vibes.js";
 // Create MCP server
 const server = new McpServer({
     name: "gemini-design-mcp",
-    version: "3.2.0",
+    version: "3.9.0",
 });
 // =============================================================================
 // TOOL 1: CREATE_FRONTEND
@@ -279,9 +282,92 @@ Keywords: minimal, whitespace, gallery, clean, sophisticated`, generateVibesSche
 // START SERVER
 // =============================================================================
 async function main() {
-    const transport = new StdioServerTransport();
-    await server.connect(transport);
-    console.error("gemini-design-mcp v3.2.0 running on stdio");
+    const transportType = process.env.TRANSPORT || "stdio";
+
+    if (transportType === "sse") {
+        // SSE/HTTP mode for remote connections
+        const app = express();
+
+        app.use(cors({
+            origin: "*",
+            methods: ["GET", "POST", "OPTIONS"],
+            allowedHeaders: ["Content-Type", "Authorization", "mcp-session-id"],
+            credentials: true,
+        }));
+        app.use(express.json());
+
+        // Store active transports by session
+        const transports = new Map();
+
+        // Health check endpoint
+        app.get("/health", (req, res) => {
+            res.json({
+                status: "ok",
+                service: "gemini-design-mcp",
+                version: "3.9.0",
+                transport: "sse",
+                timestamp: Date.now(),
+            });
+        });
+
+        // SSE endpoint - establishes connection
+        app.get("/sse", async (req, res) => {
+            const authHeader = req.headers.authorization;
+            const apiKey = authHeader?.replace("Bearer ", "");
+
+            // 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 <key>. " +
+                           "Accepted formats: gd_xxx (platform) or AIza... (Google API key)",
+                });
+            }
+
+            // Set API_KEY env var for this request (tools read from process.env)
+            process.env.API_KEY = apiKey;
+
+            // Create SSE transport
+            const transport = new SSEServerTransport("/messages", res);
+            const sessionId = `sess_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+            transports.set(sessionId, transport);
+
+            // Send session ID to client
+            res.setHeader("mcp-session-id", sessionId);
+
+            // Clean up on disconnect
+            res.on("close", () => {
+                transports.delete(sessionId);
+            });
+
+            await server.connect(transport);
+        });
+
+        // Messages endpoint - receives JSON-RPC from client
+        app.post("/messages", async (req, res) => {
+            const sessionId = req.query.sessionId || req.headers["mcp-session-id"];
+            const transport = transports.get(sessionId);
+
+            if (!transport) {
+                return res.status(400).json({ error: "Invalid or expired session" });
+            }
+
+            await transport.handlePostMessage(req, res);
+        });
+
+        const port = parseInt(process.env.PORT || "3000", 10);
+        app.listen(port, "0.0.0.0", () => {
+            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`);
+        });
+    } else {
+        // Stdio mode (default, for local IDE usage)
+        const transport = new StdioServerTransport();
+        await server.connect(transport);
+        console.error("gemini-design-mcp v3.9.0 running on stdio");
+    }
 }
 main().catch((error) => {
     console.error("Fatal error:", 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.7.2",
+  "version": "3.9.0",
   "description": "MCP server that uses Gemini 3 Pro for frontend/design code generation",
   "main": "build/index.js",
   "bin": {
@@ -11,6 +11,7 @@
     "build": "tsc",
     "dev": "tsc --watch",
     "start": "node build/index.js",
+    "start:sse": "TRANSPORT=sse node build/index.js",
     "prepublishOnly": "echo 'Publishing...'"
   },
   "keywords": [
@@ -28,10 +29,14 @@
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.0",
     "@google/genai": "^1.33.0",
-    "zod": "^3.23.0"
+    "zod": "^3.23.0",
+    "express": "^4.21.0",
+    "cors": "^2.8.5"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",
+    "@types/express": "^4.17.21",
+    "@types/cors": "^2.8.17",
     "typescript": "^5.0.0"
   },
   "engines": {