@adaline/gateway 0.13.0 → 0.15.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +2 -0
- package/dist/index.d.mts +339 -231
- package/dist/index.d.ts +339 -231
- package/dist/index.js +37 -30
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +10 -10
- package/dist/index.mjs.map +1 -1
- package/package.json +10 -3
package/dist/index.js
CHANGED
|
@@ -1,41 +1,48 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
-
var zod = require('zod');
|
|
4
3
|
var types = require('@adaline/types');
|
|
5
|
-
var
|
|
6
|
-
var
|
|
4
|
+
var uuid = require('uuid');
|
|
5
|
+
var api = require('@opentelemetry/api');
|
|
6
|
+
var zod = require('zod');
|
|
7
|
+
var semanticConventions = require('@opentelemetry/semantic-conventions');
|
|
8
|
+
var ae = require('axios');
|
|
9
|
+
var proxyAgent = require('proxy-agent');
|
|
10
|
+
var it = require('crypto-js/sha256.js');
|
|
7
11
|
var lruCache = require('lru-cache');
|
|
12
|
+
var Te = require('os');
|
|
8
13
|
|
|
9
14
|
function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
|
|
10
15
|
|
|
11
|
-
var
|
|
12
|
-
var
|
|
13
|
-
|
|
14
|
-
var be=Object.defineProperty,Re=Object.defineProperties;var Te=Object.getOwnPropertyDescriptors;var W=Object.getOwnPropertySymbols;var we=Object.prototype.hasOwnProperty,ke=Object.prototype.propertyIsEnumerable;var D=(a,t)=>(t=Symbol[a])?t:Symbol.for("Symbol."+a);var J=(a,t,e)=>t in a?be(a,t,{enumerable:!0,configurable:!0,writable:!0,value:e}):a[t]=e,w=(a,t)=>{for(var e in t||(t={}))we.call(t,e)&&J(a,e,t[e]);if(W)for(var e of W(t))ke.call(t,e)&&J(a,e,t[e]);return a},L=(a,t)=>Re(a,Te(t));var d=(a,t,e)=>new Promise((s,r)=>{var o=n=>{try{m(e.next(n));}catch(c){r(c);}},l=n=>{try{m(e.throw(n));}catch(c){r(c);}},m=n=>n.done?s(n.value):Promise.resolve(n.value).then(o,l);m((e=e.apply(a,t)).next());}),C=function(a,t){this[0]=a,this[1]=t;},v=(a,t,e)=>{var s=(l,m,n,c)=>{try{var i=e[l](m),u=(m=i.value)instanceof C,q=i.done;Promise.resolve(u?m[0]:m).then(y=>u?s(l==="return"?l:"next",m[1]?{done:y.done,value:y.value}:y,n,c):n({value:y,done:q})).catch(y=>s("throw",y,n,c));}catch(y){c(y);}},r=l=>o[l]=m=>new Promise((n,c)=>s(l,m,n,c)),o={};return e=e.apply(a,t),o[D("asyncIterator")]=()=>o,r("next"),r("throw"),r("return"),o};var E=(a,t,e)=>(t=a[D("asyncIterator")])?t.call(a):(a=a[D("iterator")](),t={},e=(s,r)=>(r=a[s])&&(t[s]=o=>new Promise((l,m,n)=>(o=r.call(a,o),n=o.done,Promise.resolve(o.value).then(c=>l({value:c,done:n}),m)))),e("next"),e("return"),t);var f=class a extends Error{constructor(t,e=500,s){super(t),this.name="GatewayError",this.status=e,this.data=s,Error.captureStackTrace&&Error.captureStackTrace(this,a);}};var Ee="HttpClientError",H=class a extends types.GatewayBaseError{constructor({info:t,cause:e}){super({info:t,cause:e},Ee),this.info=t,this.cause=e,Object.setPrototypeOf(this,new.target.prototype);}static isHttpClientError(t){return t instanceof a}},He="HttpRequestError",g=class a extends types.GatewayBaseError{constructor(t,e=500,s,r){super({info:t,cause:{status:e,headers:s,data:r}},He),this.info=t,this.cause={status:e,headers:s,data:r},Object.setPrototypeOf(this,new.target.prototype);}static isHttpRequestError(t){return t instanceof a}};var F=a=>{let t={};return a&&(typeof a=="object"||a instanceof Headers)&&Object.entries(a).forEach(([e,s])=>{Array.isArray(s)?t[e]=s.join(", "):typeof s=="string"?t[e]=s:t[e]="";}),t},X=a=>{var o,l,m;let t=(a==null?void 0:a.message)||"An unexpected error occurred",e=((o=a==null?void 0:a.response)==null?void 0:o.status)||500,s=F((l=a==null?void 0:a.response)==null?void 0:l.headers)||{},r=((m=a==null?void 0:a.response)==null?void 0:m.data)||{};return new g(t,e,s,r)},j=class{constructor(t){this.isNodeEnvironment=()=>typeof process!="undefined"&&process.versions!=null&&process.versions.node!=null;let{axiosInstance:e,timeoutInMilliseconds:s}=t;this.client=e||V__default.default.create();let r=zod.z.number().int().positive().optional();this.defaultTimeout=r.parse(s),this.client.defaults.timeout=this.defaultTimeout,this.proxyUrl=t.proxyUrl;}makeRequest(l,m,n){return d(this,arguments,function*(t,e,s,r={},o){try{let c=L(w(w({},t==="get"||t==="delete"?{params:s}:{data:s}),r),{timeout:o||this.defaultTimeout});if(this.proxyUrl&&(c.proxy={host:this.proxyUrl.hostname,port:parseInt(this.proxyUrl.port),protocol:this.proxyUrl.protocol.slice(0,-1)}),console.log("config",c),t==="get"||t==="delete"){let i=yield this.client[t](e,c);return {data:i.data,headers:F(i.headers),status:{code:i.status,text:i.statusText}}}else {let i=yield this.client[t](e,c.data,L(w({},c),{params:c.params}));return {data:i.data,headers:F(i.headers),status:{code:i.status,text:i.statusText}}}}catch(c){throw V__default.default.isAxiosError(c)?X(c):new H({info:"An unexpected error occurred",cause:c})}})}stream(t,e,s,r,o){return v(this,null,function*(){try{if(this.isNodeEnvironment()){let i=yield new C(this.client.request({method:e,url:t,headers:r,data:s,responseType:"stream",signal:o==null?void 0:o.abortSignal}));try{for(var l=E(i.data),m,n,c;m=!(n=yield new C(l.next())).done;m=!1){let u=n.value;yield u.toString();}}catch(n){c=[n];}finally{try{m&&(n=l.return)&&(yield new C(n.call(l)));}finally{if(c)throw c[0]}}}else {let i={method:e,headers:new Headers(w({},r)),body:e!=="get"?JSON.stringify(s):void 0,signal:o==null?void 0:o.abortSignal},u=yield new C(fetch(t,i));if(!u.ok){let q=yield new C(u.json());throw new g(`Request failed with status ${u.status}`,u.status,F(u.headers),q)}if(u.body){let q=u.body.getReader();for(;;){let{done:y,value:k}=yield new C(q.read());if(y){yield new TextDecoder().decode(k,{stream:!0});break}yield new TextDecoder().decode(k,{stream:!0});}}else throw new g("Cannot stream the body of the response.",500,{},u)}}catch(i){throw g.isHttpRequestError(i)?i:(i==null?void 0:i.name)==="AbortError"?new g("AbortError",408,{},{}):(i==null?void 0:i.name)==="CanceledError"?new g("AbortError",408,{},{}):V__default.default.isAxiosError(i)?X(i):new H({info:"An unexpected error occurred",cause:i})}})}get(t,e,s){return d(this,null,function*(){return this.makeRequest("get",t,e||{},{headers:s})})}post(t,e,s){return d(this,null,function*(){return this.makeRequest("post",t,e||{},{headers:s})})}put(t,e,s){return d(this,null,function*(){return this.makeRequest("put",t,e||{},{headers:s})})}delete(t,e,s){return d(this,null,function*(){return this.makeRequest("delete",t,e||{},{headers:s})})}patch(t,e,s){return d(this,null,function*(){return this.makeRequest("patch",t,e||{},{headers:s})})}};var ve="QueueTaskTimeoutError",Q=class a extends types.GatewayBaseError{constructor({info:t,cause:e}){super({info:t,cause:e},ve),this.info=t,this.cause=e,Object.setPrototypeOf(this,new.target.prototype);}static isQueueTaskTimeoutError(t){return t instanceof a}};var Z=zod.z.object({maxConcurrentTasks:zod.z.number().int().positive(),retryCount:zod.z.number().int().positive(),timeout:zod.z.number().int().positive(),retry:zod.z.object({initialDelay:zod.z.number().int().positive(),exponentialFactor:zod.z.number().int().positive()})});var U=(a,t)=>Se__default.default(a+JSON.stringify(t)).toString(),S=a=>a instanceof f?a:H.isHttpClientError(a)?new f(a.message):a instanceof Error?new f(a.message):new f(a),x=()=>typeof window!="undefined"&&typeof window.document!="undefined"&&typeof navigator!="undefined";var xe={error:"color: red",warn:"color: yellow",info:"color: green"},Pe=(a,t,...e)=>{if(x())x()&&console.log(`%c[${a.toUpperCase()}] [${t}]`,xe[a],...e);else switch(a){case"error":console.error(...e);break;case"warn":console.warn(...e);break;default:console.log(...e);}},Me=(a,t,...e)=>{var s;x()||((s=process==null?void 0:process.env)==null?void 0:s.DEBUG)==="true"&&Pe(a,t,...e);},T=(a,t,...e)=>d(void 0,null,function*(){let s=[];a.forEach(r=>{let o=r[t];if(typeof o=="function")try{let l=o(...e);l instanceof Promise&&s.push(l);}catch(l){Me("error",`SAFELY_INVOKE_CALLBACKS:${String(t)}:`,l);}}),yield Promise.allSettled(s);}),_=a=>new Promise(t=>setTimeout(t,a));var P=class{constructor(t){this.activeTasks=0;this.queue=[];this.options=t;}enqueue(t){this.queue.push(t),this.processQueue();}withTimeout(t){return new Promise((e,s)=>{let r=setTimeout(()=>s(new Q({info:"Queue task timeout",cause:new Error("Queue task timeout")})),this.options.timeout);t.execute(t.request).then(o=>{clearTimeout(r),e(o);}).catch(o=>{clearTimeout(r),s(o);});})}executeWithRetry(t,e){return d(this,null,function*(){try{return yield this.withTimeout(t)}catch(s){if(e===0)throw s;let r=!0,o=this.options.retry.initialDelay*Math.pow(this.options.retry.exponentialFactor,this.options.retryCount-e);if(g.isHttpRequestError(s)){if(s.cause.status===429){let l=M.safeParse(t.request);if(l.success){let m=l.data.model.getRetryDelay(s.cause.headers);r=m.shouldRetry,m.delayMs>0&&(o=m.delayMs);}}s.cause.status>=500&&s.cause.status<600;}if(r)return yield _(o),this.executeWithRetry(t,e-1);throw s}})}processQueue(){return d(this,null,function*(){if(this.activeTasks>=this.options.maxConcurrentTasks)return;let t=this.queue.shift();if(t){this.activeTasks+=1;try{let e=yield this.executeWithRetry(t,this.options.retryCount);t.resolve(e);}catch(e){t.reject(e);}finally{this.activeTasks-=1,this.processQueue();}}})}};var O=class{constructor(t=1e3){this.cache=new lruCache.LRUCache({max:t,allowStale:!1,updateAgeOnGet:!1});}get(t){return d(this,null,function*(){return new Promise((e,s)=>{e(this.cache.get(t));})})}set(t,e){return d(this,null,function*(){return new Promise((s,r)=>{this.cache.set(t,e),s();})})}delete(t){return d(this,null,function*(){return new Promise((e,s)=>{this.cache.delete(t),e();})})}clear(){return d(this,null,function*(){return new Promise((t,e)=>{this.cache.clear(),t();})})}};var z=class{constructor(t,e){this.events=[];this.batchSize=e,this.endpointUrl=t;}record(t,e){let s={event:t,dimensions:e,timestamp:new Date().toISOString()};this.events.push(s),this.events.length>=this.batchSize&&this.flushEvents();}flushEvents(){if(this.events.length===0)return;let t=[...this.events];this.events=[],this.sendEvents(t);}sendEvents(t){return d(this,null,function*(){try{yield V__default.default.post(this.endpointUrl,{events:t},{headers:{"Content-Type":"application/json"}});}catch(e){}})}};var ae=zod.z.object({queueOptions:Z.partial().optional(),dangerouslyAllowBrowser:zod.z.boolean().optional(),httpClient:zod.z.custom().optional(),proxyUrl:zod.z.string().url().optional(),completeChatCache:zod.z.custom().optional(),completeChatCallbacks:zod.z.array(zod.z.custom()).nonempty().optional(),getEmbeddingsCache:zod.z.custom().optional(),getEmbeddingsCallbacks:zod.z.array(zod.z.custom()).nonempty().optional(),streamChatCallbacks:zod.z.array(zod.z.custom()).nonempty().optional()}),je=zod.z.object({enableCache:zod.z.boolean().optional().default(!0),customHeaders:zod.z.record(zod.z.string()).optional(),metadataForCallbacks:zod.z.any().optional()}),M=zod.z.object({model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),options:je.optional()}),Qe=zod.z.object({customHeaders:zod.z.record(zod.z.string()).optional(),metadataForCallbacks:zod.z.any().optional()}),se=zod.z.object({model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),options:Qe.optional()}),Ue=zod.z.object({enableCache:zod.z.boolean().optional().default(!0),customHeaders:zod.z.record(zod.z.string()).optional(),metadataForCallbacks:zod.z.any().optional()}),K=zod.z.object({model:zod.z.custom(),config:types.Config(),embeddingRequests:types.EmbeddingRequests(),options:Ue.optional()});var ie=zod.z.object({cache:zod.z.custom(),model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),enableCache:zod.z.boolean(),customHeaders:zod.z.record(zod.z.string()).optional(),callbacks:zod.z.array(zod.z.custom()).nonempty().optional(),metadataForCallbacks:zod.z.any().optional()}),St=zod.z.object({request:zod.z.object({config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional()}),response:types.ChatResponse,cached:zod.z.boolean(),latencyInMs:zod.z.number().int().positive(),metadataForCallbacks:zod.z.any().optional(),provider:zod.z.object({request:zod.z.any(),response:zod.z.any()})});function pe(a,t){return d(this,null,function*(){let e=ie.parse(a),s=a.callbacks||[];try{T(s,"onChatStart",a.metadataForCallbacks);let r={config:e.config,messages:e.messages,tools:e.tools},o={url:yield e.model.getCompleteChatUrl(e.config,e.messages,e.tools),headers:yield e.model.getCompleteChatHeaders(e.config,e.messages,e.tools),data:yield e.model.getCompleteChatData(e.config,e.messages,e.tools)};e.customHeaders&&(o.headers=w(w({},o.headers),e.customHeaders));let l=U(`complete-chat:${o.url}:${e.model.modelSchema.name}`,r);if(e.enableCache){let u=yield a.cache.get(l);if(u)return u.cached=!0,T(s,"onChatCached",a.metadataForCallbacks,u),u}let m=Date.now(),n=yield t.post(o.url,o.data,o.headers),c=Date.now()-m,i={request:r,response:e.model.transformCompleteChatResponse(n.data),cached:!1,latencyInMs:c,metadataForCallbacks:a.metadataForCallbacks,provider:{request:o,response:n}};return e.enableCache&&(yield a.cache.set(l,i)),T(s,"onChatComplete",a.metadataForCallbacks,i),i}catch(r){let o;throw g.isHttpRequestError(r)||r instanceof f?o=r:o=S(r),T(s,"onChatError",a.metadataForCallbacks,o),o}})}var de=zod.z.object({cache:zod.z.custom(),model:zod.z.custom(),config:types.Config(),embeddingRequests:types.EmbeddingRequests(),enableCache:zod.z.boolean(),customHeaders:zod.z.record(zod.z.string()).optional(),callbacks:zod.z.array(zod.z.custom()).nonempty().optional(),metadataForCallbacks:zod.z.any().optional()}),Vt=zod.z.object({request:zod.z.object({config:types.Config(),embeddingRequests:types.EmbeddingRequests()}),response:types.EmbeddingResponse,cached:zod.z.boolean(),latencyInMs:zod.z.number().int().positive(),metadataForCallbacks:zod.z.any().optional(),provider:zod.z.object({request:zod.z.any(),response:zod.z.any()})});function me(a,t){return d(this,null,function*(){let e=de.parse(a),s=a.callbacks||[];try{T(s,"onGetEmbeddingsStart",a.metadataForCallbacks);let r={config:e.config,embeddingRequests:e.embeddingRequests},o={url:yield e.model.getGetEmbeddingsUrl(e.config,e.embeddingRequests),headers:yield e.model.getGetEmbeddingsHeaders(e.config,e.embeddingRequests),data:yield e.model.getGetEmbeddingsData(e.config,e.embeddingRequests)};e.customHeaders&&(o.headers=w(w({},o.headers),e.customHeaders));let l=U(`get-embeddings:${o.url}:${e.model.modelSchema.name}`,r);if(e.enableCache){let u=yield a.cache.get(l);if(u)return u.cached=!0,T(s,"onGetEmbeddingsCached",a.metadataForCallbacks,u),u}let m=Date.now(),n=yield t.post(o.url,o.data,o.headers),c=Date.now()-m,i={request:r,response:e.model.transformGetEmbeddingsResponse(n.data),cached:!1,latencyInMs:c,metadataForCallbacks:a.metadataForCallbacks,provider:{request:o,response:n}};return e.enableCache&&(yield a.cache.set(l,i)),T(s,"onGetEmbeddingsComplete",a.metadataForCallbacks,i),i}catch(r){let o;throw g.isHttpRequestError(r)||r instanceof f?o=r:o=S(r),T(s,"onGetEmbeddingsError",a.metadataForCallbacks,o),o}})}var Ce=zod.z.object({model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),customHeaders:zod.z.record(zod.z.string()).optional(),callbacks:zod.z.array(zod.z.custom()).nonempty().optional(),metadataForCallbacks:zod.z.any().optional()}),sa=zod.z.object({request:zod.z.object({config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional()}),response:types.PartialChatResponse,metadataForCallbacks:zod.z.any().optional(),provider:zod.z.object({request:zod.z.any(),response:zod.z.any()})});function fe(a,t){return v(this,null,function*(){let e=Ce.parse(a),s=a.callbacks||[],r={config:e.config,messages:e.messages,tools:e.tools};try{T(s,"onStreamStart",a.metadataForCallbacks);let y={url:yield new C(e.model.getStreamChatUrl(e.config,e.messages,e.tools)),headers:yield new C(e.model.getStreamChatHeaders(e.config,e.messages,e.tools)),data:yield new C(e.model.getStreamChatData(e.config,e.messages,e.tools))};e.customHeaders&&(y.headers=w(w({},y.headers),e.customHeaders));let k="",A=!0;try{for(var c=E(t.stream(y.url,"post",y.data,y.headers)),i,u,q;i=!(u=yield new C(c.next())).done;i=!1){let I=u.value;try{for(var o=E(e.model.transformStreamChatResponseChunk(I,k)),l,m,n;l=!(m=yield new C(o.next())).done;l=!1){let ge=m.value;let $={request:r,response:ge.partialResponse,metadataForCallbacks:a.metadataForCallbacks,provider:{request:y,response:I}};T(s,A?"onStreamFirstResponse":"onStreamNewResponse",a.metadataForCallbacks,$),A&&(A=!1),yield $;}}catch(m){n=[m];}finally{try{l&&(m=o.return)&&(yield new C(m.call(o)));}finally{if(n)throw n[0]}}}}catch(u){q=[u];}finally{try{i&&(u=c.return)&&(yield new C(u.call(c)));}finally{if(q)throw q[0]}}T(s,"onStreamEnd",a.metadataForCallbacks);}catch(y){let k;throw g.isHttpRequestError(y)||y instanceof f?k=y:k=S(y),T(s,"onStreamError",a.metadataForCallbacks,k),k}})}var N=class{constructor(t){this.analytics=new z("https://j954t34pkh.execute-api.us-east-1.amazonaws.com/v0/analytics",5);var s,r,o,l;if(!t.dangerouslyAllowBrowser&&x())throw new f(`It looks like you're running in a browser-like environment.
|
|
16
|
+
var ae__default = /*#__PURE__*/_interopDefault(ae);
|
|
17
|
+
var it__default = /*#__PURE__*/_interopDefault(it);
|
|
18
|
+
var Te__default = /*#__PURE__*/_interopDefault(Te);
|
|
15
19
|
|
|
16
|
-
This is disabled by default, as it risks exposing your provider secrets to attackers.
|
|
17
|
-
If you understand the risks and have appropriate mitigation in place,
|
|
18
|
-
you can set the \`dangerouslyAllowBrowser\` option to \`true\`.
|
|
19
|
-
`);this.options=ae.parse(t);let e={maxConcurrentTasks:((s=this.options.queueOptions)==null?void 0:s.maxConcurrentTasks)||4,retryCount:((r=this.options.queueOptions)==null?void 0:r.retryCount)||3,retry:((o=this.options.queueOptions)==null?void 0:o.retry)||{initialDelay:1e3,exponentialFactor:2},timeout:((l=this.options.queueOptions)==null?void 0:l.timeout)||12e4};this.queues={completeChat:new P(e),getEmbeddings:new P(e)},this.httpClient=t.httpClient||new j({timeoutInMilliseconds:e.timeout*.9,proxyUrl:t.proxyUrl?new URL(t.proxyUrl):void 0}),this.caches={completeChat:t.completeChatCache||new O,getEmbeddings:t.getEmbeddingsCache||new O};}completeChat(t){return d(this,null,function*(){let e=M.parse(t);return this.analytics.record("completeChat",{modelName:e.model.modelSchema.name}),new Promise((s,r)=>{let o={request:e,cache:this.caches.completeChat,resolve:s,reject:r,execute:this.executeCompleteChat.bind(this)};this.queues.completeChat.enqueue(o);})})}executeCompleteChat(t){return d(this,null,function*(){var s,r,o;let e=M.parse(t);return pe({cache:this.caches.completeChat,model:e.model,config:e.config,messages:e.messages,tools:e.tools,enableCache:(r=(s=e.options)==null?void 0:s.enableCache)!=null?r:!0,callbacks:this.options.completeChatCallbacks,metadataForCallbacks:(o=e.options)==null?void 0:o.metadataForCallbacks},this.httpClient)})}streamChat(t){return v(this,null,function*(){var e,m;try{let n=se.parse(t);this.analytics.record("streamChat",{modelName:n.model.modelSchema.name});try{for(var s=E(fe({model:n.model,config:n.config,messages:n.messages,tools:n.tools,callbacks:this.options.streamChatCallbacks,metadataForCallbacks:(e=n.options)==null?void 0:e.metadataForCallbacks},this.httpClient)),r,o,l;r=!(o=yield new C(s.next())).done;r=!1){let c=o.value;yield c;}}catch(o){l=[o];}finally{try{r&&(o=s.return)&&(yield new C(o.call(s)));}finally{if(l)throw l[0]}}}catch(n){throw n instanceof f?n:new f(n==null?void 0:n.message,500,(m=n==null?void 0:n.response)==null?void 0:m.data)}})}getEmbeddings(t){return d(this,null,function*(){let e=K.parse(t);return this.analytics.record("getEmbeddings",{modelName:e.model.modelSchema.name}),new Promise((s,r)=>{let o={request:e,cache:this.caches.getEmbeddings,resolve:s,reject:r,execute:this.executeGetEmbeddingsTask.bind(this)};this.queues.getEmbeddings.enqueue(o);})})}executeGetEmbeddingsTask(t){return d(this,null,function*(){var s,r,o;let e=K.parse(t);return me({cache:this.caches.getEmbeddings,model:e.model,config:e.config,embeddingRequests:e.embeddingRequests,enableCache:(r=(s=e.options)==null?void 0:s.enableCache)!=null?r:!0,callbacks:this.options.getEmbeddingsCallbacks,metadataForCallbacks:(o=e.options)==null?void 0:o.metadataForCallbacks},this.httpClient)})}};N.GatewayError=f;
|
|
20
|
+
var _e=Object.defineProperty,We=Object.defineProperties;var Ye=Object.getOwnPropertyDescriptors;var ie=Object.getOwnPropertySymbols;var Je=Object.prototype.hasOwnProperty,Xe=Object.prototype.propertyIsEnumerable;var $=(r,t)=>(t=Symbol[r])?t:Symbol.for("Symbol."+r),Ze=r=>{throw TypeError(r)};var ce=(r,t,a)=>t in r?_e(r,t,{enumerable:!0,configurable:!0,writable:!0,value:a}):r[t]=a,q=(r,t)=>{for(var a in t||(t={}))Je.call(t,a)&&ce(r,a,t[a]);if(ie)for(var a of ie(t))Xe.call(t,a)&&ce(r,a,t[a]);return r},te=(r,t)=>We(r,Ye(t));var m=(r,t,a)=>new Promise((e,n)=>{var s=l=>{try{c(a.next(l));}catch(i){n(i);}},o=l=>{try{c(a.throw(l));}catch(i){n(i);}},c=l=>l.done?e(l.value):Promise.resolve(l.value).then(s,o);c((a=a.apply(r,t)).next());}),g=function(r,t){this[0]=r,this[1]=t;},M=(r,t,a)=>{var e=(o,c,l,i)=>{try{var d=a[o](c),p=(c=d.value)instanceof g,T=d.done;Promise.resolve(p?c[0]:c).then(y=>p?e(o==="return"?o:"next",c[1]?{done:y.done,value:y.value}:y,l,i):l({value:y,done:T})).catch(y=>e("throw",y,l,i));}catch(y){i(y);}},n=o=>s[o]=c=>new Promise((l,i)=>e(o,c,l,i)),s={};return a=a.apply(r,t),s[$("asyncIterator")]=()=>s,n("next"),n("throw"),n("return"),s},L=r=>{var t=r[$("asyncIterator")],a=!1,e,n={};return t==null?(t=r[$("iterator")](),e=s=>n[s]=o=>t[s](o)):(t=t.call(r),e=s=>n[s]=o=>{if(a){if(a=!1,s==="throw")throw o;return o}return a=!0,{done:!1,value:new g(new Promise(c=>{var l=t[s](o);l instanceof Object||Ze("Object expected"),c(l);}),1)}}),n[$("iterator")]=()=>n,e("next"),"throw"in t?e("throw"):n.throw=s=>{throw s},"return"in t&&e("return"),n},D=(r,t,a)=>(t=r[$("asyncIterator")])?t.call(r):(r=r[$("iterator")](),t={},a=(e,n)=>(n=r[e])&&(t[e]=s=>new Promise((o,c,l)=>(s=n.call(r,s),l=s.done,Promise.resolve(s.value).then(i=>o({value:i,done:l}),c)))),a("next"),a("return"),t);var E=class r extends Error{constructor(t,a=500,e){super(t),this.name="GatewayError",this.status=a,this.data=e,Error.captureStackTrace&&Error.captureStackTrace(this,r);}},tt="GatewayTelemetryError",de=class r extends types.GatewayBaseError{constructor({info:t,cause:a}){super({info:t,cause:a},tt),this.info=t,this.cause=a,Object.setPrototypeOf(this,new.target.prototype);}static isGatewayTelemetryError(t){return t instanceof r}};var at="HttpClientError",P=class r extends types.GatewayBaseError{constructor({info:t,cause:a}){super({info:t,cause:a},at),this.info=t,this.cause=a,Object.setPrototypeOf(this,new.target.prototype);}static isHttpClientError(t){return t instanceof r}},st="HttpRequestError",v=class r extends types.GatewayBaseError{constructor(t,a=500,e,n){super({info:t,cause:{status:a,headers:e,data:n}},st),this.info=t,this.cause={status:a,headers:e,data:n},Object.setPrototypeOf(this,new.target.prototype);}static isHttpRequestError(t){return t instanceof r}};var b=class{static setTracer(t){this.tracer||(this.tracer=t||api.trace.getTracer(this.DEFAULT_TRACER_KEY));}static getTracer(){return this.tracer||api.trace.getTracer(this.DEFAULT_TRACER_KEY)}static setMeter(t){this.meter||(this.meter=t||api.metrics.getMeter(this.DEFAULT_METER_KEY));}static getMeter(){return this.meter||api.metrics.getMeter(this.DEFAULT_METER_KEY)}};b.DEFAULT_TRACER_KEY="gateway",b.DEFAULT_METER_KEY="gateway",b.tracer=void 0,b.meter=void 0;var N=r=>{let t={};return r&&(typeof r=="object"||r instanceof Headers)&&Object.entries(r).forEach(([a,e])=>{Array.isArray(e)?t[a]=e.join(", "):typeof e=="string"?t[a]=e:t[a]="";}),t},be=r=>{var s,o,c;let t=(r==null?void 0:r.message)||"An unexpected error occurred",a=((s=r==null?void 0:r.response)==null?void 0:s.status)||500,e=N((o=r==null?void 0:r.response)==null?void 0:o.headers)||{},n=((c=r==null?void 0:r.response)==null?void 0:c.data)||{};return new v(t,a,e,n)},V=class{constructor(t){this.isNodeEnvironment=()=>typeof process!="undefined"&&process.versions!=null&&process.versions.node!=null;let{axiosInstance:a,timeoutInMilliseconds:e}=t;this.client=a||ae__default.default.create();let n=zod.z.number().int().positive().optional();this.defaultTimeout=n.parse(e),this.client.defaults.timeout=this.defaultTimeout,this.httpProxyAgent=new proxyAgent.ProxyAgent,this.httpsProxyAgent=new proxyAgent.ProxyAgent({rejectUnauthorized:!1});let s=h.getLogger();s==null||s.debug(`IsomorphicHttpClient initialized with defaultTimeout: ${this.defaultTimeout}`);}makeRequest(o,c,l){return m(this,arguments,function*(t,a,e,n={},s){let i=h.getLogger(),d=p=>m(this,null,function*(){try{let T=te(q(q({},t==="get"||t==="delete"?{params:e}:{data:e}),n),{timeout:this.defaultTimeout,httpAgent:this.httpProxyAgent,httpsAgent:this.httpsProxyAgent});if(t==="get"||t==="delete"){let y=yield this.client[t](a,T);p==null||p.setStatus({code:api.SpanStatusCode.OK,message:"request successful"});let C={data:y.data,headers:N(y.headers),status:{code:y.status,text:y.statusText}};return i==null||i.debug("IsomorphicHttpClient.makeRequest response: ",C),C}else {let y=yield this.client[t](a,T.data,te(q({},T),{params:T.params}));p==null||p.setStatus({code:api.SpanStatusCode.OK,message:"request successful"});let C={data:y.data,headers:N(y.headers),status:{code:y.status,text:y.statusText}};return i==null||i.debug("IsomorphicHttpClient.makeRequest response: ",C),C}}catch(T){throw i==null||i.warn("IsomorphicHttpClient.makeRequest error: ",T),p==null||p.setStatus({code:api.SpanStatusCode.ERROR,message:"request failed"}),ae__default.default.isAxiosError(T)?be(T):new P({info:"An unexpected error occurred",cause:T})}finally{p==null||p.end();}});return s?yield api.context.with(s,()=>m(this,null,function*(){return yield b.getTracer().startActiveSpan("http.request",T=>m(this,null,function*(){return T.setAttribute(semanticConventions.ATTR_HTTP_REQUEST_METHOD,t.toUpperCase()),T.setAttribute(semanticConventions.ATTR_URL_FULL,a),yield d(T)}))})):d()})}stream(t,a,e,n,s,o){return M(this,null,function*(){let c=h.getLogger();c==null||c.debug(`IsomorphicHttpClient.STREAM request to ${t}`,{data:e,headers:n});let l=function(i){return M(this,null,function*(){try{if(this.isNodeEnvironment()){c==null||c.debug("IsomorphicHttpClient.stream in node environment");let C=yield new g(this.client.request({method:a,url:t,headers:n,data:e,responseType:"stream",signal:s==null?void 0:s.abortSignal}));try{for(var d=D(C.data),p,T,y;p=!(T=yield new g(d.next())).done;p=!1){let f=T.value;i==null||i.addEvent("stream.chunk",{message:"stream chunk received"});let w=f.toString();c==null||c.debug("IsomorphicHttpClient.stream chunk: ",w),yield w;}}catch(T){y=[T];}finally{try{p&&(T=d.return)&&(yield new g(T.call(d)));}finally{if(y)throw y[0]}}i==null||i.setStatus({code:api.SpanStatusCode.OK,message:"stream successful"});}else {c==null||c.debug("IsomorphicHttpClient.stream in browser environment");let C={method:a,headers:new Headers(q({},n)),body:a!=="get"?JSON.stringify(e):void 0,signal:s==null?void 0:s.abortSignal},f=yield new g(fetch(t,C));if(!f.ok){c==null||c.warn("IsomorphicHttpClient.stream response not ok: ",f),i==null||i.setStatus({code:api.SpanStatusCode.ERROR,message:"stream failed"});let w=yield new g(f.json());throw new v(`Request failed with status ${f.status}`,f.status,N(f.headers),w)}if(f.body){let w=f.body.getReader();for(;;){let{done:H,value:A}=yield new g(w.read());if(H){i==null||i.addEvent("stream.chunk",{message:"stream chunk received"});let j=new TextDecoder().decode(A,{stream:!0});c==null||c.debug("IsomorphicHttpClient.stream chunk: ",j),yield j;break}i==null||i.addEvent("stream.chunk",{message:"stream chunk received"});let I=new TextDecoder().decode(A,{stream:!0});c==null||c.debug("IsomorphicHttpClient.stream chunk: ",I),yield I;}i==null||i.setStatus({code:api.SpanStatusCode.OK,message:"stream successful"});}else throw c==null||c.warn("IsomorphicHttpClient.stream response has no body"),i==null||i.setStatus({code:api.SpanStatusCode.ERROR,message:"stream failed"}),new v("Cannot stream the body of the response.",500,{},f)}}catch(C){throw c==null||c.warn("IsomorphicHttpClient.stream error: ",C),i==null||i.setStatus({code:api.SpanStatusCode.ERROR,message:"stream failed"}),v.isHttpRequestError(C)?C:(C==null?void 0:C.name)==="AbortError"?new v("AbortError",408,{},{}):(C==null?void 0:C.name)==="CanceledError"?new v("AbortError",408,{},{}):ae__default.default.isAxiosError(C)?be(C):new P({info:"An unexpected error occurred",cause:C})}finally{i==null||i.end();}})}.bind(this);return o?yield*L(yield new g(api.context.with(o,()=>m(this,null,function*(){return yield b.getTracer().startActiveSpan("http.stream",d=>m(this,null,function*(){return d.setAttribute(semanticConventions.ATTR_HTTP_REQUEST_METHOD,a.toUpperCase()),d.setAttribute(semanticConventions.ATTR_URL_FULL,t),yield l(d)}))})))):yield*L(l())})}get(t,a,e,n){return m(this,null,function*(){let s=h.getLogger();return s==null||s.debug(`IsomorphicHttpClient.GET request to ${t}`,{params:a,headers:e}),this.makeRequest("get",t,a||{},{headers:e},n)})}post(t,a,e,n){return m(this,null,function*(){let s=h.getLogger();return s==null||s.debug(`IsomorphicHttpClient.POST request to ${t}`,{data:a,headers:e}),this.makeRequest("post",t,a||{},{headers:e},n)})}put(t,a,e,n){return m(this,null,function*(){let s=h.getLogger();return s==null||s.debug(`IsomorphicHttpClient.PUT request to ${t}`,{data:a,headers:e}),this.makeRequest("put",t,a||{},{headers:e},n)})}delete(t,a,e,n){return m(this,null,function*(){let s=h.getLogger();return s==null||s.debug(`IsomorphicHttpClient.DELETE request to ${t}`,{params:a,headers:e}),this.makeRequest("delete",t,a||{},{headers:e},n)})}patch(t,a,e,n){return m(this,null,function*(){let s=h.getLogger();return s==null||s.debug(`IsomorphicHttpClient.PATCH request to ${t}`,{data:a,headers:e}),this.makeRequest("patch",t,a||{},{headers:e},n)})}};var nt="QueueTaskTimeoutError",B=class r extends types.GatewayBaseError{constructor({info:t,cause:a}){super({info:t,cause:a},nt),this.info=t,this.cause=a,Object.setPrototypeOf(this,new.target.prototype);}static isQueueTaskTimeoutError(t){return t instanceof r}};var fe=zod.z.object({maxConcurrentTasks:zod.z.number().int().positive(),retryCount:zod.z.number().int().positive(),timeout:zod.z.number().int().positive(),retry:zod.z.object({initialDelay:zod.z.number().int().positive(),exponentialFactor:zod.z.number().int().positive()})});var h=class{static setLogger(t){this.logger=t;}static getLogger(){return this.logger}};var _=(r,t)=>it__default.default(r+JSON.stringify(t)).toString(),F=r=>r instanceof E?r:P.isHttpClientError(r)?new E(r.message):r instanceof Error?new E(r.message):new E(r),x=()=>typeof window!="undefined"&&typeof window.document!="undefined"&&typeof navigator!="undefined";var ct={error:"color: red",warn:"color: yellow",info:"color: green"},dt=(r,t,...a)=>{if(x())x()&&console.log(`%c[${r.toUpperCase()}] [${t}]`,ct[r],...a);else switch(r){case"error":console.error(...a);break;case"warn":console.warn(...a);break;default:console.log(...a);}},mt=(r,t,...a)=>{var e;x()||((e=process==null?void 0:process.env)==null?void 0:e.DEBUG)==="true"&&dt(r,t,...a);},G=(r,t,...a)=>m(void 0,null,function*(){let e=[];r.forEach(n=>{let s=n[t];if(typeof s=="function")try{let o=s(...a);o instanceof Promise&&e.push(o);}catch(o){mt("error",`SAFELY_INVOKE_CALLBACKS:${String(t)}:`,o);}}),yield Promise.allSettled(e);}),ge=r=>new Promise(t=>setTimeout(t,r));var Re=class{debug(t,...a){console.debug(t,...a);}info(t,...a){console.info(t,...a);}warn(t,...a){x()?console.warn(`%WARN: %c${t}`,"color: yellow; font-weight: bold;","",...a):console.warn(`\x1B[33mWARN:\x1B[0m ${t}`,...a);}error(t,...a){x()?console.error(`%ERROR: %c${t}`,"color: lightcoral; font-weight: bold;","",...a):console.error(`\x1B[91mERROR:\x1B[0m ${t}`,...a);}critical(t,...a){x()?console.error(`%cCRITICAL: %c${t}`,"color: red; font-weight: bold;","",...a):console.error(`\x1B[31;1mCRITICAL:\x1B[0m ${t}`,...a);}};var U=class{constructor(t){this.activeTasks=0;this.queue=[];this.options=t;}enqueue(t){let a=h.getLogger();a==null||a.debug(`SimpleQueue.enqueue invoked, id: ${t.id}`),api.context.with(t.telemetryContext,()=>m(this,null,function*(){return b.getTracer().startActiveSpan("queue.task.pickup-wait",n=>m(this,null,function*(){n.setAttribute("id",t.id),this.queue.push({task:t,taskSpan:n}),a==null||a.debug(`SimpleQueue.enqueue task enqueued, id: ${t.id}`);}))})),this.processQueue();}executeWithTimeout(t,a){let e=h.getLogger();return e==null||e.debug(`SimpleQueue.executeWithTimeout invoked with timeout: ${this.options.timeout}, id: ${t.id}`),new Promise((n,s)=>{let o=setTimeout(()=>{e==null||e.warn(`SimpleQueue.executeWithTimeout timed out, id: ${t.id}`),s(new B({info:"Queue task timeout",cause:new Error("Queue task timeout")}));},this.options.timeout);e==null||e.debug(`SimpleQueue.executeWithTimeout task executing, id: ${t.id}`),t.execute(t.request,a).then(c=>{e==null||e.debug(`SimpleQueue.executeWithTimeout task completed, id: ${t.id}`),clearTimeout(o),n(c);}).catch(c=>{e==null||e.warn(`SimpleQueue.executeWithTimeout task errored, id: ${t.id}`),clearTimeout(o),s(c);});})}executeWithRetry(t,a){return m(this,null,function*(){let e=h.getLogger();return yield api.context.with(t.telemetryContext,()=>m(this,null,function*(){let n=b.getTracer();return yield n.startActiveSpan("queue.task.execute",s=>m(this,null,function*(){e==null||e.debug(`SimpleQueue.executeWithRetry invoked, attempt: ${this.options.retryCount-a}, id: ${t.id}`),s.setAttribute("attempt",this.options.retryCount-a);try{let o=api.context.active(),c=yield this.executeWithTimeout(t,o);return s.setStatus({code:api.SpanStatusCode.OK}),s.end(),c}catch(o){if(a===0)throw e==null||e.warn(`SimpleQueue.executeWithRetry retry count reached, id: ${t.id}`),s.end(),o;let c=!0,l=this.options.retry.initialDelay*Math.pow(this.options.retry.exponentialFactor,this.options.retryCount-a);if(v.isHttpRequestError(o)){if(o.cause.status===429){e==null||e.warn(`SimpleQueue.executeWithRetry rate limiting error, id: ${t.id}`);let i=z.safeParse(t.request);if(i.success){let d=i.data.model.getRetryDelay(o.cause.headers);c=d.shouldRetry,d.delayMs>0&&(l=d.delayMs);}}o.cause.status>=500&&o.cause.status<600&&(e==null||e.warn(`SimpleQueue.executeWithRetry ${o.cause.status} error, id: ${t.id}`));}else e==null||e.warn(`SimpleQueue.executeWithRetry non http-request error, id: ${t.id}`,{error:o});if(c)return yield n.startActiveSpan("queue.task.retry-wait",i=>m(this,null,function*(){return e==null||e.debug(`SimpleQueue.executeWithRetry retry wait: ${l}ms, id: ${t.id}`),yield ge(l),i.end(),s.end(),this.executeWithRetry(t,a-1)}));throw e==null||e.warn(`SimpleQueue.executeWithRetry model returned should not retry, id: ${t.id}`),s.end(),o}finally{}}))}))})}processQueue(){return m(this,null,function*(){var s;let t=h.getLogger();if(this.activeTasks>=this.options.maxConcurrentTasks){t==null||t.debug("SimpleQueue.processQueue max concurrent tasks reached");return}let a=this.queue.shift();if(!a){t==null||t.debug("SimpleQueue.processQueue no item to process");return}let{task:e,taskSpan:n}=a;n&&n.end(),this.activeTasks+=1,t==null||t.debug(`SimpleQueue.processQueue active tasks: ${this.activeTasks}`),t==null||t.debug(`SimpleQueue.processQueue processing task, id: ${e.id}`);try{let o=yield this.executeWithRetry(e,this.options.retryCount);t==null||t.debug(`SimpleQueue.processQueue task completed, id: ${e.id}`),e.resolve(o);}catch(o){t==null||t.warn(`SimpleQueue.processQueue task errored, id: ${e.id}`),e.reject(o);}finally{this.activeTasks-=1,t==null||t.debug(`SimpleQueue.processQueue active tasks: ${this.activeTasks}`),(s=api.trace.getSpan(e.telemetryContext))==null||s.end(),this.processQueue();}})}};var K=class{constructor(t=1e3){this.cache=new lruCache.LRUCache({max:t,allowStale:!1,updateAgeOnGet:!1});let a=h.getLogger();a==null||a.debug(`LRUCache initialized with maxEntries: ${t}`);}get(t){return m(this,null,function*(){let a=h.getLogger();return a==null||a.debug(`LRUCache.get invoked, key: ${t}`),new Promise(e=>{let n=this.cache.get(t);a==null||a.debug("LRUCache.get completed, value: ",n),e(n);})})}set(t,a){return m(this,null,function*(){let e=h.getLogger();return e==null||e.debug(`LRUCache.set invoked, key: ${t}, value: `,a),new Promise(n=>{this.cache.set(t,a),e==null||e.debug("LRUCache.set completed"),n();})})}delete(t){return m(this,null,function*(){let a=h.getLogger();return a==null||a.debug(`LRUCache.delete invoked, key: ${t}`),new Promise(e=>{this.cache.delete(t),a==null||a.debug("LRUCache.delete completed"),e();})})}clear(){return m(this,null,function*(){let t=h.getLogger();return t==null||t.debug("LRUCache.clear invoked"),new Promise(a=>{this.cache.clear(),t==null||t.debug("LRUCache.clear completed"),a();})})}};var we=()=>({node:{version:process.version,platform:Te__default.default.platform(),architecture:Te__default.default.arch()}}),Ee=()=>({browser:{version:navigator.userAgent,userAgent:navigator.userAgent}}),ve=()=>{try{throw new Error("Not implemented")}catch(r){return "unknown"}};var W=class{constructor(){this.eventVersion="0.1";this.gatewayVersion=ve();this.flushInterval=1e4;this.batchSize=1;this.maxAttempts=3;this.environment=x()?Ee():we();this.analyticsEndpointUrl="https://j954t34pkh.execute-api.us-east-1.amazonaws.com/v0/analytics";this.events=[];}startFlushTimer(){x()?this.flushTimer=window.setInterval(()=>this.flushEvents(),this.flushInterval):this.flushTimer=setInterval(()=>this.flushEvents(),this.flushInterval);}stopFlushTimer(){x()?window.clearInterval(this.flushTimer):clearInterval(this.flushTimer);}record(t,a,e){let n={event:t,status:a,dimensions:e,timestamp:new Date().toISOString(),eventVersion:this.eventVersion,gatewayVersion:this.gatewayVersion,environment:this.environment};this.events.push({event:n,attempt:0}),this.events.length>=this.batchSize&&this.flushEvents();}flushEvents(){return m(this,null,function*(){if(this.events.length===0)return;let t=[...this.events];this.events=[],(yield this.sendEvents(t.map(e=>e.event)))||this.events.push(...t.filter(e=>e.attempt<this.maxAttempts).map(e=>({event:e.event,attempt:e.attempt+1})));})}sendEvents(t){return m(this,null,function*(){try{return (yield ae__default.default.post(this.analyticsEndpointUrl,{events:t},{headers:{"Content-Type":"application/json"}})).status===200}catch(a){return !1}})}stopRecorder(){this.stopFlushTimer(),this.flushEvents();}};var Y=class{record(t,a,e){}stopRecorder(){}};var J=class{static getAnalyticsRecorder(t){return this.analytics!==void 0?this.analytics:(this.analytics=t?new W:new Y,this.analytics)}};var He=zod.z.object({queueOptions:zod.z.lazy(()=>fe.partial()).optional(),dangerouslyAllowBrowser:zod.z.boolean().optional(),httpClient:zod.z.custom().optional(),completeChatCache:zod.z.custom().optional(),completeChatCallbacks:zod.z.array(zod.z.custom()).nonempty().optional(),getEmbeddingsCache:zod.z.custom().optional(),getEmbeddingsCallbacks:zod.z.array(zod.z.custom()).nonempty().optional(),streamChatCallbacks:zod.z.array(zod.z.custom()).nonempty().optional(),logger:zod.z.custom().optional(),telemetry:zod.z.object({tracer:zod.z.custom().optional(),meter:zod.z.custom().optional()}).optional(),analyticsEnabled:zod.z.boolean().optional()}),Ct=zod.z.object({enableCache:zod.z.boolean().optional().default(!0),customHeaders:zod.z.record(zod.z.string()).optional(),metadataForCallbacks:zod.z.any().optional()}),z=zod.z.object({model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),options:Ct.optional()}),bt=zod.z.object({customHeaders:zod.z.record(zod.z.string()).optional(),metadataForCallbacks:zod.z.any().optional()}),Ge=zod.z.object({model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),options:bt.optional()}),ft=zod.z.object({enableCache:zod.z.boolean().optional().default(!0),customHeaders:zod.z.record(zod.z.string()).optional(),metadataForCallbacks:zod.z.any().optional()}),re=zod.z.object({model:zod.z.custom(),config:types.Config(),embeddingRequests:types.EmbeddingRequests(),options:ft.optional()});var Me=zod.z.object({cache:zod.z.custom(),model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),enableCache:zod.z.boolean(),customHeaders:zod.z.record(zod.z.string()).optional(),callbacks:zod.z.array(zod.z.custom()).nonempty().optional(),metadataForCallbacks:zod.z.any().optional()}),Ka=zod.z.object({request:zod.z.object({config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional()}),response:types.ChatResponse,cached:zod.z.boolean(),latencyInMs:zod.z.number().int().positive(),metadataForCallbacks:zod.z.any().optional(),provider:zod.z.object({request:zod.z.any(),response:zod.z.any()})});function Pe(r,t,a){return m(this,null,function*(){let e=h.getLogger(),n=s=>m(this,null,function*(){e==null||e.debug("handleCompleteChat invoked"),e==null||e.debug("handleCompleteChat request: ",{request:r});let o=Me.parse(r),c=r.callbacks||[],l=api.context.active();try{G(c,"onChatStart",r.metadataForCallbacks);let i={config:o.config,messages:o.messages,tools:o.tools},d={url:yield o.model.getCompleteChatUrl(o.config,o.messages,o.tools),headers:yield o.model.getCompleteChatHeaders(o.config,o.messages,o.tools),data:yield o.model.getCompleteChatData(o.config,o.messages,o.tools)};o.customHeaders&&(d.headers=q(q({},d.headers),o.customHeaders)),e==null||e.debug("handleCompleteChat providerRequest: ",{providerRequest:d});let p=_(`complete-chat:${d.url}:${o.model.modelSchema.name}`,i);if(o.enableCache){e==null||e.debug("handleCompleteChat checking cache");let w=yield r.cache.get(p);if(w)return w.cached=!0,e==null||e.debug("handleCompleteChat cached hit"),s==null||s.setAttribute("cached",!0),s==null||s.setStatus({code:api.SpanStatusCode.OK}),G(c,"onChatCached",r.metadataForCallbacks,w),e==null||e.debug("handleCompleteChat cached response: ",{cachedResponse:w}),w}e==null||e.debug("handleCompleteChat cache miss");let T=Date.now(),y=yield t.post(d.url,d.data,d.headers,l),C=Date.now()-T;e==null||e.debug("handleCompleteChat providerResponse: ",{providerResponse:y});let f={request:i,response:o.model.transformCompleteChatResponse(y.data),cached:!1,latencyInMs:C,metadataForCallbacks:r.metadataForCallbacks,provider:{request:d,response:y}};return e==null||e.debug("handleCompleteChat response: ",{response:f}),o.enableCache&&(yield r.cache.set(p,f),e==null||e.debug("handleCompleteChat response cached")),s==null||s.setAttribute("cached",!1),s==null||s.setStatus({code:api.SpanStatusCode.OK}),G(c,"onChatComplete",r.metadataForCallbacks,f),f}catch(i){e==null||e.warn("handleCompleteChat error: ",{error:i});let d;throw v.isHttpRequestError(i)||i instanceof E?d=i:d=F(i),G(c,"onChatError",r.metadataForCallbacks,d),d}finally{s==null||s.end();}});return a?yield api.context.with(a,()=>m(this,null,function*(){return yield b.getTracer().startActiveSpan("complete-chat.handler",o=>m(this,null,function*(){return yield n(o)}))})):yield n()})}var $e=zod.z.object({cache:zod.z.custom(),model:zod.z.custom(),config:types.Config(),embeddingRequests:types.EmbeddingRequests(),enableCache:zod.z.boolean(),customHeaders:zod.z.record(zod.z.string()).optional(),callbacks:zod.z.array(zod.z.custom()).nonempty().optional(),metadataForCallbacks:zod.z.any().optional()}),is=zod.z.object({request:zod.z.object({config:types.Config(),embeddingRequests:types.EmbeddingRequests()}),response:types.EmbeddingResponse,cached:zod.z.boolean(),latencyInMs:zod.z.number().int().positive(),metadataForCallbacks:zod.z.any().optional(),provider:zod.z.object({request:zod.z.any(),response:zod.z.any()})});function De(r,t,a){return m(this,null,function*(){let e=h.getLogger(),n=s=>m(this,null,function*(){e==null||e.debug("handleGetEmbeddings invoked"),e==null||e.debug("handleGetEmbeddings request: ",{request:r});let o=$e.parse(r),c=r.callbacks||[],l=api.context.active();try{G(c,"onGetEmbeddingsStart",r.metadataForCallbacks);let i={config:o.config,embeddingRequests:o.embeddingRequests},d={url:yield o.model.getGetEmbeddingsUrl(o.config,o.embeddingRequests),headers:yield o.model.getGetEmbeddingsHeaders(o.config,o.embeddingRequests),data:yield o.model.getGetEmbeddingsData(o.config,o.embeddingRequests)};o.customHeaders&&(d.headers=q(q({},d.headers),o.customHeaders)),e==null||e.debug("handleGetEmbeddings providerRequest: ",{providerRequest:d});let p=_(`get-embeddings:${d.url}:${o.model.modelSchema.name}`,i);if(o.enableCache){e==null||e.debug("handleGetEmbeddings checking cache");let w=yield r.cache.get(p);if(w)return w.cached=!0,e==null||e.debug("handleGetEmbeddings cached hit"),s==null||s.setAttribute("cached",!0),s==null||s.setStatus({code:api.SpanStatusCode.OK}),G(c,"onGetEmbeddingsCached",r.metadataForCallbacks,w),e==null||e.debug("handleGetEmbeddings cached response: ",{cachedResponse:w}),w}e==null||e.debug("handleGetEmbeddings cache miss");let T=Date.now(),y=yield t.post(d.url,d.data,d.headers,l),C=Date.now()-T;e==null||e.debug("handleGetEmbeddings providerResponse: ",{providerResponse:y});let f={request:i,response:o.model.transformGetEmbeddingsResponse(y.data),cached:!1,latencyInMs:C,metadataForCallbacks:r.metadataForCallbacks,provider:{request:d,response:y}};return e==null||e.debug("handleGetEmbeddings response: ",{response:f}),o.enableCache&&(yield r.cache.set(p,f),e==null||e.debug("handleGetEmbeddings response cached")),s==null||s.setAttribute("cached",!1),s==null||s.setStatus({code:api.SpanStatusCode.OK}),G(c,"onGetEmbeddingsComplete",r.metadataForCallbacks,f),f}catch(i){e==null||e.warn("handleGetEmbeddings error: ",{error:i});let d;throw v.isHttpRequestError(i)||i instanceof E?d=i:d=F(i),G(c,"onGetEmbeddingsError",r.metadataForCallbacks,d),d}finally{s==null||s.end();}});return a?yield api.context.with(a,()=>m(this,null,function*(){return yield b.getTracer().startActiveSpan("get-embeddings.handler",o=>m(this,null,function*(){return yield n(o)}))})):yield n()})}var Ne=zod.z.object({model:zod.z.custom(),config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional(),customHeaders:zod.z.record(zod.z.string()).optional(),callbacks:zod.z.array(zod.z.custom()).nonempty().optional(),metadataForCallbacks:zod.z.any().optional()}),vs=zod.z.object({request:zod.z.object({config:types.Config(),messages:zod.z.array(types.Message()),tools:zod.z.array(types.Tool()).optional()}),response:types.PartialChatResponse,metadataForCallbacks:zod.z.any().optional(),provider:zod.z.object({request:zod.z.any(),response:zod.z.any()})});function Ve(r,t,a){return M(this,null,function*(){let e=h.getLogger(),n=function(s){return M(this,null,function*(){e==null||e.debug("handleStreamChat invoked"),e==null||e.debug("handleStreamChat request: ",{request:r});let o=Ne.parse(r),c=r.callbacks||[],l={config:o.config,messages:o.messages,tools:o.tools};try{G(c,"onStreamStart",r.metadataForCallbacks);let H={url:yield new g(o.model.getStreamChatUrl(o.config,o.messages,o.tools)),headers:yield new g(o.model.getStreamChatHeaders(o.config,o.messages,o.tools)),data:yield new g(o.model.getStreamChatData(o.config,o.messages,o.tools))};o.customHeaders&&(H.headers=q(q({},H.headers),o.customHeaders)),e==null||e.debug("handleStreamChat providerRequest: ",{providerRequest:H});let A="",I=!0;try{for(var y=D(t.stream(H.url,"post",H.data,H.headers)),C,f,w;C=!(f=yield new g(y.next())).done;C=!1){let j=f.value;try{for(var i=D(o.model.transformStreamChatResponseChunk(j,A)),d,p,T;d=!(p=yield new g(i.next())).done;d=!1){let Z=p.value;if(Z.partialResponse.partialMessages.length>0){let ee={request:l,response:Z.partialResponse,metadataForCallbacks:r.metadataForCallbacks,provider:{request:H,response:j}};G(c,I?"onStreamFirstResponse":"onStreamNewResponse",r.metadataForCallbacks,ee),I&&(I=!1),e==null||e.debug("handleStreamChat streamResponse: ",{streamResponse:ee}),yield ee;}else A=Z.buffer;}}catch(p){T=[p];}finally{try{d&&(p=i.return)&&(yield new g(p.call(i)));}finally{if(T)throw T[0]}}}}catch(f){w=[f];}finally{try{C&&(f=y.return)&&(yield new g(f.call(y)));}finally{if(w)throw w[0]}}s==null||s.setStatus({code:api.SpanStatusCode.OK}),G(c,"onStreamEnd",r.metadataForCallbacks);}catch(H){e==null||e.warn("handleStreamChat error: ",{error:H});let A;throw v.isHttpRequestError(H)||H instanceof E?A=H:A=F(H),G(c,"onStreamError",r.metadataForCallbacks,A),A}finally{s==null||s.end();}})};return a?yield*L(yield new g(api.context.with(a,()=>m(this,null,function*(){return yield b.getTracer().startActiveSpan("stream-chat.handler",o=>m(this,null,function*(){return yield n(o)}))})))):yield*L(n())})}var ne=class{constructor(t){var e,n,s,o,c,l,i;if(!t.dangerouslyAllowBrowser&&x())throw new E("It looks like you're running in a browser-like environment. This is disabled by default, as it risks exposing your provider secrets to attackers. If you understand the risks and have appropriate mitigation in place, you can set the `dangerouslyAllowBrowser` option to `true`.");this.options=He.parse(t),h.setLogger(t.logger),this.logger=t.logger,this.analytics=J.getAnalyticsRecorder(this.options.analyticsEnabled===void 0?!0:this.options.analyticsEnabled),b.setTracer((e=t.telemetry)==null?void 0:e.tracer),this.tracer=b.getTracer(),b.setMeter((n=t.telemetry)==null?void 0:n.meter),this.meter=b.getMeter();let a={maxConcurrentTasks:((s=this.options.queueOptions)==null?void 0:s.maxConcurrentTasks)||4,retryCount:((o=this.options.queueOptions)==null?void 0:o.retryCount)||3,retry:((c=this.options.queueOptions)==null?void 0:c.retry)||{initialDelay:1e3,exponentialFactor:2},timeout:((l=this.options.queueOptions)==null?void 0:l.timeout)||12e4};this.queues={completeChat:new U(a),getEmbeddings:new U(a)},this.httpClient=t.httpClient||new V({timeoutInMilliseconds:a.timeout*.9}),this.caches={completeChat:t.completeChatCache||new K,getEmbeddings:t.getEmbeddingsCache||new K},(i=this.logger)==null||i.debug("gateway initialized");}completeChat(t){return m(this,null,function*(){var n,s;(n=this.logger)==null||n.info("gateway.completeChat invoked"),(s=this.logger)==null||s.debug("request: ",{request:t});let a=z.parse(t),e=a.model.modelSchema.name;return yield this.tracer.startActiveSpan("complete-chat",o=>m(this,null,function*(){return o.setAttribute("modelName",e),new Promise((c,l)=>{var d;let i={id:uuid.v4(),request:a,cache:this.caches.completeChat,resolve:p=>{this.analytics.record("completeChat","success",{modelName:e,usage:p.response.usage||{}}),c(p);},reject:p=>{console.log("completeChat error",p),this.analytics.record("completeChat","error",{modelName:e}),l(p);},execute:this.executeCompleteChat.bind(this),telemetryContext:api.context.active()};this.queues.completeChat.enqueue(i),(d=this.logger)==null||d.debug(`gateway.completeChat task enqueued, id: ${i.id}`);})}))})}executeCompleteChat(t,a){return m(this,null,function*(){var n,s,o,c;(n=this.logger)==null||n.debug("gateway.executeCompleteChat invoked");let e=z.parse(t);return Pe({cache:this.caches.completeChat,model:e.model,config:e.config,messages:e.messages,tools:e.tools,enableCache:(o=(s=e.options)==null?void 0:s.enableCache)!=null?o:!0,callbacks:this.options.completeChatCallbacks,metadataForCallbacks:(c=e.options)==null?void 0:c.metadataForCallbacks},this.httpClient,a)})}streamChat(t){return M(this,null,function*(){var c,l,i,d;(c=this.logger)==null||c.info("gateway.streamChat invoked"),(l=this.logger)==null||l.debug("request: ",{request:t});let a=Ge.parse(t),e=a.model.modelSchema.name,n="success",s=this.tracer.startSpan("stream-chat"),o=api.trace.setSpan(api.context.active(),s);try{return s.setAttribute("modelName",e),yield*L(yield new g(api.context.with(o,()=>m(this,null,function*(){var p;return Ve({model:a.model,config:a.config,messages:a.messages,tools:a.tools,callbacks:this.options.streamChatCallbacks,metadataForCallbacks:(p=a.options)==null?void 0:p.metadataForCallbacks},this.httpClient,o)}))))}catch(p){throw n="error",s.setStatus({code:api.SpanStatusCode.ERROR,message:"stream failed"}),(i=this.logger)==null||i.error("gateway.streamChat error: ",{error:p}),p instanceof E?p:new E(p==null?void 0:p.message,500,(d=p==null?void 0:p.response)==null?void 0:d.data)}finally{this.analytics.record("streamChat",n,{modelName:e}),s.end();}})}getEmbeddings(t){return m(this,null,function*(){var n,s;(n=this.logger)==null||n.info("gateway.getEmbeddings invoked"),(s=this.logger)==null||s.debug("request: ",{request:t});let a=re.parse(t),e=a.model.modelSchema.name;return yield this.tracer.startActiveSpan("get-embeddings",o=>m(this,null,function*(){return o.setAttribute("modelName",e),new Promise((c,l)=>{var d;let i={id:uuid.v4(),request:a,cache:this.caches.getEmbeddings,resolve:p=>{this.analytics.record("getEmbeddings","success",{modelName:e,usage:p.response.usage||{}}),c(p);},reject:p=>{this.analytics.record("getEmbeddings","error",{modelName:e}),l(p);},execute:this.executeGetEmbeddings.bind(this),telemetryContext:api.context.active()};this.queues.getEmbeddings.enqueue(i),(d=this.logger)==null||d.debug(`gateway.getEmbeddings task enqueued, id: ${i.id}`);})}))})}executeGetEmbeddings(t,a){return m(this,null,function*(){var n,s,o,c;(n=this.logger)==null||n.debug("gateway.executeGetEmbeddings invoked");let e=re.parse(t);return De({cache:this.caches.getEmbeddings,model:e.model,config:e.config,embeddingRequests:e.embeddingRequests,enableCache:(o=(s=e.options)==null?void 0:s.enableCache)!=null?o:!0,callbacks:this.options.getEmbeddingsCallbacks,metadataForCallbacks:(c=e.options)==null?void 0:c.metadataForCallbacks},this.httpClient,a)})}};ne.GatewayError=E;
|
|
20
21
|
|
|
21
|
-
exports.
|
|
22
|
-
exports.
|
|
23
|
-
exports.
|
|
24
|
-
exports.
|
|
25
|
-
exports.
|
|
26
|
-
exports.
|
|
27
|
-
exports.
|
|
28
|
-
exports.
|
|
29
|
-
exports.
|
|
30
|
-
exports.
|
|
31
|
-
exports.
|
|
32
|
-
exports.
|
|
33
|
-
exports.
|
|
34
|
-
exports.
|
|
35
|
-
exports.
|
|
36
|
-
exports.
|
|
37
|
-
exports.
|
|
38
|
-
exports.
|
|
39
|
-
exports.
|
|
22
|
+
exports.AnalyticsManager = J;
|
|
23
|
+
exports.CompleteChatHandlerRequest = Me;
|
|
24
|
+
exports.CompleteChatHandlerResponse = Ka;
|
|
25
|
+
exports.ConsoleLogger = Re;
|
|
26
|
+
exports.Gateway = ne;
|
|
27
|
+
exports.GatewayError = E;
|
|
28
|
+
exports.GatewayTelemetryError = de;
|
|
29
|
+
exports.GetEmbeddingsHandlerRequest = $e;
|
|
30
|
+
exports.GetEmbeddingsHandlerResponse = is;
|
|
31
|
+
exports.HttpClientError = P;
|
|
32
|
+
exports.HttpRequestError = v;
|
|
33
|
+
exports.IsomorphicHttpClient = V;
|
|
34
|
+
exports.LRUCache = K;
|
|
35
|
+
exports.LoggerManager = h;
|
|
36
|
+
exports.NoOpAnalytics = Y;
|
|
37
|
+
exports.PostAnalytics = W;
|
|
38
|
+
exports.QueueOptions = fe;
|
|
39
|
+
exports.QueueTaskTimeoutError = B;
|
|
40
|
+
exports.SimpleQueue = U;
|
|
41
|
+
exports.StreamChatHandlerRequest = Ne;
|
|
42
|
+
exports.StreamChatHandlerResponse = vs;
|
|
43
|
+
exports.TelemetryManager = b;
|
|
44
|
+
exports.handleCompleteChat = Pe;
|
|
45
|
+
exports.handleGetEmbeddings = De;
|
|
46
|
+
exports.handleStreamChat = Ve;
|
|
40
47
|
//# sourceMappingURL=index.js.map
|
|
41
48
|
//# sourceMappingURL=index.js.map
|