@aigur/client - npm Package Compare versions

Comparing version 0.33.0 to 0.34.0

dist/index.d.ts

@@ -21,2 +21,7 @@ import { z } from 'zod';
 
+interface AigurConfiguration {
+    apiKeys: APIKeys;
+    eventListener?: (cb: (event: PipelineEvent) => void) => void;
+    eventPublisher?: (event: PipelineEvent) => Promise<any>;
+}
 interface PipelineConf<Input extends Record<string, unknown>, Output extends Record<string, unknown> | ReadableStream> {
@@ -29,2 +34,4 @@ id: string;
     updateProgress?: boolean;
+    eventListener?: (cb: (event: PipelineEvent) => void) => void;
+    eventPublisher?: (event: PipelineEvent) => Promise<any>;
     validateInput?: (input: Input) => {
@@ -41,2 +48,9 @@ valid: boolean;
 };
+type PipelineEvent = {
+    type: EventType;
+    data?: Record<any, any>;
+    pipelineId: string;
+};
+type EventType = PipelineEventType | ProgressEventType;
+type PipelineEventType = 'pipeline:start' | 'pipeline:finish';
 type ProgressEventType = 'node:start' | 'node:finish' | 'node:stream';
@@ -76,14 +90,12 @@ type APIKeys = Record<string, string> & {
     onFinish(cb: () => void): () => void;
+    private processPipeline;
+    private executeAction;
     private listenToEvents;
     private triggerListeners;
-    private processPipeline;
-    private executeAction;
     private notifyEvent;
 }
 
-interface AigurConfiguration {
-    apiKeys: APIKeys;
-}
 type Aigur = ReturnType<typeof createClient>;
 declare const createClient: (opts: AigurConfiguration) => {
+    apiKeys: APIKeys;
     pipeline: {
@@ -90,0 +102,0 @@ create: <Input extends Record<string, unknown>, Output extends Record<string, unknown> | ReadableStream<any>>(conf: PipelineConf<Input, Output>) => Pipeline<Input, Output>;

dist/index.js

@@ -1,7 +0,7 @@
-var E=Object.defineProperty;var z=Object.getOwnPropertyDescriptor;var L=Object.getOwnPropertyNames;var D=Object.prototype.hasOwnProperty;var k=(o,e)=>{for(var t in e)E(o,t,{get:e[t],enumerable:!0})},K=(o,e,t,n)=>{if(e&&typeof e=="object"||typeof e=="function")for(let r of L(e))!D.call(o,r)&&r!==t&&E(o,r,{get:()=>e[r],enumerable:!(n=z(e,r))||n.enumerable});return o};var M=o=>K(E({},"__esModule",{value:!0}),o);var xe={};k(xe,{Pipeline:()=>w,createClient:()=>J,dalleBase64TextToImage:()=>Z,dalleUrlTextToImage:()=>q,enhanceWithKeywords:()=>re,googleImageLabeling:()=>V,googleTextToSpeech:()=>ye,gpt3Prediction:()=>pe,gpt3PredictionStream:()=>ue,inputSchema:()=>N,output:()=>R,outputSchema:()=>se,replaceString:()=>ae,stabilityClipGuidancePreset:()=>C,stabilityModel:()=>U,stabilityTextToImage:()=>ee,stringToArrayBuffer:()=>le,vercelGenericEdge:()=>G,whisperApi:()=>Se});module.exports=M(xe);function O(o){return new Promise(e=>setTimeout(e,o))}function j(o,e){if(typeof o=="string")return r(o);let t={...o};return n(t);function n(i){let s={};for(let d in i)s[d]=r(i[d]);return s}function r(i){if(Array.isArray(i))return i.map(u=>n(u));if(typeof i=="object"&&i!==null)return n(i);let s=i,d=a(i);for(let u of d){let h=e[u.nodeId][u.property];if(h instanceof ArrayBuffer){s=h;continue}s=s?.replace(new RegExp(c(u.value)),h),s==="undefined"?s=void 0:s!==h&&s===h.toString()&&(s=h)}return s}function a(i){if(typeof i!="string")return[];let s=/\$context\.(\d+|input)\.(\w+)\$/g,d=i.matchAll(s),u=[];for(let f of d)u.push({value:f[0],nodeId:f[1],property:f[2]});return u}function c(i){return i.replace(/[.*+?^${}()|[\]\\]/g,"\\$&")}}function T(o=16){let e="",t="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789",n=t.length;for(let r=0;r<o;r++)e+=t.charAt(Math.floor(Math.random()*n));return e}var B=2,$=350,w=class{constructor(e,t,n){this.conf=e;this.flow=t;this.apiKeys=n;this.onProgressListeners=new Map;this.onStartListeners=new Map;this.onFinishListeners=new Map;this.vercel={invoke:e=>this.invokeRemote(`/api/pipelines/${this.conf.id}?_vercel_no_cache=1`,e),invokeStream:(e,t)=>this.invokeStream(`/api/pipelines/${this.conf.id}?_vercel_no_cache=1`,e,t)};this.listenToEvents()}invoke(e){return this.processPipeline(e)}invokeRemote(e,t){return fetch(e,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(t)}).then(n=>n.json())}async invokeStream(e,t,n){let r=await fetch(e,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(t)});if(!r.ok)throw new Error(r.statusText);let a=r.body;if(!a)return;let c=a.getReader(),i=new TextDecoder,s=!1;for(;!s;){let{value:d,done:u}=await c.read();s=u;let f=i.decode(d);n(f)}}onProgress(e){let t=T();return this.onProgressListeners.set(t,e),()=>{this.onProgressListeners.delete(t)}}onStart(e){let t=T();return this.onStartListeners.set(t,e),()=>{this.onStartListeners.delete(t)}}onFinish(e){let t=T();return this.onFinishListeners.set(t,e),()=>{this.onFinishListeners.delete(t)}}listenToEvents(){if(!this.conf.updateProgress||typeof window>"u"||!this.apiKeys?.ablySubscribe)return;let e=`https://realtime.ably.io/event-stream?channels=aigur-client&v=1.2&key=${this.apiKeys.ablySubscribe}&enveloped=false`,t=new EventSource(e);t.onmessage=n=>{let r=JSON.parse(n.data);r.pipelineId===this.conf.id&&(r.type==="pipeline:start"?this.triggerListeners(this.onStartListeners):r.type==="pipeline:finish"?this.triggerListeners(this.onFinishListeners):(r.type==="node:start"||r.type==="node:finish")&&this.triggerListeners(this.onProgressListeners,{...r.data,type:r.type}))}}triggerListeners(e,...t){for(let n of e.values())n(...t)}async processPipeline(e){let t=this.conf.retries??B;try{if(await this.notifyEvent("pipeline:start"),this.conf.validateInput){let i=this.conf.validateInput(e);if(!i.valid)throw new Error(i.message)}let n={input:e},r={},a=this.flow.getNodes(),c;for(let i=0;i<a.length;i++){c=this.notifyEvent("node:start",{node:a[i].name,index:i});let s=0,d=!1;do{s++;try{r=await this.executeAction(a,i,n),n[i]=r,d=!0}catch(u){if(s>t)throw u;await O((this.conf.retryDelayInMs??$)*s)}}while(!d&&s<=t);await c,await this.notifyEvent("node:finish",{node:a[i].name,index:i})}return await this.notifyEvent("pipeline:finish"),r}catch(n){throw console.error(n),n}}async executeAction(e,t,n){let{action:r,input:a}=e[t],c=j(a,n);return r(c,this.apiKeys)}notifyEvent(e,t){if(!(!this.conf.updateProgress||!this.apiKeys.ablyPublish))return fetch("https://rest.ably.io/channels/aigur-client/messages?enveloped=false ",{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Basic ${btoa(this.apiKeys.ablyPublish)}`},body:JSON.stringify({type:e,data:t,pipelineId:this.conf.id})})}};async function R(o){return o}var x=class{constructor(e){this.nodes=e}static create(){return new x([])}node(e,t){let n=this.createDynamicPlaceholders("input"),r=this.nodes.length>0?this.nodes[this.nodes.length-1]:n,a={action:e,input:t({nodes:this.nodes,prev:r.output,input:n}),output:this.createDynamicPlaceholders(this.nodes.length)};return this.nodes.push(a),this}createDynamicPlaceholders(e){let t={},n={get:function(a,c){return`$context.${e}.${c}$`}};return new Proxy(t,n)}output(e){return this.node(R,e)}getNodes(){return this.nodes}};var W=2,F=350,J=o=>{let{apiKeys:e}=o;return{pipeline:{create:t=>{let n={...t,retries:t.retries??W,retryDelayInMs:t.retryDelayInMs??F},r=t.flow(new x([]));return new w(n,r,e)}}}};async function G(o,e){let t=await e.json(),{searchParams:n}=new URL(e.url);if(!n.has("id"))return new Response("Missing id",{status:400});let r=n.get("id"),a=o[r];return{output:await a.invoke(t),pipeline:a}}var b=require("zod"),H=b.z.object({image:b.z.string()}),ze=b.z.object({labels:b.z.array(b.z.string())});async function V(o,e){let t=H.parse(o),n=`https://vision.googleapis.com/v1/images:annotate?key=${e.googleapis}`,r={requests:[{image:{content:t.image},features:[{type:"LABEL_DETECTION"}]}]};return{labels:(await(await fetch(n,{method:"POST",body:JSON.stringify(r)})).json()).responses[0].labelAnnotations.map(i=>i.description)}}var P=require("zod"),Y=P.z.object({prompt:P.z.string(),size:P.z.enum(["256x256","512x512","1024x1024"]).default("512x512")}),De=P.z.object({url:P.z.string().url()});async function q(o,e){let t=Y.parse(o),r=await fetch("https://api.openai.com/v1/images/generations",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)}),{data:a}=await r.json();return{url:a[0].url}}var S=require("zod"),X=S.z.object({prompt:S.z.string(),response_format:S.z.literal("b64_json").default("b64_json"),size:S.z.enum(["256x256","512x512","1024x1024"]).default("512x512")}),Ke=S.z.object({result:S.z.string()});async function Z(o,e){let t=X.parse(o),r=await fetch("https://api.openai.com/v1/images/generations",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)}),{data:a}=await r.json();return{result:a[0].b64_json}}var p=require("zod"),U=p.z.enum(["stable-diffusion-v1-5","stable-diffusion-512-v2-0","stable-diffusion-768-v2-0","stable-diffusion-512-v2-1","stable-diffusion-768-v2-1"]),C=p.z.enum(["NONE","FAST_BLUE","FAST_GREEN","SIMPLE","SLOW","SLOWER","SLOWEST"]),Q=p.z.object({text_prompts:p.z.array(p.z.object({text:p.z.string(),weight:p.z.number().min(-1).max(1).default(1)})).refine(o=>o.length>0,"Must have at least one text prompt"),model:U.default("stable-diffusion-v1-5"),clip_guidance_preset:C.optional(),steps:p.z.number().min(0).max(150).optional(),sampler:p.z.enum(["DDIM","DDPM","K_DPMPP_2M","K_DPMPP_2S_ANCESTRAL","K_DPM_2","K_DPM_2_ANCESTRAL","K_EULER","K_EULER_ANCESTRAL","K_HEUN","K_LMS"]).optional(),cfg_scale:p.z.number().min(0).max(35).optional(),seed:p.z.number().min(0).optional(),height:p.z.number().min(128).optional().refine(o=>typeof o<"u"?o%64===0:!0,"Must be a multiple of 64"),width:p.z.number().min(128).optional().refine(o=>typeof o<"u"?o%64===0:!0,"Must be a multiple of 64")}),Be=p.z.object({result:p.z.instanceof(ArrayBuffer)});async function ee(o,e){let t=Q.parse(o),n=`https://api.stability.ai/v1beta/generation/${t.model}/text-to-image`;return{result:await(await fetch(n,{headers:{"Content-Type":"application/json",Accept:"image/png",Authorization:e.stability},method:"POST",body:JSON.stringify(t)})).arrayBuffer()}}var v=require("zod"),te=v.z.object({text:v.z.string(),amount:v.z.number().optional().default(8)}),We=v.z.object({text:v.z.string()}),ne={"Colonial-style home":"Colonial, traditional, classic, historical, timeless, elegant, regal, grand, spacious, architectural, wood-framed, brick-exterior, symmetrical, gabled roof, columns, portico, fireplace, formal, ornate, landscaped".split(", "),"High-end penthouse apartment":"Luxury, high-end, penthouse, apartment, upscale, contemporary, modern, stylish, designer, elite, high-rise, rooftop, panoramic, views, spacious, open-plan, top-floor, amenities, concierge, service, exclusive".split(", ")};function oe(o){return Object.entries(ne).map(([e,t])=>`Title: ${e}
-Description: ${t.slice(0,o).join(", ")}
+var E=Object.defineProperty;var L=Object.getOwnPropertyDescriptor;var z=Object.getOwnPropertyNames;var D=Object.prototype.hasOwnProperty;var k=(n,e)=>{for(var t in e)E(n,t,{get:e[t],enumerable:!0})},M=(n,e,t,o)=>{if(e&&typeof e=="object"||typeof e=="function")for(let r of z(e))!D.call(n,r)&&r!==t&&E(n,r,{get:()=>e[r],enumerable:!(o=L(e,r))||o.enumerable});return n};var K=n=>M(E({},"__esModule",{value:!0}),n);var xe={};k(xe,{Pipeline:()=>v,createClient:()=>J,dalleBase64TextToImage:()=>Z,dalleUrlTextToImage:()=>q,enhanceWithKeywords:()=>re,googleImageLabeling:()=>V,googleTextToSpeech:()=>ye,gpt3Prediction:()=>pe,gpt3PredictionStream:()=>ue,inputSchema:()=>N,output:()=>R,outputSchema:()=>se,replaceString:()=>ae,stabilityClipGuidancePreset:()=>C,stabilityModel:()=>U,stabilityTextToImage:()=>ee,stringToArrayBuffer:()=>le,vercelGenericEdge:()=>G,whisperApi:()=>Se});module.exports=K(xe);async function R(n){return n}var x=class{constructor(e){this.nodes=e}static create(){return new x([])}node(e,t){let o=this.createDynamicPlaceholders("input"),r=this.nodes.length>0?this.nodes[this.nodes.length-1]:o,a={action:e,input:t({nodes:this.nodes,prev:r.output,input:o}),output:this.createDynamicPlaceholders(this.nodes.length)};return this.nodes.push(a),this}createDynamicPlaceholders(e){let t={},o={get:function(a,c){return`$context.${e}.${c}$`}};return new Proxy(t,o)}output(e){return this.node(R,e)}getNodes(){return this.nodes}};function O(n){return new Promise(e=>setTimeout(e,n))}function j(n,e){if(typeof n=="string")return r(n);let t={...n};return o(t);function o(i){let s={};for(let d in i)s[d]=r(i[d]);return s}function r(i){if(Array.isArray(i))return i.map(u=>o(u));if(typeof i=="object"&&i!==null)return o(i);let s=i,d=a(i);for(let u of d){let h=e[u.nodeId][u.property];if(h instanceof ArrayBuffer){s=h;continue}s=s?.replace(new RegExp(c(u.value)),h),s==="undefined"?s=void 0:s!==h&&s===h.toString()&&(s=h)}return s}function a(i){if(typeof i!="string")return[];let s=/\$context\.(\d+|input)\.(\w+)\$/g,d=i.matchAll(s),u=[];for(let f of d)u.push({value:f[0],nodeId:f[1],property:f[2]});return u}function c(i){return i.replace(/[.*+?^${}()|[\]\\]/g,"\\$&")}}function T(n=16){let e="",t="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789",o=t.length;for(let r=0;r<n;r++)e+=t.charAt(Math.floor(Math.random()*o));return e}var B=2,$=350,v=class{constructor(e,t,o){this.conf=e;this.flow=t;this.apiKeys=o;this.onProgressListeners=new Map;this.onStartListeners=new Map;this.onFinishListeners=new Map;this.vercel={invoke:e=>this.invokeRemote(`/api/pipelines/${this.conf.id}?_vercel_no_cache=1`,e),invokeStream:(e,t)=>this.invokeStream(`/api/pipelines/${this.conf.id}?_vercel_no_cache=1`,e,t)};this.listenToEvents()}invoke(e){return this.processPipeline(e)}invokeRemote(e,t){return fetch(e,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(t)}).then(o=>o.json())}async invokeStream(e,t,o){let r=await fetch(e,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(t)});if(!r.ok)throw new Error(r.statusText);let a=r.body;if(!a)return;let c=a.getReader(),i=new TextDecoder,s=!1;for(;!s;){let{value:d,done:u}=await c.read();s=u;let f=i.decode(d);o(f)}}onProgress(e){let t=T();return this.onProgressListeners.set(t,e),()=>{this.onProgressListeners.delete(t)}}onStart(e){let t=T();return this.onStartListeners.set(t,e),()=>{this.onStartListeners.delete(t)}}onFinish(e){let t=T();return this.onFinishListeners.set(t,e),()=>{this.onFinishListeners.delete(t)}}async processPipeline(e){let t=this.conf.retries??B;try{if(await this.notifyEvent("pipeline:start"),this.conf.validateInput){let i=this.conf.validateInput(e);if(!i.valid)throw new Error(i.message)}let o={input:e},r={},a=this.flow.getNodes(),c;for(let i=0;i<a.length;i++){c=this.notifyEvent("node:start",{node:a[i].name,index:i});let s=0,d=!1;do{s++;try{r=await this.executeAction(a,i,o),o[i]=r,d=!0}catch(u){if(s>t)throw u;await O((this.conf.retryDelayInMs??$)*s)}}while(!d&&s<=t);await c,await this.notifyEvent("node:finish",{node:a[i].name,index:i})}return await this.notifyEvent("pipeline:finish"),r}catch(o){throw console.error(o),o}}async executeAction(e,t,o){let{action:r,input:a}=e[t],c=j(a,o);return r(c,this.apiKeys)}listenToEvents(){!this.conf.updateProgress||typeof window>"u"||!this.conf.eventListener||this.conf.eventListener(e=>{e.pipelineId===this.conf.id&&(e.type==="pipeline:start"?this.triggerListeners(this.onStartListeners):e.type==="pipeline:finish"?this.triggerListeners(this.onFinishListeners):(e.type==="node:start"||e.type==="node:finish")&&this.triggerListeners(this.onProgressListeners,{...e.data,type:e.type}))})}triggerListeners(e,...t){for(let o of e.values())o(...t)}notifyEvent(e,t){if(!(!this.conf.updateProgress||!this.conf.eventPublisher))return this.conf.eventPublisher({type:e,data:t,pipelineId:this.conf.id})}};var W=2,F=350,J=n=>{let{apiKeys:e}=n;return{apiKeys:e,pipeline:{create:t=>{let o={...t,retries:t.retries??W,retryDelayInMs:t.retryDelayInMs??F,eventListener:n.eventListener,eventPublisher:n.eventPublisher},r=t.flow(new x([]));return new v(o,r,e)}}}};async function G(n,e){let t=await e.json(),{searchParams:o}=new URL(e.url);if(!o.has("id"))return new Response("Missing id",{status:400});let r=o.get("id"),a=n[r];return{output:await a.invoke(t),pipeline:a}}var b=require("zod"),H=b.z.object({image:b.z.string()}),Le=b.z.object({labels:b.z.array(b.z.string())});async function V(n,e){let t=H.parse(n),o=`https://vision.googleapis.com/v1/images:annotate?key=${e.googleapis}`,r={requests:[{image:{content:t.image},features:[{type:"LABEL_DETECTION"}]}]};return{labels:(await(await fetch(o,{method:"POST",body:JSON.stringify(r)})).json()).responses[0].labelAnnotations.map(i=>i.description)}}var P=require("zod"),Y=P.z.object({prompt:P.z.string(),size:P.z.enum(["256x256","512x512","1024x1024"]).default("512x512")}),De=P.z.object({url:P.z.string().url()});async function q(n,e){let t=Y.parse(n),r=await fetch("https://api.openai.com/v1/images/generations",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)}),{data:a}=await r.json();return{url:a[0].url}}var S=require("zod"),X=S.z.object({prompt:S.z.string(),response_format:S.z.literal("b64_json").default("b64_json"),size:S.z.enum(["256x256","512x512","1024x1024"]).default("512x512")}),Me=S.z.object({result:S.z.string()});async function Z(n,e){let t=X.parse(n),r=await fetch("https://api.openai.com/v1/images/generations",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)}),{data:a}=await r.json();return{result:a[0].b64_json}}var p=require("zod"),U=p.z.enum(["stable-diffusion-v1-5","stable-diffusion-512-v2-0","stable-diffusion-768-v2-0","stable-diffusion-512-v2-1","stable-diffusion-768-v2-1"]),C=p.z.enum(["NONE","FAST_BLUE","FAST_GREEN","SIMPLE","SLOW","SLOWER","SLOWEST"]),Q=p.z.object({text_prompts:p.z.array(p.z.object({text:p.z.string(),weight:p.z.number().min(-1).max(1).default(1)})).refine(n=>n.length>0,"Must have at least one text prompt"),model:U.default("stable-diffusion-v1-5"),clip_guidance_preset:C.optional(),steps:p.z.number().min(0).max(150).optional(),sampler:p.z.enum(["DDIM","DDPM","K_DPMPP_2M","K_DPMPP_2S_ANCESTRAL","K_DPM_2","K_DPM_2_ANCESTRAL","K_EULER","K_EULER_ANCESTRAL","K_HEUN","K_LMS"]).optional(),cfg_scale:p.z.number().min(0).max(35).optional(),seed:p.z.number().min(0).optional(),height:p.z.number().min(128).optional().refine(n=>typeof n<"u"?n%64===0:!0,"Must be a multiple of 64"),width:p.z.number().min(128).optional().refine(n=>typeof n<"u"?n%64===0:!0,"Must be a multiple of 64")}),Be=p.z.object({result:p.z.instanceof(ArrayBuffer)});async function ee(n,e){let t=Q.parse(n),o=`https://api.stability.ai/v1beta/generation/${t.model}/text-to-image`;return{result:await(await fetch(o,{headers:{"Content-Type":"application/json",Accept:"image/png",Authorization:e.stability},method:"POST",body:JSON.stringify(t)})).arrayBuffer()}}var w=require("zod"),te=w.z.object({text:w.z.string(),amount:w.z.number().optional().default(8)}),We=w.z.object({text:w.z.string()}),ne={"Colonial-style home":"Colonial, traditional, classic, historical, timeless, elegant, regal, grand, spacious, architectural, wood-framed, brick-exterior, symmetrical, gabled roof, columns, portico, fireplace, formal, ornate, landscaped".split(", "),"High-end penthouse apartment":"Luxury, high-end, penthouse, apartment, upscale, contemporary, modern, stylish, designer, elite, high-rise, rooftop, panoramic, views, spacious, open-plan, top-floor, amenities, concierge, service, exclusive".split(", ")};function oe(n){return Object.entries(ne).map(([e,t])=>`Title: ${e}
+Description: ${t.slice(0,n).join(", ")}
 `).join(`
-`)}async function re(o){let e=te.parse(o);return{text:`Write a maximum of ${e.amount} keywords in a csv list that describes the following
+`)}async function re(n){let e=te.parse(n);return{text:`Write a maximum of ${e.amount} keywords in a csv list that describes the following
 ${oe(e.amount)}
 Title: ${e.text}
-Description:`}}var g=require("zod"),ie=g.z.object({text:g.z.string().or(g.z.array(g.z.string())),modifier:g.z.string()}),Je=g.z.object({text:g.z.string()});async function ae(o){let e=ie.parse(o);return{text:e.modifier.replace(/\$\(text\)\$/gm,Array.isArray(e.text)?e.text.join(", "):e.text)}}var m=require("zod"),N=m.z.object({prompt:m.z.string(),model:m.z.enum(["text-davinci-003","text-curie-001","text-babbage-001","text-ada-001","code-davinci-002","code-cushman-002"]).default("text-davinci-003"),temperature:m.z.number().min(0).max(2).default(.7),top_p:m.z.number().min(0).max(1).default(1),frequency_penalty:m.z.number().min(-2).max(2).default(0),presence_penalty:m.z.number().min(-2).max(2).default(0),max_tokens:m.z.number().default(200),n:m.z.number().default(1)}),se=m.z.object({text:m.z.string()});async function pe(o,e){let t=N.parse(o);return{text:(await(await fetch("https://api.openai.com/v1/completions",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)})).json()).choices[0]?.text.replace(/^(?:\n)+/gm,"")}}var _=require("eventsource-parser"),I=require("zod");var ce=N.merge(I.z.object({stream:I.z.literal(!0).optional().default(!0)})),Xe=I.z.object({stream:I.z.instanceof(globalThis.ReadableStream??Object)});async function ue(o,e){let t=ce.parse(o),n=await fetch("https://api.openai.com/v1/completions",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)});return{stream:await de(n)}}async function de(o){let e=new TextEncoder,t=new TextDecoder,n=0;return new ReadableStream({async start(a){function c(s){if(s.type==="event"){let d=s.data;if(d==="[DONE]"){a.close();return}try{let f=JSON.parse(d).choices[0].text;if(n<2&&(f.match(/\n/)||[]).length)return;let h=e.encode(f);a.enqueue(h),n++}catch(u){a.error(u)}}}let i=(0,_.createParser)(c);for await(let s of o.body)i.feed(t.decode(s))}})}var A=require("zod"),me=A.z.object({string:A.z.string()}),Qe=A.z.object({arrayBuffer:A.z.instanceof(ArrayBuffer)});async function le(o){let e=me.parse(o);return{arrayBuffer:Uint8Array.from(atob(e.string),n=>n.charCodeAt(0)).buffer}}var l=require("zod"),fe=l.z.object({text:l.z.string(),speakingRate:l.z.number().min(.25).max(4).optional().default(1),pitch:l.z.number().min(-20).max(20).optional().default(0),encoding:l.z.enum(["MP3","FLAC","LINEAR16","MULAW","AMR","AMR_WB","OGG_OPUS","SPEEX_WITH_HEADER_BYTE","WEBM_OPUS"]).optional().default("MP3"),voice:l.z.object({language:l.z.string().optional().default("en-US"),name:l.z.enum(["en-US-Standard-A","en-US-Standard-C","en-US-Standard-D","en-US-Standard-E","en-US-Standard-F","en-US-Standard-G","en-US-Standard-H","en-US-Standard-I","en-US-Standard-J","en-US-Studio-M","en-US-Studio-O","en-US-Wavenet-A","en-US-Wavenet-B","en-US-Wavenet-C","en-US-Wavenet-D","en-US-Wavenet-E","en-US-Wavenet-F","en-US-Wavenet-G","en-US-Wavenet-H","en-US-Wavenet-I","en-US-Wavenet-J","en-US-News-K","en-US-News-L","en-US-News-M","en-US-News-N","en-US-Standard-A","en-US-Standard-B","en-US-Standard-C","en-US-Standard-D","en-US-Standard-E","en-US-Standard-F","en-US-Standard-G","en-US-Standard-H","en-US-Standard-I","en-US-Standard-J"]).or(l.z.string()).optional().default("en-US-Neural2-C")}).optional().default({language:"en-US",name:"en-US-Neural2-C"})}),tt=l.z.object({audio:l.z.string()});async function ye(o,e){let t=fe.parse(o),n=`https://us-central1-texttospeech.googleapis.com/v1beta1/text:synthesize?key=${e.googleapis}`,r={input:{text:t.text},voice:{languageCode:t.voice.language,name:t.voice.name},audioConfig:{audioEncoding:t.encoding,speakingRate:t.speakingRate,pitch:t.pitch}};return{audio:(await(await fetch(n,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(r)})).json()).audioContent}}var y=require("zod"),he="https://transcribe.whisperapi.com",ge=y.z.object({audioUrl:y.z.string().url(),language:y.z.string().default("en"),autoDetectLanguage:y.z.boolean().default(!1),fileType:y.z.string().default("mp3"),task:y.z.enum(["transcribe","translate"]).default("transcribe")}),ot=y.z.object({text:y.z.string()});async function Se(o,e){let t=ge.parse(o),n=new FormData;n.append("url",t.audioUrl),t.autoDetectLanguage||n.append("language",t.language),n.append("fileType",t.fileType),n.append("task",t.task);let r=await fetch(he,{method:"POST",headers:{contentType:"application/json",Authorization:"Bearer "+e.whisperapi},body:n});if(!r.ok)throw new Error(r.statusText);return{text:(await r.json()).text.trim()}}0&&(module.exports={Pipeline,createClient,dalleBase64TextToImage,dalleUrlTextToImage,enhanceWithKeywords,googleImageLabeling,googleTextToSpeech,gpt3Prediction,gpt3PredictionStream,inputSchema,output,outputSchema,replaceString,stabilityClipGuidancePreset,stabilityModel,stabilityTextToImage,stringToArrayBuffer,vercelGenericEdge,whisperApi});
+Description:`}}var g=require("zod"),ie=g.z.object({text:g.z.string().or(g.z.array(g.z.string())),modifier:g.z.string()}),Je=g.z.object({text:g.z.string()});async function ae(n){let e=ie.parse(n);return{text:e.modifier.replace(/\$\(text\)\$/gm,Array.isArray(e.text)?e.text.join(", "):e.text)}}var m=require("zod"),N=m.z.object({prompt:m.z.string(),model:m.z.enum(["text-davinci-003","text-curie-001","text-babbage-001","text-ada-001","code-davinci-002","code-cushman-002"]).default("text-davinci-003"),temperature:m.z.number().min(0).max(2).default(.7),top_p:m.z.number().min(0).max(1).default(1),frequency_penalty:m.z.number().min(-2).max(2).default(0),presence_penalty:m.z.number().min(-2).max(2).default(0),max_tokens:m.z.number().default(200),n:m.z.number().default(1)}),se=m.z.object({text:m.z.string()});async function pe(n,e){let t=N.parse(n);return{text:(await(await fetch("https://api.openai.com/v1/completions",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)})).json()).choices[0]?.text.replace(/^(?:\n)+/gm,"")}}var _=require("eventsource-parser"),I=require("zod");var ce=N.merge(I.z.object({stream:I.z.literal(!0).optional().default(!0)})),Xe=I.z.object({stream:I.z.instanceof(globalThis.ReadableStream??Object)});async function ue(n,e){let t=ce.parse(n),o=await fetch("https://api.openai.com/v1/completions",{headers:{"Content-Type":"application/json",Authorization:`Bearer ${e.openai}`},method:"POST",body:JSON.stringify(t)});return{stream:await de(o)}}async function de(n){let e=new TextEncoder,t=new TextDecoder,o=0;return new ReadableStream({async start(a){function c(s){if(s.type==="event"){let d=s.data;if(d==="[DONE]"){a.close();return}try{let f=JSON.parse(d).choices[0].text;if(o<2&&(f.match(/\n/)||[]).length)return;let h=e.encode(f);a.enqueue(h),o++}catch(u){a.error(u)}}}let i=(0,_.createParser)(c);for await(let s of n.body)i.feed(t.decode(s))}})}var A=require("zod"),me=A.z.object({string:A.z.string()}),Qe=A.z.object({arrayBuffer:A.z.instanceof(ArrayBuffer)});async function le(n){let e=me.parse(n);return{arrayBuffer:Uint8Array.from(atob(e.string),o=>o.charCodeAt(0)).buffer}}var l=require("zod"),fe=l.z.object({text:l.z.string(),speakingRate:l.z.number().min(.25).max(4).optional().default(1),pitch:l.z.number().min(-20).max(20).optional().default(0),encoding:l.z.enum(["MP3","FLAC","LINEAR16","MULAW","AMR","AMR_WB","OGG_OPUS","SPEEX_WITH_HEADER_BYTE","WEBM_OPUS"]).optional().default("MP3"),voice:l.z.object({language:l.z.string().optional().default("en-US"),name:l.z.enum(["en-US-Standard-A","en-US-Standard-C","en-US-Standard-D","en-US-Standard-E","en-US-Standard-F","en-US-Standard-G","en-US-Standard-H","en-US-Standard-I","en-US-Standard-J","en-US-Studio-M","en-US-Studio-O","en-US-Wavenet-A","en-US-Wavenet-B","en-US-Wavenet-C","en-US-Wavenet-D","en-US-Wavenet-E","en-US-Wavenet-F","en-US-Wavenet-G","en-US-Wavenet-H","en-US-Wavenet-I","en-US-Wavenet-J","en-US-News-K","en-US-News-L","en-US-News-M","en-US-News-N","en-US-Standard-A","en-US-Standard-B","en-US-Standard-C","en-US-Standard-D","en-US-Standard-E","en-US-Standard-F","en-US-Standard-G","en-US-Standard-H","en-US-Standard-I","en-US-Standard-J"]).or(l.z.string()).optional().default("en-US-Neural2-C")}).optional().default({language:"en-US",name:"en-US-Neural2-C"})}),tt=l.z.object({audio:l.z.string()});async function ye(n,e){let t=fe.parse(n),o=`https://us-central1-texttospeech.googleapis.com/v1beta1/text:synthesize?key=${e.googleapis}`,r={input:{text:t.text},voice:{languageCode:t.voice.language,name:t.voice.name},audioConfig:{audioEncoding:t.encoding,speakingRate:t.speakingRate,pitch:t.pitch}};return{audio:(await(await fetch(o,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(r)})).json()).audioContent}}var y=require("zod"),he="https://transcribe.whisperapi.com",ge=y.z.object({audioUrl:y.z.string().url(),language:y.z.string().default("en"),autoDetectLanguage:y.z.boolean().default(!1),fileType:y.z.string().default("mp3"),task:y.z.enum(["transcribe","translate"]).default("transcribe")}),ot=y.z.object({text:y.z.string()});async function Se(n,e){let t=ge.parse(n),o=new FormData;o.append("url",t.audioUrl),t.autoDetectLanguage||o.append("language",t.language),o.append("fileType",t.fileType),o.append("task",t.task);let r=await fetch(he,{method:"POST",headers:{contentType:"application/json",Authorization:"Bearer "+e.whisperapi},body:o});if(!r.ok)throw new Error(r.statusText);return{text:(await r.json()).text.trim()}}0&&(module.exports={Pipeline,createClient,dalleBase64TextToImage,dalleUrlTextToImage,enhanceWithKeywords,googleImageLabeling,googleTextToSpeech,gpt3Prediction,gpt3PredictionStream,inputSchema,output,outputSchema,replaceString,stabilityClipGuidancePreset,stabilityModel,stabilityTextToImage,stringToArrayBuffer,vercelGenericEdge,whisperApi});

package.json

 {
 	"name": "@aigur/client",
-	"version": "0.33.0",
+	"version": "0.34.0",
 	"description": "",
@@ -5,0 +5,0 @@ "main": "./dist/index.js",

src/Aigur.ts

@@ -1,9 +0,5 @@
-import { APIKeys, PipelineConf } from './types';
+import { FlowBuilder } from './builder';
 import { Pipeline } from './Pipeline';
-import { FlowBuilder } from './builder';
+import { AigurConfiguration, PipelineConf } from './types';
 
-interface AigurConfiguration {
-	apiKeys: APIKeys;
-}
-
 const DEFAULT_RETRIES = 2;
@@ -17,2 +13,3 @@ const RETRY_DELAY_IN_MS = 350;
 	return {
+		apiKeys,
 		pipeline: {
@@ -29,2 +26,4 @@ create: <
 					retryDelayInMs: conf.retryDelayInMs ?? RETRY_DELAY_IN_MS,
+					eventListener: opts.eventListener,
+					eventPublisher: opts.eventPublisher,
 				};
@@ -31,0 +30,0 @@ const flow = conf.flow(new FlowBuilder<Input, Output, [], null>([]));

src/Pipeline.ts

@@ -118,36 +118,2 @@ import { FlowBuilder } from './builder';
 
-	private listenToEvents() {
-		if (
-			!this.conf.updateProgress ||
-			typeof window === 'undefined' ||
-			!this.apiKeys?.ablySubscribe
-		) {
-			return;
-		}
-		// TODO: move channel name to config
-		const dataEndpoint = `https://realtime.ably.io/event-stream?channels=aigur-client&v=1.2&key=${this.apiKeys.ablySubscribe}&enveloped=false`;
-		const eventSource = new EventSource(dataEndpoint);
-		eventSource.onmessage = (event) => {
-			const e: { pipelineId: string; type: EventType; data: Record<any, any> } = JSON.parse(
-				event.data
-			);
-			if (e.pipelineId !== this.conf.id) {
-				return;
-			}
-			if (e.type === 'pipeline:start') {
-				this.triggerListeners(this.onStartListeners);
-			} else if (e.type === 'pipeline:finish') {
-				this.triggerListeners(this.onFinishListeners);
-			} else if (e.type === 'node:start' || e.type === 'node:finish') {
-				this.triggerListeners(this.onProgressListeners, { ...e.data, type: e.type });
-			}
-		};
-	}
-
-	private triggerListeners(listeners: Map<string, (...args: any[]) => void>, ...args: any[]) {
-		for (let listener of listeners.values()) {
-			listener(...args);
-		}
-	}
-
 	private async processPipeline(input: Input): Promise<Output> {
@@ -205,20 +171,33 @@ const retriesCount = this.conf.retries ?? DEFAULT_RETRIES;
 
-	private notifyEvent(type: EventType, data?: Record<any, any>) {
-		if (!this.conf.updateProgress || !this.apiKeys.ablyPublish) {
+	private listenToEvents() {
+		if (!this.conf.updateProgress || typeof window === 'undefined' || !this.conf.eventListener) {
 			return;
 		}
 
-		return fetch('https://rest.ably.io/channels/aigur-client/messages?enveloped=false ', {
-			method: 'POST',
-			headers: {
-				'Content-Type': 'application/json',
-				Authorization: `Basic ${btoa(this.apiKeys.ablyPublish)}`,
-			},
-			body: JSON.stringify({
-				type,
-				data,
-				pipelineId: this.conf.id,
-			}),
+		this.conf.eventListener((event) => {
+			if (event.pipelineId !== this.conf.id) {
+				return;
+			}
+			if (event.type === 'pipeline:start') {
+				this.triggerListeners(this.onStartListeners);
+			} else if (event.type === 'pipeline:finish') {
+				this.triggerListeners(this.onFinishListeners);
+			} else if (event.type === 'node:start' || event.type === 'node:finish') {
+				this.triggerListeners(this.onProgressListeners, { ...event.data, type: event.type });
+			}
 		});
 	}
+
+	private triggerListeners(listeners: Map<string, (...args: any[]) => void>, ...args: any[]) {
+		for (let listener of listeners.values()) {
+			listener(...args);
+		}
+	}
+
+	private notifyEvent(type: EventType, data?: Record<any, any>) {
+		if (!this.conf.updateProgress || !this.conf.eventPublisher) {
+			return;
+		}
+		return this.conf.eventPublisher({ type, data, pipelineId: this.conf.id });
+	}
 }

src/types.ts

@@ -5,2 +5,8 @@ import { z } from 'zod';
 
+export interface AigurConfiguration {
+	apiKeys: APIKeys;
+	eventListener?: (cb: (event: PipelineEvent) => void) => void;
+	eventPublisher?: (event: PipelineEvent) => Promise<any>;
+}
+
 export interface PipelineConf<
@@ -18,2 +24,4 @@ Input extends Record<string, unknown>,
 	updateProgress?: boolean;
+	eventListener?: (cb: (event: PipelineEvent) => void) => void;
+	eventPublisher?: (event: PipelineEvent) => Promise<any>;
 	validateInput?: (input: Input) => { valid: boolean; message?: string };
@@ -36,2 +44,3 @@ }
 
+export type PipelineEvent = { type: EventType; data?: Record<any, any>; pipelineId: string };
 export type EventType = PipelineEventType | ProgressEventType;
@@ -38,0 +47,0 @@ export type PipelineEventType = 'pipeline:start' | 'pipeline:finish';

New alerts

License Policy Violation

License

This package is not allowed per your license policy. Review the package's license to ensure compliance.

Found 1 instance in 1 package

Fixed alerts

License Policy Violation

License

This package is not allowed per your license policy. Review the package's license to ensure compliance.

Found 1 instance in 1 package

Improved metrics

Total package byte prevSize

61430

increased by0.13%

Number of medium supply chain risk alerts

10

decreased by-9.09%

Worsened metrics

Lines of code

1427

decreased by-0.14%

No dependency changes