@yetter/client 0.0.11 → 0.0.13

package/dist/index.d.ts

@@ -1,3 +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'>;
@@ -90,3 +91,65 @@ export interface YetterStream extends AsyncIterable<StreamEvent> {
     cancel(): Promise<void>;
     getRequestId(): string;
 }
+/**
+ * Options for yetter.uploadFile() and yetter.uploadBlob()
+ */
+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;
+}
+/**
+ * Request to get presigned upload URL(s)
+ */
+export interface GetUploadUrlRequest {
+    /** Original filename with extension */
+    file_name: string;
+    /** MIME type (e.g., "image/jpeg") */
+    content_type: string;
+    /** File size in bytes */
+    size: number;
+}
+/**
+ * Response containing presigned URL(s) for upload
+ */
+export interface GetUploadUrlResponse {
+    /** Upload mode: "single" for small files, "multipart" for large files */
+    mode: "single" | "multipart";
+    /** S3 object key for tracking */
+    key: string;
+    /** Presigned PUT URL (single mode only) */
+    put_url?: string;
+    /** Size of each part in bytes (multipart mode only) */
+    part_size?: number;
+    /** Array of part URLs with part numbers (multipart mode only) */
+    part_urls?: Array<{
+        part_number: number;
+        url: string;
+    }>;
+}
+/**
+ * Request to notify upload completion
+ */
+export interface UploadCompleteRequest {
+    /** S3 object key from GetUploadUrlResponse */
+    key: string;
+}
+/**
+ * Response after successful upload completion
+ */
+export interface UploadCompleteResponse {
+    /** Public URL to access the uploaded file */
+    url: string;
+    /** S3 object key */
+    key: string;
+    /** Optional metadata */
+    metadata?: {
+        size: number;
+        content_type: string;
+        uploaded_at?: string;
+    };
+}

package/package.json

@@ -1,9 +1,10 @@
 {
   "name": "@yetter/client",
-  "version": "0.0.11",
+  "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",
@@ -23,9 +28,11 @@
   "description": "",
   "dependencies": {
     "@types/eventsource": "^1.1.15",
+    "@types/mime-types": "^3.0.1",
     "cross-fetch": "^4.1.0",
     "event-source-polyfill": "^1.0.31",
-    "eventsource": "^4.0.0"
+    "eventsource": "^4.0.0",
+    "mime-types": "^3.0.2"
   },
   "devDependencies": {
     "@types/event-source-polyfill": "^1.0.5",

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 === "FAILED") {
-                        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 === "FAILED") {
-                    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/src/api.ts

@@ -1,105 +0,0 @@
-import fetch from "cross-fetch";
-import {
-    ClientOptions,
-    GenerateRequest,
-    GenerateResponse,
-    GetStatusRequest,
-    GetStatusResponse,
-    CancelRequest,
-    CancelResponse,
-    GetResponseRequest,
-    GetResponseResponse,
-} 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;
-    }
-}