import { WiroClient } from '@wiroai/sdk';

const client = new WiroClient({
  apiKey: process.env.WIRO_API_KEY!,
  apiSecret: process.env.WIRO_API_SECRET!,
});

// Submit task
const run = await client.runModel('google/nano-banana-pro', {
  prompt: 'A cute cat wearing an astronaut helmet in space',
  aspectRatio: '16:9',
  resolution: '2K',
});

// Wait for completion
const result = await client.waitForTask(run.socketaccesstoken);

for (const output of result.outputs) {
  console.log(`${output.name}: ${output.url}`);
}

import { getModelRegistry } from '@wiroai/sdk';

const registry = getModelRegistry();

// 76 models available instantly — unknown models auto-fetched on first use
const model = await registry.ensureSpec(client, 'openai/sora-2');

if (model) {
  console.log(model.parameters);        // All parameters with types & defaults
  console.log(model.requiredParameters); // Required parameter names
}

// Validate parameters before running
const errors = registry.validateParams('openai/sora-2', {
  prompt: 'A sunset',
  duration: 'invalid',
});
// → ['Invalid value for "duration": got "invalid", expected one of: 5, 10, 15, 20']

// WebSocket-based monitoring
const unsubscribe = client.connectTaskWebSocket(run.socketaccesstoken, {
  onQueued: () => console.log('Queued...'),
  onProcessing: () => console.log('Processing...'),
  onCompleted: (outputs) => console.log('Done!', outputs),
  onFailed: (error) => console.error('Failed:', error),
  onError: (error) => console.error('WebSocket error:', error),
});

// Call unsubscribe() to disconnect

// Check task status
const detail = await client.getTaskDetail({ tasktoken: token });
const detailById = await client.getTaskDetailById(taskId);

// Kill a running task
await client.killTask(taskId);

// Cancel a queued task
await client.cancelTask(taskId);

import { KNOWN_MODELS, MODEL_CATEGORIES, getModelRegistry } from '@wiroai/sdk';

const registry = getModelRegistry();

// Search models by keyword
const results = registry.search('image');

// Get all models
const all = registry.getAll();

// Filter by category
const imageModels = KNOWN_MODELS.filter(m => m.category === 'text-to-image');

// Available categories
console.log(MODEL_CATEGORIES);
// → ['text-to-image', 'image-to-image', 'text-to-video', 'llm', 'text-to-speech', ...]