@yetter/client 0.0.6 → 0.0.7

package/README.md

@@ -61,6 +61,7 @@ async function main() {
         const result = await yetter.subscribe(model, {
             input: {
                 prompt: "a vibrant coral reef, underwater photography",
+                num_inference_steps: 28, // Required
             },
             logs: true,
             onQueueUpdate: (update) => {
@@ -120,6 +121,7 @@ async function main() {
         const streamInstance = await yetter.stream(model, {
             input: {
                 prompt: "a bioluminescent forest at night, fantasy art",
+                num_inference_steps: 28, // Required
             },
         });
         streamRequestId = streamInstance.getRequestId();
@@ -189,6 +191,7 @@ async function main() {
         const { request_id } = await yetter.queue.submit(model, {
             input: {
                 prompt: "a fluffy white kitten playing with a yarn ball",
+                num_inference_steps: 28, // Required
             },
         });
         console.log("Request ID:", request_id);

package/dist/types.d.ts

@@ -3,11 +3,17 @@ export interface ClientOptions {
     endpoint?: string;
     is_bearer?: boolean;
 }
+export interface LoraWeight {
+    path: string;
+    scale?: number;
+}
 export interface GenerateImageRequest {
     model: string;
     prompt: string;
-    num_inference_steps?: number;
+    num_inference_steps: number;
     seed?: number;
+    loras?: LoraWeight[];
+    image_url?: string;
 }
 export interface GenerateImageResponse {
     status: string;

package/examples/stream.ts

@@ -3,14 +3,14 @@ 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",
-                seed: 42
-                // You can add other parameters like seed or num_inference_steps here
+                num_inference_steps: 28,
             },
         });
         streamRequestId = streamInstance.getRequestId();
@@ -37,7 +37,7 @@ async function main() {
             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);

package/examples/submit.ts

@@ -8,6 +8,7 @@ async function main() {
         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,
             },
         });
 

package/examples/subscribe.ts

@@ -5,9 +5,11 @@ async function main() {
     // 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) => {
@@ -25,7 +27,7 @@ async function main() {
 
         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);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yetter/client",
-  "version": "0.0.6",
+  "version": "0.0.7",
   "type": "module",
   "scripts": {
     "build": "tsc",

package/src/types.ts

@@ -4,11 +4,18 @@ export interface ClientOptions {
     is_bearer?: boolean;
 }
 
+export interface LoraWeight {
+    path: string;
+    scale?: number;
+}
+
 export interface GenerateImageRequest {
     model: string;
     prompt: string;
-    num_inference_steps?: number;
+    num_inference_steps: number;
     seed?: number;
+    loras?: LoraWeight[];
+    image_url?: string;
 }
 
 export interface GenerateImageResponse {