@yetter/client 0.0.8 → 0.0.10

package/dist/api.d.ts

@@ -1,10 +1,10 @@
-import { ClientOptions, GenerateImageRequest, GenerateImageResponse, GetStatusRequest, GetStatusResponse, CancelRequest, CancelResponse, GetResponseRequest, GetResponseResponse } from "./types";
+import { ClientOptions, GenerateRequest, GenerateResponse, GetStatusRequest, GetStatusResponse, CancelRequest, CancelResponse, GetResponseRequest, GetResponseResponse } from "./types";
 export declare class YetterImageClient {
     private apiKey;
     private endpoint;
     constructor(options: ClientOptions);
     getApiEndpoint(): string;
-    generateImage(body: GenerateImageRequest): Promise<GenerateImageResponse>;
+    generate(body: GenerateRequest): Promise<GenerateResponse>;
     getStatus(body: GetStatusRequest): Promise<GetStatusResponse>;
     cancel(body: CancelRequest): Promise<CancelResponse>;
     getResponse(body: GetResponseRequest): Promise<GetResponseResponse>;

package/dist/api.js

@@ -10,7 +10,7 @@ export class YetterImageClient {
     getApiEndpoint() {
         return this.endpoint;
     }
-    async generateImage(body) {
+    async generate(body) {
         const url = `${this.endpoint}/${body.model}`;
         const res = await fetch(url, {
             method: "POST",

package/dist/client.d.ts

@@ -1,11 +1,11 @@
-import { ClientOptions, GetResponseResponse, SubscribeOptions, GenerateImageResponse, SubmitQueueOptions, GetResultOptions, GetResultResponse, StatusOptions, StatusResponse, StreamOptions, YetterStream } from "./types.js";
+import { ClientOptions, GetResponseResponse, SubscribeOptions, GenerateResponse, SubmitQueueOptions, GetResultOptions, GetResultResponse, StatusOptions, StatusResponse, StreamOptions, YetterStream } from "./types.js";
 export declare class yetter {
     private static apiKey;
     private static endpoint;
     static configure(options: ClientOptions): void;
     static subscribe(model: string, options: SubscribeOptions): Promise<GetResponseResponse>;
     static queue: {
-        submit: (model: string, options: SubmitQueueOptions) => Promise<GenerateImageResponse>;
+        submit: (model: string, options: SubmitQueueOptions) => Promise<GenerateResponse>;
         status: (model: string, options: StatusOptions) => Promise<StatusResponse>;
         result: (model: string, options: GetResultOptions) => Promise<GetResultResponse>;
     };

package/dist/client.js

@@ -24,7 +24,7 @@ export class yetter {
             apiKey: _a.apiKey,
             endpoint: _a.endpoint
         });
-        const generateResponse = await client.generateImage({
+        const generateResponse = await client.generate({
             model: model,
             ...options.input,
         });
@@ -76,10 +76,11 @@ export class yetter {
             apiKey: _a.apiKey,
             endpoint: _a.endpoint
         });
-        const initialApiResponse = await client.generateImage({
+        const initialApiResponse = await client.generate({
             model: model,
             ...options.input,
         });
+        console.timeEnd("Initial API Response");
         const requestId = initialApiResponse.request_id;
         const responseUrl = initialApiResponse.response_url;
         const cancelUrl = initialApiResponse.cancel_url;
@@ -169,13 +170,14 @@ export class yetter {
             }
         };
         eventSource = new EventSourcePolyfill(sseStreamUrl, {
-            headers: { 'Authorization': `${_a.apiKey}` }
+            headers: { 'Authorization': `${_a.apiKey}` },
+            heartbeatTimeout: 3000,
         });
         eventSource.onopen = (event) => {
             console.log("SSE Connection Opened:", event);
         };
         eventSource.addEventListener('data', (event) => {
-            console.log("SSE 'data' event received, raw data:", event.data);
+            // console.log("SSE 'data' event received, raw data:", event.data);
             try {
                 const statusData = JSON.parse(event.data);
                 controller.push(statusData);
@@ -186,12 +188,18 @@ export class yetter {
             }
         });
         eventSource.addEventListener('done', (event) => {
-            console.log("SSE 'done' event received, raw data:", event.data);
+            // console.log("SSE 'done' event received, raw data:", event.data);
             controller.close();
         });
         eventSource.onerror = (err) => {
-            console.error("SSE Connection Error (onerror):", err);
-            controller.error(err);
+            var _b;
+            const message = ((_b = err === null || err === void 0 ? void 0 : err.error) === null || _b === void 0 ? void 0 : _b.message) || (err === null || err === void 0 ? void 0 : err.message) || '';
+            const isIdleTimeout = typeof message === 'string' && message.includes('No activity within') && message.includes('Reconnecting');
+            if (isIdleTimeout) {
+                console.warn("SSE idle timeout; letting EventSource auto-reconnect.", err);
+                return;
+            }
+            console.warn("SSE Connection Error (onerror) - will allow auto-reconnect:", err);
         };
         // Handle if API immediately returns a terminal status in initialApiResponse (e.g. already completed/failed)
         if (initialApiResponse.status === "COMPLETED") {
@@ -240,7 +248,7 @@ yetter.queue = {
             apiKey: _a.apiKey,
             endpoint: _a.endpoint
         });
-        const generateResponse = await client.generateImage({
+        const generateResponse = await client.generate({
             model: model,
             ...options.input,
         });

package/dist/types.d.ts

@@ -2,15 +2,16 @@ export interface ClientOptions {
     apiKey: string;
     endpoint?: string;
     is_bearer?: boolean;
+    promptEnhancerModel?: string;
 }
 export interface LoraWeight {
     path: string;
     scale?: number;
 }
-export interface GenerateImageRequest {
+export interface GenerateRequest {
     [key: string]: any;
 }
-export interface GenerateImageResponse {
+export interface GenerateResponse {
     status: string;
     request_id: string;
     response_url: string;
@@ -54,12 +55,12 @@ export interface GetResponseResponse {
     prompt: string;
 }
 export interface SubscribeOptions {
-    input: Omit<GenerateImageRequest, 'model'>;
+    input: Omit<GenerateRequest, 'model'>;
     logs?: boolean;
     onQueueUpdate?: (update: GetStatusResponse) => void;
 }
 export interface SubmitQueueOptions {
-    input: Omit<GenerateImageRequest, 'model'>;
+    input: Omit<GenerateRequest, 'model'>;
 }
 export interface GetResultOptions {
     requestId: string;
@@ -76,7 +77,7 @@ export interface StatusResponse {
     requestId: string;
 }
 export interface StreamOptions {
-    input: Omit<GenerateImageRequest, 'model'>;
+    input: Omit<GenerateRequest, 'model'>;
 }
 /**
  * Represents an event received from the SSE stream.

package/package.json

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

package/src/api.ts

@@ -1,8 +1,8 @@
 import fetch from "cross-fetch";
 import {
     ClientOptions,
-    GenerateImageRequest,
-    GenerateImageResponse,
+    GenerateRequest,
+    GenerateResponse,
     GetStatusRequest,
     GetStatusResponse,
     CancelRequest,
@@ -27,9 +27,9 @@ export class YetterImageClient {
         return this.endpoint;
     }
 
-    public async generateImage(
-        body: GenerateImageRequest
-    ): Promise<GenerateImageResponse> {
+    public async generate(
+        body: GenerateRequest
+    ): Promise<GenerateResponse> {
         const url = `${this.endpoint}/${body.model}`;
         const res = await fetch(url, {
             method: "POST",
@@ -45,7 +45,7 @@ export class YetterImageClient {
             throw new Error(`API error (${res.status}): ${errorText}`);
         }
         
-        return (await res.json()) as GenerateImageResponse;
+        return (await res.json()) as GenerateResponse;
     }
     
     public async getStatus(body: GetStatusRequest): Promise<GetStatusResponse> {

package/src/client.ts

@@ -5,7 +5,7 @@ import {
     GetStatusResponse,
     GetResponseResponse,
     SubscribeOptions,
-    GenerateImageResponse,
+    GenerateResponse,
     SubmitQueueOptions,
     GetResultOptions,
     GetResultResponse,
@@ -45,7 +45,7 @@ export class yetter {
             endpoint: yetter.endpoint
         });
 
-        const generateResponse = await client.generateImage({
+        const generateResponse = await client.generate({
             model: model,
             ...options.input,
         });
@@ -100,7 +100,7 @@ export class yetter {
         submit: async (
             model: string,
             options: SubmitQueueOptions
-        ): Promise<GenerateImageResponse> => {
+        ): Promise<GenerateResponse> => {
             if (!yetter.apiKey) {
                 throw new Error("API key is not configured. Call yetter.configure({ apiKey: 'your_key' }) or set YTR_API_KEY.");
             }
@@ -109,7 +109,7 @@ export class yetter {
                 endpoint: yetter.endpoint
             });
 
-            const generateResponse = await client.generateImage({
+            const generateResponse = await client.generate({
                 model: model,
                 ...options.input,
             });
@@ -178,11 +178,11 @@ export class yetter {
             endpoint: yetter.endpoint
         });
 
-        const initialApiResponse = await client.generateImage({
+        const initialApiResponse = await client.generate({
             model: model,
             ...options.input,
         });
-
+        console.timeEnd("Initial API Response");
         const requestId = initialApiResponse.request_id;
         const responseUrl = initialApiResponse.response_url;
         const cancelUrl = initialApiResponse.cancel_url;
@@ -271,7 +271,8 @@ export class yetter {
         };
 
         eventSource = new EventSourcePolyfill(sseStreamUrl, {
-            headers: { 'Authorization': `${yetter.apiKey}` }
+            headers: { 'Authorization': `${yetter.apiKey}` },
+            heartbeatTimeout: 3000,
         } as any);
 
         eventSource.onopen = (event: Event) => {
@@ -279,7 +280,7 @@ export class yetter {
         };
 
         eventSource.addEventListener('data', (event: MessageEvent) => {
-            console.log("SSE 'data' event received, raw data:", event.data);
+            // console.log("SSE 'data' event received, raw data:", event.data);
             try {
                 const statusData: GetStatusResponse = JSON.parse(event.data as string);
                 controller.push(statusData as StreamEvent);
@@ -290,15 +291,20 @@ export class yetter {
         });
 
         eventSource.addEventListener('done', (event: MessageEvent) => {
-            console.log("SSE 'done' event received, raw data:", event.data);
+            // console.log("SSE 'done' event received, raw data:", event.data);
             controller.close();
         });
 
         eventSource.onerror = (err: Event | MessageEvent) => { 
-            console.error("SSE Connection Error (onerror):", err);
-            controller.error(err);
+            const message = (err as any)?.error?.message || (err as any)?.message || '';
+            const isIdleTimeout = typeof message === 'string' && message.includes('No activity within') && message.includes('Reconnecting');
+            if (isIdleTimeout) {
+                console.warn("SSE idle timeout; letting EventSource auto-reconnect.", err);
+                return;
+            }
+            console.warn("SSE Connection Error (onerror) - will allow auto-reconnect:", err);
         };
-        
+
         // Handle if API immediately returns a terminal status in initialApiResponse (e.g. already completed/failed)
         if (initialApiResponse.status === "COMPLETED"){
             controller.push(initialApiResponse as any as StreamEvent);

package/src/types.ts

@@ -2,6 +2,7 @@ export interface ClientOptions {
     apiKey: string;
     endpoint?: string;
     is_bearer?: boolean;
+    promptEnhancerModel?: string;
 }
 
 export interface LoraWeight {
@@ -9,11 +10,11 @@ export interface LoraWeight {
     scale?: number;
 }
 
-export interface GenerateImageRequest {
+export interface GenerateRequest {
     [key: string]: any;
 }
 
-export interface GenerateImageResponse {
+export interface GenerateResponse {
     status: string;
     request_id: string;
     response_url: string;
@@ -66,13 +67,13 @@ export interface GetResponseResponse {
 }
 
 export interface SubscribeOptions {
-  input: Omit<GenerateImageRequest, 'model'>; // model is a direct param to subscribe
+  input: Omit<GenerateRequest, 'model'>; // model is a direct param to subscribe
   logs?: boolean;
   onQueueUpdate?: (update: GetStatusResponse) => void;
 }
 
 export interface SubmitQueueOptions {
-  input: Omit<GenerateImageRequest, 'model'>;
+  input: Omit<GenerateRequest, 'model'>;
 }
 
 export interface GetResultOptions {
@@ -95,7 +96,7 @@ export interface StatusResponse {
 
 // --- Stream Types ---
 export interface StreamOptions {
-    input: Omit<GenerateImageRequest, 'model'>;
+    input: Omit<GenerateRequest, 'model'>;
     // logs?: boolean; // Optional: if you want to request logs via stream if API supports it
 }