gemini-design-mcp 3.2.2 → 3.3.1

package/build/lib/gemini.d.ts

@@ -1,4 +1,4 @@
 import { GoogleGenAI } from "@google/genai";
-export declare const ai: GoogleGenAI;
+export declare const ai: GoogleGenAI | null;
 export declare const DEFAULT_MODEL = "gemini-3-flash-preview";
 export declare function generateWithGemini(systemPrompt: string, userPrompt: string, model?: string): Promise<string>;

package/build/lib/gemini.js

@@ -1,27 +1,81 @@
-import { GoogleGenAI, ThinkingLevel } from "@google/genai";
-const apiKey = process.env.GEMINI_API_KEY;
+import { GoogleGenAI } from "@google/genai";
+const apiKey = process.env.API_KEY;
 if (!apiKey) {
-    console.error("ERROR: GEMINI_API_KEY environment variable is required");
+    console.error("ERROR: API_KEY environment variable is required");
     console.error("Get your API key at: https://aistudio.google.com/app/apikey");
+    console.error("Or use a Gemini Design key (gd_xxx) from: https://gemini-design.com");
     process.exit(1);
 }
-export const ai = new GoogleGenAI({ apiKey });
+// Check if using hosted Gemini Design service
+const isHostedKey = apiKey.startsWith("gd_");
+const PROXY_URL = process.env.GEMINI_DESIGN_PROXY_URL || "https://brave-turtle-253.convex.site/v1/generate";
+// Only initialize SDK if not using hosted service
+export const ai = isHostedKey ? null : new GoogleGenAI({ apiKey });
 // Default model - Gemini 3 Flash Preview (best for design)
 export const DEFAULT_MODEL = "gemini-3-flash-preview";
-export async function generateWithGemini(systemPrompt, userPrompt, model = DEFAULT_MODEL) {
-    try {
-        const response = await ai.models.generateContent({
+/**
+ * Generate content via the hosted proxy service
+ */
+async function generateViaProxy(systemPrompt, userPrompt, model) {
+    const response = await fetch(PROXY_URL, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            "Authorization": `Bearer ${apiKey}`,
+        },
+        body: JSON.stringify({
             model,
-            contents: userPrompt,
-            config: {
-                systemInstruction: systemPrompt,
-                temperature: 1,
-                thinkingConfig: {
-                    thinkingLevel: ThinkingLevel.LOW,
+            contents: [
+                {
+                    role: "user",
+                    parts: [{ text: userPrompt }],
                 },
+            ],
+            systemInstruction: {
+                parts: [{ text: systemPrompt }],
             },
-        });
-        return response.text ?? "";
+            generationConfig: {
+                temperature: 1,
+            },
+        }),
+    });
+    if (!response.ok) {
+        const error = await response.json().catch(() => ({ error: "Unknown error" }));
+        throw new Error(error.error || `Proxy error: ${response.status}`);
+    }
+    const result = await response.json();
+    // Extract text from Gemini response format
+    const text = result.candidates?.[0]?.content?.parts?.[0]?.text;
+    if (!text) {
+        throw new Error("No text in response");
+    }
+    return text;
+}
+/**
+ * Generate content directly via Google SDK
+ */
+async function generateViaSdk(systemPrompt, userPrompt, model) {
+    if (!ai) {
+        throw new Error("SDK not initialized");
+    }
+    const response = await ai.models.generateContent({
+        model,
+        contents: userPrompt,
+        config: {
+            systemInstruction: systemPrompt,
+            temperature: 1,
+        },
+    });
+    return response.text ?? "";
+}
+export async function generateWithGemini(systemPrompt, userPrompt, model = DEFAULT_MODEL) {
+    try {
+        if (isHostedKey) {
+            return await generateViaProxy(systemPrompt, userPrompt, model);
+        }
+        else {
+            return await generateViaSdk(systemPrompt, userPrompt, model);
+        }
     }
     catch (error) {
         const errorMessage = error instanceof Error ? error.message : "Unknown error";

package/package.json

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