gemini-design-mcp 3.2.1 → 3.3.0

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,87 @@
 import { GoogleGenAI, ThinkingLevel } from "@google/genai";
-const apiKey = process.env.GEMINI_API_KEY;
+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,
+            contents: [
+                {
+                    role: "user",
+                    parts: [{ text: userPrompt }],
+                },
+            ],
+            systemInstruction: {
+                parts: [{ text: systemPrompt }],
+            },
+            generationConfig: {
                 temperature: 1,
                 thinkingConfig: {
-                    thinkingLevel: ThinkingLevel.LOW,
+                    thinkingBudget: 1024, // LOW thinking level
                 },
             },
-        });
-        return response.text ?? "";
+        }),
+    });
+    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,
+            thinkingConfig: {
+                thinkingLevel: ThinkingLevel.LOW,
+            },
+        },
+    });
+    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/build/tools/create-frontend.d.ts

@@ -3,7 +3,7 @@ export declare const createFrontendSchema: {
     request: z.ZodString;
     filePath: z.ZodString;
     techStack: z.ZodString;
-    context: z.ZodNullable<z.ZodString>;
+    context: z.ZodOptional<z.ZodString>;
     designSystem: z.ZodOptional<z.ZodObject<{
         vibe: z.ZodObject<{
             name: z.ZodString;
@@ -36,7 +36,7 @@ export declare function createFrontend(params: {
     request: string;
     filePath: string;
     techStack: string;
-    context: string | null;
+    context?: string;
     designSystem?: {
         vibe: {
             name: string;

package/build/tools/create-frontend.js

@@ -9,9 +9,9 @@ export const createFrontendSchema = {
         "Example: 'src/components/PricingPage.tsx' or 'app/dashboard/page.tsx'"),
     techStack: z.string().describe("The tech stack to use. Be specific. " +
         "Examples: 'React + TypeScript + Tailwind CSS', 'Next.js 14 App Router + shadcn/ui', 'Vue 3 + Composition API + CSS Modules'"),
-    context: z.string().nullable().describe("Existing project code for design consistency. " +
+    context: z.string().optional().describe("Existing project code for design consistency. " +
         "Pass relevant files (components, styles, theme config) so Gemini matches your design system. " +
-        "Null only if this is the FIRST file in a new project."),
+        "Omit this parameter if this is the FIRST file in a new project."),
     designSystem: z.object({
         vibe: z.object({
             name: z.string().describe("Vibe name (e.g., 'Pristine Museum', 'Dark Luxe')"),
@@ -42,9 +42,9 @@ Interpret this vibe creatively:
 - You have freedom in the specific implementation, but the overall feel MUST match this vibe
 `;
     }
-    // Build context instructions
+    // Build context instructions (ignore empty strings and "null" strings)
     let contextInstructions = '';
-    if (context) {
+    if (context && context !== "null" && context.trim() !== "") {
         contextInstructions = `
 EXISTING PROJECT CONTEXT (match this design system):
 ${context}

package/package.json

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