@yetter/client 0.0.12 → 0.0.13

package/dist/index.d.ts

@@ -1,4 +1,4 @@
 export { YetterImageClient } from "./api.js";
 export * from "./types.js";
-export { yetter } from "./client.js";
+export { yetter, YetterClient } from "./client.js";
 export type { UploadOptions, GetUploadUrlRequest, GetUploadUrlResponse, UploadCompleteRequest, UploadCompleteResponse, } from "./types.js";

package/dist/index.js

@@ -1,3 +1,3 @@
 export { YetterImageClient } from "./api.js";
 export * from "./types.js";
-export { yetter } from "./client.js";
+export { yetter, YetterClient } from "./client.js";

package/dist/types.d.ts

@@ -58,6 +58,7 @@ export interface SubscribeOptions {
     input: Omit<GenerateRequest, 'model'>;
     logs?: boolean;
     onQueueUpdate?: (update: GetStatusResponse) => void;
+    pollIntervalMs?: number;
 }
 export interface SubmitQueueOptions {
     input: Omit<GenerateRequest, 'model'>;
@@ -96,6 +97,8 @@ export interface YetterStream extends AsyncIterable<StreamEvent> {
 export interface UploadOptions {
     /** Optional callback for upload progress (0-100) */
     onProgress?: (progress: number) => void;
+    /** Optional upload timeout in milliseconds (default: 5 minutes) */
+    timeout?: number;
     /** Optional custom filename (browser File uploads) */
     filename?: string;
 }

package/package.json

@@ -1,9 +1,10 @@
 {
   "name": "@yetter/client",
-  "version": "0.0.12",
+  "version": "0.0.13",
   "type": "module",
   "scripts": {
     "build": "tsc",
+    "test": "npm run build && node --test tests/*.test.mjs",
     "test:submit": "node --loader ts-node/esm examples/submit.ts",
     "test:subscribe": "node --loader ts-node/esm examples/subscribe.ts",
     "test:stream": "node --loader ts-node/esm examples/stream.ts",
@@ -11,6 +12,10 @@
   },
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "README.md"
+  ],
   "exports": {
     ".": {
       "import": "./dist/index.js",

package/examples/stream.ts

@@ -1,54 +0,0 @@
-import { yetter } from "../src/client.js";
-
-async function main() {
-    const model = "ytr-ai/flux/dev";
-    console.log("\n--- Starting Stream Test ---");
-    console.time("Stream Test");
-    let streamRequestId = "";
-
-    try {
-        const streamInstance = await yetter.stream(model, {
-            input: {
-                prompt: "a bioluminescent forest at night, fantasy art",
-                num_inference_steps: 28,
-            },
-        });
-        streamRequestId = streamInstance.getRequestId();
-        console.log(`[${new Date().toLocaleTimeString()}] Stream initiated for Request ID: ${streamRequestId}`);
-
-        // Iterate over stream events
-        for await (const event of streamInstance) {
-            console.log(`[${new Date().toLocaleTimeString()}][STREAM EVENT - ${streamRequestId}] Status: ${event.status}, QPos: ${event.queue_position}`);
-            if (event.logs && event.logs.length > 0) {
-                console.log(`  Logs for ${streamRequestId}:`);
-                event.logs.forEach(log => console.log(`    - ${log.message}`));
-            }
-        }
-        console.log(`[${new Date().toLocaleTimeString()}] Stream for ${streamRequestId} finished iterating events.`);
-
-        // Wait for the final result from the done() method
-        console.log(`[${new Date().toLocaleTimeString()}] Waiting for done() on stream for ${streamRequestId}...`);
-        const result = await streamInstance.done();
-        console.log("\n--- Stream Test Final Result ---");
-        console.log(`Request ID: ${streamRequestId}`);
-        if (result.images && result.images.length > 0) {
-            console.log("Generated Images:", result.images);
-        } else {
-            console.log("No images in final result.");
-        }''
-        console.log("Original Prompt:", result.prompt);
-        console.timeEnd("Stream Test");
-    } catch (err: any) {
-        console.error(`\n--- Stream Test Failed (Request ID: ${streamRequestId || 'UNKNOWN'}) ---`);
-        console.error("Error during stream test:", err.message || err);
-        if (err.stack) {
-            console.error(err.stack);
-        }
-        process.exit(1);
-    }
-}
-
-main().catch(err => {
-    console.error("Unhandled error in main:", err);
-    process.exit(1);
-}); 

package/examples/submit.ts

@@ -1,80 +0,0 @@
-import { yetter } from "../src/client.js";
-
-async function main() {
-    const model = "ytr-ai/flux/dev";
-    // Queue Submit test
-    try {
-        console.log("\n--- Starting Queue Submit Test ---");
-        const { request_id, status, queue_position } = await yetter.queue.submit(model, {
-            input: {
-                prompt: "a fluffy white kitten playing with a yarn ball",
-                num_inference_steps: 28,
-            },
-        });
-
-        console.log("\n--- Queue Submit Test Result ---");
-        console.log("Request ID:", request_id);
-        console.log("Status:", status);
-        console.log("Queue Position:", queue_position);
-
-        if (request_id) {
-            console.log(`\n--- Polling for status of Request ID: ${request_id} (3-minute timeout) ---`);
-            
-            const startTime = Date.now();
-            const timeoutMilliseconds = 3 * 60 * 1000; // 3 minutes
-            const pollIntervalMilliseconds = 10 * 1000; // Poll every 10 seconds
-            let success = false;
-
-            while (Date.now() - startTime < timeoutMilliseconds) {
-                try {
-                    const statusResult = await yetter.queue.status(model, {
-                        requestId: request_id,
-                    });
-                    const currentStatus = statusResult.data.status;
-                    console.log(`[${new Date().toLocaleTimeString()}] Request ${request_id} status: ${currentStatus}, Queue Position: ${statusResult.data.queue_position}`);
-
-                    if (currentStatus === "COMPLETED") {
-                        console.log("Status is COMPLETED. Fetching final result...");
-                        const finalResult = await yetter.queue.result(model, {
-                            requestId: request_id,
-                        });
-                        console.log("\n--- Get Result Test Succeeded ---");
-                        console.log("Request ID:", finalResult.requestId);
-                        console.log("Image Data:", finalResult.data.images);
-                        console.log("Prompt:", finalResult.data.prompt);
-                        success = true;
-                        break; // Exit loop on success
-                    } else if (currentStatus === "ERROR") {
-                        console.error(`Request ${request_id} FAILED. Logs:`, statusResult.data.logs);
-                        break; // Exit loop on failure
-                    }
-                    // For other statuses (e.g., IN_QUEUE, IN_PROGRESS), continue polling
-
-                } catch (pollError: any) {
-                    console.error(`Error during status polling for ${request_id} (will retry):`, pollError.message || pollError);
-                }
-
-                if (Date.now() - startTime + pollIntervalMilliseconds >= timeoutMilliseconds) {
-                    if (!success) console.error("Timeout approaching, last poll cycle completed or error occurred.");
-                    break; // Don't wait if the next interval exceeds timeout
-                }
-                await new Promise(resolve => setTimeout(resolve, pollIntervalMilliseconds));
-            }
-
-            if (!success) {
-                console.error(`\n--- Polling Timed Out or Failed for Request ID: ${request_id} ---`);
-                console.error(`Failed to get a COMPLETED status for ${request_id} within 3 minutes.`);
-            }
-        }
-
-    } catch (err: any) {
-        console.error("\n--- Queue Submit Test Failed ---");
-        console.error("Error during queue submit:", err.message || err);
-        if (err.stack) {
-            console.error(err.stack);
-        }
-        process.exit(1);
-    }
-}
-
-main();

package/examples/subscribe.ts

@@ -1,41 +0,0 @@
-import { yetter } from "../dist/index.js";
-
-async function main() {
-    const model = "ytr-ai/flux/dev";
-    // Subscribe test
-    try {
-        console.log("\n--- Starting Subscribe Test ---");
-        console.time("Subscribe Test");
-        const result = await yetter.subscribe(model, {
-            input: {
-                prompt: "a vibrant coral reef, underwater photography",
-                num_inference_steps: 28,
-            },
-            logs: true,
-            onQueueUpdate: (update) => {
-                console.log(`[Queue Update] Status: ${update.status}, Position: ${update.queue_position}`);
-                if (update.status === "IN_PROGRESS" && update.logs) {
-                    console.log("Logs:");
-                    update.logs.map((log) => log.message).forEach(logMessage => console.log(`  - ${logMessage}`));
-                } else if (update.status === "COMPLETED") {
-                    console.log("Processing completed!");
-                } else if (update.status === "ERROR") {
-                    console.error("Processing failed. Logs:", update.logs);
-                }
-            },
-        });
-
-        console.log("\n--- Subscribe Test Result ---");
-        console.log("Results:", result);
-        console.timeEnd("Subscribe Test");
-    } catch (err: any) {
-        console.error("\n--- Subscribe Test Failed ---");
-        console.error("Error during subscribe:", err.message || err);
-        if (err.stack) {
-            console.error(err.stack);
-        }
-        process.exit(1);
-    }
-}
-
-main();

package/examples/upload-and-generate.ts

@@ -1,39 +0,0 @@
-import { yetter } from "../src/client.js";
-
-async function main() {
-    // Configure with API key
-    yetter.configure({
-        apiKey: process.env.YTR_API_KEY || "your_api_key_here"
-    });
-
-    console.log("Step 1: Uploading input image...");
-
-    // Replace with your actual image path
-    const imagePath = "./test-image.jpg";
-
-    try {
-        const uploadResult = await yetter.uploadFile(imagePath, {
-            onProgress: (pct) => console.log(`  Upload progress: ${pct}%`),
-        });
-        console.log(`✓ Uploaded: ${uploadResult.url}\n`);
-
-        console.log("Step 2: Generating with uploaded image...");
-        const genResult = await yetter.subscribe("ytr-ai/qwen/image-edit/i2i", {
-            input: {
-                prompt: "Transform to watercolor painting style",
-                image_url: [uploadResult.url],
-                num_inference_steps: 28,
-            },
-            onQueueUpdate: (status) => {
-                console.log(`  Status: ${status.status}, Queue: ${status.queue_position}`);
-            },
-        });
-
-        console.log("\n✓ Generation complete!");
-        console.log("Generated images:", genResult.images);
-    } catch (error: any) {
-        console.error("Error:", error.message);
-    }
-}
-
-main().catch(console.error);

package/src/api.ts

@@ -1,159 +0,0 @@
-import fetch from "cross-fetch";
-import {
-    ClientOptions,
-    GenerateRequest,
-    GenerateResponse,
-    GetStatusRequest,
-    GetStatusResponse,
-    CancelRequest,
-    CancelResponse,
-    GetResponseRequest,
-    GetResponseResponse,
-    GetUploadUrlRequest,
-    GetUploadUrlResponse,
-    UploadCompleteRequest,
-    UploadCompleteResponse,
-} from "./types";
-
-export class YetterImageClient {
-    private apiKey: string;
-    private endpoint: string;
-
-    constructor(options: ClientOptions) {
-        if (!options.apiKey) {
-            throw new Error("`apiKey` is required");
-        }
-        this.apiKey = options.apiKey;
-        this.endpoint = options.endpoint || "https://api.yetter.ai";
-    }
-
-    public getApiEndpoint(): string {
-        return this.endpoint;
-    }
-
-    public async generate(
-        body: GenerateRequest
-    ): Promise<GenerateResponse> {
-        const url = `${this.endpoint}/${body.model}`;
-        const res = await fetch(url, {
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: this.apiKey,
-            },
-            body: JSON.stringify(body),
-        });
-        
-        if (!res.ok) {
-            const errorText = await res.text();
-            throw new Error(`API error (${res.status}): ${errorText}`);
-        }
-        
-        return (await res.json()) as GenerateResponse;
-    }
-    
-    public async getStatus(body: GetStatusRequest): Promise<GetStatusResponse> {
-        const url = new URL(body.url);
-        if (body.logs) {
-            url.searchParams.append('logs', '1');
-        }
-        
-        const res = await fetch(url.toString(), {
-            method: "GET",
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: this.apiKey,
-            },
-        });
-        
-        if (!res.ok) {
-            const errorText = await res.text();
-            throw new Error(`API error (${res.status}): ${errorText}`);
-        }
-        
-        return (await res.json()) as GetStatusResponse;
-    }
-    
-    public async cancel(body: CancelRequest): Promise<CancelResponse> {
-        const res = await fetch(body.url, {
-            method: "PUT",
-            headers: {
-                "Content-Type": "application/json", 
-                Authorization: this.apiKey,
-            },
-        });
-        
-        if (!res.ok) {
-            const errorText = await res.text();
-            throw new Error(`API error (${res.status}): ${errorText}`);
-        }
-        
-        return (await res.json()) as CancelResponse;
-    }
-
-    public async getResponse(body: GetResponseRequest): Promise<GetResponseResponse> {
-        const res = await fetch(body.url, {
-            method: "GET",
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: this.apiKey,
-            },
-        });
-
-        if (!res.ok) {
-            const errorText = await res.text();
-            throw new Error(`API error (${res.status}): ${errorText}`);
-        }
-        return (await res.json()) as GetResponseResponse;
-    }
-
-    /**
-     * Request presigned URL(s) for file upload
-     * @param body Upload request parameters
-     * @returns Presigned URL response with mode (single/multipart)
-     */
-    public async getUploadUrl(
-        body: GetUploadUrlRequest
-    ): Promise<GetUploadUrlResponse> {
-        const res = await fetch(`${this.endpoint}/uploads`, {
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: this.apiKey,
-            },
-            body: JSON.stringify(body),
-        });
-
-        if (!res.ok) {
-            const errorText = await res.text();
-            throw new Error(`Upload URL request failed (${res.status}): ${errorText}`);
-        }
-
-        return (await res.json()) as GetUploadUrlResponse;
-    }
-
-    /**
-     * Notify server that upload is complete
-     * @param body Completion request with S3 key
-     * @returns Uploaded file metadata with public URL
-     */
-    public async uploadComplete(
-        body: UploadCompleteRequest
-    ): Promise<UploadCompleteResponse> {
-        const res = await fetch(`${this.endpoint}/uploads/complete`, {
-            method: "POST",
-            headers: {
-                "Content-Type": "application/json",
-                Authorization: this.apiKey,
-            },
-            body: JSON.stringify(body),
-        });
-
-        if (!res.ok) {
-            const errorText = await res.text();
-            throw new Error(`Upload completion failed (${res.status}): ${errorText}`);
-        }
-
-        return (await res.json()) as UploadCompleteResponse;
-    }
-}