Generations

flowchart LR
    subgraph input [Input]
        Prompt["Prompt / Image URL"]
        Ref["refType + refId"]
    end

    subgraph pipeline [Pipeline]
        Create["1. Create"]
        Process["2. Processing"]
        Complete["3. Complete"]
        Error["3. Error"]
    end

    Prompt --> Create
    Ref --> Create
    Create --> Process
    Process --> Complete
    Process --> Error

import { PUClient } from 'pu-client';

const client = new PUClient({ apiKey: process.env.PU_API_KEY! });

const id = await client.generations.create({
  type: 'character',
  input: { prompt: 'A wise owl wizard in robes' },
});

const { events, unsubscribe } = client.generations.listen(id);

events.addEventListener('change', (e: MessageEvent) => {
  const detail = e.data;
  console.log('Progress:', detail);
});

events.addEventListener('error', (e: MessageEvent) => {
  console.error('Stream error:', e.data);
});

// When done, stop listening
unsubscribe();

const channel = client.generations.subscribe({ refType: 'npc', refId: 'my-npc-id' });

channel.addEventListener('update', (e: MessageEvent) => {
  const { generation } = e.data;
  console.log(`${generation.type} ${generation.id}: ${generation.state}`);
});

// Unsubscribe when done
channel.unsubscribe();

const id = await client.generations.create({
  type: 'sceneImage',
  input: {
    prompt: 'cozy cabin in snow',
    smartPrompt: true,
  },
});

const id = await client.generations.create({
  type: 'character',
  input: { prompt: 'Merchant with a satchel' },
  refType: 'npc',
  refId: npc.id,
});

flowchart TB
    subgraph client [Client]
        SDK["pu-client SDK"]
        WebApp["Web App /generations"]
    end

    subgraph api [API]
        CreateAPI["POST /api/tasks/generationNext/{type}"]
        EventsAPI["GET /api/generations/events"]
        LogsAPI["GET /api/generations/{id}/logs"]
    end

    subgraph workers [AI Workers]
        CharWorker["Character Pipeline"]
        SceneWorker["Scene Pipelines"]
        AudioWorker["Audio Pipelines"]
        ImageWorker["Image Pipelines"]
    end

    subgraph storage [Storage]
        Assets["Asset Storage"]
    end

    SDK --> CreateAPI
    WebApp --> CreateAPI
    SDK --> EventsAPI
    SDK --> LogsAPI
    CreateAPI --> CharWorker
    CreateAPI --> SceneWorker
    CreateAPI --> AudioWorker
    CreateAPI --> ImageWorker

    CharWorker --> Assets
    SceneWorker --> Assets
    AudioWorker --> Assets
    ImageWorker --> Assets

    CharWorker -.->|"emit"| EventsAPI
    SceneWorker -.->|"emit"| EventsAPI
    AudioWorker -.->|"emit"| EventsAPI
    ImageWorker -.->|"emit"| EventsAPI

import { PUClient } from 'pu-client';

const client = new PUClient({ apiKey: process.env.PU_API_KEY! });

const id = await client.generations.create({
  type: 'character',
  input: {
    prompt: 'A steampunk fox inventor with goggles',
    smartPrompt: true,
  },
});

console.log(`Generation started: ${id}`);

const id = await client.generations.create({
  type: 'sceneSplat',
  input: {
    splatUrl: 'https://example.com/reference-room.jpg',
  },
});

const id = await client.generations.create({
  type: 'music',
  input: {
    prompt: 'Ambient forest with soft piano, 2 minutes',
    musicLengthMs: 120000,
  },
});

const id = await client.generations.create({
  type: 'video',
  input: {
    prompt: 'A dragon flying over a mountain at sunset',
  },
});

const generations = await client.generations.list();
const failed = generations.filter(g => g.state === 'error');

for (const g of failed) {
  const newId = await client.generations.retry(g.id);
  console.log(`Retrying ${g.id} as ${newId}`);
}