@kvasar/google-stitch 0.1.6 → 0.1.7

package/openclaw.plugin.json

@@ -1,7 +1,7 @@
 {
   "id": "openclaw-google-stitch",
    "name": "Google Stitch MCP",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "Integrates Google Stitch MCP services into OpenClaw",
    "skills": ["skills"],
   "configSchema": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kvasar/google-stitch",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "OpenClaw plugin for Google Stitch UI generation, screen design, variants, and design systems",
   "type": "module",
   "main": "./src/index.ts",
@@ -40,11 +40,12 @@
     ]
   },
   "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.29.0",
     "@sinclair/typebox": "^0.32.0"
   },
   "devDependencies": {
-    "typescript": "^5.4.0",
     "@types/node": "^22.0.0",
+    "typescript": "^5.4.0",
     "vitest": "^1.6.0"
   },
   "keywords": [

package/src/services/stitch-mcp-client.ts

@@ -1,48 +1,44 @@
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+
 export class StitchMCPClient {
+  private client: Client;
+  private transport: StreamableHTTPClientTransport;
+  private connected = false;
+
   constructor(
     private config: { apiKey: string; endpoint: string }
   ) {
-    if (!config.apiKey) {
-      throw new Error("Missing Stitch apiKey");
-    }
-
-    if (!config.endpoint) {
-      throw new Error("Missing Stitch endpoint");
-    }
-  }
-
-  private async request(path: string, body?: unknown) {
-    const cleanPath = path.startsWith("/")
-      ? path.slice(1)
-      : path;
+    this.client = new Client({
+      name: "openclaw-google-stitch",
+      version: "1.0.0",
+    });
 
-    const response = await fetch(
-      `${this.config.endpoint}/${cleanPath}`,
+    this.transport = new StreamableHTTPClientTransport(
+      new URL(this.config.endpoint),
       {
-        method: "POST",
-        headers: {
-          "Content-Type": "application/json",
-          "X-Goog-Api-Key": this.config.apiKey,
+        requestInit: {
+          headers: {
+            "X-Goog-Api-Key": this.config.apiKey,
+          },
         },
-        body: body
-          ? JSON.stringify(body)
-          : undefined,
       }
     );
+  }
 
-    if (!response.ok) {
-      const errorText = await response.text();
-      throw new Error(
-        `Stitch API error: ${response.status} - ${errorText}`
-      );
+  async connect() {
+    if (!this.connected) {
+      await this.client.connect(this.transport);
+      this.connected = true;
     }
-
-    return response.json();
   }
 
   async generateScreen(prompt: string) {
-    return this.request("generate-screen", {
-      prompt,
+    await this.connect();
+
+    return this.client.callTool({
+      name: "generate_screen_from_text",
+      arguments: { prompt },
     });
   }
 }