@yetter/client 0.0.4 → 0.0.6

package/dist/api.js

@@ -16,7 +16,7 @@ export class YetterImageClient {
             method: "POST",
             headers: {
                 "Content-Type": "application/json",
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
             body: JSON.stringify(body),
         });
@@ -35,7 +35,7 @@ export class YetterImageClient {
             method: "GET",
             headers: {
                 "Content-Type": "application/json",
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
         });
         if (!res.ok) {
@@ -49,7 +49,7 @@ export class YetterImageClient {
             method: "PUT",
             headers: {
                 "Content-Type": "application/json",
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
         });
         if (!res.ok) {
@@ -63,7 +63,7 @@ export class YetterImageClient {
             method: "GET",
             headers: {
                 "Content-Type": "application/json",
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
         });
         if (!res.ok) {

package/dist/client.js

@@ -4,7 +4,12 @@ import { EventSourcePolyfill } from 'event-source-polyfill';
 export class yetter {
     static configure(options) {
         if (options.apiKey) {
-            _a.apiKey = options.apiKey;
+            if (options.is_bearer) {
+                _a.apiKey = 'Bearer ' + options.apiKey;
+            }
+            else {
+                _a.apiKey = 'Key ' + options.apiKey;
+            }
         }
         if (options.endpoint) {
             _a.endpoint = options.endpoint;
@@ -17,7 +22,7 @@ export class yetter {
         }
         const client = new YetterImageClient({
             apiKey: _a.apiKey,
-            endpoint: _a.endpoint,
+            endpoint: _a.endpoint
         });
         const generateResponse = await client.generateImage({
             model: model,
@@ -67,18 +72,18 @@ export class yetter {
         if (!_a.apiKey) {
             throw new Error("API key is not configured. Call yetter.configure({ apiKey: 'your_key' }) or set YTR_API_KEY.");
         }
-        const apiClient = new YetterImageClient({
+        const client = new YetterImageClient({
             apiKey: _a.apiKey,
-            endpoint: _a.endpoint,
+            endpoint: _a.endpoint
         });
-        const initialApiResponse = await apiClient.generateImage({
+        const initialApiResponse = await client.generateImage({
             model: model,
             ...options.input,
         });
         const requestId = initialApiResponse.request_id;
         const responseUrl = initialApiResponse.response_url;
         const cancelUrl = initialApiResponse.cancel_url;
-        const sseStreamUrl = `${apiClient.getApiEndpoint()}/${model}/requests/${requestId}/status/stream`;
+        const sseStreamUrl = `${client.getApiEndpoint()}/${model}/requests/${requestId}/status/stream`;
         let eventSource;
         let streamEnded = false;
         // Setup the promise for the done() method
@@ -107,7 +112,7 @@ export class yetter {
                 // Check for terminal events to resolve/reject the donePromise
                 if (event.status === "COMPLETED") {
                     streamEnded = true;
-                    apiClient.getResponse({ url: responseUrl })
+                    client.getResponse({ url: responseUrl })
                         .then(resolveDonePromise)
                         .catch(rejectDonePromise)
                         .finally(() => this.close());
@@ -208,7 +213,7 @@ export class yetter {
             cancel: async () => {
                 controller.close();
                 try {
-                    await apiClient.cancel({ url: cancelUrl });
+                    await client.cancel({ url: cancelUrl });
                     console.log(`Stream for ${requestId} - underlying request cancelled.`);
                 }
                 catch (e) {
@@ -224,7 +229,7 @@ export class yetter {
     }
 }
 _a = yetter;
-yetter.apiKey = process.env.YTR_API_KEY || process.env.REACT_APP_YTR_API_KEY || "";
+yetter.apiKey = 'Key ' + (process.env.YTR_API_KEY || process.env.REACT_APP_YTR_API_KEY || "");
 yetter.endpoint = "https://api.yetter.ai";
 yetter.queue = {
     submit: async (model, options) => {
@@ -233,7 +238,7 @@ yetter.queue = {
         }
         const client = new YetterImageClient({
             apiKey: _a.apiKey,
-            endpoint: _a.endpoint,
+            endpoint: _a.endpoint
         });
         const generateResponse = await client.generateImage({
             model: model,
@@ -247,7 +252,7 @@ yetter.queue = {
         }
         const client = new YetterImageClient({
             apiKey: _a.apiKey,
-            endpoint: _a.endpoint,
+            endpoint: _a.endpoint
         });
         const endpoint = client.getApiEndpoint();
         const statusUrl = `${endpoint}/${model}/requests/${options.requestId}/status`;
@@ -263,7 +268,7 @@ yetter.queue = {
         }
         const client = new YetterImageClient({
             apiKey: _a.apiKey,
-            endpoint: _a.endpoint,
+            endpoint: _a.endpoint
         });
         const endpoint = client.getApiEndpoint();
         const responseUrl = `${endpoint}/${model}/requests/${options.requestId}`;

package/dist/types.d.ts

@@ -1,6 +1,7 @@
 export interface ClientOptions {
     apiKey: string;
     endpoint?: string;
+    is_bearer?: boolean;
 }
 export interface GenerateImageRequest {
     model: string;
@@ -48,7 +49,7 @@ export interface GetResponseRequest {
     url: string;
 }
 export interface GetResponseResponse {
-    images: string[];
+    images: string[] | Record<string, any>[];
     prompt: string;
 }
 export interface SubscribeOptions {

package/examples/stream.ts

@@ -9,6 +9,7 @@ async function main() {
         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
             },
         });
@@ -34,7 +35,7 @@ async function main() {
             console.log("Generated Images:", result.images);
         } else {
             console.log("No images in final result.");
-        }
+        }''
         console.log("Original Prompt:", result.prompt);
 
     } catch (err: any) {

package/package.json

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

package/src/api.ts

@@ -35,7 +35,7 @@ export class YetterImageClient {
             method: "POST",
             headers: {
                 "Content-Type": "application/json",
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
             body: JSON.stringify(body),
         });
@@ -58,7 +58,7 @@ export class YetterImageClient {
             method: "GET",
             headers: {
                 "Content-Type": "application/json",
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
         });
         
@@ -75,7 +75,7 @@ export class YetterImageClient {
             method: "PUT",
             headers: {
                 "Content-Type": "application/json", 
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
         });
         
@@ -92,7 +92,7 @@ export class YetterImageClient {
             method: "GET",
             headers: {
                 "Content-Type": "application/json",
-                Authorization: `${this.apiKey}`,
+                Authorization: this.apiKey,
             },
         });
 

package/src/client.ts

@@ -17,12 +17,16 @@ import {
 } from "./types.js";
 
 export class yetter {
-    private static apiKey: string = process.env.YTR_API_KEY || process.env.REACT_APP_YTR_API_KEY || "";
+    private static apiKey: string = 'Key ' + (process.env.YTR_API_KEY || process.env.REACT_APP_YTR_API_KEY || "");
     private static endpoint: string = "https://api.yetter.ai";
 
     public static configure(options: ClientOptions) {
         if (options.apiKey) {
-            yetter.apiKey = options.apiKey;
+            if (options.is_bearer) {
+                yetter.apiKey = 'Bearer ' + options.apiKey;
+            } else {
+                yetter.apiKey = 'Key ' + options.apiKey;
+            }
         }
         if (options.endpoint) {
             yetter.endpoint = options.endpoint;
@@ -38,7 +42,7 @@ export class yetter {
         }
         const client = new YetterImageClient({
             apiKey: yetter.apiKey,
-            endpoint: yetter.endpoint,
+            endpoint: yetter.endpoint
         });
 
         const generateResponse = await client.generateImage({
@@ -102,7 +106,7 @@ export class yetter {
             }
             const client = new YetterImageClient({
                 apiKey: yetter.apiKey,
-                endpoint: yetter.endpoint,
+                endpoint: yetter.endpoint
             });
 
             const generateResponse = await client.generateImage({
@@ -122,7 +126,7 @@ export class yetter {
             }
             const client = new YetterImageClient({
                 apiKey: yetter.apiKey,
-                endpoint: yetter.endpoint,
+                endpoint: yetter.endpoint
             });
 
             const endpoint = client.getApiEndpoint();
@@ -146,7 +150,7 @@ export class yetter {
             }
             const client = new YetterImageClient({
                 apiKey: yetter.apiKey,
-                endpoint: yetter.endpoint,
+                endpoint: yetter.endpoint
             });
 
             const endpoint = client.getApiEndpoint();
@@ -169,12 +173,12 @@ export class yetter {
         if (!yetter.apiKey) {
             throw new Error("API key is not configured. Call yetter.configure({ apiKey: 'your_key' }) or set YTR_API_KEY.");
         }
-        const apiClient = new YetterImageClient({
+        const client = new YetterImageClient({
             apiKey: yetter.apiKey,
-            endpoint: yetter.endpoint,
+            endpoint: yetter.endpoint
         });
 
-        const initialApiResponse = await apiClient.generateImage({
+        const initialApiResponse = await client.generateImage({
             model: model,
             ...options.input,
         });
@@ -182,7 +186,7 @@ export class yetter {
         const requestId = initialApiResponse.request_id;
         const responseUrl = initialApiResponse.response_url;
         const cancelUrl = initialApiResponse.cancel_url;
-        const sseStreamUrl = `${apiClient.getApiEndpoint()}/${model}/requests/${requestId}/status/stream`;
+        const sseStreamUrl = `${client.getApiEndpoint()}/${model}/requests/${requestId}/status/stream`;
 
         let eventSource: EventSource;
         let streamEnded = false;
@@ -211,7 +215,7 @@ export class yetter {
                 // Check for terminal events to resolve/reject the donePromise
                 if (event.status === "COMPLETED") {
                     streamEnded = true;
-                    apiClient.getResponse({ url: responseUrl })
+                    client.getResponse({ url: responseUrl })
                         .then(resolveDonePromise)
                         .catch(rejectDonePromise)
                         .finally(() => this.close());
@@ -314,7 +318,7 @@ export class yetter {
             cancel: async () => {
                 controller.close();
                 try {
-                    await apiClient.cancel({ url: cancelUrl });
+                    await client.cancel({ url: cancelUrl });
                     console.log(`Stream for ${requestId} - underlying request cancelled.`);
                 } catch (e: any) {
                     console.error(`Error cancelling underlying request for stream ${requestId}:`, e.message);

package/src/types.ts

@@ -1,6 +1,7 @@
 export interface ClientOptions {
     apiKey: string;
     endpoint?: string;
+    is_bearer?: boolean;
 }
 
 export interface GenerateImageRequest {
@@ -58,7 +59,7 @@ export interface GetResponseRequest {
 }
 
 export interface GetResponseResponse {
-    images: string[];
+    images: string[] | Record<string, any>[];
     prompt: string;
 }