Labsco
google-labs-code logo

stitch-sdk-usage

1,700

by google-labs-code · part of google-labs-code/stitch-sdk

stitch-sdk-usage — an installable skill for AI agents, published by google-labs-code/stitch-sdk.

🔥🔥FreeQuick setup
🧰 Not standalone. This skill ships with google-labs-code/stitch-sdk and only works together with that tool — install the tool first, then add this skill.

This is the playbook your agent receives when the skill activates — you don't need to read it to use the skill, but it's here to audit before installing.


name: stitch-sdk-usage description: Use the Stitch SDK to generate, edit, and iterate on UI screens from text prompts, manage projects, and retrieve screen HTML/images. Use when the user wants to consume the SDK in their application.

Using the Stitch SDK

The Stitch SDK provides a TypeScript interface for Google Stitch, an AI-powered UI generation service.

Environment Variables

export STITCH_API_KEY="your-api-key"

Working with Projects

import { stitch } from "@google/stitch-sdk";

// List all projects
const projects = await stitch.projects();

// Reference a project by ID (no network call)
const project = stitch.project("4044680601076201931");

// Create a new project
const newProject = await stitch.createProject("My App");

Design Systems

// Create a design system for a project
const ds = await project.createDesignSystem({ displayName: "My Theme" });

// List design systems
const systems = await project.listDesignSystems();

// Reference by ID (no network call)
const dsRef = project.designSystem("existing-asset-id");

// Update a design system
const updated = await ds.update({ displayName: "Updated Theme" });

// Apply to screens (requires SelectedScreenInstance objects from project.data.screenInstances)
const screens = await ds.apply([
  { id: "instance-id", sourceScreen: "projects/123/screens/456" },
]);

Uploading Images

Upload an existing image file (PNG, JPG, JPEG, WEBP) to create a screen directly from a mockup or asset.

import { stitch } from "@google/stitch-sdk";

const project = stitch.project("your-project-id");

// Upload a local image file
const [screen] = await project.uploadImage("./mockup.png", {
  title: "Home Screen",
});

console.log(screen.id);
const html = await screen.getHtml();
const imageUrl = await screen.getImage();

The method reads the file from disk and posts it directly to the Stitch REST API — no output token constraints apply (unlike agent-driven MCP calls).

Supported formats: .png, .jpg, .jpeg, .webp

Options:

OptionTypeDefaultDescription
titlestringDisplay title for the created screen
createScreenInstancesbooleantrueWhether to add the screen to the canvas

Throws StitchError with codes: NOT_FOUND (file not found), UNKNOWN_ERROR (unsupported format or upload failure), AUTH_FAILED (invalid API key).

Generating and Iterating on Screens

// Generate a new screen from a prompt
const screen = await project.generate(
  "Login page with email and password fields",
);

// Edit an existing screen
const edited = await screen.edit("Make the background dark and add a subtitle");

// Generate variants of a screen
const variants = await screen.variants("Try different color schemes", {
  variantCount: 2,
  creativeRange: "EXPLORE",
  aspects: ["COLOR_SCHEME", "LAYOUT"],
});

Retrieving Screen Assets

// Get screen HTML download URL
const html = await screen.getHtml();

// Get screen screenshot download URL
const imageUrl = await screen.getImage();

Both methods use cached data from the generation response when available, falling back to an API call when needed.

Dynamic Tool Client (for agents)

For agents and orchestration scripts that forward JSON payloads to MCP tools:

import { stitch } from "@google/stitch-sdk";

// Find available tools
const { tools } = await stitch.listTools();
for (const tool of tools) {
  console.log(`${tool.name}: ${tool.description}`);
}
// Call a tool with a JSON payload
const result = await stitch.callTool("generate_screen_from_text", {
  projectId: "123",
  prompt: "A login page",
});
await stitch.close();

Error Handling

All SDK methods throw StitchError on failure. Use try/catch:

import { stitch, StitchError } from "@google/stitch-sdk";

try {
  const project = stitch.project("bad-id");
  await project.screens();
} catch (e) {
  if (e instanceof StitchError) {
    console.log(e.code); // "AUTH_FAILED", "NOT_FOUND", etc.
    console.log(e.message); // Human-readable description
    console.log(e.recoverable); // Whether retrying might succeed
  }
}

Error codes: AUTH_FAILED, NOT_FOUND, PERMISSION_DENIED, RATE_LIMITED, NETWORK_ERROR, VALIDATION_ERROR, UNKNOWN_ERROR

API Reference

Stitch Class

MethodReturnsDescription
createProject(title)Promise<Project>Create a new project
projects()Promise<Project[]>List all projects
project(id)ProjectReference a project by ID (no network call)

Project Class

MethodReturnsDescription
generate(prompt, deviceType?)Promise<Screen>Generate a screen from a text prompt
screens()Promise<Screen[]>List all screens in the project
getScreen(screenId)Promise<Screen>Retrieve a specific screen by ID
uploadImage(filePath, opts?)Promise<Screen[]>Upload an image file and create a screen from it
createDesignSystem(designSystem)Promise<DesignSystem>Create a design system for this project
listDesignSystems()Promise<DesignSystem[]>List all design systems
designSystem(id)DesignSystemReference by ID (no API call)

deviceType: "MOBILE" | "DESKTOP" | "TABLET" | "AGNOSTIC"

uploadImage supported formats: .png .jpg .jpeg .webp

DesignSystem Class

MethodReturnsDescription
update(designSystem)Promise<DesignSystem>Update the design system's theme
apply(selectedScreenInstances)Promise<Screen[]>Apply this design system to screens

Screen Class

MethodReturnsDescription
getHtml()Promise<string>Get the screen's HTML download URL
getImage()Promise<string>Get the screen's screenshot download URL
edit(prompt, deviceType?, modelId?)Promise<Screen>Edit the screen using a text prompt
variants(prompt, options, deviceType?, modelId?)Promise<Screen[]>Generate variants of the screen

modelId: "GEMINI_3_PRO" | "GEMINI_3_FLASH"

StitchToolClient (for agents)

MethodReturnsDescription
callTool(name, args)Promise<T>Call any MCP tool by name
listTools()Promise<Tools>Discover available tools
connect()Promise<void>Establish MCP connection (auto-called by callTool)
close()Promise<void>Close the connection

Explicit Configuration

import { Stitch, StitchToolClient } from "@google/stitch-sdk";

const client = new StitchToolClient({
  apiKey: "your-api-key",
  baseUrl: "https://stitch.googleapis.com/mcp",
  timeout: 300_000,
});

const sdk = new Stitch(client);
const projects = await sdk.projects();
OptionEnv VariableDescription
apiKeySTITCH_API_KEYAPI key for authentication
accessTokenSTITCH_ACCESS_TOKENOAuth access token
projectIdGOOGLE_CLOUD_PROJECTGCP project ID (required with OAuth)
baseUrlMCP server URL (default: https://stitch.googleapis.com/mcp)
timeoutRequest timeout in ms (default: 300000)