@ax-llm/ax 14.0.1 → 14.0.3

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/index.cjs CHANGED
@@ -1,127 +1,127 @@
1
- "use strict";var li=Object.create;var vt=Object.defineProperty;var pi=Object.getOwnPropertyDescriptor;var ci=Object.getOwnPropertyNames;var ui=Object.getPrototypeOf,di=Object.prototype.hasOwnProperty;var mi=(s,e)=>{for(var t in e)vt(s,t,{get:e[t],enumerable:!0})},Jo=(s,e,t,n)=>{if(e&&typeof e=="object"||typeof e=="function")for(let o of ci(e))!di.call(s,o)&&o!==t&&vt(s,o,{get:()=>e[o],enumerable:!(n=pi(e,o))||n.enumerable});return s};var Ot=(s,e,t)=>(t=s!=null?li(ui(s)):{},Jo(e||!s||!s.__esModule?vt(t,"default",{value:s,enumerable:!0}):t,s)),gi=s=>Jo(vt({},"__esModule",{value:!0}),s);var Ma={};mi(Ma,{AxAI:()=>gt,AxAIAnthropic:()=>Ae,AxAIAnthropicModel:()=>Ze,AxAIAnthropicVertexModel:()=>Dt,AxAIAzureOpenAI:()=>be,AxAICohere:()=>Re,AxAICohereEmbedModel:()=>ot,AxAICohereModel:()=>nt,AxAIDeepSeek:()=>Ce,AxAIDeepSeekModel:()=>rt,AxAIGoogleGemini:()=>Te,AxAIGoogleGeminiEmbedModel:()=>Ht,AxAIGoogleGeminiEmbedTypes:()=>co,AxAIGoogleGeminiModel:()=>st,AxAIGoogleGeminiSafetyCategory:()=>Kt,AxAIGoogleGeminiSafetyThreshold:()=>Vt,AxAIGrok:()=>De,AxAIGrokEmbedModels:()=>To,AxAIGrokModel:()=>mt,AxAIGroq:()=>Se,AxAIGroqModel:()=>it,AxAIHuggingFace:()=>ve,AxAIHuggingFaceModel:()=>Yt,AxAIMistral:()=>Oe,AxAIMistralEmbedModels:()=>xo,AxAIMistralModel:()=>at,AxAIOllama:()=>Me,AxAIOpenAI:()=>Ie,AxAIOpenAIBase:()=>F,AxAIOpenAIEmbedModel:()=>xe,AxAIOpenAIModel:()=>tt,AxAIOpenAIResponses:()=>Ee,AxAIOpenAIResponsesBase:()=>lt,AxAIOpenAIResponsesImpl:()=>ke,AxAIOpenAIResponsesModel:()=>fe,AxAIRefusalError:()=>O,AxAIReka:()=>Pe,AxAIRekaModel:()=>ct,AxAIServiceAbortedError:()=>ge,AxAIServiceAuthenticationError:()=>ne,AxAIServiceError:()=>U,AxAIServiceNetworkError:()=>Z,AxAIServiceResponseError:()=>te,AxAIServiceStatusError:()=>le,AxAIServiceStreamTerminatedError:()=>K,AxAIServiceTimeoutError:()=>pe,AxAITogether:()=>Fe,AxAIWebLLM:()=>_e,AxAIWebLLMModel:()=>dt,AxAgent:()=>St,AxApacheTika:()=>Sn,AxAssertionError:()=>re,AxBalancer:()=>Ut,AxBaseAI:()=>E,AxBaseOptimizer:()=>ae,AxBootstrapFewShot:()=>ze,AxContentProcessingError:()=>V,AxDB:()=>mn,AxDBBase:()=>j,AxDBCloudflare:()=>Le,AxDBManager:()=>gn,AxDBMemory:()=>oe,AxDBPinecone:()=>Ge,AxDBWeaviate:()=>Ne,AxDefaultCostTracker:()=>Ct,AxDefaultResultReranker:()=>wn,AxDockerSession:()=>$n,AxEmbeddingAdapter:()=>Un,AxEvalUtil:()=>Ks,AxFlow:()=>wt,AxFlowDependencyAnalyzer:()=>je,AxFlowExecutionPlanner:()=>Ke,AxFlowSubContextImpl:()=>Je,AxFlowTypedSubContextImpl:()=>Gn,AxFluentFieldType:()=>L,AxFunctionError:()=>xt,AxFunctionProcessor:()=>ft,AxGen:()=>N,AxGenerateError:()=>Rt,AxHFDataLoader:()=>kn,AxInstanceRegistry:()=>Ue,AxLLMRequestTypeValues:()=>Vn,AxMCPClient:()=>Bn,AxMCPHTTPSSETransport:()=>qn,AxMCPStreambleHTTPTransport:()=>zn,AxMediaNotSupportedError:()=>z,AxMemory:()=>$e,AxMiPRO:()=>Dn,AxMockAIService:()=>en,AxMultiServiceRouter:()=>tn,AxProgram:()=>ie,AxPromptTemplate:()=>de,AxProviderRouter:()=>rn,AxRateLimiterTokenUsage:()=>we,AxSignature:()=>P,AxSignatureBuilder:()=>bt,AxSimpleClassifier:()=>On,AxSimpleClassifierClass:()=>vn,AxSpanKindValues:()=>Wn,AxStringUtil:()=>Tn,AxTestPrompt:()=>Mn,agent:()=>ii,ai:()=>ss,ax:()=>Ln,axAIAnthropicDefaultConfig:()=>oo,axAIAnthropicVertexDefaultConfig:()=>_r,axAIAzureOpenAIBestConfig:()=>Nr,axAIAzureOpenAICreativeConfig:()=>Lr,axAIAzureOpenAIDefaultConfig:()=>so,axAIAzureOpenAIFastConfig:()=>Gr,axAICohereCreativeConfig:()=>zr,axAICohereDefaultConfig:()=>lo,axAIDeepSeekCodeConfig:()=>jr,axAIDeepSeekDefaultConfig:()=>po,axAIGoogleGeminiDefaultConfig:()=>go,axAIGoogleGeminiDefaultCreativeConfig:()=>Kr,axAIGrokBestConfig:()=>rs,axAIGrokDefaultConfig:()=>un,axAIHuggingFaceCreativeConfig:()=>Vr,axAIHuggingFaceDefaultConfig:()=>Ao,axAIMistralBestConfig:()=>Wr,axAIMistralDefaultConfig:()=>Zt,axAIOllamaDefaultConfig:()=>fo,axAIOllamaDefaultCreativeConfig:()=>Jr,axAIOpenAIBestConfig:()=>Gt,axAIOpenAICreativeConfig:()=>Nt,axAIOpenAIDefaultConfig:()=>ce,axAIOpenAIFastConfig:()=>$t,axAIOpenAIResponsesBestConfig:()=>Qr,axAIOpenAIResponsesCreativeConfig:()=>Yr,axAIOpenAIResponsesDefaultConfig:()=>pt,axAIRekaBestConfig:()=>es,axAIRekaCreativeConfig:()=>ts,axAIRekaDefaultConfig:()=>ut,axAIRekaFastConfig:()=>ns,axAITogetherDefaultConfig:()=>bo,axAIWebLLMCreativeConfig:()=>os,axAIWebLLMDefaultConfig:()=>Co,axAnalyzeChatPromptRequirements:()=>Xr,axAnalyzeRequestRequirements:()=>ue,axBaseAIDefaultConfig:()=>w,axBaseAIDefaultCreativeConfig:()=>_,axCheckMetricsHealth:()=>is,axCreateDefaultColorLogger:()=>Kn,axCreateDefaultOptimizerColorLogger:()=>Ko,axCreateDefaultOptimizerTextLogger:()=>Ws,axCreateDefaultTextLogger:()=>rr,axCreateFlowColorLogger:()=>Tt,axCreateFlowTextLogger:()=>Zs,axDefaultFlowLogger:()=>ei,axDefaultMetricsConfig:()=>Oo,axDefaultOptimizerLogger:()=>En,axDefaultOptimizerMetricsConfig:()=>Vo,axGetCompatibilityReport:()=>$r,axGetFormatCompatibility:()=>Br,axGetMetricsConfig:()=>ls,axGetOptimizerMetricsConfig:()=>Qs,axGetProvidersWithMediaSupport:()=>Ur,axGlobals:()=>M,axModelInfoAnthropic:()=>et,axModelInfoCohere:()=>zt,axModelInfoDeepSeek:()=>jt,axModelInfoGoogleGemini:()=>Wt,axModelInfoGrok:()=>cn,axModelInfoGroq:()=>Jt,axModelInfoHuggingFace:()=>Qt,axModelInfoMistral:()=>Xt,axModelInfoOpenAI:()=>ye,axModelInfoOpenAIResponses:()=>Lt,axModelInfoReka:()=>on,axModelInfoTogether:()=>sn,axModelInfoWebLLM:()=>pn,axProcessContentForProvider:()=>nn,axRAG:()=>ai,axScoreProvidersForRequest:()=>Bt,axSelectOptimalProvider:()=>qt,axSpanAttributes:()=>T,axSpanEvents:()=>W,axUpdateMetricsConfig:()=>as,axUpdateOptimizerMetricsConfig:()=>Js,axValidateChatRequestMessage:()=>an,axValidateChatResponseResult:()=>ln,axValidateProviderCapabilities:()=>io,f:()=>Be,flow:()=>Nn,s:()=>Ys});module.exports=gi(Ma);function q({model:s,modelInfo:e,models:t}){let n=t?.find(l=>l.key===s),o=n&&"model"in n?n.model:s,r=e.find(l=>l.name===s);if(r)return r;let i=o.replace(/^(anthropic\.|openai\.)/,"").replace(/-latest$/,"").replace(/-\d{8}$/,"").replace(/-v\d+:\d+$/,"").replace(/@\d{8}$/,"").replace(/-\d{2,}(-[a-zA-Z0-9-]+)?$/,"").replace(/-v\d+@\d{8}$/,"").replace(/-v\d+$/,""),a=e.find(l=>l.name===i);return a||null}var Qo=(()=>{if(globalThis.crypto&&typeof globalThis.crypto.randomUUID=="function")return globalThis.crypto;throw new Error("Web Crypto API with randomUUID support not available. Requires Node.js 16+ or modern browser.")})();function $(){return Qo.randomUUID()}async function hi(s){let e=new TextEncoder,t=typeof s=="string"?e.encode(s):s,n=await Qo.subtle.digest("SHA-256",t);return Array.from(new Uint8Array(n)).map(i=>i.toString(16).padStart(2,"0")).join("")}var jn=class{data="";update(e){return this.data+=e,this}digest(e){if(e!=="hex")throw new Error("Only hex encoding is supported");let n=new TextEncoder().encode(this.data),o=0;for(let r=0;r<n.length;r++){let i=n[r];o=(o<<5)-o+i,o=o&o}return Math.abs(o).toString(16).padStart(8,"0")}async digestAsync(){return hi(this.data)}};function Mt(s){if(s!=="sha256")throw new Error("Only SHA-256 algorithm is supported");return new jn}var kt=class extends TransformStream{buffer="";currentEvent={rawData:""};dataParser;onError;constructor(e={}){super({transform:(t,n)=>this.handleChunk(t,n),flush:t=>this.handleFlush(t)}),this.dataParser=e.dataParser||JSON.parse,this.onError=e.onError||((t,n)=>{console.warn("Failed to parse event data:",t),console.log("Raw data that failed to parse:",n)})}handleChunk(e,t){this.buffer+=e,this.processBuffer(t)}handleFlush(e){this.processBuffer(e),this.currentEvent.rawData&&this.processEvent(e)}processBuffer(e){let n=this.buffer.replace(/\r\n|\r/g,`
1
+ "use strict";var ci=Object.create;var Ot=Object.defineProperty;var ui=Object.getOwnPropertyDescriptor;var di=Object.getOwnPropertyNames;var mi=Object.getPrototypeOf,gi=Object.prototype.hasOwnProperty;var hi=(s,e)=>{for(var t in e)Ot(s,t,{get:e[t],enumerable:!0})},Qr=(s,e,t,n)=>{if(e&&typeof e=="object"||typeof e=="function")for(let r of di(e))!gi.call(s,r)&&r!==t&&Ot(s,r,{get:()=>e[r],enumerable:!(n=ui(e,r))||n.enumerable});return s};var Mt=(s,e,t)=>(t=s!=null?ci(mi(s)):{},Qr(e||!s||!s.__esModule?Ot(t,"default",{value:s,enumerable:!0}):t,s)),Ai=s=>Qr(Ot({},"__esModule",{value:!0}),s);var Ea={};hi(Ea,{AxAI:()=>gt,AxAIAnthropic:()=>Ae,AxAIAnthropicModel:()=>Ze,AxAIAnthropicVertexModel:()=>Lt,AxAIAzureOpenAI:()=>be,AxAICohere:()=>Re,AxAICohereEmbedModel:()=>rt,AxAICohereModel:()=>nt,AxAIDeepSeek:()=>Te,AxAIDeepSeekModel:()=>ot,AxAIGoogleGemini:()=>Ce,AxAIGoogleGeminiEmbedModel:()=>Kt,AxAIGoogleGeminiEmbedTypes:()=>dr,AxAIGoogleGeminiModel:()=>st,AxAIGoogleGeminiSafetyCategory:()=>Vt,AxAIGoogleGeminiSafetyThreshold:()=>Wt,AxAIGrok:()=>De,AxAIGrokEmbedModels:()=>wr,AxAIGrokModel:()=>mt,AxAIGroq:()=>Se,AxAIGroqModel:()=>it,AxAIHuggingFace:()=>ve,AxAIHuggingFaceModel:()=>Xt,AxAIMistral:()=>Oe,AxAIMistralEmbedModels:()=>fr,AxAIMistralModel:()=>at,AxAIOllama:()=>Me,AxAIOpenAI:()=>Ie,AxAIOpenAIBase:()=>F,AxAIOpenAIEmbedModel:()=>xe,AxAIOpenAIModel:()=>tt,AxAIOpenAIResponses:()=>Ee,AxAIOpenAIResponsesBase:()=>lt,AxAIOpenAIResponsesImpl:()=>ke,AxAIOpenAIResponsesModel:()=>fe,AxAIRefusalError:()=>O,AxAIReka:()=>Pe,AxAIRekaModel:()=>ct,AxAIServiceAbortedError:()=>ge,AxAIServiceAuthenticationError:()=>ne,AxAIServiceError:()=>U,AxAIServiceNetworkError:()=>Z,AxAIServiceResponseError:()=>te,AxAIServiceStatusError:()=>le,AxAIServiceStreamTerminatedError:()=>K,AxAIServiceTimeoutError:()=>pe,AxAITogether:()=>Fe,AxAIWebLLM:()=>_e,AxAIWebLLMModel:()=>dt,AxAgent:()=>vt,AxApacheTika:()=>On,AxAssertionError:()=>oe,AxBalancer:()=>Bt,AxBaseAI:()=>E,AxBaseOptimizer:()=>ae,AxBootstrapFewShot:()=>ze,AxContentProcessingError:()=>V,AxDB:()=>gn,AxDBBase:()=>j,AxDBCloudflare:()=>Le,AxDBManager:()=>hn,AxDBMemory:()=>re,AxDBPinecone:()=>Ge,AxDBWeaviate:()=>$e,AxDefaultCostTracker:()=>Ct,AxDefaultResultReranker:()=>vn,AxDockerSession:()=>Bn,AxEmbeddingAdapter:()=>qn,AxEvalUtil:()=>Ws,AxFlow:()=>St,AxFlowDependencyAnalyzer:()=>je,AxFlowExecutionPlanner:()=>Ke,AxFlowSubContextImpl:()=>Je,AxFlowTypedSubContextImpl:()=>Nn,AxFluentFieldType:()=>L,AxFunctionError:()=>xt,AxFunctionProcessor:()=>ft,AxGen:()=>$,AxGenerateError:()=>Tt,AxHFDataLoader:()=>Pn,AxInstanceRegistry:()=>Ue,AxLLMRequestTypeValues:()=>Jn,AxMCPClient:()=>zn,AxMCPHTTPSSETransport:()=>jn,AxMCPStreambleHTTPTransport:()=>Hn,AxMediaNotSupportedError:()=>z,AxMemory:()=>Ne,AxMiPRO:()=>Gn,AxMockAIService:()=>tn,AxMultiServiceRouter:()=>nn,AxProgram:()=>ie,AxPromptTemplate:()=>de,AxProviderRouter:()=>sn,AxRateLimiterTokenUsage:()=>we,AxSignature:()=>P,AxSignatureBuilder:()=>bt,AxSimpleClassifier:()=>kn,AxSimpleClassifierClass:()=>Mn,AxSpanKindValues:()=>Qn,AxStringUtil:()=>Sn,AxTestPrompt:()=>En,agent:()=>li,ai:()=>as,ax:()=>$n,axAIAnthropicDefaultConfig:()=>sr,axAIAnthropicVertexDefaultConfig:()=>Lo,axAIAzureOpenAIBestConfig:()=>Uo,axAIAzureOpenAICreativeConfig:()=>$o,axAIAzureOpenAIDefaultConfig:()=>ar,axAIAzureOpenAIFastConfig:()=>No,axAICohereCreativeConfig:()=>Ho,axAICohereDefaultConfig:()=>cr,axAIDeepSeekCodeConfig:()=>Ko,axAIDeepSeekDefaultConfig:()=>ur,axAIGoogleGeminiDefaultConfig:()=>hr,axAIGoogleGeminiDefaultCreativeConfig:()=>Wo,axAIGrokBestConfig:()=>is,axAIGrokDefaultConfig:()=>dn,axAIHuggingFaceCreativeConfig:()=>Jo,axAIHuggingFaceDefaultConfig:()=>xr,axAIMistralBestConfig:()=>Qo,axAIMistralDefaultConfig:()=>en,axAIOllamaDefaultConfig:()=>yr,axAIOllamaDefaultCreativeConfig:()=>Yo,axAIOpenAIBestConfig:()=>$t,axAIOpenAICreativeConfig:()=>Nt,axAIOpenAIDefaultConfig:()=>ce,axAIOpenAIFastConfig:()=>Ut,axAIOpenAIResponsesBestConfig:()=>Xo,axAIOpenAIResponsesCreativeConfig:()=>Zo,axAIOpenAIResponsesDefaultConfig:()=>pt,axAIRekaBestConfig:()=>ns,axAIRekaCreativeConfig:()=>rs,axAIRekaDefaultConfig:()=>ut,axAIRekaFastConfig:()=>os,axAITogetherDefaultConfig:()=>Rr,axAIWebLLMCreativeConfig:()=>ss,axAIWebLLMDefaultConfig:()=>Cr,axAnalyzeChatPromptRequirements:()=>es,axAnalyzeRequestRequirements:()=>ue,axBaseAIDefaultConfig:()=>S,axBaseAIDefaultCreativeConfig:()=>_,axCheckMetricsHealth:()=>ls,axCreateDefaultColorLogger:()=>Wn,axCreateDefaultOptimizerColorLogger:()=>Vr,axCreateDefaultOptimizerTextLogger:()=>Qs,axCreateDefaultTextLogger:()=>so,axCreateFlowColorLogger:()=>wt,axCreateFlowTextLogger:()=>ti,axDefaultFlowLogger:()=>ni,axDefaultMetricsConfig:()=>Mr,axDefaultOptimizerLogger:()=>Fn,axDefaultOptimizerMetricsConfig:()=>Wr,axGetCompatibilityReport:()=>Bo,axGetFormatCompatibility:()=>zo,axGetMetricsConfig:()=>cs,axGetOptimizerMetricsConfig:()=>Xs,axGetProvidersWithMediaSupport:()=>qo,axGlobals:()=>M,axModelInfoAnthropic:()=>et,axModelInfoCohere:()=>jt,axModelInfoDeepSeek:()=>Ht,axModelInfoGoogleGemini:()=>Jt,axModelInfoGrok:()=>un,axModelInfoGroq:()=>Qt,axModelInfoHuggingFace:()=>Yt,axModelInfoMistral:()=>Zt,axModelInfoOpenAI:()=>ye,axModelInfoOpenAIResponses:()=>Gt,axModelInfoReka:()=>on,axModelInfoTogether:()=>an,axModelInfoWebLLM:()=>cn,axProcessContentForProvider:()=>rn,axRAG:()=>pi,axScoreProvidersForRequest:()=>qt,axSelectOptimalProvider:()=>zt,axSpanAttributes:()=>C,axSpanEvents:()=>W,axUpdateMetricsConfig:()=>ps,axUpdateOptimizerMetricsConfig:()=>Ys,axValidateChatRequestMessage:()=>ln,axValidateChatResponseResult:()=>pn,axValidateProviderCapabilities:()=>lr,f:()=>Be,flow:()=>Un,s:()=>Zs});module.exports=Ai(Ea);function q({model:s,modelInfo:e,models:t}){let n=t?.find(l=>l.key===s),r=n&&"model"in n?n.model:s,o=e.find(l=>l.name===s);if(o)return o;let i=r.replace(/^(anthropic\.|openai\.)/,"").replace(/-latest$/,"").replace(/-\d{8}$/,"").replace(/-v\d+:\d+$/,"").replace(/@\d{8}$/,"").replace(/-\d{2,}(-[a-zA-Z0-9-]+)?$/,"").replace(/-v\d+@\d{8}$/,"").replace(/-v\d+$/,""),a=e.find(l=>l.name===i);return a||null}var Yr=(()=>{if(globalThis.crypto&&typeof globalThis.crypto.randomUUID=="function")return globalThis.crypto;throw new Error("Web Crypto API with randomUUID support not available. Requires Node.js 16+ or modern browser.")})();function N(){return Yr.randomUUID()}async function xi(s){let e=new TextEncoder,t=typeof s=="string"?e.encode(s):s,n=await Yr.subtle.digest("SHA-256",t);return Array.from(new Uint8Array(n)).map(i=>i.toString(16).padStart(2,"0")).join("")}var Kn=class{data="";update(e){return this.data+=e,this}digest(e){if(e!=="hex")throw new Error("Only hex encoding is supported");let n=new TextEncoder().encode(this.data),r=0;for(let o=0;o<n.length;o++){let i=n[o];r=(r<<5)-r+i,r=r&r}return Math.abs(r).toString(16).padStart(8,"0")}async digestAsync(){return xi(this.data)}};function kt(s){if(s!=="sha256")throw new Error("Only SHA-256 algorithm is supported");return new Kn}var Et=class extends TransformStream{buffer="";currentEvent={rawData:""};dataParser;onError;constructor(e={}){super({transform:(t,n)=>this.handleChunk(t,n),flush:t=>this.handleFlush(t)}),this.dataParser=e.dataParser||JSON.parse,this.onError=e.onError||((t,n)=>{console.warn("Failed to parse event data:",t),console.log("Raw data that failed to parse:",n)})}handleChunk(e,t){this.buffer+=e,this.processBuffer(t)}handleFlush(e){this.processBuffer(e),this.currentEvent.rawData&&this.processEvent(e)}processBuffer(e){let n=this.buffer.replace(/\r\n|\r/g,`
2
2
  `).split(`
3
- `);this.buffer=n.pop()||"";for(let o of n)o===""?this.processEvent(e):this.parseLine(o)}parseLine(e){if(e.startsWith(":"))return;let t=e.indexOf(":");if(t===-1){this.currentEvent.rawData+=(this.currentEvent.rawData&&!this.currentEvent.rawData.endsWith(`
3
+ `);this.buffer=n.pop()||"";for(let r of n)r===""?this.processEvent(e):this.parseLine(r)}parseLine(e){if(e.startsWith(":"))return;let t=e.indexOf(":");if(t===-1){this.currentEvent.rawData+=(this.currentEvent.rawData&&!this.currentEvent.rawData.endsWith(`
4
4
  `)?`
5
- `:"")+e.trim();return}let n=e.slice(0,t).trim(),o=e.slice(t+1).trim();switch(n){case"event":this.currentEvent.event=o;break;case"data":this.currentEvent.rawData+=(this.currentEvent.rawData&&!this.currentEvent.rawData.endsWith(`
5
+ `:"")+e.trim();return}let n=e.slice(0,t).trim(),r=e.slice(t+1).trim();switch(n){case"event":this.currentEvent.event=r;break;case"data":this.currentEvent.rawData+=(this.currentEvent.rawData&&!this.currentEvent.rawData.endsWith(`
6
6
  `)?`
7
- `:"")+o;break;case"id":this.currentEvent.id=o;break;case"retry":{let r=Number.parseInt(o,10);Number.isNaN(r)||(this.currentEvent.retry=r);break}}}processEvent(e){if(this.currentEvent.rawData){if(this.currentEvent.event||(this.currentEvent.event="message"),this.currentEvent.rawData.trim()==="[DONE]"){this.currentEvent={rawData:""};return}try{let t=this.dataParser(this.currentEvent.rawData);e.enqueue(t)}catch(t){this.onError(t,this.currentEvent.rawData)}this.currentEvent={rawData:""}}}};var Hn=class{decoder;constructor(){this.decoder=new TextDecoder}transform(e,t){if(!(e instanceof ArrayBuffer||ArrayBuffer.isView(e)))throw new TypeError("Input data must be a BufferSource");let n=this.decoder.decode(e,{stream:!0});n.length!==0&&t.enqueue(n)}flush(e){let t=this.decoder.decode();t.length!==0&&e.enqueue(t)}},Et=class extends TransformStream{constructor(){super(new Hn)}};var Ai={maxRetries:3,initialDelayMs:1e3,maxDelayMs:6e4,backoffFactor:2,retryableStatusCodes:[500,408,429,502,503,504]},xi=globalThis.TextDecoderStream??Et,U=class extends Error{constructor(t,n,o,r,i={}){super(t);this.url=n;this.requestBody=o;this.responseBody=r;this.name=this.constructor.name,this.timestamp=new Date().toISOString(),this.errorId=$(),this.context=i,this.stack=this.toString()}timestamp;errorId;context;toString(){return[`${this.name}: ${this.message}`,`URL: ${this.url}`,`Request Body: ${JSON.stringify(this.requestBody,null,2)}`,`Response Body: ${JSON.stringify(this.responseBody,null,2)}`,`Context: ${JSON.stringify(this.context,null,2)}`,`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].join(`
8
- `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},le=class extends U{constructor(t,n,o,r,i,a){super(`HTTP ${t} - ${n}`,o,r,{httpStatus:t,httpStatusText:n,responseBody:i,...a});this.status=t;this.statusText=n;this.name=this.constructor.name}},Z=class extends U{constructor(t,n,o,r,i){super(`Network Error: ${t.message}`,n,o,r,{originalErrorName:t.name,originalErrorStack:t.stack,...i});this.originalError=t;this.name=this.constructor.name,this.stack=t.stack}},te=class extends U{constructor(e,t,n,o){super(e,t,n,void 0,o),this.name=this.constructor.name}},K=class extends U{constructor(t,n,o,r){super("Stream terminated unexpectedly by remote host",t,n,void 0,{lastChunk:o,...r});this.lastChunk=o;this.name=this.constructor.name}},pe=class extends U{constructor(e,t,n,o){super(`Request timed out after ${t}ms`,e,n,void 0,{timeoutMs:t,...o}),this.name=this.constructor.name}},ge=class extends U{constructor(e,t,n,o){super(`Request aborted${t?`: ${t}`:""}`,e,n,void 0,{abortReason:t,...o}),this.name=this.constructor.name}},ne=class extends U{constructor(e,t,n,o){super("Authentication failed",e,t,n,o),this.name=this.constructor.name}},O=class extends Error{constructor(t,n,o){super(`Model refused to fulfill request: ${t}`);this.refusalMessage=t;this.model=n;this.requestId=o;this.name="AxAIRefusalError",this.timestamp=new Date().toISOString(),this.errorId=$()}timestamp;errorId;toString(){return[`${this.name}: ${this.message}`,`Refusal: ${this.refusalMessage}`,this.model?`Model: ${this.model}`:"",this.requestId?`Request ID: ${this.requestId}`:"",`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].filter(Boolean).join(`
9
- `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},z=class extends Error{constructor(t,n,o=!1){super(`${t} not supported by ${n}${o?" (fallback available)":""}`);this.mediaType=t;this.provider=n;this.fallbackAvailable=o;this.name="AxMediaNotSupportedError",this.timestamp=new Date().toISOString(),this.errorId=$()}timestamp;errorId;toString(){return[`${this.name}: ${this.message}`,`Media Type: ${this.mediaType}`,`Provider: ${this.provider}`,`Fallback Available: ${this.fallbackAvailable}`,`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].join(`
10
- `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},V=class extends Error{constructor(t,n,o){super(`Failed to process ${n} during ${o}: ${t.message}`);this.originalError=t;this.contentType=n;this.processingStep=o;this.name="AxContentProcessingError",this.timestamp=new Date().toISOString(),this.errorId=$()}timestamp;errorId;toString(){return[`${this.name}: ${this.message}`,`Content Type: ${this.contentType}`,`Processing Step: ${this.processingStep}`,`Original Error: ${this.originalError.message}`,`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].join(`
11
- `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}};async function Yo(s){try{return s.headers.get("content-type")?.includes("application/json")?await s.json():await s.clone().text()}catch(e){return`[ReadableStream - read failed: ${e.message}]`}}function Xo(s,e){return Math.min(e.maxDelayMs,e.initialDelayMs*e.backoffFactor**s)*(.75+Math.random()*.5)}function fi(){return{startTime:Date.now(),retryCount:0}}function Zo(s){s.retryCount++,s.lastRetryTime=Date.now()}function er(s,e,t,n){return t>=n.maxRetries?!1:e&&n.retryableStatusCodes.includes(e)?!0:s instanceof Z&&!(s instanceof ne)}var B=async(s,e)=>{if(s.localCall)return await s.localCall(e,s.stream);if(!s.url)throw new Error("API URL is required when localCall is not provided");let t={...Ai,...s.retry},n=s.timeout,o=fi(),r,i=new URL(s.url),a=`${[i.pathname,s.name].filter(Boolean).join("/").replace(/\/+/g,"/")}${i.search}`,l=new URL(a,i);if(s.corsProxy){let u=l.href;l=new URL(`${s.corsProxy}?url=${encodeURIComponent(u)}`)}let p=$();if(s.validateRequest&&!await s.validateRequest(e))throw new te("Invalid request data",l.href,e,{validation:"request"});s.span?.setAttributes({"http.request.method":s.put?"PUT":"POST","url.full":l.href,"request.id":p,"request.startTime":o.startTime});let c=0;for(;;){let u=new AbortController;if(s.abortSignal){if(s.abortSignal.aborted)throw new ge(l.href,s.abortSignal.reason,e,{metrics:o});let d=()=>{u.abort(s.abortSignal.reason||"User aborted request")};s.abortSignal.addEventListener("abort",d,{once:!0});let m=u.abort.bind(u);u.abort=h=>{s.abortSignal.removeEventListener("abort",d),m(h)}}n&&(r=setTimeout(()=>{u.abort("Request timeout")},n));try{let d=await(s.fetch??fetch)(l,{method:s.put?"PUT":"POST",headers:{"Content-Type":"application/json","X-Request-ID":p,"X-Retry-Count":c.toString(),...s.headers},body:JSON.stringify(e),signal:u.signal});if(r&&clearTimeout(r),d.status===401||d.status===403){let x=await Yo(d);throw new ne(l.href,e,x,{metrics:o})}if(d.status>=400&&er(new Error,d.status,c,t)){let x=Xo(c,t);c++,Zo(o),s.span?.addEvent("retry",{attempt:c,delay:x,status:d.status,"metrics.startTime":o.startTime,"metrics.retryCount":o.retryCount,"metrics.lastRetryTime":o.lastRetryTime}),await new Promise(y=>setTimeout(y,x));continue}if(d.status>=400){let x=await Yo(d);throw new le(d.status,d.statusText,l.href,e,x,{metrics:o})}if(!s.stream){let x=await d.json();if(s.validateResponse&&!await s.validateResponse(x))throw new te("Invalid response data",l.href,e,{validation:"response"});return s.span?.setAttributes({"response.time":Date.now()-o.startTime,"response.retries":o.retryCount}),x}if(!d.body)throw new te("Response body is null",l.href,e,{metrics:o});let m,h=0;if(typeof window<"u"&&typeof EventSource<"u")return new ReadableStream({start(x){let y=d.body.getReader(),S=new TextDecoder,I="";async function R(){try{for(;;){let{done:k,value:v}=await y.read();if(k){A=!0,x.close();break}I+=S.decode(v,{stream:!0});let H=I.split(`
7
+ `:"")+r;break;case"id":this.currentEvent.id=r;break;case"retry":{let o=Number.parseInt(r,10);Number.isNaN(o)||(this.currentEvent.retry=o);break}}}processEvent(e){if(this.currentEvent.rawData){if(this.currentEvent.event||(this.currentEvent.event="message"),this.currentEvent.rawData.trim()==="[DONE]"){this.currentEvent={rawData:""};return}try{let t=this.dataParser(this.currentEvent.rawData);e.enqueue(t)}catch(t){this.onError(t,this.currentEvent.rawData)}this.currentEvent={rawData:""}}}};var Vn=class{decoder;constructor(){this.decoder=new TextDecoder}transform(e,t){if(!(e instanceof ArrayBuffer||ArrayBuffer.isView(e)))throw new TypeError("Input data must be a BufferSource");let n=this.decoder.decode(e,{stream:!0});n.length!==0&&t.enqueue(n)}flush(e){let t=this.decoder.decode();t.length!==0&&e.enqueue(t)}},Pt=class extends TransformStream{constructor(){super(new Vn)}};var fi={maxRetries:3,initialDelayMs:1e3,maxDelayMs:6e4,backoffFactor:2,retryableStatusCodes:[500,408,429,502,503,504]},yi=globalThis.TextDecoderStream??Pt,U=class extends Error{constructor(t,n,r,o,i={}){super(t);this.url=n;this.requestBody=r;this.responseBody=o;this.name=this.constructor.name,this.timestamp=new Date().toISOString(),this.errorId=N(),this.context=i,this.stack=this.toString()}timestamp;errorId;context;toString(){return[`${this.name}: ${this.message}`,`URL: ${this.url}`,`Request Body: ${JSON.stringify(this.requestBody,null,2)}`,`Response Body: ${JSON.stringify(this.responseBody,null,2)}`,`Context: ${JSON.stringify(this.context,null,2)}`,`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].join(`
8
+ `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},le=class extends U{constructor(t,n,r,o,i,a){super(`HTTP ${t} - ${n}`,r,o,{httpStatus:t,httpStatusText:n,responseBody:i,...a});this.status=t;this.statusText=n;this.name=this.constructor.name}},Z=class extends U{constructor(t,n,r,o,i){super(`Network Error: ${t.message}`,n,r,o,{originalErrorName:t.name,originalErrorStack:t.stack,...i});this.originalError=t;this.name=this.constructor.name,this.stack=t.stack}},te=class extends U{constructor(e,t,n,r){super(e,t,n,void 0,r),this.name=this.constructor.name}},K=class extends U{constructor(t,n,r,o){super("Stream terminated unexpectedly by remote host",t,n,void 0,{lastChunk:r,...o});this.lastChunk=r;this.name=this.constructor.name}},pe=class extends U{constructor(e,t,n,r){super(`Request timed out after ${t}ms`,e,n,void 0,{timeoutMs:t,...r}),this.name=this.constructor.name}},ge=class extends U{constructor(e,t,n,r){super(`Request aborted${t?`: ${t}`:""}`,e,n,void 0,{abortReason:t,...r}),this.name=this.constructor.name}},ne=class extends U{constructor(e,t,n,r){super("Authentication failed",e,t,n,r),this.name=this.constructor.name}},O=class extends Error{constructor(t,n,r){super(`Model refused to fulfill request: ${t}`);this.refusalMessage=t;this.model=n;this.requestId=r;this.name="AxAIRefusalError",this.timestamp=new Date().toISOString(),this.errorId=N()}timestamp;errorId;toString(){return[`${this.name}: ${this.message}`,`Refusal: ${this.refusalMessage}`,this.model?`Model: ${this.model}`:"",this.requestId?`Request ID: ${this.requestId}`:"",`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].filter(Boolean).join(`
9
+ `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},z=class extends Error{constructor(t,n,r=!1){super(`${t} not supported by ${n}${r?" (fallback available)":""}`);this.mediaType=t;this.provider=n;this.fallbackAvailable=r;this.name="AxMediaNotSupportedError",this.timestamp=new Date().toISOString(),this.errorId=N()}timestamp;errorId;toString(){return[`${this.name}: ${this.message}`,`Media Type: ${this.mediaType}`,`Provider: ${this.provider}`,`Fallback Available: ${this.fallbackAvailable}`,`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].join(`
10
+ `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},V=class extends Error{constructor(t,n,r){super(`Failed to process ${n} during ${r}: ${t.message}`);this.originalError=t;this.contentType=n;this.processingStep=r;this.name="AxContentProcessingError",this.timestamp=new Date().toISOString(),this.errorId=N()}timestamp;errorId;toString(){return[`${this.name}: ${this.message}`,`Content Type: ${this.contentType}`,`Processing Step: ${this.processingStep}`,`Original Error: ${this.originalError.message}`,`Timestamp: ${this.timestamp}`,`Error ID: ${this.errorId}`].join(`
11
+ `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}};async function Xr(s){try{return s.headers.get("content-type")?.includes("application/json")?await s.json():await s.clone().text()}catch(e){return`[ReadableStream - read failed: ${e.message}]`}}function Zr(s,e){return Math.min(e.maxDelayMs,e.initialDelayMs*e.backoffFactor**s)*(.75+Math.random()*.5)}function Ii(){return{startTime:Date.now(),retryCount:0}}function eo(s){s.retryCount++,s.lastRetryTime=Date.now()}function to(s,e,t,n){return t>=n.maxRetries?!1:e&&n.retryableStatusCodes.includes(e)?!0:s instanceof Z&&!(s instanceof ne)}var B=async(s,e)=>{if(s.localCall)return await s.localCall(e,s.stream);if(!s.url)throw new Error("API URL is required when localCall is not provided");let t={...fi,...s.retry},n=s.timeout,r=Ii(),o,i=new URL(s.url),a=`${[i.pathname,s.name].filter(Boolean).join("/").replace(/\/+/g,"/")}${i.search}`,l=new URL(a,i);if(s.corsProxy){let u=l.href;l=new URL(`${s.corsProxy}?url=${encodeURIComponent(u)}`)}let p=N();if(s.validateRequest&&!await s.validateRequest(e))throw new te("Invalid request data",l.href,e,{validation:"request"});s.span?.setAttributes({"http.request.method":s.put?"PUT":"POST","url.full":l.href,"request.id":p,"request.startTime":r.startTime});let c=0;for(;;){let u=new AbortController;if(s.abortSignal){if(s.abortSignal.aborted)throw new ge(l.href,s.abortSignal.reason,e,{metrics:r});let d=()=>{u.abort(s.abortSignal.reason||"User aborted request")};s.abortSignal.addEventListener("abort",d,{once:!0});let m=u.abort.bind(u);u.abort=h=>{s.abortSignal.removeEventListener("abort",d),m(h)}}n&&(o=setTimeout(()=>{u.abort("Request timeout")},n));try{let d=await(s.fetch??fetch)(l,{method:s.put?"PUT":"POST",headers:{"Content-Type":"application/json","X-Request-ID":p,"X-Retry-Count":c.toString(),...s.headers},body:JSON.stringify(e),signal:u.signal});if(o&&clearTimeout(o),d.status===401||d.status===403){let x=await Xr(d);throw new ne(l.href,e,x,{metrics:r})}if(d.status>=400&&to(new Error,d.status,c,t)){let x=Zr(c,t);c++,eo(r),s.span?.addEvent("retry",{attempt:c,delay:x,status:d.status,"metrics.startTime":r.startTime,"metrics.retryCount":r.retryCount,"metrics.lastRetryTime":r.lastRetryTime}),await new Promise(y=>setTimeout(y,x));continue}if(d.status>=400){let x=await Xr(d);throw new le(d.status,d.statusText,l.href,e,x,{metrics:r})}if(!s.stream){let x=await d.json();if(s.validateResponse&&!await s.validateResponse(x))throw new te("Invalid response data",l.href,e,{validation:"response"});return s.span?.setAttributes({"response.time":Date.now()-r.startTime,"response.retries":r.retryCount}),x}if(!d.body)throw new te("Response body is null",l.href,e,{metrics:r});let m,h=0;if(typeof window<"u"&&typeof EventSource<"u")return new ReadableStream({start(x){let y=d.body.getReader(),w=new TextDecoder,I="";async function R(){try{for(;;){let{done:k,value:v}=await y.read();if(k){A=!0,x.close();break}I+=w.decode(v,{stream:!0});let H=I.split(`
12
12
 
13
13
  `);I=H.pop()||"";for(let Y of H){if(!Y.trim())continue;let Qe=Y.split(`
14
- `),Ye="",Wo="message";for(let X of Qe)X.startsWith("data: ")?Ye=X.slice(6):X.startsWith("event: ")&&(Wo=X.slice(7));if(Ye){if(Ye==="[DONE]"){x.close();return}try{let X=JSON.parse(Ye);m=X,h++,o.streamChunks=h,o.lastChunkTime=Date.now(),console.log("!!!!>>>>>>>>>parsed",X),x.enqueue(X),s.span?.addEvent("stream.chunk",{"stream.chunks":h,"stream.duration":Date.now()-o.startTime,"response.retries":o.retryCount,"sse.event.type":Wo})}catch(X){s.debug&&console.warn("Skipping non-JSON SSE data:",Ye,X)}}}}}catch(k){let v=k,H={...o,streamDuration:Date.now()-o.startTime};v.name==="AbortError"||v.message?.includes("aborted")?x.error(new K(l.href,e,m,{streamMetrics:H})):x.error(new Z(v,l.href,e,"[ReadableStream - consumed during streaming]",{streamMetrics:H}))}finally{y.releaseLock()}}R()}});let f=new TransformStream({transform(x,y){m=x,h++,o.streamChunks=h,o.lastChunkTime=Date.now(),y.enqueue(x),s.span?.addEvent("stream.chunk",{"stream.chunks":h,"stream.duration":Date.now()-o.startTime,"response.retries":o.retryCount})}}),A=!1;return new ReadableStream({start(x){let y=d.body.pipeThrough(new xi).pipeThrough(new kt).pipeThrough(f).getReader();async function S(){try{for(;;){let{done:I,value:R}=await y.read();if(I){A||(A=!0,x.close());break}if(A)break;x.enqueue(R)}}catch(I){let R=I,k={...o,streamDuration:Date.now()-o.startTime};throw R.name==="AbortError"||R.message?.includes("aborted")?x.error(new K(l.href,e,m,{streamMetrics:k})):R instanceof TypeError&&R.message.includes("cancelled")?x.error(new K(l.href,e,m,{streamMetrics:k,cancelReason:"Stream cancelled by client"})):x.error(new Z(R,l.href,e,"[ReadableStream - consumed during streaming]",{streamMetrics:k})),R}finally{r&&clearTimeout(r),y.releaseLock()}}S()},cancel(){A=!0}})}catch(d){if(d instanceof Error&&d.name==="AbortError")throw s.abortSignal?.aborted?new ge(l.href,s.abortSignal.reason,e,{metrics:o}):new pe(l.href,n||0,e,{metrics:o});if(s.span?.isRecording()&&(s.span.recordException(d),s.span.setAttributes({"error.time":Date.now()-o.startTime,"error.retries":o.retryCount})),d instanceof Z&&er(d,void 0,c,t)){let m=Xo(c,t);c++,Zo(o),s.span?.addEvent("retry",{attempt:c,delay:m,error:d.message,"metrics.startTime":o.startTime,"metrics.retryCount":o.retryCount,"metrics.lastRetryTime":o.lastRetryTime}),await new Promise(h=>setTimeout(h,m));continue}throw d instanceof U&&(d.context.metrics=o),d}finally{r!==void 0&&clearTimeout(r)}}};var he=require("@opentelemetry/api");var M={signatureStrict:!0,tracer:void 0,meter:void 0,logger:void 0,optimizerLogger:void 0,functionResultFormatter:s=>typeof s=="string"?s:s==null?"":JSON.stringify(s,null,2)};var G=class{ANSI_WHITE_BRIGHT="\x1B[97m";ANSI_GREEN_BRIGHT="\x1B[92m";ANSI_BLUE_BRIGHT="\x1B[94m";ANSI_RED_BRIGHT="\x1B[91m";ANSI_YELLOW="\x1B[93m";ANSI_RED="\x1B[91m";ANSI_RESET="\x1B[0m";ANSI_ORANGE="\x1B[38;5;208m";ANSI_WHITE="\x1B[37m";ANSI_CYAN_BRIGHT="\x1B[96m";ANSI_MAGENTA_BRIGHT="\x1B[95m";ANSI_GRAY="\x1B[90m";ANSI_GREEN="\x1B[32m";ANSI_CYAN="\x1B[36m";ANSI_MAGENTA="\x1B[35m";ANSI_BLUE="\x1B[34m";ANSI_YELLOW_DIM="\x1B[33m";colorize(e,t){return`${t}${e}${this.ANSI_RESET}`}whiteBright(e){return this.colorize(e,this.ANSI_WHITE_BRIGHT)}greenBright(e){return this.colorize(e,this.ANSI_GREEN_BRIGHT)}blueBright(e){return this.colorize(e,this.ANSI_BLUE_BRIGHT)}redBright(e){return this.colorize(e,this.ANSI_RED_BRIGHT)}white(e){return this.colorize(e,this.ANSI_WHITE)}yellow(e){return this.colorize(e,this.ANSI_YELLOW)}red(e){return this.colorize(e,this.ANSI_RED)}orange(e){return this.colorize(e,this.ANSI_ORANGE)}cyanBright(e){return this.colorize(e,this.ANSI_CYAN_BRIGHT)}magentaBright(e){return this.colorize(e,this.ANSI_MAGENTA_BRIGHT)}gray(e){return this.colorize(e,this.ANSI_GRAY)}green(e){return this.colorize(e,this.ANSI_GREEN)}cyan(e){return this.colorize(e,this.ANSI_CYAN)}magenta(e){return this.colorize(e,this.ANSI_MAGENTA)}blue(e){return this.colorize(e,this.ANSI_BLUE)}yellowDim(e){return this.colorize(e,this.ANSI_YELLOW_DIM)}};var qa=new G,tr=s=>{console.log(s)},nr=(s,e,t)=>{let n=(o,r)=>t&&r&&r in t?t[r](o):o;switch(s.role){case"system":return`${n("[ SYSTEM ]","magentaBright")}
14
+ `),Ye="",Jr="message";for(let X of Qe)X.startsWith("data: ")?Ye=X.slice(6):X.startsWith("event: ")&&(Jr=X.slice(7));if(Ye){if(Ye==="[DONE]"){x.close();return}try{let X=JSON.parse(Ye);m=X,h++,r.streamChunks=h,r.lastChunkTime=Date.now(),console.log("!!!!>>>>>>>>>parsed",X),x.enqueue(X),s.span?.addEvent("stream.chunk",{"stream.chunks":h,"stream.duration":Date.now()-r.startTime,"response.retries":r.retryCount,"sse.event.type":Jr})}catch(X){s.debug&&console.warn("Skipping non-JSON SSE data:",Ye,X)}}}}}catch(k){let v=k,H={...r,streamDuration:Date.now()-r.startTime};v.name==="AbortError"||v.message?.includes("aborted")?x.error(new K(l.href,e,m,{streamMetrics:H})):x.error(new Z(v,l.href,e,"[ReadableStream - consumed during streaming]",{streamMetrics:H}))}finally{y.releaseLock()}}R()}});let f=new TransformStream({transform(x,y){m=x,h++,r.streamChunks=h,r.lastChunkTime=Date.now(),y.enqueue(x),s.span?.addEvent("stream.chunk",{"stream.chunks":h,"stream.duration":Date.now()-r.startTime,"response.retries":r.retryCount})}}),A=!1;return new ReadableStream({start(x){let y=d.body.pipeThrough(new yi).pipeThrough(new Et).pipeThrough(f).getReader();async function w(){try{for(;;){let{done:I,value:R}=await y.read();if(I){A||(A=!0,x.close());break}if(A)break;x.enqueue(R)}}catch(I){let R=I,k={...r,streamDuration:Date.now()-r.startTime};throw R.name==="AbortError"||R.message?.includes("aborted")?x.error(new K(l.href,e,m,{streamMetrics:k})):R instanceof TypeError&&R.message.includes("cancelled")?x.error(new K(l.href,e,m,{streamMetrics:k,cancelReason:"Stream cancelled by client"})):x.error(new Z(R,l.href,e,"[ReadableStream - consumed during streaming]",{streamMetrics:k})),R}finally{o&&clearTimeout(o),y.releaseLock()}}w()},cancel(){A=!0}})}catch(d){if(d instanceof Error&&d.name==="AbortError")throw s.abortSignal?.aborted?new ge(l.href,s.abortSignal.reason,e,{metrics:r}):new pe(l.href,n||0,e,{metrics:r});if(s.span?.isRecording()&&(s.span.recordException(d),s.span.setAttributes({"error.time":Date.now()-r.startTime,"error.retries":r.retryCount})),d instanceof Z&&to(d,void 0,c,t)){let m=Zr(c,t);c++,eo(r),s.span?.addEvent("retry",{attempt:c,delay:m,error:d.message,"metrics.startTime":r.startTime,"metrics.retryCount":r.retryCount,"metrics.lastRetryTime":r.lastRetryTime}),await new Promise(h=>setTimeout(h,m));continue}throw d instanceof U&&(d.context.metrics=r),d}finally{o!==void 0&&clearTimeout(o)}}};var he=require("@opentelemetry/api");var M={signatureStrict:!0,tracer:void 0,meter:void 0,logger:void 0,optimizerLogger:void 0,functionResultFormatter:s=>typeof s=="string"?s:s==null?"":JSON.stringify(s,null,2)};var G=class{ANSI_WHITE_BRIGHT="\x1B[97m";ANSI_GREEN_BRIGHT="\x1B[92m";ANSI_BLUE_BRIGHT="\x1B[94m";ANSI_RED_BRIGHT="\x1B[91m";ANSI_YELLOW="\x1B[93m";ANSI_RED="\x1B[91m";ANSI_RESET="\x1B[0m";ANSI_ORANGE="\x1B[38;5;208m";ANSI_WHITE="\x1B[37m";ANSI_CYAN_BRIGHT="\x1B[96m";ANSI_MAGENTA_BRIGHT="\x1B[95m";ANSI_GRAY="\x1B[90m";ANSI_GREEN="\x1B[32m";ANSI_CYAN="\x1B[36m";ANSI_MAGENTA="\x1B[35m";ANSI_BLUE="\x1B[34m";ANSI_YELLOW_DIM="\x1B[33m";colorize(e,t){return`${t}${e}${this.ANSI_RESET}`}whiteBright(e){return this.colorize(e,this.ANSI_WHITE_BRIGHT)}greenBright(e){return this.colorize(e,this.ANSI_GREEN_BRIGHT)}blueBright(e){return this.colorize(e,this.ANSI_BLUE_BRIGHT)}redBright(e){return this.colorize(e,this.ANSI_RED_BRIGHT)}white(e){return this.colorize(e,this.ANSI_WHITE)}yellow(e){return this.colorize(e,this.ANSI_YELLOW)}red(e){return this.colorize(e,this.ANSI_RED)}orange(e){return this.colorize(e,this.ANSI_ORANGE)}cyanBright(e){return this.colorize(e,this.ANSI_CYAN_BRIGHT)}magentaBright(e){return this.colorize(e,this.ANSI_MAGENTA_BRIGHT)}gray(e){return this.colorize(e,this.ANSI_GRAY)}green(e){return this.colorize(e,this.ANSI_GREEN)}cyan(e){return this.colorize(e,this.ANSI_CYAN)}magenta(e){return this.colorize(e,this.ANSI_MAGENTA)}blue(e){return this.colorize(e,this.ANSI_BLUE)}yellowDim(e){return this.colorize(e,this.ANSI_YELLOW_DIM)}};var ja=new G,no=s=>{console.log(s)},ro=(s,e,t)=>{let n=(r,o)=>t&&o&&o in t?t[o](r):r;switch(s.role){case"system":return`${n("[ SYSTEM ]","magentaBright")}
15
15
  ${n(s.content,"magenta")}`;case"function":return`${n("[ FUNCTION RESULT ]","yellow")}
16
- ${n(s.result??"[No result]","yellowDim")}`;case"user":{let o=`${n("[ USER ]","greenBright")}
17
- `;if(typeof s.content=="string")return o+n(s.content,"green");let r=s.content.map(i=>{if(i.type==="text")return n(i.text,"green");if(i.type==="image"){let a=e?"[Image]":`[Image: ${i.image}]`;return n(a,"green")}if(i.type==="audio"){let a=e?"[Audio]":`[Audio: ${i.data}]`;return n(a,"green")}return n("[Unknown content type]","gray")});return o+r.join(`
18
- `)}case"assistant":{let o=n("[ ASSISTANT","cyanBright");s.name&&(o+=` ${s.name}`),o+=" ]";let r=`${o}
19
- `;return s.content&&(r+=`${n(s.content,"cyan")}
20
- `),s.functionCalls&&s.functionCalls.length>0&&(r+=`${n("[ FUNCTION CALLS ]","yellow")}
21
- `,s.functionCalls.forEach((i,a)=>{let l=typeof i.function.params=="string"?i.function.params:JSON.stringify(i.function.params,null,2);r+=n(`${a+1}. ${i.function.name}(${l}) [id: ${i.id}]`,"yellowDim"),a<(s.functionCalls?.length??0)-1&&(r+=`
22
- `)}),r+=`
23
- `),!s.content&&(!s.functionCalls||s.functionCalls.length===0)&&(r+=n("[No content]","gray")),r}default:return`${n("[ UNKNOWN ]","redBright")}
24
- ${n(JSON.stringify(s),"gray")}`}},Kn=(s=tr)=>{let e=new G,t=e.gray(`${"\u2500".repeat(60)}
25
- `);return n=>{let o=n,r="";switch(o.name){case"ChatRequestChatPrompt":r=`
26
- ${e.blueBright(`[ CHAT REQUEST Step ${o.step} ]`)}
16
+ ${n(s.result??"[No result]","yellowDim")}`;case"user":{let r=`${n("[ USER ]","greenBright")}
17
+ `;if(typeof s.content=="string")return r+n(s.content,"green");let o=s.content.map(i=>{if(i.type==="text")return n(i.text,"green");if(i.type==="image"){let a=e?"[Image]":`[Image: ${i.image}]`;return n(a,"green")}if(i.type==="audio"){let a=e?"[Audio]":`[Audio: ${i.data}]`;return n(a,"green")}return n("[Unknown content type]","gray")});return r+o.join(`
18
+ `)}case"assistant":{let r=n("[ ASSISTANT","cyanBright");s.name&&(r+=` ${s.name}`),r+=" ]";let o=`${r}
19
+ `;return s.content&&(o+=`${n(s.content,"cyan")}
20
+ `),s.functionCalls&&s.functionCalls.length>0&&(o+=`${n("[ FUNCTION CALLS ]","yellow")}
21
+ `,s.functionCalls.forEach((i,a)=>{let l=typeof i.function.params=="string"?i.function.params:JSON.stringify(i.function.params,null,2);o+=n(`${a+1}. ${i.function.name}(${l}) [id: ${i.id}]`,"yellowDim"),a<(s.functionCalls?.length??0)-1&&(o+=`
22
+ `)}),o+=`
23
+ `),!s.content&&(!s.functionCalls||s.functionCalls.length===0)&&(o+=n("[No content]","gray")),o}default:return`${n("[ UNKNOWN ]","redBright")}
24
+ ${n(JSON.stringify(s),"gray")}`}},Wn=(s=no)=>{let e=new G,t=e.gray(`${"\u2500".repeat(60)}
25
+ `);return n=>{let r=n,o="";switch(r.name){case"ChatRequestChatPrompt":o=`
26
+ ${e.blueBright(`[ CHAT REQUEST Step ${r.step} ]`)}
27
27
  ${t}
28
- `,o.value.forEach((i,a)=>{r+=nr(i,void 0,e),a<o.value.length-1&&(r+=`
28
+ `,r.value.forEach((i,a)=>{o+=ro(i,void 0,e),a<r.value.length-1&&(o+=`
29
29
  ${t}
30
- `)}),r+=`
31
- ${t}`;break;case"FunctionResults":r=`
30
+ `)}),o+=`
31
+ ${t}`;break;case"FunctionResults":o=`
32
32
  ${e.yellow("[ FUNCTION RESULTS ]")}
33
- `,o.value.forEach((i,a)=>{r+=e.yellowDim(`Function: ${i.functionId}
34
- Result: ${i.result}`),a<o.value.length-1&&(r+=`
33
+ `,r.value.forEach((i,a)=>{o+=e.yellowDim(`Function: ${i.functionId}
34
+ Result: ${i.result}`),a<r.value.length-1&&(o+=`
35
35
  ${t}
36
- `)});break;case"ChatResponseResults":r=`
36
+ `)});break;case"ChatResponseResults":o=`
37
37
  ${e.cyanBright("[ CHAT RESPONSE ]")}
38
- `,o.value.forEach((i,a)=>{r+=e.cyan(i.content||"[No content]"),a<o.value.length-1&&(r+=`
38
+ `,r.value.forEach((i,a)=>{o+=e.cyan(i.content||"[No content]"),a<r.value.length-1&&(o+=`
39
39
  ${t}
40
- `)});break;case"ChatResponseStreamingResult":{let i=o.value.delta||o.value.content||"";r=e.cyanBright(i);return}case"ChatResponseStreamingDoneResult":{r=`
40
+ `)});break;case"ChatResponseStreamingResult":{let i=r.value.delta||r.value.content||"";o=e.cyanBright(i);return}case"ChatResponseStreamingDoneResult":{o=`
41
41
  ${e.cyanBright("[ CHAT RESPONSE ]")}
42
42
  ${t}
43
- `,o.value.content&&(r+=e.cyanBright(o.value.content)),o.value.functionCalls&&(r+=e.cyanBright(JSON.stringify(o.value.functionCalls,null,2)));break}case"FunctionError":r=`
44
- ${e.redBright(`[ FUNCTION ERROR #${o.index} ]`)}
43
+ `,r.value.content&&(o+=e.cyanBright(r.value.content)),r.value.functionCalls&&(o+=e.cyanBright(JSON.stringify(r.value.functionCalls,null,2)));break}case"FunctionError":o=`
44
+ ${e.redBright(`[ FUNCTION ERROR #${r.index} ]`)}
45
45
  ${t}
46
- ${e.white(o.fixingInstructions)}
47
- ${e.red(`Error: ${o.error}`)}`;break;case"ValidationError":r=`
48
- ${e.redBright(`[ VALIDATION ERROR #${o.index} ]`)}
46
+ ${e.white(r.fixingInstructions)}
47
+ ${e.red(`Error: ${r.error}`)}`;break;case"ValidationError":o=`
48
+ ${e.redBright(`[ VALIDATION ERROR #${r.index} ]`)}
49
49
  ${t}
50
- ${e.white(o.fixingInstructions)}
51
- ${e.red(`Error: ${o.error}`)}`;break;case"AssertionError":r=`
52
- ${e.redBright(`[ ASSERTION ERROR #${o.index} ]`)}
50
+ ${e.white(r.fixingInstructions)}
51
+ ${e.red(`Error: ${r.error}`)}`;break;case"AssertionError":o=`
52
+ ${e.redBright(`[ ASSERTION ERROR #${r.index} ]`)}
53
53
  ${t}
54
- ${e.white(o.fixingInstructions)}
55
- ${e.red(`Error: ${o.error}`)}`;break;case"ResultPickerUsed":r=`${e.greenBright("[ RESULT PICKER ]")}
54
+ ${e.white(r.fixingInstructions)}
55
+ ${e.red(`Error: ${r.error}`)}`;break;case"ResultPickerUsed":o=`${e.greenBright("[ RESULT PICKER ]")}
56
56
  ${t}
57
- ${e.green(`Selected sample ${o.selectedIndex+1} of ${o.sampleCount} (${o.latency.toFixed(2)}ms)`)}`;break;case"Notification":r=`${e.gray(`[ NOTIFICATION ${o.id} ]`)}
57
+ ${e.green(`Selected sample ${r.selectedIndex+1} of ${r.sampleCount} (${r.latency.toFixed(2)}ms)`)}`;break;case"Notification":o=`${e.gray(`[ NOTIFICATION ${r.id} ]`)}
58
58
  ${t}
59
- ${e.white(o.value)}`;break;case"EmbedRequest":r=`${e.orange(`[ EMBED REQUEST ${o.embedModel} ]`)}
59
+ ${e.white(r.value)}`;break;case"EmbedRequest":o=`${e.orange(`[ EMBED REQUEST ${r.embedModel} ]`)}
60
60
  ${t}
61
- `,o.value.forEach((i,a)=>{r+=e.white(`Text ${a+1}: ${i.substring(0,100)}${i.length>100?"...":""}`),a<o.value.length-1&&(r+=`
61
+ `,r.value.forEach((i,a)=>{o+=e.white(`Text ${a+1}: ${i.substring(0,100)}${i.length>100?"...":""}`),a<r.value.length-1&&(o+=`
62
62
  ${t}
63
- `)});break;case"EmbedResponse":r=`${e.orange(`[ EMBED RESPONSE (${o.totalEmbeddings} embeddings) ]`)}
63
+ `)});break;case"EmbedResponse":o=`${e.orange(`[ EMBED RESPONSE (${r.totalEmbeddings} embeddings) ]`)}
64
64
  ${t}
65
- `,o.value.forEach((i,a)=>{r+=e.white(`Embedding ${a+1}: [${i.sample.join(", ")}${i.truncated?", ...":""}] (length: ${i.length})`),a<o.value.length-1&&(r+=`
65
+ `,r.value.forEach((i,a)=>{o+=e.white(`Embedding ${a+1}: [${i.sample.join(", ")}${i.truncated?", ...":""}] (length: ${i.length})`),a<r.value.length-1&&(o+=`
66
66
  ${t}
67
- `)});break;default:r=e.gray(JSON.stringify(o,null,2))}s(r)}},or=Kn(),rr=(s=tr)=>{let e="\u2500".repeat(60);return t=>{let n=t,o="";switch(n.name){case"ChatRequestChatPrompt":o=`
67
+ `)});break;default:o=e.gray(JSON.stringify(r,null,2))}s(o)}},oo=Wn(),so=(s=no)=>{let e="\u2500".repeat(60);return t=>{let n=t,r="";switch(n.name){case"ChatRequestChatPrompt":r=`
68
68
  [ CHAT REQUEST Step ${n.step} ]
69
69
  ${e}
70
- `,n.value.forEach((r,i)=>{o+=nr(r),i<n.value.length-1&&(o+=`
70
+ `,n.value.forEach((o,i)=>{r+=ro(o),i<n.value.length-1&&(r+=`
71
71
  ${e}
72
- `)}),o+=`
73
- ${e}`;break;case"FunctionResults":o=`
72
+ `)}),r+=`
73
+ ${e}`;break;case"FunctionResults":r=`
74
74
  [ FUNCTION RESULTS ]
75
75
  ${e}
76
- `,n.value.forEach((r,i)=>{o+=`Function: ${r.functionId}
77
- Result: ${r.result}`,i<n.value.length-1&&(o+=`
76
+ `,n.value.forEach((o,i)=>{r+=`Function: ${o.functionId}
77
+ Result: ${o.result}`,i<n.value.length-1&&(r+=`
78
78
  ${e}
79
- `)});break;case"ChatResponseResults":o=`
79
+ `)});break;case"ChatResponseResults":r=`
80
80
  [ CHAT RESPONSE ]
81
- `,n.value.forEach((r,i)=>{o+=r.content||"[No content]",i<n.value.length-1&&(o+=`
81
+ `,n.value.forEach((o,i)=>{r+=o.content||"[No content]",i<n.value.length-1&&(r+=`
82
82
  ${e}
83
- `)});break;case"ChatResponseStreamingResult":return;case"ChatResponseStreamingDoneResult":{o=`
83
+ `)});break;case"ChatResponseStreamingResult":return;case"ChatResponseStreamingDoneResult":{r=`
84
84
  [ CHAT RESPONSE ]
85
- `,n.value.content&&(o+=n.value.content),n.value.functionCalls&&(o+=JSON.stringify(n.value.functionCalls,null,2));break}case"FunctionError":o=`
85
+ `,n.value.content&&(r+=n.value.content),n.value.functionCalls&&(r+=JSON.stringify(n.value.functionCalls,null,2));break}case"FunctionError":r=`
86
86
  [ FUNCTION ERROR #${n.index} ]
87
87
  ${e}
88
88
  ${n.fixingInstructions}
89
- Error: ${n.error}`;break;case"ValidationError":o=`
89
+ Error: ${n.error}`;break;case"ValidationError":r=`
90
90
  [ VALIDATION ERROR #${n.index} ]
91
91
  ${e}
92
92
  ${n.fixingInstructions}
93
- Error: ${n.error}`;break;case"AssertionError":o=`
93
+ Error: ${n.error}`;break;case"AssertionError":r=`
94
94
  [ ASSERTION ERROR #${n.index} ]
95
95
  ${e}
96
96
  ${n.fixingInstructions}
97
- Error: ${n.error}`;break;case"ResultPickerUsed":o=`[ RESULT PICKER ]
97
+ Error: ${n.error}`;break;case"ResultPickerUsed":r=`[ RESULT PICKER ]
98
98
  ${e}
99
- Selected sample ${n.selectedIndex+1} of ${n.sampleCount} (${n.latency.toFixed(2)}ms)`;break;case"Notification":o=`[ NOTIFICATION ${n.id} ]
99
+ Selected sample ${n.selectedIndex+1} of ${n.sampleCount} (${n.latency.toFixed(2)}ms)`;break;case"Notification":r=`[ NOTIFICATION ${n.id} ]
100
100
  ${e}
101
- ${n.value}`;break;case"EmbedRequest":o=`[ EMBED REQUEST ${n.embedModel} ]
101
+ ${n.value}`;break;case"EmbedRequest":r=`[ EMBED REQUEST ${n.embedModel} ]
102
102
  ${e}
103
- `,n.value.forEach((r,i)=>{o+=`Text ${i+1}: ${r.substring(0,100)}${r.length>100?"...":""}`,i<n.value.length-1&&(o+=`
103
+ `,n.value.forEach((o,i)=>{r+=`Text ${i+1}: ${o.substring(0,100)}${o.length>100?"...":""}`,i<n.value.length-1&&(r+=`
104
104
  ${e}
105
- `)});break;case"EmbedResponse":o=`[ EMBED RESPONSE (${n.totalEmbeddings} embeddings) ]
105
+ `)});break;case"EmbedResponse":r=`[ EMBED RESPONSE (${n.totalEmbeddings} embeddings) ]
106
106
  ${e}
107
- `,n.value.forEach((r,i)=>{o+=`Embedding ${i+1}: [${r.sample.join(", ")}${r.truncated?", ...":""}] (length: ${r.length})`,i<n.value.length-1&&(o+=`
107
+ `,n.value.forEach((o,i)=>{r+=`Embedding ${i+1}: [${o.sample.join(", ")}${o.truncated?", ...":""}] (length: ${o.length})`,i<n.value.length-1&&(r+=`
108
108
  ${e}
109
- `)});break;default:o=JSON.stringify(n,null,2)}s(o)}};var T={LLM_SYSTEM:"gen_ai.system",LLM_OPERATION_NAME:"gen_ai.operation.name",LLM_REQUEST_MODEL:"gen_ai.request.model",LLM_REQUEST_MAX_TOKENS:"gen_ai.request.max_tokens",LLM_REQUEST_TEMPERATURE:"gen_ai.request.temperature",LLM_REQUEST_TOP_K:"gen_ai.request.top_k",LLM_REQUEST_FREQUENCY_PENALTY:"gen_ai.request.frequency_penalty",LLM_REQUEST_PRESENCE_PENALTY:"gen_ai.request.presence_penalty",LLM_REQUEST_STOP_SEQUENCES:"gen_ai.request.stop_sequences",LLM_REQUEST_LLM_IS_STREAMING:"gen_ai.request.llm_is_streaming",LLM_REQUEST_TOP_P:"gen_ai.request.top_p",LLM_USAGE_INPUT_TOKENS:"gen_ai.usage.input_tokens",LLM_USAGE_OUTPUT_TOKENS:"gen_ai.usage.output_tokens",LLM_USAGE_TOTAL_TOKENS:"gen_ai.usage.total_tokens",LLM_USAGE_THOUGHTS_TOKENS:"gen_ai.usage.thoughts_tokens",DB_SYSTEM:"db.system",DB_TABLE:"db.table",DB_NAMESPACE:"db.namespace",DB_ID:"db.id",DB_QUERY_TEXT:"db.query.text",DB_VECTOR:"db.vector",DB_OPERATION_NAME:"db.operation.name",DB_VECTOR_QUERY_TOP_K:"db.vector.query.top_k",DB_QUERY_EMBEDDINGS:"db.query.embeddings",DB_QUERY_RESULT:"db.query.result",DB_QUERY_EMBEDDINGS_VECTOR:"db.query.embeddings.vector",DB_QUERY_RESULT_ID:"db.query.result.id",DB_QUERY_RESULT_SCORE:"db.query.result.score",DB_QUERY_RESULT_DISTANCE:"db.query.result.distance",DB_QUERY_RESULT_METADATA:"db.query.result.metadata",DB_QUERY_RESULT_VECTOR:"db.query.result.vector",DB_QUERY_RESULT_DOCUMENT:"db.query.result.document"},W={GEN_AI_USER_MESSAGE:"gen_ai.user.message",GEN_AI_SYSTEM_MESSAGE:"gen_ai.system.message",GEN_AI_ASSISTANT_MESSAGE:"gen_ai.assistant.message",GEN_AI_TOOL_MESSAGE:"gen_ai.tool.message",GEN_AI_CHOICE:"gen_ai.choice",GEN_AI_USAGE:"gen_ai.usage"},Vn=(o=>(o.COMPLETION="completion",o.CHAT="chat",o.RERANK="rerank",o.UNKNOWN="unknown",o))(Vn||{}),Wn=(r=>(r.WORKFLOW="workflow",r.TASK="task",r.AGENT="agent",r.TOOL="tool",r.UNKNOWN="unknown",r))(Wn||{});var Jn=class{buffer;doneCallback;transformFn;constructor(e,t){this.transformFn=e,this.doneCallback=t,this.buffer=t?[]:void 0}async transform(e,t){let n=this.transformFn(e);n&&(t.enqueue(n),this.buffer?.push(n))}async flush(e){await this.doneCallback?.(this.buffer??[]),e.terminate()}},Pt=class extends TransformStream{constructor(e,t){super(new Jn(e,t))}};function Ft(s,e){for(let t of e){let n=s.find(o=>o.id===t.id);n?(typeof t.function.name=="string"&&t.function.name.length>0&&(n.function.name+=t.function.name),typeof t.function.params=="string"&&t.function.params.length>0&&(n.function.params+=t.function.params),typeof t.function.params=="object"&&(n.function.params=t.function.params)):s.push(t)}}var sr=(s,e,t,n)=>{let o=n?s.filter(i=>i.role!=="system"):[...s];t({name:"ChatRequestChatPrompt",step:e,value:o})};var ir=(s,e)=>{if(!s.results)return;let t={name:"ChatResponseResults",value:s.results};e(t)},ar=(s,e,t)=>{t({name:"ChatResponseStreamingResult",index:e,value:s})};function lr(s,e){let t=new Map;for(let n of s)for(let o of n.results){if(!o)continue;let r=t.get(o.index);r?(o.content&&(r.content=(r.content??"")+o.content),o.thought&&(r.thought=(r.thought??"")+o.thought),o.finishReason&&(r.finishReason=o.finishReason),o.functionCalls&&(r.functionCalls?Ft(r.functionCalls,structuredClone(o.functionCalls)):r.functionCalls=structuredClone(o.functionCalls))):(r=structuredClone(o),t.set(o.index,r))}for(let n of t.values()){let o={name:"ChatResponseStreamingDoneResult",index:n.index,value:n};e(o)}}var pr=(s,e)=>{e({name:"FunctionResults",value:s})},cr=(s,e,t,n)=>{n({name:"FunctionError",index:e,fixingInstructions:t,error:s})},ur=(s,e,t,n)=>{n({name:"ValidationError",index:e,fixingInstructions:t,error:s})},dr=(s,e,t,n)=>{n({name:"AssertionError",index:e,fixingInstructions:t,error:s})},mr=(s,e,t)=>{t({name:"RefusalError",index:e,error:s})};var gr=(s,e,t)=>{t({name:"EmbedRequest",embedModel:e,value:s})},hr=(s,e)=>{let t=s.slice(0,3).map(o=>({length:o.length,sample:o.slice(0,5),truncated:o.length>5})),n={name:"EmbedResponse",totalEmbeddings:s.length,value:t};e(n)},Ar=(s,e,t,n)=>{n({name:"ResultPickerUsed",sampleCount:s,selectedIndex:e,latency:t})};var Qn=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let o=String(n);e[t]=o.length>100?o.substring(0,100):o}return e},_t,xr=s=>{if(_t)return _t;if(s)return _t=yi(s),_t};var yi=s=>({latencyHistogram:s.createHistogram("ax_llm_request_duration_ms",{description:"Duration of LLM requests in milliseconds",unit:"ms"}),errorCounter:s.createCounter("ax_llm_errors_total",{description:"Total number of LLM request errors"}),requestCounter:s.createCounter("ax_llm_requests_total",{description:"Total number of LLM requests"}),tokenCounter:s.createCounter("ax_llm_tokens_total",{description:"Total number of LLM tokens consumed"}),inputTokenCounter:s.createCounter("ax_llm_input_tokens_total",{description:"Total number of input/prompt tokens consumed"}),outputTokenCounter:s.createCounter("ax_llm_output_tokens_total",{description:"Total number of output/completion tokens generated"}),errorRateGauge:s.createGauge("ax_llm_error_rate",{description:"Current error rate as a percentage (0-100)"}),meanLatencyGauge:s.createGauge("ax_llm_mean_latency_ms",{description:"Mean latency of LLM requests in milliseconds",unit:"ms"}),p95LatencyGauge:s.createGauge("ax_llm_p95_latency_ms",{description:"95th percentile latency of LLM requests in milliseconds",unit:"ms"}),p99LatencyGauge:s.createGauge("ax_llm_p99_latency_ms",{description:"99th percentile latency of LLM requests in milliseconds",unit:"ms"}),streamingRequestsCounter:s.createCounter("ax_llm_streaming_requests_total",{description:"Total number of streaming LLM requests"}),functionCallsCounter:s.createCounter("ax_llm_function_calls_total",{description:"Total number of function/tool calls made"}),functionCallLatencyHistogram:s.createHistogram("ax_llm_function_call_latency_ms",{description:"Latency of function calls in milliseconds",unit:"ms"}),requestSizeHistogram:s.createHistogram("ax_llm_request_size_bytes",{description:"Size of LLM request payloads in bytes",unit:"By"}),responseSizeHistogram:s.createHistogram("ax_llm_response_size_bytes",{description:"Size of LLM response payloads in bytes",unit:"By"}),temperatureGauge:s.createGauge("ax_llm_temperature_gauge",{description:"Temperature setting used for LLM requests"}),maxTokensGauge:s.createGauge("ax_llm_max_tokens_gauge",{description:"Maximum tokens setting used for LLM requests"}),estimatedCostCounter:s.createCounter("ax_llm_estimated_cost_total",{description:"Estimated cost of LLM requests in USD",unit:"$"}),promptLengthHistogram:s.createHistogram("ax_llm_prompt_length_chars",{description:"Length of prompts in characters"}),contextWindowUsageGauge:s.createGauge("ax_llm_context_window_usage_ratio",{description:"Context window utilization ratio (0-1)"}),timeoutsCounter:s.createCounter("ax_llm_timeouts_total",{description:"Total number of timed out LLM requests"}),abortsCounter:s.createCounter("ax_llm_aborts_total",{description:"Total number of aborted LLM requests"}),thinkingBudgetUsageCounter:s.createCounter("ax_llm_thinking_budget_usage_total",{description:"Total thinking budget tokens used"}),multimodalRequestsCounter:s.createCounter("ax_llm_multimodal_requests_total",{description:"Total number of multimodal requests (with images/audio)"})}),fr=(s,e,t,n,o)=>{try{if(s.latencyHistogram){let r=Qn({operation:e,ai_service:n,...o?{model:o}:{}});s.latencyHistogram.record(t,r)}}catch(r){console.warn("Failed to record latency metric:",r)}},yr=(s,e,t,n,o,r,i)=>{let a={operation:e,ai_service:r,...i?{model:i}:{}};s.meanLatencyGauge&&s.meanLatencyGauge.record(t,a),s.p95LatencyGauge&&s.p95LatencyGauge.record(n,a),s.p99LatencyGauge&&s.p99LatencyGauge.record(o,a)},Ir=(s,e,t,n)=>{try{if(s.errorCounter){let o=Qn({operation:e,ai_service:t,...n?{model:n}:{}});s.errorCounter.add(1,o)}}catch(o){console.warn("Failed to record error metric:",o)}},br=(s,e,t,n,o)=>{s.errorRateGauge&&s.errorRateGauge.record(t*100,{operation:e,ai_service:n,...o?{model:o}:{}})},Rr=(s,e,t,n)=>{s.requestCounter&&s.requestCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},Xe=(s,e,t,n,o)=>{try{let r=Qn({ai_service:n,...o?{model:o}:{}});s.tokenCounter&&s.tokenCounter.add(t,{token_type:e,...r}),e==="input"&&s.inputTokenCounter&&s.inputTokenCounter.add(t,r),e==="output"&&s.outputTokenCounter&&s.outputTokenCounter.add(t,r)}catch(r){console.warn("Failed to record token metric:",r)}},Cr=(s,e,t,n,o)=>{t&&s.streamingRequestsCounter&&s.streamingRequestsCounter.add(1,{operation:e,ai_service:n,...o?{model:o}:{}})},Tr=(s,e,t,n,o)=>{let r={function_name:e,...n?{ai_service:n}:{},...o?{model:o}:{}};s.functionCallsCounter&&s.functionCallsCounter.add(1,r),t&&s.functionCallLatencyHistogram&&s.functionCallLatencyHistogram.record(t,r)},Yn=(s,e,t,n,o)=>{s.requestSizeHistogram&&s.requestSizeHistogram.record(t,{operation:e,ai_service:n,...o?{model:o}:{}})},Xn=(s,e,t,n,o)=>{s.responseSizeHistogram&&s.responseSizeHistogram.record(t,{operation:e,ai_service:n,...o?{model:o}:{}})},wr=(s,e,t,n,o)=>{let r={...n?{ai_service:n}:{},...o?{model:o}:{}};e!==void 0&&s.temperatureGauge&&s.temperatureGauge.record(e,r),t!==void 0&&s.maxTokensGauge&&s.maxTokensGauge.record(t,r)},Zn=(s,e,t,n,o)=>{s.estimatedCostCounter&&s.estimatedCostCounter.add(t,{operation:e,ai_service:n,...o?{model:o}:{}})},Sr=(s,e,t,n)=>{s.promptLengthHistogram&&s.promptLengthHistogram.record(e,{ai_service:t,...n?{model:n}:{}})},vr=(s,e,t,n)=>{s.contextWindowUsageGauge&&s.contextWindowUsageGauge.record(e,{ai_service:t,...n?{model:n}:{}})},Or=(s,e,t,n)=>{s.timeoutsCounter&&s.timeoutsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},Mr=(s,e,t,n)=>{s.abortsCounter&&s.abortsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},kr=(s,e,t,n)=>{s.thinkingBudgetUsageCounter&&s.thinkingBudgetUsageCounter.add(e,{ai_service:t,...n?{model:n}:{}})},Er=(s,e,t,n,o)=>{(e||t)&&s.multimodalRequestsCounter&&s.multimodalRequestsCounter.add(1,{ai_service:n,has_images:e.toString(),has_audio:t.toString(),...o?{model:o}:{}})};var w=()=>structuredClone({temperature:0,topK:40,topP:.9}),_=()=>structuredClone({temperature:.4,topP:.7,frequencyPenalty:.2}),E=class{constructor(e,{name:t,apiURL:n,headers:o,modelInfo:r,defaults:i,options:a={},supportFor:l,models:p}){this.aiImpl=e;this.name=t,this.apiURL=n||"",this.headers=o,this.supportFor=l,this.tracer=a.tracer??M.tracer,this.meter=a.meter??M.meter,this.modelInfo=r,this.models=p,this.id=$();let c=this.getModel(i.model)??i.model,u=this.getEmbedModel(i.embedModel)??i.embedModel;if(this.defaults={model:c,embedModel:u},!i.model||typeof i.model!="string"||i.model==="")throw new Error("No model defined");this.setOptions(a),p&&bi(p)}debug=!1;rt;fetch;tracer;meter;timeout;excludeContentFromTrace;models;abortSignal;logger=M.logger??or;corsProxy;modelInfo;modelUsage;embedModelUsage;defaults;lastUsedModelConfig;lastUsedChatModel;lastUsedEmbedModel;apiURL;name;id;headers;supportFor;metrics={latency:{chat:{mean:0,p95:0,p99:0,samples:[]},embed:{mean:0,p95:0,p99:0,samples:[]}},errors:{chat:{count:0,rate:0,total:0},embed:{count:0,rate:0,total:0}}};getMetricsInstruments(){return xr(this.meter)}setName(e){this.name=e}getId(){return this.id}setAPIURL(e){this.apiURL=e}setHeaders(e){this.headers=e}setOptions(e){this.debug=e.debug??!1,this.rt=e.rateLimiter,this.fetch=e.fetch,this.timeout=e.timeout,this.tracer=e.tracer??M.tracer,this.meter=e.meter??M.meter,this.excludeContentFromTrace=e.excludeContentFromTrace,this.abortSignal=e.abortSignal,this.logger=e.logger??M.logger??this.logger,this.corsProxy=e.corsProxy}getOptions(){return{debug:this.debug,rateLimiter:this.rt,fetch:this.fetch,tracer:this.tracer,meter:this.meter,timeout:this.timeout,excludeContentFromTrace:this.excludeContentFromTrace,abortSignal:this.abortSignal,logger:this.logger,corsProxy:this.corsProxy}}getLogger(){return this.logger}getModelList(){let e=[];for(let t of this.models??[])t.isInternal||("model"in t&&t.model&&e.push({key:t.key,description:t.description,model:t.model}),"embedModel"in t&&t.embedModel&&e.push({key:t.key,description:t.description,embedModel:t.embedModel}));return e}getName(){return this.name}getFeatures(e){return typeof this.supportFor=="function"?this.supportFor(e??this.defaults.model):this.supportFor}getLastUsedChatModel(){return this.lastUsedChatModel}getLastUsedEmbedModel(){return this.lastUsedEmbedModel}getLastUsedModelConfig(){return this.lastUsedModelConfig}calculatePercentile(e,t){if(e.length===0)return 0;let n=[...e].sort((r,i)=>r-i),o=Math.ceil(t/100*n.length)-1;return n[o]??0}updateLatencyMetrics(e,t){let n=this.metrics.latency[e];n.samples.push(t),n.samples.length>1e3&&n.samples.shift(),n.mean=n.samples.reduce((r,i)=>r+i,0)/n.samples.length,n.p95=this.calculatePercentile(n.samples,95),n.p99=this.calculatePercentile(n.samples,99);let o=this.getMetricsInstruments();if(o){let r=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;fr(o,e,t,this.name,r),yr(o,e,n.mean,n.p95,n.p99,this.name,r)}}updateErrorMetrics(e,t){let n=this.metrics.errors[e];n.total++,t&&n.count++,n.rate=n.count/n.total;let o=this.getMetricsInstruments();if(o){let r=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Rr(o,e,this.name,r),t&&Ir(o,e,this.name,r),br(o,e,n.rate,this.name,r)}}recordTokenUsage(e){let t=this.getMetricsInstruments();if(t&&e?.tokens){let{promptTokens:n,completionTokens:o,totalTokens:r,thoughtsTokens:i}=e.tokens;n&&Xe(t,"input",n,this.name,e.model),o&&Xe(t,"output",o,this.name,e.model),r&&Xe(t,"total",r,this.name,e.model),i&&Xe(t,"thoughts",i,this.name,e.model)}}calculateRequestSize(e){try{return new TextEncoder().encode(JSON.stringify(e)).length}catch{return 0}}calculateResponseSize(e){try{return new TextEncoder().encode(JSON.stringify(e)).length}catch{return 0}}detectMultimodalContent(e){let t=!1,n=!1;if(e.chatPrompt&&Array.isArray(e.chatPrompt)){for(let o of e.chatPrompt)if(o.role==="user"&&Array.isArray(o.content))for(let r of o.content)r.type==="image"?t=!0:r.type==="audio"&&(n=!0)}return{hasImages:t,hasAudio:n}}calculatePromptLength(e){let t=0;if(e.chatPrompt&&Array.isArray(e.chatPrompt))for(let n of e.chatPrompt)if(n.role==="system"||n.role==="assistant")n.content&&(t+=n.content.length);else if(n.role==="user"){if(typeof n.content=="string")t+=n.content.length;else if(Array.isArray(n.content))for(let o of n.content)o.type==="text"&&(t+=o.text.length)}else n.role==="function"&&n.result&&(t+=n.result.length);return t}calculateContextWindowUsage(e,t){if(!t?.tokens?.promptTokens)return 0;let n=this.modelInfo.find(o=>o.name===e);return n?.contextWindow?t.tokens.promptTokens/n.contextWindow:0}estimateCost(e,t){if(!t?.tokens)return 0;let n=this.modelInfo.find(l=>l.name===e);if(!n||!n.promptTokenCostPer1M&&!n.completionTokenCostPer1M)return 0;let{promptTokens:o=0,completionTokens:r=0}=t.tokens,i=n.promptTokenCostPer1M||0,a=n.completionTokenCostPer1M||0;return o*i/1e6+r*a/1e6}estimateCostByName(e,t){if(!t?.tokens)return 0;let n=this.modelInfo.find(l=>l.name===e);if(!n||!n.promptTokenCostPer1M&&!n.completionTokenCostPer1M)return 0;let{promptTokens:o=0,completionTokens:r=0}=t.tokens,i=n.promptTokenCostPer1M||0,a=n.completionTokenCostPer1M||0;return o*i/1e6+r*a/1e6}recordFunctionCallMetrics(e,t){let n=this.getMetricsInstruments();if(!(!n||!e))for(let o of e)o&&typeof o=="object"&&"function"in o&&o.function&&typeof o.function=="object"&&"name"in o.function&&Tr(n,o.function.name,void 0,this.name,t)}recordTimeoutMetric(e){let t=this.getMetricsInstruments();if(t){let n=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Or(t,e,this.name,n)}}recordAbortMetric(e){let t=this.getMetricsInstruments();if(t){let n=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Mr(t,e,this.name,n)}}recordChatMetrics(e,t,n){let o=this.getMetricsInstruments();if(!o)return;let r=this.lastUsedChatModel,i=this.lastUsedModelConfig,a=i?.stream??!1;Cr(o,"chat",a,this.name,r);let{hasImages:l,hasAudio:p}=this.detectMultimodalContent(e);Er(o,l,p,this.name,r);let c=this.calculatePromptLength(e);Sr(o,c,this.name,r),wr(o,i?.temperature,i?.maxTokens,this.name,r),t?.thinkingTokenBudget&&this.modelUsage?.tokens?.thoughtsTokens&&kr(o,this.modelUsage.tokens.thoughtsTokens,this.name,r);let u=this.calculateRequestSize(e);if(Yn(o,"chat",u,this.name,r),n&&!a){let d=n,m=this.calculateResponseSize(d);if(Xn(o,"chat",m,this.name,r),d.results)for(let f of d.results)f.functionCalls&&this.recordFunctionCallMetrics(f.functionCalls,this.lastUsedChatModel);let h=this.calculateContextWindowUsage(this.lastUsedChatModel,d.modelUsage);h>0&&vr(o,h,this.name,r);let g=this.estimateCost(this.lastUsedChatModel,d.modelUsage);g>0&&Zn(o,"chat",g,this.name,r)}}recordEmbedMetrics(e,t){let n=this.getMetricsInstruments();if(!n)return;let o=this.lastUsedEmbedModel,r=this.calculateRequestSize(e);Yn(n,"embed",r,this.name,o);let i=this.calculateResponseSize(t);Xn(n,"embed",i,this.name,o);let a=this.estimateCostByName(o,t.modelUsage);a>0&&Zn(n,"embed",a,this.name,o)}getMetrics(){return structuredClone(this.metrics)}async chat(e,t){let n=performance.now(),o=!1,r;try{return r=await this._chat1(e,t),r}catch(i){throw o=!0,i instanceof Error&&(i.message.includes("timeout")||i.name==="TimeoutError"?this.recordTimeoutMetric("chat"):(i.message.includes("abort")||i.name==="AbortError")&&this.recordAbortMetric("chat")),i}finally{let i=performance.now()-n;this.updateLatencyMetrics("chat",i),this.updateErrorMetrics("chat",o),o||this.recordChatMetrics(e,t,r)}}async _chat1(e,t){let n=this.getModel(e.model)??e.model??this.defaults.model;e.chatPrompt&&Array.isArray(e.chatPrompt)&&eo(e.chatPrompt);let o={...this.aiImpl.getModelConfig(),...e.modelConfig};if(t?.thinkingTokenBudget&&!this.getFeatures(n).hasThinkingBudget)throw new Error(`Model ${n} does not support thinkingTokenBudget.`);if(t?.showThoughts&&!this.getFeatures(n).hasShowThoughts)throw new Error(`Model ${n} does not support showThoughts.`);if(this.modelInfo.find(a=>a.name===n)?.isExpensive&&t?.useExpensiveModel!=="yes")throw new Error(`Model ${n} is marked as expensive and requires explicit confirmation. Set useExpensiveModel: "yes" to proceed.`);return o.stream=(t?.stream!==void 0?t.stream:o.stream)??!0,this.getFeatures(n).streaming||(o.stream=!1),this.tracer?await this.tracer.startActiveSpan("AI Chat Request",{kind:he.SpanKind.SERVER,attributes:{[T.LLM_SYSTEM]:this.name,[T.LLM_OPERATION_NAME]:"chat",[T.LLM_REQUEST_MODEL]:n,[T.LLM_REQUEST_MAX_TOKENS]:o.maxTokens??"Not set",[T.LLM_REQUEST_TEMPERATURE]:o.temperature,[T.LLM_REQUEST_TOP_P]:o.topP??"Not set",[T.LLM_REQUEST_TOP_K]:o.topK??"Not set",[T.LLM_REQUEST_FREQUENCY_PENALTY]:o.frequencyPenalty??"Not set",[T.LLM_REQUEST_PRESENCE_PENALTY]:o.presencePenalty??"Not set",[T.LLM_REQUEST_STOP_SEQUENCES]:o.stopSequences?.join(", ")??"Not set",[T.LLM_REQUEST_LLM_IS_STREAMING]:o.stream??"Not set"}},t?.traceContext??he.context.active(),async a=>await this._chat2(n,o,e,t,a)):await this._chat2(n,o,e,t)}cleanupFunctionSchema(e){let t={...e};if(t.parameters){let n={...t.parameters};Array.isArray(n.required)&&n.required.length===0&&delete n.required,n.properties&&Object.keys(n.properties).length===0&&delete n.properties,Object.keys(n).length===0||Object.keys(n).length===1&&n.type==="object"?delete t.parameters:t.parameters=n}return t}async _chat2(e,t,n,o,r){if(!this.aiImpl.createChatReq)throw new Error("generateChatReq not implemented");let i=o?.debug??this.debug,a;n.functions&&n.functions.length>0&&(a=n.functions.map(m=>this.cleanupFunctionSchema(m)));let l={...n,model:e,functions:a,modelConfig:t};this.lastUsedChatModel=e,this.lastUsedModelConfig=t;let p=async()=>{let[m,h]=await this.aiImpl.createChatReq(l,o);return r?.isRecording()&&Ii(n,r,this.excludeContentFromTrace),await B({name:m.name,url:this.apiURL,localCall:m.localCall,headers:await this.buildHeaders(m.headers),stream:t.stream,timeout:this.timeout,debug:i,fetch:this.fetch,span:r,abortSignal:o?.abortSignal??this.abortSignal,corsProxy:this.corsProxy},h)};i&&sr(l.chatPrompt,o?.stepIndex??0,o?.logger??this.logger,o?.debugHideSystemPrompt);let c=o?.rateLimiter??this.rt,u=c?await c(p,{modelUsage:this.modelUsage}):await p();if(t.stream){if(!this.aiImpl.createChatStreamResp)throw new Error("generateChatStreamResp not implemented");let m=this.aiImpl.createChatStreamResp.bind(this),h=x=>y=>{let S=m(y,x);if(S.sessionId=o?.sessionId,!S.modelUsage){let I=this.aiImpl.getTokenUsage();I&&(S.modelUsage={ai:this.name,model:e,tokens:I})}if(this.modelUsage=S.modelUsage,this.recordTokenUsage(S.modelUsage),r?.isRecording()&&Pr(S,r,this.excludeContentFromTrace),i)for(let I of S.results)ar(I,I.index,o?.logger??this.logger);return S},g=async x=>{r?.isRecording()&&r.end(),i&&lr(x,o?.logger??this.logger)};if(typeof window<"u"){let x=u,y={},S=[];return new ReadableStream({start(I){let R=x.getReader();async function k(){try{for(;;){let{done:v,value:H}=await R.read();if(v){g&&await g(S),I.close();break}let Y=h(y)(H);Y&&(S.push(Y),I.enqueue(Y))}}catch(v){I.error(v)}finally{R.releaseLock()}}k()}})}return u.pipeThrough(new Pt(h({}),g))}if(!this.aiImpl.createChatResp)throw new Error("generateChatResp not implemented");let d=this.aiImpl.createChatResp(u);if(d.sessionId=o?.sessionId,!d.modelUsage){let m=this.aiImpl.getTokenUsage();m&&(d.modelUsage={ai:this.name,model:e,tokens:m})}return d.modelUsage&&(this.modelUsage=d.modelUsage,this.recordTokenUsage(d.modelUsage)),r?.isRecording()&&(Pr(d,r,this.excludeContentFromTrace),r.end()),i&&ir(d,o?.logger??this.logger),d}async embed(e,t){let n=performance.now(),o=!1,r;try{return r=await this._embed1(e,t),r}catch(i){throw o=!0,i instanceof Error&&(i.message.includes("timeout")||i.name==="TimeoutError"?this.recordTimeoutMetric("embed"):(i.message.includes("abort")||i.name==="AbortError")&&this.recordAbortMetric("embed")),i}finally{let i=performance.now()-n;this.updateLatencyMetrics("embed",i),this.updateErrorMetrics("embed",o),!o&&r&&this.recordEmbedMetrics(e,r)}}async _embed1(e,t){let n=this.getEmbedModel(e.embedModel)??e.embedModel??this.defaults.embedModel;if(!n)throw new Error("No embed model defined");return this.tracer&&await this.tracer?.startActiveSpan("AI Embed Request",{kind:he.SpanKind.SERVER,attributes:{[T.LLM_SYSTEM]:this.name,[T.LLM_OPERATION_NAME]:"embeddings",[T.LLM_REQUEST_MODEL]:n}},t?.traceContext??he.context.active(),async o=>{try{return await this._embed2(n,e,t,o)}finally{o.end()}}),this._embed2(n,e,t)}async _embed2(e,t,n,o){if(!this.aiImpl.createEmbedReq)throw new Error("generateEmbedReq not implemented");if(!this.aiImpl.createEmbedResp)throw new Error("generateEmbedResp not implemented");let r=this.aiImpl.createEmbedReq,i=n?.debug??this.debug,a={...t,embedModel:e};this.lastUsedEmbedModel=e,i&&gr(a.texts??[],e,n?.logger??this.logger);let l=async()=>{let[u,d]=await r(a);return await B({name:u.name,url:this.apiURL,localCall:u.localCall,headers:await this.buildHeaders(u.headers),debug:i,fetch:this.fetch,timeout:this.timeout,span:o,abortSignal:n?.abortSignal??this.abortSignal,corsProxy:this.corsProxy},d)},p=this.rt?await this.rt(l,{modelUsage:this.embedModelUsage}):await l(),c=this.aiImpl.createEmbedResp?.(p);if(c.sessionId=n?.sessionId,!c.modelUsage){let u=this.aiImpl.getTokenUsage();u&&(c.modelUsage={ai:this.name,model:e,tokens:u})}return this.embedModelUsage=c.modelUsage,this.recordTokenUsage(c.modelUsage),o?.isRecording()&&c.modelUsage?.tokens&&o.addEvent(W.GEN_AI_USAGE,{[T.LLM_USAGE_INPUT_TOKENS]:c.modelUsage.tokens.promptTokens,[T.LLM_USAGE_OUTPUT_TOKENS]:c.modelUsage.tokens.completionTokens??0,[T.LLM_USAGE_TOTAL_TOKENS]:c.modelUsage.tokens.totalTokens}),i&&hr(c.embeddings,n?.logger??this.logger),o?.end(),c}async buildHeaders(e={}){return{...e,...await this.headers()}}getModelByKey(e){return e?this.models?.find(n=>n.key===e):void 0}getModel(e){let t=this.getModelByKey(e);return t&&"model"in t?t.model:void 0}getEmbedModel(e){let t=this.getModelByKey(e);return t&&"embedModel"in t?t.embedModel:void 0}};function Ii(s,e,t){let n=[];if(s.chatPrompt&&Array.isArray(s.chatPrompt)&&s.chatPrompt.length>0)for(let r of s.chatPrompt)switch(r.role){case"system":if(r.content){let i={};t||(i.content=r.content),e.addEvent(W.GEN_AI_SYSTEM_MESSAGE,i)}break;case"user":if(typeof r.content=="string")n.push(r.content);else if(Array.isArray(r.content))for(let i of r.content)i.type==="text"&&n.push(i.text);break;case"assistant":{let i=r.functionCalls?.map(a=>({id:a.id,type:a.type,function:a.function.name,arguments:a.function.params}));if(i&&i.length>0){let a={function_calls:JSON.stringify(i,null,2)};!t&&r.content&&(a.content=r.content),e.addEvent(W.GEN_AI_ASSISTANT_MESSAGE,a)}else if(r.content){let a={};t||(a.content=r.content),e.addEvent(W.GEN_AI_ASSISTANT_MESSAGE,a)}break}case"function":{let i={id:r.functionId};t||(i.content=r.result),e.addEvent(W.GEN_AI_TOOL_MESSAGE,i);break}}let o={};t||(o.content=n.join(`
110
- `)),e.addEvent(W.GEN_AI_USER_MESSAGE,o)}function Pr(s,e,t){if(s.modelUsage?.tokens){let n=s.modelUsage.tokens.thoughtsTokens?{[T.LLM_USAGE_THOUGHTS_TOKENS]:s.modelUsage.tokens.thoughtsTokens}:{};e.addEvent(W.GEN_AI_USAGE,{[T.LLM_USAGE_INPUT_TOKENS]:s.modelUsage.tokens.promptTokens,[T.LLM_USAGE_OUTPUT_TOKENS]:s.modelUsage.tokens.completionTokens??0,[T.LLM_USAGE_TOTAL_TOKENS]:s.modelUsage.tokens.totalTokens,...n})}if(s.results)for(let n=0;n<s.results.length;n++){let o=s.results[n];if(!o||!o.content&&!o.thought&&!o.functionCalls?.length&&!o.finishReason)continue;let r=o.functionCalls?.map(a=>({id:a.id,type:a.type,function:a.function.name,arguments:a.function.params})),i={};r&&r.length>0?(t||(i.content=o.content),i.tool_calls=r):t||(i.content=o.content??""),e.addEvent(W.GEN_AI_CHOICE,{finish_reason:o.finishReason,index:n,message:JSON.stringify(i,null,2)})}}function eo(s){for(let e=0;e<s.length;e++){let t=s[e];if(!t||typeof t!="object")throw new Error(`AxMessage array validation failed: Item at index ${e} is not a valid message object`);if("content"in t&&typeof t.content=="string"&&t.content.trim()==="")throw new Error(`AxMessage array validation failed: Item at index ${e} has empty content`)}}function bi(s){let e=new Set;for(let t of s){if(e.has(t.key))throw new Error(`Duplicate model key detected: "${t.key}". Each model key must be unique.`);e.add(t.key)}}var Ze=(c=>(c.Claude4Opus="claude-opus-4-20250514",c.Claude4Sonnet="claude-sonnet-4-20250514",c.Claude37Sonnet="claude-3-7-sonnet-latest",c.Claude35Sonnet="claude-3-5-sonnet-latest",c.Claude35Haiku="claude-3-5-haiku-latest",c.Claude3Opus="claude-3-opus-latest",c.Claude3Sonnet="claude-3-sonnet-20240229",c.Claude3Haiku="claude-3-haiku-20240307",c.Claude21="claude-2.1",c.ClaudeInstant12="claude-instant-1.2",c))(Ze||{}),Dt=(i=>(i.Claude37Sonnet="claude-3-7-sonnet",i.Claude35Haiku="claude-3-5-haiku",i.Claude35Sonnet="claude-3-5-sonnet",i.Claude35SonnetV2="claude-3-5-sonnet-v2",i.Claude3Haiku="claude-3-haiku",i.Claude3Opus="claude-3-opus",i))(Dt||{});var et=[{name:"claude-opus-4-20250514",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:75,maxTokens:32e3,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"claude-sonnet-4-20250514",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:64e3,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"claude-3-7-sonnet-latest",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:64e3,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"claude-3-5-sonnet-latest",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:8192},{name:"claude-3-5-haiku-latest",currency:"usd",promptTokenCostPer1M:.8,completionTokenCostPer1M:4,maxTokens:8192},{name:"claude-3-opus-latest",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:75,maxTokens:4096},{name:"claude-3-sonnet-20240229",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:4096},{name:"claude-3-haiku-20240307",currency:"usd",promptTokenCostPer1M:.25,completionTokenCostPer1M:1.25,maxTokens:4096},{name:"claude-2.1",currency:"usd",promptTokenCostPer1M:8,completionTokenCostPer1M:25,maxTokens:4096},{name:"claude-instant-1.2",currency:"usd",promptTokenCostPer1M:.8,completionTokenCostPer1M:2.24,maxTokens:4096}];var to=s=>{if(!s||typeof s!="object")return s;let e={...s};return delete e.additionalProperties,delete e.default,delete e.optional,delete e.oneOf,delete e.anyOf,delete e.allOf,e.properties&&typeof e.properties=="object"&&(e.properties=Object.fromEntries(Object.entries(e.properties).map(([t,n])=>[t,to(n)]))),e.items&&(e.items=to(e.items)),e},oo=()=>structuredClone({model:"claude-3-7-sonnet-latest",maxTokens:4e4,thinkingTokenBudgetLevels:{minimal:1024,low:5e3,medium:1e4,high:2e4,highest:32e3},...w()}),_r=()=>structuredClone({model:"claude-3-7-sonnet",maxTokens:4e4,thinkingTokenBudgetLevels:{minimal:1024,low:5e3,medium:1e4,high:2e4,highest:32e3},...w()}),no=class{constructor(e,t){this.config=e;this.isVertex=t}tokensUsed;currentPromptConfig;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens??4096,temperature:e.temperature,topP:e.topP,topK:e.topK,stream:e.stream,stopSequences:e.stopSequences,endSequences:e.endSequences,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,n:e.n}}createChatReq=async(e,t)=>{this.currentPromptConfig=t;let n=e.model,o=e.modelConfig?.stream??this.config.stream,r;this.isVertex?r={name:o?`/models/${n}:streamRawPredict?alt=sse`:`/models/${n}:rawPredict`}:r={name:"/messages"};let i;if(e.functionCall&&e.functions&&e.functions.length>0)if(typeof e.functionCall=="string")switch(e.functionCall){case"auto":i={tool_choice:{type:"auto"}};break;case"required":i={tool_choice:{type:"any"}};break;case"none":throw new Error("functionCall none not supported")}else if("function"in e.functionCall)i={tool_choice:{type:"tool",name:e.functionCall.function.name}};else throw new Error("Invalid function call type, must be string or object");let a=e.chatPrompt.filter(y=>y.role==="system").map(y=>({type:"text",text:y.content,...y.cache?{cache:{type:"ephemeral"}}:{}})),l=e.chatPrompt.filter(y=>y.role!=="system"),p=Ri(l),c=e.functions?.map(y=>({name:y.name,description:y.description,input_schema:y.parameters?to(y.parameters):void 0})),u=e.modelConfig?.maxTokens??this.config.maxTokens,d=e.modelConfig?.stopSequences??this.config.stopSequences,m=e.modelConfig?.temperature??this.config.temperature,h=e.modelConfig?.topP??this.config.topP,g=e.modelConfig?.topK??this.config.topK,f=e.modelConfig?.n??this.config.n;if(f&&f>1)throw new Error("Anthropic does not support sampling (n > 1)");let A;if(this.config.thinking?.budget_tokens&&(A=this.config.thinking),t?.thinkingTokenBudget){let y=this.config.thinkingTokenBudgetLevels;switch(t.thinkingTokenBudget){case"none":A=void 0;break;case"minimal":A={type:"enabled",budget_tokens:y?.minimal??1024};break;case"low":A={type:"enabled",budget_tokens:y?.low??5e3};break;case"medium":A={type:"enabled",budget_tokens:y?.medium??1e4};break;case"high":A={type:"enabled",budget_tokens:y?.high??2e4};break;case"highest":A={type:"enabled",budget_tokens:y?.highest??32e3};break}}let x={...this.isVertex?{anthropic_version:"vertex-2023-10-16"}:{model:n},...u?{max_tokens:u}:{},...d&&d.length>0?{stop_sequences:d}:{},...m&&!A?{temperature:m}:{},...h&&(!A||h>=.95)?{top_p:h}:{},...g&&!A?{top_k:g}:{},...i,...c&&c.length>0?{tools:c}:{},...o?{stream:!0}:{},...a?{system:a}:{},...A?{thinking:A}:{},messages:p};return[r,x]};createChatResp=e=>{if(e.type==="error")throw new O(e.error.message,void 0,void 0);let t=Fr(e.stop_reason),n=this.currentPromptConfig?.thinkingTokenBudget!=="none"&&this.currentPromptConfig?.showThoughts!==!1,o=e.content.map((r,i)=>r.type==="tool_use"?{index:i,id:r.id,functionCalls:[{id:r.id,type:"function",function:{name:r.name,params:r.input}}],finishReason:t}:(r.type==="thinking"||r.type==="redacted_thinking")&&n?{index:i,thought:r.thinking,id:e.id,finishReason:t}:{index:i,content:r.type==="text"?r.text:"",id:e.id,finishReason:t}).filter(r=>r.content!==""||r.thought!==void 0||r.functionCalls!==void 0);return this.tokensUsed={promptTokens:e.usage.input_tokens,completionTokens:e.usage.output_tokens,totalTokens:e.usage.input_tokens+e.usage.output_tokens},{results:o,remoteId:e.id}};createChatStreamResp=(e,t)=>{if(!("type"in e))throw new Error("Invalid Anthropic streaming event");let n=t;if(n.indexIdMap||(n.indexIdMap={}),e.type==="error"){let{error:r}=e;throw new O(r.message,void 0,void 0)}let o=0;if(e.type==="message_start"){let{message:r}=e,i=[{index:o,content:"",id:r.id}];return this.tokensUsed={promptTokens:r.usage?.input_tokens??0,completionTokens:r.usage?.output_tokens??0,totalTokens:(r.usage?.input_tokens??0)+(r.usage?.output_tokens??0)},{results:i}}if(e.type==="content_block_start"){let{content_block:r}=e;if(r.type==="text")return{results:[{index:o,content:r.text}]};if(r.type==="thinking")return this.currentPromptConfig?.thinkingTokenBudget!=="none"&&this.currentPromptConfig?.showThoughts!==!1?{results:[{index:o,thought:r.thinking}]}:{results:[{index:o,content:""}]};if(r.type==="tool_use"&&typeof r.id=="string"&&typeof e.index=="number"&&!n.indexIdMap[e.index]){n.indexIdMap[e.index]=r.id;let i=[{id:r.id,type:"function",function:{name:r.name,params:""}}];return{results:[{index:o,functionCalls:i}]}}}if(e.type==="content_block_delta"){let{delta:r}=e;if(r.type==="text_delta")return{results:[{index:o,content:r.text}]};if(r.type==="thinking_delta")return this.currentPromptConfig?.thinkingTokenBudget!=="none"&&this.currentPromptConfig?.showThoughts!==!1?{results:[{index:o,thought:r.thinking}]}:{results:[{index:o,content:""}]};if(r.type==="signature_delta")return{results:[{index:o,content:""}]};if(r.type==="input_json_delta"){let i=n.indexIdMap[e.index];if(!i)throw new Error(`invalid streaming index no id found: ${e.index}`);let a=[{id:i,type:"function",function:{name:"",params:r.partial_json}}];return{results:[{index:o,functionCalls:a}]}}}if(e.type==="message_delta"){let{delta:r,usage:i}=e;return this.tokensUsed={promptTokens:0,completionTokens:i.output_tokens,totalTokens:i.output_tokens},{results:[{index:o,content:"",finishReason:Fr(r.stop_reason)}]}}return{results:[{index:o,content:""}]}}},Ae=class s extends E{static create(e){return new s(e)}constructor({apiKey:e,projectId:t,region:n,config:o,options:r,models:i}){let a=t!==void 0&&n!==void 0,l,p;if(a){if(!e)throw new Error("Anthropic Vertex API key not set");if(typeof e!="function")throw new Error("Anthropic Vertex API key must be a function for token-based authentication");l=`https://${n}-aiplatform.googleapis.com/v1/projects/${t}/locations/${n}/publishers/anthropic/`,p=async()=>({Authorization:`Bearer ${await e()}`})}else{if(!e)throw new Error("Anthropic API key not set");l="https://api.anthropic.com/v1",p=async()=>({"anthropic-version":"2023-06-01","anthropic-beta":"prompt-caching-2024-07-31","x-api-key":typeof e=="function"?await e():e})}let c={...oo(),...o},u=new no(c,a),d=m=>{let h=q({model:m,modelInfo:et,models:i});return{functions:!0,streaming:!0,hasThinkingBudget:h?.hasThinkingBudget??!1,hasShowThoughts:h?.hasShowThoughts??!1,functionCot:!0,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:5*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!1,formats:[],maxDuration:0},files:{supported:!1,formats:[],maxSize:0,uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!0,types:["ephemeral"]},thinking:h?.hasThinkingBudget??!1,multiTurn:!0}};super(u,{name:"Anthropic",apiURL:l,headers:p,modelInfo:et,defaults:{model:c.model},options:r,supportFor:d,models:i})}};function Ri(s){let e=s.map(t=>{switch(t.role){case"function":return{role:"user",content:[{type:"tool_result",content:t.result,tool_use_id:t.functionId,...t.isError?{is_error:!0}:{},...t.cache?{cache:{type:"ephemeral"}}:{}}]};case"user":return typeof t.content=="string"?{role:"user",content:t.content}:{role:"user",content:t.content.map(o=>{switch(o.type){case"text":return{type:"text",text:o.text,...o.cache?{cache:{type:"ephemeral"}}:{}};case"image":return{type:"image",source:{type:"base64",media_type:o.mimeType,data:o.image},...o.cache?{cache:{type:"ephemeral"}}:{}};default:throw new Error("Invalid content type")}})};case"assistant":{let n="";return typeof t.content=="string"&&(n=t.content),typeof t.functionCalls<"u"&&(n=t.functionCalls.map(o=>{let r={};return typeof o.function.params=="string"?r=JSON.parse(o.function.params):typeof o.function.params=="object"&&(r=o.function.params),{type:"tool_use",id:o.id,name:o.function.name,input:r,...t.cache?{cache:{type:"ephemeral"}}:{}}})),{role:"assistant",content:n}}default:throw new Error("Invalid role")}});return Ci(e)}function Ci(s){let e=[];for(let[t,n]of s.entries()){if(n.role!=="assistant"){e.push(n);continue}if(t>0&&s.at(t-1)?.role==="assistant"){let o=e.pop();e.push({...o||{},...n})}else e.push(n)}return e}function Fr(s){if(s)switch(s){case"stop_sequence":return"stop";case"max_tokens":return"length";case"tool_use":return"function_call";case"end_turn":return"stop";default:return"stop"}}var tt=(A=>(A.GPT4="gpt-4",A.GPT41="gpt-4.1",A.GPT41Mini="gpt-4.1-mini",A.GPT4O="gpt-4o",A.GPT4OMini="gpt-4o-mini",A.GPT4ChatGPT4O="chatgpt-4o-latest",A.GPT4Turbo="gpt-4-turbo",A.GPT35Turbo="gpt-3.5-turbo",A.GPT35TurboInstruct="gpt-3.5-turbo-instruct",A.GPT35TextDavinci002="text-davinci-002",A.GPT3TextBabbage002="text-babbage-002",A.GPT3TextAda001="text-ada-001",A.O1="o1",A.O1Mini="o1-mini",A.O3="o3",A.O3Mini="o3-mini",A.O4Mini="o4-mini",A))(tt||{}),xe=(n=>(n.TextEmbeddingAda002="text-embedding-ada-002",n.TextEmbedding3Small="text-embedding-3-small",n.TextEmbedding3Large="text-embedding-3-large",n))(xe||{});var fe=(y=>(y.GPT4="gpt-4",y.GPT41="gpt-4.1",y.GPT41Mini="gpt-4.1-mini",y.GPT4O="gpt-4o",y.GPT4OMini="gpt-4o-mini",y.GPT4ChatGPT4O="chatgpt-4o-latest",y.GPT4Turbo="gpt-4-turbo",y.GPT35Turbo="gpt-3.5-turbo",y.GPT35TurboInstruct="gpt-3.5-turbo-instruct",y.GPT35TextDavinci002="text-davinci-002",y.GPT3TextBabbage002="text-babbage-002",y.GPT3TextAda001="text-ada-001",y.O1Pro="o1-pro",y.O1="o1",y.O1Mini="o1-mini",y.O3Pro="o3-pro",y.O3="o3",y.O3Mini="o3-mini",y.O4Mini="o4-mini",y))(fe||{});var ye=[{name:"gpt-4",currency:"usd",promptTokenCostPer1M:30,completionTokenCostPer1M:60},{name:"gpt-4.1",currency:"usd",promptTokenCostPer1M:2,completionTokenCostPer1M:8},{name:"gpt-4.1-mini",currency:"usd",promptTokenCostPer1M:.4,completionTokenCostPer1M:1.6},{name:"gpt-4o",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4o-mini",currency:"usd",promptTokenCostPer1M:.15,completionTokenCostPer1M:.6},{name:"chatgpt-4o-latest",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4-turbo",currency:"usd",promptTokenCostPer1M:10,completionTokenCostPer1M:30},{name:"gpt-3.5-turbo",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"o1",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60},{name:"o1-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:14.4},{name:"o3",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60},{name:"o3-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4},{name:"o4-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4},{name:"text-embedding-ada-002",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"text-embedding-3-small",currency:"usd",promptTokenCostPer1M:.02,completionTokenCostPer1M:.02},{name:"text-embedding-3-large",currency:"usd",promptTokenCostPer1M:.13,completionTokenCostPer1M:.13}],Lt=[{name:"gpt-4",currency:"usd",promptTokenCostPer1M:30,completionTokenCostPer1M:60},{name:"gpt-4.1",currency:"usd",promptTokenCostPer1M:2,completionTokenCostPer1M:8},{name:"gpt-4.1-mini",currency:"usd",promptTokenCostPer1M:.4,completionTokenCostPer1M:1.6},{name:"gpt-4o",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4o-mini",currency:"usd",promptTokenCostPer1M:.15,completionTokenCostPer1M:.6},{name:"chatgpt-4o-latest",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4-turbo",currency:"usd",promptTokenCostPer1M:10,completionTokenCostPer1M:30},{name:"gpt-3.5-turbo",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"o1-pro",currency:"usd",promptTokenCostPer1M:150,completionTokenCostPer1M:600,hasThinkingBudget:!0,hasShowThoughts:!0,isExpensive:!0},{name:"o1",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"o3-pro",currency:"usd",promptTokenCostPer1M:20,completionTokenCostPer1M:80,hasThinkingBudget:!0,hasShowThoughts:!0,isExpensive:!0},{name:"o3",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"o3-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"o4-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4,hasThinkingBudget:!0,hasShowThoughts:!0}];var Ti=s=>{let e=["o1","o1-mini","o3","o3-mini","o4-mini","o1-pro","o3-pro"];return e.includes(s)||e.includes(s)},ce=()=>structuredClone({model:"gpt-4.1",embedModel:"text-embedding-3-small",...w()}),Gt=()=>structuredClone({...ce(),model:"gpt-4.1"}),Nt=()=>structuredClone({model:"gpt-4.1",embedModel:"text-embedding-3-small",..._()}),$t=()=>({...ce(),model:"gpt-4.1-mini"}),ro=class{constructor(e,t,n){this.config=e;this.streamingUsage=t;this.chatReqUpdater=n}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,endSequences:e.endSequences,topP:e.topP,n:e.n,stream:e.stream}}createChatReq(e,t){let n=e.model;if(!e.chatPrompt||e.chatPrompt.length===0)throw new Error("Chat prompt is empty");let o={name:"/chat/completions"},r=e.functions?.map(m=>({type:"function",function:{name:m.name,description:m.description,parameters:m.parameters}})),i=!e.functionCall&&e.functions&&e.functions.length>0?"auto":e.functionCall,a=wi(e),l=e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,p=e.modelConfig?.stream??this.config.stream,c=this.config.store,u=Ti(n),d={model:n,messages:a,response_format:this.config?.responseFormat?{type:this.config.responseFormat}:void 0,tools:r,tool_choice:i,...u?{}:{max_completion_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,top_p:e.modelConfig?.topP??this.config.topP??1,n:e.modelConfig?.n??this.config.n,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,...l?{frequency_penalty:l}:{}},stop:e.modelConfig?.stopSequences??this.config.stop,logit_bias:this.config.logitBias,...p&&this.streamingUsage?{stream:!0,stream_options:{include_usage:!0}}:{},...c?{store:c}:{},...this.config.serviceTier?{service_tier:this.config.serviceTier}:{},...this.config.user?{user:this.config.user}:{}};if(this.config.reasoningEffort&&(d.reasoning_effort=this.config.reasoningEffort),this.config.webSearchOptions&&(d.web_search_options={...this.config.webSearchOptions.searchContextSize&&{search_context_size:this.config.webSearchOptions.searchContextSize},...this.config.webSearchOptions.userLocation&&{user_location:{approximate:{type:"approximate",...this.config.webSearchOptions.userLocation.approximate.city&&{city:this.config.webSearchOptions.userLocation.approximate.city},...this.config.webSearchOptions.userLocation.approximate.country&&{country:this.config.webSearchOptions.userLocation.approximate.country},...this.config.webSearchOptions.userLocation.approximate.region&&{region:this.config.webSearchOptions.userLocation.approximate.region},...this.config.webSearchOptions.userLocation.approximate.timezone&&{timezone:this.config.webSearchOptions.userLocation.approximate.timezone}}}}}),t?.thinkingTokenBudget)switch(t.thinkingTokenBudget){case"none":d.reasoning_effort=void 0;break;case"minimal":d.reasoning_effort="low";break;case"low":d.reasoning_effort="medium";break;case"medium":d.reasoning_effort="high";break;case"high":d.reasoning_effort="high";break;case"highest":d.reasoning_effort="high";break}return this.chatReqUpdater&&(d=this.chatReqUpdater(d)),[o,d]}createEmbedReq(e){let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n={name:"/embeddings"},o={model:t,input:e.texts,dimensions:this.config.dimensions};return[n,o]}createChatResp(e){let{id:t,usage:n,choices:o,error:r}=e;if(r)throw r;return this.tokensUsed=n?{promptTokens:n.prompt_tokens,completionTokens:n.completion_tokens,totalTokens:n.total_tokens}:void 0,{results:o.map(a=>{if(a.message.refusal)throw new O(a.message.refusal,e.model,e.id);let l=Dr(a.finish_reason),p=a.message.tool_calls?.map(({id:c,function:{arguments:u,name:d}})=>({id:c,type:"function",function:{name:d,params:u}}));return{index:a.index,id:`${a.index}`,content:a.message.content??void 0,thought:a.message.reasoning_content,annotations:a.message.annotations,functionCalls:p,finishReason:l}}),remoteId:t}}createChatStreamResp(e,t){let{id:n,usage:o,choices:r}=e;this.tokensUsed=o?{promptTokens:o.prompt_tokens,completionTokens:o.completion_tokens,totalTokens:o.total_tokens}:void 0;let i=t;return i.indexIdMap||(i.indexIdMap={}),{results:r.map(({index:l,delta:{content:p,role:c,refusal:u,tool_calls:d,reasoning_content:m,annotations:h},finish_reason:g})=>{if(u)throw new O(u,void 0,n);let f=Dr(g),A=d?.map(({id:x,index:y,function:{name:S,arguments:I}})=>{typeof x=="string"&&typeof y=="number"&&!i.indexIdMap[y]&&(i.indexIdMap[y]=x);let R=i.indexIdMap[y];return R?{id:R,type:"function",function:{name:S,params:I}}:null}).filter(x=>x!==null);return{index:l,content:p??void 0,role:c,thought:m,annotations:h,functionCalls:A,finishReason:f,id:n}})}}createEmbedResp(e){let{data:t,usage:n}=e;return this.tokensUsed=n?{promptTokens:n.prompt_tokens,completionTokens:n.completion_tokens,totalTokens:n.total_tokens}:void 0,{embeddings:t.map(o=>o.embedding)}}},Dr=s=>{switch(s){case"stop":return"stop";case"length":return"length";case"content_filter":return"error";case"tool_calls":return"function_call"}};function wi(s){return s.chatPrompt.map(t=>{switch(t.role){case"system":return{role:"system",content:t.content};case"user":{let n=Array.isArray(t.content)?t.content.map(o=>{switch(o.type){case"text":return{type:"text",text:o.text};case"image":return{type:"image_url",image_url:{url:`data:${o.mimeType};base64,${o.image}`,details:o.details??"auto"}};case"audio":return{type:"input_audio",input_audio:{data:o.data,format:o.format==="wav"?"wav":void 0}};default:throw new Error("Invalid content type")}}):t.content;return{role:"user",...t.name?{name:t.name}:{},content:n}}case"assistant":{let n=t.functionCalls?.map(o=>({id:o.id,type:"function",function:{name:o.function.name,arguments:typeof o.function.params=="object"?JSON.stringify(o.function.params):o.function.params}}));if(n&&n.length>0)return{role:"assistant",...t.content?{content:t.content}:{},name:t.name,tool_calls:n};if(t.content===void 0)throw new Error("Assistant content is required when no tool calls are provided");return{role:"assistant",content:t.content,...t.name?{name:t.name}:{}}}case"function":return{role:"tool",content:t.result,tool_call_id:t.functionId};default:throw new Error("Invalid role")}})}var F=class extends E{constructor({apiKey:e,config:t,options:n,apiURL:o,modelInfo:r,models:i,chatReqUpdater:a,supportFor:l}){if(!e||e==="")throw new Error("OpenAI API key not set");let p=new ro(t,n?.streamingUsage??!0,a);super(p,{name:"OpenAI",apiURL:o||"https://api.openai.com/v1",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:r,defaults:{model:t.model,embedModel:t.embedModel},options:n,supportFor:l,models:i})}},Ie=class extends F{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("OpenAI API key not set");r=[...ye,...r??[]];let i=a=>{let l=q({model:a,modelInfo:r,models:o});return{functions:!0,streaming:!0,hasThinkingBudget:l?.hasThinkingBudget??!1,hasShowThoughts:l?.hasShowThoughts??!1,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:20*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!0,formats:["wav","mp3","ogg"],maxDuration:25*60},files:{supported:!0,formats:["text/plain","application/pdf","image/jpeg","image/png"],maxSize:512*1024*1024,uploadMethod:"upload"},urls:{supported:!1,webSearch:!0,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:l?.hasThinkingBudget??!1,multiTurn:!0}};super({apiKey:e,config:{...ce(),...t},options:n,modelInfo:r,models:o,supportFor:i}),super.setName("OpenAI")}};var so=ce,Lr=Nt,Gr=$t,Nr=Gt,be=class extends F{constructor({apiKey:e,resourceName:t,deploymentName:n,version:o="api-version=2024-02-15-preview",config:r,options:i,models:a,modelInfo:l}){if(!e||e==="")throw new Error("Azure OpenAPI API key not set");if(!t||t==="")throw new Error("Azure OpenAPI resource name not set");if(!n||n==="")throw new Error("Azure OpenAPI deployment id not set");let p={...so(),...r};l=[...ye,...l??[]];let c=d=>{let m=q({model:d,modelInfo:l,models:a});return{functions:!0,streaming:!0,hasThinkingBudget:m?.hasThinkingBudget??!1,hasShowThoughts:m?.hasShowThoughts??!1,functionCot:!1,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:20*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!1,formats:[],maxDuration:0},files:{supported:!1,formats:[],maxSize:0,uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:m?.hasThinkingBudget??!1,multiTurn:!0}};super({apiKey:e,config:p,options:i,models:a,modelInfo:l,supportFor:c});let u=t.includes("://")?t:`https://${t}.openai.azure.com/`;super.setName("Azure OpenAI"),super.setAPIURL(new URL(`/openai/deployments/${n}?api-version=${o}`,u).href),super.setHeaders(async()=>({"api-key":e}))}};var Ut=class s{services;currentServiceIndex=0;currentService;debug;initialBackoffMs;maxBackoffMs;maxRetries;serviceFailures=new Map;constructor(e,t){if(e.length===0)throw new Error("No AI services provided.");Si(e),this.services=[...e].sort(t?.comparator??s.metricComparator);let n=this.services[this.currentServiceIndex];if(n===void 0)throw new Error("Error initializing the AI services.");this.currentService=n,this.debug=t?.debug??!0,this.initialBackoffMs=t?.initialBackoffMs??1e3,this.maxBackoffMs=t?.maxBackoffMs??32e3,this.maxRetries=t?.maxRetries??3}static create(e,t){return new s(e,t)}getLastUsedChatModel(){return this.currentService.getLastUsedChatModel()}getLastUsedEmbedModel(){return this.currentService.getLastUsedEmbedModel()}getLastUsedModelConfig(){return this.currentService.getLastUsedModelConfig()}static inputOrderComparator=()=>0;static metricComparator=(e,t)=>{let n=e.getMetrics(),o=t.getMetrics();return n.latency.chat.mean-o.latency.chat.mean};getModelList(){return this.currentService.getModelList()}getNextService(){let e=this.services[++this.currentServiceIndex];return e===void 0?!1:(this.currentService=e,!0)}reset(){this.currentServiceIndex=0;let e=this.services[this.currentServiceIndex];if(e===void 0)throw new Error("No AI services provided.");this.currentService=e}getName(){return this.currentService.getName()}getId(){return this.currentService.getId()}getFeatures(e){return this.currentService.getFeatures(e)}getMetrics(){return this.currentService.getMetrics()}canRetryService(){let e=this.serviceFailures.get(this.currentService.getId());if(!e)return!0;let{retries:t,lastFailureTime:n}=e,o=Date.now()-n,r=Math.min(this.initialBackoffMs*2**t,this.maxBackoffMs);return o>=r}handleFailure(){let t=(this.serviceFailures.get(this.currentService.getId())?.retries??0)+1;if(this.serviceFailures.set(this.currentService.getId(),{retries:t,lastFailureTime:Date.now()}),this.debug&&console.warn(`AxBalancer: Service ${this.currentService.getName()} failed (retry ${t}/${this.maxRetries})`),t>=this.maxRetries){let n=this.getNextService();return this.debug&&console.warn(`AxBalancer: Switching to service ${this.currentService.getName()}`),n}return!0}handleSuccess(){this.serviceFailures.delete(this.currentService.getId())}async chat(e,t){for(this.reset();;){if(!this.canRetryService()){if(!this.getNextService())throw new Error("All services exhausted");continue}try{let n=await this.currentService.chat(e,t);return this.handleSuccess(),n}catch(n){if(!(n instanceof U))throw n;switch(n.constructor){case ne:throw n;case le:break;case Z:break;case te:break;case K:break;case pe:break;default:throw n}if(!this.handleFailure())throw n}}}async embed(e,t){for(this.reset();;){if(!this.canRetryService()){if(!this.getNextService())throw new Error("All services exhausted");continue}try{let n=await this.currentService.embed(e,t);return this.handleSuccess(),n}catch(n){if(!this.handleFailure())throw n}}}setOptions(e){this.currentService.setOptions(e)}getOptions(){return this.currentService.getOptions()}getLogger(){return this.currentService.getLogger()}};function Si(s){let e=s.find(o=>o.getModelList()!==void 0);if(!e)return;let t=e.getModelList();if(!t)throw new Error("No model list found in any service.");let n=new Set(t.map(o=>o.key));for(let o=0;o<s.length;o++){let r=s[o];if(!r)throw new Error(`Service at index ${o} is undefined`);let i=r.getModelList();if(!i)throw new Error(`Service at index ${o} (${r.getName()}) has no model list while another service does.`);let a=new Set(i.map(l=>l.key));for(let l of n)if(!a.has(l))throw new Error(`Service at index ${o} (${r.getName()}) is missing model "${l}"`);for(let l of a)if(!n.has(l))throw new Error(`Service at index ${o} (${r.getName()}) has extra model "${l}"`)}}function ue(s){let e=!1,t=!1,n=!1,o=!1,r=!1,i=!1,a=!1,l=new Set,p=0;if(s.chatPrompt&&Array.isArray(s.chatPrompt))for(let c of s.chatPrompt){if(c.role==="user"&&Array.isArray(c.content))for(let u of c.content)switch(l.add(u.type),u.type){case"image":e=!0,u.cache&&(a=!0),p+=85;break;case"audio":t=!0,u.cache&&(a=!0),p+=u.duration||60;break;case"file":n=!0,u.cache&&(a=!0),p+=Math.ceil((u.extractedText?.length||1e3)/4);break;case"url":o=!0,u.cache&&(a=!0),p+=Math.ceil((u.cachedContent?.length||2e3)/4);break;case"text":u.cache&&(a=!0),p+=Math.ceil(u.text.length/4);break}else"content"in c&&typeof c.content=="string"&&(p+=Math.ceil(c.content.length/4));"cache"in c&&c.cache&&(a=!0)}return s.functions&&s.functions.length>0&&(r=!0),s.modelConfig?.stream===!0&&(i=!0),s.capabilities&&(s.capabilities.requiresImages&&(e=!0),s.capabilities.requiresAudio&&(t=!0),s.capabilities.requiresFiles&&(n=!0),s.capabilities.requiresWebSearch&&(o=!0)),{hasImages:e,hasAudio:t,hasFiles:n,hasUrls:o,requiresFunctions:r,requiresStreaming:i,requiresCaching:a,contentTypes:l,estimatedTokens:p}}function io(s,e){let t=s.getFeatures(),n=[],o=[],r=[];if(e.hasImages&&!t.media.images.supported&&(n.push("Image support"),r.push("Use altText for images or imageToText service")),e.hasAudio&&!t.media.audio.supported&&(n.push("Audio support"),r.push("Pre-transcribe audio or use transcription field")),e.hasFiles&&!t.media.files.supported&&(n.push("File support"),r.push("Pre-extract text content or use extractedText field")),e.hasUrls&&!t.media.urls.supported&&(n.push("URL/Web search support"),r.push("Pre-fetch content or use cachedContent field")),e.requiresFunctions&&!t.functions&&n.push("Function calling"),e.requiresStreaming&&!t.streaming&&(n.push("Streaming responses"),r.push("Use non-streaming mode")),e.requiresCaching&&!t.caching.supported&&(n.push("Content caching"),r.push("Repeated content will not be cached")),e.hasImages&&t.media.images.supported){let a=t.media.images.maxSize;a&&a<10*1024*1024&&o.push(`Image size limit is ${Math.round(a/(1024*1024))}MB`)}if(e.hasAudio&&t.media.audio.supported){let a=t.media.audio.maxDuration;a&&a<600&&o.push(`Audio duration limit is ${Math.round(a/60)} minutes`)}return{isSupported:n.length===0,missingCapabilities:n,warnings:o,alternatives:r}}function Bt(s,e){return s.map(t=>{let n=t.getFeatures(),o=io(t,e),r=0,i=[];return r+=10,e.hasImages&&n.media.images.supported&&(r+=25,i.push("Images"),n.media.images.detailLevels?.includes("high")&&(r+=5),n.media.images.maxSize&&n.media.images.maxSize>10*1024*1024&&(r+=3)),e.hasAudio&&n.media.audio.supported&&(r+=25,i.push("Audio"),n.media.audio.maxDuration&&n.media.audio.maxDuration>600&&(r+=5)),e.hasFiles&&n.media.files.supported&&(r+=25,i.push("Files"),n.media.files.uploadMethod==="cloud"&&(r+=3)),e.hasUrls&&n.media.urls.supported&&(r+=25,i.push("URLs"),n.media.urls.webSearch&&(r+=5)),e.requiresFunctions&&n.functions&&(r+=15,i.push("Functions"),n.functionCot&&(r+=3)),e.requiresStreaming&&n.streaming&&(r+=10,i.push("Streaming")),e.requiresCaching&&n.caching.supported&&(r+=8,i.push("Caching"),n.caching.types.includes("persistent")&&(r+=3)),n.thinking&&(r+=2),n.multiTurn&&(r+=2),n.hasThinkingBudget&&(r+=1),n.hasShowThoughts&&(r+=1),r-=o.missingCapabilities.length*10,{provider:t,score:r,missingCapabilities:o.missingCapabilities,supportedCapabilities:i}}).sort((t,n)=>n.score-t.score)}function qt(s,e,t={}){if(e.length===0)throw new Error("No providers available");let n=ue(s),o=Bt(e,n);if(t.requireExactMatch){let r=o.filter(i=>i.missingCapabilities.length===0);if(r.length===0)throw new Error(`No providers fully support the request requirements: ${o[0]?.missingCapabilities.join(", ")||"unknown requirements"}`);return r[0].provider}if(!t.allowDegradation){let r=o[0];if(r.missingCapabilities.length>0)throw new Error(`Best available provider (${r.provider.getName()}) is missing: ${r.missingCapabilities.join(", ")}`)}return o[0].provider}function $r(s,e){let t=ue(s),n=Bt(e,t),o=n[0]?.provider||null,r=[t.hasImages&&"images",t.hasAudio&&"audio",t.hasFiles&&"files",t.hasUrls&&"URLs",t.requiresFunctions&&"functions",t.requiresStreaming&&"streaming",t.requiresCaching&&"caching"].filter(Boolean).length,i=o?n[0].supportedCapabilities.length:0,a=o?`${o.getName()} supports ${i}/${r} requirements (${Math.round(i/Math.max(r,1)*100)}% compatibility)`:"No suitable providers found";return{requirements:t,providerScores:n,recommendedProvider:o,summary:a}}function Ur(s,e){return s.filter(t=>t.getFeatures().media[e].supported)}function Br(s,e){let t={};for(let n of s){let r=n.getFeatures().media[e];if(r.supported)for(let i of r.formats)t[i]||(t[i]=[]),t[i].push(n)}return t}var nt=(o=>(o.CommandRPlus="command-r-plus",o.CommandR="command-r",o.Command="command",o.CommandLight="command-light",o))(nt||{}),ot=(o=>(o.EmbedEnglishV30="embed-english-v3.0",o.EmbedEnglishLightV30="embed-english-light-v3.0",o.EmbedMultiLingualV30="embed-multilingual-v3.0",o.EmbedMultiLingualLightV30="embed-multilingual-light-v3.0",o))(ot||{});var zt=[{name:"command-r-plus",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15},{name:"command-r",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"command",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"command-light",currency:"usd",promptTokenCostPer1M:.3,completionTokenCostPer1M:.6},{name:"embed-english-light-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"embed-english-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"embed-multilingual-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"embed-multilingual-light-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1}];var lo=()=>structuredClone({model:"command-r-plus",embedModel:"embed-english-v3.0",...w()}),zr=()=>structuredClone({model:"command-r",embedModel:"embed-english-v3.0",..._()}),ao=class{constructor(e){this.config=e}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,frequencyPenalty:e.frequencyPenalty,presencePenalty:e.presencePenalty,endSequences:e.endSequences,stopSequences:e.stopSequences,stream:e.stream,n:e.n}}createChatReq(e){let t=e.model,n=e.chatPrompt.at(-1),o=e.chatPrompt.slice(0,-1),r;n&&n.role==="user"&&typeof n.content=="string"&&(r=n?.content);let i=vi(o),a=e.functions?.map(u=>{let d={};if(u.parameters?.properties)for(let[m,h]of Object.entries(u.parameters.properties))d[m]={description:h.description,type:h.type,required:u.parameters.required?.includes(m)??!1};return{name:u.name,description:u.description,parameter_definitions:d}}),l=e.chatPrompt.filter(u=>u.role==="function").map(u=>{let d=a?.find(m=>m.name===u.functionId);if(!d)throw new Error("Function not found");return{call:{name:d.name,parameters:d.parameter_definitions},outputs:[{result:u.result??""}]}}),p={name:"/chat"},c={message:r,model:t,tools:a,...l&&!r?{tool_results:l}:{},chat_history:i,max_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,k:e.modelConfig?.topK??this.config.topK,p:e.modelConfig?.topP??this.config.topP,frequency_penalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,end_sequences:this.config.endSequences,stop_sequences:e.modelConfig?.stopSequences??this.config.stopSequences};return[p,c]}createEmbedReq=e=>{let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n={name:"/embed"},o={model:t,texts:e.texts??[],input_type:"classification",truncate:""};return[n,o]};createChatResp=e=>{this.tokensUsed=e.meta.billed_units?{promptTokens:e.meta.billed_units.input_tokens,completionTokens:e.meta.billed_units.output_tokens,totalTokens:e.meta.billed_units.input_tokens+e.meta.billed_units.output_tokens}:void 0;let t;if("finish_reason"in e)switch(e.finish_reason){case"COMPLETE":t="stop";break;case"MAX_TOKENS":t="length";break;case"ERROR":throw new Error("Finish reason: ERROR");case"ERROR_TOXIC":throw new Error("Finish reason: CONTENT_FILTER");default:t="stop";break}let n;return"tool_calls"in e&&(n=e.tool_calls?.map(r=>({id:r.name,type:"function",function:{name:r.name,params:r.parameters}}))),{results:[{index:0,id:e.generation_id,content:e.text,functionCalls:n,finishReason:t}],remoteId:e.response_id}};createChatStreamResp=(e,t)=>{let n=t;e.event_type==="stream-start"&&(n.generation_id=e.generation_id),this.tokensUsed={promptTokens:0,completionTokens:e.meta.billed_units?.output_tokens??0,totalTokens:e.meta.billed_units?.output_tokens??0};let{results:o}=this.createChatResp(e),r=o[0];if(!r)throw new Error("No result");return r.id=n.generation_id??"",{results:o}};createEmbedResp(e){return{remoteId:e.id,embeddings:e.embeddings}}},Re=class extends E{constructor({apiKey:e,config:t,options:n,models:o}){if(!e||e==="")throw new Error("Cohere API key not set");let r={...lo(),...t},i=new ao(r);super(i,{name:"Cohere",apiURL:"https://api.cohere.ai/v1",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:zt,defaults:{model:r.model},supportFor:{functions:!0,streaming:!0,media:{images:{supported:!1,formats:[],maxSize:0,detailLevels:[]},audio:{supported:!1,formats:[],maxDuration:0},files:{supported:!1,formats:[],maxSize:0,uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},options:n,models:o})}};function vi(s){return s.map(e=>{let t="";if(e.role==="system"||e.role==="assistant"||e.role==="user")if(typeof e.content=="string")t=e.content;else throw new Error("Multi-modal content not supported");switch(e.role){case"user":return{role:"USER",message:t};case"system":return{role:"SYSTEM",message:t};case"assistant":{let n=qr(e.functionCalls);return{role:"CHATBOT",message:t,tool_calls:n}}case"function":{let n=s.map(i=>{if(i.role==="assistant")return i.functionCalls?.find(a=>a.id===e.functionId)}).filter(i=>i!==void 0),o=qr(n)?.at(0);if(!o)throw new Error("Function call not found");let r=[{result:e.result}];return{role:"TOOL",tool_results:[{call:o,outputs:r}]}}default:throw new Error("Unknown role")}})}function qr(s){return s?.map(e=>{let t=typeof e.function.params=="string"?JSON.parse(e.function.params):e.function.params;return{name:e.function.name,parameters:t}})}var rt=(n=>(n.DeepSeekChat="deepseek-chat",n.DeepSeekCoder="deepseek-coder",n.DeepSeekReasoner="deepseek-reasoner",n))(rt||{});var jt=[{name:"deepseek-chat",currency:"USD",promptTokenCostPer1M:.27,completionTokenCostPer1M:1.1},{name:"deepseek-reasoner",currency:"USD",promptTokenCostPer1M:.55,completionTokenCostPer1M:2.19}];var po=()=>structuredClone({model:"deepseek-chat",...w()}),jr=()=>structuredClone({model:"deepseek-coder",..._()}),Ce=class extends F{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("DeepSeek API key not set");let i={...po(),...t};r=[...jt,...r??[]],super({apiKey:e,config:i,options:n,apiURL:"https://api.deepseek.com",modelInfo:r,supportFor:{functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},models:o}),super.setName("DeepSeek")}};var st=(c=>(c.Gemini25Pro="gemini-2.5-pro",c.Gemini25Flash="gemini-2.5-flash",c.Gemini25FlashLite="gemini-2.5-flash-lite",c.Gemini20Flash="gemini-2.0-flash",c.Gemini20FlashLite="gemini-2.0-flash-lite",c.Gemini1Pro="gemini-1.0-pro",c.Gemini15Flash="gemini-1.5-flash",c.Gemini15Flash002="gemini-1.5-flash-002",c.Gemini15Flash8B="gemini-1.5-flash-8b",c.Gemini15Pro="gemini-1.5-pro",c))(st||{}),Ht=(o=>(o.GeminiEmbedding="gemini-embedding-exp",o.TextEmbeddingLarge="text-embedding-large-exp-03-07",o.TextEmbedding004="text-embedding-004",o.TextEmbedding005="text-embedding-005",o))(Ht||{}),Kt=(o=>(o.HarmCategoryHarassment="HARM_CATEGORY_HARASSMENT",o.HarmCategoryHateSpeech="HARM_CATEGORY_HATE_SPEECH",o.HarmCategorySexuallyExplicit="HARM_CATEGORY_SEXUALLY_EXPLICIT",o.HarmCategoryDangerousContent="HARM_CATEGORY_DANGEROUS_CONTENT",o))(Kt||{}),Vt=(r=>(r.BlockNone="BLOCK_NONE",r.BlockOnlyHigh="BLOCK_ONLY_HIGH",r.BlockMediumAndAbove="BLOCK_MEDIUM_AND_ABOVE",r.BlockLowAndAbove="BLOCK_LOW_AND_ABOVE",r.BlockDefault="HARM_BLOCK_THRESHOLD_UNSPECIFIED",r))(Vt||{}),co=(l=>(l.SemanticSimilarity="SEMANTIC_SIMILARITY",l.Classification="CLASSIFICATION",l.Clustering="CLUSTERING",l.RetrievalDocument="RETRIEVAL_DOCUMENT",l.RetrievalQuery="RETRIEVAL_QUERY",l.QuestionAnswering="QUESTION_ANSWERING",l.FactVerification="FACT_VERIFICATION",l.CodeRetrievalQuery="CODE_RETRIEVAL_QUERY",l))(co||{});var Wt=[{name:"gemini-2.5-pro",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:2.5,completionTokenCostPer1M:15,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"gemini-2.5-flash",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:15,completionTokenCostPer1M:3.5,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"gemini-2.5-flash-lite",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.1,completionTokenCostPer1M:.4,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"gemini-2.0-flash",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.01,completionTokenCostPer1M:.4},{name:"gemini-2.0-flash-lite",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:0,completionTokenCostPer1M:0},{name:"gemini-1.5-flash",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.075,completionTokenCostPer1M:.3},{name:"gemini-1.5-flash-8b",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.0375,completionTokenCostPer1M:.15},{name:"gemini-1.5-pro",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:1.25,completionTokenCostPer1M:5},{name:"gemini-1.0-pro",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5}];var uo=s=>{if(!s||typeof s!="object")return s;let e={...s};return delete e.additionalProperties,delete e.default,delete e.optional,delete e.maximum,delete e.oneOf,delete e.anyOf,e.properties&&typeof e.properties=="object"&&(e.properties=Object.fromEntries(Object.entries(e.properties).map(([t,n])=>[t,uo(n)]))),e.items&&(e.items=uo(e.items)),e},Hr=[{category:"HARM_CATEGORY_HARASSMENT",threshold:"BLOCK_NONE"},{category:"HARM_CATEGORY_HATE_SPEECH",threshold:"BLOCK_NONE"},{category:"HARM_CATEGORY_SEXUALLY_EXPLICIT",threshold:"BLOCK_NONE"},{category:"HARM_CATEGORY_DANGEROUS_CONTENT",threshold:"BLOCK_NONE"}],go=()=>structuredClone({model:"gemini-2.5-flash",embedModel:"text-embedding-005",safetySettings:Hr,thinkingTokenBudgetLevels:{minimal:200,low:800,medium:5e3,high:1e4,highest:24500},...w()}),Kr=()=>structuredClone({model:"gemini-2.0-flash",embedModel:"text-embedding-005",safetySettings:Hr,thinkingTokenBudgetLevels:{minimal:200,low:800,medium:5e3,high:1e4,highest:24500},..._()}),mo=class{constructor(e,t,n,o,r){this.config=e;this.isVertex=t;this.endpointId=n;this.apiKey=o;this.options=r;if(!this.isVertex&&this.config.autoTruncate)throw new Error("Auto truncate is not supported for GoogleGemini")}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,endSequences:e.endSequences,stream:e.stream,n:e.n}}createChatReq=async(e,t)=>{let n=e.model,o=e.modelConfig?.stream??this.config.stream;if(!e.chatPrompt||e.chatPrompt.length===0)throw new Error("Chat prompt is empty");let r;if(this.endpointId?r={name:o?`/${this.endpointId}:streamGenerateContent?alt=sse`:`/${this.endpointId}:generateContent`}:r={name:o?`/models/${n}:streamGenerateContent?alt=sse`:`/models/${n}:generateContent`},!this.isVertex){let g=o?"&":"?",f=typeof this.apiKey=="function"?await this.apiKey():this.apiKey;r.name+=`${g}key=${f}`}let i=e.chatPrompt.filter(g=>g.role==="system").map(g=>g.content),a=i.length>0?{role:"user",parts:[{text:i.join(" ")}]}:void 0,l=e.chatPrompt.filter(g=>g.role!=="system").map((g,f)=>{switch(g.role){case"user":return{role:"user",parts:Array.isArray(g.content)?g.content.map((x,y)=>{switch(x.type){case"text":return{text:x.text};case"image":return{inlineData:{mimeType:x.mimeType,data:x.image}};case"file":return"fileUri"in x?{fileData:{mimeType:x.mimeType,fileUri:x.fileUri}}:{inlineData:{mimeType:x.mimeType,data:x.data}};default:throw new Error(`Chat prompt content type not supported (index: ${y})`)}}):[{text:g.content}]};case"assistant":{let A=[];if(g.functionCalls){if(A=g.functionCalls.map(x=>{let y=typeof x.function.params=="string"?JSON.parse(x.function.params):x.function.params;return{functionCall:{name:x.function.name,args:y}}}),!A)throw new Error("Function call is empty");return{role:"model",parts:A}}if(!g.content)throw new Error("Assistant content is empty");return A=[{text:g.content}],{role:"model",parts:A}}case"function":{if(!("functionId"in g))throw new Error(`Chat prompt functionId is empty (index: ${f})`);return{role:"user",parts:[{functionResponse:{name:g.functionId,response:{result:g.result}}}]}}default:throw new Error(`Invalid role: ${JSON.stringify(g)} (index: ${f})`)}}),p=[];if(e.functions&&e.functions.length>0){let g=e.functions.map(f=>({...f,parameters:f.parameters?uo(f.parameters):void 0}));p.push({function_declarations:g})}this.options?.codeExecution&&p.push({code_execution:{}}),this.options?.googleSearchRetrieval&&p.push({google_search_retrieval:{dynamic_retrieval_config:this.options.googleSearchRetrieval}}),this.options?.googleSearch&&p.push({google_search:{}}),this.options?.urlContext&&p.push({url_context:{}}),p.length===0&&(p=void 0);let c;if(e.functionCall)if(e.functionCall==="none")c={function_calling_config:{mode:"NONE"}};else if(e.functionCall==="auto")c={function_calling_config:{mode:"AUTO"}};else if(e.functionCall==="required")c={function_calling_config:{mode:"ANY"}};else{let g=e.functionCall.function?.name?{allowedFunctionNames:[e.functionCall.function.name]}:{};c={function_calling_config:{mode:"ANY"},...g}}else p&&p.length>0&&(c={function_calling_config:{mode:"AUTO"}});let u={};if(this.config.thinking?.includeThoughts&&(u.includeThoughts=!0),this.config.thinking?.thinkingTokenBudget&&(u.thinkingBudget=this.config.thinking.thinkingTokenBudget),t?.thinkingTokenBudget){let g=this.config.thinkingTokenBudgetLevels;switch(t.thinkingTokenBudget){case"none":u.thinkingBudget=0,u.includeThoughts=!1;break;case"minimal":u.thinkingBudget=g?.minimal??200;break;case"low":u.thinkingBudget=g?.low??800;break;case"medium":u.thinkingBudget=g?.medium??5e3;break;case"high":u.thinkingBudget=g?.high??1e4;break;case"highest":u.thinkingBudget=g?.highest??24500;break}}t?.showThoughts!==void 0&&t?.thinkingTokenBudget!=="none"&&(u.includeThoughts=t.showThoughts);let d={maxOutputTokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,topP:e.modelConfig?.topP??this.config.topP,topK:e.modelConfig?.topK??this.config.topK,frequencyPenalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,candidateCount:1,stopSequences:e.modelConfig?.stopSequences??this.config.stopSequences,responseMimeType:"text/plain",...Object.keys(u).length>0?{thinkingConfig:u}:{}},m=this.config.safetySettings;return[r,{contents:l,tools:p,toolConfig:c,systemInstruction:a,generationConfig:d,safetySettings:m}]};createEmbedReq=async e=>{let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n,o;if(this.isVertex)this.endpointId?n={name:`/${this.endpointId}:predict`}:n={name:`/models/${t}:predict`},o={instances:e.texts.map(r=>({content:r,...this.config.embedType&&{taskType:this.config.embedType}})),parameters:{autoTruncate:this.config.autoTruncate,outputDimensionality:this.config.dimensions}};else{let r=typeof this.apiKey=="function"?this.apiKey():this.apiKey;n={name:`/models/${t}:batchEmbedContents?key=${r}`},o={requests:e.texts.map(i=>({model:`models/${t}`,content:{parts:[{text:i}]},outputDimensionality:this.config.dimensions,...this.config.embedType&&{taskType:this.config.embedType}}))}}return[n,o]};createChatResp=e=>{let t=e.candidates?.map(n=>{let o={index:0};switch(n.finishReason){case"MAX_TOKENS":o.finishReason="length";break;case"STOP":o.finishReason="stop";break;case"SAFETY":throw new O("Content was blocked due to safety settings",void 0,void 0);case"RECITATION":throw new O("Content was blocked due to recitation policy",void 0,void 0);case"MALFORMED_FUNCTION_CALL":throw new O("Function call was malformed and blocked",void 0,void 0);case"UNEXPECTED_TOOL_CALL":throw new O("Unexpected tool call",void 0,void 0);case"FINISH_REASON_UNSPECIFIED":throw new O("Finish reason unspecified",void 0,void 0);case"BLOCKLIST":throw new O("Content was blocked due to blocklist",void 0,void 0);case"PROHIBITED_CONTENT":throw new O("Content was blocked due to prohibited content",void 0,void 0);case"SPII":throw new O("Content was blocked due to SPII",void 0,void 0);case"OTHER":throw new O("Other finish reason",void 0,void 0)}if(!n.content||!n.content.parts)return o;for(let r of n.content.parts){if("text"in r){"thought"in r&&r.thought?o.thought=r.text:o.content=r.text;continue}"functionCall"in r&&(o.functionCalls=[{id:$(),type:"function",function:{name:r.functionCall.name,params:r.functionCall.args}}])}return o});return e.usageMetadata&&(this.tokensUsed={totalTokens:e.usageMetadata.totalTokenCount,promptTokens:e.usageMetadata.promptTokenCount,completionTokens:e.usageMetadata.candidatesTokenCount,thoughtsTokens:e.usageMetadata.thoughtsTokenCount}),{results:t}};createChatStreamResp=e=>this.createChatResp(e);createEmbedResp=e=>{let t;return this.isVertex?t=e.predictions.map(n=>n.embeddings.values):t=e.embeddings.map(n=>n.values),{embeddings:t}}},Te=class s extends E{static create(e){return new s(e)}constructor({apiKey:e,projectId:t,region:n,endpointId:o,config:r,options:i,models:a,modelInfo:l}){let p=t!==void 0&&n!==void 0,c,u;if(p){if(!e)throw new Error("GoogleGemini Vertex API key not set");if(typeof e!="function")throw new Error("GoogleGemini Vertex API key must be a function for token-based authentication");let g;o?g="endpoints":g="publishers/google",c=`https://${n==="global"?"aiplatform":`${n}-aiplatform`}.googleapis.com/v1/projects/${t}/locations/${n}/${g}`,u=async()=>({Authorization:`Bearer ${typeof e=="function"?await e():e}`})}else{if(!e)throw new Error("GoogleGemini AI API key not set");c="https://generativelanguage.googleapis.com/v1beta",u=async()=>({})}let d={...go(),...r},m=new mo(d,p,o,e,i);l=[...Wt,...l??[]];let h=g=>{let f=q({model:g,modelInfo:l,models:a});return{functions:!0,streaming:!0,hasThinkingBudget:f?.hasThinkingBudget??!1,hasShowThoughts:f?.hasShowThoughts??!1,functionCot:!1,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:20*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!0,formats:["wav","mp3","aac","ogg"],maxDuration:9.5*60},files:{supported:!0,formats:["application/pdf","text/plain","text/csv","text/html","text/xml"],maxSize:2*1024*1024*1024,uploadMethod:"cloud"},urls:{supported:!0,webSearch:!0,contextFetching:!0}},caching:{supported:!1,types:[]},thinking:f?.hasThinkingBudget??!1,multiTurn:!0}};super(m,{name:"GoogleGeminiAI",apiURL:c,headers:u,modelInfo:l,defaults:{model:d.model,embedModel:d.embedModel},options:i,supportFor:h,models:a})}};var Oi=new G,we=class{options;maxTokens;refillRate;currentTokens;lastRefillTime;constructor(e,t,n){this.maxTokens=e,this.refillRate=t,this.currentTokens=e,this.lastRefillTime=Date.now(),this.options=n}refillTokens(){let e=Date.now(),n=(e-this.lastRefillTime)/1e3*this.refillRate;this.currentTokens=Math.min(this.maxTokens,this.currentTokens+n),this.lastRefillTime=e}async waitUntilTokensAvailable(e){if(this.refillTokens(),this.currentTokens>=e){this.currentTokens-=e;return}return this.options?.debug&&console.log(Oi.red(`Rate limiter: Waiting for ${e-this.currentTokens} tokens`)),await new Promise(t=>setTimeout(t,100)),this.waitUntilTokensAvailable(e)}async acquire(e){await this.waitUntilTokensAvailable(e)}};var it=(o=>(o.Llama3_8B="llama3-8b-8192",o.Llama33_70B="llama-3.3-70b-versatile",o.Mixtral_8x7B="mixtral-8x7b-32768",o.Gemma2_9B="gemma2-9b-it",o))(it||{});var Jt=[{name:"gemma2-9b-it",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.2,completionTokenCostPer1M:.2},{name:"llama-3.3-70b-versatile",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.59,completionTokenCostPer1M:.79},{name:"llama3-8b-8192",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.05,completionTokenCostPer1M:.08},{name:"mixtral-8x7b-32768",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.24,completionTokenCostPer1M:.24}];var Mi=()=>structuredClone({model:"llama-3.3-70b-versatile",...w()}),Se=class extends F{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("Groq API key not set");let i={...Mi(),...t},a={...n,streamingUsage:!1};r=[...Jt,...r??[]];let l={functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0};super({apiKey:e,config:i,options:a,modelInfo:r,apiURL:"https://api.groq.com/openai/v1",models:o,supportFor:l}),super.setName("Groq"),this.setOptions(a)}setOptions=e=>{let t=this.newRateLimiter(e);super.setOptions({...e,rateLimiter:t})};newRateLimiter=e=>{if(e?.rateLimiter)return e.rateLimiter;let t=e?.tokensPerMinute??4800,n=new we(t,t/60,{debug:e?.debug});return async(r,i)=>{let a=i.modelUsage?.tokens?.totalTokens||0;return await n.acquire(a),await r()}}};var Qt=[];var Yt=(e=>(e.MetaLlama270BChatHF="meta-llama/Llama-2-70b-chat-hf",e))(Yt||{});var Ao=()=>structuredClone({model:"meta-llama/Llama-2-70b-chat-hf",...w()}),Vr=()=>structuredClone({model:"meta-llama/Llama-2-70b-chat-hf",..._()}),ho=class{constructor(e){this.config=e}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,n:e.n,presencePenalty:e.presencePenalty}}createChatReq=e=>{let t=e.model,n=e.functions?`Functions:
109
+ `)});break;default:r=JSON.stringify(n,null,2)}s(r)}};var C={LLM_SYSTEM:"gen_ai.system",LLM_OPERATION_NAME:"gen_ai.operation.name",LLM_REQUEST_MODEL:"gen_ai.request.model",LLM_REQUEST_MAX_TOKENS:"gen_ai.request.max_tokens",LLM_REQUEST_TEMPERATURE:"gen_ai.request.temperature",LLM_REQUEST_TOP_K:"gen_ai.request.top_k",LLM_REQUEST_FREQUENCY_PENALTY:"gen_ai.request.frequency_penalty",LLM_REQUEST_PRESENCE_PENALTY:"gen_ai.request.presence_penalty",LLM_REQUEST_STOP_SEQUENCES:"gen_ai.request.stop_sequences",LLM_REQUEST_LLM_IS_STREAMING:"gen_ai.request.llm_is_streaming",LLM_REQUEST_TOP_P:"gen_ai.request.top_p",LLM_USAGE_INPUT_TOKENS:"gen_ai.usage.input_tokens",LLM_USAGE_OUTPUT_TOKENS:"gen_ai.usage.output_tokens",LLM_USAGE_TOTAL_TOKENS:"gen_ai.usage.total_tokens",LLM_USAGE_THOUGHTS_TOKENS:"gen_ai.usage.thoughts_tokens",DB_SYSTEM:"db.system",DB_TABLE:"db.table",DB_NAMESPACE:"db.namespace",DB_ID:"db.id",DB_QUERY_TEXT:"db.query.text",DB_VECTOR:"db.vector",DB_OPERATION_NAME:"db.operation.name",DB_VECTOR_QUERY_TOP_K:"db.vector.query.top_k",DB_QUERY_EMBEDDINGS:"db.query.embeddings",DB_QUERY_RESULT:"db.query.result",DB_QUERY_EMBEDDINGS_VECTOR:"db.query.embeddings.vector",DB_QUERY_RESULT_ID:"db.query.result.id",DB_QUERY_RESULT_SCORE:"db.query.result.score",DB_QUERY_RESULT_DISTANCE:"db.query.result.distance",DB_QUERY_RESULT_METADATA:"db.query.result.metadata",DB_QUERY_RESULT_VECTOR:"db.query.result.vector",DB_QUERY_RESULT_DOCUMENT:"db.query.result.document"},W={GEN_AI_USER_MESSAGE:"gen_ai.user.message",GEN_AI_SYSTEM_MESSAGE:"gen_ai.system.message",GEN_AI_ASSISTANT_MESSAGE:"gen_ai.assistant.message",GEN_AI_TOOL_MESSAGE:"gen_ai.tool.message",GEN_AI_CHOICE:"gen_ai.choice",GEN_AI_USAGE:"gen_ai.usage"},Jn=(r=>(r.COMPLETION="completion",r.CHAT="chat",r.RERANK="rerank",r.UNKNOWN="unknown",r))(Jn||{}),Qn=(o=>(o.WORKFLOW="workflow",o.TASK="task",o.AGENT="agent",o.TOOL="tool",o.UNKNOWN="unknown",o))(Qn||{});var Yn=class{buffer;doneCallback;transformFn;constructor(e,t){this.transformFn=e,this.doneCallback=t,this.buffer=t?[]:void 0}async transform(e,t){let n=this.transformFn(e);n&&(t.enqueue(n),this.buffer?.push(n))}async flush(e){await this.doneCallback?.(this.buffer??[]),e.terminate()}},Ft=class extends TransformStream{constructor(e,t){super(new Yn(e,t))}};function _t(s,e){for(let t of e){let n=s.find(r=>r.id===t.id);n?(typeof t.function.name=="string"&&t.function.name.length>0&&(n.function.name+=t.function.name),typeof t.function.params=="string"&&t.function.params.length>0&&(n.function.params+=t.function.params),typeof t.function.params=="object"&&(n.function.params=t.function.params)):s.push(t)}}var io=(s,e,t,n)=>{let r=n?s.filter(i=>i.role!=="system"):[...s];t({name:"ChatRequestChatPrompt",step:e,value:r})};var ao=(s,e)=>{if(!s.results)return;let t={name:"ChatResponseResults",value:s.results};e(t)},lo=(s,e,t)=>{t({name:"ChatResponseStreamingResult",index:e,value:s})};function po(s,e){let t=new Map;for(let n of s)for(let r of n.results){if(!r)continue;let o=t.get(r.index);o?(r.content&&(o.content=(o.content??"")+r.content),r.thought&&(o.thought=(o.thought??"")+r.thought),r.finishReason&&(o.finishReason=r.finishReason),r.functionCalls&&(o.functionCalls?_t(o.functionCalls,structuredClone(r.functionCalls)):o.functionCalls=structuredClone(r.functionCalls))):(o=structuredClone(r),t.set(r.index,o))}for(let n of t.values()){let r={name:"ChatResponseStreamingDoneResult",index:n.index,value:n};e(r)}}var co=(s,e)=>{e({name:"FunctionResults",value:s})},uo=(s,e,t,n)=>{n({name:"FunctionError",index:e,fixingInstructions:t,error:s})},mo=(s,e,t,n)=>{n({name:"ValidationError",index:e,fixingInstructions:t,error:s})},go=(s,e,t,n)=>{n({name:"AssertionError",index:e,fixingInstructions:t,error:s})},ho=(s,e,t)=>{t({name:"RefusalError",index:e,error:s})};var Ao=(s,e,t)=>{t({name:"EmbedRequest",embedModel:e,value:s})},xo=(s,e)=>{let t=s.slice(0,3).map(r=>({length:r.length,sample:r.slice(0,5),truncated:r.length>5})),n={name:"EmbedResponse",totalEmbeddings:s.length,value:t};e(n)},fo=(s,e,t,n)=>{n({name:"ResultPickerUsed",sampleCount:s,selectedIndex:e,latency:t})};var Xn=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let r=String(n);e[t]=r.length>100?r.substring(0,100):r}return e},Dt,yo=s=>{if(Dt)return Dt;if(s)return Dt=bi(s),Dt};var bi=s=>({latencyHistogram:s.createHistogram("ax_llm_request_duration_ms",{description:"Duration of LLM requests in milliseconds",unit:"ms"}),errorCounter:s.createCounter("ax_llm_errors_total",{description:"Total number of LLM request errors"}),requestCounter:s.createCounter("ax_llm_requests_total",{description:"Total number of LLM requests"}),tokenCounter:s.createCounter("ax_llm_tokens_total",{description:"Total number of LLM tokens consumed"}),inputTokenCounter:s.createCounter("ax_llm_input_tokens_total",{description:"Total number of input/prompt tokens consumed"}),outputTokenCounter:s.createCounter("ax_llm_output_tokens_total",{description:"Total number of output/completion tokens generated"}),errorRateGauge:s.createGauge("ax_llm_error_rate",{description:"Current error rate as a percentage (0-100)"}),meanLatencyGauge:s.createGauge("ax_llm_mean_latency_ms",{description:"Mean latency of LLM requests in milliseconds",unit:"ms"}),p95LatencyGauge:s.createGauge("ax_llm_p95_latency_ms",{description:"95th percentile latency of LLM requests in milliseconds",unit:"ms"}),p99LatencyGauge:s.createGauge("ax_llm_p99_latency_ms",{description:"99th percentile latency of LLM requests in milliseconds",unit:"ms"}),streamingRequestsCounter:s.createCounter("ax_llm_streaming_requests_total",{description:"Total number of streaming LLM requests"}),functionCallsCounter:s.createCounter("ax_llm_function_calls_total",{description:"Total number of function/tool calls made"}),functionCallLatencyHistogram:s.createHistogram("ax_llm_function_call_latency_ms",{description:"Latency of function calls in milliseconds",unit:"ms"}),requestSizeHistogram:s.createHistogram("ax_llm_request_size_bytes",{description:"Size of LLM request payloads in bytes",unit:"By"}),responseSizeHistogram:s.createHistogram("ax_llm_response_size_bytes",{description:"Size of LLM response payloads in bytes",unit:"By"}),temperatureGauge:s.createGauge("ax_llm_temperature_gauge",{description:"Temperature setting used for LLM requests"}),maxTokensGauge:s.createGauge("ax_llm_max_tokens_gauge",{description:"Maximum tokens setting used for LLM requests"}),estimatedCostCounter:s.createCounter("ax_llm_estimated_cost_total",{description:"Estimated cost of LLM requests in USD",unit:"$"}),promptLengthHistogram:s.createHistogram("ax_llm_prompt_length_chars",{description:"Length of prompts in characters"}),contextWindowUsageGauge:s.createGauge("ax_llm_context_window_usage_ratio",{description:"Context window utilization ratio (0-1)"}),timeoutsCounter:s.createCounter("ax_llm_timeouts_total",{description:"Total number of timed out LLM requests"}),abortsCounter:s.createCounter("ax_llm_aborts_total",{description:"Total number of aborted LLM requests"}),thinkingBudgetUsageCounter:s.createCounter("ax_llm_thinking_budget_usage_total",{description:"Total thinking budget tokens used"}),multimodalRequestsCounter:s.createCounter("ax_llm_multimodal_requests_total",{description:"Total number of multimodal requests (with images/audio)"})}),Io=(s,e,t,n,r)=>{try{if(s.latencyHistogram){let o=Xn({operation:e,ai_service:n,...r?{model:r}:{}});s.latencyHistogram.record(t,o)}}catch(o){console.warn("Failed to record latency metric:",o)}},bo=(s,e,t,n,r,o,i)=>{let a={operation:e,ai_service:o,...i?{model:i}:{}};s.meanLatencyGauge&&s.meanLatencyGauge.record(t,a),s.p95LatencyGauge&&s.p95LatencyGauge.record(n,a),s.p99LatencyGauge&&s.p99LatencyGauge.record(r,a)},Ro=(s,e,t,n)=>{try{if(s.errorCounter){let r=Xn({operation:e,ai_service:t,...n?{model:n}:{}});s.errorCounter.add(1,r)}}catch(r){console.warn("Failed to record error metric:",r)}},To=(s,e,t,n,r)=>{s.errorRateGauge&&s.errorRateGauge.record(t*100,{operation:e,ai_service:n,...r?{model:r}:{}})},Co=(s,e,t,n)=>{s.requestCounter&&s.requestCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},Xe=(s,e,t,n,r)=>{try{let o=Xn({ai_service:n,...r?{model:r}:{}});s.tokenCounter&&s.tokenCounter.add(t,{token_type:e,...o}),e==="input"&&s.inputTokenCounter&&s.inputTokenCounter.add(t,o),e==="output"&&s.outputTokenCounter&&s.outputTokenCounter.add(t,o)}catch(o){console.warn("Failed to record token metric:",o)}},wo=(s,e,t,n,r)=>{t&&s.streamingRequestsCounter&&s.streamingRequestsCounter.add(1,{operation:e,ai_service:n,...r?{model:r}:{}})},So=(s,e,t,n,r)=>{let o={function_name:e,...n?{ai_service:n}:{},...r?{model:r}:{}};s.functionCallsCounter&&s.functionCallsCounter.add(1,o),t&&s.functionCallLatencyHistogram&&s.functionCallLatencyHistogram.record(t,o)},Zn=(s,e,t,n,r)=>{s.requestSizeHistogram&&s.requestSizeHistogram.record(t,{operation:e,ai_service:n,...r?{model:r}:{}})},er=(s,e,t,n,r)=>{s.responseSizeHistogram&&s.responseSizeHistogram.record(t,{operation:e,ai_service:n,...r?{model:r}:{}})},vo=(s,e,t,n,r)=>{let o={...n?{ai_service:n}:{},...r?{model:r}:{}};e!==void 0&&s.temperatureGauge&&s.temperatureGauge.record(e,o),t!==void 0&&s.maxTokensGauge&&s.maxTokensGauge.record(t,o)},tr=(s,e,t,n,r)=>{s.estimatedCostCounter&&s.estimatedCostCounter.add(t,{operation:e,ai_service:n,...r?{model:r}:{}})},Oo=(s,e,t,n)=>{s.promptLengthHistogram&&s.promptLengthHistogram.record(e,{ai_service:t,...n?{model:n}:{}})},Mo=(s,e,t,n)=>{s.contextWindowUsageGauge&&s.contextWindowUsageGauge.record(e,{ai_service:t,...n?{model:n}:{}})},ko=(s,e,t,n)=>{s.timeoutsCounter&&s.timeoutsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},Eo=(s,e,t,n)=>{s.abortsCounter&&s.abortsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},Po=(s,e,t,n)=>{s.thinkingBudgetUsageCounter&&s.thinkingBudgetUsageCounter.add(e,{ai_service:t,...n?{model:n}:{}})},Fo=(s,e,t,n,r)=>{(e||t)&&s.multimodalRequestsCounter&&s.multimodalRequestsCounter.add(1,{ai_service:n,has_images:e.toString(),has_audio:t.toString(),...r?{model:r}:{}})};var S=()=>structuredClone({temperature:0,topK:40,topP:.9}),_=()=>structuredClone({temperature:.4,topP:.7,frequencyPenalty:.2}),E=class{constructor(e,{name:t,apiURL:n,headers:r,modelInfo:o,defaults:i,options:a={},supportFor:l,models:p}){this.aiImpl=e;this.name=t,this.apiURL=n||"",this.headers=r,this.supportFor=l,this.tracer=a.tracer??M.tracer,this.meter=a.meter??M.meter,this.modelInfo=o,this.models=p,this.id=N();let c=this.getModel(i.model)??i.model,u=this.getEmbedModel(i.embedModel)??i.embedModel;if(this.defaults={model:c,embedModel:u},!i.model||typeof i.model!="string"||i.model==="")throw new Error("No model defined");this.setOptions(a),p&&Ti(p)}debug=!1;rt;fetch;tracer;meter;timeout;excludeContentFromTrace;models;abortSignal;logger=M.logger??oo;corsProxy;modelInfo;modelUsage;embedModelUsage;defaults;lastUsedModelConfig;lastUsedChatModel;lastUsedEmbedModel;apiURL;name;id;headers;supportFor;metrics={latency:{chat:{mean:0,p95:0,p99:0,samples:[]},embed:{mean:0,p95:0,p99:0,samples:[]}},errors:{chat:{count:0,rate:0,total:0},embed:{count:0,rate:0,total:0}}};getMetricsInstruments(){return yo(this.meter)}setName(e){this.name=e}getId(){return this.id}setAPIURL(e){this.apiURL=e}setHeaders(e){this.headers=e}setOptions(e){this.debug=e.debug??!1,this.rt=e.rateLimiter,this.fetch=e.fetch,this.timeout=e.timeout,this.tracer=e.tracer??M.tracer,this.meter=e.meter??M.meter,this.excludeContentFromTrace=e.excludeContentFromTrace,this.abortSignal=e.abortSignal,this.logger=e.logger??M.logger??this.logger,this.corsProxy=e.corsProxy}getOptions(){return{debug:this.debug,rateLimiter:this.rt,fetch:this.fetch,tracer:this.tracer,meter:this.meter,timeout:this.timeout,excludeContentFromTrace:this.excludeContentFromTrace,abortSignal:this.abortSignal,logger:this.logger,corsProxy:this.corsProxy}}getLogger(){return this.logger}getModelList(){let e=[];for(let t of this.models??[])t.isInternal||("model"in t&&t.model&&e.push({key:t.key,description:t.description,model:t.model}),"embedModel"in t&&t.embedModel&&e.push({key:t.key,description:t.description,embedModel:t.embedModel}));return e}getName(){return this.name}getFeatures(e){return typeof this.supportFor=="function"?this.supportFor(e??this.defaults.model):this.supportFor}getLastUsedChatModel(){return this.lastUsedChatModel}getLastUsedEmbedModel(){return this.lastUsedEmbedModel}getLastUsedModelConfig(){return this.lastUsedModelConfig}calculatePercentile(e,t){if(e.length===0)return 0;let n=[...e].sort((o,i)=>o-i),r=Math.ceil(t/100*n.length)-1;return n[r]??0}updateLatencyMetrics(e,t){let n=this.metrics.latency[e];n.samples.push(t),n.samples.length>1e3&&n.samples.shift(),n.mean=n.samples.reduce((o,i)=>o+i,0)/n.samples.length,n.p95=this.calculatePercentile(n.samples,95),n.p99=this.calculatePercentile(n.samples,99);let r=this.getMetricsInstruments();if(r){let o=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Io(r,e,t,this.name,o),bo(r,e,n.mean,n.p95,n.p99,this.name,o)}}updateErrorMetrics(e,t){let n=this.metrics.errors[e];n.total++,t&&n.count++,n.rate=n.count/n.total;let r=this.getMetricsInstruments();if(r){let o=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Co(r,e,this.name,o),t&&Ro(r,e,this.name,o),To(r,e,n.rate,this.name,o)}}recordTokenUsage(e){let t=this.getMetricsInstruments();if(t&&e?.tokens){let{promptTokens:n,completionTokens:r,totalTokens:o,thoughtsTokens:i}=e.tokens;n&&Xe(t,"input",n,this.name,e.model),r&&Xe(t,"output",r,this.name,e.model),o&&Xe(t,"total",o,this.name,e.model),i&&Xe(t,"thoughts",i,this.name,e.model)}}calculateRequestSize(e){try{return new TextEncoder().encode(JSON.stringify(e)).length}catch{return 0}}calculateResponseSize(e){try{return new TextEncoder().encode(JSON.stringify(e)).length}catch{return 0}}detectMultimodalContent(e){let t=!1,n=!1;if(e.chatPrompt&&Array.isArray(e.chatPrompt)){for(let r of e.chatPrompt)if(r.role==="user"&&Array.isArray(r.content))for(let o of r.content)o.type==="image"?t=!0:o.type==="audio"&&(n=!0)}return{hasImages:t,hasAudio:n}}calculatePromptLength(e){let t=0;if(e.chatPrompt&&Array.isArray(e.chatPrompt))for(let n of e.chatPrompt)if(n.role==="system"||n.role==="assistant")n.content&&(t+=n.content.length);else if(n.role==="user"){if(typeof n.content=="string")t+=n.content.length;else if(Array.isArray(n.content))for(let r of n.content)r.type==="text"&&(t+=r.text.length)}else n.role==="function"&&n.result&&(t+=n.result.length);return t}calculateContextWindowUsage(e,t){if(!t?.tokens?.promptTokens)return 0;let n=this.modelInfo.find(r=>r.name===e);return n?.contextWindow?t.tokens.promptTokens/n.contextWindow:0}estimateCost(e,t){if(!t?.tokens)return 0;let n=this.modelInfo.find(l=>l.name===e);if(!n||!n.promptTokenCostPer1M&&!n.completionTokenCostPer1M)return 0;let{promptTokens:r=0,completionTokens:o=0}=t.tokens,i=n.promptTokenCostPer1M||0,a=n.completionTokenCostPer1M||0;return r*i/1e6+o*a/1e6}estimateCostByName(e,t){if(!t?.tokens)return 0;let n=this.modelInfo.find(l=>l.name===e);if(!n||!n.promptTokenCostPer1M&&!n.completionTokenCostPer1M)return 0;let{promptTokens:r=0,completionTokens:o=0}=t.tokens,i=n.promptTokenCostPer1M||0,a=n.completionTokenCostPer1M||0;return r*i/1e6+o*a/1e6}recordFunctionCallMetrics(e,t){let n=this.getMetricsInstruments();if(!(!n||!e))for(let r of e)r&&typeof r=="object"&&"function"in r&&r.function&&typeof r.function=="object"&&"name"in r.function&&So(n,r.function.name,void 0,this.name,t)}recordTimeoutMetric(e){let t=this.getMetricsInstruments();if(t){let n=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;ko(t,e,this.name,n)}}recordAbortMetric(e){let t=this.getMetricsInstruments();if(t){let n=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Eo(t,e,this.name,n)}}recordChatMetrics(e,t,n){let r=this.getMetricsInstruments();if(!r)return;let o=this.lastUsedChatModel,i=this.lastUsedModelConfig,a=i?.stream??!1;wo(r,"chat",a,this.name,o);let{hasImages:l,hasAudio:p}=this.detectMultimodalContent(e);Fo(r,l,p,this.name,o);let c=this.calculatePromptLength(e);Oo(r,c,this.name,o),vo(r,i?.temperature,i?.maxTokens,this.name,o),t?.thinkingTokenBudget&&this.modelUsage?.tokens?.thoughtsTokens&&Po(r,this.modelUsage.tokens.thoughtsTokens,this.name,o);let u=this.calculateRequestSize(e);if(Zn(r,"chat",u,this.name,o),n&&!a){let d=n,m=this.calculateResponseSize(d);if(er(r,"chat",m,this.name,o),d.results)for(let f of d.results)f.functionCalls&&this.recordFunctionCallMetrics(f.functionCalls,this.lastUsedChatModel);let h=this.calculateContextWindowUsage(this.lastUsedChatModel,d.modelUsage);h>0&&Mo(r,h,this.name,o);let g=this.estimateCost(this.lastUsedChatModel,d.modelUsage);g>0&&tr(r,"chat",g,this.name,o)}}recordEmbedMetrics(e,t){let n=this.getMetricsInstruments();if(!n)return;let r=this.lastUsedEmbedModel,o=this.calculateRequestSize(e);Zn(n,"embed",o,this.name,r);let i=this.calculateResponseSize(t);er(n,"embed",i,this.name,r);let a=this.estimateCostByName(r,t.modelUsage);a>0&&tr(n,"embed",a,this.name,r)}getMetrics(){return structuredClone(this.metrics)}async chat(e,t){let n=performance.now(),r=!1,o;try{return o=await this._chat1(e,t),o}catch(i){throw r=!0,i instanceof Error&&(i.message.includes("timeout")||i.name==="TimeoutError"?this.recordTimeoutMetric("chat"):(i.message.includes("abort")||i.name==="AbortError")&&this.recordAbortMetric("chat")),i}finally{let i=performance.now()-n;this.updateLatencyMetrics("chat",i),this.updateErrorMetrics("chat",r),r||this.recordChatMetrics(e,t,o)}}async _chat1(e,t){let n=this.getModel(e.model)??e.model??this.defaults.model;e.chatPrompt&&Array.isArray(e.chatPrompt)&&nr(e.chatPrompt);let r={...this.aiImpl.getModelConfig(),...e.modelConfig};if(t?.thinkingTokenBudget&&!this.getFeatures(n).hasThinkingBudget)throw new Error(`Model ${n} does not support thinkingTokenBudget.`);if(t?.showThoughts&&!this.getFeatures(n).hasShowThoughts)throw new Error(`Model ${n} does not support showThoughts.`);if(this.modelInfo.find(a=>a.name===n)?.isExpensive&&t?.useExpensiveModel!=="yes")throw new Error(`Model ${n} is marked as expensive and requires explicit confirmation. Set useExpensiveModel: "yes" to proceed.`);return r.stream=(t?.stream!==void 0?t.stream:r.stream)??!0,this.getFeatures(n).streaming||(r.stream=!1),this.tracer?await this.tracer.startActiveSpan("AI Chat Request",{kind:he.SpanKind.SERVER,attributes:{[C.LLM_SYSTEM]:this.name,[C.LLM_OPERATION_NAME]:"chat",[C.LLM_REQUEST_MODEL]:n,[C.LLM_REQUEST_MAX_TOKENS]:r.maxTokens??"Not set",[C.LLM_REQUEST_TEMPERATURE]:r.temperature,[C.LLM_REQUEST_TOP_P]:r.topP??"Not set",[C.LLM_REQUEST_TOP_K]:r.topK??"Not set",[C.LLM_REQUEST_FREQUENCY_PENALTY]:r.frequencyPenalty??"Not set",[C.LLM_REQUEST_PRESENCE_PENALTY]:r.presencePenalty??"Not set",[C.LLM_REQUEST_STOP_SEQUENCES]:r.stopSequences?.join(", ")??"Not set",[C.LLM_REQUEST_LLM_IS_STREAMING]:r.stream??"Not set"}},t?.traceContext??he.context.active(),async a=>await this._chat2(n,r,e,t,a)):await this._chat2(n,r,e,t)}cleanupFunctionSchema(e){let t={...e};if(t.parameters){let n={...t.parameters};Array.isArray(n.required)&&n.required.length===0&&delete n.required,n.properties&&Object.keys(n.properties).length===0&&delete n.properties,Object.keys(n).length===0||Object.keys(n).length===1&&n.type==="object"?delete t.parameters:t.parameters=n}return t}async _chat2(e,t,n,r,o){if(!this.aiImpl.createChatReq)throw new Error("generateChatReq not implemented");let i=r?.debug??this.debug,a;n.functions&&n.functions.length>0&&(a=n.functions.map(m=>this.cleanupFunctionSchema(m)));let l={...n,model:e,functions:a,modelConfig:t};this.lastUsedChatModel=e,this.lastUsedModelConfig=t;let p=async()=>{let[m,h]=await this.aiImpl.createChatReq(l,r);return o?.isRecording()&&Ri(n,o,this.excludeContentFromTrace),await B({name:m.name,url:this.apiURL,localCall:m.localCall,headers:await this.buildHeaders(m.headers),stream:t.stream,timeout:this.timeout,debug:i,fetch:this.fetch,span:o,abortSignal:r?.abortSignal??this.abortSignal,corsProxy:this.corsProxy},h)};i&&io(l.chatPrompt,r?.stepIndex??0,r?.logger??this.logger,r?.debugHideSystemPrompt);let c=r?.rateLimiter??this.rt,u=c?await c(p,{modelUsage:this.modelUsage}):await p();if(t.stream){if(!this.aiImpl.createChatStreamResp)throw new Error("generateChatStreamResp not implemented");let m=this.aiImpl.createChatStreamResp.bind(this),h=x=>y=>{let w=m(y,x);if(w.sessionId=r?.sessionId,!w.modelUsage){let I=this.aiImpl.getTokenUsage();I&&(w.modelUsage={ai:this.name,model:e,tokens:I})}if(this.modelUsage=w.modelUsage,this.recordTokenUsage(w.modelUsage),o?.isRecording()&&_o(w,o,this.excludeContentFromTrace),i)for(let I of w.results)lo(I,I.index,r?.logger??this.logger);return w},g=async x=>{o?.isRecording()&&o.end(),i&&po(x,r?.logger??this.logger)};if(typeof window<"u"){let x=u,y={},w=[];return new ReadableStream({start(I){let R=x.getReader();async function k(){try{for(;;){let{done:v,value:H}=await R.read();if(v){g&&await g(w),I.close();break}let Y=h(y)(H);Y&&(w.push(Y),I.enqueue(Y))}}catch(v){I.error(v)}finally{R.releaseLock()}}k()}})}return u.pipeThrough(new Ft(h({}),g))}if(!this.aiImpl.createChatResp)throw new Error("generateChatResp not implemented");let d=this.aiImpl.createChatResp(u);if(d.sessionId=r?.sessionId,!d.modelUsage){let m=this.aiImpl.getTokenUsage();m&&(d.modelUsage={ai:this.name,model:e,tokens:m})}return d.modelUsage&&(this.modelUsage=d.modelUsage,this.recordTokenUsage(d.modelUsage)),o?.isRecording()&&(_o(d,o,this.excludeContentFromTrace),o.end()),i&&ao(d,r?.logger??this.logger),d}async embed(e,t){let n=performance.now(),r=!1,o;try{return o=await this._embed1(e,t),o}catch(i){throw r=!0,i instanceof Error&&(i.message.includes("timeout")||i.name==="TimeoutError"?this.recordTimeoutMetric("embed"):(i.message.includes("abort")||i.name==="AbortError")&&this.recordAbortMetric("embed")),i}finally{let i=performance.now()-n;this.updateLatencyMetrics("embed",i),this.updateErrorMetrics("embed",r),!r&&o&&this.recordEmbedMetrics(e,o)}}async _embed1(e,t){let n=this.getEmbedModel(e.embedModel)??e.embedModel??this.defaults.embedModel;if(!n)throw new Error("No embed model defined");return this.tracer&&await this.tracer?.startActiveSpan("AI Embed Request",{kind:he.SpanKind.SERVER,attributes:{[C.LLM_SYSTEM]:this.name,[C.LLM_OPERATION_NAME]:"embeddings",[C.LLM_REQUEST_MODEL]:n}},t?.traceContext??he.context.active(),async r=>{try{return await this._embed2(n,e,t,r)}finally{r.end()}}),this._embed2(n,e,t)}async _embed2(e,t,n,r){if(!this.aiImpl.createEmbedReq)throw new Error("generateEmbedReq not implemented");if(!this.aiImpl.createEmbedResp)throw new Error("generateEmbedResp not implemented");let o=this.aiImpl.createEmbedReq,i=n?.debug??this.debug,a={...t,embedModel:e};this.lastUsedEmbedModel=e,i&&Ao(a.texts??[],e,n?.logger??this.logger);let l=async()=>{let[u,d]=await o(a);return await B({name:u.name,url:this.apiURL,localCall:u.localCall,headers:await this.buildHeaders(u.headers),debug:i,fetch:this.fetch,timeout:this.timeout,span:r,abortSignal:n?.abortSignal??this.abortSignal,corsProxy:this.corsProxy},d)},p=this.rt?await this.rt(l,{modelUsage:this.embedModelUsage}):await l(),c=this.aiImpl.createEmbedResp?.(p);if(c.sessionId=n?.sessionId,!c.modelUsage){let u=this.aiImpl.getTokenUsage();u&&(c.modelUsage={ai:this.name,model:e,tokens:u})}return this.embedModelUsage=c.modelUsage,this.recordTokenUsage(c.modelUsage),r?.isRecording()&&c.modelUsage?.tokens&&r.addEvent(W.GEN_AI_USAGE,{[C.LLM_USAGE_INPUT_TOKENS]:c.modelUsage.tokens.promptTokens,[C.LLM_USAGE_OUTPUT_TOKENS]:c.modelUsage.tokens.completionTokens??0,[C.LLM_USAGE_TOTAL_TOKENS]:c.modelUsage.tokens.totalTokens}),i&&xo(c.embeddings,n?.logger??this.logger),r?.end(),c}async buildHeaders(e={}){return{...e,...await this.headers()}}getModelByKey(e){return e?this.models?.find(n=>n.key===e):void 0}getModel(e){let t=this.getModelByKey(e);return t&&"model"in t?t.model:void 0}getEmbedModel(e){let t=this.getModelByKey(e);return t&&"embedModel"in t?t.embedModel:void 0}};function Ri(s,e,t){let n=[];if(s.chatPrompt&&Array.isArray(s.chatPrompt)&&s.chatPrompt.length>0)for(let o of s.chatPrompt)switch(o.role){case"system":if(o.content){let i={};t||(i.content=o.content),e.addEvent(W.GEN_AI_SYSTEM_MESSAGE,i)}break;case"user":if(typeof o.content=="string")n.push(o.content);else if(Array.isArray(o.content))for(let i of o.content)i.type==="text"&&n.push(i.text);break;case"assistant":{let i=o.functionCalls?.map(a=>({id:a.id,type:a.type,function:a.function.name,arguments:a.function.params}));if(i&&i.length>0){let a={function_calls:JSON.stringify(i,null,2)};!t&&o.content&&(a.content=o.content),e.addEvent(W.GEN_AI_ASSISTANT_MESSAGE,a)}else if(o.content){let a={};t||(a.content=o.content),e.addEvent(W.GEN_AI_ASSISTANT_MESSAGE,a)}break}case"function":{let i={id:o.functionId};t||(i.content=o.result),e.addEvent(W.GEN_AI_TOOL_MESSAGE,i);break}}let r={};t||(r.content=n.join(`
110
+ `)),e.addEvent(W.GEN_AI_USER_MESSAGE,r)}function _o(s,e,t){if(s.modelUsage?.tokens){let n=s.modelUsage.tokens.thoughtsTokens?{[C.LLM_USAGE_THOUGHTS_TOKENS]:s.modelUsage.tokens.thoughtsTokens}:{};e.addEvent(W.GEN_AI_USAGE,{[C.LLM_USAGE_INPUT_TOKENS]:s.modelUsage.tokens.promptTokens,[C.LLM_USAGE_OUTPUT_TOKENS]:s.modelUsage.tokens.completionTokens??0,[C.LLM_USAGE_TOTAL_TOKENS]:s.modelUsage.tokens.totalTokens,...n})}if(s.results)for(let n=0;n<s.results.length;n++){let r=s.results[n];if(!r||!r.content&&!r.thought&&!r.functionCalls?.length&&!r.finishReason)continue;let o=r.functionCalls?.map(a=>({id:a.id,type:a.type,function:a.function.name,arguments:a.function.params})),i={};o&&o.length>0?(t||(i.content=r.content),i.tool_calls=o):t||(i.content=r.content??""),e.addEvent(W.GEN_AI_CHOICE,{finish_reason:r.finishReason,index:n,message:JSON.stringify(i,null,2)})}}function nr(s){for(let e=0;e<s.length;e++){let t=s[e];if(!t||typeof t!="object")throw new Error(`AxMessage array validation failed: Item at index ${e} is not a valid message object`);if("content"in t&&typeof t.content=="string"&&t.content.trim()==="")throw new Error(`AxMessage array validation failed: Item at index ${e} has empty content`)}}function Ti(s){let e=new Set;for(let t of s){if(e.has(t.key))throw new Error(`Duplicate model key detected: "${t.key}". Each model key must be unique.`);e.add(t.key)}}var Ze=(c=>(c.Claude4Opus="claude-opus-4-20250514",c.Claude4Sonnet="claude-sonnet-4-20250514",c.Claude37Sonnet="claude-3-7-sonnet-latest",c.Claude35Sonnet="claude-3-5-sonnet-latest",c.Claude35Haiku="claude-3-5-haiku-latest",c.Claude3Opus="claude-3-opus-latest",c.Claude3Sonnet="claude-3-sonnet-20240229",c.Claude3Haiku="claude-3-haiku-20240307",c.Claude21="claude-2.1",c.ClaudeInstant12="claude-instant-1.2",c))(Ze||{}),Lt=(i=>(i.Claude37Sonnet="claude-3-7-sonnet",i.Claude35Haiku="claude-3-5-haiku",i.Claude35Sonnet="claude-3-5-sonnet",i.Claude35SonnetV2="claude-3-5-sonnet-v2",i.Claude3Haiku="claude-3-haiku",i.Claude3Opus="claude-3-opus",i))(Lt||{});var et=[{name:"claude-opus-4-20250514",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:75,maxTokens:32e3,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"claude-sonnet-4-20250514",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:64e3,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"claude-3-7-sonnet-latest",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:64e3,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"claude-3-5-sonnet-latest",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:8192},{name:"claude-3-5-haiku-latest",currency:"usd",promptTokenCostPer1M:.8,completionTokenCostPer1M:4,maxTokens:8192},{name:"claude-3-opus-latest",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:75,maxTokens:4096},{name:"claude-3-sonnet-20240229",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15,maxTokens:4096},{name:"claude-3-haiku-20240307",currency:"usd",promptTokenCostPer1M:.25,completionTokenCostPer1M:1.25,maxTokens:4096},{name:"claude-2.1",currency:"usd",promptTokenCostPer1M:8,completionTokenCostPer1M:25,maxTokens:4096},{name:"claude-instant-1.2",currency:"usd",promptTokenCostPer1M:.8,completionTokenCostPer1M:2.24,maxTokens:4096}];var rr=s=>{if(!s||typeof s!="object")return s;let e={...s};return delete e.additionalProperties,delete e.default,delete e.optional,delete e.oneOf,delete e.anyOf,delete e.allOf,e.properties&&typeof e.properties=="object"&&(e.properties=Object.fromEntries(Object.entries(e.properties).map(([t,n])=>[t,rr(n)]))),e.items&&(e.items=rr(e.items)),e},sr=()=>structuredClone({model:"claude-3-7-sonnet-latest",maxTokens:4e4,thinkingTokenBudgetLevels:{minimal:1024,low:5e3,medium:1e4,high:2e4,highest:32e3},...S()}),Lo=()=>structuredClone({model:"claude-3-7-sonnet",maxTokens:4e4,thinkingTokenBudgetLevels:{minimal:1024,low:5e3,medium:1e4,high:2e4,highest:32e3},...S()}),or=class{constructor(e,t){this.config=e;this.isVertex=t}tokensUsed;currentPromptConfig;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens??4096,temperature:e.temperature,topP:e.topP,topK:e.topK,stream:e.stream,stopSequences:e.stopSequences,endSequences:e.endSequences,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,n:e.n}}createChatReq=async(e,t)=>{this.currentPromptConfig=t;let n=e.model,r=e.modelConfig?.stream??this.config.stream,o;this.isVertex?o={name:r?`/models/${n}:streamRawPredict?alt=sse`:`/models/${n}:rawPredict`}:o={name:"/messages"};let i;if(e.functionCall&&e.functions&&e.functions.length>0)if(typeof e.functionCall=="string")switch(e.functionCall){case"auto":i={tool_choice:{type:"auto"}};break;case"required":i={tool_choice:{type:"any"}};break;case"none":throw new Error("functionCall none not supported")}else if("function"in e.functionCall)i={tool_choice:{type:"tool",name:e.functionCall.function.name}};else throw new Error("Invalid function call type, must be string or object");let a=e.chatPrompt.filter(y=>y.role==="system").map(y=>({type:"text",text:y.content,...y.cache?{cache:{type:"ephemeral"}}:{}})),l=e.chatPrompt.filter(y=>y.role!=="system"),p=Ci(l),c=e.functions?.map(y=>({name:y.name,description:y.description,input_schema:y.parameters?rr(y.parameters):void 0})),u=e.modelConfig?.maxTokens??this.config.maxTokens,d=e.modelConfig?.stopSequences??this.config.stopSequences,m=e.modelConfig?.temperature??this.config.temperature,h=e.modelConfig?.topP??this.config.topP,g=e.modelConfig?.topK??this.config.topK,f=e.modelConfig?.n??this.config.n;if(f&&f>1)throw new Error("Anthropic does not support sampling (n > 1)");let A;if(this.config.thinking?.budget_tokens&&(A=this.config.thinking),t?.thinkingTokenBudget){let y=this.config.thinkingTokenBudgetLevels;switch(t.thinkingTokenBudget){case"none":A=void 0;break;case"minimal":A={type:"enabled",budget_tokens:y?.minimal??1024};break;case"low":A={type:"enabled",budget_tokens:y?.low??5e3};break;case"medium":A={type:"enabled",budget_tokens:y?.medium??1e4};break;case"high":A={type:"enabled",budget_tokens:y?.high??2e4};break;case"highest":A={type:"enabled",budget_tokens:y?.highest??32e3};break}}let x={...this.isVertex?{anthropic_version:"vertex-2023-10-16"}:{model:n},...u?{max_tokens:u}:{},...d&&d.length>0?{stop_sequences:d}:{},...m&&!A?{temperature:m}:{},...h&&(!A||h>=.95)?{top_p:h}:{},...g&&!A?{top_k:g}:{},...i,...c&&c.length>0?{tools:c}:{},...r?{stream:!0}:{},...a?{system:a}:{},...A?{thinking:A}:{},messages:p};return[o,x]};createChatResp=e=>{if(e.type==="error")throw new O(e.error.message,void 0,void 0);let t=Do(e.stop_reason),n=this.currentPromptConfig?.thinkingTokenBudget!=="none"&&this.currentPromptConfig?.showThoughts!==!1,r=e.content.map((o,i)=>o.type==="tool_use"?{index:i,id:o.id,functionCalls:[{id:o.id,type:"function",function:{name:o.name,params:o.input}}],finishReason:t}:(o.type==="thinking"||o.type==="redacted_thinking")&&n?{index:i,thought:o.thinking,id:e.id,finishReason:t}:{index:i,content:o.type==="text"?o.text:"",id:e.id,finishReason:t}).filter(o=>o.content!==""||o.thought!==void 0||o.functionCalls!==void 0);return this.tokensUsed={promptTokens:e.usage.input_tokens,completionTokens:e.usage.output_tokens,totalTokens:e.usage.input_tokens+e.usage.output_tokens},{results:r,remoteId:e.id}};createChatStreamResp=(e,t)=>{if(!("type"in e))throw new Error("Invalid Anthropic streaming event");let n=t;if(n.indexIdMap||(n.indexIdMap={}),e.type==="error"){let{error:o}=e;throw new O(o.message,void 0,void 0)}let r=0;if(e.type==="message_start"){let{message:o}=e,i=[{index:r,content:"",id:o.id}];return this.tokensUsed={promptTokens:o.usage?.input_tokens??0,completionTokens:o.usage?.output_tokens??0,totalTokens:(o.usage?.input_tokens??0)+(o.usage?.output_tokens??0)},{results:i}}if(e.type==="content_block_start"){let{content_block:o}=e;if(o.type==="text")return{results:[{index:r,content:o.text}]};if(o.type==="thinking")return this.currentPromptConfig?.thinkingTokenBudget!=="none"&&this.currentPromptConfig?.showThoughts!==!1?{results:[{index:r,thought:o.thinking}]}:{results:[{index:r,content:""}]};if(o.type==="tool_use"&&typeof o.id=="string"&&typeof e.index=="number"&&!n.indexIdMap[e.index]){n.indexIdMap[e.index]=o.id;let i=[{id:o.id,type:"function",function:{name:o.name,params:""}}];return{results:[{index:r,functionCalls:i}]}}}if(e.type==="content_block_delta"){let{delta:o}=e;if(o.type==="text_delta")return{results:[{index:r,content:o.text}]};if(o.type==="thinking_delta")return this.currentPromptConfig?.thinkingTokenBudget!=="none"&&this.currentPromptConfig?.showThoughts!==!1?{results:[{index:r,thought:o.thinking}]}:{results:[{index:r,content:""}]};if(o.type==="signature_delta")return{results:[{index:r,content:""}]};if(o.type==="input_json_delta"){let i=n.indexIdMap[e.index];if(!i)throw new Error(`invalid streaming index no id found: ${e.index}`);let a=[{id:i,type:"function",function:{name:"",params:o.partial_json}}];return{results:[{index:r,functionCalls:a}]}}}if(e.type==="message_delta"){let{delta:o,usage:i}=e;return this.tokensUsed={promptTokens:0,completionTokens:i.output_tokens,totalTokens:i.output_tokens},{results:[{index:r,content:"",finishReason:Do(o.stop_reason)}]}}return{results:[{index:r,content:""}]}}},Ae=class s extends E{static create(e){return new s(e)}constructor({apiKey:e,projectId:t,region:n,config:r,options:o,models:i}){let a=t!==void 0&&n!==void 0,l,p;if(a){if(!e)throw new Error("Anthropic Vertex API key not set");if(typeof e!="function")throw new Error("Anthropic Vertex API key must be a function for token-based authentication");l=`https://${n}-aiplatform.googleapis.com/v1/projects/${t}/locations/${n}/publishers/anthropic/`,p=async()=>({Authorization:`Bearer ${await e()}`})}else{if(!e)throw new Error("Anthropic API key not set");l="https://api.anthropic.com/v1",p=async()=>({"anthropic-version":"2023-06-01","anthropic-beta":"prompt-caching-2024-07-31","x-api-key":typeof e=="function"?await e():e})}let c={...sr(),...r},u=new or(c,a),d=m=>{let h=q({model:m,modelInfo:et,models:i});return{functions:!0,streaming:!0,hasThinkingBudget:h?.hasThinkingBudget??!1,hasShowThoughts:h?.hasShowThoughts??!1,functionCot:!0,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:5*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!1,formats:[],maxDuration:0},files:{supported:!1,formats:[],maxSize:0,uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!0,types:["ephemeral"]},thinking:h?.hasThinkingBudget??!1,multiTurn:!0}};super(u,{name:"Anthropic",apiURL:l,headers:p,modelInfo:et,defaults:{model:c.model},options:o,supportFor:d,models:i})}};function Ci(s){let e=s.map(t=>{switch(t.role){case"function":return{role:"user",content:[{type:"tool_result",content:t.result,tool_use_id:t.functionId,...t.isError?{is_error:!0}:{},...t.cache?{cache:{type:"ephemeral"}}:{}}]};case"user":return typeof t.content=="string"?{role:"user",content:t.content}:{role:"user",content:t.content.map(r=>{switch(r.type){case"text":return{type:"text",text:r.text,...r.cache?{cache:{type:"ephemeral"}}:{}};case"image":return{type:"image",source:{type:"base64",media_type:r.mimeType,data:r.image},...r.cache?{cache:{type:"ephemeral"}}:{}};default:throw new Error("Invalid content type")}})};case"assistant":{let n="";return typeof t.content=="string"&&(n=t.content),typeof t.functionCalls<"u"&&(n=t.functionCalls.map(r=>{let o={};return typeof r.function.params=="string"?o=JSON.parse(r.function.params):typeof r.function.params=="object"&&(o=r.function.params),{type:"tool_use",id:r.id,name:r.function.name,input:o,...t.cache?{cache:{type:"ephemeral"}}:{}}})),{role:"assistant",content:n}}default:throw new Error("Invalid role")}});return wi(e)}function wi(s){let e=[];for(let[t,n]of s.entries()){if(n.role!=="assistant"){e.push(n);continue}if(t>0&&s.at(t-1)?.role==="assistant"){let r=e.pop();e.push({...r||{},...n})}else e.push(n)}return e}function Do(s){if(s)switch(s){case"stop_sequence":return"stop";case"max_tokens":return"length";case"tool_use":return"function_call";case"end_turn":return"stop";default:return"stop"}}var tt=(A=>(A.GPT4="gpt-4",A.GPT41="gpt-4.1",A.GPT41Mini="gpt-4.1-mini",A.GPT4O="gpt-4o",A.GPT4OMini="gpt-4o-mini",A.GPT4ChatGPT4O="chatgpt-4o-latest",A.GPT4Turbo="gpt-4-turbo",A.GPT35Turbo="gpt-3.5-turbo",A.GPT35TurboInstruct="gpt-3.5-turbo-instruct",A.GPT35TextDavinci002="text-davinci-002",A.GPT3TextBabbage002="text-babbage-002",A.GPT3TextAda001="text-ada-001",A.O1="o1",A.O1Mini="o1-mini",A.O3="o3",A.O3Mini="o3-mini",A.O4Mini="o4-mini",A))(tt||{}),xe=(n=>(n.TextEmbeddingAda002="text-embedding-ada-002",n.TextEmbedding3Small="text-embedding-3-small",n.TextEmbedding3Large="text-embedding-3-large",n))(xe||{});var fe=(y=>(y.GPT4="gpt-4",y.GPT41="gpt-4.1",y.GPT41Mini="gpt-4.1-mini",y.GPT4O="gpt-4o",y.GPT4OMini="gpt-4o-mini",y.GPT4ChatGPT4O="chatgpt-4o-latest",y.GPT4Turbo="gpt-4-turbo",y.GPT35Turbo="gpt-3.5-turbo",y.GPT35TurboInstruct="gpt-3.5-turbo-instruct",y.GPT35TextDavinci002="text-davinci-002",y.GPT3TextBabbage002="text-babbage-002",y.GPT3TextAda001="text-ada-001",y.O1Pro="o1-pro",y.O1="o1",y.O1Mini="o1-mini",y.O3Pro="o3-pro",y.O3="o3",y.O3Mini="o3-mini",y.O4Mini="o4-mini",y))(fe||{});var ye=[{name:"gpt-4",currency:"usd",promptTokenCostPer1M:30,completionTokenCostPer1M:60},{name:"gpt-4.1",currency:"usd",promptTokenCostPer1M:2,completionTokenCostPer1M:8},{name:"gpt-4.1-mini",currency:"usd",promptTokenCostPer1M:.4,completionTokenCostPer1M:1.6},{name:"gpt-4o",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4o-mini",currency:"usd",promptTokenCostPer1M:.15,completionTokenCostPer1M:.6},{name:"chatgpt-4o-latest",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4-turbo",currency:"usd",promptTokenCostPer1M:10,completionTokenCostPer1M:30},{name:"gpt-3.5-turbo",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"o1",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60},{name:"o1-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:14.4},{name:"o3",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60},{name:"o3-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4},{name:"o4-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4},{name:"text-embedding-ada-002",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"text-embedding-3-small",currency:"usd",promptTokenCostPer1M:.02,completionTokenCostPer1M:.02},{name:"text-embedding-3-large",currency:"usd",promptTokenCostPer1M:.13,completionTokenCostPer1M:.13}],Gt=[{name:"gpt-4",currency:"usd",promptTokenCostPer1M:30,completionTokenCostPer1M:60},{name:"gpt-4.1",currency:"usd",promptTokenCostPer1M:2,completionTokenCostPer1M:8},{name:"gpt-4.1-mini",currency:"usd",promptTokenCostPer1M:.4,completionTokenCostPer1M:1.6},{name:"gpt-4o",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4o-mini",currency:"usd",promptTokenCostPer1M:.15,completionTokenCostPer1M:.6},{name:"chatgpt-4o-latest",currency:"usd",promptTokenCostPer1M:5,completionTokenCostPer1M:15},{name:"gpt-4-turbo",currency:"usd",promptTokenCostPer1M:10,completionTokenCostPer1M:30},{name:"gpt-3.5-turbo",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"o1-pro",currency:"usd",promptTokenCostPer1M:150,completionTokenCostPer1M:600,hasThinkingBudget:!0,hasShowThoughts:!0,isExpensive:!0},{name:"o1",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"o3-pro",currency:"usd",promptTokenCostPer1M:20,completionTokenCostPer1M:80,hasThinkingBudget:!0,hasShowThoughts:!0,isExpensive:!0},{name:"o3",currency:"usd",promptTokenCostPer1M:15,completionTokenCostPer1M:60,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"o3-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"o4-mini",currency:"usd",promptTokenCostPer1M:1.1,completionTokenCostPer1M:4.4,hasThinkingBudget:!0,hasShowThoughts:!0}];var Si=s=>{let e=["o1","o1-mini","o3","o3-mini","o4-mini","o1-pro","o3-pro"];return e.includes(s)||e.includes(s)},ce=()=>structuredClone({model:"gpt-4.1",embedModel:"text-embedding-3-small",...S()}),$t=()=>structuredClone({...ce(),model:"gpt-4.1"}),Nt=()=>structuredClone({model:"gpt-4.1",embedModel:"text-embedding-3-small",..._()}),Ut=()=>({...ce(),model:"gpt-4.1-mini"}),ir=class{constructor(e,t,n){this.config=e;this.streamingUsage=t;this.chatReqUpdater=n}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,endSequences:e.endSequences,topP:e.topP,n:e.n,stream:e.stream}}createChatReq(e,t){let n=e.model;if(!e.chatPrompt||e.chatPrompt.length===0)throw new Error("Chat prompt is empty");let r={name:"/chat/completions"},o=e.functions?.map(m=>({type:"function",function:{name:m.name,description:m.description,parameters:m.parameters}})),i=!e.functionCall&&e.functions&&e.functions.length>0?"auto":e.functionCall,a=vi(e),l=e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,p=e.modelConfig?.stream??this.config.stream,c=this.config.store,u=Si(n),d={model:n,messages:a,response_format:this.config?.responseFormat?{type:this.config.responseFormat}:void 0,tools:o,tool_choice:i,...u?{}:{max_completion_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,top_p:e.modelConfig?.topP??this.config.topP??1,n:e.modelConfig?.n??this.config.n,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,...l?{frequency_penalty:l}:{}},stop:e.modelConfig?.stopSequences??this.config.stop,logit_bias:this.config.logitBias,...p&&this.streamingUsage?{stream:!0,stream_options:{include_usage:!0}}:{},...c?{store:c}:{},...this.config.serviceTier?{service_tier:this.config.serviceTier}:{},...this.config.user?{user:this.config.user}:{}};if(this.config.reasoningEffort&&(d.reasoning_effort=this.config.reasoningEffort),this.config.webSearchOptions&&(d.web_search_options={...this.config.webSearchOptions.searchContextSize&&{search_context_size:this.config.webSearchOptions.searchContextSize},...this.config.webSearchOptions.userLocation&&{user_location:{approximate:{type:"approximate",...this.config.webSearchOptions.userLocation.approximate.city&&{city:this.config.webSearchOptions.userLocation.approximate.city},...this.config.webSearchOptions.userLocation.approximate.country&&{country:this.config.webSearchOptions.userLocation.approximate.country},...this.config.webSearchOptions.userLocation.approximate.region&&{region:this.config.webSearchOptions.userLocation.approximate.region},...this.config.webSearchOptions.userLocation.approximate.timezone&&{timezone:this.config.webSearchOptions.userLocation.approximate.timezone}}}}}),t?.thinkingTokenBudget)switch(t.thinkingTokenBudget){case"none":d.reasoning_effort=void 0;break;case"minimal":d.reasoning_effort="low";break;case"low":d.reasoning_effort="medium";break;case"medium":d.reasoning_effort="high";break;case"high":d.reasoning_effort="high";break;case"highest":d.reasoning_effort="high";break}return this.chatReqUpdater&&(d=this.chatReqUpdater(d)),[r,d]}createEmbedReq(e){let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n={name:"/embeddings"},r={model:t,input:e.texts,dimensions:this.config.dimensions};return[n,r]}createChatResp(e){let{id:t,usage:n,choices:r,error:o}=e;if(o)throw o;return this.tokensUsed=n?{promptTokens:n.prompt_tokens,completionTokens:n.completion_tokens,totalTokens:n.total_tokens}:void 0,{results:r.map(a=>{if(a.message.refusal)throw new O(a.message.refusal,e.model,e.id);let l=Go(a.finish_reason),p=a.message.tool_calls?.map(({id:c,function:{arguments:u,name:d}})=>({id:c,type:"function",function:{name:d,params:u}}));return{index:a.index,id:`${a.index}`,content:a.message.content??void 0,thought:a.message.reasoning_content,annotations:a.message.annotations,functionCalls:p,finishReason:l}}),remoteId:t}}createChatStreamResp(e,t){let{id:n,usage:r,choices:o}=e;this.tokensUsed=r?{promptTokens:r.prompt_tokens,completionTokens:r.completion_tokens,totalTokens:r.total_tokens}:void 0;let i=t;return i.indexIdMap||(i.indexIdMap={}),{results:o.map(({index:l,delta:{content:p,role:c,refusal:u,tool_calls:d,reasoning_content:m,annotations:h},finish_reason:g})=>{if(u)throw new O(u,void 0,n);let f=Go(g),A=d?.map(({id:x,index:y,function:{name:w,arguments:I}})=>{typeof x=="string"&&typeof y=="number"&&!i.indexIdMap[y]&&(i.indexIdMap[y]=x);let R=i.indexIdMap[y];return R?{id:R,type:"function",function:{name:w,params:I}}:null}).filter(x=>x!==null);return{index:l,content:p??void 0,role:c,thought:m,annotations:h,functionCalls:A,finishReason:f,id:n}})}}createEmbedResp(e){let{data:t,usage:n}=e;return this.tokensUsed=n?{promptTokens:n.prompt_tokens,completionTokens:n.completion_tokens,totalTokens:n.total_tokens}:void 0,{embeddings:t.map(r=>r.embedding)}}},Go=s=>{switch(s){case"stop":return"stop";case"length":return"length";case"content_filter":return"error";case"tool_calls":return"function_call"}};function vi(s){return s.chatPrompt.map(t=>{switch(t.role){case"system":return{role:"system",content:t.content};case"user":{let n=Array.isArray(t.content)?t.content.map(r=>{switch(r.type){case"text":return{type:"text",text:r.text};case"image":return{type:"image_url",image_url:{url:`data:${r.mimeType};base64,${r.image}`,details:r.details??"auto"}};case"audio":return{type:"input_audio",input_audio:{data:r.data,format:r.format==="wav"?"wav":void 0}};default:throw new Error("Invalid content type")}}):t.content;return{role:"user",...t.name?{name:t.name}:{},content:n}}case"assistant":{let n=t.functionCalls?.map(r=>({id:r.id,type:"function",function:{name:r.function.name,arguments:typeof r.function.params=="object"?JSON.stringify(r.function.params):r.function.params}}));if(n&&n.length>0)return{role:"assistant",...t.content?{content:t.content}:{},name:t.name,tool_calls:n};if(t.content===void 0)throw new Error("Assistant content is required when no tool calls are provided");return{role:"assistant",content:t.content,...t.name?{name:t.name}:{}}}case"function":return{role:"tool",content:t.result,tool_call_id:t.functionId};default:throw new Error("Invalid role")}})}var F=class extends E{constructor({apiKey:e,config:t,options:n,apiURL:r,modelInfo:o,models:i,chatReqUpdater:a,supportFor:l}){if(!e||e==="")throw new Error("OpenAI API key not set");let p=new ir(t,n?.streamingUsage??!0,a);super(p,{name:"OpenAI",apiURL:r||"https://api.openai.com/v1",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:o,defaults:{model:t.model,embedModel:t.embedModel},options:n,supportFor:l,models:i})}},Ie=class extends F{constructor({apiKey:e,apiURL:t,config:n,options:r,models:o,modelInfo:i}){if(!e||e==="")throw new Error("OpenAI API key not set");i=[...ye,...i??[]];let a=l=>{let p=q({model:l,modelInfo:i,models:o});return{functions:!0,streaming:!0,hasThinkingBudget:p?.hasThinkingBudget??!1,hasShowThoughts:p?.hasShowThoughts??!1,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:20*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!0,formats:["wav","mp3","ogg"],maxDuration:25*60},files:{supported:!0,formats:["text/plain","application/pdf","image/jpeg","image/png"],maxSize:512*1024*1024,uploadMethod:"upload"},urls:{supported:!1,webSearch:!0,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:p?.hasThinkingBudget??!1,multiTurn:!0}};super({apiKey:e,apiURL:t,config:{...ce(),...n},options:r,modelInfo:i,models:o,supportFor:a}),super.setName("OpenAI")}};var ar=ce,$o=Nt,No=Ut,Uo=$t,be=class extends F{constructor({apiKey:e,resourceName:t,deploymentName:n,version:r="api-version=2024-02-15-preview",config:o,options:i,models:a,modelInfo:l}){if(!e||e==="")throw new Error("Azure OpenAPI API key not set");if(!t||t==="")throw new Error("Azure OpenAPI resource name not set");if(!n||n==="")throw new Error("Azure OpenAPI deployment id not set");let p={...ar(),...o};l=[...ye,...l??[]];let c=d=>{let m=q({model:d,modelInfo:l,models:a});return{functions:!0,streaming:!0,hasThinkingBudget:m?.hasThinkingBudget??!1,hasShowThoughts:m?.hasShowThoughts??!1,functionCot:!1,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:20*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!1,formats:[],maxDuration:0},files:{supported:!1,formats:[],maxSize:0,uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:m?.hasThinkingBudget??!1,multiTurn:!0}};super({apiKey:e,config:p,options:i,models:a,modelInfo:l,supportFor:c});let u=t.includes("://")?t:`https://${t}.openai.azure.com/`;super.setName("Azure OpenAI"),super.setAPIURL(new URL(`/openai/deployments/${n}?api-version=${r}`,u).href),super.setHeaders(async()=>({"api-key":e}))}};var Bt=class s{services;currentServiceIndex=0;currentService;debug;initialBackoffMs;maxBackoffMs;maxRetries;serviceFailures=new Map;constructor(e,t){if(e.length===0)throw new Error("No AI services provided.");Oi(e),this.services=[...e].sort(t?.comparator??s.metricComparator);let n=this.services[this.currentServiceIndex];if(n===void 0)throw new Error("Error initializing the AI services.");this.currentService=n,this.debug=t?.debug??!0,this.initialBackoffMs=t?.initialBackoffMs??1e3,this.maxBackoffMs=t?.maxBackoffMs??32e3,this.maxRetries=t?.maxRetries??3}static create(e,t){return new s(e,t)}getLastUsedChatModel(){return this.currentService.getLastUsedChatModel()}getLastUsedEmbedModel(){return this.currentService.getLastUsedEmbedModel()}getLastUsedModelConfig(){return this.currentService.getLastUsedModelConfig()}static inputOrderComparator=()=>0;static metricComparator=(e,t)=>{let n=e.getMetrics(),r=t.getMetrics();return n.latency.chat.mean-r.latency.chat.mean};getModelList(){return this.currentService.getModelList()}getNextService(){let e=this.services[++this.currentServiceIndex];return e===void 0?!1:(this.currentService=e,!0)}reset(){this.currentServiceIndex=0;let e=this.services[this.currentServiceIndex];if(e===void 0)throw new Error("No AI services provided.");this.currentService=e}getName(){return this.currentService.getName()}getId(){return this.currentService.getId()}getFeatures(e){return this.currentService.getFeatures(e)}getMetrics(){return this.currentService.getMetrics()}canRetryService(){let e=this.serviceFailures.get(this.currentService.getId());if(!e)return!0;let{retries:t,lastFailureTime:n}=e,r=Date.now()-n,o=Math.min(this.initialBackoffMs*2**t,this.maxBackoffMs);return r>=o}handleFailure(){let t=(this.serviceFailures.get(this.currentService.getId())?.retries??0)+1;if(this.serviceFailures.set(this.currentService.getId(),{retries:t,lastFailureTime:Date.now()}),this.debug&&console.warn(`AxBalancer: Service ${this.currentService.getName()} failed (retry ${t}/${this.maxRetries})`),t>=this.maxRetries){let n=this.getNextService();return this.debug&&console.warn(`AxBalancer: Switching to service ${this.currentService.getName()}`),n}return!0}handleSuccess(){this.serviceFailures.delete(this.currentService.getId())}async chat(e,t){for(this.reset();;){if(!this.canRetryService()){if(!this.getNextService())throw new Error("All services exhausted");continue}try{let n=await this.currentService.chat(e,t);return this.handleSuccess(),n}catch(n){if(!(n instanceof U))throw n;switch(n.constructor){case ne:throw n;case le:break;case Z:break;case te:break;case K:break;case pe:break;default:throw n}if(!this.handleFailure())throw n}}}async embed(e,t){for(this.reset();;){if(!this.canRetryService()){if(!this.getNextService())throw new Error("All services exhausted");continue}try{let n=await this.currentService.embed(e,t);return this.handleSuccess(),n}catch(n){if(!this.handleFailure())throw n}}}setOptions(e){this.currentService.setOptions(e)}getOptions(){return this.currentService.getOptions()}getLogger(){return this.currentService.getLogger()}};function Oi(s){let e=s.find(r=>r.getModelList()!==void 0);if(!e)return;let t=e.getModelList();if(!t)throw new Error("No model list found in any service.");let n=new Set(t.map(r=>r.key));for(let r=0;r<s.length;r++){let o=s[r];if(!o)throw new Error(`Service at index ${r} is undefined`);let i=o.getModelList();if(!i)throw new Error(`Service at index ${r} (${o.getName()}) has no model list while another service does.`);let a=new Set(i.map(l=>l.key));for(let l of n)if(!a.has(l))throw new Error(`Service at index ${r} (${o.getName()}) is missing model "${l}"`);for(let l of a)if(!n.has(l))throw new Error(`Service at index ${r} (${o.getName()}) has extra model "${l}"`)}}function ue(s){let e=!1,t=!1,n=!1,r=!1,o=!1,i=!1,a=!1,l=new Set,p=0;if(s.chatPrompt&&Array.isArray(s.chatPrompt))for(let c of s.chatPrompt){if(c.role==="user"&&Array.isArray(c.content))for(let u of c.content)switch(l.add(u.type),u.type){case"image":e=!0,u.cache&&(a=!0),p+=85;break;case"audio":t=!0,u.cache&&(a=!0),p+=u.duration||60;break;case"file":n=!0,u.cache&&(a=!0),p+=Math.ceil((u.extractedText?.length||1e3)/4);break;case"url":r=!0,u.cache&&(a=!0),p+=Math.ceil((u.cachedContent?.length||2e3)/4);break;case"text":u.cache&&(a=!0),p+=Math.ceil(u.text.length/4);break}else"content"in c&&typeof c.content=="string"&&(p+=Math.ceil(c.content.length/4));"cache"in c&&c.cache&&(a=!0)}return s.functions&&s.functions.length>0&&(o=!0),s.modelConfig?.stream===!0&&(i=!0),s.capabilities&&(s.capabilities.requiresImages&&(e=!0),s.capabilities.requiresAudio&&(t=!0),s.capabilities.requiresFiles&&(n=!0),s.capabilities.requiresWebSearch&&(r=!0)),{hasImages:e,hasAudio:t,hasFiles:n,hasUrls:r,requiresFunctions:o,requiresStreaming:i,requiresCaching:a,contentTypes:l,estimatedTokens:p}}function lr(s,e){let t=s.getFeatures(),n=[],r=[],o=[];if(e.hasImages&&!t.media.images.supported&&(n.push("Image support"),o.push("Use altText for images or imageToText service")),e.hasAudio&&!t.media.audio.supported&&(n.push("Audio support"),o.push("Pre-transcribe audio or use transcription field")),e.hasFiles&&!t.media.files.supported&&(n.push("File support"),o.push("Pre-extract text content or use extractedText field")),e.hasUrls&&!t.media.urls.supported&&(n.push("URL/Web search support"),o.push("Pre-fetch content or use cachedContent field")),e.requiresFunctions&&!t.functions&&n.push("Function calling"),e.requiresStreaming&&!t.streaming&&(n.push("Streaming responses"),o.push("Use non-streaming mode")),e.requiresCaching&&!t.caching.supported&&(n.push("Content caching"),o.push("Repeated content will not be cached")),e.hasImages&&t.media.images.supported){let a=t.media.images.maxSize;a&&a<10*1024*1024&&r.push(`Image size limit is ${Math.round(a/(1024*1024))}MB`)}if(e.hasAudio&&t.media.audio.supported){let a=t.media.audio.maxDuration;a&&a<600&&r.push(`Audio duration limit is ${Math.round(a/60)} minutes`)}return{isSupported:n.length===0,missingCapabilities:n,warnings:r,alternatives:o}}function qt(s,e){return s.map(t=>{let n=t.getFeatures(),r=lr(t,e),o=0,i=[];return o+=10,e.hasImages&&n.media.images.supported&&(o+=25,i.push("Images"),n.media.images.detailLevels?.includes("high")&&(o+=5),n.media.images.maxSize&&n.media.images.maxSize>10*1024*1024&&(o+=3)),e.hasAudio&&n.media.audio.supported&&(o+=25,i.push("Audio"),n.media.audio.maxDuration&&n.media.audio.maxDuration>600&&(o+=5)),e.hasFiles&&n.media.files.supported&&(o+=25,i.push("Files"),n.media.files.uploadMethod==="cloud"&&(o+=3)),e.hasUrls&&n.media.urls.supported&&(o+=25,i.push("URLs"),n.media.urls.webSearch&&(o+=5)),e.requiresFunctions&&n.functions&&(o+=15,i.push("Functions"),n.functionCot&&(o+=3)),e.requiresStreaming&&n.streaming&&(o+=10,i.push("Streaming")),e.requiresCaching&&n.caching.supported&&(o+=8,i.push("Caching"),n.caching.types.includes("persistent")&&(o+=3)),n.thinking&&(o+=2),n.multiTurn&&(o+=2),n.hasThinkingBudget&&(o+=1),n.hasShowThoughts&&(o+=1),o-=r.missingCapabilities.length*10,{provider:t,score:o,missingCapabilities:r.missingCapabilities,supportedCapabilities:i}}).sort((t,n)=>n.score-t.score)}function zt(s,e,t={}){if(e.length===0)throw new Error("No providers available");let n=ue(s),r=qt(e,n);if(t.requireExactMatch){let o=r.filter(i=>i.missingCapabilities.length===0);if(o.length===0)throw new Error(`No providers fully support the request requirements: ${r[0]?.missingCapabilities.join(", ")||"unknown requirements"}`);return o[0].provider}if(!t.allowDegradation){let o=r[0];if(o.missingCapabilities.length>0)throw new Error(`Best available provider (${o.provider.getName()}) is missing: ${o.missingCapabilities.join(", ")}`)}return r[0].provider}function Bo(s,e){let t=ue(s),n=qt(e,t),r=n[0]?.provider||null,o=[t.hasImages&&"images",t.hasAudio&&"audio",t.hasFiles&&"files",t.hasUrls&&"URLs",t.requiresFunctions&&"functions",t.requiresStreaming&&"streaming",t.requiresCaching&&"caching"].filter(Boolean).length,i=r?n[0].supportedCapabilities.length:0,a=r?`${r.getName()} supports ${i}/${o} requirements (${Math.round(i/Math.max(o,1)*100)}% compatibility)`:"No suitable providers found";return{requirements:t,providerScores:n,recommendedProvider:r,summary:a}}function qo(s,e){return s.filter(t=>t.getFeatures().media[e].supported)}function zo(s,e){let t={};for(let n of s){let o=n.getFeatures().media[e];if(o.supported)for(let i of o.formats)t[i]||(t[i]=[]),t[i].push(n)}return t}var nt=(r=>(r.CommandRPlus="command-r-plus",r.CommandR="command-r",r.Command="command",r.CommandLight="command-light",r))(nt||{}),rt=(r=>(r.EmbedEnglishV30="embed-english-v3.0",r.EmbedEnglishLightV30="embed-english-light-v3.0",r.EmbedMultiLingualV30="embed-multilingual-v3.0",r.EmbedMultiLingualLightV30="embed-multilingual-light-v3.0",r))(rt||{});var jt=[{name:"command-r-plus",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15},{name:"command-r",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"command",currency:"usd",promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5},{name:"command-light",currency:"usd",promptTokenCostPer1M:.3,completionTokenCostPer1M:.6},{name:"embed-english-light-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"embed-english-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"embed-multilingual-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1},{name:"embed-multilingual-light-v3.0",currency:"usd",promptTokenCostPer1M:.1,completionTokenCostPer1M:.1}];var cr=()=>structuredClone({model:"command-r-plus",embedModel:"embed-english-v3.0",...S()}),Ho=()=>structuredClone({model:"command-r",embedModel:"embed-english-v3.0",..._()}),pr=class{constructor(e){this.config=e}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,frequencyPenalty:e.frequencyPenalty,presencePenalty:e.presencePenalty,endSequences:e.endSequences,stopSequences:e.stopSequences,stream:e.stream,n:e.n}}createChatReq(e){let t=e.model,n=e.chatPrompt.at(-1),r=e.chatPrompt.slice(0,-1),o;n&&n.role==="user"&&typeof n.content=="string"&&(o=n?.content);let i=Mi(r),a=e.functions?.map(u=>{let d={};if(u.parameters?.properties)for(let[m,h]of Object.entries(u.parameters.properties))d[m]={description:h.description,type:h.type,required:u.parameters.required?.includes(m)??!1};return{name:u.name,description:u.description,parameter_definitions:d}}),l=e.chatPrompt.filter(u=>u.role==="function").map(u=>{let d=a?.find(m=>m.name===u.functionId);if(!d)throw new Error("Function not found");return{call:{name:d.name,parameters:d.parameter_definitions},outputs:[{result:u.result??""}]}}),p={name:"/chat"},c={message:o,model:t,tools:a,...l&&!o?{tool_results:l}:{},chat_history:i,max_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,k:e.modelConfig?.topK??this.config.topK,p:e.modelConfig?.topP??this.config.topP,frequency_penalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,end_sequences:this.config.endSequences,stop_sequences:e.modelConfig?.stopSequences??this.config.stopSequences};return[p,c]}createEmbedReq=e=>{let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n={name:"/embed"},r={model:t,texts:e.texts??[],input_type:"classification",truncate:""};return[n,r]};createChatResp=e=>{this.tokensUsed=e.meta.billed_units?{promptTokens:e.meta.billed_units.input_tokens,completionTokens:e.meta.billed_units.output_tokens,totalTokens:e.meta.billed_units.input_tokens+e.meta.billed_units.output_tokens}:void 0;let t;if("finish_reason"in e)switch(e.finish_reason){case"COMPLETE":t="stop";break;case"MAX_TOKENS":t="length";break;case"ERROR":throw new Error("Finish reason: ERROR");case"ERROR_TOXIC":throw new Error("Finish reason: CONTENT_FILTER");default:t="stop";break}let n;return"tool_calls"in e&&(n=e.tool_calls?.map(o=>({id:o.name,type:"function",function:{name:o.name,params:o.parameters}}))),{results:[{index:0,id:e.generation_id,content:e.text,functionCalls:n,finishReason:t}],remoteId:e.response_id}};createChatStreamResp=(e,t)=>{let n=t;e.event_type==="stream-start"&&(n.generation_id=e.generation_id),this.tokensUsed={promptTokens:0,completionTokens:e.meta.billed_units?.output_tokens??0,totalTokens:e.meta.billed_units?.output_tokens??0};let{results:r}=this.createChatResp(e),o=r[0];if(!o)throw new Error("No result");return o.id=n.generation_id??"",{results:r}};createEmbedResp(e){return{remoteId:e.id,embeddings:e.embeddings}}},Re=class extends E{constructor({apiKey:e,config:t,options:n,models:r}){if(!e||e==="")throw new Error("Cohere API key not set");let o={...cr(),...t},i=new pr(o);super(i,{name:"Cohere",apiURL:"https://api.cohere.ai/v1",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:jt,defaults:{model:o.model},supportFor:{functions:!0,streaming:!0,media:{images:{supported:!1,formats:[],maxSize:0,detailLevels:[]},audio:{supported:!1,formats:[],maxDuration:0},files:{supported:!1,formats:[],maxSize:0,uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},options:n,models:r})}};function Mi(s){return s.map(e=>{let t="";if(e.role==="system"||e.role==="assistant"||e.role==="user")if(typeof e.content=="string")t=e.content;else throw new Error("Multi-modal content not supported");switch(e.role){case"user":return{role:"USER",message:t};case"system":return{role:"SYSTEM",message:t};case"assistant":{let n=jo(e.functionCalls);return{role:"CHATBOT",message:t,tool_calls:n}}case"function":{let n=s.map(i=>{if(i.role==="assistant")return i.functionCalls?.find(a=>a.id===e.functionId)}).filter(i=>i!==void 0),r=jo(n)?.at(0);if(!r)throw new Error("Function call not found");let o=[{result:e.result}];return{role:"TOOL",tool_results:[{call:r,outputs:o}]}}default:throw new Error("Unknown role")}})}function jo(s){return s?.map(e=>{let t=typeof e.function.params=="string"?JSON.parse(e.function.params):e.function.params;return{name:e.function.name,parameters:t}})}var ot=(n=>(n.DeepSeekChat="deepseek-chat",n.DeepSeekCoder="deepseek-coder",n.DeepSeekReasoner="deepseek-reasoner",n))(ot||{});var Ht=[{name:"deepseek-chat",currency:"USD",promptTokenCostPer1M:.27,completionTokenCostPer1M:1.1},{name:"deepseek-reasoner",currency:"USD",promptTokenCostPer1M:.55,completionTokenCostPer1M:2.19}];var ur=()=>structuredClone({model:"deepseek-chat",...S()}),Ko=()=>structuredClone({model:"deepseek-coder",..._()}),Te=class extends F{constructor({apiKey:e,config:t,options:n,models:r,modelInfo:o}){if(!e||e==="")throw new Error("DeepSeek API key not set");let i={...ur(),...t};o=[...Ht,...o??[]],super({apiKey:e,config:i,options:n,apiURL:"https://api.deepseek.com",modelInfo:o,supportFor:{functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},models:r}),super.setName("DeepSeek")}};var st=(c=>(c.Gemini25Pro="gemini-2.5-pro",c.Gemini25Flash="gemini-2.5-flash",c.Gemini25FlashLite="gemini-2.5-flash-lite",c.Gemini20Flash="gemini-2.0-flash",c.Gemini20FlashLite="gemini-2.0-flash-lite",c.Gemini1Pro="gemini-1.0-pro",c.Gemini15Flash="gemini-1.5-flash",c.Gemini15Flash002="gemini-1.5-flash-002",c.Gemini15Flash8B="gemini-1.5-flash-8b",c.Gemini15Pro="gemini-1.5-pro",c))(st||{}),Kt=(r=>(r.GeminiEmbedding="gemini-embedding-exp",r.TextEmbeddingLarge="text-embedding-large-exp-03-07",r.TextEmbedding004="text-embedding-004",r.TextEmbedding005="text-embedding-005",r))(Kt||{}),Vt=(r=>(r.HarmCategoryHarassment="HARM_CATEGORY_HARASSMENT",r.HarmCategoryHateSpeech="HARM_CATEGORY_HATE_SPEECH",r.HarmCategorySexuallyExplicit="HARM_CATEGORY_SEXUALLY_EXPLICIT",r.HarmCategoryDangerousContent="HARM_CATEGORY_DANGEROUS_CONTENT",r))(Vt||{}),Wt=(o=>(o.BlockNone="BLOCK_NONE",o.BlockOnlyHigh="BLOCK_ONLY_HIGH",o.BlockMediumAndAbove="BLOCK_MEDIUM_AND_ABOVE",o.BlockLowAndAbove="BLOCK_LOW_AND_ABOVE",o.BlockDefault="HARM_BLOCK_THRESHOLD_UNSPECIFIED",o))(Wt||{}),dr=(l=>(l.SemanticSimilarity="SEMANTIC_SIMILARITY",l.Classification="CLASSIFICATION",l.Clustering="CLUSTERING",l.RetrievalDocument="RETRIEVAL_DOCUMENT",l.RetrievalQuery="RETRIEVAL_QUERY",l.QuestionAnswering="QUESTION_ANSWERING",l.FactVerification="FACT_VERIFICATION",l.CodeRetrievalQuery="CODE_RETRIEVAL_QUERY",l))(dr||{});var Jt=[{name:"gemini-2.5-pro",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:2.5,completionTokenCostPer1M:15,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"gemini-2.5-flash",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:15,completionTokenCostPer1M:3.5,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"gemini-2.5-flash-lite",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.1,completionTokenCostPer1M:.4,hasThinkingBudget:!0,hasShowThoughts:!0},{name:"gemini-2.0-flash",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.01,completionTokenCostPer1M:.4},{name:"gemini-2.0-flash-lite",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:0,completionTokenCostPer1M:0},{name:"gemini-1.5-flash",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.075,completionTokenCostPer1M:.3},{name:"gemini-1.5-flash-8b",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.0375,completionTokenCostPer1M:.15},{name:"gemini-1.5-pro",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:1.25,completionTokenCostPer1M:5},{name:"gemini-1.0-pro",currency:"usd",characterIsToken:!1,promptTokenCostPer1M:.5,completionTokenCostPer1M:1.5}];var mr=s=>{if(!s||typeof s!="object")return s;let e={...s};return delete e.additionalProperties,delete e.default,delete e.optional,delete e.maximum,delete e.oneOf,delete e.anyOf,e.properties&&typeof e.properties=="object"&&(e.properties=Object.fromEntries(Object.entries(e.properties).map(([t,n])=>[t,mr(n)]))),e.items&&(e.items=mr(e.items)),e},Vo=[{category:"HARM_CATEGORY_HARASSMENT",threshold:"BLOCK_NONE"},{category:"HARM_CATEGORY_HATE_SPEECH",threshold:"BLOCK_NONE"},{category:"HARM_CATEGORY_SEXUALLY_EXPLICIT",threshold:"BLOCK_NONE"},{category:"HARM_CATEGORY_DANGEROUS_CONTENT",threshold:"BLOCK_NONE"}],hr=()=>structuredClone({model:"gemini-2.5-flash",embedModel:"text-embedding-005",safetySettings:Vo,thinkingTokenBudgetLevels:{minimal:200,low:800,medium:5e3,high:1e4,highest:24500},...S()}),Wo=()=>structuredClone({model:"gemini-2.0-flash",embedModel:"text-embedding-005",safetySettings:Vo,thinkingTokenBudgetLevels:{minimal:200,low:800,medium:5e3,high:1e4,highest:24500},..._()}),gr=class{constructor(e,t,n,r,o){this.config=e;this.isVertex=t;this.endpointId=n;this.apiKey=r;this.options=o;if(!this.isVertex&&this.config.autoTruncate)throw new Error("Auto truncate is not supported for GoogleGemini")}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,endSequences:e.endSequences,stream:e.stream,n:e.n}}createChatReq=async(e,t)=>{let n=e.model,r=e.modelConfig?.stream??this.config.stream;if(!e.chatPrompt||e.chatPrompt.length===0)throw new Error("Chat prompt is empty");let o;if(this.endpointId?o={name:r?`/${this.endpointId}:streamGenerateContent?alt=sse`:`/${this.endpointId}:generateContent`}:o={name:r?`/models/${n}:streamGenerateContent?alt=sse`:`/models/${n}:generateContent`},!this.isVertex){let g=r?"&":"?",f=typeof this.apiKey=="function"?await this.apiKey():this.apiKey;o.name+=`${g}key=${f}`}let i=e.chatPrompt.filter(g=>g.role==="system").map(g=>g.content),a=i.length>0?{role:"user",parts:[{text:i.join(" ")}]}:void 0,l=e.chatPrompt.filter(g=>g.role!=="system").map((g,f)=>{switch(g.role){case"user":return{role:"user",parts:Array.isArray(g.content)?g.content.map((x,y)=>{switch(x.type){case"text":return{text:x.text};case"image":return{inlineData:{mimeType:x.mimeType,data:x.image}};case"file":return"fileUri"in x?{fileData:{mimeType:x.mimeType,fileUri:x.fileUri}}:{inlineData:{mimeType:x.mimeType,data:x.data}};default:throw new Error(`Chat prompt content type not supported (index: ${y})`)}}):[{text:g.content}]};case"assistant":{let A=[];if(g.functionCalls){if(A=g.functionCalls.map(x=>{let y=typeof x.function.params=="string"?JSON.parse(x.function.params):x.function.params;return{functionCall:{name:x.function.name,args:y}}}),!A)throw new Error("Function call is empty");return{role:"model",parts:A}}if(!g.content)throw new Error("Assistant content is empty");return A=[{text:g.content}],{role:"model",parts:A}}case"function":{if(!("functionId"in g))throw new Error(`Chat prompt functionId is empty (index: ${f})`);return{role:"user",parts:[{functionResponse:{name:g.functionId,response:{result:g.result}}}]}}default:throw new Error(`Invalid role: ${JSON.stringify(g)} (index: ${f})`)}}),p=[];if(e.functions&&e.functions.length>0){let g=e.functions.map(f=>({...f,parameters:f.parameters?mr(f.parameters):void 0}));p.push({function_declarations:g})}this.options?.codeExecution&&p.push({code_execution:{}}),this.options?.googleSearchRetrieval&&p.push({google_search_retrieval:{dynamic_retrieval_config:this.options.googleSearchRetrieval}}),this.options?.googleSearch&&p.push({google_search:{}}),this.options?.urlContext&&p.push({url_context:{}}),p.length===0&&(p=void 0);let c;if(e.functionCall)if(e.functionCall==="none")c={function_calling_config:{mode:"NONE"}};else if(e.functionCall==="auto")c={function_calling_config:{mode:"AUTO"}};else if(e.functionCall==="required")c={function_calling_config:{mode:"ANY"}};else{let g=e.functionCall.function?.name?{allowedFunctionNames:[e.functionCall.function.name]}:{};c={function_calling_config:{mode:"ANY"},...g}}else p&&p.length>0&&(c={function_calling_config:{mode:"AUTO"}});let u={};if(this.config.thinking?.includeThoughts&&(u.includeThoughts=!0),this.config.thinking?.thinkingTokenBudget&&(u.thinkingBudget=this.config.thinking.thinkingTokenBudget),t?.thinkingTokenBudget){let g=this.config.thinkingTokenBudgetLevels;switch(t.thinkingTokenBudget){case"none":u.thinkingBudget=0,u.includeThoughts=!1;break;case"minimal":u.thinkingBudget=g?.minimal??200;break;case"low":u.thinkingBudget=g?.low??800;break;case"medium":u.thinkingBudget=g?.medium??5e3;break;case"high":u.thinkingBudget=g?.high??1e4;break;case"highest":u.thinkingBudget=g?.highest??24500;break}}t?.showThoughts!==void 0&&t?.thinkingTokenBudget!=="none"&&(u.includeThoughts=t.showThoughts);let d={maxOutputTokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,topP:e.modelConfig?.topP??this.config.topP,topK:e.modelConfig?.topK??this.config.topK,frequencyPenalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,candidateCount:1,stopSequences:e.modelConfig?.stopSequences??this.config.stopSequences,responseMimeType:"text/plain",...Object.keys(u).length>0?{thinkingConfig:u}:{}},m=this.config.safetySettings;return[o,{contents:l,tools:p,toolConfig:c,systemInstruction:a,generationConfig:d,safetySettings:m}]};createEmbedReq=async e=>{let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n,r;if(this.isVertex)this.endpointId?n={name:`/${this.endpointId}:predict`}:n={name:`/models/${t}:predict`},r={instances:e.texts.map(o=>({content:o,...this.config.embedType&&{taskType:this.config.embedType}})),parameters:{autoTruncate:this.config.autoTruncate,outputDimensionality:this.config.dimensions}};else{let o=typeof this.apiKey=="function"?this.apiKey():this.apiKey;n={name:`/models/${t}:batchEmbedContents?key=${o}`},r={requests:e.texts.map(i=>({model:`models/${t}`,content:{parts:[{text:i}]},outputDimensionality:this.config.dimensions,...this.config.embedType&&{taskType:this.config.embedType}}))}}return[n,r]};createChatResp=e=>{let t=e.candidates?.map(n=>{let r={index:0};switch(n.finishReason){case"MAX_TOKENS":r.finishReason="length";break;case"STOP":r.finishReason="stop";break;case"SAFETY":throw new O("Content was blocked due to safety settings",void 0,void 0);case"RECITATION":throw new O("Content was blocked due to recitation policy",void 0,void 0);case"MALFORMED_FUNCTION_CALL":throw new O("Function call was malformed and blocked",void 0,void 0);case"UNEXPECTED_TOOL_CALL":throw new O("Unexpected tool call",void 0,void 0);case"FINISH_REASON_UNSPECIFIED":throw new O("Finish reason unspecified",void 0,void 0);case"BLOCKLIST":throw new O("Content was blocked due to blocklist",void 0,void 0);case"PROHIBITED_CONTENT":throw new O("Content was blocked due to prohibited content",void 0,void 0);case"SPII":throw new O("Content was blocked due to SPII",void 0,void 0);case"OTHER":throw new O("Other finish reason",void 0,void 0)}if(!n.content||!n.content.parts)return r;for(let o of n.content.parts){if("text"in o){"thought"in o&&o.thought?r.thought=o.text:r.content=o.text;continue}"functionCall"in o&&(r.functionCalls=[{id:N(),type:"function",function:{name:o.functionCall.name,params:o.functionCall.args}}])}return r});return e.usageMetadata&&(this.tokensUsed={totalTokens:e.usageMetadata.totalTokenCount,promptTokens:e.usageMetadata.promptTokenCount,completionTokens:e.usageMetadata.candidatesTokenCount,thoughtsTokens:e.usageMetadata.thoughtsTokenCount}),{results:t}};createChatStreamResp=e=>this.createChatResp(e);createEmbedResp=e=>{let t;return this.isVertex?t=e.predictions.map(n=>n.embeddings.values):t=e.embeddings.map(n=>n.values),{embeddings:t}}},Ce=class s extends E{static create(e){return new s(e)}constructor({apiKey:e,projectId:t,region:n,endpointId:r,config:o,options:i,models:a,modelInfo:l}){let p=t!==void 0&&n!==void 0,c,u;if(p){if(!e)throw new Error("GoogleGemini Vertex API key not set");if(typeof e!="function")throw new Error("GoogleGemini Vertex API key must be a function for token-based authentication");let g;r?g="endpoints":g="publishers/google",c=`https://${n==="global"?"aiplatform":`${n}-aiplatform`}.googleapis.com/v1/projects/${t}/locations/${n}/${g}`,u=async()=>({Authorization:`Bearer ${typeof e=="function"?await e():e}`})}else{if(!e)throw new Error("GoogleGemini AI API key not set");c="https://generativelanguage.googleapis.com/v1beta",u=async()=>({})}let d={...hr(),...o},m=new gr(d,p,r,e,i);l=[...Jt,...l??[]];let h=g=>{let f=q({model:g,modelInfo:l,models:a});return{functions:!0,streaming:!0,hasThinkingBudget:f?.hasThinkingBudget??!1,hasShowThoughts:f?.hasShowThoughts??!1,functionCot:!1,media:{images:{supported:!0,formats:["image/jpeg","image/png","image/gif","image/webp"],maxSize:20*1024*1024,detailLevels:["high","low","auto"]},audio:{supported:!0,formats:["wav","mp3","aac","ogg"],maxDuration:9.5*60},files:{supported:!0,formats:["application/pdf","text/plain","text/csv","text/html","text/xml"],maxSize:2*1024*1024*1024,uploadMethod:"cloud"},urls:{supported:!0,webSearch:!0,contextFetching:!0}},caching:{supported:!1,types:[]},thinking:f?.hasThinkingBudget??!1,multiTurn:!0}};super(m,{name:"GoogleGeminiAI",apiURL:c,headers:u,modelInfo:l,defaults:{model:d.model,embedModel:d.embedModel},options:i,supportFor:h,models:a})}};var ki=new G,we=class{options;maxTokens;refillRate;currentTokens;lastRefillTime;constructor(e,t,n){this.maxTokens=e,this.refillRate=t,this.currentTokens=e,this.lastRefillTime=Date.now(),this.options=n}refillTokens(){let e=Date.now(),n=(e-this.lastRefillTime)/1e3*this.refillRate;this.currentTokens=Math.min(this.maxTokens,this.currentTokens+n),this.lastRefillTime=e}async waitUntilTokensAvailable(e){if(this.refillTokens(),this.currentTokens>=e){this.currentTokens-=e;return}return this.options?.debug&&console.log(ki.red(`Rate limiter: Waiting for ${e-this.currentTokens} tokens`)),await new Promise(t=>setTimeout(t,100)),this.waitUntilTokensAvailable(e)}async acquire(e){await this.waitUntilTokensAvailable(e)}};var it=(r=>(r.Llama3_8B="llama3-8b-8192",r.Llama33_70B="llama-3.3-70b-versatile",r.Mixtral_8x7B="mixtral-8x7b-32768",r.Gemma2_9B="gemma2-9b-it",r))(it||{});var Qt=[{name:"gemma2-9b-it",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.2,completionTokenCostPer1M:.2},{name:"llama-3.3-70b-versatile",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.59,completionTokenCostPer1M:.79},{name:"llama3-8b-8192",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.05,completionTokenCostPer1M:.08},{name:"mixtral-8x7b-32768",currency:"usd",characterIsToken:!0,promptTokenCostPer1M:.24,completionTokenCostPer1M:.24}];var Ei=()=>structuredClone({model:"llama-3.3-70b-versatile",...S()}),Se=class extends F{constructor({apiKey:e,config:t,options:n,models:r,modelInfo:o}){if(!e||e==="")throw new Error("Groq API key not set");let i={...Ei(),...t},a={...n,streamingUsage:!1};o=[...Qt,...o??[]];let l={functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0};super({apiKey:e,config:i,options:a,modelInfo:o,apiURL:"https://api.groq.com/openai/v1",models:r,supportFor:l}),super.setName("Groq"),this.setOptions(a)}setOptions=e=>{let t=this.newRateLimiter(e);super.setOptions({...e,rateLimiter:t})};newRateLimiter=e=>{if(e?.rateLimiter)return e.rateLimiter;let t=e?.tokensPerMinute??4800,n=new we(t,t/60,{debug:e?.debug});return async(o,i)=>{let a=i.modelUsage?.tokens?.totalTokens||0;return await n.acquire(a),await o()}}};var Yt=[];var Xt=(e=>(e.MetaLlama270BChatHF="meta-llama/Llama-2-70b-chat-hf",e))(Xt||{});var xr=()=>structuredClone({model:"meta-llama/Llama-2-70b-chat-hf",...S()}),Jo=()=>structuredClone({model:"meta-llama/Llama-2-70b-chat-hf",..._()}),Ar=class{constructor(e){this.config=e}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,n:e.n,presencePenalty:e.presencePenalty}}createChatReq=e=>{let t=e.model,n=e.functions?`Functions:
111
111
  ${JSON.stringify(e.functions,null,2)}
112
- `:"",o=e.chatPrompt?.map(l=>{switch(l.role){case"user":return`User: ${l.content}`;case"system":return`System: ${l.content}`;case"function":return`Function Result: ${l.result}`;case"assistant":{let p=l.functionCalls?.map(c=>{let u=typeof c.function.params=="string"?c.function.params:JSON.stringify(c.function.params);return`${c.function.name}(${u})`}).join(`
112
+ `:"",r=e.chatPrompt?.map(l=>{switch(l.role){case"user":return`User: ${l.content}`;case"system":return`System: ${l.content}`;case"function":return`Function Result: ${l.result}`;case"assistant":{let p=l.functionCalls?.map(c=>{let u=typeof c.function.params=="string"?c.function.params:JSON.stringify(c.function.params);return`${c.function.name}(${u})`}).join(`
113
113
  `);return p?`Assistant: ${l.content}
114
114
  Functions:
115
115
  ${p}`:`Assistant: ${l.content}`}default:throw new Error("Unknown role")}}).join(`
116
- `),r=`${n} ${o}`.trim(),i={name:"/models"},a={model:t,inputs:r,parameters:{max_new_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,repetition_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,temperature:e.modelConfig?.temperature??this.config.temperature,top_p:e.modelConfig?.topP??this.config.topP,top_k:e.modelConfig?.topK??this.config.topK,return_full_text:this.config.returnFullText,num_return_sequences:this.config.n,do_sample:this.config.doSample,max_time:this.config.maxTime},options:{use_cache:this.config.useCache,wait_for_model:this.config.waitForModel}};return[i,a]};createChatResp=e=>({results:[{index:0,content:e.generated_text}]})},ve=class extends E{constructor({apiKey:e,config:t,options:n,models:o}){if(!e||e==="")throw new Error("HuggingFace API key not set");let r={...Ao(),...t},i=new ho(r);super(i,{name:"HuggingFace",apiURL:"https://api-inference.huggingface.co",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:Qt,defaults:{model:r.model},options:n,supportFor:{functions:!1,streaming:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},models:o})}};var at=(l=>(l.Mistral7B="open-mistral-7b",l.Mistral8x7B="open-mixtral-8x7b",l.MistralSmall="mistral-small-latest",l.MistralNemo="mistral-nemo-latest",l.MistralLarge="mistral-large-latest",l.Codestral="codestral-latest",l.OpenCodestralMamba="open-codestral-mamba",l.OpenMistralNemo="open-mistral-nemo-latest",l))(at||{}),xo=(e=>(e.MistralEmbed="mistral-embed",e))(xo||{});var Xt=[{name:"open-mistral-7b",currency:"USD",promptTokenCostPer1M:.25,completionTokenCostPer1M:.25},{name:"open-mixtral-8x7b",currency:"USD",promptTokenCostPer1M:.7,completionTokenCostPer1M:.7},{name:"mistral-nemo-latest",currency:"USD",promptTokenCostPer1M:.15,completionTokenCostPer1M:.15},{name:"mistral-small-latest",currency:"USD",promptTokenCostPer1M:.2,completionTokenCostPer1M:.6},{name:"mistral-large-latest",currency:"USD",promptTokenCostPer1M:2,completionTokenCostPer1M:6},{name:"codestral-latest",currency:"USD",promptTokenCostPer1M:.2,completionTokenCostPer1M:.6},{name:"open-codestral-mamba",currency:"USD",promptTokenCostPer1M:.25,completionTokenCostPer1M:.25},{name:"open-mistral-nemo-latest",currency:"USD",promptTokenCostPer1M:.3,completionTokenCostPer1M:.3}];var Zt=()=>structuredClone({model:"mistral-small-latest",...w(),topP:1}),Wr=()=>structuredClone({...Zt(),model:"mistral-large-latest"}),Oe=class extends F{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("Mistral API key not set");let i={...Zt(),...t};r=[...Xt,...r??[]];let a={functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},l=p=>{let{max_completion_tokens:c,messages:u,...d}=p;return{...d,messages:this.updateMessages(u),max_tokens:c}};super({apiKey:e,config:i,options:n,apiURL:"https://api.mistral.ai/v1",modelInfo:r,models:o,supportFor:a,chatReqUpdater:l}),super.setName("Mistral")}updateMessages(e){let t=[];if(!Array.isArray(e))return e;for(let n of e)if(n.role==="user"&&Array.isArray(n.content)){let o=n.content.map(r=>typeof r=="object"&&r!==null&&"image_url"in r?{type:"image_url",image_url:{url:r.image_url?.url}}:r);t.push({...n,content:o})}else t.push(n);return t}};var en=class{constructor(e={}){this.config=e;this.config.id=this.config.id??$()}metrics={latency:{chat:{mean:0,p95:0,p99:0,samples:[]},embed:{mean:0,p95:0,p99:0,samples:[]}},errors:{chat:{count:0,rate:0,total:0},embed:{count:0,rate:0,total:0}}};getLastUsedChatModel(){return this.config.modelInfo?.name??"mock-model"}getLastUsedEmbedModel(){return this.config.embedModelInfo?.name??"mock-embed-model"}getLastUsedModelConfig(){return this.config.modelInfo?{maxTokens:this.config.modelInfo.maxTokens,temperature:.7,stream:this.config.features?.streaming??!1}:void 0}getName(){return this.config.name??"mock-ai-service"}getId(){return this.config.id??"mock-ai-service-id"}getFeatures(e){return{functions:this.config.features?.functions??!1,streaming:this.config.features?.streaming??!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}getModelList(){return this.config.models}getMetrics(){return this.metrics}async chat(e,t){if(this.config.latencyMs&&await new Promise(n=>setTimeout(n,this.config.latencyMs)),this.config.shouldError)throw new Error(this.config.errorMessage??"Mock chat error");return this.updateMetrics("chat"),typeof this.config.chatResponse=="function"?await this.config.chatResponse(e):this.config.chatResponse??{results:[{index:0,content:"Mock response",finishReason:"stop"}],modelUsage:{ai:this.getName(),model:"mock-model",tokens:{promptTokens:10,completionTokens:5,totalTokens:15}}}}async embed(e,t){if(this.config.latencyMs&&await new Promise(n=>setTimeout(n,this.config.latencyMs)),this.config.shouldError)throw new Error(this.config.errorMessage??"Mock embed error");return this.updateMetrics("embed"),typeof this.config.embedResponse=="function"?this.config.embedResponse(e):this.config.embedResponse??{embeddings:[[.1,.2,.3]],modelUsage:{ai:this.getName(),model:"mock-model",tokens:{promptTokens:5,completionTokens:0,totalTokens:5}}}}setOptions(e){this.config.options=e}getOptions(){return this.config.options??{}}getLogger(){return this.config.options?.logger??(e=>{console.log(e)})}updateMetrics(e){let t=this.config.latencyMs??0;this.metrics.latency[e].samples.push(t);let n=this.metrics.latency[e].samples;if(this.metrics.latency[e].mean=n.reduce((o,r)=>o+r,0)/n.length,n.length>0){let o=[...n].sort((a,l)=>a-l),r=Math.max(0,Math.floor(o.length*.95)-1);this.metrics.latency[e].p95=o[r]??t;let i=Math.max(0,Math.floor(o.length*.99)-1);this.metrics.latency[e].p99=o[i]??t}if(this.config.shouldError){this.metrics.errors[e].count++,this.metrics.errors[e].total++;let o=this.metrics.latency[e].samples.length;this.metrics.errors[e].rate=o>0?this.metrics.errors[e].count/o:0}}};var tn=class s{options;lastUsedService;services=new Map;constructor(e){if(e.length===0)throw new Error("No AI services provided.");for(let[t,n]of e.entries())if("key"in n){if(this.services.has(n.key))throw new Error(`Duplicate model key: ${n.key}`);let{service:r,description:i,isInternal:a}=n;this.services.set(n.key,{service:r,description:i,isInternal:a})}else{let r=n.getModelList();if(!r)throw new Error(`Service ${t} \`${n.getName()}\` has no model list.`);for(let i of r){if(this.services.has(i.key)){let a=this.services.get(i.key)?.service;throw new Error(`Service ${t} \`${n.getName()}\` has duplicate model key: ${i.key} as service ${a?.getName()}`)}if("model"in i&&typeof i.model)this.services.set(i.key,{description:i.description,service:n,model:i.model});else if("embedModel"in i&&i.embedModel)this.services.set(i.key,{description:i.description,service:n,embedModel:i.embedModel});else throw new Error(`Key ${i.key} in model list for service ${t} \`${n.getName()}\` is missing a model or embedModel property.`)}}}static create(e){return new s(e)}getLastUsedChatModel(){return this.lastUsedService?.getLastUsedChatModel()}getLastUsedEmbedModel(){return this.lastUsedService?.getLastUsedEmbedModel()}getLastUsedModelConfig(){return this.lastUsedService?.getLastUsedModelConfig()}async chat(e,t){let n=e.model;if(!n)throw new Error("Model key must be specified for multi-service");let o=this.services.get(n);if(!o)throw new Error(`No service found for model key: ${n}`);if(this.lastUsedService=o.service,!o.model){let{model:r,...i}=e;return await o.service.chat(i,t)}return await o.service.chat({model:n,...e},t)}async embed(e,t){let n=e.embedModel;if(!n)throw new Error("Embed model key must be specified for multi-service");let o=this.services.get(n);if(!o)throw new Error(`No service found for embed model key: ${n}`);if(this.lastUsedService=o.service,!o.model){let{embedModel:r,...i}=e;return await o.service.embed(i,t)}return await o.service.embed({embedModel:n,...e},t)}getId(){return`MultiServiceRouter:${Array.from(this.services.values()).map(e=>e.service.getId()).join(",")}`}getName(){return"MultiServiceRouter"}getModelList(){return Array.from(this.services).filter(([,e])=>!e.isInternal).map(([e,t])=>{if(t.model)return{key:e,description:t.description,model:t.model};if(t.embedModel)return{key:e,description:t.description,embedModel:t.embedModel};throw new Error(`Service ${e} has no model or embedModel`)})}getFeatures(e){if(e){let t=this.services.get(e);if(t)return t.service.getFeatures(e)}return{functions:!1,streaming:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}getMetrics(){let e=this.lastUsedService;if(!e){let t=this.services.values().next().value;t&&(e="service"in t?t.service:t)}if(!e)throw new Error("No service available to get metrics.");return e.getMetrics()}setOptions(e){for(let t of this.services.values())t.service.setOptions(e);this.options=e}getOptions(){return this.options??{}}getLogger(){let e=this.lastUsedService;if(!e){let t=this.services.values().next().value;t&&(e=t.service)}if(!e)throw new Error("No service available to get logger.");return e.getLogger()}setServiceEntry(e,t){this.services.set(e,t)}};var fo=()=>structuredClone({...w(),model:"nous-hermes2",embedModel:"all-minilm"}),Jr=()=>structuredClone({..._(),model:"nous-hermes2",embedModel:"all-minilm"}),Me=class extends F{constructor({apiKey:e="not-set",url:t="http://localhost:11434/v1",config:n,options:o,models:r}){let i={...fo(),...n};super({apiKey:e,options:o,config:i,apiURL:t,models:r,modelInfo:[],supportFor:{functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}),super.setName("Ollama")}};var ki=s=>["o1","o1-mini","o1-pro","o3","o3-mini","o3-pro","o4-mini"].includes(s),ke=class{constructor(e,t,n){this.config=e;this.streamingUsage=t;this.responsesReqUpdater=n}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,stopSequences:e.stopSequences,topP:e.topP,stream:e.stream}}mapInternalContentToResponsesInput(e){return e.map(n=>{if(n.type==="text")return{type:"text",text:n.text};if(n.type==="image")return{type:"image_url",image_url:{url:`data:${n.mimeType};base64,${n.image}`,details:n.details??"auto"}};if(n.type==="audio")return{type:"input_audio",input_audio:{data:n.data,format:n.format==="wav"?"wav":void 0}};let o=n;throw new Error(`Unsupported content part: ${JSON.stringify(o)}`)})}createResponsesReqInternalInput(e,t=!1){let n=[];for(let o of e){if(t&&o.role==="system")continue;let r;if(o.role==="system"||o.role==="user"||o.role==="assistant"&&o.content)if(typeof o.content=="string")r=o.content;else if(Array.isArray(o.content))r=this.mapInternalContentToResponsesInput(o.content);else{if(!(o.role==="assistant"&&!o.content&&o.functionCalls))throw new Error(`Invalid content type for role ${o.role}`);r=""}else o.role,r="";switch(o.role){case"system":n.push({type:"message",role:"system",content:r});break;case"user":n.push({type:"message",role:"user",content:r,name:o.name});break;case"assistant":if(o.content||o.functionCalls){let i={type:"message",role:"assistant",content:""};if(o.content&&(i.content=r),o.name&&(i.name=o.name),o.content&&n.push(i),o.functionCalls)for(let a of o.functionCalls)n.push({type:"function_call",call_id:a.id,name:a.function.name,arguments:typeof a.function.params=="object"?JSON.stringify(a.function.params):a.function.params||""})}break;case"function":n.push({type:"function_call_output",call_id:o.functionId,output:o.result});break;default:{let i=o.role;throw new Error(`Invalid role in chat prompt: ${i}`)}}}return n}createChatReq(e,t){let n=e.model,o={name:"/responses"},r=null,i=!1;if(e.chatPrompt){for(let A of e.chatPrompt)if(A.role==="system"&&typeof A.content=="string"){r=A.content,i=!0;break}}let a=r??this.config.systemPrompt??null,l=e.functions?.map(A=>({type:"function",name:A.name,description:A.description,parameters:A.parameters??{}})),p=[],c=ki(n),u=this.config.reasoningSummary;t?.showThoughts?u||(u="auto"):u=void 0;let d=this.config.reasoningEffort;if(t?.thinkingTokenBudget)switch(t.thinkingTokenBudget){case"none":d=void 0;break;case"minimal":d="low";break;case"low":d="medium";break;case"medium":case"high":case"highest":d="high";break}let m={model:n,input:"",instructions:a,tools:l?.length?l:void 0,tool_choice:e.functionCall==="none"||e.functionCall==="auto"||e.functionCall==="required"?e.functionCall:typeof e.functionCall=="object"&&e.functionCall.function?{type:"function",name:e.functionCall.function.name}:void 0,...c?{max_output_tokens:e.modelConfig?.maxTokens??this.config.maxTokens??void 0}:{temperature:e.modelConfig?.temperature??this.config.temperature??void 0,top_p:e.modelConfig?.topP??this.config.topP??void 0,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty??void 0,frequency_penalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty??void 0},stream:e.modelConfig?.stream??this.config.stream??!1,background:void 0,include:p.length>0?p:void 0,metadata:void 0,parallel_tool_calls:this.config.parallelToolCalls,previous_response_id:void 0,...d?{reasoning:{effort:d,summary:u}}:{},service_tier:this.config.serviceTier,store:this.config.store,text:void 0,truncation:void 0,user:this.config.user,seed:this.config.seed};this.config.user&&(m.user=this.config.user),this.config.parallelToolCalls!==void 0&&(m.parallel_tool_calls=this.config.parallelToolCalls),this.config.responseFormat&&(m.text={format:{type:this.config.responseFormat}}),this.config.seed&&(m.seed=this.config.seed);let h=e.chatPrompt?this.createResponsesReqInternalInput(e.chatPrompt,i):[];if(h.length>0)m.input=h;else if(e.chatPrompt&&e.chatPrompt.length===1&&e.chatPrompt[0]?.role==="user"&&e.chatPrompt[0]?.content&&typeof e.chatPrompt[0].content=="string"&&!a)m.input=e.chatPrompt[0].content;else if(h.length===0&&!a)throw new Error("Responses API request must have input or instructions.");let g=m.reasoning??{};if(this.config.reasoningEffort&&(g={...g,effort:this.config.reasoningEffort}),t?.thinkingTokenBudget)switch(t.thinkingTokenBudget){case"none":g={};break;case"minimal":g={...g,effort:"low"};break;case"low":g={...g,effort:"medium"};break;case"medium":case"high":case"highest":g={...g,effort:"high"};break}Object.keys(g).length>0&&g.effort?m.reasoning=g:m.reasoning=void 0;let f=m;return this.responsesReqUpdater&&(f=this.responsesReqUpdater(f)),[o,f]}createChatResp(e){let{id:t,output:n,usage:o}=e;o&&(this.tokensUsed={promptTokens:o.prompt_tokens,completionTokens:o.completion_tokens,totalTokens:o.total_tokens});let r={};for(let i of n??[])switch(i.type){case"message":r.id=i.id,r.content=yo(i.content,t),r.finishReason=i.status==="completed"?"stop":"content_filter";break;case"reasoning":r.id=i.id,i.encrypted_content?r.thought=i.encrypted_content:r.thought=i.summary.map(a=>typeof a=="object"?JSON.stringify(a):a).join(`
117
- `);break;case"file_search_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:"file_search",params:{queries:i.queries,results:i.results}}}],r.finishReason="function_call";break;case"web_search_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:"web_search",params:{queries:i.queries}}}],r.finishReason="function_call";break;case"computer_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:"computer_use",params:{action:i.action}}}],r.finishReason="function_call";break;case"code_interpreter_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:"code_interpreter",params:{code:i.code,results:i.results}}}],r.finishReason="function_call";break;case"image_generation_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:"image_generation",params:{result:i.result}}}],r.finishReason="function_call";break;case"local_shell_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:"local_shell",params:{action:i.action}}}],r.finishReason="function_call";break;case"mcp_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:"mcp",params:{name:i.name,args:i.args,serverLabel:i.server_label,output:i.output,error:i.error}}}],r.finishReason="function_call";break;case"function_call":r.id=i.id,r.functionCalls=[{id:i.id,type:"function",function:{name:i.name,params:i.arguments}}],r.finishReason="function_call";break}return{results:[{...r,index:0}],remoteId:t}}createChatStreamResp(e){let t=e,n={index:0,id:"",content:"",finishReason:"stop"},o;switch(t.type){case"response.created":case"response.in_progress":case"response.queued":o=t.response.id,n.id=`${t.response.id}_res_0`;break;case"response.output_item.added":switch(t.item.type){case"message":n.id=t.item.id,n.content=yo(t.item.content,t.item.id);break;case"function_call":n.id=t.item.id,n.functionCalls=[{id:t.item.id,type:"function",function:{name:t.item.name,params:t.item.arguments}}];break;case"file_search_call":{let r=t.item;n.id=t.item.id,n.functionCalls=[{id:r.id,type:"function",function:{name:"file_search",params:{queries:r.queries||[],results:r.results?.map(i=>({fileId:i.file_id,filename:i.filename,score:i.score,text:i.text,attributes:i.attributes}))}}}]}break;case"web_search_call":{let r=t.item;n.id=t.item.id,n.functionCalls=[{id:r.id,type:"function",function:{name:"web_search",params:{queries:r.queries||[]}}}]}break;case"computer_call":{let r=t.item;n.id=t.item.id,n.functionCalls=[{id:r.id,type:"function",function:{name:"computer_use",params:{action:r.action||{}}}}]}break;case"code_interpreter_call":{let r=t.item;n.id=t.item.id,n.functionCalls=[{id:r.id,type:"function",function:{name:"code_interpreter",params:{code:r.code||"",results:r.results}}}]}break;case"image_generation_call":{let r=t.item;n.id=t.item.id,n.functionCalls=[{id:r.id,type:"function",function:{name:"image_generation",params:{result:r.result}}}]}break;case"local_shell_call":{let r=t.item;n.id=t.item.id,n.functionCalls=[{id:r.id,type:"function",function:{name:"local_shell",params:{action:r.action||{}}}}]}break;case"mcp_call":{let r=t.item;n.id=t.item.id,n.functionCalls=[{id:r.id,type:"function",function:{name:"mcp",params:{name:r.name||"",args:r.args||"",serverLabel:r.server_label||"",output:r.output,error:r.error}}}]}break}break;case"response.content_part.added":n.id=t.item_id,n.content=yo([t.part],t.item_id);break;case"response.output_text.delta":n.id=t.item_id,n.content=t.delta;break;case"response.output_text.done":break;case"response.function_call_arguments.delta":n.id=t.item_id,n.functionCalls=[{id:t.item_id,type:"function",function:{name:"",params:t.delta}}];break;case"response.reasoning_summary_text.delta":n.id=t.item_id,n.thought=t.delta;break;case"response.file_search_call.in_progress":case"response.file_search_call.searching":n.id=t.item_id,n.finishReason="function_call";break;case"response.file_search_call.completed":n.id=t.item_id,n.finishReason="function_call";break;case"response.web_search_call.in_progress":case"response.web_search_call.searching":n.id=t.item_id,n.finishReason="function_call";break;case"response.web_search_call.completed":n.id=t.item_id,n.finishReason="function_call";break;case"response.image_generation_call.in_progress":case"response.image_generation_call.generating":n.id=t.item_id,n.finishReason="function_call";break;case"response.image_generation_call.completed":n.id=t.item_id,n.finishReason="function_call";break;case"response.image_generation_call.partial_image":n.id=t.item_id,n.finishReason="function_call";break;case"response.mcp_call.in_progress":n.id=t.item_id,n.finishReason="function_call";break;case"response.mcp_call.arguments.delta":n.id=t.item_id,n.functionCalls=[{id:t.item_id,type:"function",function:{name:"",params:t.delta}}];break;case"response.mcp_call.arguments.done":n.id=t.item_id,n.functionCalls=[{id:t.item_id,type:"function",function:{name:"",params:t.arguments}}];break;case"response.mcp_call.completed":case"response.mcp_call.failed":n.id="mcp_call_event",n.finishReason="function_call";break;case"response.mcp_list_tools.in_progress":case"response.mcp_list_tools.completed":case"response.mcp_list_tools.failed":n.id="mcp_list_tools_event",n.finishReason="function_call";break;case"response.output_item.done":switch(t.item.type){case"message":n.id=t.item.id,n.finishReason=t.item.status==="completed"?"stop":"error";break;case"function_call":case"file_search_call":case"web_search_call":case"computer_call":case"code_interpreter_call":case"image_generation_call":case"local_shell_call":case"mcp_call":n.id=t.item.id,n.finishReason="function_call";break}break;case"response.completed":t.response.usage&&(this.tokensUsed={promptTokens:t.response.usage.prompt_tokens,completionTokens:t.response.usage.completion_tokens,totalTokens:t.response.usage.total_tokens}),o=t.response.id,n.id=`${t.response.id}_completed`,n.finishReason="stop";break;case"response.failed":o=t.response.id,n.id=`${t.response.id}_failed`,n.finishReason="error";break;case"response.incomplete":o=t.response.id,n.id=`${t.response.id}_incomplete`,n.finishReason="length";break;case"error":n.id="error",n.content=`Error: ${t.message}`,n.finishReason="error";break;default:n.id="unknown";break}return{results:[n],remoteId:o}}createEmbedReq(e){let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n={name:"/embeddings"},o={model:t,input:e.texts,dimensions:this.config.dimensions};return[n,o]}},yo=(s,e)=>{let t=s.filter(n=>n.type==="refusal");if(t.length>0){let n=t.map(o=>o.refusal).join(`
116
+ `),o=`${n} ${r}`.trim(),i={name:"/models"},a={model:t,inputs:o,parameters:{max_new_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,repetition_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,temperature:e.modelConfig?.temperature??this.config.temperature,top_p:e.modelConfig?.topP??this.config.topP,top_k:e.modelConfig?.topK??this.config.topK,return_full_text:this.config.returnFullText,num_return_sequences:this.config.n,do_sample:this.config.doSample,max_time:this.config.maxTime},options:{use_cache:this.config.useCache,wait_for_model:this.config.waitForModel}};return[i,a]};createChatResp=e=>({results:[{index:0,content:e.generated_text}]})},ve=class extends E{constructor({apiKey:e,config:t,options:n,models:r}){if(!e||e==="")throw new Error("HuggingFace API key not set");let o={...xr(),...t},i=new Ar(o);super(i,{name:"HuggingFace",apiURL:"https://api-inference.huggingface.co",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:Yt,defaults:{model:o.model},options:n,supportFor:{functions:!1,streaming:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},models:r})}};var at=(l=>(l.Mistral7B="open-mistral-7b",l.Mistral8x7B="open-mixtral-8x7b",l.MistralSmall="mistral-small-latest",l.MistralNemo="mistral-nemo-latest",l.MistralLarge="mistral-large-latest",l.Codestral="codestral-latest",l.OpenCodestralMamba="open-codestral-mamba",l.OpenMistralNemo="open-mistral-nemo-latest",l))(at||{}),fr=(e=>(e.MistralEmbed="mistral-embed",e))(fr||{});var Zt=[{name:"open-mistral-7b",currency:"USD",promptTokenCostPer1M:.25,completionTokenCostPer1M:.25},{name:"open-mixtral-8x7b",currency:"USD",promptTokenCostPer1M:.7,completionTokenCostPer1M:.7},{name:"mistral-nemo-latest",currency:"USD",promptTokenCostPer1M:.15,completionTokenCostPer1M:.15},{name:"mistral-small-latest",currency:"USD",promptTokenCostPer1M:.2,completionTokenCostPer1M:.6},{name:"mistral-large-latest",currency:"USD",promptTokenCostPer1M:2,completionTokenCostPer1M:6},{name:"codestral-latest",currency:"USD",promptTokenCostPer1M:.2,completionTokenCostPer1M:.6},{name:"open-codestral-mamba",currency:"USD",promptTokenCostPer1M:.25,completionTokenCostPer1M:.25},{name:"open-mistral-nemo-latest",currency:"USD",promptTokenCostPer1M:.3,completionTokenCostPer1M:.3}];var en=()=>structuredClone({model:"mistral-small-latest",...S(),topP:1}),Qo=()=>structuredClone({...en(),model:"mistral-large-latest"}),Oe=class extends F{constructor({apiKey:e,config:t,options:n,models:r,modelInfo:o}){if(!e||e==="")throw new Error("Mistral API key not set");let i={...en(),...t};o=[...Zt,...o??[]];let a={functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},l=p=>{let{max_completion_tokens:c,messages:u,...d}=p;return{...d,messages:this.updateMessages(u),max_tokens:c}};super({apiKey:e,config:i,options:n,apiURL:"https://api.mistral.ai/v1",modelInfo:o,models:r,supportFor:a,chatReqUpdater:l}),super.setName("Mistral")}updateMessages(e){let t=[];if(!Array.isArray(e))return e;for(let n of e)if(n.role==="user"&&Array.isArray(n.content)){let r=n.content.map(o=>typeof o=="object"&&o!==null&&"image_url"in o?{type:"image_url",image_url:{url:o.image_url?.url}}:o);t.push({...n,content:r})}else t.push(n);return t}};var tn=class{constructor(e={}){this.config=e;this.config.id=this.config.id??N()}metrics={latency:{chat:{mean:0,p95:0,p99:0,samples:[]},embed:{mean:0,p95:0,p99:0,samples:[]}},errors:{chat:{count:0,rate:0,total:0},embed:{count:0,rate:0,total:0}}};getLastUsedChatModel(){return this.config.modelInfo?.name??"mock-model"}getLastUsedEmbedModel(){return this.config.embedModelInfo?.name??"mock-embed-model"}getLastUsedModelConfig(){return this.config.modelInfo?{maxTokens:this.config.modelInfo.maxTokens,temperature:.7,stream:this.config.features?.streaming??!1}:void 0}getName(){return this.config.name??"mock-ai-service"}getId(){return this.config.id??"mock-ai-service-id"}getFeatures(e){return{functions:this.config.features?.functions??!1,streaming:this.config.features?.streaming??!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}getModelList(){return this.config.models}getMetrics(){return this.metrics}async chat(e,t){if(this.config.latencyMs&&await new Promise(n=>setTimeout(n,this.config.latencyMs)),this.config.shouldError)throw new Error(this.config.errorMessage??"Mock chat error");return this.updateMetrics("chat"),typeof this.config.chatResponse=="function"?await this.config.chatResponse(e):this.config.chatResponse??{results:[{index:0,content:"Mock response",finishReason:"stop"}],modelUsage:{ai:this.getName(),model:"mock-model",tokens:{promptTokens:10,completionTokens:5,totalTokens:15}}}}async embed(e,t){if(this.config.latencyMs&&await new Promise(n=>setTimeout(n,this.config.latencyMs)),this.config.shouldError)throw new Error(this.config.errorMessage??"Mock embed error");return this.updateMetrics("embed"),typeof this.config.embedResponse=="function"?this.config.embedResponse(e):this.config.embedResponse??{embeddings:[[.1,.2,.3]],modelUsage:{ai:this.getName(),model:"mock-model",tokens:{promptTokens:5,completionTokens:0,totalTokens:5}}}}setOptions(e){this.config.options=e}getOptions(){return this.config.options??{}}getLogger(){return this.config.options?.logger??(e=>{console.log(e)})}updateMetrics(e){let t=this.config.latencyMs??0;this.metrics.latency[e].samples.push(t);let n=this.metrics.latency[e].samples;if(this.metrics.latency[e].mean=n.reduce((r,o)=>r+o,0)/n.length,n.length>0){let r=[...n].sort((a,l)=>a-l),o=Math.max(0,Math.floor(r.length*.95)-1);this.metrics.latency[e].p95=r[o]??t;let i=Math.max(0,Math.floor(r.length*.99)-1);this.metrics.latency[e].p99=r[i]??t}if(this.config.shouldError){this.metrics.errors[e].count++,this.metrics.errors[e].total++;let r=this.metrics.latency[e].samples.length;this.metrics.errors[e].rate=r>0?this.metrics.errors[e].count/r:0}}};var nn=class s{options;lastUsedService;services=new Map;constructor(e){if(e.length===0)throw new Error("No AI services provided.");for(let[t,n]of e.entries())if("key"in n){if(this.services.has(n.key))throw new Error(`Duplicate model key: ${n.key}`);let{service:o,description:i,isInternal:a}=n;this.services.set(n.key,{service:o,description:i,isInternal:a})}else{let o=n.getModelList();if(!o)throw new Error(`Service ${t} \`${n.getName()}\` has no model list.`);for(let i of o){if(this.services.has(i.key)){let a=this.services.get(i.key)?.service;throw new Error(`Service ${t} \`${n.getName()}\` has duplicate model key: ${i.key} as service ${a?.getName()}`)}if("model"in i&&typeof i.model)this.services.set(i.key,{description:i.description,service:n,model:i.model});else if("embedModel"in i&&i.embedModel)this.services.set(i.key,{description:i.description,service:n,embedModel:i.embedModel});else throw new Error(`Key ${i.key} in model list for service ${t} \`${n.getName()}\` is missing a model or embedModel property.`)}}}static create(e){return new s(e)}getLastUsedChatModel(){return this.lastUsedService?.getLastUsedChatModel()}getLastUsedEmbedModel(){return this.lastUsedService?.getLastUsedEmbedModel()}getLastUsedModelConfig(){return this.lastUsedService?.getLastUsedModelConfig()}async chat(e,t){let n=e.model;if(!n)throw new Error("Model key must be specified for multi-service");let r=this.services.get(n);if(!r)throw new Error(`No service found for model key: ${n}`);if(this.lastUsedService=r.service,!r.model){let{model:o,...i}=e;return await r.service.chat(i,t)}return await r.service.chat({model:n,...e},t)}async embed(e,t){let n=e.embedModel;if(!n)throw new Error("Embed model key must be specified for multi-service");let r=this.services.get(n);if(!r)throw new Error(`No service found for embed model key: ${n}`);if(this.lastUsedService=r.service,!r.model){let{embedModel:o,...i}=e;return await r.service.embed(i,t)}return await r.service.embed({embedModel:n,...e},t)}getId(){return`MultiServiceRouter:${Array.from(this.services.values()).map(e=>e.service.getId()).join(",")}`}getName(){return"MultiServiceRouter"}getModelList(){return Array.from(this.services).filter(([,e])=>!e.isInternal).map(([e,t])=>{if(t.model)return{key:e,description:t.description,model:t.model};if(t.embedModel)return{key:e,description:t.description,embedModel:t.embedModel};throw new Error(`Service ${e} has no model or embedModel`)})}getFeatures(e){if(e){let t=this.services.get(e);if(t)return t.service.getFeatures(e)}return{functions:!1,streaming:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}getMetrics(){let e=this.lastUsedService;if(!e){let t=this.services.values().next().value;t&&(e="service"in t?t.service:t)}if(!e)throw new Error("No service available to get metrics.");return e.getMetrics()}setOptions(e){for(let t of this.services.values())t.service.setOptions(e);this.options=e}getOptions(){return this.options??{}}getLogger(){let e=this.lastUsedService;if(!e){let t=this.services.values().next().value;t&&(e=t.service)}if(!e)throw new Error("No service available to get logger.");return e.getLogger()}setServiceEntry(e,t){this.services.set(e,t)}};var yr=()=>structuredClone({...S(),model:"nous-hermes2",embedModel:"all-minilm"}),Yo=()=>structuredClone({..._(),model:"nous-hermes2",embedModel:"all-minilm"}),Me=class extends F{constructor({apiKey:e="not-set",url:t="http://localhost:11434/v1",config:n,options:r,models:o}){let i={...yr(),...n};super({apiKey:e,options:r,config:i,apiURL:t,models:o,modelInfo:[],supportFor:{functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}),super.setName("Ollama")}};var Pi=s=>["o1","o1-mini","o1-pro","o3","o3-mini","o3-pro","o4-mini"].includes(s),ke=class{constructor(e,t,n){this.config=e;this.streamingUsage=t;this.responsesReqUpdater=n}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,stopSequences:e.stopSequences,topP:e.topP,stream:e.stream}}mapInternalContentToResponsesInput(e){return e.map(n=>{if(n.type==="text")return{type:"text",text:n.text};if(n.type==="image")return{type:"image_url",image_url:{url:`data:${n.mimeType};base64,${n.image}`,details:n.details??"auto"}};if(n.type==="audio")return{type:"input_audio",input_audio:{data:n.data,format:n.format==="wav"?"wav":void 0}};let r=n;throw new Error(`Unsupported content part: ${JSON.stringify(r)}`)})}createResponsesReqInternalInput(e,t=!1){let n=[];for(let r of e){if(t&&r.role==="system")continue;let o;if(r.role==="system"||r.role==="user"||r.role==="assistant"&&r.content)if(typeof r.content=="string")o=r.content;else if(Array.isArray(r.content))o=this.mapInternalContentToResponsesInput(r.content);else{if(!(r.role==="assistant"&&!r.content&&r.functionCalls))throw new Error(`Invalid content type for role ${r.role}`);o=""}else r.role,o="";switch(r.role){case"system":n.push({type:"message",role:"system",content:o});break;case"user":n.push({type:"message",role:"user",content:o,name:r.name});break;case"assistant":if(r.content||r.functionCalls){let i={type:"message",role:"assistant",content:""};if(r.content&&(i.content=o),r.name&&(i.name=r.name),r.content&&n.push(i),r.functionCalls)for(let a of r.functionCalls)n.push({type:"function_call",call_id:a.id,name:a.function.name,arguments:typeof a.function.params=="object"?JSON.stringify(a.function.params):a.function.params||""})}break;case"function":n.push({type:"function_call_output",call_id:r.functionId,output:r.result});break;default:{let i=r.role;throw new Error(`Invalid role in chat prompt: ${i}`)}}}return n}createChatReq(e,t){let n=e.model,r={name:"/responses"},o=null,i=!1;if(e.chatPrompt){for(let A of e.chatPrompt)if(A.role==="system"&&typeof A.content=="string"){o=A.content,i=!0;break}}let a=o??this.config.systemPrompt??null,l=e.functions?.map(A=>({type:"function",name:A.name,description:A.description,parameters:A.parameters??{}})),p=[],c=Pi(n),u=this.config.reasoningSummary;t?.showThoughts?u||(u="auto"):u=void 0;let d=this.config.reasoningEffort;if(t?.thinkingTokenBudget)switch(t.thinkingTokenBudget){case"none":d=void 0;break;case"minimal":d="low";break;case"low":d="medium";break;case"medium":case"high":case"highest":d="high";break}let m={model:n,input:"",instructions:a,tools:l?.length?l:void 0,tool_choice:e.functionCall==="none"||e.functionCall==="auto"||e.functionCall==="required"?e.functionCall:typeof e.functionCall=="object"&&e.functionCall.function?{type:"function",name:e.functionCall.function.name}:void 0,...c?{max_output_tokens:e.modelConfig?.maxTokens??this.config.maxTokens??void 0}:{temperature:e.modelConfig?.temperature??this.config.temperature??void 0,top_p:e.modelConfig?.topP??this.config.topP??void 0,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty??void 0,frequency_penalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty??void 0},stream:e.modelConfig?.stream??this.config.stream??!1,background:void 0,include:p.length>0?p:void 0,metadata:void 0,parallel_tool_calls:this.config.parallelToolCalls,previous_response_id:void 0,...d?{reasoning:{effort:d,summary:u}}:{},service_tier:this.config.serviceTier,store:this.config.store,text:void 0,truncation:void 0,user:this.config.user,seed:this.config.seed};this.config.user&&(m.user=this.config.user),this.config.parallelToolCalls!==void 0&&(m.parallel_tool_calls=this.config.parallelToolCalls),this.config.responseFormat&&(m.text={format:{type:this.config.responseFormat}}),this.config.seed&&(m.seed=this.config.seed);let h=e.chatPrompt?this.createResponsesReqInternalInput(e.chatPrompt,i):[];if(h.length>0)m.input=h;else if(e.chatPrompt&&e.chatPrompt.length===1&&e.chatPrompt[0]?.role==="user"&&e.chatPrompt[0]?.content&&typeof e.chatPrompt[0].content=="string"&&!a)m.input=e.chatPrompt[0].content;else if(h.length===0&&!a)throw new Error("Responses API request must have input or instructions.");let g=m.reasoning??{};if(this.config.reasoningEffort&&(g={...g,effort:this.config.reasoningEffort}),t?.thinkingTokenBudget)switch(t.thinkingTokenBudget){case"none":g={};break;case"minimal":g={...g,effort:"low"};break;case"low":g={...g,effort:"medium"};break;case"medium":case"high":case"highest":g={...g,effort:"high"};break}Object.keys(g).length>0&&g.effort?m.reasoning=g:m.reasoning=void 0;let f=m;return this.responsesReqUpdater&&(f=this.responsesReqUpdater(f)),[r,f]}createChatResp(e){let{id:t,output:n,usage:r}=e;r&&(this.tokensUsed={promptTokens:r.prompt_tokens,completionTokens:r.completion_tokens,totalTokens:r.total_tokens});let o={};for(let i of n??[])switch(i.type){case"message":o.id=i.id,o.content=Ir(i.content,t),o.finishReason=i.status==="completed"?"stop":"content_filter";break;case"reasoning":o.id=i.id,i.encrypted_content?o.thought=i.encrypted_content:o.thought=i.summary.map(a=>typeof a=="object"?JSON.stringify(a):a).join(`
117
+ `);break;case"file_search_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:"file_search",params:{queries:i.queries,results:i.results}}}],o.finishReason="function_call";break;case"web_search_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:"web_search",params:{queries:i.queries}}}],o.finishReason="function_call";break;case"computer_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:"computer_use",params:{action:i.action}}}],o.finishReason="function_call";break;case"code_interpreter_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:"code_interpreter",params:{code:i.code,results:i.results}}}],o.finishReason="function_call";break;case"image_generation_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:"image_generation",params:{result:i.result}}}],o.finishReason="function_call";break;case"local_shell_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:"local_shell",params:{action:i.action}}}],o.finishReason="function_call";break;case"mcp_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:"mcp",params:{name:i.name,args:i.args,serverLabel:i.server_label,output:i.output,error:i.error}}}],o.finishReason="function_call";break;case"function_call":o.id=i.id,o.functionCalls=[{id:i.id,type:"function",function:{name:i.name,params:i.arguments}}],o.finishReason="function_call";break}return{results:[{...o,index:0}],remoteId:t}}createChatStreamResp(e){let t=e,n={index:0,id:"",content:"",finishReason:"stop"},r;switch(t.type){case"response.created":case"response.in_progress":case"response.queued":r=t.response.id,n.id=`${t.response.id}_res_0`;break;case"response.output_item.added":switch(t.item.type){case"message":n.id=t.item.id,n.content=Ir(t.item.content,t.item.id);break;case"function_call":n.id=t.item.id,n.functionCalls=[{id:t.item.id,type:"function",function:{name:t.item.name,params:t.item.arguments}}];break;case"file_search_call":{let o=t.item;n.id=t.item.id,n.functionCalls=[{id:o.id,type:"function",function:{name:"file_search",params:{queries:o.queries||[],results:o.results?.map(i=>({fileId:i.file_id,filename:i.filename,score:i.score,text:i.text,attributes:i.attributes}))}}}]}break;case"web_search_call":{let o=t.item;n.id=t.item.id,n.functionCalls=[{id:o.id,type:"function",function:{name:"web_search",params:{queries:o.queries||[]}}}]}break;case"computer_call":{let o=t.item;n.id=t.item.id,n.functionCalls=[{id:o.id,type:"function",function:{name:"computer_use",params:{action:o.action||{}}}}]}break;case"code_interpreter_call":{let o=t.item;n.id=t.item.id,n.functionCalls=[{id:o.id,type:"function",function:{name:"code_interpreter",params:{code:o.code||"",results:o.results}}}]}break;case"image_generation_call":{let o=t.item;n.id=t.item.id,n.functionCalls=[{id:o.id,type:"function",function:{name:"image_generation",params:{result:o.result}}}]}break;case"local_shell_call":{let o=t.item;n.id=t.item.id,n.functionCalls=[{id:o.id,type:"function",function:{name:"local_shell",params:{action:o.action||{}}}}]}break;case"mcp_call":{let o=t.item;n.id=t.item.id,n.functionCalls=[{id:o.id,type:"function",function:{name:"mcp",params:{name:o.name||"",args:o.args||"",serverLabel:o.server_label||"",output:o.output,error:o.error}}}]}break}break;case"response.content_part.added":n.id=t.item_id,n.content=Ir([t.part],t.item_id);break;case"response.output_text.delta":n.id=t.item_id,n.content=t.delta;break;case"response.output_text.done":break;case"response.function_call_arguments.delta":n.id=t.item_id,n.functionCalls=[{id:t.item_id,type:"function",function:{name:"",params:t.delta}}];break;case"response.reasoning_summary_text.delta":n.id=t.item_id,n.thought=t.delta;break;case"response.file_search_call.in_progress":case"response.file_search_call.searching":n.id=t.item_id,n.finishReason="function_call";break;case"response.file_search_call.completed":n.id=t.item_id,n.finishReason="function_call";break;case"response.web_search_call.in_progress":case"response.web_search_call.searching":n.id=t.item_id,n.finishReason="function_call";break;case"response.web_search_call.completed":n.id=t.item_id,n.finishReason="function_call";break;case"response.image_generation_call.in_progress":case"response.image_generation_call.generating":n.id=t.item_id,n.finishReason="function_call";break;case"response.image_generation_call.completed":n.id=t.item_id,n.finishReason="function_call";break;case"response.image_generation_call.partial_image":n.id=t.item_id,n.finishReason="function_call";break;case"response.mcp_call.in_progress":n.id=t.item_id,n.finishReason="function_call";break;case"response.mcp_call.arguments.delta":n.id=t.item_id,n.functionCalls=[{id:t.item_id,type:"function",function:{name:"",params:t.delta}}];break;case"response.mcp_call.arguments.done":n.id=t.item_id,n.functionCalls=[{id:t.item_id,type:"function",function:{name:"",params:t.arguments}}];break;case"response.mcp_call.completed":case"response.mcp_call.failed":n.id="mcp_call_event",n.finishReason="function_call";break;case"response.mcp_list_tools.in_progress":case"response.mcp_list_tools.completed":case"response.mcp_list_tools.failed":n.id="mcp_list_tools_event",n.finishReason="function_call";break;case"response.output_item.done":switch(t.item.type){case"message":n.id=t.item.id,n.finishReason=t.item.status==="completed"?"stop":"error";break;case"function_call":case"file_search_call":case"web_search_call":case"computer_call":case"code_interpreter_call":case"image_generation_call":case"local_shell_call":case"mcp_call":n.id=t.item.id,n.finishReason="function_call";break}break;case"response.completed":t.response.usage&&(this.tokensUsed={promptTokens:t.response.usage.prompt_tokens,completionTokens:t.response.usage.completion_tokens,totalTokens:t.response.usage.total_tokens}),r=t.response.id,n.id=`${t.response.id}_completed`,n.finishReason="stop";break;case"response.failed":r=t.response.id,n.id=`${t.response.id}_failed`,n.finishReason="error";break;case"response.incomplete":r=t.response.id,n.id=`${t.response.id}_incomplete`,n.finishReason="length";break;case"error":n.id="error",n.content=`Error: ${t.message}`,n.finishReason="error";break;default:n.id="unknown";break}return{results:[n],remoteId:r}}createEmbedReq(e){let t=e.embedModel;if(!t)throw new Error("Embed model not set");if(!e.texts||e.texts.length===0)throw new Error("Embed texts is empty");let n={name:"/embeddings"},r={model:t,input:e.texts,dimensions:this.config.dimensions};return[n,r]}},Ir=(s,e)=>{let t=s.filter(n=>n.type==="refusal");if(t.length>0){let n=t.map(r=>r.refusal).join(`
118
118
  `);throw new O(n,void 0,e)}return s.filter(n=>n.type==="output_text").map(n=>n.text).join(`
119
- `)};var pt=()=>({model:"gpt-4o",embedModel:"text-embedding-ada-002",temperature:.7,topP:1,stream:!0}),Qr=()=>({...pt(),model:"gpt-4o",temperature:.5}),Yr=()=>({...pt(),model:"gpt-4o",temperature:.9}),lt=class extends E{constructor({apiKey:e,config:t,options:n,apiURL:o,modelInfo:r=[],models:i,responsesReqUpdater:a,supportFor:l={functions:!0,streaming:!0,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}){if(!e||e==="")throw new Error("OpenAI API key not set");let p=new ke(t,n?.streamingUsage??!0,a),c=i;super(p,{name:"OpenAI",apiURL:o||"https://api.openai.com/v1",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:r,defaults:{model:t.model,embedModel:t.embedModel},options:n,supportFor:l,models:c})}},Ee=class extends lt{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("OpenAI API key not set");r=[...Lt,...r??[]];let i=a=>{let l=q({model:a,modelInfo:r,models:o});return{functions:!0,streaming:!0,hasThinkingBudget:l?.hasThinkingBudget??!1,hasShowThoughts:l?.hasShowThoughts??!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}};super({apiKey:e,config:{...pt(),...t},options:n,modelInfo:r,models:o,supportFor:i})}};async function nn(s,e,t={}){if(typeof s=="string")return[{type:"text",text:s}];if(!Array.isArray(s))return[{type:"text",text:String(s)}];let n=e.getFeatures(),o=[];for(let r of s)try{switch(r.type){case"text":o.push({type:"text",text:r.text});break;case"image":if(n.media.images.supported)r.altText?o.push({type:"text",text:`[Image: ${r.altText}]`}):o.push({type:"text",text:"[Image content]"});else if(r.altText)o.push({type:"text",text:r.altText});else if(t.imageToText)try{let i=await t.imageToText(r.image);o.push({type:"text",text:i})}catch(i){throw new V(i,"image","vision analysis")}else switch(t.fallbackBehavior){case"error":throw new z("Images",e.getName(),!1);case"skip":continue;default:o.push({type:"text",text:"[Image content not supported by this provider]"})}break;case"audio":if(n.media.audio.supported)r.transcription?o.push({type:"text",text:r.transcription}):o.push({type:"text",text:"[Audio content]"});else if(r.transcription)o.push({type:"text",text:r.transcription});else if(t.audioToText)try{let i=await t.audioToText(r.data,r.format);o.push({type:"text",text:i})}catch(i){throw new V(i,"audio","transcription")}else switch(t.fallbackBehavior){case"error":throw new z("Audio",e.getName(),!1);case"skip":continue;case"degrade":o.push({type:"text",text:"[Audio content not supported by this provider]"})}break;case"file":if(n.media.files.supported)r.extractedText?o.push({type:"text",text:r.extractedText}):o.push({type:"text",text:`[File: ${r.filename}]`});else if(r.extractedText)o.push({type:"text",text:r.extractedText});else if(t.fileToText)try{let i=await t.fileToText(r.data,r.mimeType);o.push({type:"text",text:i})}catch(i){throw new V(i,"file","text extraction")}else switch(t.fallbackBehavior){case"error":throw new z("Files",e.getName(),!1);case"skip":continue;default:o.push({type:"text",text:`[File: ${r.filename} - content not accessible by this provider]`})}break;case"url":if(n.media.urls.supported)r.cachedContent?o.push({type:"text",text:r.cachedContent}):o.push({type:"text",text:`[Link: ${r.url}${r.title?` - ${r.title}`:""}]`});else if(r.cachedContent)o.push({type:"text",text:r.cachedContent});else if(t.urlToText)try{let i=await t.urlToText(r.url);o.push({type:"text",text:i})}catch(i){throw new V(i,"url","content fetching")}else switch(t.fallbackBehavior){case"error":throw new z("URLs",e.getName(),!1);case"skip":continue;case"degrade":o.push({type:"text",text:`[Link: ${r.url}${r.title?` - ${r.title}`:""}]`})}break;default:typeof r=="object"&&r.text?o.push({type:"text",text:r.text}):o.push({type:"text",text:String(r)})}}catch(i){throw i instanceof z||i instanceof V?i:new V(i,r.type||"unknown","content processing")}return o}function Xr(s){let e=!1,t=!1,n=!1,o=!1;for(let r of s)if(r.role==="user"&&Array.isArray(r.content))for(let i of r.content)switch(i.type){case"image":e=!0;break;case"audio":t=!0;break;case"file":n=!0;break;case"url":o=!0;break}return{hasImages:e,hasAudio:t,hasFiles:n,hasUrls:o}}var ct=(n=>(n.RekaCore="reka-core",n.RekaFlash="reka-flash",n.RekaEdge="reka-edge",n))(ct||{});var on=[{name:"reka-core",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15},{name:"reka-flash",currency:"usd",promptTokenCostPer1M:.8,completionTokenCostPer1M:2},{name:"reka-edge",currency:"usd",promptTokenCostPer1M:.4,completionTokenCostPer1M:1}];var ut=()=>structuredClone({model:"reka-core",...w()}),es=()=>structuredClone({...ut(),model:"reka-core"}),ts=()=>structuredClone({model:"reka-core",..._()}),ns=()=>({...ut(),model:"reka-flash"}),Io=class{constructor(e){this.config=e}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,topP:e.topP,n:e.n,stream:e.stream}}createChatReq=e=>{let t=e.model;if(!e.chatPrompt||e.chatPrompt.length===0)throw new Error("Chat prompt is empty");let n={name:"/chat/completions"},o=Ei(e),r=e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,i=e.modelConfig?.stream??this.config.stream,a={model:t,messages:o,max_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,top_k:e.modelConfig?.n??this.config.n,top_p:e.modelConfig?.topP??this.config.topP??1,stop:e.modelConfig?.stopSequences??this.config.stop,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,...r?{frequency_penalty:r}:{},...i?{stream:!0}:{}};return[n,a]};createChatResp=e=>{let{id:t,usage:n,responses:o}=e;return this.tokensUsed=n?{promptTokens:n.input_tokens,completionTokens:n.output_tokens,totalTokens:n.input_tokens+n.output_tokens}:void 0,{results:o.map((i,a)=>{let l=Zr(i.finish_reason),p;return typeof i.message.content=="string"?p=i.message.content:p=i.message.content.text,{index:a,id:`${t}`,content:p,finishReason:l}}),remoteId:t}};createChatStreamResp=e=>{let{id:t,usage:n,responses:o}=e;return this.tokensUsed=n?{promptTokens:n.input_tokens,completionTokens:n.output_tokens,totalTokens:n.input_tokens+n.output_tokens}:void 0,{results:o.map((i,a)=>{let l=Zr(i.finish_reason),p;return typeof i.chunk.content=="string"?p=i.chunk.content:p=i.chunk.content.text,{index:a,id:`${t}`,content:p,finishReason:l}})}}},Zr=s=>{switch(s){case"stop":return"stop";case"context":return"length";case"length":return"length"}};function Ei(s){return s.chatPrompt.map(e=>{switch(e.role){case"system":return{role:"user",content:e.content};case"user":return Array.isArray(e.content)?{role:"user",content:e.content.map(t=>{switch(t.type){case"text":return{type:"text",text:t.text};case"image":throw new Error("Image type not supported");default:throw new Error("Invalid content type")}})}:{role:"user",content:e.content};case"assistant":if(Array.isArray(e.content))return{role:"assistant",content:e.content.map(t=>{switch(t.type){case"text":return{type:"text",text:t.text};case"image":throw new Error("Image type not supported");default:throw new Error("Invalid content type")}})};if(!e.content)throw new Error("Assistant content is empty");return{role:"user",content:e.content};default:throw new Error("Invalid role")}})}var Pe=class extends E{constructor({apiKey:e,config:t,options:n,apiURL:o,modelInfo:r=on,models:i}){if(!e||e==="")throw new Error("Reka API key not set");let a={...ut(),...t},l=new Io(a);super(l,{name:"Reka",apiURL:o||"https://api.reka.ai/v1/chat",headers:async()=>({"X-Api-Key":e}),modelInfo:r,defaults:{model:a.model},options:n,supportFor:{functions:!0,streaming:!0,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},models:i})}};var rn=class{providers;processingServices;config;constructor(e){this.providers=[e.providers.primary,...e.providers.alternatives],this.processingServices=e.processing,this.config=e.routing}async chat(e,t={}){let n=await this.selectProviderWithDegradation(e,t.routingOptions||{}),o=await this.preprocessRequest(e,n.provider,t.processingOptions);try{return{response:await n.provider.chat(o,t),routing:n}}catch(r){if(r instanceof z&&t.fallbackProviders?.length)return await this.tryFallbackProviders(e,t.fallbackProviders,t);throw r}}async preprocessRequest(e,t,n){let o={...n,fallbackBehavior:n?.fallbackBehavior||"degrade",imageToText:n?.imageToText||this.processingServices.imageToText,audioToText:n?.audioToText||this.processingServices.audioToText,fileToText:n?.fileToText||this.processingServices.fileToText,urlToText:n?.urlToText||this.processingServices.urlToText},r=[];for(let i of e.chatPrompt)if(i.role==="user"&&Array.isArray(i.content)){let a=await nn(i.content,t,o);a.every(p=>p.type==="text")&&a.length===1?r.push({...i,content:a[0].text}):r.push({...i,content:a.map(p=>({type:"text",text:p.text}))})}else r.push(i);return{...e,chatPrompt:r}}async selectProviderWithDegradation(e,t){let n=ue(e),o=[],r=[],i=[];try{let a=qt(e,this.providers,{requireExactMatch:t.requireExactMatch??this.config.capability.requireExactMatch,allowDegradation:t.allowDegradation??this.config.capability.allowDegradation}),l=a.getFeatures();return n.hasImages&&!l.media.images.supported&&(r.push("Images will be converted to text descriptions"),o.push("Image-to-text conversion")),n.hasAudio&&!l.media.audio.supported&&(r.push("Audio will be transcribed to text"),o.push("Audio-to-text transcription")),n.hasFiles&&!l.media.files.supported&&(r.push("File content will be extracted to text"),o.push("File-to-text extraction")),n.hasUrls&&!l.media.urls.supported&&(r.push("URL content will be pre-fetched"),o.push("URL content fetching")),n.requiresStreaming&&!l.streaming&&i.push("Streaming not supported - will use non-streaming mode"),n.requiresCaching&&!l.caching.supported&&i.push("Content caching not supported"),{provider:a,processingApplied:o,degradations:r,warnings:i}}catch(a){throw new Error(`Provider selection failed: ${a instanceof Error?a.message:"Unknown error"}`)}}async tryFallbackProviders(e,t,n){for(let o of t)try{let r={provider:o,processingApplied:["Fallback provider selection"],degradations:["Using fallback provider due to primary provider failure"],warnings:[]},i=await this.preprocessRequest(e,o,{fallbackBehavior:"degrade"});return{response:await o.chat(i,n),routing:r}}catch{}throw new Error("All fallback providers failed")}async getRoutingRecommendation(e){return await this.selectProviderWithDegradation(e,{})}async validateRequest(e){let t=ue(e),n=[],o=[];try{let r=await this.selectProviderWithDegradation(e,{});return r.degradations.length>0&&(n.push(...r.degradations),o.push("Consider using a provider that natively supports all media types")),r.warnings.length>0&&n.push(...r.warnings),t.hasImages&&this.processingServices.imageToText===void 0&&(this.providers.some(a=>a.getFeatures().media.images.supported)||(n.push("No image processing service available and no providers support images"),o.push("Add imageToText processing service or use image-capable provider"))),t.hasAudio&&this.processingServices.audioToText===void 0&&(this.providers.some(a=>a.getFeatures().media.audio.supported)||(n.push("No audio processing service available and no providers support audio"),o.push("Add audioToText processing service or use audio-capable provider"))),{canHandle:n.length===0||r.degradations.length>0,issues:n,recommendations:o}}catch(r){return{canHandle:!1,issues:[`Cannot route request: ${r instanceof Error?r.message:"Unknown error"}`],recommendations:["Add more providers or processing services to handle this request"]}}}getRoutingStats(){let e={};for(let t of this.providers){let n=t.getFeatures(),o=t.getName();n.functions&&(e.Functions=e.Functions||[],e.Functions.push(o)),n.streaming&&(e.Streaming=e.Streaming||[],e.Streaming.push(o)),n.media.images.supported&&(e.Images=e.Images||[],e.Images.push(o)),n.media.audio.supported&&(e.Audio=e.Audio||[],e.Audio.push(o)),n.media.files.supported&&(e.Files=e.Files||[],e.Files.push(o)),n.media.urls.supported&&(e.URLs=e.URLs||[],e.URLs.push(o)),n.caching.supported&&(e.Caching=e.Caching||[],e.Caching.push(o))}return{totalProviders:this.providers.length,capabilityMatrix:e,recommendedProvider:this.providers[0]?.getName()||"None"}}};var sn=[];var bo=()=>structuredClone({model:"mistralai/Mixtral-8x7B-Instruct-v0.1",...w()}),Fe=class extends F{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("Together API key not set");let i={...bo(),...t};r=[...sn,...r??[]];let a={functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0};super({apiKey:e,config:i,options:n,apiURL:"https://api.together.xyz/v1",modelInfo:r,models:o,supportFor:a}),super.setName("Together")}};function an(s){let e=n=>JSON.stringify(n,null,2);if(!s)throw new Error(`Chat request message item cannot be null or undefined, received: ${e(s)}`);let t=s?.role;if(!t)throw new Error(`Chat request message must have a role, received: ${e(t)}`);switch(t){case"system":{let n=s;if(!n.content||n.content.trim()==="")throw new Error(`System message content cannot be empty or whitespace-only, received: ${e(n.content)}`);break}case"user":{let n=s;if(!n.content)throw new Error(`User message content cannot be undefined, received: ${e(n.content)}`);if(typeof n.content=="string"){if(n.content.trim()==="")throw new Error(`User message content cannot be empty or whitespace-only, received: ${e(n.content)}`)}else if(Array.isArray(n.content)){if(n.content.length===0)throw new Error(`User message content array cannot be empty, received: ${e(n.content)}`);for(let o=0;o<n.content.length;o++){let r=n.content[o];if(!r||typeof r!="object")throw new Error(`User message content item at index ${o} must be an object, received: ${e(r)}`);let i=r?.type;if(!i)throw new Error(`User message content item at index ${o} must have a type, received: ${e(i)}`);switch(i){case"text":{let a=r;if(!a.text||a.text.trim()==="")throw new Error(`User message text content at index ${o} cannot be empty or whitespace-only, received: ${e(a.text)}`);break}case"image":{let a=r;if(!a.image||a.image.trim()==="")throw new Error(`User message image content at index ${o} cannot be empty, received: ${e(a.image)}`);if(!a.mimeType||a.mimeType.trim()==="")throw new Error(`User message image content at index ${o} must have a mimeType, received: ${e(a.mimeType)}`);break}case"audio":{let a=r;if(!a.data||a.data.trim()==="")throw new Error(`User message audio content at index ${o} cannot be empty, received: ${e(a.data)}`);break}default:throw new Error(`User message content item at index ${o} has unsupported type: ${e(i)}`)}}}else throw new Error(`User message content must be a string or array of content objects, received: ${e(n.content)}`);break}case"assistant":{let n=s;if(!n.content&&!n.functionCalls)throw new Error(`Assistant message must have either content or function calls, received content: ${e(n.content)}, functionCalls: ${e(n.functionCalls)}`);if(n.content&&typeof n.content!="string")throw new Error(`Assistant message content must be a string, received: ${e(n.content)}`);if(n.functionCalls&&!Array.isArray(n.functionCalls))throw new Error(`Assistant message function calls must be an array, received: ${e(n.functionCalls)}`);break}case"function":{let n=s;if(!n.functionId||n.functionId.trim()==="")throw new Error(`Function message must have a non-empty functionId, received: ${e(n.functionId)}`);if(n.result===void 0||n.result===null)throw new Error(`Function message must have a result, received: ${e(n.result)}`);if(typeof n.result!="string")throw new Error(`Function message result must be a string, received: ${e(n.result)}`);break}default:throw new Error(`Unsupported message role: ${e(t)}`)}}function ln(s){let e=n=>JSON.stringify(n,null,2),t=Array.isArray(s)?s:[s];if(t.length===0)throw new Error(`Chat response results cannot be empty, received: ${e(t)}`);for(let n=0;n<t.length;n++){let o=t[n];if(!o)throw new Error(`Chat response result at index ${n} cannot be null or undefined, received: ${e(o)}`);if(typeof o.index!="number")throw new Error(`Chat response result at index ${n} must have a numeric index, received: ${e(o.index)}`);if(o.index<0)throw new Error(`Chat response result at index ${n} must have a non-negative index, received: ${e(o.index)}`);if(!o.content&&!o.thought&&!o.functionCalls&&!o.finishReason)throw new Error(`Chat response result at index ${n} must have at least one of: content, thought, functionCalls, or finishReason, received: ${e({content:o.content,thought:o.thought,functionCalls:o.functionCalls,finishReason:o.finishReason})}`);if(o.content!==void 0&&typeof o.content!="string")throw new Error(`Chat response result content at index ${n} must be a string, received: ${e(o.content)}`);if(o.thought!==void 0&&typeof o.thought!="string")throw new Error(`Chat response result thought at index ${n} must be a string, received: ${e(o.thought)}`);if(o.name!==void 0){if(typeof o.name!="string")throw new Error(`Chat response result name at index ${n} must be a string, received: ${e(o.name)}`);if(o.name.trim()==="")throw new Error(`Chat response result name at index ${n} cannot be empty or whitespace-only, received: ${e(o.name)}`)}if(o.annotations!==void 0){if(!Array.isArray(o.annotations))throw new Error(`Chat response result annotations at index ${n} must be an array, received: ${e(o.annotations)}`);for(let r=0;r<o.annotations.length;r++){let i=o.annotations[r];if(!i||typeof i!="object")throw new Error(`Chat response result annotation at index ${n}[${r}] must be an object, received: ${e(i)}`);if(i.type!=="url_citation")throw new Error(`Chat response result annotation at index ${n}[${r}] must have type 'url_citation', received: ${e(i.type)}`);if(!i.url_citation||typeof i.url_citation!="object")throw new Error(`Chat response result annotation at index ${n}[${r}] must have a valid url_citation object, received: ${e(i.url_citation)}`);if(typeof i.url_citation.url!="string")throw new Error(`Chat response result annotation at index ${n}[${r}] url_citation.url must be a string, received: ${e(i.url_citation.url)}`)}}if(o.id!==void 0){if(typeof o.id!="string")throw new Error(`Chat response result id at index ${n} must be a string, received: ${e(o.id)}`);if(o.id.trim()==="")throw new Error(`Chat response result id at index ${n} cannot be empty or whitespace-only, received: ${e(o.id)}`)}if(o.functionCalls!==void 0){if(!Array.isArray(o.functionCalls))throw new Error(`Chat response result functionCalls at index ${n} must be an array, received: ${e(o.functionCalls)}`);for(let r=0;r<o.functionCalls.length;r++){let i=o.functionCalls[r];if(!i)throw new Error(`Function call at index ${r} in result ${n} cannot be null or undefined, received: ${e(i)}`);if(!i.id||typeof i.id!="string"||i.id.trim()==="")throw new Error(`Function call at index ${r} in result ${n} must have a non-empty string id, received: ${e(i.id)}`);if(i.type!=="function")throw new Error(`Function call at index ${r} in result ${n} must have type 'function', received: ${e(i.type)}`);if(!i.function)throw new Error(`Function call at index ${r} in result ${n} must have a function object, received: ${e(i.function)}`);if(!i.function.name||typeof i.function.name!="string"||i.function.name.trim()==="")throw new Error(`Function call at index ${r} in result ${n} must have a non-empty function name, received: ${e(i.function.name)}`);if(i.function.params!==void 0&&typeof i.function.params!="string"&&typeof i.function.params!="object")throw new Error(`Function call params at index ${r} in result ${n} must be a string or object, received: ${e(i.function.params)}`)}}if(o.finishReason!==void 0){let r=["stop","length","function_call","content_filter","error"];if(!r.includes(o.finishReason))throw new Error(`Chat response result finishReason at index ${n} must be one of: ${r.join(", ")}, received: ${e(o.finishReason)}`)}}}var dt=(d=>(d.Llama31_8B_Instruct="Llama-3.1-8B-Instruct-q4f32_1-MLC",d.Llama31_70B_Instruct="Llama-3.1-70B-Instruct-q4f16_1-MLC",d.Llama32_1B_Instruct="Llama-3.2-1B-Instruct-q4f32_1-MLC",d.Llama32_3B_Instruct="Llama-3.2-3B-Instruct-q4f32_1-MLC",d.Mistral7B_Instruct="Mistral-7B-Instruct-v0.3-q4f32_1-MLC",d.Phi35_Mini_Instruct="Phi-3.5-mini-instruct-q4f32_1-MLC",d.Gemma2_2B_Instruct="gemma-2-2b-it-q4f32_1-MLC",d.Gemma2_9B_Instruct="gemma-2-9b-it-q4f32_1-MLC",d.Qwen2_5_0_5B_Instruct="Qwen2.5-0.5B-Instruct-q4f32_1-MLC",d.Qwen2_5_1_5B_Instruct="Qwen2.5-1.5B-Instruct-q4f32_1-MLC",d.Qwen2_5_3B_Instruct="Qwen2.5-3B-Instruct-q4f32_1-MLC",d.Qwen2_5_7B_Instruct="Qwen2.5-7B-Instruct-q4f32_1-MLC",d))(dt||{});var pn=[{name:"Llama-3.1-8B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:4096},{name:"Llama-3.1-70B-Instruct-q4f16_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:4096,isExpensive:!0},{name:"Llama-3.2-1B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:2048},{name:"Llama-3.2-3B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:2048},{name:"Mistral-7B-Instruct-v0.3-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:4096},{name:"Phi-3.5-mini-instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:4096},{name:"gemma-2-2b-it-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:8192,maxTokens:2048},{name:"gemma-2-9b-it-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:8192,maxTokens:2048},{name:"Qwen2.5-0.5B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:2048},{name:"Qwen2.5-1.5B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:2048},{name:"Qwen2.5-3B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:2048},{name:"Qwen2.5-7B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:4096}];var Co=()=>structuredClone({model:"Llama-3.2-3B-Instruct-q4f32_1-MLC",...w()}),os=()=>structuredClone({model:"Llama-3.2-3B-Instruct-q4f32_1-MLC",..._()}),Ro=class{constructor(e,t){this.config=e;this.engine=t}tokensUsed;engine;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,endSequences:e.endSequences,stream:e.stream,n:e.n}}createChatReq(e){let t=e.model,n=e.chatPrompt.map(a=>{if(a.role==="function")return{role:"function",name:a.functionId,content:typeof a.result=="string"?a.result:JSON.stringify(a.result)};let l="";typeof a.content=="string"?l=a.content:Array.isArray(a.content)&&(l=a.content.filter(c=>c.type==="text").map(c=>c.text).join(`
120
- `));let p={role:a.role,content:l};return a.role==="assistant"&&a.functionCalls?.length?{...p,tool_calls:a.functionCalls.map(c=>({id:c.id,type:"function",function:{name:c.function.name,arguments:typeof c.function.params=="string"?c.function.params:JSON.stringify(c.function.params||{})}}))}:p}),o=e.functions?.map(a=>({type:"function",function:{name:a.name,description:a.description,parameters:a.parameters||{type:"object",properties:{}}}})),r={name:"/chat/completions",localCall:async(a,l)=>{try{let p=await this.engine.chat.completions.create({...a,stream:l||!1});return l?new ReadableStream({async start(c){try{for await(let u of p)c.enqueue(u);c.close()}catch(u){c.error(u)}}}):p}catch(p){throw new Error(`WebLLM API error: ${p}`)}}},i={model:t,messages:n,...o?.length?{tools:o}:{},max_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,top_p:e.modelConfig?.topP??this.config.topP,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,frequency_penalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,stop:e.modelConfig?.stopSequences??this.config.stopSequences,stream:e.modelConfig?.stream??this.config.stream,n:e.modelConfig?.n??this.config.n};return[r,i]}createEmbedReq=e=>{throw new Error("WebLLM does not support embeddings")};createChatResp=e=>(this.tokensUsed={promptTokens:e.usage?.prompt_tokens??0,completionTokens:e.usage?.completion_tokens??0,totalTokens:e.usage?.total_tokens??0},{results:e.choices.map((n,o)=>{let r="stop";switch(n.finish_reason){case"stop":r="stop";break;case"length":r="length";break;case"tool_calls":r="function_call";break;case"content_filter":r="content_filter";break;default:r="stop";break}let i=n.message.tool_calls?.map(a=>({id:a.id,type:"function",function:{name:a.function.name,params:a.function.arguments}}));return{index:o,id:e.id,content:n.message.content||"",functionCalls:i,finishReason:r}}),remoteId:e.id});createChatStreamResp=(e,t)=>{let n=t,o=e.choices[0];if(!o)throw new Error("No choice in WebLLM stream response");if(o.delta.content&&(n.content=(n.content||"")+o.delta.content),o.delta.tool_calls){n.toolCalls||(n.toolCalls=[]);for(let l of o.delta.tool_calls){let p=n.toolCalls[l.index];p?l.function?.arguments&&(p.function.arguments=(p.function?.arguments||"")+l.function.arguments):n.toolCalls[l.index]={id:l.id,type:l.type,function:{name:l.function?.name,arguments:l.function?.arguments||""}}}}e.usage&&(this.tokensUsed={promptTokens:e.usage.prompt_tokens,completionTokens:e.usage.completion_tokens,totalTokens:e.usage.total_tokens});let r;if(o.finish_reason)switch(o.finish_reason){case"stop":r="stop";break;case"length":r="length";break;case"tool_calls":r="function_call";break;case"content_filter":r="content_filter";break;default:r="stop";break}let i=n.toolCalls?.map(l=>({id:l.id||"",type:"function",function:{name:l.function?.name||"",params:l.function?.arguments||""}}));return{results:[{index:0,id:e.id,content:n.content||"",functionCalls:i,finishReason:r}],remoteId:e.id}};createEmbedResp(e){throw new Error("WebLLM does not support embeddings")}},_e=class extends E{constructor({engine:e,config:t,options:n,models:o}){if(!e)throw new Error("WebLLM engine instance is required");let r={...Co(),...t},i=new Ro(r,e);super(i,{name:"WebLLM",apiURL:void 0,headers:async()=>({}),modelInfo:pn,defaults:{model:r.model},supportFor:a=>({functions:!0,streaming:!0,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}),options:n,models:o})}};var mt=(o=>(o.Grok3="grok-3",o.Grok3Mini="grok-3-mini",o.Grok3Fast="grok-3-fast",o.Grok3MiniFast="grok-3-mini-fast",o))(mt||{}),To=(e=>(e.GrokEmbedSmall="grok-embed-small",e))(To||{});var cn=[{name:"grok-3",currency:"USD",promptTokenCostPer1M:3,completionTokenCostPer1M:15},{name:"grok-3-mini",currency:"USD",promptTokenCostPer1M:.3,completionTokenCostPer1M:.5,hasThinkingBudget:!0},{name:"grok-3-fast",currency:"USD",promptTokenCostPer1M:5,completionTokenCostPer1M:25},{name:"grok-3-mini-fast",currency:"USD",promptTokenCostPer1M:.6,completionTokenCostPer1M:4,hasThinkingBudget:!0}];var un=()=>structuredClone({model:"grok-3-mini",...w()}),rs=()=>structuredClone({...un(),model:"grok-3"}),De=class extends F{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("Grok API key not set");let i={...un(),...t};r=[...cn,...r??[]];let a=p=>{let c=q({model:p,modelInfo:r,models:o});return{functions:!0,streaming:!0,hasThinkingBudget:c?.hasThinkingBudget??!1,hasShowThoughts:c?.hasShowThoughts??!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}},l=p=>{if(n?.searchParameters){let c=n.searchParameters;return{...p,search_parameters:{mode:c.mode,return_citations:c.returnCitations,from_date:c.fromDate,to_date:c.toDate,max_search_results:c.maxSearchResults,sources:c.sources?.map(u=>({type:u.type,country:u.country,excluded_websites:u.excludedWebsites,allowed_websites:u.allowedWebsites,safe_search:u.safeSearch,x_handles:u.xHandles,links:u.links}))}}}return p};super({apiKey:e,config:i,options:n,apiURL:"https://api.x.ai/v1",modelInfo:r,models:o,supportFor:a,chatReqUpdater:l}),super.setName("Grok")}};function ss(s){return gt.create(s)}var gt=class s{ai;static create(e){return new s(e)}constructor(e){switch(e.name){case"openai":this.ai=new Ie(e);break;case"openai-responses":this.ai=new Ee(e);break;case"azure-openai":this.ai=new be(e);break;case"grok":this.ai=new De(e);break;case"huggingface":this.ai=new ve(e);break;case"groq":this.ai=new Se(e);break;case"together":this.ai=new Fe(e);break;case"cohere":this.ai=new Re(e);break;case"google-gemini":this.ai=new Te(e);break;case"anthropic":this.ai=new Ae(e);break;case"mistral":this.ai=new Oe(e);break;case"deepseek":this.ai=new Ce(e);break;case"ollama":this.ai=new Me(e);break;case"reka":this.ai=new Pe(e);break;case"webllm":this.ai=new _e(e);break;default:throw new Error("Unknown AI")}}getName(){return this.ai.getName()}getId(){return this.ai.getId()}getFeatures(e){return this.ai.getFeatures(e)}getModelList(){return this.ai.getModelList()}getLastUsedChatModel(){return this.ai.getLastUsedChatModel()}getLastUsedEmbedModel(){return this.ai.getLastUsedEmbedModel()}getLastUsedModelConfig(){return this.ai.getLastUsedModelConfig()}getMetrics(){return this.ai.getMetrics()}async chat(e,t){return await this.ai.chat(e,t)}async embed(e,t){return await this.ai.embed(e,t)}setOptions(e){this.ai.setOptions(e)}getOptions(){return this.ai.getOptions()}getLogger(){return this.ai.getLogger()}};var dn=require("@opentelemetry/api");var j=class{name;fetch;tracer;_upsert;_batchUpsert;_query;constructor({name:e,fetch:t,tracer:n}){this.name=e,this.fetch=t,this.tracer=n}async upsert(e,t){if(!this._upsert)throw new Error("upsert() not implemented");return this.tracer?await this.tracer.startActiveSpan("DB Upsert Request",{kind:dn.SpanKind.SERVER,attributes:{[T.DB_SYSTEM]:this.name,[T.DB_OPERATION_NAME]:"upsert",[T.DB_TABLE]:e.table,[T.DB_NAMESPACE]:e.namespace,[T.DB_OPERATION_NAME]:t?"update":"insert"}},async n=>{try{return await this._upsert(e,t,{span:n})}finally{n.end()}}):await this._upsert(e,t)}async batchUpsert(e,t){if(!this._batchUpsert)throw new Error("batchUpsert() not implemented");if(e.length===0)throw new Error("Batch request is empty");if(!e[0])throw new Error("Batch request is invalid first element is undefined");return this.tracer?await this.tracer.startActiveSpan("DB Batch Upsert Request",{kind:dn.SpanKind.SERVER,attributes:{[T.DB_SYSTEM]:this.name,[T.DB_OPERATION_NAME]:"upsert",[T.DB_TABLE]:e[0].table,[T.DB_NAMESPACE]:e[0].namespace,[T.DB_OPERATION_NAME]:t?"update":"insert"}},async n=>{try{return await this._batchUpsert(e,t,{span:n})}finally{n.end()}}):await this._batchUpsert(e,t)}async query(e){if(!this._query)throw new Error("query() not implemented");return this.tracer?await this.tracer.startActiveSpan("DB Query Request",{kind:dn.SpanKind.SERVER,attributes:{[T.DB_SYSTEM]:this.name,[T.DB_OPERATION_NAME]:"upsert",[T.DB_TABLE]:e.table,[T.DB_NAMESPACE]:e.namespace,[T.DB_OPERATION_NAME]:"query"}},async t=>{try{return await this._query(e,{span:t})}finally{t.end()}}):await this._query(e)}};var wo="https://api.cloudflare.com/client/v4/accounts/",Le=class extends j{apiKey;accountId;constructor({apiKey:e,accountId:t,fetch:n,tracer:o}){if(!e||!t)throw new Error("Cloudflare credentials not set");super({name:"Cloudflare",fetch:n,tracer:o}),this.apiKey=e,this.accountId=t}_upsert=async(e,t,n)=>{let o=await B({url:new URL(`${this.accountId}/vectorize/indexes/${e.table}/upsert`,wo),headers:{"X-Auth-Key":this.apiKey},fetch:this.fetch,span:n?.span},{id:e.id,values:e.values,namespace:e.namespace,metadata:e.metadata});if(o.errors)throw new Error(`Cloudflare upsert failed: ${o.errors.map(({message:r})=>r).join(", ")}`);return{ids:o.result.ids}};batchUpsert=async(e,t,n)=>{if(t)throw new Error("Weaviate does not support batch update");if(e.length<1)throw new Error("Batch request is empty");if(!e[0]||!e[0].table)throw new Error("Table name is empty");let o=e[0].table,r=await B({url:new URL(`${this.accountId}/vectorize/indexes/${o}/upsert`,wo),headers:{"X-Auth-Key":this.apiKey},fetch:this.fetch,span:n?.span},e.map(i=>({id:i.id,values:i.values,namespace:i.namespace,metadata:i.metadata})));if(r.errors)throw new Error(`Cloudflare batch upsert failed: ${r.errors.map(({message:i})=>i).join(", ")}`);return{ids:r.result.ids}};query=async(e,t)=>{let n=await B({url:new URL(`${this.accountId}/vectorize/indexes/${e.table}/query`,wo),headers:{"X-Auth-Key":this.apiKey},fetch:this.fetch,span:t?.span},{vector:e.values,topK:e.limit||10,returnValues:!0});if(n.errors)throw new Error(`Cloudflare query failed: ${n.errors.map(({message:r})=>r).join(", ")}`);return{matches:n.result.matches.map(({id:r,score:i,values:a,metadata:l})=>({id:r,score:i,values:a,metadata:l}))}}};var oe=class extends j{state;constructor({tracer:e}={}){super({name:"Memory",tracer:e}),this.state={}}_upsert=async(e,t,n)=>{if(!this.state[e.table])this.state[e.table]={[e.id]:e};else{let o=this.state[e.table];if(!o)throw new Error(`Table not found: ${e.table}`);o[e.id]=e}return{ids:[e.id]}};_batchUpsert=async(e,t,n)=>{let o=[];for(let r of e){let i=await this.upsert(r,t);o.push(...i.ids)}return{ids:o}};_query=async(e,t)=>{let n=this.state[e.table];if(!n)return{matches:[]};let o=[];return Object.entries(n).forEach(([r,i])=>{if(e.values&&i.values){let a=Pi(e.values,i.values);o.push({id:r,score:a,metadata:i.metadata})}}),o.sort((r,i)=>r.score-i.score),e.limit&&(o.length=e.limit),{matches:o}};getDB=()=>structuredClone(this.state);setDB=e=>{this.state=structuredClone(e)};clearDB=()=>{this.state={}}},Pi=(s,e)=>{if(s.length!==e.length)throw new Error("Vectors must be of the same length.");let t=0,n=0,o=0,r=!0,i=!0,a=new Float64Array(s),l=new Float64Array(e);for(let d=0;d<a.length;d++)t+=a[d]*l[d],n+=a[d]*a[d],o+=l[d]*l[d],a[d]!==0&&(r=!1),l[d]!==0&&(i=!1);if(r||i)return 1;let p=Math.sqrt(n),c=Math.sqrt(o);return 1-t/(p*c)};var Fi=s=>({namespace:s.namespace,topK:s.limit||10,filter:{},includeValues:!0,includeMetadata:!0,vector:s.values??[],id:s.id}),Ge=class extends j{apiKey;apiURL;constructor({apiKey:e,host:t,fetch:n,tracer:o}){if(!e||e==="")throw new Error("Pinecone API key not set");super({name:"Pinecone",fetch:n,tracer:o}),this.apiKey=e,this.apiURL=t}_upsert=async(e,t,n)=>(await this._batchUpsert([e],t,n),{ids:[e.id]});_batchUpsert=async(e,t,n)=>{if(e.length===0)throw new Error("Batch request is empty");return await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/vectors/upsert",fetch:this.fetch,span:n?.span},e.map(({id:o,values:r=[],metadata:i})=>({id:o,values:r,metadata:i}))),{ids:e.map(({id:o})=>o)}};query=async(e,t)=>{if(e.text)throw new Error("Pinecone does not support text");return{matches:(await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/query",fetch:this.fetch,span:t?.span},Fi(e))).matches.map(({id:r,score:i,values:a,metadata:l})=>({id:r,score:i,metadata:l,values:a}))}}};var Ne=class extends j{apiKey;apiURL;constructor({apiKey:e,host:t,fetch:n,tracer:o}){if(!e||e==="")throw new Error("Weaviate API key not set");super({name:"Weaviate",fetch:n,tracer:o}),this.apiKey=e,this.apiURL=t}_upsert=async(e,t,n)=>{let o=await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:`/v1/objects/${e.table}/${e.id}`,put:!!t,fetch:this.fetch,span:n?.span},{id:e.id,class:e.table,tenant:e.namespace,vector:e.values,properties:e.metadata??{}});if(o?.result?.errors)throw new Error(`Weaviate upsert failed: ${o.result.errors.error.map(({message:r})=>r).join(", ")}`);return{ids:[o.id]}};_batchUpsert=async(e,t,n)=>{if(t)throw new Error("Weaviate does not support batch update");if(e.length===0)throw new Error("Batch request is empty");let o=e.map(i=>({id:i.id,class:i.table,tenant:i.namespace,vector:i.values,properties:i.metadata??{}})),r=await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/v1/batch/objects",fetch:this.fetch,span:n?.span},{objects:o});if(r?.some(({result:i})=>i?.errors))throw new Error(`Weaviate batch upsert failed: ${r.map(({result:i})=>i?.errors?.error.map(({message:a})=>a).join(", ")).join(", ")}`);return{ids:r.map(({id:i})=>i)}};_query=async(e,t)=>{let n="";if(e.columns&&e.columns.length===0)throw new Error("Weaviate requires at least one column");if(e.values)n=`nearVector: {
119
+ `)};var pt=()=>({model:"gpt-4o",embedModel:"text-embedding-ada-002",temperature:.7,topP:1,stream:!0}),Xo=()=>({...pt(),model:"gpt-4o",temperature:.5}),Zo=()=>({...pt(),model:"gpt-4o",temperature:.9}),lt=class extends E{constructor({apiKey:e,config:t,options:n,apiURL:r,modelInfo:o=[],models:i,responsesReqUpdater:a,supportFor:l={functions:!0,streaming:!0,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}}){if(!e||e==="")throw new Error("OpenAI API key not set");let p=new ke(t,n?.streamingUsage??!0,a),c=i;super(p,{name:"OpenAI",apiURL:r||"https://api.openai.com/v1",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:o,defaults:{model:t.model,embedModel:t.embedModel},options:n,supportFor:l,models:c})}},Ee=class extends lt{constructor({apiKey:e,config:t,options:n,models:r,modelInfo:o}){if(!e||e==="")throw new Error("OpenAI API key not set");o=[...Gt,...o??[]];let i=a=>{let l=q({model:a,modelInfo:o,models:r});return{functions:!0,streaming:!0,hasThinkingBudget:l?.hasThinkingBudget??!1,hasShowThoughts:l?.hasShowThoughts??!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}};super({apiKey:e,config:{...pt(),...t},options:n,modelInfo:o,models:r,supportFor:i})}};async function rn(s,e,t={}){if(typeof s=="string")return[{type:"text",text:s}];if(!Array.isArray(s))return[{type:"text",text:String(s)}];let n=e.getFeatures(),r=[];for(let o of s)try{switch(o.type){case"text":r.push({type:"text",text:o.text});break;case"image":if(n.media.images.supported)o.altText?r.push({type:"text",text:`[Image: ${o.altText}]`}):r.push({type:"text",text:"[Image content]"});else if(o.altText)r.push({type:"text",text:o.altText});else if(t.imageToText)try{let i=await t.imageToText(o.image);r.push({type:"text",text:i})}catch(i){throw new V(i,"image","vision analysis")}else switch(t.fallbackBehavior){case"error":throw new z("Images",e.getName(),!1);case"skip":continue;default:r.push({type:"text",text:"[Image content not supported by this provider]"})}break;case"audio":if(n.media.audio.supported)o.transcription?r.push({type:"text",text:o.transcription}):r.push({type:"text",text:"[Audio content]"});else if(o.transcription)r.push({type:"text",text:o.transcription});else if(t.audioToText)try{let i=await t.audioToText(o.data,o.format);r.push({type:"text",text:i})}catch(i){throw new V(i,"audio","transcription")}else switch(t.fallbackBehavior){case"error":throw new z("Audio",e.getName(),!1);case"skip":continue;case"degrade":r.push({type:"text",text:"[Audio content not supported by this provider]"})}break;case"file":if(n.media.files.supported)o.extractedText?r.push({type:"text",text:o.extractedText}):r.push({type:"text",text:`[File: ${o.filename}]`});else if(o.extractedText)r.push({type:"text",text:o.extractedText});else if(t.fileToText)try{let i=await t.fileToText(o.data,o.mimeType);r.push({type:"text",text:i})}catch(i){throw new V(i,"file","text extraction")}else switch(t.fallbackBehavior){case"error":throw new z("Files",e.getName(),!1);case"skip":continue;default:r.push({type:"text",text:`[File: ${o.filename} - content not accessible by this provider]`})}break;case"url":if(n.media.urls.supported)o.cachedContent?r.push({type:"text",text:o.cachedContent}):r.push({type:"text",text:`[Link: ${o.url}${o.title?` - ${o.title}`:""}]`});else if(o.cachedContent)r.push({type:"text",text:o.cachedContent});else if(t.urlToText)try{let i=await t.urlToText(o.url);r.push({type:"text",text:i})}catch(i){throw new V(i,"url","content fetching")}else switch(t.fallbackBehavior){case"error":throw new z("URLs",e.getName(),!1);case"skip":continue;case"degrade":r.push({type:"text",text:`[Link: ${o.url}${o.title?` - ${o.title}`:""}]`})}break;default:typeof o=="object"&&o.text?r.push({type:"text",text:o.text}):r.push({type:"text",text:String(o)})}}catch(i){throw i instanceof z||i instanceof V?i:new V(i,o.type||"unknown","content processing")}return r}function es(s){let e=!1,t=!1,n=!1,r=!1;for(let o of s)if(o.role==="user"&&Array.isArray(o.content))for(let i of o.content)switch(i.type){case"image":e=!0;break;case"audio":t=!0;break;case"file":n=!0;break;case"url":r=!0;break}return{hasImages:e,hasAudio:t,hasFiles:n,hasUrls:r}}var ct=(n=>(n.RekaCore="reka-core",n.RekaFlash="reka-flash",n.RekaEdge="reka-edge",n))(ct||{});var on=[{name:"reka-core",currency:"usd",promptTokenCostPer1M:3,completionTokenCostPer1M:15},{name:"reka-flash",currency:"usd",promptTokenCostPer1M:.8,completionTokenCostPer1M:2},{name:"reka-edge",currency:"usd",promptTokenCostPer1M:.4,completionTokenCostPer1M:1}];var ut=()=>structuredClone({model:"reka-core",...S()}),ns=()=>structuredClone({...ut(),model:"reka-core"}),rs=()=>structuredClone({model:"reka-core",..._()}),os=()=>({...ut(),model:"reka-flash"}),br=class{constructor(e){this.config=e}tokensUsed;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,topP:e.topP,n:e.n,stream:e.stream}}createChatReq=e=>{let t=e.model;if(!e.chatPrompt||e.chatPrompt.length===0)throw new Error("Chat prompt is empty");let n={name:"/chat/completions"},r=Fi(e),o=e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,i=e.modelConfig?.stream??this.config.stream,a={model:t,messages:r,max_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,top_k:e.modelConfig?.n??this.config.n,top_p:e.modelConfig?.topP??this.config.topP??1,stop:e.modelConfig?.stopSequences??this.config.stop,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,...o?{frequency_penalty:o}:{},...i?{stream:!0}:{}};return[n,a]};createChatResp=e=>{let{id:t,usage:n,responses:r}=e;return this.tokensUsed=n?{promptTokens:n.input_tokens,completionTokens:n.output_tokens,totalTokens:n.input_tokens+n.output_tokens}:void 0,{results:r.map((i,a)=>{let l=ts(i.finish_reason),p;return typeof i.message.content=="string"?p=i.message.content:p=i.message.content.text,{index:a,id:`${t}`,content:p,finishReason:l}}),remoteId:t}};createChatStreamResp=e=>{let{id:t,usage:n,responses:r}=e;return this.tokensUsed=n?{promptTokens:n.input_tokens,completionTokens:n.output_tokens,totalTokens:n.input_tokens+n.output_tokens}:void 0,{results:r.map((i,a)=>{let l=ts(i.finish_reason),p;return typeof i.chunk.content=="string"?p=i.chunk.content:p=i.chunk.content.text,{index:a,id:`${t}`,content:p,finishReason:l}})}}},ts=s=>{switch(s){case"stop":return"stop";case"context":return"length";case"length":return"length"}};function Fi(s){return s.chatPrompt.map(e=>{switch(e.role){case"system":return{role:"user",content:e.content};case"user":return Array.isArray(e.content)?{role:"user",content:e.content.map(t=>{switch(t.type){case"text":return{type:"text",text:t.text};case"image":throw new Error("Image type not supported");default:throw new Error("Invalid content type")}})}:{role:"user",content:e.content};case"assistant":if(Array.isArray(e.content))return{role:"assistant",content:e.content.map(t=>{switch(t.type){case"text":return{type:"text",text:t.text};case"image":throw new Error("Image type not supported");default:throw new Error("Invalid content type")}})};if(!e.content)throw new Error("Assistant content is empty");return{role:"user",content:e.content};default:throw new Error("Invalid role")}})}var Pe=class extends E{constructor({apiKey:e,config:t,options:n,apiURL:r,modelInfo:o=on,models:i}){if(!e||e==="")throw new Error("Reka API key not set");let a={...ut(),...t},l=new br(a);super(l,{name:"Reka",apiURL:r||"https://api.reka.ai/v1/chat",headers:async()=>({"X-Api-Key":e}),modelInfo:o,defaults:{model:a.model},options:n,supportFor:{functions:!0,streaming:!0,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0},models:i})}};var sn=class{providers;processingServices;config;constructor(e){this.providers=[e.providers.primary,...e.providers.alternatives],this.processingServices=e.processing,this.config=e.routing}async chat(e,t={}){let n=await this.selectProviderWithDegradation(e,t.routingOptions||{}),r=await this.preprocessRequest(e,n.provider,t.processingOptions);try{return{response:await n.provider.chat(r,t),routing:n}}catch(o){if(o instanceof z&&t.fallbackProviders?.length)return await this.tryFallbackProviders(e,t.fallbackProviders,t);throw o}}async preprocessRequest(e,t,n){let r={...n,fallbackBehavior:n?.fallbackBehavior||"degrade",imageToText:n?.imageToText||this.processingServices.imageToText,audioToText:n?.audioToText||this.processingServices.audioToText,fileToText:n?.fileToText||this.processingServices.fileToText,urlToText:n?.urlToText||this.processingServices.urlToText},o=[];for(let i of e.chatPrompt)if(i.role==="user"&&Array.isArray(i.content)){let a=await rn(i.content,t,r);a.every(p=>p.type==="text")&&a.length===1?o.push({...i,content:a[0].text}):o.push({...i,content:a.map(p=>({type:"text",text:p.text}))})}else o.push(i);return{...e,chatPrompt:o}}async selectProviderWithDegradation(e,t){let n=ue(e),r=[],o=[],i=[];try{let a=zt(e,this.providers,{requireExactMatch:t.requireExactMatch??this.config.capability.requireExactMatch,allowDegradation:t.allowDegradation??this.config.capability.allowDegradation}),l=a.getFeatures();return n.hasImages&&!l.media.images.supported&&(o.push("Images will be converted to text descriptions"),r.push("Image-to-text conversion")),n.hasAudio&&!l.media.audio.supported&&(o.push("Audio will be transcribed to text"),r.push("Audio-to-text transcription")),n.hasFiles&&!l.media.files.supported&&(o.push("File content will be extracted to text"),r.push("File-to-text extraction")),n.hasUrls&&!l.media.urls.supported&&(o.push("URL content will be pre-fetched"),r.push("URL content fetching")),n.requiresStreaming&&!l.streaming&&i.push("Streaming not supported - will use non-streaming mode"),n.requiresCaching&&!l.caching.supported&&i.push("Content caching not supported"),{provider:a,processingApplied:r,degradations:o,warnings:i}}catch(a){throw new Error(`Provider selection failed: ${a instanceof Error?a.message:"Unknown error"}`)}}async tryFallbackProviders(e,t,n){for(let r of t)try{let o={provider:r,processingApplied:["Fallback provider selection"],degradations:["Using fallback provider due to primary provider failure"],warnings:[]},i=await this.preprocessRequest(e,r,{fallbackBehavior:"degrade"});return{response:await r.chat(i,n),routing:o}}catch{}throw new Error("All fallback providers failed")}async getRoutingRecommendation(e){return await this.selectProviderWithDegradation(e,{})}async validateRequest(e){let t=ue(e),n=[],r=[];try{let o=await this.selectProviderWithDegradation(e,{});return o.degradations.length>0&&(n.push(...o.degradations),r.push("Consider using a provider that natively supports all media types")),o.warnings.length>0&&n.push(...o.warnings),t.hasImages&&this.processingServices.imageToText===void 0&&(this.providers.some(a=>a.getFeatures().media.images.supported)||(n.push("No image processing service available and no providers support images"),r.push("Add imageToText processing service or use image-capable provider"))),t.hasAudio&&this.processingServices.audioToText===void 0&&(this.providers.some(a=>a.getFeatures().media.audio.supported)||(n.push("No audio processing service available and no providers support audio"),r.push("Add audioToText processing service or use audio-capable provider"))),{canHandle:n.length===0||o.degradations.length>0,issues:n,recommendations:r}}catch(o){return{canHandle:!1,issues:[`Cannot route request: ${o instanceof Error?o.message:"Unknown error"}`],recommendations:["Add more providers or processing services to handle this request"]}}}getRoutingStats(){let e={};for(let t of this.providers){let n=t.getFeatures(),r=t.getName();n.functions&&(e.Functions=e.Functions||[],e.Functions.push(r)),n.streaming&&(e.Streaming=e.Streaming||[],e.Streaming.push(r)),n.media.images.supported&&(e.Images=e.Images||[],e.Images.push(r)),n.media.audio.supported&&(e.Audio=e.Audio||[],e.Audio.push(r)),n.media.files.supported&&(e.Files=e.Files||[],e.Files.push(r)),n.media.urls.supported&&(e.URLs=e.URLs||[],e.URLs.push(r)),n.caching.supported&&(e.Caching=e.Caching||[],e.Caching.push(r))}return{totalProviders:this.providers.length,capabilityMatrix:e,recommendedProvider:this.providers[0]?.getName()||"None"}}};var an=[];var Rr=()=>structuredClone({model:"mistralai/Mixtral-8x7B-Instruct-v0.1",...S()}),Fe=class extends F{constructor({apiKey:e,config:t,options:n,models:r,modelInfo:o}){if(!e||e==="")throw new Error("Together API key not set");let i={...Rr(),...t};o=[...an,...o??[]];let a={functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0};super({apiKey:e,config:i,options:n,apiURL:"https://api.together.xyz/v1",modelInfo:o,models:r,supportFor:a}),super.setName("Together")}};function ln(s){let e=n=>JSON.stringify(n,null,2);if(!s)throw new Error(`Chat request message item cannot be null or undefined, received: ${e(s)}`);let t=typeof s=="object"&&s!==null&&"role"in s&&typeof s.role=="string"?s.role:void 0;if(!t)throw new Error(`Chat request message must have a role, received: ${e(t)}`);switch(t){case"system":{let n=typeof s=="object"&&s!==null&&"content"in s&&typeof s.content=="string"?s.content:void 0;if(!n||n.trim()==="")throw new Error(`System message content cannot be empty or whitespace-only, received: ${e(n)}`);break}case"user":{let n=typeof s=="object"&&s!==null&&"content"in s?s.content:void 0;if(!n)throw new Error(`User message content cannot be undefined, received: ${e(n)}`);if(typeof n=="string"){if(n.trim()==="")throw new Error(`User message content cannot be empty or whitespace-only, received: ${e(n)}`)}else if(Array.isArray(n)){if(n.length===0)throw new Error(`User message content array cannot be empty, received: ${e(n)}`);for(let r=0;r<n.length;r++){let o=n[r];if(!o||typeof o!="object")throw new Error(`User message content item at index ${r} must be an object, received: ${e(o)}`);let i=typeof o=="object"&&o!==null&&"type"in o&&typeof o.type=="string"?o.type:void 0;if(!i)throw new Error(`User message content item at index ${r} must have a type, received: ${e(i)}`);switch(i){case"text":{let a="text"in o&&typeof o.text=="string"?o.text:void 0;if(!a||a.trim()==="")throw new Error(`User message text content at index ${r} cannot be empty or whitespace-only, received: ${e(a)}`);break}case"image":{let a="image"in o&&typeof o.image=="string"?o.image:void 0,l="mimeType"in o&&typeof o.mimeType=="string"?o.mimeType:void 0;if(!a||a.trim()==="")throw new Error(`User message image content at index ${r} cannot be empty, received: ${e(a)}`);if(!l||l.trim()==="")throw new Error(`User message image content at index ${r} must have a mimeType, received: ${e(l)}`);break}case"audio":{let a="data"in o&&typeof o.data=="string"?o.data:void 0;if(!a||a.trim()==="")throw new Error(`User message audio content at index ${r} cannot be empty, received: ${e(a)}`);break}case"file":{let a="fileUri"in o&&typeof o.fileUri=="string",l="data"in o&&typeof o.data=="string";if(!a&&!l)throw new Error(`User message file content at index ${r} must have either 'data' or 'fileUri', received: ${e(o)}`);if(a&&l)throw new Error(`User message file content at index ${r} cannot have both 'data' and 'fileUri', received: ${e(o)}`);if(a){let c=o.fileUri;if(!c||c.trim()==="")throw new Error(`User message file content at index ${r} fileUri cannot be empty, received: ${e(c)}`)}if(l){let c=o.data;if(!c||c.trim()==="")throw new Error(`User message file content at index ${r} data cannot be empty, received: ${e(c)}`)}let p="mimeType"in o&&typeof o.mimeType=="string"?o.mimeType:null;if(!p||p.trim()==="")throw new Error(`User message file content at index ${r} must have a mimeType, received: ${e(p)}`);break}case"url":{let a="url"in o&&typeof o.url=="string"?o.url:void 0;if(!a||a.trim()==="")throw new Error(`User message url content at index ${r} cannot be empty, received: ${e(a)}`);break}default:throw new Error(`User message content item at index ${r} has unsupported type: ${e(i)}`)}}}else throw new Error(`User message content must be a string or array of content objects, received: ${e(n)}`);break}case"assistant":{let n=typeof s=="object"&&s!==null&&"content"in s?s.content:void 0,r=typeof s=="object"&&s!==null&&"functionCalls"in s?s.functionCalls:void 0;if(!n&&!r)throw new Error(`Assistant message must have either content or function calls, received content: ${e(n)}, functionCalls: ${e(r)}`);if(n&&typeof n!="string")throw new Error(`Assistant message content must be a string, received: ${e(n)}`);if(r&&!Array.isArray(r))throw new Error(`Assistant message function calls must be an array, received: ${e(r)}`);break}case"function":{let n=typeof s=="object"&&s!==null&&"functionId"in s&&typeof s.functionId=="string"?s.functionId:void 0,r=typeof s=="object"&&s!==null&&"result"in s?s.result:void 0;if(!n||n.trim()==="")throw new Error(`Function message must have a non-empty functionId, received: ${e(n)}`);if(r==null)throw new Error(`Function message must have a result, received: ${e(r)}`);if(typeof r!="string")throw new Error(`Function message result must be a string, received: ${e(r)}`);break}default:throw new Error(`Unsupported message role: ${e(t)}`)}}function pn(s){let e=n=>JSON.stringify(n,null,2),t=Array.isArray(s)?s:[s];if(t.length===0)throw new Error(`Chat response results cannot be empty, received: ${e(t)}`);for(let n=0;n<t.length;n++){let r=t[n];if(!r)throw new Error(`Chat response result at index ${n} cannot be null or undefined, received: ${e(r)}`);if(typeof r.index!="number")throw new Error(`Chat response result at index ${n} must have a numeric index, received: ${e(r.index)}`);if(r.index<0)throw new Error(`Chat response result at index ${n} must have a non-negative index, received: ${e(r.index)}`);if(!r.content&&!r.thought&&!r.functionCalls&&!r.finishReason)throw new Error(`Chat response result at index ${n} must have at least one of: content, thought, functionCalls, or finishReason, received: ${e({content:r.content,thought:r.thought,functionCalls:r.functionCalls,finishReason:r.finishReason})}`);if(r.content!==void 0&&typeof r.content!="string")throw new Error(`Chat response result content at index ${n} must be a string, received: ${e(r.content)}`);if(r.thought!==void 0&&typeof r.thought!="string")throw new Error(`Chat response result thought at index ${n} must be a string, received: ${e(r.thought)}`);if(r.name!==void 0){if(typeof r.name!="string")throw new Error(`Chat response result name at index ${n} must be a string, received: ${e(r.name)}`);if(r.name.trim()==="")throw new Error(`Chat response result name at index ${n} cannot be empty or whitespace-only, received: ${e(r.name)}`)}if(r.annotations!==void 0){if(!Array.isArray(r.annotations))throw new Error(`Chat response result annotations at index ${n} must be an array, received: ${e(r.annotations)}`);for(let o=0;o<r.annotations.length;o++){let i=r.annotations[o];if(!i||typeof i!="object")throw new Error(`Chat response result annotation at index ${n}[${o}] must be an object, received: ${e(i)}`);if(i.type!=="url_citation")throw new Error(`Chat response result annotation at index ${n}[${o}] must have type 'url_citation', received: ${e(i.type)}`);if(!i.url_citation||typeof i.url_citation!="object")throw new Error(`Chat response result annotation at index ${n}[${o}] must have a valid url_citation object, received: ${e(i.url_citation)}`);if(typeof i.url_citation.url!="string")throw new Error(`Chat response result annotation at index ${n}[${o}] url_citation.url must be a string, received: ${e(i.url_citation.url)}`)}}if(r.id!==void 0){if(typeof r.id!="string")throw new Error(`Chat response result id at index ${n} must be a string, received: ${e(r.id)}`);if(r.id.trim()==="")throw new Error(`Chat response result id at index ${n} cannot be empty or whitespace-only, received: ${e(r.id)}`)}if(r.functionCalls!==void 0){if(!Array.isArray(r.functionCalls))throw new Error(`Chat response result functionCalls at index ${n} must be an array, received: ${e(r.functionCalls)}`);for(let o=0;o<r.functionCalls.length;o++){let i=r.functionCalls[o];if(!i)throw new Error(`Function call at index ${o} in result ${n} cannot be null or undefined, received: ${e(i)}`);if(!i.id||typeof i.id!="string"||i.id.trim()==="")throw new Error(`Function call at index ${o} in result ${n} must have a non-empty string id, received: ${e(i.id)}`);if(i.type!=="function")throw new Error(`Function call at index ${o} in result ${n} must have type 'function', received: ${e(i.type)}`);if(!i.function)throw new Error(`Function call at index ${o} in result ${n} must have a function object, received: ${e(i.function)}`);if(!i.function.name||typeof i.function.name!="string"||i.function.name.trim()==="")throw new Error(`Function call at index ${o} in result ${n} must have a non-empty function name, received: ${e(i.function.name)}`);if(i.function.params!==void 0&&typeof i.function.params!="string"&&typeof i.function.params!="object")throw new Error(`Function call params at index ${o} in result ${n} must be a string or object, received: ${e(i.function.params)}`)}}if(r.finishReason!==void 0){let o=["stop","length","function_call","content_filter","error"];if(!o.includes(r.finishReason))throw new Error(`Chat response result finishReason at index ${n} must be one of: ${o.join(", ")}, received: ${e(r.finishReason)}`)}}}var dt=(d=>(d.Llama31_8B_Instruct="Llama-3.1-8B-Instruct-q4f32_1-MLC",d.Llama31_70B_Instruct="Llama-3.1-70B-Instruct-q4f16_1-MLC",d.Llama32_1B_Instruct="Llama-3.2-1B-Instruct-q4f32_1-MLC",d.Llama32_3B_Instruct="Llama-3.2-3B-Instruct-q4f32_1-MLC",d.Mistral7B_Instruct="Mistral-7B-Instruct-v0.3-q4f32_1-MLC",d.Phi35_Mini_Instruct="Phi-3.5-mini-instruct-q4f32_1-MLC",d.Gemma2_2B_Instruct="gemma-2-2b-it-q4f32_1-MLC",d.Gemma2_9B_Instruct="gemma-2-9b-it-q4f32_1-MLC",d.Qwen2_5_0_5B_Instruct="Qwen2.5-0.5B-Instruct-q4f32_1-MLC",d.Qwen2_5_1_5B_Instruct="Qwen2.5-1.5B-Instruct-q4f32_1-MLC",d.Qwen2_5_3B_Instruct="Qwen2.5-3B-Instruct-q4f32_1-MLC",d.Qwen2_5_7B_Instruct="Qwen2.5-7B-Instruct-q4f32_1-MLC",d))(dt||{});var cn=[{name:"Llama-3.1-8B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:4096},{name:"Llama-3.1-70B-Instruct-q4f16_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:4096,isExpensive:!0},{name:"Llama-3.2-1B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:2048},{name:"Llama-3.2-3B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:2048},{name:"Mistral-7B-Instruct-v0.3-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:4096},{name:"Phi-3.5-mini-instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:128e3,maxTokens:4096},{name:"gemma-2-2b-it-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:8192,maxTokens:2048},{name:"gemma-2-9b-it-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:8192,maxTokens:2048},{name:"Qwen2.5-0.5B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:2048},{name:"Qwen2.5-1.5B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:2048},{name:"Qwen2.5-3B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:2048},{name:"Qwen2.5-7B-Instruct-q4f32_1-MLC",currency:"usd",promptTokenCostPer1M:0,completionTokenCostPer1M:0,contextWindow:32768,maxTokens:4096}];var Cr=()=>structuredClone({model:"Llama-3.2-3B-Instruct-q4f32_1-MLC",...S()}),ss=()=>structuredClone({model:"Llama-3.2-3B-Instruct-q4f32_1-MLC",..._()}),Tr=class{constructor(e,t){this.config=e;this.engine=t}tokensUsed;engine;getTokenUsage(){return this.tokensUsed}getModelConfig(){let{config:e}=this;return{maxTokens:e.maxTokens,temperature:e.temperature,topP:e.topP,topK:e.topK,presencePenalty:e.presencePenalty,frequencyPenalty:e.frequencyPenalty,stopSequences:e.stopSequences,endSequences:e.endSequences,stream:e.stream,n:e.n}}createChatReq(e){let t=e.model,n=e.chatPrompt.map(a=>{if(a.role==="function")return{role:"function",name:a.functionId,content:typeof a.result=="string"?a.result:JSON.stringify(a.result)};let l="";typeof a.content=="string"?l=a.content:Array.isArray(a.content)&&(l=a.content.filter(c=>c.type==="text").map(c=>c.text).join(`
120
+ `));let p={role:a.role,content:l};return a.role==="assistant"&&a.functionCalls?.length?{...p,tool_calls:a.functionCalls.map(c=>({id:c.id,type:"function",function:{name:c.function.name,arguments:typeof c.function.params=="string"?c.function.params:JSON.stringify(c.function.params||{})}}))}:p}),r=e.functions?.map(a=>({type:"function",function:{name:a.name,description:a.description,parameters:a.parameters||{type:"object",properties:{}}}})),o={name:"/chat/completions",localCall:async(a,l)=>{try{let p=await this.engine.chat.completions.create({...a,stream:l||!1});return l?new ReadableStream({async start(c){try{for await(let u of p)c.enqueue(u);c.close()}catch(u){c.error(u)}}}):p}catch(p){throw new Error(`WebLLM API error: ${p}`)}}},i={model:t,messages:n,...r?.length?{tools:r}:{},max_tokens:e.modelConfig?.maxTokens??this.config.maxTokens,temperature:e.modelConfig?.temperature??this.config.temperature,top_p:e.modelConfig?.topP??this.config.topP,presence_penalty:e.modelConfig?.presencePenalty??this.config.presencePenalty,frequency_penalty:e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,stop:e.modelConfig?.stopSequences??this.config.stopSequences,stream:e.modelConfig?.stream??this.config.stream,n:e.modelConfig?.n??this.config.n};return[o,i]}createEmbedReq=e=>{throw new Error("WebLLM does not support embeddings")};createChatResp=e=>(this.tokensUsed={promptTokens:e.usage?.prompt_tokens??0,completionTokens:e.usage?.completion_tokens??0,totalTokens:e.usage?.total_tokens??0},{results:e.choices.map((n,r)=>{let o="stop";switch(n.finish_reason){case"stop":o="stop";break;case"length":o="length";break;case"tool_calls":o="function_call";break;case"content_filter":o="content_filter";break;default:o="stop";break}let i=n.message.tool_calls?.map(a=>({id:a.id,type:"function",function:{name:a.function.name,params:a.function.arguments}}));return{index:r,id:e.id,content:n.message.content||"",functionCalls:i,finishReason:o}}),remoteId:e.id});createChatStreamResp=(e,t)=>{let n=t,r=e.choices[0];if(!r)throw new Error("No choice in WebLLM stream response");if(r.delta.content&&(n.content=(n.content||"")+r.delta.content),r.delta.tool_calls){n.toolCalls||(n.toolCalls=[]);for(let l of r.delta.tool_calls){let p=n.toolCalls[l.index];p?l.function?.arguments&&(p.function.arguments=(p.function?.arguments||"")+l.function.arguments):n.toolCalls[l.index]={id:l.id,type:l.type,function:{name:l.function?.name,arguments:l.function?.arguments||""}}}}e.usage&&(this.tokensUsed={promptTokens:e.usage.prompt_tokens,completionTokens:e.usage.completion_tokens,totalTokens:e.usage.total_tokens});let o;if(r.finish_reason)switch(r.finish_reason){case"stop":o="stop";break;case"length":o="length";break;case"tool_calls":o="function_call";break;case"content_filter":o="content_filter";break;default:o="stop";break}let i=n.toolCalls?.map(l=>({id:l.id||"",type:"function",function:{name:l.function?.name||"",params:l.function?.arguments||""}}));return{results:[{index:0,id:e.id,content:n.content||"",functionCalls:i,finishReason:o}],remoteId:e.id}};createEmbedResp(e){throw new Error("WebLLM does not support embeddings")}},_e=class extends E{constructor({engine:e,config:t,options:n,models:r}){if(!e)throw new Error("WebLLM engine instance is required");let o={...Cr(),...t},i=new Tr(o,e);super(i,{name:"WebLLM",apiURL:void 0,headers:async()=>({}),modelInfo:cn,defaults:{model:o.model},supportFor:a=>({functions:!0,streaming:!0,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}),options:n,models:r})}};var mt=(r=>(r.Grok3="grok-3",r.Grok3Mini="grok-3-mini",r.Grok3Fast="grok-3-fast",r.Grok3MiniFast="grok-3-mini-fast",r))(mt||{}),wr=(e=>(e.GrokEmbedSmall="grok-embed-small",e))(wr||{});var un=[{name:"grok-3",currency:"USD",promptTokenCostPer1M:3,completionTokenCostPer1M:15},{name:"grok-3-mini",currency:"USD",promptTokenCostPer1M:.3,completionTokenCostPer1M:.5,hasThinkingBudget:!0},{name:"grok-3-fast",currency:"USD",promptTokenCostPer1M:5,completionTokenCostPer1M:25},{name:"grok-3-mini-fast",currency:"USD",promptTokenCostPer1M:.6,completionTokenCostPer1M:4,hasThinkingBudget:!0}];var dn=()=>structuredClone({model:"grok-3-mini",...S()}),is=()=>structuredClone({...dn(),model:"grok-3"}),De=class extends F{constructor({apiKey:e,config:t,options:n,models:r,modelInfo:o}){if(!e||e==="")throw new Error("Grok API key not set");let i={...dn(),...t};o=[...un,...o??[]];let a=p=>{let c=q({model:p,modelInfo:o,models:r});return{functions:!0,streaming:!0,hasThinkingBudget:c?.hasThinkingBudget??!1,hasShowThoughts:c?.hasShowThoughts??!1,media:{images:{supported:!1,formats:[]},audio:{supported:!1,formats:[]},files:{supported:!1,formats:[],uploadMethod:"none"},urls:{supported:!1,webSearch:!1,contextFetching:!1}},caching:{supported:!1,types:[]},thinking:!1,multiTurn:!0}},l=p=>{if(n?.searchParameters){let c=n.searchParameters;return{...p,search_parameters:{mode:c.mode,return_citations:c.returnCitations,from_date:c.fromDate,to_date:c.toDate,max_search_results:c.maxSearchResults,sources:c.sources?.map(u=>({type:u.type,country:u.country,excluded_websites:u.excludedWebsites,allowed_websites:u.allowedWebsites,safe_search:u.safeSearch,x_handles:u.xHandles,links:u.links}))}}}return p};super({apiKey:e,config:i,options:n,apiURL:"https://api.x.ai/v1",modelInfo:o,models:r,supportFor:a,chatReqUpdater:l}),super.setName("Grok")}};function as(s){return gt.create(s)}var gt=class s{ai;static create(e){return new s(e)}constructor(e){switch(e.name){case"openai":this.ai=new Ie(e);break;case"openai-responses":this.ai=new Ee(e);break;case"azure-openai":this.ai=new be(e);break;case"grok":this.ai=new De(e);break;case"huggingface":this.ai=new ve(e);break;case"groq":this.ai=new Se(e);break;case"together":this.ai=new Fe(e);break;case"cohere":this.ai=new Re(e);break;case"google-gemini":this.ai=new Ce(e);break;case"anthropic":this.ai=new Ae(e);break;case"mistral":this.ai=new Oe(e);break;case"deepseek":this.ai=new Te(e);break;case"ollama":this.ai=new Me(e);break;case"reka":this.ai=new Pe(e);break;case"webllm":this.ai=new _e(e);break;default:throw new Error("Unknown AI")}}getName(){return this.ai.getName()}getId(){return this.ai.getId()}getFeatures(e){return this.ai.getFeatures(e)}getModelList(){return this.ai.getModelList()}getLastUsedChatModel(){return this.ai.getLastUsedChatModel()}getLastUsedEmbedModel(){return this.ai.getLastUsedEmbedModel()}getLastUsedModelConfig(){return this.ai.getLastUsedModelConfig()}getMetrics(){return this.ai.getMetrics()}async chat(e,t){return await this.ai.chat(e,t)}async embed(e,t){return await this.ai.embed(e,t)}setOptions(e){this.ai.setOptions(e)}getOptions(){return this.ai.getOptions()}getLogger(){return this.ai.getLogger()}};var mn=require("@opentelemetry/api");var j=class{name;fetch;tracer;_upsert;_batchUpsert;_query;constructor({name:e,fetch:t,tracer:n}){this.name=e,this.fetch=t,this.tracer=n}async upsert(e,t){if(!this._upsert)throw new Error("upsert() not implemented");return this.tracer?await this.tracer.startActiveSpan("DB Upsert Request",{kind:mn.SpanKind.SERVER,attributes:{[C.DB_SYSTEM]:this.name,[C.DB_OPERATION_NAME]:"upsert",[C.DB_TABLE]:e.table,[C.DB_NAMESPACE]:e.namespace,[C.DB_OPERATION_NAME]:t?"update":"insert"}},async n=>{try{return await this._upsert(e,t,{span:n})}finally{n.end()}}):await this._upsert(e,t)}async batchUpsert(e,t){if(!this._batchUpsert)throw new Error("batchUpsert() not implemented");if(e.length===0)throw new Error("Batch request is empty");if(!e[0])throw new Error("Batch request is invalid first element is undefined");return this.tracer?await this.tracer.startActiveSpan("DB Batch Upsert Request",{kind:mn.SpanKind.SERVER,attributes:{[C.DB_SYSTEM]:this.name,[C.DB_OPERATION_NAME]:"upsert",[C.DB_TABLE]:e[0].table,[C.DB_NAMESPACE]:e[0].namespace,[C.DB_OPERATION_NAME]:t?"update":"insert"}},async n=>{try{return await this._batchUpsert(e,t,{span:n})}finally{n.end()}}):await this._batchUpsert(e,t)}async query(e){if(!this._query)throw new Error("query() not implemented");return this.tracer?await this.tracer.startActiveSpan("DB Query Request",{kind:mn.SpanKind.SERVER,attributes:{[C.DB_SYSTEM]:this.name,[C.DB_OPERATION_NAME]:"upsert",[C.DB_TABLE]:e.table,[C.DB_NAMESPACE]:e.namespace,[C.DB_OPERATION_NAME]:"query"}},async t=>{try{return await this._query(e,{span:t})}finally{t.end()}}):await this._query(e)}};var Sr="https://api.cloudflare.com/client/v4/accounts/",Le=class extends j{apiKey;accountId;constructor({apiKey:e,accountId:t,fetch:n,tracer:r}){if(!e||!t)throw new Error("Cloudflare credentials not set");super({name:"Cloudflare",fetch:n,tracer:r}),this.apiKey=e,this.accountId=t}_upsert=async(e,t,n)=>{let r=await B({url:new URL(`${this.accountId}/vectorize/indexes/${e.table}/upsert`,Sr),headers:{"X-Auth-Key":this.apiKey},fetch:this.fetch,span:n?.span},{id:e.id,values:e.values,namespace:e.namespace,metadata:e.metadata});if(r.errors)throw new Error(`Cloudflare upsert failed: ${r.errors.map(({message:o})=>o).join(", ")}`);return{ids:r.result.ids}};batchUpsert=async(e,t,n)=>{if(t)throw new Error("Weaviate does not support batch update");if(e.length<1)throw new Error("Batch request is empty");if(!e[0]||!e[0].table)throw new Error("Table name is empty");let r=e[0].table,o=await B({url:new URL(`${this.accountId}/vectorize/indexes/${r}/upsert`,Sr),headers:{"X-Auth-Key":this.apiKey},fetch:this.fetch,span:n?.span},e.map(i=>({id:i.id,values:i.values,namespace:i.namespace,metadata:i.metadata})));if(o.errors)throw new Error(`Cloudflare batch upsert failed: ${o.errors.map(({message:i})=>i).join(", ")}`);return{ids:o.result.ids}};query=async(e,t)=>{let n=await B({url:new URL(`${this.accountId}/vectorize/indexes/${e.table}/query`,Sr),headers:{"X-Auth-Key":this.apiKey},fetch:this.fetch,span:t?.span},{vector:e.values,topK:e.limit||10,returnValues:!0});if(n.errors)throw new Error(`Cloudflare query failed: ${n.errors.map(({message:o})=>o).join(", ")}`);return{matches:n.result.matches.map(({id:o,score:i,values:a,metadata:l})=>({id:o,score:i,values:a,metadata:l}))}}};var re=class extends j{state;constructor({tracer:e}={}){super({name:"Memory",tracer:e}),this.state={}}_upsert=async(e,t,n)=>{if(!this.state[e.table])this.state[e.table]={[e.id]:e};else{let r=this.state[e.table];if(!r)throw new Error(`Table not found: ${e.table}`);r[e.id]=e}return{ids:[e.id]}};_batchUpsert=async(e,t,n)=>{let r=[];for(let o of e){let i=await this.upsert(o,t);r.push(...i.ids)}return{ids:r}};_query=async(e,t)=>{let n=this.state[e.table];if(!n)return{matches:[]};let r=[];return Object.entries(n).forEach(([o,i])=>{if(e.values&&i.values){let a=_i(e.values,i.values);r.push({id:o,score:a,metadata:i.metadata})}}),r.sort((o,i)=>o.score-i.score),e.limit&&(r.length=e.limit),{matches:r}};getDB=()=>structuredClone(this.state);setDB=e=>{this.state=structuredClone(e)};clearDB=()=>{this.state={}}},_i=(s,e)=>{if(s.length!==e.length)throw new Error("Vectors must be of the same length.");let t=0,n=0,r=0,o=!0,i=!0,a=new Float64Array(s),l=new Float64Array(e);for(let d=0;d<a.length;d++)t+=a[d]*l[d],n+=a[d]*a[d],r+=l[d]*l[d],a[d]!==0&&(o=!1),l[d]!==0&&(i=!1);if(o||i)return 1;let p=Math.sqrt(n),c=Math.sqrt(r);return 1-t/(p*c)};var Di=s=>({namespace:s.namespace,topK:s.limit||10,filter:{},includeValues:!0,includeMetadata:!0,vector:s.values??[],id:s.id}),Ge=class extends j{apiKey;apiURL;constructor({apiKey:e,host:t,fetch:n,tracer:r}){if(!e||e==="")throw new Error("Pinecone API key not set");super({name:"Pinecone",fetch:n,tracer:r}),this.apiKey=e,this.apiURL=t}_upsert=async(e,t,n)=>(await this._batchUpsert([e],t,n),{ids:[e.id]});_batchUpsert=async(e,t,n)=>{if(e.length===0)throw new Error("Batch request is empty");return await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/vectors/upsert",fetch:this.fetch,span:n?.span},e.map(({id:r,values:o=[],metadata:i})=>({id:r,values:o,metadata:i}))),{ids:e.map(({id:r})=>r)}};query=async(e,t)=>{if(e.text)throw new Error("Pinecone does not support text");return{matches:(await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/query",fetch:this.fetch,span:t?.span},Di(e))).matches.map(({id:o,score:i,values:a,metadata:l})=>({id:o,score:i,metadata:l,values:a}))}}};var $e=class extends j{apiKey;apiURL;constructor({apiKey:e,host:t,fetch:n,tracer:r}){if(!e||e==="")throw new Error("Weaviate API key not set");super({name:"Weaviate",fetch:n,tracer:r}),this.apiKey=e,this.apiURL=t}_upsert=async(e,t,n)=>{let r=await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:`/v1/objects/${e.table}/${e.id}`,put:!!t,fetch:this.fetch,span:n?.span},{id:e.id,class:e.table,tenant:e.namespace,vector:e.values,properties:e.metadata??{}});if(r?.result?.errors)throw new Error(`Weaviate upsert failed: ${r.result.errors.error.map(({message:o})=>o).join(", ")}`);return{ids:[r.id]}};_batchUpsert=async(e,t,n)=>{if(t)throw new Error("Weaviate does not support batch update");if(e.length===0)throw new Error("Batch request is empty");let r=e.map(i=>({id:i.id,class:i.table,tenant:i.namespace,vector:i.values,properties:i.metadata??{}})),o=await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/v1/batch/objects",fetch:this.fetch,span:n?.span},{objects:r});if(o?.some(({result:i})=>i?.errors))throw new Error(`Weaviate batch upsert failed: ${o.map(({result:i})=>i?.errors?.error.map(({message:a})=>a).join(", ")).join(", ")}`);return{ids:o.map(({id:i})=>i)}};_query=async(e,t)=>{let n="";if(e.columns&&e.columns.length===0)throw new Error("Weaviate requires at least one column");if(e.values)n=`nearVector: {
121
121
  vector: [${e.values.join(",")}],
122
122
  }`;else if(e.text)n=`nearText: {
123
123
  concepts: ['${e.text}'],
124
- }`;else throw new Error("Weaviate requires either text or values");let o=await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/v1/graphql",fetch:this.fetch,span:t?.span},{query:`{
124
+ }`;else throw new Error("Weaviate requires either text or values");let r=await B({url:this.apiURL,headers:{Authorization:`Bearer ${this.apiKey}`},name:"/v1/graphql",fetch:this.fetch,span:t?.span},{query:`{
125
125
  Get {
126
126
  ${e.table} (
127
127
  limit: ${e.limit||10},
@@ -131,98 +131,98 @@ ${p}`:`Assistant: ${l.content}`}default:throw new Error("Unknown role")}}).join(
131
131
  `)}
132
132
  }
133
133
  }
134
- }`});if(o.errors)throw new Error(`Weaviate query failed: ${o.errors.map(({message:a})=>a).join(", ")}`);let r=o.data.Get[e.table];return r?{matches:r.map(a=>({id:a.id,score:1,metadata:a}))}:{matches:[]}}};var mn=class{db;constructor(e){switch(e.name){case"weaviate":this.db=new Ne(e);break;case"pinecone":this.db=new Ge(e);break;case"cloudflare":this.db=new Le(e);break;case"memory":this.db=new oe(e);break;default:throw new Error("Unknown DB")}}async upsert(e,t){return await this.db.upsert(e,t)}async batchUpsert(e,t){return await this.db.batchUpsert(e,t)}async query(e){return await this.db.query(e)}};var So="_internal",gn=class{ai;db;chunker;rewriter;reranker;constructor({ai:e,db:t,config:n}){this.ai=e,this.db=t,this.chunker=n?.chunker??this.defaultChunker,this.reranker=n?.reranker,this.rewriter=n?.rewriter}defaultChunker=e=>e.split(/\n\n+/);insert=async(e,t)=>{try{let n=Array.isArray(e)?e.join(`
134
+ }`});if(r.errors)throw new Error(`Weaviate query failed: ${r.errors.map(({message:a})=>a).join(", ")}`);let o=r.data.Get[e.table];return o?{matches:o.map(a=>({id:a.id,score:1,metadata:a}))}:{matches:[]}}};var gn=class{db;constructor(e){switch(e.name){case"weaviate":this.db=new $e(e);break;case"pinecone":this.db=new Ge(e);break;case"cloudflare":this.db=new Le(e);break;case"memory":this.db=new re(e);break;default:throw new Error("Unknown DB")}}async upsert(e,t){return await this.db.upsert(e,t)}async batchUpsert(e,t){return await this.db.batchUpsert(e,t)}async query(e){return await this.db.query(e)}};var vr="_internal",hn=class{ai;db;chunker;rewriter;reranker;constructor({ai:e,db:t,config:n}){this.ai=e,this.db=t,this.chunker=n?.chunker??this.defaultChunker,this.reranker=n?.reranker,this.rewriter=n?.rewriter}defaultChunker=e=>e.split(/\n\n+/);insert=async(e,t)=>{try{let n=Array.isArray(e)?e.join(`
135
135
 
136
- `):e,o=this.chunker(n).filter(p=>p.length>0),r=t?.maxWordsPerChunk,i=t?.minWordsPerChunk,a=_i({initialChunks:o,minWordsPerChunk:i,maxWordsPerChunk:r}),l=t?.batchSize??10;for(let p=0;p<a.length;p+=l){let c=a.slice(p,p+l),d=(await this.ai.embed({texts:c},{abortSignal:t?.abortSignal})).embeddings.map((m,h)=>({id:`chunk_${Date.now()+h}`,table:So,values:m,metadata:{text:c[h]??""}})).filter(m=>m.metadata?.text&&m.metadata?.text.length>0);await this.db.batchUpsert(d)}}catch(n){throw new Error(`Error processing text: ${n}`)}};query=async(e,{topPercent:t,abortSignal:n}={})=>{let o=Array.isArray(e)?e:[e];if(typeof o[0]=="string"&&this.rewriter)for(let[l,p]of o.entries()){let{rewrittenQuery:c}=await this.rewriter.forward(this.ai,{query:p});o[l]=c}let r;typeof o[0]=="string"?r=(await this.ai.embed({texts:o},{abortSignal:n})).embeddings.map(p=>this.db.query({table:So,values:p})):r=o.map(l=>this.db.query({table:So,values:l}));let i=await Promise.all(r),a=[];for(let{matches:l}of i){let p=l.filter(d=>d.metadata?.text&&d.metadata?.text.length>0).map(({score:d,metadata:m})=>({score:d,text:m?.text??""})),c=t&&t>1?t/100:t,u=c?Di(p,c):p;if(this.reranker){let{rankedItems:d}=await this.reranker.forward(this.ai,{query:o[0],items:u.map(h=>h.text)}),m=d.map(h=>u.find(g=>g.text===h)).filter(h=>h!==void 0);a.push(m)}else a.push(u)}return a}},_i=({initialChunks:s,maxWordsPerChunk:e=350,minWordsPerChunk:t=250})=>{let n=[],o="",r=0;return s.forEach(i=>{let a=i.split(/\s+/),l=a.length;if(r+l<=e)o+=`${i}
136
+ `):e,r=this.chunker(n).filter(p=>p.length>0),o=t?.maxWordsPerChunk,i=t?.minWordsPerChunk,a=Li({initialChunks:r,minWordsPerChunk:i,maxWordsPerChunk:o}),l=t?.batchSize??10;for(let p=0;p<a.length;p+=l){let c=a.slice(p,p+l),d=(await this.ai.embed({texts:c},{abortSignal:t?.abortSignal})).embeddings.map((m,h)=>({id:`chunk_${Date.now()+h}`,table:vr,values:m,metadata:{text:c[h]??""}})).filter(m=>m.metadata?.text&&m.metadata?.text.length>0);await this.db.batchUpsert(d)}}catch(n){throw new Error(`Error processing text: ${n}`)}};query=async(e,{topPercent:t,abortSignal:n}={})=>{let r=Array.isArray(e)?e:[e];if(typeof r[0]=="string"&&this.rewriter)for(let[l,p]of r.entries()){let{rewrittenQuery:c}=await this.rewriter.forward(this.ai,{query:p});r[l]=c}let o;typeof r[0]=="string"?o=(await this.ai.embed({texts:r},{abortSignal:n})).embeddings.map(p=>this.db.query({table:vr,values:p})):o=r.map(l=>this.db.query({table:vr,values:l}));let i=await Promise.all(o),a=[];for(let{matches:l}of i){let p=l.filter(d=>d.metadata?.text&&d.metadata?.text.length>0).map(({score:d,metadata:m})=>({score:d,text:m?.text??""})),c=t&&t>1?t/100:t,u=c?Gi(p,c):p;if(this.reranker){let{rankedItems:d}=await this.reranker.forward(this.ai,{query:r[0],items:u.map(h=>h.text)}),m=d.map(h=>u.find(g=>g.text===h)).filter(h=>h!==void 0);a.push(m)}else a.push(u)}return a}},Li=({initialChunks:s,maxWordsPerChunk:e=350,minWordsPerChunk:t=250})=>{let n=[],r="",o=0;return s.forEach(i=>{let a=i.split(/\s+/),l=a.length;if(o+l<=e)r+=`${i}
137
137
 
138
- `,r+=l;else if(r>0&&r+l<=e*1.5)o+=`${i}
138
+ `,o+=l;else if(o>0&&o+l<=e*1.5)r+=`${i}
139
139
 
140
- `,r+=l;else if(r>t&&(n.push(o.trim()),o="",r=0),l>e){let p=a;for(;p.length>e*1.5;){let c=p.splice(0,e);n.push(c.join(" "))}p.length>0&&(o+=`${p.join(" ")}
140
+ `,o+=l;else if(o>t&&(n.push(r.trim()),r="",o=0),l>e){let p=a;for(;p.length>e*1.5;){let c=p.splice(0,e);n.push(c.join(" "))}p.length>0&&(r+=`${p.join(" ")}
141
141
 
142
- `,r+=p.length)}else o=`${i}
142
+ `,o+=p.length)}else r=`${i}
143
143
 
144
- `,r=l}),(r>t||n.length===0)&&n.push(o.trim()),n},Di=(s,e=.1)=>{let t=[...s].sort((o,r)=>o.score-r.score),n=Math.ceil(t.length*e);return t.slice(0,n)};var qe=require("@opentelemetry/api");var ht=class{data=[];addRequest(e,t){this.data.push(...e.map(n=>{let o=structuredClone(n);return{role:n.role,chat:[{index:t,value:o}]}}))}addFunctionResults(e){let t=e.map(({index:o,...r})=>({index:o,value:structuredClone(r)})),n=this.getLast();n?.role==="function"?n.chat.push(...t):this.data.push({role:"function",chat:t})}addResponse(e){let t=e.map(({index:n,...o})=>({index:n,value:structuredClone(o)}));this.data.push({role:"assistant",chat:t})}updateResult({content:e,name:t,functionCalls:n,index:o}){let r=this.data.at(-1);if(!r||r.role!=="assistant"||r.role==="assistant"&&!r.updatable){this.data.push({role:"assistant",updatable:!0,chat:[{index:o,value:structuredClone({content:e,name:t,functionCalls:n})}]});return}let i=r.chat.find(a=>a.index===o);if(!i){r.chat.push({index:o,value:structuredClone({content:e,name:t,functionCalls:n})});return}typeof e=="string"&&e.trim()!==""&&(i.value.content=e),typeof t=="string"&&t.trim()!==""&&(i.value.name=t),Array.isArray(n)&&n.length>0&&(i.value.functionCalls=n)}addTag(e){let t=this.data.at(-1);t&&(t.tags||(t.tags=[]),t.tags.includes(e)||t.tags.push(e))}rewindToTag(e){let t=this.data.findIndex(n=>n.tags?.includes(e));if(t===-1)throw new Error(`Tag "${e}" not found`);return this.data.splice(t)}removeByTag(e){let t=this.data.reduce((n,o,r)=>(o.tags?.includes(e)&&n.push(r),n),[]);if(t.length===0)throw new Error(`No items found with tag "${e}"`);return t.reverse().map(n=>this.data.splice(n,1).at(0)).filter(n=>n!==void 0).reverse()}history(e){let t=[];for(let{role:n,chat:o}of this.data){let r;n==="function"?r=o.filter(i=>i.index===e).map(i=>i.value):r=o.find(i=>i.index===e)?.value,Array.isArray(r)&&r.length>0?t.push(...r.map(i=>({...i,role:n}))):typeof r=="object"&&r!==null&&t.push({...r,role:n})}return t}getLast(){return this.data.at(-1)}reset(){this.data=[]}},$e=class{memories=new Map;defaultMemory;constructor(){this.defaultMemory=new ht}getMemory(e){return e?(this.memories.has(e)||this.memories.set(e,new ht),this.memories.get(e)):this.defaultMemory}addRequest(e,t){for(let n of e)an(n);this.getMemory(t).addRequest(e,0)}addResponse(e,t){ln(e),this.getMemory(t).addResponse(e)}addFunctionResults(e,t){this.getMemory(t).addFunctionResults(e)}updateResult(e,t){this.getMemory(t).updateResult(e)}addTag(e,t){this.getMemory(t).addTag(e)}rewindToTag(e,t){return this.getMemory(t).rewindToTag(e)}history(e,t){return this.getMemory(t).history(e)}getLast(e){return this.getMemory(e).getLast()}reset(e){e?this.memories.set(e,new ht):this.defaultMemory.reset()}};var re=class extends Error{constructor({message:e}){super(e),this.name=this.constructor.name}getFixingInstructions=()=>{let e=[],t=this.message.trim();return e.push({name:"error",title:"Follow these instructions",description:t+(t.endsWith(".")?"":".")}),e};toString(){return`${this.name}: ${this.message}`}[Symbol.for("nodejs.util.inspect.custom")](e,t){return this.toString()}},hn=async(s,e)=>{for(let t of s){let{fn:n,message:o}=t,r=await n(e);if(r!==void 0&&!r)throw o?new re({message:o}):new Error("Assertion Failed: No message provided for assertion")}},vo=async(s,e,t,n=!1)=>{if(!e.currField||e.s===-1||!s||s.length===0)return;let o=s.filter(i=>i.fieldName===e.currField?.name);if(o.length===0)return;let r=t.substring(e.s);for(let i of o){let{message:a,fn:l}=i,p=await l(r,n);if(p!==void 0&&!p&&a)throw new re({message:a})}};var Oo={enabled:!0,enabledCategories:["generation","streaming","functions","errors","performance"],maxLabelLength:100,samplingRate:1},At,Mo=s=>{if(At)return At;let e=s??M.meter;if(e)return At=Li(e),At};var is=()=>{let s=[];return M.meter||s.push("Global meter not initialized"),!At&&M.meter&&s.push("Metrics instruments not created despite available meter"),{healthy:s.length===0,issues:s}},Li=s=>({generationLatencyHistogram:s.createHistogram("ax_gen_generation_duration_ms",{description:"End-to-end duration of AxGen generation requests",unit:"ms"}),generationRequestsCounter:s.createCounter("ax_gen_generation_requests_total",{description:"Total number of AxGen generation requests"}),generationErrorsCounter:s.createCounter("ax_gen_generation_errors_total",{description:"Total number of failed AxGen generations"}),multiStepGenerationsCounter:s.createCounter("ax_gen_multistep_generations_total",{description:"Total number of generations that required multiple steps"}),stepsPerGenerationHistogram:s.createHistogram("ax_gen_steps_per_generation",{description:"Number of steps taken per generation"}),maxStepsReachedCounter:s.createCounter("ax_gen_max_steps_reached_total",{description:"Total number of generations that hit max steps limit"}),validationErrorsCounter:s.createCounter("ax_gen_validation_errors_total",{description:"Total number of validation errors encountered"}),assertionErrorsCounter:s.createCounter("ax_gen_assertion_errors_total",{description:"Total number of assertion errors encountered"}),errorCorrectionAttemptsHistogram:s.createHistogram("ax_gen_error_correction_attempts",{description:"Number of error correction attempts per generation"}),errorCorrectionSuccessCounter:s.createCounter("ax_gen_error_correction_success_total",{description:"Total number of successful error corrections"}),errorCorrectionFailureCounter:s.createCounter("ax_gen_error_correction_failure_total",{description:"Total number of failed error corrections"}),maxRetriesReachedCounter:s.createCounter("ax_gen_max_retries_reached_total",{description:"Total number of generations that hit max retries limit"}),functionsEnabledGenerationsCounter:s.createCounter("ax_gen_functions_enabled_generations_total",{description:"Total number of generations with functions enabled"}),functionCallStepsCounter:s.createCounter("ax_gen_function_call_steps_total",{description:"Total number of steps that included function calls"}),functionsExecutedPerGenerationHistogram:s.createHistogram("ax_gen_functions_executed_per_generation",{description:"Number of unique functions executed per generation"}),functionErrorCorrectionCounter:s.createCounter("ax_gen_function_error_correction_total",{description:"Total number of function-related error corrections"}),fieldProcessorsExecutedCounter:s.createCounter("ax_gen_field_processors_executed_total",{description:"Total number of field processors executed"}),streamingFieldProcessorsExecutedCounter:s.createCounter("ax_gen_streaming_field_processors_executed_total",{description:"Total number of streaming field processors executed"}),streamingGenerationsCounter:s.createCounter("ax_gen_streaming_generations_total",{description:"Total number of streaming generations"}),streamingDeltasEmittedCounter:s.createCounter("ax_gen_streaming_deltas_emitted_total",{description:"Total number of streaming deltas emitted"}),streamingFinalizationLatencyHistogram:s.createHistogram("ax_gen_streaming_finalization_duration_ms",{description:"Duration of streaming response finalization",unit:"ms"}),samplesGeneratedHistogram:s.createHistogram("ax_gen_samples_generated",{description:"Number of samples generated per request"}),resultPickerUsageCounter:s.createCounter("ax_gen_result_picker_usage_total",{description:"Total number of times result picker was used"}),resultPickerLatencyHistogram:s.createHistogram("ax_gen_result_picker_duration_ms",{description:"Duration of result picker execution",unit:"ms"}),inputFieldsGauge:s.createGauge("ax_gen_input_fields",{description:"Number of input fields in signature"}),outputFieldsGauge:s.createGauge("ax_gen_output_fields",{description:"Number of output fields in signature"}),examplesUsedGauge:s.createGauge("ax_gen_examples_used",{description:"Number of examples used in generation"}),demosUsedGauge:s.createGauge("ax_gen_demos_used",{description:"Number of demos used in generation"}),promptRenderLatencyHistogram:s.createHistogram("ax_gen_prompt_render_duration_ms",{description:"Duration of prompt template rendering",unit:"ms"}),extractionLatencyHistogram:s.createHistogram("ax_gen_extraction_duration_ms",{description:"Duration of value extraction from responses",unit:"ms"}),assertionLatencyHistogram:s.createHistogram("ax_gen_assertion_duration_ms",{description:"Duration of assertion checking",unit:"ms"}),stateCreationLatencyHistogram:s.createHistogram("ax_gen_state_creation_duration_ms",{description:"Duration of state creation for multiple samples",unit:"ms"}),memoryUpdateLatencyHistogram:s.createHistogram("ax_gen_memory_update_duration_ms",{description:"Duration of memory updates during generation",unit:"ms"})}),An=Oo,as=s=>{An={...An,...s}},ls=()=>({...An}),J=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let o=String(n),r=An.maxLabelLength;e[t]=o.length>r?o.substring(0,r):o}return e},ps=(s,e,t,n,o,r)=>{try{let i=J({success:t.toString(),...n?{signature:n}:{},...o?{ai_service:o}:{},...r?{model:r}:{}});s.generationLatencyHistogram&&s.generationLatencyHistogram.record(e,i),s.generationRequestsCounter&&s.generationRequestsCounter.add(1,i),!t&&s.generationErrorsCounter&&s.generationErrorsCounter.add(1,i)}catch(i){console.warn("Failed to record generation metric:",i)}},xn=(s,e,t,n)=>{try{let o=J({...n?{signature:n}:{}});e>1&&s.multiStepGenerationsCounter&&s.multiStepGenerationsCounter.add(1,o),s.stepsPerGenerationHistogram&&s.stepsPerGenerationHistogram.record(e,o),e>=t&&s.maxStepsReachedCounter&&s.maxStepsReachedCounter.add(1,o)}catch(o){console.warn("Failed to record multi-step metric:",o)}},ko=(s,e,t)=>{try{let n=J({error_type:e,...t?{signature:t}:{}});e==="validation"&&s.validationErrorsCounter&&s.validationErrorsCounter.add(1,n),e==="assertion"&&s.assertionErrorsCounter&&s.assertionErrorsCounter.add(1,n)}catch(n){console.warn("Failed to record validation error metric:",n)}},cs=(s,e)=>{try{let t=J({error_type:"refusal",...e?{signature:e}:{}});s.validationErrorsCounter&&s.validationErrorsCounter.add(1,t)}catch(t){console.warn("Failed to record refusal error metric:",t)}},Eo=(s,e,t,n,o)=>{try{let r=J({success:t.toString(),...o?{signature:o}:{}});s.errorCorrectionAttemptsHistogram&&s.errorCorrectionAttemptsHistogram.record(e,r),t&&s.errorCorrectionSuccessCounter&&s.errorCorrectionSuccessCounter.add(1,r),t||(s.errorCorrectionFailureCounter&&s.errorCorrectionFailureCounter.add(1,r),e>=n&&s.maxRetriesReachedCounter&&s.maxRetriesReachedCounter.add(1,r))}catch(r){console.warn("Failed to record error correction metric:",r)}},Po=(s,e,t,n,o=!1,r)=>{try{let i=J({functions_enabled:e.toString(),had_function_calls:n.toString(),...r?{signature:r}:{}});e&&s.functionsEnabledGenerationsCounter&&s.functionsEnabledGenerationsCounter.add(1,i),n&&s.functionCallStepsCounter&&s.functionCallStepsCounter.add(1,i),t>0&&s.functionsExecutedPerGenerationHistogram&&s.functionsExecutedPerGenerationHistogram.record(t,i),o&&s.functionErrorCorrectionCounter&&s.functionErrorCorrectionCounter.add(1,i)}catch(i){console.warn("Failed to record function calling metric:",i)}},us=(s,e,t,n)=>{try{let o=J({...n?{signature:n}:{}});e>0&&s.fieldProcessorsExecutedCounter&&s.fieldProcessorsExecutedCounter.add(e,o),t>0&&s.streamingFieldProcessorsExecutedCounter&&s.streamingFieldProcessorsExecutedCounter.add(t,o)}catch(o){console.warn("Failed to record field processing metric:",o)}},ds=(s,e,t,n,o)=>{try{let r=J({is_streaming:e.toString(),...o?{signature:o}:{}});e&&s.streamingGenerationsCounter&&s.streamingGenerationsCounter.add(1,r),t>0&&s.streamingDeltasEmittedCounter&&s.streamingDeltasEmittedCounter.add(t,r),n&&s.streamingFinalizationLatencyHistogram&&s.streamingFinalizationLatencyHistogram.record(n,r)}catch(r){console.warn("Failed to record streaming metric:",r)}},ms=(s,e,t,n,o)=>{try{let r=J({result_picker_used:t.toString(),...o?{signature:o}:{}});s.samplesGeneratedHistogram&&s.samplesGeneratedHistogram.record(e,r),t&&s.resultPickerUsageCounter&&s.resultPickerUsageCounter.add(1,r),n&&s.resultPickerLatencyHistogram&&s.resultPickerLatencyHistogram.record(n,r)}catch(r){console.warn("Failed to record samples metric:",r)}},gs=(s,e,t,n,o,r)=>{try{let i=J({...r?{signature:r}:{}});s.inputFieldsGauge&&s.inputFieldsGauge.record(e,i),s.outputFieldsGauge&&s.outputFieldsGauge.record(t,i),s.examplesUsedGauge&&s.examplesUsedGauge.record(n,i),s.demosUsedGauge&&s.demosUsedGauge.record(o,i)}catch(i){console.warn("Failed to record signature complexity metrics:",i)}},fn=(s,e,t,n)=>{try{let o=J({metric_type:e,...n?{signature:n}:{}});switch(e){case"prompt_render":s.promptRenderLatencyHistogram&&s.promptRenderLatencyHistogram.record(t,o);break;case"extraction":s.extractionLatencyHistogram&&s.extractionLatencyHistogram.record(t,o);break;case"assertion":s.assertionLatencyHistogram&&s.assertionLatencyHistogram.record(t,o);break;case"state_creation":s.stateCreationLatencyHistogram&&s.stateCreationLatencyHistogram.record(t,o);break;case"memory_update":s.memoryUpdateLatencyHistogram&&s.memoryUpdateLatencyHistogram.record(t,o);break}}catch(o){console.warn("Failed to record performance metric:",o)}};var D=class extends Error{fields;constructor({message:e,fields:t}){super(e),this.fields=t,this.name=this.constructor.name}getFixingInstructions=()=>{let e=t=>{let n=(()=>{switch(t?.name){case"string":return"string";case"number":return"number";case"boolean":return"boolean";case"date":return'date ("YYYY-MM-DD" format)';case"datetime":return'date time ("YYYY-MM-DD HH:mm Timezone" format)';case"json":return"JSON object";case"class":return"classification class";case"code":return"code";default:return"string"}})();return t?.isArray?`json array of ${n} items`:n};return this.fields.map(t=>({name:"outputError",title:"Output Correction Required",description:`The section labeled '${t.title}' does not match the expected format of '${e(t.type)}'. ${this.message} Please revise your response to ensure it conforms to the specified format.`}))};toString(){let e=t=>{let n=(()=>{switch(t?.name){case"string":return"string";case"number":return"number";case"boolean":return"boolean";case"date":return'date ("YYYY-MM-DD" format)';case"datetime":return'date time ("YYYY-MM-DD HH:mm Timezone" format)';case"json":return"JSON object";case"class":return"classification class";case"code":return"code";default:return"string"}})();return t?.isArray?`json array of ${n} items`:n};return[`${this.name}: ${this.message}`,...this.fields.map(t=>` - ${t.title}: Expected format '${e(t.type)}'`)].join(`
145
- `)}[Symbol.for("nodejs.util.inspect.custom")](e,t){return this.toString()}},hs=({error:s,errCount:e,debug:t,logger:n,metricsInstruments:o,signatureName:r,span:i})=>{let a=s.getFixingInstructions();if(t&&n){let l=a?.map(p=>p.title).join(", ")??"";ur(s,e,l,n)}return o&&ko(o,"validation",r),i&&i.addEvent("validation.error",{message:s.toString(),fixing_instructions:a?.map(l=>l.title).join(", ")??""}),a},As=({error:s,errCount:e,debug:t,logger:n,metricsInstruments:o,signatureName:r,span:i})=>{let a=s.getFixingInstructions();if(t&&n){let l=a?.map(p=>p.title).join(", ")??"";dr(s,e,l,n)}return o&&ko(o,"assertion",r),i&&i.addEvent("assertion.error",{message:s.toString(),fixing_instructions:a?.map(l=>l.title).join(", ")??""}),a},xs=({error:s,errCount:e,debug:t,logger:n,metricsInstruments:o,signatureName:r,span:i})=>{t&&n&&mr(s,e,n),o&&cs(o,r),i&&i.addEvent("refusal.error",{message:s.toString()})};var fs=s=>{let e=[],t=(n,o="")=>{if(!n||typeof n!="object")return;let r=["array","integer","number","string","boolean","null","object"];if(n.anyOf&&Array.isArray(n.anyOf)){n.anyOf.length===0&&e.push({path:o||"root",issue:"anyOf array is empty",fix:"Add at least one schema to the anyOf array",example:'anyOf: [{ type: "string" }, { type: "null" }]'}),n.anyOf.forEach((i,a)=>{t(i,`${o}anyOf[${a}].`)});return}if(n.oneOf&&Array.isArray(n.oneOf)){n.oneOf.length===0&&e.push({path:o||"root",issue:"oneOf array is empty",fix:"Add at least one schema to the oneOf array",example:'oneOf: [{ type: "string" }, { type: "number" }]'}),n.oneOf.forEach((i,a)=>{t(i,`${o}oneOf[${a}].`)});return}if(n.allOf&&Array.isArray(n.allOf)){n.allOf.length===0&&e.push({path:o||"root",issue:"allOf array is empty",fix:"Add at least one schema to the allOf array",example:'allOf: [{ type: "object" }, { properties: { name: { type: "string" } } }]'}),n.allOf.forEach((i,a)=>{t(i,`${o}allOf[${a}].`)});return}if(n.type){if(!r.includes(n.type)){e.push({path:o||"root",issue:`Invalid type '${n.type}'`,fix:`Change type to one of: ${r.join(", ")}`,example:'{ type: "string" } or { type: "object" }'});return}if(n.type==="object"){if(n.properties)if(typeof n.properties!="object"||Array.isArray(n.properties))e.push({path:o||"root",issue:"properties must be an object, not an array or primitive",fix:"Change properties to be an object with property names as keys",example:'properties: { name: { type: "string" }, age: { type: "number" } }'});else for(let i in n.properties){let a=n.properties[i];if(a!=null){if(typeof a!="object"){e.push({path:`${o}${i}`,issue:`Property schema must be an object, got ${typeof a}`,fix:"Define the property as a proper schema object",example:`${i}: { type: "string", description: "..." }`});continue}t(a,`${o}${i}.`)}}if(n.required){if(!Array.isArray(n.required))e.push({path:o||"root",issue:`'required' must be an array, got ${typeof n.required}`,fix:"Change required to be an array of property names",example:'required: ["name", "email"] instead of required: "name,email"'});else if(n.required.length!==0){if(n.properties)for(let i of n.required)typeof i!="string"?e.push({path:`${o}required`,issue:`Required property names must be strings, got ${typeof i}`,fix:"Ensure all items in required array are strings",example:'required: ["name", "email"] not required: [123, "email"]'}):i in n.properties||e.push({path:`${o}required`,issue:`Required property '${i}' is not defined in properties`,fix:`Either add '${i}' to properties or remove it from required`,example:`properties: { ${i}: { type: "string" } }`})}}}n.type==="array"&&n.items&&(typeof n.items!="object"?e.push({path:`${o}items`,issue:`Array items schema must be an object, got ${typeof n.items}`,fix:"Define items as a proper schema object",example:'items: { type: "string" } or items: { type: "object", properties: {...} }'}):t(n.items,`${o}items.`))}};if(t(s),e.length>0){let n=["JSON Schema validation failed:","",...e.map((o,r)=>{let i=[`${r+1}. Path: ${o.path}`,` Issue: ${o.issue}`,` Fix: ${o.fix}`];return o.example&&i.push(` Example: ${o.example}`),i.join(`
144
+ `,o=l}),(o>t||n.length===0)&&n.push(r.trim()),n},Gi=(s,e=.1)=>{let t=[...s].sort((r,o)=>r.score-o.score),n=Math.ceil(t.length*e);return t.slice(0,n)};var qe=require("@opentelemetry/api");var ht=class{data=[];addRequest(e,t){this.data.push(...e.map(n=>{let r=structuredClone(n);return{role:n.role,chat:[{index:t,value:r}]}}))}addFunctionResults(e){let t=e.map(({index:r,...o})=>({index:r,value:structuredClone(o)})),n=this.getLast();n?.role==="function"?n.chat.push(...t):this.data.push({role:"function",chat:t})}addResponse(e){let t=e.map(({index:n,...r})=>({index:n,value:structuredClone(r)}));this.data.push({role:"assistant",chat:t})}updateResult({content:e,name:t,functionCalls:n,index:r}){let o=this.data.at(-1);if(!o||o.role!=="assistant"||o.role==="assistant"&&!o.updatable){this.data.push({role:"assistant",updatable:!0,chat:[{index:r,value:structuredClone({content:e,name:t,functionCalls:n})}]});return}let i=o.chat.find(a=>a.index===r);if(!i){o.chat.push({index:r,value:structuredClone({content:e,name:t,functionCalls:n})});return}typeof e=="string"&&e.trim()!==""&&(i.value.content=e),typeof t=="string"&&t.trim()!==""&&(i.value.name=t),Array.isArray(n)&&n.length>0&&(i.value.functionCalls=n)}addTag(e){let t=this.data.at(-1);t&&(t.tags||(t.tags=[]),t.tags.includes(e)||t.tags.push(e))}rewindToTag(e){let t=this.data.findIndex(n=>n.tags?.includes(e));if(t===-1)throw new Error(`Tag "${e}" not found`);return this.data.splice(t)}removeByTag(e){let t=this.data.reduce((n,r,o)=>(r.tags?.includes(e)&&n.push(o),n),[]);if(t.length===0)throw new Error(`No items found with tag "${e}"`);return t.reverse().map(n=>this.data.splice(n,1).at(0)).filter(n=>n!==void 0).reverse()}history(e){let t=[];for(let{role:n,chat:r}of this.data){let o;n==="function"?o=r.filter(i=>i.index===e).map(i=>i.value):o=r.find(i=>i.index===e)?.value,Array.isArray(o)&&o.length>0?t.push(...o.map(i=>({...i,role:n}))):typeof o=="object"&&o!==null&&t.push({...o,role:n})}return t}getLast(){return this.data.at(-1)}reset(){this.data=[]}},Ne=class{memories=new Map;defaultMemory;constructor(){this.defaultMemory=new ht}getMemory(e){return e?(this.memories.has(e)||this.memories.set(e,new ht),this.memories.get(e)):this.defaultMemory}addRequest(e,t){for(let n of e)ln(n);this.getMemory(t).addRequest(e,0)}addResponse(e,t){pn(e),this.getMemory(t).addResponse(e)}addFunctionResults(e,t){this.getMemory(t).addFunctionResults(e)}updateResult(e,t){this.getMemory(t).updateResult(e)}addTag(e,t){this.getMemory(t).addTag(e)}rewindToTag(e,t){return this.getMemory(t).rewindToTag(e)}history(e,t){return this.getMemory(t).history(e)}getLast(e){return this.getMemory(e).getLast()}reset(e){e?this.memories.set(e,new ht):this.defaultMemory.reset()}};var oe=class extends Error{constructor({message:e}){super(e),this.name=this.constructor.name}getFixingInstructions=()=>{let e=[],t=this.message.trim();return e.push({name:"error",title:"Follow these instructions",description:t+(t.endsWith(".")?"":".")}),e};toString(){return`${this.name}: ${this.message}`}[Symbol.for("nodejs.util.inspect.custom")](e,t){return this.toString()}},An=async(s,e)=>{for(let t of s){let{fn:n,message:r}=t,o=await n(e);if(o!==void 0&&!o)throw r?new oe({message:r}):new Error("Assertion Failed: No message provided for assertion")}},Or=async(s,e,t,n=!1)=>{if(!e.currField||e.s===-1||!s||s.length===0)return;let r=s.filter(i=>i.fieldName===e.currField?.name);if(r.length===0)return;let o=t.substring(e.s);for(let i of r){let{message:a,fn:l}=i,p=await l(o,n);if(p!==void 0&&!p&&a)throw new oe({message:a})}};var Mr={enabled:!0,enabledCategories:["generation","streaming","functions","errors","performance"],maxLabelLength:100,samplingRate:1},At,kr=s=>{if(At)return At;let e=s??M.meter;if(e)return At=$i(e),At};var ls=()=>{let s=[];return M.meter||s.push("Global meter not initialized"),!At&&M.meter&&s.push("Metrics instruments not created despite available meter"),{healthy:s.length===0,issues:s}},$i=s=>({generationLatencyHistogram:s.createHistogram("ax_gen_generation_duration_ms",{description:"End-to-end duration of AxGen generation requests",unit:"ms"}),generationRequestsCounter:s.createCounter("ax_gen_generation_requests_total",{description:"Total number of AxGen generation requests"}),generationErrorsCounter:s.createCounter("ax_gen_generation_errors_total",{description:"Total number of failed AxGen generations"}),multiStepGenerationsCounter:s.createCounter("ax_gen_multistep_generations_total",{description:"Total number of generations that required multiple steps"}),stepsPerGenerationHistogram:s.createHistogram("ax_gen_steps_per_generation",{description:"Number of steps taken per generation"}),maxStepsReachedCounter:s.createCounter("ax_gen_max_steps_reached_total",{description:"Total number of generations that hit max steps limit"}),validationErrorsCounter:s.createCounter("ax_gen_validation_errors_total",{description:"Total number of validation errors encountered"}),assertionErrorsCounter:s.createCounter("ax_gen_assertion_errors_total",{description:"Total number of assertion errors encountered"}),errorCorrectionAttemptsHistogram:s.createHistogram("ax_gen_error_correction_attempts",{description:"Number of error correction attempts per generation"}),errorCorrectionSuccessCounter:s.createCounter("ax_gen_error_correction_success_total",{description:"Total number of successful error corrections"}),errorCorrectionFailureCounter:s.createCounter("ax_gen_error_correction_failure_total",{description:"Total number of failed error corrections"}),maxRetriesReachedCounter:s.createCounter("ax_gen_max_retries_reached_total",{description:"Total number of generations that hit max retries limit"}),functionsEnabledGenerationsCounter:s.createCounter("ax_gen_functions_enabled_generations_total",{description:"Total number of generations with functions enabled"}),functionCallStepsCounter:s.createCounter("ax_gen_function_call_steps_total",{description:"Total number of steps that included function calls"}),functionsExecutedPerGenerationHistogram:s.createHistogram("ax_gen_functions_executed_per_generation",{description:"Number of unique functions executed per generation"}),functionErrorCorrectionCounter:s.createCounter("ax_gen_function_error_correction_total",{description:"Total number of function-related error corrections"}),fieldProcessorsExecutedCounter:s.createCounter("ax_gen_field_processors_executed_total",{description:"Total number of field processors executed"}),streamingFieldProcessorsExecutedCounter:s.createCounter("ax_gen_streaming_field_processors_executed_total",{description:"Total number of streaming field processors executed"}),streamingGenerationsCounter:s.createCounter("ax_gen_streaming_generations_total",{description:"Total number of streaming generations"}),streamingDeltasEmittedCounter:s.createCounter("ax_gen_streaming_deltas_emitted_total",{description:"Total number of streaming deltas emitted"}),streamingFinalizationLatencyHistogram:s.createHistogram("ax_gen_streaming_finalization_duration_ms",{description:"Duration of streaming response finalization",unit:"ms"}),samplesGeneratedHistogram:s.createHistogram("ax_gen_samples_generated",{description:"Number of samples generated per request"}),resultPickerUsageCounter:s.createCounter("ax_gen_result_picker_usage_total",{description:"Total number of times result picker was used"}),resultPickerLatencyHistogram:s.createHistogram("ax_gen_result_picker_duration_ms",{description:"Duration of result picker execution",unit:"ms"}),inputFieldsGauge:s.createGauge("ax_gen_input_fields",{description:"Number of input fields in signature"}),outputFieldsGauge:s.createGauge("ax_gen_output_fields",{description:"Number of output fields in signature"}),examplesUsedGauge:s.createGauge("ax_gen_examples_used",{description:"Number of examples used in generation"}),demosUsedGauge:s.createGauge("ax_gen_demos_used",{description:"Number of demos used in generation"}),promptRenderLatencyHistogram:s.createHistogram("ax_gen_prompt_render_duration_ms",{description:"Duration of prompt template rendering",unit:"ms"}),extractionLatencyHistogram:s.createHistogram("ax_gen_extraction_duration_ms",{description:"Duration of value extraction from responses",unit:"ms"}),assertionLatencyHistogram:s.createHistogram("ax_gen_assertion_duration_ms",{description:"Duration of assertion checking",unit:"ms"}),stateCreationLatencyHistogram:s.createHistogram("ax_gen_state_creation_duration_ms",{description:"Duration of state creation for multiple samples",unit:"ms"}),memoryUpdateLatencyHistogram:s.createHistogram("ax_gen_memory_update_duration_ms",{description:"Duration of memory updates during generation",unit:"ms"})}),xn=Mr,ps=s=>{xn={...xn,...s}},cs=()=>({...xn}),J=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let r=String(n),o=xn.maxLabelLength;e[t]=r.length>o?r.substring(0,o):r}return e},us=(s,e,t,n,r,o)=>{try{let i=J({success:t.toString(),...n?{signature:n}:{},...r?{ai_service:r}:{},...o?{model:o}:{}});s.generationLatencyHistogram&&s.generationLatencyHistogram.record(e,i),s.generationRequestsCounter&&s.generationRequestsCounter.add(1,i),!t&&s.generationErrorsCounter&&s.generationErrorsCounter.add(1,i)}catch(i){console.warn("Failed to record generation metric:",i)}},fn=(s,e,t,n)=>{try{let r=J({...n?{signature:n}:{}});e>1&&s.multiStepGenerationsCounter&&s.multiStepGenerationsCounter.add(1,r),s.stepsPerGenerationHistogram&&s.stepsPerGenerationHistogram.record(e,r),e>=t&&s.maxStepsReachedCounter&&s.maxStepsReachedCounter.add(1,r)}catch(r){console.warn("Failed to record multi-step metric:",r)}},Er=(s,e,t)=>{try{let n=J({error_type:e,...t?{signature:t}:{}});e==="validation"&&s.validationErrorsCounter&&s.validationErrorsCounter.add(1,n),e==="assertion"&&s.assertionErrorsCounter&&s.assertionErrorsCounter.add(1,n)}catch(n){console.warn("Failed to record validation error metric:",n)}},ds=(s,e)=>{try{let t=J({error_type:"refusal",...e?{signature:e}:{}});s.validationErrorsCounter&&s.validationErrorsCounter.add(1,t)}catch(t){console.warn("Failed to record refusal error metric:",t)}},Pr=(s,e,t,n,r)=>{try{let o=J({success:t.toString(),...r?{signature:r}:{}});s.errorCorrectionAttemptsHistogram&&s.errorCorrectionAttemptsHistogram.record(e,o),t&&s.errorCorrectionSuccessCounter&&s.errorCorrectionSuccessCounter.add(1,o),t||(s.errorCorrectionFailureCounter&&s.errorCorrectionFailureCounter.add(1,o),e>=n&&s.maxRetriesReachedCounter&&s.maxRetriesReachedCounter.add(1,o))}catch(o){console.warn("Failed to record error correction metric:",o)}},Fr=(s,e,t,n,r=!1,o)=>{try{let i=J({functions_enabled:e.toString(),had_function_calls:n.toString(),...o?{signature:o}:{}});e&&s.functionsEnabledGenerationsCounter&&s.functionsEnabledGenerationsCounter.add(1,i),n&&s.functionCallStepsCounter&&s.functionCallStepsCounter.add(1,i),t>0&&s.functionsExecutedPerGenerationHistogram&&s.functionsExecutedPerGenerationHistogram.record(t,i),r&&s.functionErrorCorrectionCounter&&s.functionErrorCorrectionCounter.add(1,i)}catch(i){console.warn("Failed to record function calling metric:",i)}},ms=(s,e,t,n)=>{try{let r=J({...n?{signature:n}:{}});e>0&&s.fieldProcessorsExecutedCounter&&s.fieldProcessorsExecutedCounter.add(e,r),t>0&&s.streamingFieldProcessorsExecutedCounter&&s.streamingFieldProcessorsExecutedCounter.add(t,r)}catch(r){console.warn("Failed to record field processing metric:",r)}},gs=(s,e,t,n,r)=>{try{let o=J({is_streaming:e.toString(),...r?{signature:r}:{}});e&&s.streamingGenerationsCounter&&s.streamingGenerationsCounter.add(1,o),t>0&&s.streamingDeltasEmittedCounter&&s.streamingDeltasEmittedCounter.add(t,o),n&&s.streamingFinalizationLatencyHistogram&&s.streamingFinalizationLatencyHistogram.record(n,o)}catch(o){console.warn("Failed to record streaming metric:",o)}},hs=(s,e,t,n,r)=>{try{let o=J({result_picker_used:t.toString(),...r?{signature:r}:{}});s.samplesGeneratedHistogram&&s.samplesGeneratedHistogram.record(e,o),t&&s.resultPickerUsageCounter&&s.resultPickerUsageCounter.add(1,o),n&&s.resultPickerLatencyHistogram&&s.resultPickerLatencyHistogram.record(n,o)}catch(o){console.warn("Failed to record samples metric:",o)}},As=(s,e,t,n,r,o)=>{try{let i=J({...o?{signature:o}:{}});s.inputFieldsGauge&&s.inputFieldsGauge.record(e,i),s.outputFieldsGauge&&s.outputFieldsGauge.record(t,i),s.examplesUsedGauge&&s.examplesUsedGauge.record(n,i),s.demosUsedGauge&&s.demosUsedGauge.record(r,i)}catch(i){console.warn("Failed to record signature complexity metrics:",i)}},yn=(s,e,t,n)=>{try{let r=J({metric_type:e,...n?{signature:n}:{}});switch(e){case"prompt_render":s.promptRenderLatencyHistogram&&s.promptRenderLatencyHistogram.record(t,r);break;case"extraction":s.extractionLatencyHistogram&&s.extractionLatencyHistogram.record(t,r);break;case"assertion":s.assertionLatencyHistogram&&s.assertionLatencyHistogram.record(t,r);break;case"state_creation":s.stateCreationLatencyHistogram&&s.stateCreationLatencyHistogram.record(t,r);break;case"memory_update":s.memoryUpdateLatencyHistogram&&s.memoryUpdateLatencyHistogram.record(t,r);break}}catch(r){console.warn("Failed to record performance metric:",r)}};var D=class extends Error{fields;constructor({message:e,fields:t}){super(e),this.fields=t,this.name=this.constructor.name}getFixingInstructions=()=>{let e=t=>{let n=(()=>{switch(t?.name){case"string":return"string";case"number":return"number";case"boolean":return"boolean";case"date":return'date ("YYYY-MM-DD" format)';case"datetime":return'date time ("YYYY-MM-DD HH:mm Timezone" format)';case"json":return"JSON object";case"class":return"classification class";case"code":return"code";default:return"string"}})();return t?.isArray?`json array of ${n} items`:n};return this.fields.map(t=>({name:"outputError",title:"Output Correction Required",description:`The section labeled '${t.title}' does not match the expected format of '${e(t.type)}'. ${this.message} Please revise your response to ensure it conforms to the specified format.`}))};toString(){let e=t=>{let n=(()=>{switch(t?.name){case"string":return"string";case"number":return"number";case"boolean":return"boolean";case"date":return'date ("YYYY-MM-DD" format)';case"datetime":return'date time ("YYYY-MM-DD HH:mm Timezone" format)';case"json":return"JSON object";case"class":return"classification class";case"code":return"code";default:return"string"}})();return t?.isArray?`json array of ${n} items`:n};return[`${this.name}: ${this.message}`,...this.fields.map(t=>` - ${t.title}: Expected format '${e(t.type)}'`)].join(`
145
+ `)}[Symbol.for("nodejs.util.inspect.custom")](e,t){return this.toString()}},xs=({error:s,errCount:e,debug:t,logger:n,metricsInstruments:r,signatureName:o,span:i})=>{let a=s.getFixingInstructions();if(t&&n){let l=a?.map(p=>p.title).join(", ")??"";mo(s,e,l,n)}return r&&Er(r,"validation",o),i&&i.addEvent("validation.error",{message:s.toString(),fixing_instructions:a?.map(l=>l.title).join(", ")??""}),a},fs=({error:s,errCount:e,debug:t,logger:n,metricsInstruments:r,signatureName:o,span:i})=>{let a=s.getFixingInstructions();if(t&&n){let l=a?.map(p=>p.title).join(", ")??"";go(s,e,l,n)}return r&&Er(r,"assertion",o),i&&i.addEvent("assertion.error",{message:s.toString(),fixing_instructions:a?.map(l=>l.title).join(", ")??""}),a},ys=({error:s,errCount:e,debug:t,logger:n,metricsInstruments:r,signatureName:o,span:i})=>{t&&n&&ho(s,e,n),r&&ds(r,o),i&&i.addEvent("refusal.error",{message:s.toString()})};var Is=s=>{let e=[],t=(n,r="")=>{if(!n||typeof n!="object")return;let o=["array","integer","number","string","boolean","null","object"];if(n.anyOf&&Array.isArray(n.anyOf)){n.anyOf.length===0&&e.push({path:r||"root",issue:"anyOf array is empty",fix:"Add at least one schema to the anyOf array",example:'anyOf: [{ type: "string" }, { type: "null" }]'}),n.anyOf.forEach((i,a)=>{t(i,`${r}anyOf[${a}].`)});return}if(n.oneOf&&Array.isArray(n.oneOf)){n.oneOf.length===0&&e.push({path:r||"root",issue:"oneOf array is empty",fix:"Add at least one schema to the oneOf array",example:'oneOf: [{ type: "string" }, { type: "number" }]'}),n.oneOf.forEach((i,a)=>{t(i,`${r}oneOf[${a}].`)});return}if(n.allOf&&Array.isArray(n.allOf)){n.allOf.length===0&&e.push({path:r||"root",issue:"allOf array is empty",fix:"Add at least one schema to the allOf array",example:'allOf: [{ type: "object" }, { properties: { name: { type: "string" } } }]'}),n.allOf.forEach((i,a)=>{t(i,`${r}allOf[${a}].`)});return}if(n.type){if(!o.includes(n.type)){e.push({path:r||"root",issue:`Invalid type '${n.type}'`,fix:`Change type to one of: ${o.join(", ")}`,example:'{ type: "string" } or { type: "object" }'});return}if(n.type==="object"){if(n.properties)if(typeof n.properties!="object"||Array.isArray(n.properties))e.push({path:r||"root",issue:"properties must be an object, not an array or primitive",fix:"Change properties to be an object with property names as keys",example:'properties: { name: { type: "string" }, age: { type: "number" } }'});else for(let i in n.properties){let a=n.properties[i];if(a!=null){if(typeof a!="object"){e.push({path:`${r}${i}`,issue:`Property schema must be an object, got ${typeof a}`,fix:"Define the property as a proper schema object",example:`${i}: { type: "string", description: "..." }`});continue}t(a,`${r}${i}.`)}}if(n.required){if(!Array.isArray(n.required))e.push({path:r||"root",issue:`'required' must be an array, got ${typeof n.required}`,fix:"Change required to be an array of property names",example:'required: ["name", "email"] instead of required: "name,email"'});else if(n.required.length!==0){if(n.properties)for(let i of n.required)typeof i!="string"?e.push({path:`${r}required`,issue:`Required property names must be strings, got ${typeof i}`,fix:"Ensure all items in required array are strings",example:'required: ["name", "email"] not required: [123, "email"]'}):i in n.properties||e.push({path:`${r}required`,issue:`Required property '${i}' is not defined in properties`,fix:`Either add '${i}' to properties or remove it from required`,example:`properties: { ${i}: { type: "string" } }`})}}}n.type==="array"&&n.items&&(typeof n.items!="object"?e.push({path:`${r}items`,issue:`Array items schema must be an object, got ${typeof n.items}`,fix:"Define items as a proper schema object",example:'items: { type: "string" } or items: { type: "object", properties: {...} }'}):t(n.items,`${r}items.`))}};if(t(s),e.length>0){let n=["JSON Schema validation failed:","",...e.map((r,o)=>{let i=[`${o+1}. Path: ${r.path}`,` Issue: ${r.issue}`,` Fix: ${r.fix}`];return r.example&&i.push(` Example: ${r.example}`),i.join(`
146
146
  `)}),"","Please fix these issues and try again."].join(`
147
147
  `);throw new Error(n)}};var xt=class extends Error{constructor(t){super();this.fields=t;this.name=this.constructor.name}getFields=()=>this.fields;toString(){return[`${this.name}: Function validation error`,...this.fields.map(t=>` - ${t.field}: ${t.message}`)].join(`
148
- `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},yn=class extends Error{constructor(t,n,o){super();this.fields=t;this.func=n;this.funcId=o}getFunctionId=()=>this.funcId;getFieldDescription(t){if(!this.func.parameters?.properties?.[t])return"";let n=this.func.parameters.properties[t],o=n.description;return n.enum?.length&&(o+=` Allowed values are: ${n.enum.join(", ")}`),o}getFixingInstructions=()=>{let t=this.fields.map(n=>{let o=this.getFieldDescription(n.field)||"";return`- \`${n.field}\` - ${n.message} (${o}).`});return`Errors In Function Arguments: Fix the following invalid arguments to '${this.func.name}'
148
+ `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},In=class extends Error{constructor(t,n,r){super();this.fields=t;this.func=n;this.funcId=r}getFunctionId=()=>this.funcId;getFieldDescription(t){if(!this.func.parameters?.properties?.[t])return"";let n=this.func.parameters.properties[t],r=n.description;return n.enum?.length&&(r+=` Allowed values are: ${n.enum.join(", ")}`),r}getFixingInstructions=()=>{let t=this.fields.map(n=>{let r=this.getFieldDescription(n.field)||"";return`- \`${n.field}\` - ${n.message} (${r}).`});return`Errors In Function Arguments: Fix the following invalid arguments to '${this.func.name}'
149
149
  ${t.join(`
150
150
  `)}`};toString(){return[`${this.name}: Function execution error in '${this.func.name}'`,...this.fields.map(t=>{let n=this.getFieldDescription(t.field);return` - ${t.field}: ${t.message}${n?` (${n})`:""}`}),this.funcId?` Function ID: ${this.funcId}`:""].join(`
151
- `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},ft=class{funcList=[];constructor(e){this.funcList=e}executeFunction=async(e,t,n)=>{let o;typeof t.args=="string"&&t.args.length>0?o=JSON.parse(t.args):o=t.args;let r=n?{sessionId:n.sessionId,ai:n.ai}:void 0,i;return e.parameters?i=e.func.length===2?await e.func(o,r):await e.func(o):i=e.func.length===1?await e.func(r):await e.func(),(n?.functionResultFormatter??M.functionResultFormatter)(i)};execute=async(e,t)=>{let n=this.funcList.find(o=>o.name.localeCompare(e.name)===0);if(!n)throw new Error(`Function not found: ${e.name}`);if(!n.func)throw new Error(`No handler for function: ${e.name}`);try{return await this.executeFunction(n,e,t)}catch(o){throw o instanceof xt?new yn(o.getFields(),n,e.id):o}}},Fo=(s,e)=>{if(s.length===0)return[...e??[]];let t=s.map(n=>"toFunction"in n?n.toFunction():n).flat();for(let n of t.filter(o=>o.parameters))n.parameters&&fs(n.parameters);return[...e??[],...t]},_o=async({ai:s,functionList:e,functionCalls:t,mem:n,sessionId:o,span:r,excludeContentFromTrace:i,index:a,functionResultFormatter:l,logger:p})=>{let c=new ft(e),u=new Set,d=t.map(g=>{if(!g.id)throw new Error(`Function ${g.name} did not return an ID`);return c.execute(g,{sessionId:o,ai:s,functionResultFormatter:l}).then(A=>{if(u.add(g.name.toLowerCase()),r){let x={name:g.name};i||(x.args=g.args,x.result=A??""),r.addEvent("function.call",x)}return{result:A??"",role:"function",functionId:g.id,index:a}}).catch(A=>{if(!(A instanceof yn))throw A;let x=A.getFixingInstructions();if(r){let y={name:g.name,message:A.toString()};i||(y.args=g.args,y.fixing_instructions=x),r.addEvent("function.error",y)}return s.getOptions().debug&&cr(A,a,x,p),{functionId:g.id,isError:!0,index:a,result:x,role:"function"}})}),h=(await Promise.all(d)).filter(g=>g!==void 0);if(n.addFunctionResults(h,o),s.getOptions().debug){let g=h.filter(f=>!f.isError);g.length>0&&pr(g,p)}return h.some(g=>g.isError)&&n.addTag("error",o),u};function Do(s,e,t,n){if(!e||e.length===0)return;if(!s.getFeatures(n).functions)throw new Error("Functions are not supported by the AI service");return e.map(r=>({id:r.id,name:r.function.name,args:r.function.params}))}function ys(s,e,t){let n=e;return!t&&(n==="required"||typeof n=="function")?{functions:[],functionCall:void 0}:s?{functions:s.map(r=>"toFunction"in r?r.toFunction():r).flat(),functionCall:n}:{functions:[],functionCall:n}}var se=Ot(require("dayjs"),1),Is=Ot(require("dayjs/plugin/customParseFormat.js"),1),bs=Ot(require("dayjs/plugin/timezone.js"),1),Rs=Ot(require("dayjs/plugin/utc.js"),1);se.default.extend(Rs.default);se.default.extend(bs.default);se.default.extend(Is.default);function Cs(s,e,t=!1){try{return Gi(e)}catch(n){if(s.isOptional&&!t)return;let o=n.message;throw new D({fields:[s],message:o,value:e})}}function Gi(s){if(!(0,se.default)(s,"YYYY-MM-DD",!0).isValid())throw new Error('Invalid date format. Please provide the date in "YYYY-MM-DD" format.');return se.default.utc(s,"YYYY-MM-DD").startOf("day").toDate()}function Ts(s,e,t=!1){try{return Ni(e)}catch(n){if(s.isOptional&&!t)return;let o=n.message;throw new D({fields:[s],message:o,value:e})}}function Ni(s){let e=/^(\d{4}-\d{2}-\d{2} \d{2}:\d{2}(?::\d{2})?) (.+)$/,t=s.match(e);if(!t)throw new Error('Invalid date and time format. Please provide the date and time in "YYYY-MM-DD HH:mm" or "YYYY-MM-DD HH:mm:ss" format, followed by the timezone.');let[,n,o]=t;if(!n||!o)throw new Error('Invalid date and time format. Please provide the date and time in "YYYY-MM-DD HH:mm" or "YYYY-MM-DD HH:mm:ss" format, followed by the timezone.');try{let r=n.includes(":")&&n.split(":").length===3?"YYYY-MM-DD HH:mm:ss":"YYYY-MM-DD HH:mm",i=se.default.tz(n,r,o);if(!i.isValid())throw new Error("Invalid date and time values. Please ensure all components are correct.");return i.utc().toDate()}catch{throw new Error(`Unrecognized time zone ${o}. Please provide a valid time zone name, abbreviation, or offset. For example, "America/New_York", or "EST".`)}}var ws=s=>(0,se.default)(s).utc().format("YYYY-MM-DD HH:mm:ss [UTC]");var Cu=new G;var In=(s,e)=>{let t=s.type??{name:"string",isArray:!1},n=(p,c)=>{switch(p){case"class":return typeof c=="string";case"code":return typeof c=="string";case"string":return typeof c=="string";case"number":return typeof c=="number";case"boolean":return typeof c=="boolean";case"date":return c instanceof Date||typeof c=="string";case"datetime":return c instanceof Date||typeof c=="string";case"json":return typeof c=="object"||typeof c=="string";default:return!1}},o=p=>!(!p||typeof p!="object"||!("mimeType"in p)||!("data"in p));if(s.type?.name==="image"){let p;if(Array.isArray(e)){for(let c of e)if(!o(c)){p="object ({ mimeType: string; data: string })";break}}else o(e)||(p="object ({ mimeType: string; data: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let r=p=>!(!p||typeof p!="object"||!("data"in p));if(s.type?.name==="audio"){let p;if(Array.isArray(e)){for(let c of e)if(!r(c)){p="object ({ data: string; format?: string })";break}}else r(e)||(p="object ({ data: string; format?: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let i=p=>{if(!p||typeof p!="object"||!("mimeType"in p))return!1;let c="data"in p,u="fileUri"in p;return!(!c&&!u||c&&u)};if(s.type?.name==="file"){let p;if(Array.isArray(e)){for(let c of e)if(!i(c)){p="object ({ mimeType: string; data: string } | { mimeType: string; fileUri: string })";break}}else i(e)||(p="object ({ mimeType: string; data: string } | { mimeType: string; fileUri: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let a=p=>typeof p=="string"?!0:!(!p||typeof p!="object"||!("url"in p));if(s.type?.name==="url"){let p;if(Array.isArray(e)){for(let c of e)if(!a(c)){p="string or object ({ url: string; title?: string; description?: string })";break}}else a(e)||(p="string or object ({ url: string; title?: string; description?: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let l=!0;if(t.isArray){if(!Array.isArray(e))l=!1;else for(let p of e)if(!n(t.name,p)){l=!1;break}}else l=n(t.name,e);if(!l){let p=Array.isArray(e)?"array":typeof e;throw new Error(`Validation failed: Expected '${s.name}' to be a ${s.type?.isArray?"an array of ":""}${t.name} instead got '${p}' (${JSON.stringify(e)})`)}};function yt(s){let e={};for(let t of s){let n=`${t.ai}:${t.model}`;if(!e[n]){e[n]={...t};continue}let o=e[n];if(o){let r=o.tokens??{promptTokens:0,completionTokens:0,totalTokens:0};r.promptTokens+=t?.tokens?.promptTokens??0,r.completionTokens+=t?.tokens?.completionTokens??0,r.totalTokens+=t?.tokens?.totalTokens??0,o.tokens=r}}return Object.values(e)}var Ss=s=>{if(!s.trim())return[];let e=new Set(["-","*","+"]),t=/^\d+[\s]*[.)\]]\s*/,n=s.split(`
152
- `),o=[];for(let r of n){let i=r.trim();if(i){if(i[0]&&e.has(i[0]))o.push(i.slice(1).trim());else if(t.test(i))o.push(i.replace(t,"").trim());else if(o.length!==0)throw new Error("Could not parse markdown list: mixed content detected")}}if(o.length===0)throw new Error("Could not parse markdown list: no valid list items found");return o};function Go(s,e){let{index:t,delta:n,version:o}=e,r=s.find(i=>i.index===t)?.delta;if(!r)return s.push({index:t,delta:n,version:o}),s;for(let i of Object.keys(n)){let a=r[i],l=n[i];a===void 0&&Array.isArray(l)?r[i]=[...l]:Array.isArray(a)&&Array.isArray(l)?r[i]=[...a,...l]:(a===void 0||typeof a=="string")&&typeof l=="string"?r[i]=`${a??""}${l}`:r[i]=l}return s}var Lo=class{cache=new Map;maxSize;constructor(e){this.maxSize=e}get(e){let t=this.cache.get(e);return t&&(this.cache.delete(e),this.cache.set(e,t)),t}set(e,t){if(this.cache.has(e))this.cache.delete(e);else if(this.cache.size>=this.maxSize){let n=this.cache.keys().next().value;n&&this.cache.delete(n)}this.cache.set(e,t)}},$i=new Lo(500);function No(s,e,t=0,n=$i){if(/^```[a-zA-Z]*\s*$/.test(s))return-4;if(/^[\s`]*$/.test(s))return-3;let o=s.indexOf(e,t);if(o!==-1)return o;let r=n.get(e)??Array.from({length:e.length},(a,l)=>e.slice(0,l+1));n.get(e)||n.set(e,r);let i=-1;for(let a=r.length-1;a>=0;a--){let l=r[a];if(s.endsWith(l)){i=a;break}}return i>=0?-2:-1}var Ms=(s,e,t,n=!1)=>{let o={extractedFields:[],streamedIndex:{},s:-1};$o(s,e,o,t,{strictMode:n}),Uo(s,e,o,t,n);for(let r of s.getOutputFields())r.isInternal&&delete e[r.name]},Ui=(s,e,t)=>{let n=[];for(let o of t)o&&!o.isOptional&&e[o.name]===void 0&&n.push(o);if(n.length>0)throw new D({message:`Required ${n.length===1?"field":"fields"} not found`,fields:n})},$o=(s,e,t,n,{strictMode:o,skipEarlyFail:r}={})=>{let i=s.getOutputFields(),a;for(let[l,p]of i.entries()){if(l===t.currFieldIndex&&!t.inAssumedField||p.name in e&&!(l===t.currFieldIndex&&t.inAssumedField))continue;let u=`${(t.extractedFields.length===0?"":`
153
- `)+p.title}:`,d=No(n,u,t.s),m=u.length;switch(d){case-1:if(r)continue;if(!o&&i.length===1&&t.currField===void 0){t.inAssumedField=!0,a=p,m=0,d=0;break}if(t.currField===void 0&&t.extractedFields.length===0){if(o&&!p.isOptional)throw new D({message:"Expected (Required) field not found",fields:[p]});if(!o){let h=!1;for(let g=l;g<i.length;g++){let f=i[g];if(!f)continue;let A=`${(t.extractedFields.length===0?"":`
154
- `)+f.title}:`;if(No(n,A,t.s)>=0){h=!0;break}}if(!h){t.inAssumedField=!0,a=p,m=0,d=0;break}}}a=p.isOptional?void 0:p;continue;case-2:return!0;case-3:return!0;case-4:return t.inBlock=!0,!0}if(a&&a.name!==p.name)throw new D({message:"Expected (Required) field not found",fields:[a]});if(t.currField!==void 0&&t.inAssumedField){let h=n.substring(0,d).trim();if(h&&t.currField.name===p.name){let g=It(t.currField,h);g!==void 0&&(e[t.currField.name]=g)}else if(h){let g=It(t.currField,h);g!==void 0&&(e[t.currField.name]=g)}t.inAssumedField=!1,t.streamedIndex[t.currField.name]=0,t.currField=void 0}if(t.currField){let h=n.substring(t.s,d).trim(),g=It(t.currField,h);g!==void 0&&(e[t.currField.name]=g),t.prevFields?t.prevFields?.push({field:t.currField,s:t.s,e:d}):t.prevFields=[{field:t.currField,s:t.s,e:d}]}t.s=d+m,t.currField=p,t.currFieldIndex=l,t.extractedFields.includes(p)||t.extractedFields.push(p),t.streamedIndex[p.name]===void 0&&(t.streamedIndex[p.name]=0)}},Uo=(s,e,t,n,o=!1)=>{if(t.currField){let r=n.substring(t.s).trim(),i=It(t.currField,r);i!==void 0&&(e[t.currField.name]=i)}if(o&&!t.currField&&t.extractedFields.length===0&&n.trim()){let a=s.getOutputFields().find(l=>!l.isOptional);if(a)throw new D({message:"Expected field not found",fields:[a]})}Bi(s,e,n),Ui(t,e,s.getOutputFields())},Bi=(s,e,t)=>{let n=s.getOutputFields();for(let o of n){if(!o.isOptional||o.name in e)continue;let r=`${o.title}:`,i=t.indexOf(r);if(i===-1)continue;let a=i+r.length,l=t.length;for(let c of n){if(c.name===o.name)continue;let u=`${c.title}:`,d=t.indexOf(u,a);d!==-1&&d<l&&(l=d)}let p=t.substring(a,l).trim();if(p)try{let c=It(o,p);c!==void 0&&(e[o.name]=c)}catch{}}},vs=(s,e,t=!1)=>{switch(s.type?.name){case"code":return ks(e);case"string":return e;case"number":{let n=Number(e);if(Number.isNaN(n)){if(s.isOptional&&!t)return;throw new Error("Invalid number")}return n}case"boolean":{if(typeof e=="boolean")return e;let n=e.toLowerCase();if(n==="true")return!0;if(n==="false")return!1;if(s.isOptional&&!t)return;throw new Error("Invalid boolean")}case"date":return Cs(s,e,t);case"datetime":return Ts(s,e,t);case"class":{let n=e;if(s.type.options&&!s.type.options.includes(n)){if(s.isOptional)return;throw new Error(`Invalid class '${e}', expected one of the following: ${s.type.options.join(", ")}`)}return n}default:return e}};function*Os(s,e,t,n,o,r){let{name:i,isInternal:a}=e,{isArray:l,name:p}=e.type??{};if(a||l||p&&p!=="string"&&p!=="code")return;let c=o.streamedIndex[i]??0,u=c===0,d=s.substring(t+c,n);if(d.length===0)return;let m=d.replace(/\s+$/,"");o.currField?.type?.name==="code"&&(m=m.replace(/\s*```\s*$/,""));let h=u?m.trimStart():m;o.currField?.type?.name==="code"&&(h=h.replace(/^[ ]*```[a-zA-Z0-9]*\n\s*/,"")),h.length>0&&(yield{index:r,delta:{[i]:h}},o.streamedIndex[i]=c+m.length)}function*Bo(s,e,t,n,o){for(let i of n.prevFields??[]){let{field:a,s:l,e:p}=i;yield*Os(e,a,l,p,n,o)}if(n.prevFields=void 0,!n.currField||n.currField.isInternal)return;yield*Os(e,n.currField,n.s,e.length,n,o);let r=s.getOutputFields();for(let i of Object.keys(t)){let a=r.find(p=>p.name===i);if(!a||a.isInternal)continue;let l=t[i];if(Array.isArray(l)){let p=n.streamedIndex?.[i]??0,c=l.slice(p);c&&c.length>0&&(yield{index:o,delta:{[i]:c}},n.streamedIndex[i]=p+c.length);continue}n.streamedIndex[i]||(yield{index:o,delta:{[i]:l}},n.streamedIndex[i]=1)}}function It(s,e){if(!e||e===""||/^(null|undefined)\s*$/i.test(e)){if(s.isOptional)return;throw new D({message:"Required field is missing",fields:[s],value:e})}let t;if(s.type?.name==="json")try{let n=ks(e);return t=JSON.parse(n),t}catch(n){throw new D({message:`Invalid JSON: ${n.message}`,fields:[s],value:e})}if(s.type?.isArray)try{try{t=JSON.parse(e)}catch{t=Ss(e)}if(!Array.isArray(t))throw new Error("Expected an array")}catch(n){throw new D({message:`Invalid Array: ${n.message}`,fields:[s],value:e})}try{if(Array.isArray(t)){for(let[n,o]of t.entries())if(o!==void 0){let r=typeof o=="string"?o.trim():o;t[n]=vs(s,r,!0)}}else t=vs(s,e)}catch(n){throw new D({message:n.message,fields:[s],value:e})}if(!(typeof t=="string"&&t===""))return t}var ks=s=>{let t=/```([A-Za-z]*)\n([\s\S]*?)\n```/g.exec(s);return t?t.length===3?t[2]:t.length===2?t[1]:s:s};async function qo(s,e,t,n){for(let o of s){if(e[o.field.name]===void 0)continue;let r=o.process,i=await r(e[o.field.name],{sessionId:n,values:e,done:!0});Es(o.field,t,i,n)}}async function zo(s,e,t,n,o,r,i=!1){for(let a of s){if(t.currField?.name!==a.field.name)continue;let l=e.substring(t.s);t.currField?.type?.name==="code"&&(l=l.replace(/^[ ]*```[a-zA-Z0-9]*\n\s*/,""),l=l.replace(/\s*```\s*$/,""));let p=a.process,c=await p(l,{sessionId:r,values:o,done:i});Es(t.currField,n,c,r)}}var Es=(s,e,t,n)=>{if(t===void 0||typeof t=="string"&&(t===""||/^(null|undefined)\s*$/i.test(t)))return;let o=JSON.stringify(t,(i,a)=>typeof a=="bigint"?Number(a):a,2),r=qi(s,o);e.addRequest([{role:"user",content:[{type:"text",text:r}]}],n),e.addTag("processor",n)};function qi(s,e){let t=s.type?.name==="code",n=s.title;return t?`Code in the field "${n}" was executed. The code execution produced the following output: ${e}`:`The field "${n}" was processed. The field contents were transformed into the following output: ${e}`}async function*Ps({res:s,usage:e,states:t,...n}){let o=(n.ai.getFeatures().functionCot??!1)&&n.functions!==void 0&&n.functions.length>0,r,i=s.getReader();try{for(;;){let{done:a,value:l}=await i.read();if(a){r&&e.push(r);break}let p=l;p.modelUsage&&(r=p.modelUsage);for(let c of p.results){if((!c.content||c.content==="")&&(!c.thought||c.thought==="")&&(!c.functionCalls||c.functionCalls.length===0))continue;let u=t.find(d=>d.index===c.index);if(!u)throw new Error(`No state found for result (index: ${c.index})`);yield*zi({...n,result:c,skipEarlyFail:o,state:u})}}}finally{i.releaseLock()}for(let a of t)yield*ji({...n,state:a})}async function*zi({result:s,mem:e,sessionId:t,strictMode:n,skipEarlyFail:o,state:r,signature:i,streamingFieldProcessors:a,thoughtFieldName:l,streamingAsserts:p,asserts:c}){if(s.functionCalls&&s.functionCalls.length>0)Ft(r.functionCalls,s.functionCalls),e.updateResult({name:s.name,content:s.content,functionCalls:r.functionCalls,delta:s.functionCalls?.[0]?.function?.params,index:s.index},t);else if(s.content&&s.content.length>0){if(s.thought&&s.thought.length>0&&(yield{index:s.index,delta:{[l]:s.thought}}),r.content+=s.content,e.updateResult({name:s.name,content:r.content,delta:s.content,index:s.index},t),$o(i,r.values,r.xstate,r.content,{strictMode:n,skipEarlyFail:o}))return;p.length!==0&&await vo(p,r.xstate,r.content),a.length!==0&&await zo(a,r.content,r.xstate,e,r.values,t),yield*Bo(i,r.content,r.values,r.xstate,s.index),await hn(c,r.values)}else s.thought&&s.thought.length>0&&(r.values[l]=(r.values[l]??"")+s.thought,yield{index:s.index,delta:{[l]:s.thought}});if(s.finishReason==="length")throw new Error(`Max tokens reached before completion
155
- Content: ${r.content}`)}async function*ji({state:s,signature:e,ai:t,model:n,functions:o,mem:r,sessionId:i,traceId:a,span:l,strictMode:p,excludeContentFromTrace:c,streamingAsserts:u,asserts:d,fieldProcessors:m,streamingFieldProcessors:h,functionResultFormatter:g,logger:f}){let A=Do(t,s.functionCalls,s.values,n);if(A){if(!o)throw new Error("Functions are not defined");let x=await _o({ai:t,functionList:o,functionCalls:A,mem:r,sessionId:i,traceId:a,span:l,index:s.index,excludeContentFromTrace:c,functionResultFormatter:g,logger:f});s.functionsExecuted=new Set([...s.functionsExecuted,...x])}else Uo(e,s.values,s.xstate,s.content,p),await vo(u,s.xstate,s.content,!0),await hn(d,s.values),m.length&&await qo(m,s.values,r,i),h.length!==0&&await zo(h,s.content,s.xstate,r,s.values,i,!0),yield*Bo(e,s.content,s.values,s.xstate,s.index)}async function*Fs({ai:s,res:e,mem:t,sessionId:n,traceId:o,functions:r,span:i,strictMode:a,states:l,usage:p,excludeContentFromTrace:c,asserts:u,fieldProcessors:d,thoughtFieldName:m,signature:h,functionResultFormatter:g,logger:f}){let A=e.results??[];t.addResponse(A,n);for(let I of A){let R=l[I.index];if(!R)throw new Error(`No state found for result (index: ${I.index})`);if(e.modelUsage&&p.push(e.modelUsage),I.functionCalls?.length){let k=Do(s,I.functionCalls,R.values);if(k){if(!r)throw new Error("Functions are not defined");let v=await _o({ai:s,functionList:r,functionCalls:k,mem:t,sessionId:n,traceId:o,span:i,excludeContentFromTrace:c,index:I.index,functionResultFormatter:g,logger:f});R.functionsExecuted=new Set([...R.functionsExecuted,...v])}}else I.content&&(I.thought&&I.thought.length>0&&(R.values[m]=I.thought),Ms(h,R.values,I.content,a),await hn(u,R.values),d.length&&await qo(d,R.values,t,n));if(I.finishReason==="length")throw new Error(`Max tokens reached before completion
156
- Content: ${I.content}`)}let x=l.map(I=>I.values);for(let I of x)for(let R of h.getOutputFields())R.isInternal&&delete I[R.name];let y=h.getOutputFields(),S=x.map((I,R)=>{let k={};for(let v of y)v.isInternal||(k[v.name]=I[v.name]);return I[m]!==void 0&&(k[m]=I[m]),{index:R,delta:k}});for(let I of S)yield I}function _s(s,e,t,n){let o=s.getLast(n);if(!o)return!0;for(let[r,i]of t.entries()){let a=e&&i.functionsExecuted.has(e);if(!o.chat[r])throw new Error(`No chat message found for result (index: ${r})`);let p=o.role==="function",c=o.tags?o.tags.some(u=>u==="processor"):!1;if(p&&e&&a||!(p||c))return!1}return!0}var Ue=class{reg;constructor(){this.reg=new Set}register(e){this.reg.add(e)}*[Symbol.iterator](){let e=Array.from(this.reg);for(let t=0;t<e.length;t++)yield e[t]}};var C=class extends Error{constructor(t,n,o,r){super(t);this.position=n;this.context=o;this.suggestion=r;this.name="SignatureValidationError"}},jo=class{input;position;currentFieldName=null;currentSection="description";constructor(e){if(this.input=e.trim(),this.position=0,!this.input)throw new C("Empty signature provided",0,"",'A signature must contain at least input and output fields separated by "->". Example: "userQuery:string -> aiResponse:string"')}parse(){try{this.skipWhitespace();let e=this.parseParsedString();this.skipWhitespace(),this.currentSection="inputs";let t=this.parseFieldList(this.parseInputField.bind(this),"input");if(this.skipWhitespace(),this.position>=this.input.length)throw new C("Incomplete signature: Missing output section",this.position,this.getErrorContext(),'Add "->" followed by output fields. Example: "-> responseText:string"');if(this.expectArrow(),this.skipWhitespace(),this.position>=this.input.length)throw new C('Incomplete signature: No output fields specified after "->"',this.position,this.getErrorContext(),'Add at least one output field. Example: "-> responseText:string"');this.currentSection="outputs";let n=this.parseFieldList(this.parseOutputField.bind(this),"output");if(this.skipWhitespace(),this.position<this.input.length){let o=this.input.slice(this.position);throw new C(`Unexpected content after signature: "${o}"`,this.position,this.getErrorContext(),"Remove any extra content after the output fields")}return this.validateParsedSignature({desc:e?.trim(),inputs:t,outputs:n}),{desc:e?.trim(),inputs:t,outputs:n}}catch(e){if(e instanceof C)throw e;let t=e instanceof Error?e.message:"Unknown error";throw new C(t,this.position,this.getErrorContext())}}validateParsedSignature(e){let t=new Set;for(let o of e.inputs){if(t.has(o.name))throw new C(`Duplicate input field name: "${o.name}"`,0,"","Each field name must be unique within the signature");t.add(o.name)}let n=new Set;for(let o of e.outputs){if(n.has(o.name))throw new C(`Duplicate output field name: "${o.name}"`,0,"","Each field name must be unique within the signature");n.add(o.name)}for(let o of e.outputs)if(t.has(o.name))throw new C(`Field name "${o.name}" appears in both inputs and outputs`,0,"","Use different names for input and output fields to avoid confusion");if(e.inputs.length===0)throw new C("Signature must have at least one input field",0,"",'Add an input field before "->". Example: "userInput:string -> ..."');if(e.outputs.length===0)throw new C("Signature must have at least one output field",0,"",'Add an output field after "->". Example: "... -> responseText:string"')}getErrorContext(){let e=Math.max(0,this.position-25),t=Math.min(this.input.length,this.position+25),n=this.input.slice(e,this.position),o=this.input.slice(this.position,t),r=`${" ".repeat(n.length)}^`;return[`Position ${this.position} in signature:`,`"${n}${o}"`,` ${r}`].join(`
157
- `)}parseFieldList(e,t){let n=[];if(this.skipWhitespace(),this.position>=this.input.length)throw new C(`Empty ${t} section: Expected at least one field`,this.position,this.getErrorContext(),`Add a ${t} field. Example: ${t==="input"?"userInput:string":"responseText:string"}`);try{n.push(e())}catch(o){throw o instanceof C?o:new C(`Invalid first ${t} field: ${o instanceof Error?o.message:"Unknown error"}`,this.position,this.getErrorContext())}for(this.skipWhitespace();this.position<this.input.length&&!(this.input[this.position]==="-"&&this.position+1<this.input.length&&this.input[this.position+1]===">");)if(this.match(",")){if(this.skipWhitespace(),this.position>=this.input.length)throw new C(`Unexpected end of input after comma in ${t} section`,this.position,this.getErrorContext(),`Add another ${t} field after the comma`);try{n.push(e())}catch(o){throw o instanceof C?o:new C(`Invalid ${t} field after comma: ${o instanceof Error?o.message:"Unknown error"}`,this.position,this.getErrorContext())}this.skipWhitespace()}else break;return n}parseInputField(){this.skipWhitespace();let e=this.parseParsedIdentifier();this.currentFieldName=e,this.validateFieldName(e,"input");let t;for(;;){if(this.match("?")){t=!0;continue}if(this.match("!"))throw new C(`Input field "${e}" cannot use the internal marker "!"`,this.position-1,this.getErrorContext(),"Internal markers (!) are only allowed on output fields");break}let n;if(this.skipWhitespace(),this.match(":")){if(this.skipWhitespace(),/^class\b/.test(this.input.slice(this.position)))throw new C(`Input field "${e}" cannot use the "class" type`,this.position,this.getErrorContext(),'Class types are only allowed on output fields. Use "string" type for input classifications');try{let r=this.parseTypeNotClass(),i=this.match("[]");if(n={name:r,isArray:i},(r==="image"||r==="audio")&&i)throw new C(`Input field "${e}": Arrays of ${r} are not supported`,this.position,this.getErrorContext(),`Use a single ${r} type instead: "${r}"`)}catch(r){throw r instanceof C?r:new C(`Input field "${e}": ${r instanceof Error?r.message:"Unknown error"}`,this.position,this.getErrorContext())}}this.skipWhitespace();let o=this.parseParsedString();return{name:e,desc:o?.trim(),type:n,isOptional:t}}parseOutputField(){this.skipWhitespace();let e=this.parseParsedIdentifier();this.currentFieldName=e,this.validateFieldName(e,"output");let t=!1,n=!1;for(;;){if(this.match("?")){t=!0;continue}if(this.match("!")){n=!0;continue}break}let o;if(this.skipWhitespace(),this.match(":"))if(this.skipWhitespace(),this.match("class")){let i=this.match("[]");this.skipWhitespace();let a=this.parseParsedString();if(!a)throw new C(`Output field "${e}": Missing class options after "class" type`,this.position,this.getErrorContext(),'Add class names in quotes. Example: class "positive, negative, neutral"');let l=a.split(/[,|]/).map(p=>p.trim()).filter(p=>p.length>0);if(l.length===0)throw new C(`Output field "${e}": Empty class list provided`,this.position,this.getErrorContext(),'Provide at least one class option. Example: "positive, negative"');o={name:"class",isArray:i,options:l}}else try{let i=this.parseTypeNotClass(),a=this.match("[]");if(o={name:i,isArray:a},i==="image"&&a)throw new C(`Output field "${e}": Arrays of images are not supported`,this.position,this.getErrorContext(),'Use a single image type instead: "image"');if(i==="audio"&&a)throw new C(`Output field "${e}": Arrays of audio are not supported`,this.position,this.getErrorContext(),'Use a single audio type instead: "audio"');if(i==="image")throw new C(`Output field "${e}": Image type is not supported in output fields`,this.position,this.getErrorContext(),"Image types can only be used in input fields");if(i==="audio")throw new C(`Output field "${e}": Audio type is not supported in output fields`,this.position,this.getErrorContext(),"Audio types can only be used in input fields")}catch(i){throw i instanceof C?i:new C(`Output field "${e}": ${i instanceof Error?i.message:"Unknown error"}`,this.position,this.getErrorContext())}this.skipWhitespace();let r=this.parseParsedString();return{name:e,desc:r?.trim(),type:o,isOptional:t,isInternal:n}}validateFieldName(e,t){if(M.signatureStrict&&["text","object","image","string","number","boolean","json","array","datetime","date","time","type","class","input","output","data","value","result","response","request","item","element"].includes(e.toLowerCase())){let i=t==="input"?["userInput","questionText","documentContent","messageText"]:["responseText","analysisResult","categoryType","summaryText"];throw new C(`Field name "${e}" is too generic`,this.position,this.getErrorContext(),`Use a more descriptive name. Examples: ${i.join(", ")}`)}let n=/^[a-z][a-zA-Z0-9]*$/,o=/^[a-z]+(_[a-z0-9]+)*$/;if(!n.test(e)&&!o.test(e))throw new C(`Invalid field name "${e}"`,this.position,this.getErrorContext(),'Field names must be in camelCase (e.g., "userInput") or snake_case (e.g., "user_input")');if(e.length<2)throw new C(`Field name "${e}" is too short`,this.position,this.getErrorContext(),"Field names must be at least 2 characters long");if(e.length>50)throw new C(`Field name "${e}" is too long (${e.length} characters)`,this.position,this.getErrorContext(),"Field names should be 50 characters or less")}parseTypeNotClass(){let e=["string","number","boolean","json","image","audio","file","url","datetime","date","code"],t=e.find(n=>this.match(n));if(!t){let n=this.input.slice(this.position).match(/^\w+/)?.[0]||"",o=this.suggestType(n),r=`Invalid type "${n||"empty"}"`,i=o?`. Did you mean "${o}"?`:"",a=`${r}${i}`;throw new C(a,this.position,this.getErrorContext(),`Expected one of: ${e.join(", ")}`)}return t}suggestType(e){return{str:"string",text:"string",int:"number",integer:"number",float:"number",double:"number",bool:"boolean",object:"json",dict:"json",timestamp:"datetime",time:"datetime",img:"image",picture:"image",sound:"audio",voice:"audio",classification:"class",category:"class"}[e.toLowerCase()]||null}parseParsedIdentifier(){this.skipWhitespace();let e=/^[a-zA-Z_][a-zA-Z_0-9]*/.exec(this.input.slice(this.position));if(e)return this.position+=e[0].length,e[0];let t=/^\S+/.exec(this.input.slice(this.position)),n=t?t[0]:"";throw n===""?new C("Expected field name but found end of input",this.position,this.getErrorContext(),"Add a field name. Field names must start with a letter or underscore"):/^\d/.test(n)?new C(`Invalid field name "${n}" - cannot start with a number`,this.position,this.getErrorContext(),'Field names must start with a letter or underscore. Example: "userInput" or "_internal"'):new C(`Invalid field name "${n}"`,this.position,this.getErrorContext(),"Field names must start with a letter or underscore and contain only letters, numbers, or underscores")}parseParsedString(){let e=["'",'"'];for(let t of e)if(this.match(t)){let n="",o=!1,r=this.position-1;for(;this.position<this.input.length;){let a=this.input[this.position];if(this.position++,o)n+=a,o=!1;else if(a==="\\")o=!0;else{if(a===t)return n;n+=a}}let i=this.input.slice(r,Math.min(this.position,r+20));throw new C(`Unterminated string starting at position ${r}`,r,this.getErrorContext(),`Add closing ${t} to complete the string: ${i}${t}`)}}skipWhitespace(){let e=/^[\s\t\r\n]+/.exec(this.input.slice(this.position));e&&(this.position+=e[0].length)}match(e){let t;if(typeof e=="string"){if(this.input.startsWith(e,this.position))return this.position+=e.length,!0}else if(t=e.exec(this.input.slice(this.position)),t)return this.position+=t[0].length,!0;return!1}expectArrow(){if(!this.match("->")){let e=this.input.slice(this.position,this.position+10),t=e.includes(">")?'Use "->" (dash followed by greater-than)':e.includes("-")?'Add ">" after the dash':'Add "->" to separate input and output fields';throw new C(`Expected "->" but found "${e}..."`,this.position,this.getErrorContext(),t)}}};function Ds(s){return new jo(s).parse()}var bt=class{inputFields=[];outputFields=[];desc;input(e,t,n=!1){let o={name:e,type:{name:t.type,isArray:t.isArray,options:t.options?[...t.options]:void 0},description:t.description,isOptional:t.isOptional,isInternal:t.isInternal};return n?this.inputFields.unshift(o):this.inputFields.push(o),this}output(e,t,n=!1){let o={name:e,type:{name:t.type,isArray:t.isArray,options:t.options?[...t.options]:void 0},description:t.description,isOptional:t.isOptional,isInternal:t.isInternal};return n?this.outputFields.unshift(o):this.outputFields.push(o),this}description(e){return this.desc=e,this}build(){let e={description:this.desc,inputs:this.inputFields,outputs:this.outputFields};return new P(e)}},L=class s{type;isArray;options;description;isOptional;isInternal;constructor(e){this.type=e.type,this.isArray=e.isArray,this.options=e.options,this.description=e.description,this.isOptional=e.isOptional,this.isInternal=e.isInternal}optional(){return new s({...this,isOptional:!0})}array(){return new s({...this,isArray:!0})}internal(){return new s({...this,isInternal:!0})}},Be=Object.assign(()=>new bt,{string:s=>new L({type:"string",isArray:!1,description:s}),number:s=>new L({type:"number",isArray:!1,description:s}),boolean:s=>new L({type:"boolean",isArray:!1,description:s}),json:s=>new L({type:"json",isArray:!1,description:s}),datetime:s=>new L({type:"datetime",isArray:!1,description:s}),date:s=>new L({type:"date",isArray:!1,description:s}),class:(s,e)=>new L({type:"class",isArray:!1,options:s,description:e}),image:s=>new L({type:"image",isArray:!1,description:s}),audio:s=>new L({type:"audio",isArray:!1,description:s}),file:s=>new L({type:"file",isArray:!1,description:s}),url:s=>new L({type:"url",isArray:!1,description:s}),code:(s,e)=>new L({type:"code",isArray:!1,description:e||s}),array:s=>({...s,isArray:!0}),optional:s=>({...s,isOptional:!0}),internal:s=>({...s,isInternal:!0}),legacyArray:s=>({...s,isArray:!0}),legacyOptional:s=>({...s,isOptional:!0}),legacyInternal:s=>({...s,isInternal:!0})});function bn(s){return{type:{name:s.type,isArray:s.isArray,options:s.options?[...s.options]:void 0},description:s.description,isOptional:s.isOptional,isInternal:s.isInternal}}var b=class extends Error{constructor(t,n,o){super(t);this.fieldName=n;this.suggestion=o;this.name="AxSignatureValidationError"}},P=class s{description;inputFields;outputFields;sigHash;sigString;validatedAtHash;constructor(e){if(!e){this.inputFields=[],this.outputFields=[],this.sigHash="",this.sigString="";return}if(typeof e=="string"){let t;try{t=Ds(e)}catch(n){if(n instanceof Error){let o="suggestion"in n&&typeof n.suggestion=="string"?n.suggestion:'Please check the signature format. Example: "userInput:string -> responseText:string"';throw new b(`Invalid Signature: ${n.message}`,void 0,o)}throw new b(`Invalid Signature: ${e}`,void 0,'Please check the signature format. Example: "userInput:string -> responseText:string"')}this.description=t.desc,this.inputFields=t.inputs.map(n=>this.parseParsedField(n)),this.outputFields=t.outputs.map(n=>this.parseParsedField(n)),[this.sigHash,this.sigString]=this.updateHash()}else if(e instanceof s)this.description=e.getDescription(),this.inputFields=structuredClone(e.getInputFields()),this.outputFields=structuredClone(e.getOutputFields()),this.sigHash=e.hash(),this.sigString=e.toString(),e.validatedAtHash===this.sigHash&&(this.validatedAtHash=this.sigHash);else if(typeof e=="object"&&e!==null){if(!("inputs"in e)||!("outputs"in e))throw new b("Invalid signature object: missing inputs or outputs",void 0,'Signature object must have "inputs" and "outputs" arrays. Example: { inputs: [...], outputs: [...] }');if(!Array.isArray(e.inputs)||!Array.isArray(e.outputs))throw new b("Invalid signature object: inputs and outputs must be arrays",void 0,'Both "inputs" and "outputs" must be arrays of AxField objects');try{this.description=e.description,this.inputFields=e.inputs.map(t=>this.parseField(t)),this.outputFields=e.outputs.map(t=>this.parseField(t)),[this.sigHash,this.sigString]=this.updateHash()}catch(t){throw t instanceof b?t:new b(`Failed to create signature from object: ${t instanceof Error?t.message:"Unknown error"}`,void 0,"Check that all fields in inputs and outputs arrays are valid AxField objects")}}else throw new b("Invalid signature argument type",void 0,"Signature must be a string, another AxSignature instance, or an object with inputs and outputs arrays")}static create(e){return new s(e)}parseParsedField=e=>{if(!e.name||e.name.length===0)throw new b("Field name is required",e.name,'Every field must have a descriptive name. Example: "userInput", "responseText"');let t=this.toTitle(e.name);return{name:e.name,title:t,description:"desc"in e?e.desc:void 0,type:e.type??{name:"string",isArray:!1},..."isInternal"in e?{isInternal:e.isInternal}:{},..."isOptional"in e?{isOptional:e.isOptional}:{}}};parseField=e=>{let t=!e.title||e.title.length===0?this.toTitle(e.name):e.title;if(e.type&&(!e.type.name||e.type.name.length===0))throw new b("Field type name is required",e.name,"Specify a valid type. Available types: string, number, boolean, json, image, audio, file, url, date, datetime, class, code");return{...e,title:t}};setDescription=e=>{if(typeof e!="string")throw new b("Description must be a string",void 0,"Provide a string description for the signature");this.description=e,this.invalidateValidationCache(),this.updateHashLight()};addInputField=e=>{try{let t=this.parseField(e);ee(t,"input");for(let n of this.inputFields)if(n.name===t.name)throw new b(`Duplicate input field name: "${t.name}"`,t.name,"Each field name must be unique within the signature");for(let n of this.outputFields)if(n.name===t.name)throw new b(`Field name "${t.name}" appears in both inputs and outputs`,t.name,"Use different names for input and output fields to avoid confusion");this.inputFields.push(t),this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to add input field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};addOutputField=e=>{try{let t=this.parseField(e);ee(t,"output");for(let n of this.outputFields)if(n.name===t.name)throw new b(`Duplicate output field name: "${t.name}"`,t.name,"Each field name must be unique within the signature");for(let n of this.inputFields)if(n.name===t.name)throw new b(`Field name "${t.name}" appears in both inputs and outputs`,t.name,"Use different names for input and output fields to avoid confusion");this.outputFields.push(t),this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to add output field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};setInputFields=e=>{if(!Array.isArray(e))throw new b("Input fields must be an array",void 0,"Provide an array of field objects");try{let t=e.map(n=>{let o=this.parseField(n);return ee(o,"input"),o});this.inputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to set input fields: ${t instanceof Error?t.message:"Unknown error"}`)}};setOutputFields=e=>{if(!Array.isArray(e))throw new b("Output fields must be an array",void 0,"Provide an array of field objects");try{let t=e.map(n=>{let o=this.parseField(n);return ee(o,"output"),o});this.outputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to set output fields: ${t instanceof Error?t.message:"Unknown error"}`)}};getInputFields=()=>this.inputFields;getOutputFields=()=>this.outputFields;getDescription=()=>this.description;appendInputField=(e,t)=>{let n=new s(this);return n.addInputField({name:e,...bn(t)}),n};prependInputField=(e,t)=>{let n=new s(this),o={name:e,...bn(t)},r=n.parseField(o);ee(r,"input");for(let i of n.inputFields)if(i.name===r.name)throw new b(`Duplicate input field name: "${r.name}"`,r.name,"Each field name must be unique within the signature");for(let i of n.outputFields)if(i.name===r.name)throw new b(`Field name "${r.name}" appears in both inputs and outputs`,r.name,"Use different names for input and output fields to avoid confusion");return n.inputFields.unshift(r),n.invalidateValidationCache(),n.updateHashLight(),n};appendOutputField=(e,t)=>{let n=new s(this);return n.addOutputField({name:e,...bn(t)}),n};prependOutputField=(e,t)=>{let n=new s(this),o={name:e,...bn(t)},r=n.parseField(o);ee(r,"output");for(let i of n.outputFields)if(i.name===r.name)throw new b(`Duplicate output field name: "${r.name}"`,r.name,"Each field name must be unique within the signature");for(let i of n.inputFields)if(i.name===r.name)throw new b(`Field name "${r.name}" appears in both inputs and outputs`,r.name,"Use different names for input and output fields to avoid confusion");return n.outputFields.unshift(r),n.invalidateValidationCache(),n.updateHashLight(),n};invalidateValidationCache=()=>{this.validatedAtHash=void 0};toTitle=e=>{let t=e.replace(/_/g," ");return t=t.replace(/([A-Z]|[0-9]+)/g," $1").trim(),t.charAt(0).toUpperCase()+t.slice(1)};toJSONSchema=()=>{let e={},t=[];for(let o of this.inputFields){let r=o.type?o.type.name:"string";o.type?.isArray?e[o.name]={description:o.description,type:"array",items:{type:r,description:o.description}}:e[o.name]={description:o.description,type:r},o.isOptional||t.push(o.name)}return{type:"object",properties:e,required:t}};updateHashLight=()=>{try{return this.getInputFields().forEach(e=>{ee(e,"input")}),this.getOutputFields().forEach(e=>{ee(e,"output")}),this.sigHash=Mt("sha256").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Gs(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof b?e:new b(`Signature validation failed: ${e instanceof Error?e.message:"Unknown error"}`)}};updateHash=()=>{try{return this.getInputFields().forEach(e=>{ee(e,"input")}),this.getOutputFields().forEach(e=>{ee(e,"output")}),this.validateSignatureConsistency(),this.sigHash=Mt("sha256").update(this.description??"").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Gs(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof b?e:new b(`Signature validation failed: ${e instanceof Error?e.message:"Unknown error"}`)}};validateSignatureConsistency(){let e=new Set;for(let n of this.inputFields){if(e.has(n.name))throw new b(`Duplicate input field name: "${n.name}"`,n.name,"Each field name must be unique within the signature");e.add(n.name)}let t=new Set;for(let n of this.outputFields){if(t.has(n.name))throw new b(`Duplicate output field name: "${n.name}"`,n.name,"Each field name must be unique within the signature");t.add(n.name)}for(let n of this.outputFields)if(e.has(n.name))throw new b(`Field name "${n.name}" appears in both inputs and outputs`,n.name,"Use different names for input and output fields to avoid confusion");if(this.inputFields.length===0)throw new b("Signature must have at least one input field",void 0,'Add an input field. Example: "userInput:string -> ..."');if(this.outputFields.length===0)throw new b("Signature must have at least one output field",void 0,'Add an output field. Example: "... -> responseText:string"')}validate=()=>{if(this.validatedAtHash===this.sigHash)return!0;try{return this.updateHash(),this.validatedAtHash=this.sigHash,!0}catch(e){throw this.validatedAtHash=void 0,e}};hash=()=>this.sigHash;toString=()=>this.sigString;toJSON=()=>({id:this.hash(),description:this.description,inputFields:this.inputFields,outputFields:this.outputFields})};function Ls(s){let e=s.name;return s.isOptional&&(e+="?"),s.isInternal&&(e+="!"),s.type&&(e+=`:${s.type.name}`,s.type.isArray&&(e+="[]"),s.type.name==="class"&&s.type.options&&(e+=` "${s.type.options.join(" | ")}"`)),s.description&&s.type?.name!=="class"&&(e+=` "${s.description}"`),e}function Gs(s,e,t){let n=s?`"${s}" `:"",o=e.map(Ls).join(", "),r=t.map(Ls).join(", ");return`${n}${o} -> ${r}`}function Hi(s){let e=/^[a-z][a-zA-Z0-9]*$/,t=/^[a-z]+(_[a-z0-9]+)*$/;return e.test(s)||t.test(s)}function ee(s,e){if(!s.name||s.name.length===0)throw new b("Field name cannot be blank",s.name,"Every field must have a descriptive name");if(!Hi(s.name))throw new b(`Invalid field name '${s.name}' - must be camelCase or snake_case`,s.name,'Use camelCase (e.g., "userInput") or snake_case (e.g., "user_input")');if(M.signatureStrict&&["text","object","image","string","number","boolean","json","array","datetime","date","time","type","class","input","output","data","value","result","response","request","item","element"].includes(s.name.toLowerCase())){let n=e==="input"?["userInput","questionText","documentContent","messageText","queryString"]:["responseText","analysisResult","categoryType","summaryText","outputData"];throw new b(`Field name '${s.name}' is too generic`,s.name,`Use a more descriptive name. Examples for ${e} fields: ${n.join(", ")}`)}if(s.name.length<2)throw new b(`Field name '${s.name}' is too short`,s.name,"Field names must be at least 2 characters long");if(s.name.length>50)throw new b(`Field name '${s.name}' is too long (${s.name.length} characters)`,s.name,"Field names should be 50 characters or less");s.type&&Ki(s,e)}function Ki(s,e){if(!s.type)return;let{type:t}=s;if(t.name==="image"||t.name==="audio"||t.name==="file"||t.name==="url"){if(e==="output")throw new b(`${t.name} type is not supported in output fields`,s.name,`${t.name} types can only be used in input fields`);if(t.isArray&&(t.name==="image"||t.name==="audio"))throw new b(`Arrays of ${t.name} are not supported`,s.name,`Use a single ${t.name} type instead`)}if(t.name==="class"){if(e==="input")throw new b("Class type is not supported in input fields",s.name,'Class types are only allowed on output fields. Use "string" type for input classifications');if(!t.options||t.options.length===0)throw new b("Class type requires options",s.name,'Provide class options. Example: class "positive, negative, neutral"');for(let o of t.options){if(!o||o.trim().length===0)throw new b("Empty class option found",s.name,"All class options must be non-empty strings");let r=o.trim();if(r.includes(",")||r.includes("|"))throw new b(`Invalid class option "${r}"`,s.name,"Class options cannot contain commas (,) or pipes (|) as they are used to separate options")}if(new Set(t.options.map(o=>o.trim().toLowerCase())).size!==t.options.length)throw new b("Duplicate class options found",s.name,"Each class option must be unique (case-insensitive)")}if(t.name==="code"&&t.isArray)throw new b("Arrays of code are not commonly supported",s.name,"Consider using a single code field or an array of strings instead");if(s.isInternal&&e==="input")throw new b("Internal marker (!) is not allowed on input fields",s.name,"Internal markers are only allowed on output fields")}var ie=class{signature;sigHash;examples;examplesOptions;demos;trace;usage=[];traceLabel;key;children;constructor(e,t){this.signature=new P(e),t?.description&&this.signature.setDescription(t.description),t?.traceLabel&&(this.traceLabel=t.traceLabel),e&&this.signature.validate(),this.sigHash=this.signature?.hash(),this.children=new Ue,this.key={id:this.signature.hash()}}getSignature(){return new P(this.signature)}setSignature(e){this.signature=new P(e),e&&this.signature.validate(),this.updateSignatureHash()}setDescription(e){this.signature.setDescription(e),this.updateSignatureHash()}updateSignatureHash(){this.sigHash=this.signature.hash(),this.key={id:this.signature.hash()}}register(e){this.key&&e.setParentId(this.key.id),this.children.register(e)}setId(e){this.key={id:e,custom:!0};for(let t of Array.from(this.children))t?.setParentId(e)}setParentId(e){this.key.custom||(this.key.id=[e,this.key.id].join("/"))}setExamples(e,t){if(this._setExamples(e,t),"programId"in e)for(let n of Array.from(this.children))n?.setExamples(e,t)}_setExamples(e,t){let n=[];if("programId"in e&&e.programId===this.key.id&&(n=e.traces),Array.isArray(e)&&(n=e),n){this.examplesOptions=t;let o=this.signature,r=[...o.getInputFields(),...o.getOutputFields()];this.examples=n.map(i=>{let a={};for(let l of r){let p=i[l.name];p!==void 0&&(In(l,p),a[l.name]=p)}return a})}}getTraces(){let e=[];this.trace&&e.push({trace:this.trace,programId:this.key.id});for(let t of Array.from(this.children)){let n=t?.getTraces();e=[...e,...n??[]]}return e}getUsage(){let e=[...this.usage??[]];for(let t of Array.from(this.children)){let n=t?.getUsage();e=[...e,...n??[]]}return yt(e)}resetUsage(){this.usage=[];for(let e of Array.from(this.children))e?.resetUsage()}setDemos(e){let t=Array.from(this.children).length>0,n=e.some(o=>o.programId===this.key.id);if(t&&!n)throw new Error(`Program with id '${this.key.id}' has children but no matching programId found in demos`);this.demos=e.filter(o=>o.programId===this.key.id).map(o=>o.traces).flat();for(let o of Array.from(this.children))o?.setDemos(e)}};var Vi=`
151
+ `)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},ft=class{funcList=[];constructor(e){this.funcList=e}executeFunction=async(e,t,n)=>{let r;typeof t.args=="string"&&t.args.length>0?r=JSON.parse(t.args):r=t.args;let o=n?{sessionId:n.sessionId,ai:n.ai}:void 0,i;return e.parameters?i=e.func.length===2?await e.func(r,o):await e.func(r):i=e.func.length===1?await e.func(o):await e.func(),(n?.functionResultFormatter??M.functionResultFormatter)(i)};execute=async(e,t)=>{let n=this.funcList.find(r=>r.name.localeCompare(e.name)===0);if(!n)throw new Error(`Function not found: ${e.name}`);if(!n.func)throw new Error(`No handler for function: ${e.name}`);try{return await this.executeFunction(n,e,t)}catch(r){throw r instanceof xt?new In(r.getFields(),n,e.id):r}}},_r=(s,e)=>{if(s.length===0)return[...e??[]];let t=s.map(n=>"toFunction"in n?n.toFunction():n).flat();for(let n of t.filter(r=>r.parameters))n.parameters&&Is(n.parameters);return[...e??[],...t]},Dr=async({ai:s,functionList:e,functionCalls:t,mem:n,sessionId:r,span:o,excludeContentFromTrace:i,index:a,functionResultFormatter:l,logger:p})=>{let c=new ft(e),u=new Set,d=t.map(g=>{if(!g.id)throw new Error(`Function ${g.name} did not return an ID`);return c.execute(g,{sessionId:r,ai:s,functionResultFormatter:l}).then(A=>{if(u.add(g.name.toLowerCase()),o){let x={name:g.name};i||(x.args=g.args,x.result=A??""),o.addEvent("function.call",x)}return{result:A??"",role:"function",functionId:g.id,index:a}}).catch(A=>{if(!(A instanceof In))throw A;let x=A.getFixingInstructions();if(o){let y={name:g.name,message:A.toString()};i||(y.args=g.args,y.fixing_instructions=x),o.addEvent("function.error",y)}return s.getOptions().debug&&uo(A,a,x,p),{functionId:g.id,isError:!0,index:a,result:x,role:"function"}})}),h=(await Promise.all(d)).filter(g=>g!==void 0);if(n.addFunctionResults(h,r),s.getOptions().debug){let g=h.filter(f=>!f.isError);g.length>0&&co(g,p)}return h.some(g=>g.isError)&&n.addTag("error",r),u};function Lr(s,e,t,n){if(!e||e.length===0)return;if(!s.getFeatures(n).functions)throw new Error("Functions are not supported by the AI service");return e.map(o=>({id:o.id,name:o.function.name,args:o.function.params}))}function bs(s,e,t,n){let r=e;return n?.functionCallMode==="prompt"?{functions:[],functionCall:void 0}:!t&&(r==="required"||typeof r=="function")?{functions:[],functionCall:void 0}:s?{functions:s.map(i=>"toFunction"in i?i.toFunction():i).flat(),functionCall:r}:{functions:[],functionCall:r}}var se=Mt(require("dayjs"),1),Rs=Mt(require("dayjs/plugin/customParseFormat.js"),1),Ts=Mt(require("dayjs/plugin/timezone.js"),1),Cs=Mt(require("dayjs/plugin/utc.js"),1);se.default.extend(Cs.default);se.default.extend(Ts.default);se.default.extend(Rs.default);function ws(s,e,t=!1){try{return Ni(e)}catch(n){if(s.isOptional&&!t)return;let r=n.message;throw new D({fields:[s],message:r,value:e})}}function Ni(s){if(!(0,se.default)(s,"YYYY-MM-DD",!0).isValid())throw new Error('Invalid date format. Please provide the date in "YYYY-MM-DD" format.');return se.default.utc(s,"YYYY-MM-DD").startOf("day").toDate()}function Ss(s,e,t=!1){try{return Ui(e)}catch(n){if(s.isOptional&&!t)return;let r=n.message;throw new D({fields:[s],message:r,value:e})}}function Ui(s){let e=/^(\d{4}-\d{2}-\d{2} \d{2}:\d{2}(?::\d{2})?) (.+)$/,t=s.match(e);if(!t)throw new Error('Invalid date and time format. Please provide the date and time in "YYYY-MM-DD HH:mm" or "YYYY-MM-DD HH:mm:ss" format, followed by the timezone.');let[,n,r]=t;if(!n||!r)throw new Error('Invalid date and time format. Please provide the date and time in "YYYY-MM-DD HH:mm" or "YYYY-MM-DD HH:mm:ss" format, followed by the timezone.');try{let o=n.includes(":")&&n.split(":").length===3?"YYYY-MM-DD HH:mm:ss":"YYYY-MM-DD HH:mm",i=se.default.tz(n,o,r);if(!i.isValid())throw new Error("Invalid date and time values. Please ensure all components are correct.");return i.utc().toDate()}catch{throw new Error(`Unrecognized time zone ${r}. Please provide a valid time zone name, abbreviation, or offset. For example, "America/New_York", or "EST".`)}}var vs=s=>(0,se.default)(s).utc().format("YYYY-MM-DD HH:mm:ss [UTC]");var wu=new G;var bn=(s,e)=>{let t=s.type??{name:"string",isArray:!1},n=(p,c)=>{switch(p){case"class":return typeof c=="string";case"code":return typeof c=="string";case"string":return typeof c=="string";case"number":return typeof c=="number";case"boolean":return typeof c=="boolean";case"date":return c instanceof Date||typeof c=="string";case"datetime":return c instanceof Date||typeof c=="string";case"json":return typeof c=="object"||typeof c=="string";default:return!1}},r=p=>!(!p||typeof p!="object"||!("mimeType"in p)||!("data"in p));if(s.type?.name==="image"){let p;if(Array.isArray(e)){for(let c of e)if(!r(c)){p="object ({ mimeType: string; data: string })";break}}else r(e)||(p="object ({ mimeType: string; data: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let o=p=>!(!p||typeof p!="object"||!("data"in p));if(s.type?.name==="audio"){let p;if(Array.isArray(e)){for(let c of e)if(!o(c)){p="object ({ data: string; format?: string })";break}}else o(e)||(p="object ({ data: string; format?: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let i=p=>{if(!p||typeof p!="object"||!("mimeType"in p))return!1;let c="data"in p,u="fileUri"in p;return!(!c&&!u||c&&u)};if(s.type?.name==="file"){let p;if(Array.isArray(e)){for(let c of e)if(!i(c)){p="object ({ mimeType: string; data: string } | { mimeType: string; fileUri: string })";break}}else i(e)||(p="object ({ mimeType: string; data: string } | { mimeType: string; fileUri: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let a=p=>typeof p=="string"?!0:!(!p||typeof p!="object"||!("url"in p));if(s.type?.name==="url"){let p;if(Array.isArray(e)){for(let c of e)if(!a(c)){p="string or object ({ url: string; title?: string; description?: string })";break}}else a(e)||(p="string or object ({ url: string; title?: string; description?: string })");if(p)throw new Error(`Validation failed: Expected '${s.name}' to be type '${p}' instead got '${e}'`);return}let l=!0;if(t.isArray){if(!Array.isArray(e))l=!1;else for(let p of e)if(!n(t.name,p)){l=!1;break}}else l=n(t.name,e);if(!l){let p=Array.isArray(e)?"array":typeof e;throw new Error(`Validation failed: Expected '${s.name}' to be a ${s.type?.isArray?"an array of ":""}${t.name} instead got '${p}' (${JSON.stringify(e)})`)}};function yt(s){let e={};for(let t of s){let n=`${t.ai}:${t.model}`;if(!e[n]){e[n]={...t};continue}let r=e[n];if(r){let o=r.tokens??{promptTokens:0,completionTokens:0,totalTokens:0};o.promptTokens+=t?.tokens?.promptTokens??0,o.completionTokens+=t?.tokens?.completionTokens??0,o.totalTokens+=t?.tokens?.totalTokens??0,r.tokens=o}}return Object.values(e)}var Os=s=>{if(!s.trim())return[];let e=new Set(["-","*","+"]),t=/^\d+[\s]*[.)\]]\s*/,n=s.split(`
152
+ `),r=[];for(let o of n){let i=o.trim();if(i){if(i[0]&&e.has(i[0]))r.push(i.slice(1).trim());else if(t.test(i))r.push(i.replace(t,"").trim());else if(r.length!==0)throw new Error("Could not parse markdown list: mixed content detected")}}if(r.length===0)throw new Error("Could not parse markdown list: no valid list items found");return r};function $r(s,e){let{index:t,delta:n,version:r}=e,o=s.find(i=>i.index===t)?.delta;if(!o)return s.push({index:t,delta:n,version:r}),s;for(let i of Object.keys(n)){let a=o[i],l=n[i];a===void 0&&Array.isArray(l)?o[i]=[...l]:Array.isArray(a)&&Array.isArray(l)?o[i]=[...a,...l]:(a===void 0||typeof a=="string")&&typeof l=="string"?o[i]=`${a??""}${l}`:o[i]=l}return s}var Gr=class{cache=new Map;maxSize;constructor(e){this.maxSize=e}get(e){let t=this.cache.get(e);return t&&(this.cache.delete(e),this.cache.set(e,t)),t}set(e,t){if(this.cache.has(e))this.cache.delete(e);else if(this.cache.size>=this.maxSize){let n=this.cache.keys().next().value;n&&this.cache.delete(n)}this.cache.set(e,t)}},Bi=new Gr(500);function Nr(s,e,t=0,n=Bi){if(/^```[a-zA-Z]*\s*$/.test(s))return-4;if(/^[\s`]*$/.test(s))return-3;let r=s.indexOf(e,t);if(r!==-1)return r;let o=n.get(e)??Array.from({length:e.length},(a,l)=>e.slice(0,l+1));n.get(e)||n.set(e,o);let i=-1;for(let a=o.length-1;a>=0;a--){let l=o[a];if(s.endsWith(l)){i=a;break}}return i>=0?-2:-1}var Es=(s,e,t,n=!1)=>{let r={extractedFields:[],streamedIndex:{},s:-1};Ur(s,e,r,t,{strictMode:n}),Br(s,e,r,t,n);for(let o of s.getOutputFields())o.isInternal&&delete e[o.name]},qi=(s,e,t)=>{let n=[];for(let r of t)r&&!r.isOptional&&e[r.name]===void 0&&n.push(r);if(n.length>0)throw new D({message:`Required ${n.length===1?"field":"fields"} not found`,fields:n})},Ur=(s,e,t,n,{strictMode:r,skipEarlyFail:o}={})=>{let i=s.getOutputFields(),a;for(let[l,p]of i.entries()){if(l===t.currFieldIndex&&!t.inAssumedField||p.name in e&&!(l===t.currFieldIndex&&t.inAssumedField))continue;let u=`${(t.extractedFields.length===0?"":`
153
+ `)+p.title}:`,d=Nr(n,u,t.s),m=u.length;switch(d){case-1:if(o)continue;if(!r&&i.length===1&&t.currField===void 0){t.inAssumedField=!0,a=p,m=0,d=0;break}if(t.currField===void 0&&t.extractedFields.length===0){if(r&&!p.isOptional)throw new D({message:"Expected (Required) field not found",fields:[p]});if(!r){let h=!1;for(let g=l;g<i.length;g++){let f=i[g];if(!f)continue;let A=`${(t.extractedFields.length===0?"":`
154
+ `)+f.title}:`;if(Nr(n,A,t.s)>=0){h=!0;break}}if(!h){t.inAssumedField=!0,a=p,m=0,d=0;break}}}a=p.isOptional?void 0:p;continue;case-2:return!0;case-3:return!0;case-4:return t.inBlock=!0,!0}if(a&&a.name!==p.name)throw new D({message:"Expected (Required) field not found",fields:[a]});if(t.currField!==void 0&&t.inAssumedField){let h=n.substring(0,d).trim();if(h&&t.currField.name===p.name){let g=It(t.currField,h);g!==void 0&&(e[t.currField.name]=g)}else if(h){let g=It(t.currField,h);g!==void 0&&(e[t.currField.name]=g)}t.inAssumedField=!1,t.streamedIndex[t.currField.name]=0,t.currField=void 0}if(t.currField){let h=n.substring(t.s,d).trim(),g=It(t.currField,h);g!==void 0&&(e[t.currField.name]=g),t.prevFields?t.prevFields?.push({field:t.currField,s:t.s,e:d}):t.prevFields=[{field:t.currField,s:t.s,e:d}]}t.s=d+m,t.currField=p,t.currFieldIndex=l,t.extractedFields.includes(p)||t.extractedFields.push(p),t.streamedIndex[p.name]===void 0&&(t.streamedIndex[p.name]=0)}},Br=(s,e,t,n,r=!1)=>{if(t.currField){let o=n.substring(t.s).trim(),i=It(t.currField,o);i!==void 0&&(e[t.currField.name]=i)}if(r&&!t.currField&&t.extractedFields.length===0&&n.trim()){let a=s.getOutputFields().find(l=>!l.isOptional);if(a)throw new D({message:"Expected field not found",fields:[a]})}zi(s,e,n),qi(t,e,s.getOutputFields())},zi=(s,e,t)=>{let n=s.getOutputFields();for(let r of n){if(!r.isOptional||r.name in e)continue;let o=`${r.title}:`,i=t.indexOf(o);if(i===-1)continue;let a=i+o.length,l=t.length;for(let c of n){if(c.name===r.name)continue;let u=`${c.title}:`,d=t.indexOf(u,a);d!==-1&&d<l&&(l=d)}let p=t.substring(a,l).trim();if(p)try{let c=It(r,p);c!==void 0&&(e[r.name]=c)}catch{}}},Ms=(s,e,t=!1)=>{switch(s.type?.name){case"code":return Ps(e);case"string":return e;case"number":{let n=Number(e);if(Number.isNaN(n)){if(s.isOptional&&!t)return;throw new Error("Invalid number")}return n}case"boolean":{if(typeof e=="boolean")return e;let n=e.toLowerCase();if(n==="true")return!0;if(n==="false")return!1;if(s.isOptional&&!t)return;throw new Error("Invalid boolean")}case"date":return ws(s,e,t);case"datetime":return Ss(s,e,t);case"class":{let n=e;if(s.type.options&&!s.type.options.includes(n)){if(s.isOptional)return;throw new Error(`Invalid class '${e}', expected one of the following: ${s.type.options.join(", ")}`)}return n}default:return e}};function*ks(s,e,t,n,r,o){let{name:i,isInternal:a}=e,{isArray:l,name:p}=e.type??{};if(a||l||p&&p!=="string"&&p!=="code")return;let c=r.streamedIndex[i]??0,u=c===0,d=s.substring(t+c,n);if(d.length===0)return;let m=d.replace(/\s+$/,"");r.currField?.type?.name==="code"&&(m=m.replace(/\s*```\s*$/,""));let h=u?m.trimStart():m;r.currField?.type?.name==="code"&&(h=h.replace(/^[ ]*```[a-zA-Z0-9]*\n\s*/,"")),h.length>0&&(yield{index:o,delta:{[i]:h}},r.streamedIndex[i]=c+m.length)}function*qr(s,e,t,n,r){for(let i of n.prevFields??[]){let{field:a,s:l,e:p}=i;yield*ks(e,a,l,p,n,r)}if(n.prevFields=void 0,!n.currField||n.currField.isInternal)return;yield*ks(e,n.currField,n.s,e.length,n,r);let o=s.getOutputFields();for(let i of Object.keys(t)){let a=o.find(p=>p.name===i);if(!a||a.isInternal)continue;let l=t[i];if(Array.isArray(l)){let p=n.streamedIndex?.[i]??0,c=l.slice(p);c&&c.length>0&&(yield{index:r,delta:{[i]:c}},n.streamedIndex[i]=p+c.length);continue}n.streamedIndex[i]||(yield{index:r,delta:{[i]:l}},n.streamedIndex[i]=1)}}function It(s,e){if(!e||e===""||/^(null|undefined)\s*$/i.test(e)){if(s.isOptional)return;throw new D({message:"Required field is missing",fields:[s],value:e})}let t;if(s.type?.name==="json")try{let n=Ps(e);return t=JSON.parse(n),t}catch(n){throw new D({message:`Invalid JSON: ${n.message}`,fields:[s],value:e})}if(s.type?.isArray)try{try{t=JSON.parse(e)}catch{t=Os(e)}if(!Array.isArray(t))throw new Error("Expected an array")}catch(n){throw new D({message:`Invalid Array: ${n.message}`,fields:[s],value:e})}try{if(Array.isArray(t)){for(let[n,r]of t.entries())if(r!==void 0){let o=typeof r=="string"?r.trim():r;t[n]=Ms(s,o,!0)}}else t=Ms(s,e)}catch(n){throw new D({message:n.message,fields:[s],value:e})}if(!(typeof t=="string"&&t===""))return t}var Ps=s=>{let t=/```([A-Za-z]*)\n([\s\S]*?)\n```/g.exec(s);return t?t.length===3?t[2]:t.length===2?t[1]:s:s};async function zr(s,e,t,n){for(let r of s){if(e[r.field.name]===void 0)continue;let o=r.process,i=await o(e[r.field.name],{sessionId:n,values:e,done:!0});Fs(r.field,t,i,n)}}async function jr(s,e,t,n,r,o,i=!1){for(let a of s){if(t.currField?.name!==a.field.name)continue;let l=e.substring(t.s);t.currField?.type?.name==="code"&&(l=l.replace(/^[ ]*```[a-zA-Z0-9]*\n\s*/,""),l=l.replace(/\s*```\s*$/,""));let p=a.process,c=await p(l,{sessionId:o,values:r,done:i});Fs(t.currField,n,c,o)}}var Fs=(s,e,t,n)=>{if(t===void 0||typeof t=="string"&&(t===""||/^(null|undefined)\s*$/i.test(t)))return;let r=JSON.stringify(t,(i,a)=>typeof a=="bigint"?Number(a):a,2),o=ji(s,r);e.addRequest([{role:"user",content:[{type:"text",text:o}]}],n),e.addTag("processor",n)};function ji(s,e){let t=s.type?.name==="code",n=s.title;return t?`Code in the field "${n}" was executed. The code execution produced the following output: ${e}`:`The field "${n}" was processed. The field contents were transformed into the following output: ${e}`}async function*_s({res:s,usage:e,states:t,...n}){let r=(n.ai.getFeatures().functionCot??!1)&&n.functions!==void 0&&n.functions.length>0,o,i=s.getReader();try{for(;;){let{done:a,value:l}=await i.read();if(a){o&&e.push(o);break}let p=l;p.modelUsage&&(o=p.modelUsage);for(let c of p.results){if((!c.content||c.content==="")&&(!c.thought||c.thought==="")&&(!c.functionCalls||c.functionCalls.length===0))continue;let u=t.find(d=>d.index===c.index);if(!u)throw new Error(`No state found for result (index: ${c.index})`);yield*Hi({...n,result:c,skipEarlyFail:r,state:u})}}}finally{i.releaseLock()}for(let a of t)yield*Ki({...n,state:a})}async function*Hi({result:s,mem:e,sessionId:t,strictMode:n,skipEarlyFail:r,state:o,signature:i,streamingFieldProcessors:a,thoughtFieldName:l,streamingAsserts:p,asserts:c}){if(s.functionCalls&&s.functionCalls.length>0)_t(o.functionCalls,s.functionCalls),e.updateResult({name:s.name,content:s.content,functionCalls:o.functionCalls,delta:s.functionCalls?.[0]?.function?.params,index:s.index},t);else if(s.content&&s.content.length>0){if(s.thought&&s.thought.length>0&&(yield{index:s.index,delta:{[l]:s.thought}}),o.content+=s.content,e.updateResult({name:s.name,content:o.content,delta:s.content,index:s.index},t),Ur(i,o.values,o.xstate,o.content,{strictMode:n,skipEarlyFail:r}))return;p.length!==0&&await Or(p,o.xstate,o.content),a.length!==0&&await jr(a,o.content,o.xstate,e,o.values,t),yield*qr(i,o.content,o.values,o.xstate,s.index),await An(c,o.values)}else s.thought&&s.thought.length>0&&(o.values[l]=(o.values[l]??"")+s.thought,yield{index:s.index,delta:{[l]:s.thought}});if(s.finishReason==="length")throw new Error(`Max tokens reached before completion
155
+ Content: ${o.content}`)}async function*Ki({state:s,signature:e,ai:t,model:n,functions:r,mem:o,sessionId:i,traceId:a,span:l,strictMode:p,excludeContentFromTrace:c,streamingAsserts:u,asserts:d,fieldProcessors:m,streamingFieldProcessors:h,functionResultFormatter:g,logger:f}){let A=Lr(t,s.functionCalls,s.values,n);if(A){if(!r)throw new Error("Functions are not defined");let x=await Dr({ai:t,functionList:r,functionCalls:A,mem:o,sessionId:i,traceId:a,span:l,index:s.index,excludeContentFromTrace:c,functionResultFormatter:g,logger:f});s.functionsExecuted=new Set([...s.functionsExecuted,...x])}else Br(e,s.values,s.xstate,s.content,p),await Or(u,s.xstate,s.content,!0),await An(d,s.values),m.length&&await zr(m,s.values,o,i),h.length!==0&&await jr(h,s.content,s.xstate,o,s.values,i,!0),yield*qr(e,s.content,s.values,s.xstate,s.index)}async function*Ds({ai:s,res:e,mem:t,sessionId:n,traceId:r,functions:o,span:i,strictMode:a,states:l,usage:p,excludeContentFromTrace:c,asserts:u,fieldProcessors:d,thoughtFieldName:m,signature:h,functionResultFormatter:g,logger:f}){let A=e.results??[];t.addResponse(A,n);for(let I of A){let R=l[I.index];if(!R)throw new Error(`No state found for result (index: ${I.index})`);if(e.modelUsage&&p.push(e.modelUsage),I.functionCalls?.length){let k=Lr(s,I.functionCalls,R.values);if(k){if(!o)throw new Error("Functions are not defined");let v=await Dr({ai:s,functionList:o,functionCalls:k,mem:t,sessionId:n,traceId:r,span:i,excludeContentFromTrace:c,index:I.index,functionResultFormatter:g,logger:f});R.functionsExecuted=new Set([...R.functionsExecuted,...v])}}else I.content&&(I.thought&&I.thought.length>0&&(R.values[m]=I.thought),Es(h,R.values,I.content,a),await An(u,R.values),d.length&&await zr(d,R.values,t,n));if(I.finishReason==="length")throw new Error(`Max tokens reached before completion
156
+ Content: ${I.content}`)}let x=l.map(I=>I.values);for(let I of x)for(let R of h.getOutputFields())R.isInternal&&delete I[R.name];let y=h.getOutputFields(),w=x.map((I,R)=>{let k={};for(let v of y)v.isInternal||(k[v.name]=I[v.name]);return I[m]!==void 0&&(k[m]=I[m]),{index:R,delta:k}});for(let I of w)yield I}function Ls(s,e,t,n){let r=s.getLast(n);if(!r)return!0;for(let[o,i]of t.entries()){let a=e&&i.functionsExecuted.has(e);if(!r.chat[o])throw new Error(`No chat message found for result (index: ${o})`);let p=r.role==="function",c=r.tags?r.tags.some(u=>u==="processor"):!1;if(p&&e&&a||!(p||c))return!1}return!0}var Ue=class{reg;constructor(){this.reg=new Set}register(e){this.reg.add(e)}*[Symbol.iterator](){let e=Array.from(this.reg);for(let t=0;t<e.length;t++)yield e[t]}};var T=class extends Error{constructor(t,n,r,o){super(t);this.position=n;this.context=r;this.suggestion=o;this.name="SignatureValidationError"}},Hr=class{input;position;currentFieldName=null;currentSection="description";constructor(e){if(this.input=e.trim(),this.position=0,!this.input)throw new T("Empty signature provided",0,"",'A signature must contain at least input and output fields separated by "->". Example: "userQuery:string -> aiResponse:string"')}parse(){try{this.skipWhitespace();let e=this.parseParsedString();this.skipWhitespace(),this.currentSection="inputs";let t=this.parseFieldList(this.parseInputField.bind(this),"input");if(this.skipWhitespace(),this.position>=this.input.length)throw new T("Incomplete signature: Missing output section",this.position,this.getErrorContext(),'Add "->" followed by output fields. Example: "-> responseText:string"');if(this.expectArrow(),this.skipWhitespace(),this.position>=this.input.length)throw new T('Incomplete signature: No output fields specified after "->"',this.position,this.getErrorContext(),'Add at least one output field. Example: "-> responseText:string"');this.currentSection="outputs";let n=this.parseFieldList(this.parseOutputField.bind(this),"output");if(this.skipWhitespace(),this.position<this.input.length){let r=this.input.slice(this.position);throw new T(`Unexpected content after signature: "${r}"`,this.position,this.getErrorContext(),"Remove any extra content after the output fields")}return this.validateParsedSignature({desc:e?.trim(),inputs:t,outputs:n}),{desc:e?.trim(),inputs:t,outputs:n}}catch(e){if(e instanceof T)throw e;let t=e instanceof Error?e.message:"Unknown error";throw new T(t,this.position,this.getErrorContext())}}validateParsedSignature(e){let t=new Set;for(let r of e.inputs){if(t.has(r.name))throw new T(`Duplicate input field name: "${r.name}"`,0,"","Each field name must be unique within the signature");t.add(r.name)}let n=new Set;for(let r of e.outputs){if(n.has(r.name))throw new T(`Duplicate output field name: "${r.name}"`,0,"","Each field name must be unique within the signature");n.add(r.name)}for(let r of e.outputs)if(t.has(r.name))throw new T(`Field name "${r.name}" appears in both inputs and outputs`,0,"","Use different names for input and output fields to avoid confusion");if(e.inputs.length===0)throw new T("Signature must have at least one input field",0,"",'Add an input field before "->". Example: "userInput:string -> ..."');if(e.outputs.length===0)throw new T("Signature must have at least one output field",0,"",'Add an output field after "->". Example: "... -> responseText:string"')}getErrorContext(){let e=Math.max(0,this.position-25),t=Math.min(this.input.length,this.position+25),n=this.input.slice(e,this.position),r=this.input.slice(this.position,t),o=`${" ".repeat(n.length)}^`;return[`Position ${this.position} in signature:`,`"${n}${r}"`,` ${o}`].join(`
157
+ `)}parseFieldList(e,t){let n=[];if(this.skipWhitespace(),this.position>=this.input.length)throw new T(`Empty ${t} section: Expected at least one field`,this.position,this.getErrorContext(),`Add a ${t} field. Example: ${t==="input"?"userInput:string":"responseText:string"}`);try{n.push(e())}catch(r){throw r instanceof T?r:new T(`Invalid first ${t} field: ${r instanceof Error?r.message:"Unknown error"}`,this.position,this.getErrorContext())}for(this.skipWhitespace();this.position<this.input.length&&!(this.input[this.position]==="-"&&this.position+1<this.input.length&&this.input[this.position+1]===">");)if(this.match(",")){if(this.skipWhitespace(),this.position>=this.input.length)throw new T(`Unexpected end of input after comma in ${t} section`,this.position,this.getErrorContext(),`Add another ${t} field after the comma`);try{n.push(e())}catch(r){throw r instanceof T?r:new T(`Invalid ${t} field after comma: ${r instanceof Error?r.message:"Unknown error"}`,this.position,this.getErrorContext())}this.skipWhitespace()}else break;return n}parseInputField(){this.skipWhitespace();let e=this.parseParsedIdentifier();this.currentFieldName=e,this.validateFieldName(e,"input");let t;for(;;){if(this.match("?")){t=!0;continue}if(this.match("!"))throw new T(`Input field "${e}" cannot use the internal marker "!"`,this.position-1,this.getErrorContext(),"Internal markers (!) are only allowed on output fields");break}let n;if(this.skipWhitespace(),this.match(":")){if(this.skipWhitespace(),/^class\b/.test(this.input.slice(this.position)))throw new T(`Input field "${e}" cannot use the "class" type`,this.position,this.getErrorContext(),'Class types are only allowed on output fields. Use "string" type for input classifications');try{let o=this.parseTypeNotClass(),i=this.match("[]");if(n={name:o,isArray:i},(o==="image"||o==="audio")&&i)throw new T(`Input field "${e}": Arrays of ${o} are not supported`,this.position,this.getErrorContext(),`Use a single ${o} type instead: "${o}"`)}catch(o){throw o instanceof T?o:new T(`Input field "${e}": ${o instanceof Error?o.message:"Unknown error"}`,this.position,this.getErrorContext())}}this.skipWhitespace();let r=this.parseParsedString();return{name:e,desc:r?.trim(),type:n,isOptional:t}}parseOutputField(){this.skipWhitespace();let e=this.parseParsedIdentifier();this.currentFieldName=e,this.validateFieldName(e,"output");let t=!1,n=!1;for(;;){if(this.match("?")){t=!0;continue}if(this.match("!")){n=!0;continue}break}let r;if(this.skipWhitespace(),this.match(":"))if(this.skipWhitespace(),this.match("class")){let i=this.match("[]");this.skipWhitespace();let a=this.parseParsedString();if(!a)throw new T(`Output field "${e}": Missing class options after "class" type`,this.position,this.getErrorContext(),'Add class names in quotes. Example: class "positive, negative, neutral"');let l=a.split(/[,|]/).map(p=>p.trim()).filter(p=>p.length>0);if(l.length===0)throw new T(`Output field "${e}": Empty class list provided`,this.position,this.getErrorContext(),'Provide at least one class option. Example: "positive, negative"');r={name:"class",isArray:i,options:l}}else try{let i=this.parseTypeNotClass(),a=this.match("[]");if(r={name:i,isArray:a},i==="image"&&a)throw new T(`Output field "${e}": Arrays of images are not supported`,this.position,this.getErrorContext(),'Use a single image type instead: "image"');if(i==="audio"&&a)throw new T(`Output field "${e}": Arrays of audio are not supported`,this.position,this.getErrorContext(),'Use a single audio type instead: "audio"');if(i==="image")throw new T(`Output field "${e}": Image type is not supported in output fields`,this.position,this.getErrorContext(),"Image types can only be used in input fields");if(i==="audio")throw new T(`Output field "${e}": Audio type is not supported in output fields`,this.position,this.getErrorContext(),"Audio types can only be used in input fields")}catch(i){throw i instanceof T?i:new T(`Output field "${e}": ${i instanceof Error?i.message:"Unknown error"}`,this.position,this.getErrorContext())}this.skipWhitespace();let o=this.parseParsedString();return{name:e,desc:o?.trim(),type:r,isOptional:t,isInternal:n}}validateFieldName(e,t){if(M.signatureStrict&&["text","object","image","string","number","boolean","json","array","datetime","date","time","type","class","input","output","data","value","result","response","request","item","element"].includes(e.toLowerCase())){let i=t==="input"?["userInput","questionText","documentContent","messageText"]:["responseText","analysisResult","categoryType","summaryText"];throw new T(`Field name "${e}" is too generic`,this.position,this.getErrorContext(),`Use a more descriptive name. Examples: ${i.join(", ")}`)}let n=/^[a-z][a-zA-Z0-9]*$/,r=/^[a-z]+(_[a-z0-9]+)*$/;if(!n.test(e)&&!r.test(e))throw new T(`Invalid field name "${e}"`,this.position,this.getErrorContext(),'Field names must be in camelCase (e.g., "userInput") or snake_case (e.g., "user_input")');if(e.length<2)throw new T(`Field name "${e}" is too short`,this.position,this.getErrorContext(),"Field names must be at least 2 characters long");if(e.length>50)throw new T(`Field name "${e}" is too long (${e.length} characters)`,this.position,this.getErrorContext(),"Field names should be 50 characters or less")}parseTypeNotClass(){let e=["string","number","boolean","json","image","audio","file","url","datetime","date","code"],t=e.find(n=>this.match(n));if(!t){let n=this.input.slice(this.position).match(/^\w+/)?.[0]||"",r=this.suggestType(n),o=`Invalid type "${n||"empty"}"`,i=r?`. Did you mean "${r}"?`:"",a=`${o}${i}`;throw new T(a,this.position,this.getErrorContext(),`Expected one of: ${e.join(", ")}`)}return t}suggestType(e){return{str:"string",text:"string",int:"number",integer:"number",float:"number",double:"number",bool:"boolean",object:"json",dict:"json",timestamp:"datetime",time:"datetime",img:"image",picture:"image",sound:"audio",voice:"audio",classification:"class",category:"class"}[e.toLowerCase()]||null}parseParsedIdentifier(){this.skipWhitespace();let e=/^[a-zA-Z_][a-zA-Z_0-9]*/.exec(this.input.slice(this.position));if(e)return this.position+=e[0].length,e[0];let t=/^\S+/.exec(this.input.slice(this.position)),n=t?t[0]:"";throw n===""?new T("Expected field name but found end of input",this.position,this.getErrorContext(),"Add a field name. Field names must start with a letter or underscore"):/^\d/.test(n)?new T(`Invalid field name "${n}" - cannot start with a number`,this.position,this.getErrorContext(),'Field names must start with a letter or underscore. Example: "userInput" or "_internal"'):new T(`Invalid field name "${n}"`,this.position,this.getErrorContext(),"Field names must start with a letter or underscore and contain only letters, numbers, or underscores")}parseParsedString(){let e=["'",'"'];for(let t of e)if(this.match(t)){let n="",r=!1,o=this.position-1;for(;this.position<this.input.length;){let a=this.input[this.position];if(this.position++,r)n+=a,r=!1;else if(a==="\\")r=!0;else{if(a===t)return n;n+=a}}let i=this.input.slice(o,Math.min(this.position,o+20));throw new T(`Unterminated string starting at position ${o}`,o,this.getErrorContext(),`Add closing ${t} to complete the string: ${i}${t}`)}}skipWhitespace(){let e=/^[\s\t\r\n]+/.exec(this.input.slice(this.position));e&&(this.position+=e[0].length)}match(e){let t;if(typeof e=="string"){if(this.input.startsWith(e,this.position))return this.position+=e.length,!0}else if(t=e.exec(this.input.slice(this.position)),t)return this.position+=t[0].length,!0;return!1}expectArrow(){if(!this.match("->")){let e=this.input.slice(this.position,this.position+10),t=e.includes(">")?'Use "->" (dash followed by greater-than)':e.includes("-")?'Add ">" after the dash':'Add "->" to separate input and output fields';throw new T(`Expected "->" but found "${e}..."`,this.position,this.getErrorContext(),t)}}};function Gs(s){return new Hr(s).parse()}var bt=class{inputFields=[];outputFields=[];desc;input(e,t,n=!1){let r={name:e,type:{name:t.type,isArray:t.isArray,options:t.options?[...t.options]:void 0},description:t.description,isOptional:t.isOptional,isInternal:t.isInternal};return n?this.inputFields.unshift(r):this.inputFields.push(r),this}output(e,t,n=!1){let r={name:e,type:{name:t.type,isArray:t.isArray,options:t.options?[...t.options]:void 0},description:t.description,isOptional:t.isOptional,isInternal:t.isInternal};return n?this.outputFields.unshift(r):this.outputFields.push(r),this}description(e){return this.desc=e,this}build(){let e={description:this.desc,inputs:this.inputFields,outputs:this.outputFields};return new P(e)}},L=class s{type;isArray;options;description;isOptional;isInternal;constructor(e){this.type=e.type,this.isArray=e.isArray,this.options=e.options,this.description=e.description,this.isOptional=e.isOptional,this.isInternal=e.isInternal}optional(){return new s({...this,isOptional:!0})}array(){return new s({...this,isArray:!0})}internal(){return new s({...this,isInternal:!0})}},Be=Object.assign(()=>new bt,{string:s=>new L({type:"string",isArray:!1,description:s}),number:s=>new L({type:"number",isArray:!1,description:s}),boolean:s=>new L({type:"boolean",isArray:!1,description:s}),json:s=>new L({type:"json",isArray:!1,description:s}),datetime:s=>new L({type:"datetime",isArray:!1,description:s}),date:s=>new L({type:"date",isArray:!1,description:s}),class:(s,e)=>new L({type:"class",isArray:!1,options:s,description:e}),image:s=>new L({type:"image",isArray:!1,description:s}),audio:s=>new L({type:"audio",isArray:!1,description:s}),file:s=>new L({type:"file",isArray:!1,description:s}),url:s=>new L({type:"url",isArray:!1,description:s}),code:(s,e)=>new L({type:"code",isArray:!1,description:e||s}),array:s=>({...s,isArray:!0}),optional:s=>({...s,isOptional:!0}),internal:s=>({...s,isInternal:!0}),legacyArray:s=>({...s,isArray:!0}),legacyOptional:s=>({...s,isOptional:!0}),legacyInternal:s=>({...s,isInternal:!0})});function Rn(s){return{type:{name:s.type,isArray:s.isArray,options:s.options?[...s.options]:void 0},description:s.description,isOptional:s.isOptional,isInternal:s.isInternal}}var b=class extends Error{constructor(t,n,r){super(t);this.fieldName=n;this.suggestion=r;this.name="AxSignatureValidationError"}},P=class s{description;inputFields;outputFields;sigHash;sigString;validatedAtHash;constructor(e){if(!e){this.inputFields=[],this.outputFields=[],this.sigHash="",this.sigString="";return}if(typeof e=="string"){let t;try{t=Gs(e)}catch(n){if(n instanceof Error){let r="suggestion"in n&&typeof n.suggestion=="string"?n.suggestion:'Please check the signature format. Example: "userInput:string -> responseText:string"';throw new b(`Invalid Signature: ${n.message}`,void 0,r)}throw new b(`Invalid Signature: ${e}`,void 0,'Please check the signature format. Example: "userInput:string -> responseText:string"')}this.description=t.desc,this.inputFields=t.inputs.map(n=>this.parseParsedField(n)),this.outputFields=t.outputs.map(n=>this.parseParsedField(n)),[this.sigHash,this.sigString]=this.updateHash()}else if(e instanceof s)this.description=e.getDescription(),this.inputFields=structuredClone(e.getInputFields()),this.outputFields=structuredClone(e.getOutputFields()),this.sigHash=e.hash(),this.sigString=e.toString(),e.validatedAtHash===this.sigHash&&(this.validatedAtHash=this.sigHash);else if(typeof e=="object"&&e!==null){if(!("inputs"in e)||!("outputs"in e))throw new b("Invalid signature object: missing inputs or outputs",void 0,'Signature object must have "inputs" and "outputs" arrays. Example: { inputs: [...], outputs: [...] }');if(!Array.isArray(e.inputs)||!Array.isArray(e.outputs))throw new b("Invalid signature object: inputs and outputs must be arrays",void 0,'Both "inputs" and "outputs" must be arrays of AxField objects');try{this.description=e.description,this.inputFields=e.inputs.map(t=>this.parseField(t)),this.outputFields=e.outputs.map(t=>this.parseField(t)),[this.sigHash,this.sigString]=this.updateHash()}catch(t){throw t instanceof b?t:new b(`Failed to create signature from object: ${t instanceof Error?t.message:"Unknown error"}`,void 0,"Check that all fields in inputs and outputs arrays are valid AxField objects")}}else throw new b("Invalid signature argument type",void 0,"Signature must be a string, another AxSignature instance, or an object with inputs and outputs arrays")}static create(e){return new s(e)}parseParsedField=e=>{if(!e.name||e.name.length===0)throw new b("Field name is required",e.name,'Every field must have a descriptive name. Example: "userInput", "responseText"');let t=this.toTitle(e.name);return{name:e.name,title:t,description:"desc"in e?e.desc:void 0,type:e.type??{name:"string",isArray:!1},..."isInternal"in e?{isInternal:e.isInternal}:{},..."isOptional"in e?{isOptional:e.isOptional}:{}}};parseField=e=>{let t=!e.title||e.title.length===0?this.toTitle(e.name):e.title;if(e.type&&(!e.type.name||e.type.name.length===0))throw new b("Field type name is required",e.name,"Specify a valid type. Available types: string, number, boolean, json, image, audio, file, url, date, datetime, class, code");return{...e,title:t}};setDescription=e=>{if(typeof e!="string")throw new b("Description must be a string",void 0,"Provide a string description for the signature");this.description=e,this.invalidateValidationCache(),this.updateHashLight()};addInputField=e=>{try{let t=this.parseField(e);ee(t,"input");for(let n of this.inputFields)if(n.name===t.name)throw new b(`Duplicate input field name: "${t.name}"`,t.name,"Each field name must be unique within the signature");for(let n of this.outputFields)if(n.name===t.name)throw new b(`Field name "${t.name}" appears in both inputs and outputs`,t.name,"Use different names for input and output fields to avoid confusion");this.inputFields.push(t),this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to add input field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};addOutputField=e=>{try{let t=this.parseField(e);ee(t,"output");for(let n of this.outputFields)if(n.name===t.name)throw new b(`Duplicate output field name: "${t.name}"`,t.name,"Each field name must be unique within the signature");for(let n of this.inputFields)if(n.name===t.name)throw new b(`Field name "${t.name}" appears in both inputs and outputs`,t.name,"Use different names for input and output fields to avoid confusion");this.outputFields.push(t),this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to add output field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};setInputFields=e=>{if(!Array.isArray(e))throw new b("Input fields must be an array",void 0,"Provide an array of field objects");try{let t=e.map(n=>{let r=this.parseField(n);return ee(r,"input"),r});this.inputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to set input fields: ${t instanceof Error?t.message:"Unknown error"}`)}};setOutputFields=e=>{if(!Array.isArray(e))throw new b("Output fields must be an array",void 0,"Provide an array of field objects");try{let t=e.map(n=>{let r=this.parseField(n);return ee(r,"output"),r});this.outputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof b?t:new b(`Failed to set output fields: ${t instanceof Error?t.message:"Unknown error"}`)}};getInputFields=()=>this.inputFields;getOutputFields=()=>this.outputFields;getDescription=()=>this.description;appendInputField=(e,t)=>{let n=new s(this);return n.addInputField({name:e,...Rn(t)}),n};prependInputField=(e,t)=>{let n=new s(this),r={name:e,...Rn(t)},o=n.parseField(r);ee(o,"input");for(let i of n.inputFields)if(i.name===o.name)throw new b(`Duplicate input field name: "${o.name}"`,o.name,"Each field name must be unique within the signature");for(let i of n.outputFields)if(i.name===o.name)throw new b(`Field name "${o.name}" appears in both inputs and outputs`,o.name,"Use different names for input and output fields to avoid confusion");return n.inputFields.unshift(o),n.invalidateValidationCache(),n.updateHashLight(),n};appendOutputField=(e,t)=>{let n=new s(this);return n.addOutputField({name:e,...Rn(t)}),n};prependOutputField=(e,t)=>{let n=new s(this),r={name:e,...Rn(t)},o=n.parseField(r);ee(o,"output");for(let i of n.outputFields)if(i.name===o.name)throw new b(`Duplicate output field name: "${o.name}"`,o.name,"Each field name must be unique within the signature");for(let i of n.inputFields)if(i.name===o.name)throw new b(`Field name "${o.name}" appears in both inputs and outputs`,o.name,"Use different names for input and output fields to avoid confusion");return n.outputFields.unshift(o),n.invalidateValidationCache(),n.updateHashLight(),n};invalidateValidationCache=()=>{this.validatedAtHash=void 0};toTitle=e=>{let t=e.replace(/_/g," ");return t=t.replace(/([A-Z]|[0-9]+)/g," $1").trim(),t.charAt(0).toUpperCase()+t.slice(1)};toJSONSchema=()=>{let e={},t=[];for(let r of this.inputFields){let o=r.type?r.type.name:"string";r.type?.isArray?e[r.name]={description:r.description,type:"array",items:{type:o,description:r.description}}:e[r.name]={description:r.description,type:o},r.isOptional||t.push(r.name)}return{type:"object",properties:e,required:t}};updateHashLight=()=>{try{return this.getInputFields().forEach(e=>{ee(e,"input")}),this.getOutputFields().forEach(e=>{ee(e,"output")}),this.sigHash=kt("sha256").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Ns(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof b?e:new b(`Signature validation failed: ${e instanceof Error?e.message:"Unknown error"}`)}};updateHash=()=>{try{return this.getInputFields().forEach(e=>{ee(e,"input")}),this.getOutputFields().forEach(e=>{ee(e,"output")}),this.validateSignatureConsistency(),this.sigHash=kt("sha256").update(this.description??"").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Ns(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof b?e:new b(`Signature validation failed: ${e instanceof Error?e.message:"Unknown error"}`)}};validateSignatureConsistency(){let e=new Set;for(let n of this.inputFields){if(e.has(n.name))throw new b(`Duplicate input field name: "${n.name}"`,n.name,"Each field name must be unique within the signature");e.add(n.name)}let t=new Set;for(let n of this.outputFields){if(t.has(n.name))throw new b(`Duplicate output field name: "${n.name}"`,n.name,"Each field name must be unique within the signature");t.add(n.name)}for(let n of this.outputFields)if(e.has(n.name))throw new b(`Field name "${n.name}" appears in both inputs and outputs`,n.name,"Use different names for input and output fields to avoid confusion");if(this.inputFields.length===0)throw new b("Signature must have at least one input field",void 0,'Add an input field. Example: "userInput:string -> ..."');if(this.outputFields.length===0)throw new b("Signature must have at least one output field",void 0,'Add an output field. Example: "... -> responseText:string"')}validate=()=>{if(this.validatedAtHash===this.sigHash)return!0;try{return this.updateHash(),this.validatedAtHash=this.sigHash,!0}catch(e){throw this.validatedAtHash=void 0,e}};hash=()=>this.sigHash;toString=()=>this.sigString;injectToolFields(e){let t=new s(this);for(let n of e)if(n.parameters?.properties&&Object.keys(n.parameters.properties).length>0){let r=this.generateToolParameterFields(n);for(let o of r)t.outputFields.some(a=>a.name===o.name)||t.addOutputField(o)}else{let r=this.sanitizeFieldName(n.name),o=this.inferToolFieldType(n.parameters);t.outputFields.some(a=>a.name===r)||t.addOutputField({name:r,title:this.formatTitle(n.name),type:o,description:n.description||`Parameters for ${n.name}`,isOptional:!0})}return t}generateToolParameterFields(e){let t=[];if(!e.parameters||!e.parameters.properties)return t;let n=e.parameters.properties,r=e.parameters.required||[],o=(i,a,l)=>{for(let[p,c]of Object.entries(i)){let u=a?`${a}.${p}`:p,d=`${e.name}.${u}`;if(c.type==="object"&&c.properties)o(c.properties,u,c.required||[]);else{let m=this.inferParameterType(c);t.push({name:this.sanitizeFieldName(d),title:this.formatParameterTitle(e.name,u),type:m,description:c.description||`${p} parameter for ${e.name}`,isOptional:!0})}}};return o(n,"",r),t}inferParameterType(e){switch(e.type){case"string":return{name:"string",isArray:!1};case"number":case"integer":return{name:"number",isArray:!1};case"boolean":return{name:"boolean",isArray:!1};case"array":{let t=e.items;if(t?.type)switch(t.type){case"string":return{name:"string",isArray:!0};case"number":case"integer":return{name:"number",isArray:!0};case"boolean":return{name:"boolean",isArray:!0};default:return{name:"json",isArray:!0}}return{name:"json",isArray:!0}}case"object":return{name:"json",isArray:!1};default:return{name:"string",isArray:!1}}}formatParameterTitle(e,t){return`${e} ${t.replace(/\./g," ")}`}sanitizeFieldName(e){return e.replace(/([A-Z])/g,"_$1").toLowerCase().replace(/^_|_$/g,"").replace(/[^a-z0-9_]/g,"_")}formatTitle(e){return e.replace(/([A-Z])/g," $1").replace(/^./,t=>t.toUpperCase()).trim()}inferToolFieldType(e){return!e||!e.properties||Object.keys(e.properties).length===0?{name:"string",isArray:!1}:{name:"json",isArray:!1}}toJSON=()=>({id:this.hash(),description:this.description,inputFields:this.inputFields,outputFields:this.outputFields})};function $s(s){let e=s.name;return s.isOptional&&(e+="?"),s.isInternal&&(e+="!"),s.type&&(e+=`:${s.type.name}`,s.type.isArray&&(e+="[]"),s.type.name==="class"&&s.type.options&&(e+=` "${s.type.options.join(" | ")}"`)),s.description&&s.type?.name!=="class"&&(e+=` "${s.description}"`),e}function Ns(s,e,t){let n=s?`"${s}" `:"",r=e.map($s).join(", "),o=t.map($s).join(", ");return`${n}${r} -> ${o}`}function Vi(s){let e=/^[a-z][a-zA-Z0-9]*$/,t=/^[a-z]+(_[a-z0-9]+)*$/;return e.test(s)||t.test(s)}function ee(s,e){if(!s.name||s.name.length===0)throw new b("Field name cannot be blank",s.name,"Every field must have a descriptive name");if(!Vi(s.name))throw new b(`Invalid field name '${s.name}' - must be camelCase or snake_case`,s.name,'Use camelCase (e.g., "userInput") or snake_case (e.g., "user_input")');if(M.signatureStrict&&["text","object","image","string","number","boolean","json","array","datetime","date","time","type","class","input","output","data","value","result","response","request","item","element"].includes(s.name.toLowerCase())){let n=e==="input"?["userInput","questionText","documentContent","messageText","queryString"]:["responseText","analysisResult","categoryType","summaryText","outputData"];throw new b(`Field name '${s.name}' is too generic`,s.name,`Use a more descriptive name. Examples for ${e} fields: ${n.join(", ")}`)}if(s.name.length<2)throw new b(`Field name '${s.name}' is too short`,s.name,"Field names must be at least 2 characters long");if(s.name.length>50)throw new b(`Field name '${s.name}' is too long (${s.name.length} characters)`,s.name,"Field names should be 50 characters or less");s.type&&Wi(s,e)}function Wi(s,e){if(!s.type)return;let{type:t}=s;if(t.name==="image"||t.name==="audio"||t.name==="file"||t.name==="url"){if(e==="output")throw new b(`${t.name} type is not supported in output fields`,s.name,`${t.name} types can only be used in input fields`);if(t.isArray&&(t.name==="image"||t.name==="audio"))throw new b(`Arrays of ${t.name} are not supported`,s.name,`Use a single ${t.name} type instead`)}if(t.name==="class"){if(e==="input")throw new b("Class type is not supported in input fields",s.name,'Class types are only allowed on output fields. Use "string" type for input classifications');if(!t.options||t.options.length===0)throw new b("Class type requires options",s.name,'Provide class options. Example: class "positive, negative, neutral"');for(let r of t.options){if(!r||r.trim().length===0)throw new b("Empty class option found",s.name,"All class options must be non-empty strings");let o=r.trim();if(o.includes(",")||o.includes("|"))throw new b(`Invalid class option "${o}"`,s.name,"Class options cannot contain commas (,) or pipes (|) as they are used to separate options")}if(new Set(t.options.map(r=>r.trim().toLowerCase())).size!==t.options.length)throw new b("Duplicate class options found",s.name,"Each class option must be unique (case-insensitive)")}if(t.name==="code"&&t.isArray)throw new b("Arrays of code are not commonly supported",s.name,"Consider using a single code field or an array of strings instead");if(s.isInternal&&e==="input")throw new b("Internal marker (!) is not allowed on input fields",s.name,"Internal markers are only allowed on output fields")}var ie=class{signature;sigHash;examples;examplesOptions;demos;trace;usage=[];traceLabel;key;children;constructor(e,t){this.signature=new P(e),t?.description&&this.signature.setDescription(t.description),t?.traceLabel&&(this.traceLabel=t.traceLabel),e&&this.signature.validate(),this.sigHash=this.signature?.hash(),this.children=new Ue,this.key={id:this.signature.hash()}}getSignature(){return new P(this.signature)}setSignature(e){this.signature=new P(e),e&&this.signature.validate(),this.updateSignatureHash()}setDescription(e){this.signature.setDescription(e),this.updateSignatureHash()}updateSignatureHash(){this.sigHash=this.signature.hash(),this.key={id:this.signature.hash()}}register(e){this.key&&e.setParentId(this.key.id),this.children.register(e)}setId(e){this.key={id:e,custom:!0};for(let t of Array.from(this.children))t?.setParentId(e)}setParentId(e){this.key.custom||(this.key.id=[e,this.key.id].join("/"))}setExamples(e,t){if(this._setExamples(e,t),"programId"in e)for(let n of Array.from(this.children))n?.setExamples(e,t)}_setExamples(e,t){let n=[];if("programId"in e&&e.programId===this.key.id&&(n=e.traces),Array.isArray(e)&&(n=e),n){this.examplesOptions=t;let r=this.signature,o=[...r.getInputFields(),...r.getOutputFields()];this.examples=n.map(i=>{let a={};for(let l of o){let p=i[l.name];p!==void 0&&(bn(l,p),a[l.name]=p)}return a})}}getTraces(){let e=[];this.trace&&e.push({trace:this.trace,programId:this.key.id});for(let t of Array.from(this.children)){let n=t?.getTraces();e=[...e,...n??[]]}return e}getUsage(){let e=[...this.usage??[]];for(let t of Array.from(this.children)){let n=t?.getUsage();e=[...e,...n??[]]}return yt(e)}resetUsage(){this.usage=[];for(let e of Array.from(this.children))e?.resetUsage()}setDemos(e){let t=Array.from(this.children).length>0,n=e.some(r=>r.programId===this.key.id);if(t&&!n)throw new Error(`Program with id '${this.key.id}' has children but no matching programId found in demos`);this.demos=e.filter(r=>r.programId===this.key.id).map(r=>r.traces).flat();for(let r of Array.from(this.children))r?.setDemos(e)}};var Ji=`
158
158
  ## Function Call Instructions
159
159
  - Complete the task, using the functions defined earlier in this prompt.
160
160
  - Output fields should only be generated after all functions have been called.
161
- - Use the function results to generate the output fields.`,Wi=`
161
+ - Use the function results to generate the output fields.`,Qi=`
162
162
  ## Strict Output Formatting Rules
163
163
  - Output must strictly follow the defined plain-text \`field name: value\` field format.
164
164
  - Output field, values must strictly adhere to the specified output field formatting rules.
165
165
  - No formatting rules should override these **Strict Output Formatting Rules**
166
166
  - Do not add any text before or after the output fields, just the field name and value.
167
- - Do not use code blocks.`,de=class{sig;fieldTemplates;task;thoughtFieldName;functions;constructor(e,t,n){this.sig=e,this.fieldTemplates=n,this.thoughtFieldName=t?.thoughtFieldName??"thought",this.functions=t?.functions;let o=[],r=Ns(this.sig.getInputFields()),i=Ns(this.sig.getOutputFields());o.push(`You will be provided with the following fields: ${r}. Your task is to generate new fields: ${i}.`);let l=this.functions?.map(d=>"toFunction"in d?d.toFunction():d)?.flat()?.map(d=>`- \`${d.name}\`: ${Rn(d.description)}`).join(`
168
- `);l&&l.length>0&&o.push(`## Available Functions
169
- ${l}`);let p=Ji(this.sig.getInputFields());o.push(`## Input Fields
170
- ${p}`);let c=Qi(this.sig.getOutputFields());o.push(`## Output Fields
171
- ${c}`),l&&l.length>0&&o.push(Vi.trim()),o.push(Wi.trim());let u=this.sig.getDescription();if(u){let d=Rn(u);o.push(d)}this.task={type:"text",text:o.join(`
167
+ - Do not use code blocks.`,de=class{sig;fieldTemplates;task;thoughtFieldName;functions;constructor(e,t,n){this.sig=e,this.fieldTemplates=n,this.thoughtFieldName=t?.thoughtFieldName??"thought",this.functions=t?.functions;let r=[],o=Us(this.sig.getInputFields()),i=Us(this.sig.getOutputFields());r.push(`You will be provided with the following fields: ${o}. Your task is to generate new fields: ${i}.`);let l=this.functions?.map(d=>"toFunction"in d?d.toFunction():d)?.flat()?.map(d=>`- \`${d.name}\`: ${Tn(d.description)}`).join(`
168
+ `);l&&l.length>0&&r.push(`## Available Functions
169
+ ${l}`);let p=Yi(this.sig.getInputFields());r.push(`## Input Fields
170
+ ${p}`);let c=Xi(this.sig.getOutputFields());r.push(`## Output Fields
171
+ ${c}`),l&&l.length>0&&r.push(Ji.trim()),r.push(Qi.trim());let u=this.sig.getDescription();if(u){let d=Tn(u);r.push(d)}this.task={type:"text",text:r.join(`
172
172
 
173
- `)}}renderSingleValueUserContent=(e,t,n,o)=>{let r=this.renderInputFields(e),a=(o?r:[...t,...n,...r]).filter(l=>l!==void 0);return a.every(l=>l.type==="text")?a.map(l=>l.text).join(`
174
- `):a.reduce($s(`
175
- `),[])};render=(e,{examples:t,demos:n})=>{let o=t?[{type:"text",text:`
173
+ `)}}renderSingleValueUserContent=(e,t,n,r)=>{let o=this.renderInputFields(e),a=(r?o:[...t,...n,...o]).filter(l=>l!==void 0);return a.every(l=>l.type==="text")?a.map(l=>l.text).join(`
174
+ `):a.reduce(Bs(`
175
+ `),[])};render=(e,{examples:t,demos:n})=>{let r=t?[{type:"text",text:`
176
176
 
177
177
  ## Examples
178
- `},...this.renderExamples(t)]:[],r=n?this.renderDemos(n):[],i=o.every(d=>d.type==="text"),a=r.every(d=>d.type==="text"),l=i&&a,p=this.task.text;if(l){let d=[{type:"text",text:p},...o,...r];d.reduce($s(""),[]),d?.[0]&&(p=d[0].text)}let c={role:"system",content:p};if(Array.isArray(e)){let d=[],m=e,h=!0;for(let g of m){let f;if(h?(f=this.renderSingleValueUserContent(g.values,o,r,l),h=!1):f=this.renderSingleValueUserContent(g.values,[],[],!1),g.role==="user"){d.push({role:"user",content:f});continue}if(g.role!=="assistant")throw new Error("Invalid message role");if(typeof f!="string")throw new Error("Assistant message cannot contain non-text content like images, files,etc");d.push({role:"assistant",content:f})}return[c,...d]}let u=this.renderSingleValueUserContent(e,o,r,l);return[c,{role:"user",content:u}]};renderExtraFields=e=>{let t=[];if(!e||e.length===0)return t;let n=e.reduce((r,i)=>{let a=i.title;return r[a]||(r[a]=[]),r[a].push(i),r},{});return Object.entries(n).map(([r,i])=>{if(i.length===1){let a=i[0];return{title:r,name:a.name,description:a.description}}if(i.length>1){let a=i.map(l=>`- ${l.description}`).join(`
179
- `);return{title:r,name:i[0].name,description:a}}}).filter(Boolean).forEach(r=>{let i=this.fieldTemplates?.[r.name]??this.defaultRenderInField;t.push(...i(r,r.description))}),t};renderExamples=e=>{let t=[],n={isExample:!0};for(let[o,r]of e.entries()){let i=this.sig.getInputFields().map(p=>this.renderInField(p,r,{...n,isInputField:!0})).filter(p=>p!==void 0).flat(),a=this.sig.getOutputFields().map(p=>this.renderInField(p,r,{...n,isInputField:!1})).filter(p=>p!==void 0).flat(),l=[...i,...a];o>0&&l.length>0&&l[0]?.type==="text"&&t.push({type:"text",text:`---
178
+ `},...this.renderExamples(t)]:[],o=n?this.renderDemos(n):[],i=r.every(d=>d.type==="text"),a=o.every(d=>d.type==="text"),l=i&&a,p=this.task.text;if(l){let d=[{type:"text",text:p},...r,...o];d.reduce(Bs(""),[]),d?.[0]&&(p=d[0].text)}let c={role:"system",content:p};if(Array.isArray(e)){let d=[],m=e,h=!0;for(let g of m){let f;if(h?(f=this.renderSingleValueUserContent(g.values,r,o,l),h=!1):f=this.renderSingleValueUserContent(g.values,[],[],!1),g.role==="user"){d.push({role:"user",content:f});continue}if(g.role!=="assistant")throw new Error("Invalid message role");if(typeof f!="string")throw new Error("Assistant message cannot contain non-text content like images, files,etc");d.push({role:"assistant",content:f})}return[c,...d]}let u=this.renderSingleValueUserContent(e,r,o,l);return[c,{role:"user",content:u}]};renderExtraFields=e=>{let t=[];if(!e||e.length===0)return t;let n=e.reduce((o,i)=>{let a=i.title;return o[a]||(o[a]=[]),o[a].push(i),o},{});return Object.entries(n).map(([o,i])=>{if(i.length===1){let a=i[0];return{title:o,name:a.name,description:a.description}}if(i.length>1){let a=i.map(l=>`- ${l.description}`).join(`
179
+ `);return{title:o,name:i[0].name,description:a}}}).filter(Boolean).forEach(o=>{let i=this.fieldTemplates?.[o.name]??this.defaultRenderInField;t.push(...i(o,o.description))}),t};renderExamples=e=>{let t=[],n={isExample:!0};for(let[r,o]of e.entries()){let i=this.sig.getInputFields().map(p=>this.renderInField(p,o,{...n,isInputField:!0})).filter(p=>p!==void 0).flat(),a=this.sig.getOutputFields().map(p=>this.renderInField(p,o,{...n,isInputField:!1})).filter(p=>p!==void 0).flat(),l=[...i,...a];r>0&&l.length>0&&l[0]?.type==="text"&&t.push({type:"text",text:`---
180
180
 
181
181
  `}),l.forEach(p=>{"text"in p&&(p.text=`${p.text}
182
- `),t.push(p)})}return t};renderDemos=e=>{let t=[],n=this.sig.getInputFields(),o=this.sig.getOutputFields(),r={isExample:!0};for(let i of e){let a=n.map(c=>this.renderInField(c,i,{...r,isInputField:!0})).filter(c=>c!==void 0).flat(),l=o.map(c=>this.renderInField(c,i,{...r,isInputField:!1})).filter(c=>c!==void 0).flat();[...a,...l].slice(0,-1).forEach(c=>{"text"in c&&(c.text=`${c.text}
182
+ `),t.push(p)})}return t};renderDemos=e=>{let t=[],n=this.sig.getInputFields(),r=this.sig.getOutputFields(),o={isExample:!0};for(let i of e){let a=n.map(c=>this.renderInField(c,i,{...o,isInputField:!0})).filter(c=>c!==void 0).flat(),l=r.map(c=>this.renderInField(c,i,{...o,isInputField:!1})).filter(c=>c!==void 0).flat();[...a,...l].slice(0,-1).forEach(c=>{"text"in c&&(c.text=`${c.text}
183
183
  `),t.push(c)})}return t};renderInputFields=e=>{let t=this.sig.getInputFields().map(n=>this.renderInField(n,e,void 0)).filter(n=>n!==void 0).flat();return t.filter(n=>n.type==="text").forEach(n=>{n.text=`${n.text}
184
- `}),t};renderInField=(e,t,n)=>{let o=t[e.name];if(Xi(e,o,n))return;e.type&&In(e,o);let r=Yi(e,o);return(this.fieldTemplates?.[e.name]??this.defaultRenderInField)(e,r)};defaultRenderInField=(e,t)=>{if(e.type?.name==="image"){let o=i=>{if(!i)throw new Error("Image field value is required.");if(typeof i!="object")throw new Error("Image field value must be an object.");if(!("mimeType"in i))throw new Error("Image field must have mimeType");if(!("data"in i))throw new Error("Image field must have data");return i},r=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("Image field value must be an array.");r=r.concat(t.map(i=>{let a=o(i);return{type:"image",mimeType:a.mimeType,image:a.data}}))}else{let i=o(t);r.push({type:"image",mimeType:i.mimeType,image:i.data})}return r}if(e.type?.name==="audio"){let o=i=>{if(!i)throw new Error("Audio field value is required.");if(typeof i!="object")throw new Error("Audio field value must be an object.");if(!("data"in i))throw new Error("Audio field must have data");return i},r=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("Audio field value must be an array.");r=r.concat(t.map(i=>{let a=o(i);return{type:"audio",format:a.format??"wav",data:a.data}}))}else{let i=o(t);r.push({type:"audio",format:i.format??"wav",data:i.data})}return r}if(e.type?.name==="file"){let o=i=>{if(!i)throw new Error("File field value is required.");if(typeof i!="object")throw new Error("File field value must be an object.");if(!("mimeType"in i))throw new Error("File field must have mimeType");let a="data"in i,l="fileUri"in i;if(!a&&!l)throw new Error("File field must have either data or fileUri");if(a&&l)throw new Error("File field cannot have both data and fileUri");return i},r=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("File field value must be an array.");r=r.concat(t.map(i=>{let a=o(i);return"fileUri"in a?{type:"file",mimeType:a.mimeType,fileUri:a.fileUri}:{type:"file",mimeType:a.mimeType,data:a.data}}))}else{let i=o(t);r.push("fileUri"in i?{type:"file",mimeType:i.mimeType,fileUri:i.fileUri}:{type:"file",mimeType:i.mimeType,data:i.data})}return r}if(e.type?.name==="url"){let o=i=>{if(!i)throw new Error("URL field value is required.");if(typeof i=="string")return{url:i};if(typeof i!="object")throw new Error("URL field value must be a string or object.");if(!("url"in i))throw new Error("URL field must have url property");return i},r=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("URL field value must be an array.");r=r.concat(t.map(i=>{let a=o(i);return{type:"url",url:a.url,...a.title?{title:a.title}:{},...a.description?{description:a.description}:{}}}))}else{let i=o(t);r.push({type:"url",url:i.url,...i.title?{title:i.title}:{},...i.description?{description:i.description}:{}})}return r}let n=[e.title,": "];return Array.isArray(t)?(n.push(`
185
- `),n.push(t.map(o=>`- ${o}`).join(`
186
- `))):n.push(t),[{type:"text",text:n.join("")}]}},Ns=s=>s.map(e=>`\`${e.title}\``).join(", "),Ji=s=>s.map(t=>{let n=t.title,o=t.type?.name?Us(t.type):"string",r=t.isOptional?`This optional ${o} field may be omitted`:`A ${o} field`,i=t.description?` ${Rn(t.description)}`:"";return`${n}: (${r})${i}`.trim()}).join(`
187
- `),Qi=s=>s.map(t=>{let n=t.title,o=t.type?.name?Us(t.type):"string",r=t.isOptional?`Only include this ${o} field if its value is available`:`This ${o} field must be included`,i="";return t.description&&t.description.length>0&&(i=` ${t.type?.name==="class"?t.description:Rn(t.description)}`),t.type?.options&&t.type.options.length>0&&(i.length>0&&(i+=". "),i+=`Allowed values: ${t.type.options.join(", ")}`),`${n}: (${r})${i}`.trim()}).join(`
188
- `),Yi=(s,e)=>{if(s.type?.name==="date"&&e instanceof Date){let t=e.toISOString();return t.slice(0,t.indexOf("T"))}return s.type?.name==="datetime"&&e instanceof Date?ws(e):s.type?.name==="image"&&typeof e=="object"||s.type?.name==="audio"&&typeof e=="object"||s.type?.name==="file"&&typeof e=="object"||s.type?.name==="url"&&(typeof e=="string"||typeof e=="object")||typeof e=="string"?e:JSON.stringify(e,null,2)},Us=s=>{let e=(()=>{switch(s?.name){case"string":return"string";case"number":return"number";case"boolean":return"boolean (true or false)";case"date":return'date ("YYYY-MM-DD" format)';case"datetime":return'date time ("YYYY-MM-DD HH:mm Timezone" format)';case"json":return"JSON object";case"class":return"classification class";case"code":return"code";case"file":return"file (with filename, mimeType, and data)";case"url":return"URL (string or object with url, title, description)";default:return"string"}})();return s?.isArray?`json array of ${e} items`:e};function $s(s){return(e,t)=>{if(t.type==="text"){let n=e.length>0?e[e.length-1]:null;n&&n.type==="text"?n.text+=s+t.text:e.push(t)}else e.push(t);return e}}var Xi=(s,e,t)=>{if(typeof e=="boolean")return!1;if(!e||(Array.isArray(e)||typeof e=="string")&&e.length===0){if(t?.isExample||s.isOptional||s.isInternal)return!0;let n=t?.isInputField!==!1?"input":"output";throw new Error(`Value for ${n} field '${s.name}' is required.`)}return!1};function Rn(s){let e=s.trim();return e.length>0?`${e.charAt(0).toUpperCase()}${e.slice(1)}${e.endsWith(".")?"":"."}`:""}function Zi(s,e){let t=s.history(0,e),n=t.some(r=>r.role==="function");return t.some(r=>r.role==="assistant"&&"functionCalls"in r&&Array.isArray(r.functionCalls)&&r.functionCalls.length>0)&&n}function ea(s,e){let t=s.history(0,e),n=[],o=t.filter(i=>i.role==="assistant"&&"functionCalls"in i&&Array.isArray(i.functionCalls)&&i.functionCalls.length>0),r=t.filter(i=>i.role==="function");for(let i of o)if("functionCalls"in i&&i.functionCalls)for(let a of i.functionCalls){let l=r.find(p=>"functionId"in p&&p.functionId===a.id);l&&"result"in l&&"functionId"in l&&n.push({index:n.length,functionName:a.function.name,functionId:a.id,args:a.function.params||"",result:String(l.result),isError:"isError"in l?!!l.isError:!1})}return n}async function Cn(s,e,t,n){if(!e?.resultPicker||s.length<=1)return 0;let o=e.resultPicker;if((t?Zi(t,n):!1)&&t){let l=ea(t,n),p=await o({type:"function",results:l});if(p<0||p>=l.length)throw new Error(`Result picker returned invalid index: ${p}. Must be between 0 and ${l.length-1}`);return p}let i=s.map((l,p)=>({index:p,sample:l.delta})),a=await o({type:"fields",results:i});if(a<0||a>=s.length)throw new Error(`Result picker returned invalid index: ${a}. Must be between 0 and ${s.length-1}`);return a}async function Bs(s,e,t){let n=s?.getLast(e);if(!n||n.role!=="assistant"||n.chat.length<=1)return 0;let o=n.chat.map(i=>({version:0,index:i.index,delta:i.value}));return await Cn(o,t,s,e)}var N=class extends ie{promptTemplate;asserts;streamingAsserts;options;functions;fieldProcessors=[];streamingFieldProcessors=[];excludeContentFromTrace=!1;thoughtFieldName;constructor(e,t){super(e,{description:t?.description,traceLabel:t?.traceLabel}),this.options=t,this.thoughtFieldName=t?.thoughtFieldName??"thought";let n={functions:t?.functions,thoughtFieldName:this.thoughtFieldName};this.promptTemplate=new(t?.promptTemplate??de)(this.signature,n),this.asserts=this.options?.asserts??[],this.streamingAsserts=this.options?.streamingAsserts??[],this.excludeContentFromTrace=t?.excludeContentFromTrace??!1,this.usage=[],t?.functions&&(this.functions=Fo(t.functions))}getSignatureName(){return this.signature.getDescription()||"unknown_signature"}getMetricsInstruments(){return Mo()}updateMeter(e){Mo(e)}createStates(e){return Array.from({length:e},(t,n)=>({index:n,functionCalls:[],values:{},content:"",functionsExecuted:new Set,xstate:{extractedFields:[],streamedIndex:{},s:-1}}))}addAssert=(e,t)=>{this.asserts.push({fn:e,message:t})};addStreamingAssert=(e,t,n)=>{this.streamingAsserts.push({fieldName:e,fn:t,message:n})};addFieldProcessorInternal=(e,t,n=!1)=>{let o=this.signature.getOutputFields().find(r=>r.name===e);if(!o)throw new Error(`addFieldProcessor: field ${e} not found`);if(n){let r=o.type?.name;if(!(!r||r==="string"||r==="code"))throw new Error(`addFieldProcessor: field ${e} is must be a text field`);this.streamingFieldProcessors.push({field:o,process:t})}else this.fieldProcessors.push({field:o,process:t})};addStreamingFieldProcessor=(e,t)=>{this.addFieldProcessorInternal(e,t,!0)};addFieldProcessor=(e,t)=>{this.addFieldProcessorInternal(e,t,!1)};async forwardSendRequest({ai:e,mem:t,options:n,traceContext:o,functions:r,functionCall:i,stepIndex:a}){let{sessionId:l,model:p,rateLimiter:c,stream:u,thinkingTokenBudget:d,showThoughts:m}=n??{},h=await Bs(t,l,{resultPicker:n?.resultPicker}),g=t?.history(h,l)??[];if(g.length===0)throw new Error("No chat prompt found");let f={...n?.modelConfig,...n?.sampleCount?{n:n.sampleCount}:{},...n?.sampleCount&&n?.modelConfig?.temperature===1?{temperature:.8}:{}},A=this.isDebug(e,n),x=a===0,y=this.getLogger(e,n);return await e.chat({chatPrompt:g,functions:r,functionCall:i,modelConfig:f,model:p},{sessionId:l,rateLimiter:c,stream:u,debug:A,debugHideSystemPrompt:!x,thinkingTokenBudget:d,showThoughts:m,traceContext:o,abortSignal:n?.abortSignal,stepIndex:a,logger:y})}async*forwardCore({ai:e,mem:t,options:n,stepIndex:o,span:r,traceContext:i}){let{sessionId:a,functions:l}=n??{},p=n?.functionCall??this.options?.functionCall,c=n?.strictMode??!1,u=n.model,d=this.createStates(n.sampleCount??1),m=this.usage,h=o===0,g=this.getLogger(e,n),{functions:f,functionCall:A}=ys(l,p,h),x=await this.forwardSendRequest({ai:e,mem:t,options:n,traceContext:i,functions:f,functionCall:A,stepIndex:o});x instanceof ReadableStream?yield*Ps({ai:e,model:u,res:x,mem:t,sessionId:a,functions:f,strictMode:c,span:r,states:d,usage:m,asserts:this.asserts,streamingAsserts:this.streamingAsserts,fieldProcessors:this.fieldProcessors,streamingFieldProcessors:this.streamingFieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,logger:g,functionResultFormatter:n?.functionResultFormatter??this.options?.functionResultFormatter}):yield*Fs({ai:e,model:u,res:x,mem:t,sessionId:a,functions:f,span:r,strictMode:c,states:d,usage:m,asserts:this.asserts,fieldProcessors:this.fieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,logger:g,functionResultFormatter:n?.functionResultFormatter??this.options?.functionResultFormatter})}async*_forward2(e,t,n,o,r,i){let a=(o?.stopFunction??this.options?.stopFunction)?.toLowerCase(),l=o.maxRetries??this.options?.maxRetries??10,p=o.maxSteps??this.options?.maxSteps??10,c=o.mem??this.options?.mem??new $e,u;if(o?.functions&&o.functions.length>0){let x=this.options?.promptTemplate??de,y={functions:o.functions,thoughtFieldName:this.thoughtFieldName};this.promptTemplate=new x(this.signature,y)}let d,m=performance.now();Array.isArray(t)?(eo(t),d=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos})):d=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos});let h=performance.now()-m,g=this.getMetricsInstruments();g&&fn(g,"prompt_render",h,this.getSignatureName());let f=performance.now();c.addRequest(d,o.sessionId);let A=performance.now()-f;g&&fn(g,"memory_update",A,this.getSignatureName());e:for(let x=0;x<p;x++){for(let S=0;S<l;S++)try{let I=this.forwardCore({options:o,ai:e,mem:c,stepIndex:x,span:r,traceContext:i});for await(let v of I)v!==void 0&&(yield{version:S,index:v.index,delta:v.delta});if(_s(c,a,n,o?.sessionId)){let v=this.getMetricsInstruments();v&&xn(v,x+1,p,this.getSignatureName());continue e}let k=this.getMetricsInstruments();if(k){xn(k,x+1,p,this.getSignatureName());let v=new Set;n.forEach(H=>{H.functionsExecuted.forEach(Y=>v.add(Y))}),v.size>0&&Po(k,!0,v.size,!0,!1,this.getSignatureName()),us(k,this.fieldProcessors.length,this.streamingFieldProcessors.length,this.getSignatureName())}return}catch(I){let R,k=this.isDebug(e,o),v=this.getLogger(e,o),H=this.getMetricsInstruments(),Y=this.getSignatureName(),Qe={error:I,errCount:S,logger:v,metricsInstruments:H,signatureName:Y,span:r,debug:k};if(r?.recordException(I),I instanceof D)R=hs(Qe),u=I;else if(I instanceof re)R=As(Qe),u=I;else if(I instanceof O)xs(Qe);else if(!(I instanceof K))throw Ho(I,e,this.signature);R&&(c.addRequest([{role:"user",content:this.promptTemplate.renderExtraFields(R)}],o.sessionId),c.addTag("error",o.sessionId))}let y=this.getMetricsInstruments();throw y&&Eo(y,l,!1,l,this.getSignatureName()),Ho(new Error(`Unable to fix validation error: ${u?.toString()}`),e,this.signature)}throw g&&xn(g,p,p,this.getSignatureName()),Ho(new Error(`Max steps reached: ${p}`),e,this.signature)}async*_forward1(e,t,n){let o=performance.now(),r=this.createStates(n.sampleCount??1),i=performance.now()-o,a=this.getMetricsInstruments();a&&fn(a,"state_creation",i,this.getSignatureName());let l=n?.tracer??this.options?.tracer??e.getOptions().tracer,p=this.functions;if(n?.functions&&(p=Fo(n.functions,this.functions)),!l){yield*this._forward2(e,t,r,{...n,functions:p});return}let c=p?.map(A=>A.name).join(","),u={signature:JSON.stringify(this.signature.toJSON(),null,2),...this.examples?{examples:JSON.stringify(this.examples,null,2)}:{},...c?{provided_functions:c}:{},...n?.model?{model:n.model}:{},...n?.thinkingTokenBudget?{thinking_token_budget:n.thinkingTokenBudget}:{},...n?.showThoughts?{show_thoughts:n.showThoughts}:{},...n?.maxSteps?{max_steps:n.maxSteps}:{},...n?.maxRetries?{max_retries:n.maxRetries}:{}},d=this.traceLabel&&n.traceLabel?`${this.traceLabel} > ${n.traceLabel}`:n.traceLabel??this.traceLabel,m=d?`AxGen > ${d}`:"AxGen",h=l.startSpan(m,{kind:qe.SpanKind.SERVER,attributes:u}),g=qe.context.active(),f=qe.trace.setSpan(g,h);try{if(this.excludeContentFromTrace||h.addEvent("input",{content:JSON.stringify(t,null,2)}),yield*this._forward2(e,t,r,{...n,functions:p},h,f),!this.excludeContentFromTrace){let A=r.map(y=>y.values),x=A.length===1?A[0]:A;h.addEvent("output",{content:JSON.stringify(x,null,2)})}}finally{h.end()}}async forward(e,t,n){let o=performance.now(),r=this.getSignatureName(),i=n?.stream??!1,a=!1,l=0,p=!1,c=0,u=!1;try{let d=this.getMetricsInstruments();d&&gs(d,this.signature.getInputFields().length,this.signature.getOutputFields().length,this.examples?.length??0,this.demos?.length??0,r),p=!!(n?.functions||this.functions);let m=this._forward1(e,t,n??{}),h=[],g=0,f=0;for await(let R of m)R.version!==g&&(h=[]),g=R.version,h=Go(h,R),f++;l=g;let A=performance.now();u=!!n?.resultPicker;let x=await Cn(h,{resultPicker:n?.resultPicker},n?.mem,n?.sessionId),y=performance.now()-A,I=h[x]?.delta??{};if(this.trace={...t,...I},u&&this.isDebug(e,n)){let R=this.getLogger(e,n);Ar(h.length,x,y,R)}return a=!0,d&&(ms(d,h.length,u,u?y:void 0,r),ds(d,i,f,void 0,r)),I}catch(d){throw a=!1,d}finally{let d=performance.now()-o,m=this.getMetricsInstruments();m&&(ps(m,d,a,r,e.getName(),n?.model?String(n.model):void 0),p&&Po(m,p,c,c>0,!1,r),l>0&&Eo(m,l,a,n?.maxRetries??10,r))}}async*streamingForward(e,t,n){if(!n?.resultPicker){yield*this._forward1(e,t,{...n,stream:!0});return}let o=this._forward1(e,t,{...n,stream:!0}),r=[],i=0;for await(let p of o)p.version!==i&&(r=[]),i=p.version,r=Go(r,p);let a=await Cn(r,{resultPicker:n?.resultPicker},n?.mem,n?.sessionId),l=r[a];l&&(yield{version:i,index:a,delta:l.delta})}setExamples(e,t){super.setExamples(e,t)}isDebug(e,t){return t?.debug??this.options?.debug??e.getOptions().debug??!1}getLogger(e,t){return t?.logger??this.options?.logger??M.logger??e.getLogger()}},Rt=class extends Error{details;constructor(e,t,n){super(e),this.name="AxGenerateError",this.details=t,n?.cause&&(this.cause=n.cause)}};function Ho(s,e,t){let n=s instanceof Error?s:new Error(String(s)),o=e.getLastUsedChatModel(),r=e.getLastUsedModelConfig(),i={model:o,maxTokens:r?.maxTokens,streaming:r?.stream??!1,signature:{input:t.getInputFields(),output:t.getOutputFields(),description:t.getDescription()}};return new Rt("Generate failed",i,{cause:n})}var ta=s=>s.replace(/^\W+|\W+$/g,""),na=(s,e)=>{let t=s.search(e);if(t===-1)return[s];let n=s.match(e);if(!n)throw new Error("Match failed unexpectedly.");let o=s.substring(0,t),r=s.substring(t+n[0].length);return[o,r]},oa=s=>{let e=new Set,t=[];for(let n of s)e.has(n)||(e.add(n),t.push(n));return t},ra=s=>{let e=s.match(/^(\d+)[.,\s]+(.*)$/);if(!e||e.length<3)throw new Error('line must start with a number, a dot and then text. e.g. "1. hello"');let t=Number.parseInt(e[1],10),n=e[2].trim();return{id:t,text:n}},sa=s=>{let e=s.match(/^(\d+)[.,\s]+(.*)$/);return e&&e[2]!==void 0?e[2].trim():s},ia=(s,e)=>{let t=[];for(let n=0;n<s.length;n+=e)t.push(s.slice(n,n+e));return t},Tn={trimNonAlphaNum:ta,splitIntoTwo:na,dedup:oa,extractIdAndText:ra,extractIndexPrefixedText:sa,batchArray:ia};var wn=class extends N{constructor(e){super(`"You are a re-ranker assistant tasked with evaluating a set of content items in relation to a specific question. Your role involves critically analyzing each content item to determine its relevance to the question and re-ranking them accordingly. This process includes assigning a relevance score from 0 to 10 to each content item based on how well it answers the question, its coverage of the topic, and the reliability of its information. This re-ranked list should start with the content item that is most relevant to the question and end with the least relevant. Output only the list."
189
- query: string, items: string[] -> rankedItems: string[] "list of id, 5-words Rationale, relevance score"`,e)}forward=async(e,t,n)=>{let{rankedItems:o}=await super.forward(e,t,n),r=o.map(a=>{let{id:l}=Tn.extractIdAndText(a);return l});return{rankedItems:t.items.map((a,l)=>{let p=r[l];return p!==void 0?t.items[p]:void 0}).filter(a=>a!==void 0)}}};var Sn=class{tikaUrl;fetch;constructor(e){let t=e??{url:"http://localhost:9998/"};this.tikaUrl=new URL("/tika",t.url),this.fetch=t.fetch}async _convert(e,t){if(!e)throw new Error("Failed to read file data");let n=t?.format==="html"?"text/html":"text/plain";try{let o={body:e,headers:{Accept:n},method:"PUT"};typeof window>"u"&&typeof process<"u"&&(o.duplex="half");let r=await(this.fetch??fetch)(this.tikaUrl,o);if(!r.ok)throw new Error(`Failed to upload file: ${r.statusText}`);return await r.text()}catch(o){throw new Error(`Error converting file: ${o}`)}}async convert(e,t){let n=[],o=t?.batchSize??10;for(let r=0;r<e.length;r+=o){let a=e.slice(r,r+o).map(p=>this._convert(p,{format:t?.format})),l=await Promise.all(a);n.push(...l)}return n}};var qs=new G,vn=class{name;context;constructor(e,t){this.name=e,this.context=t}getName(){return this.name}getContext(){return this.context}},On=class{ai;db;debug;constructor(e){this.db=new oe,this.ai=e}getState(){return this.db.getDB()}setState(e){this.db.setDB(e)}setClasses=async(e,t)=>{for(let n of e){let o=await this.ai.embed({texts:n.getContext()},{abortSignal:t?.abortSignal});await this.db.upsert({id:n.getName(),table:"classes",values:o.embeddings[0]})}};async forward(e,t){let{embeddings:n}=await this.ai.embed({texts:[e]},{abortSignal:t?.abortSignal}),r=(await this.db.query({table:"classes",values:n[0]})).matches;if(typeof t?.cutoff=="number"){let{cutoff:a}=t;r=r.filter(l=>l.score<=a)}this.debug&&console.log(`${qs.whiteBright(`query: ${e}`)}
190
- ${qs.greenBright(JSON.stringify(r.map(a=>`${a.id}, ${a.score}`)))}`);let i=r.at(0);return i?i.id:""}setOptions(e){typeof e.debug=="boolean"&&(this.debug=e.debug)}};var zs=new Set(["0o","0s","3a","3b","3d","6b","6o","a","a1","a2","a3","a4","ab","able","about","above","abst","ac","accordance","according","accordingly","across","act","actually","ad","added","adj","ae","af","affected","affecting","affects","after","afterwards","ag","again","against","ah","ain","ain't","aj","al","all","allow","allows","almost","alone","along","already","also","although","always","am","among","amongst","amoungst","amount","an","and","announce","another","any","anybody","anyhow","anymore","anyone","anything","anyway","anyways","anywhere","ao","ap","apart","apparently","appear","appreciate","appropriate","approximately","ar","are","aren","arent","aren't","arise","around","as","a's","aside","ask","asking","associated","at","au","auth","av","available","aw","away","awfully","ax","ay","az","b","b1","b2","b3","ba","back","bc","bd","be","became","because","become","becomes","becoming","been","before","beforehand","begin","beginning","beginnings","begins","behind","being","believe","below","beside","besides","best","better","between","beyond","bi","bill","biol","bj","bk","bl","bn","both","bottom","bp","br","brief","briefly","bs","bt","bu","but","bx","by","c","c1","c2","c3","ca","call","came","can","cannot","cant","can't","cause","causes","cc","cd","ce","certain","certainly","cf","cg","ch","changes","ci","cit","cj","cl","clearly","cm","c'mon","cn","co","com","come","comes","con","concerning","consequently","consider","considering","contain","containing","contains","corresponding","could","couldn","couldnt","couldn't","course","cp","cq","cr","cry","cs","c's","ct","cu","currently","cv","cx","cy","cz","d","d2","da","date","dc","dd","de","definitely","describe","described","despite","detail","df","di","did","didn","didn't","different","dj","dk","dl","do","does","doesn","doesn't","doing","don","done","don't","down","downwards","dp","dr","ds","dt","du","due","during","dx","dy","e","e2","e3","ea","each","ec","ed","edu","ee","ef","effect","eg","ei","eight","eighty","either","ej","el","eleven","else","elsewhere","em","empty","en","end","ending","enough","entirely","eo","ep","eq","er","es","especially","est","et","et-al","etc","eu","ev","even","ever","every","everybody","everyone","everything","everywhere","ex","exactly","example","except","ey","f","f2","fa","far","fc","few","ff","fi","fifteen","fifth","fify","fill","find","fire","first","five","fix","fj","fl","fn","fo","followed","following","follows","for","former","formerly","forth","forty","found","four","fr","from","front","ft","fu","full","further","furthermore","fy","g","ga","gave","ge","get","gets","getting","gi","give","given","gives","giving","gj","gl","go","goes","going","gone","got","gotten","gr","greetings","gs","gy","h","h2","h3","had","hadn","hadn't","happens","hardly","has","hasn","hasnt","hasn't","have","haven","haven't","having","he","hed","he'd","he'll","hello","help","hence","her","here","hereafter","hereby","herein","heres","here's","hereupon","hers","herself","hes","he's","hh","hi","hid","him","himself","his","hither","hj","ho","home","hopefully","how","howbeit","however","how's","hr","hs","http","hu","hundred","hy","i","i2","i3","i4","i6","i7","i8","ia","ib","ibid","ic","id","i'd","ie","if","ig","ignored","ih","ii","ij","il","i'll","im","i'm","immediate","immediately","importance","important","in","inasmuch","inc","indeed","index","indicate","indicated","indicates","information","inner","insofar","instead","interest","into","invention","inward","io","ip","iq","ir","is","isn","isn't","it","itd","it'd","it'll","its","it's","itself","iv","i've","ix","iy","iz","j","jj","jr","js","jt","ju","just","k","ke","keep","keeps","kept","kg","kj","km","know","known","knows","ko","l","l2","la","largely","last","lately","later","latter","latterly","lb","lc","le","least","les","less","lest","let","lets","let's","lf","like","liked","likely","line","little","lj","ll","ll","ln","lo","look","looking","looks","los","lr","ls","lt","ltd","m","m2","ma","made","mainly","make","makes","many","may","maybe","me","mean","means","meantime","meanwhile","merely","mg","might","mightn","mightn't","mill","million","mine","miss","ml","mn","mo","more","moreover","most","mostly","move","mr","mrs","ms","mt","mu","much","mug","must","mustn","mustn't","my","myself","model","n","n2","na","name","namely","nay","nc","nd","ne","near","nearly","necessarily","necessary","need","needn","needn't","needs","neither","never","nevertheless","new","next","ng","ni","nine","ninety","nj","nl","nn","no","nobody","non","none","nonetheless","noone","nor","normally","nos","not","noted","nothing","novel","now","nowhere","nr","ns","nt","ny","o","oa","ob","obtain","obtained","obviously","oc","od","of","off","often","og","oh","oi","oj","ok","okay","ol","old","om","omitted","on","once","one","ones","only","onto","oo","op","oq","or","ord","os","ot","other","others","otherwise","ou","ought","our","ours","ourselves","out","outside","over","overall","ow","owing","own","ox","oz","p","p1","p2","p3","page","pagecount","pages","par","part","particular","particularly","pas","past","pc","pd","pe","per","perhaps","pf","ph","pi","pj","pk","pl","placed","please","plus","pm","pn","po","poorly","possible","possibly","potentially","pp","pq","pr","predominantly","present","presumably","previously","primarily","probably","promptly","proud","provides","ps","pt","pu","put","py","q","qj","qu","que","quickly","quite","qv","r","r2","ra","ran","rather","rc","rd","re","readily","really","reasonably","recent","recently","ref","refs","regarding","regardless","regards","related","relatively","research","research-articl","respectively","resulted","resulting","results","rf","rh","ri","right","rj","rl","rm","rn","ro","rq","rr","rs","rt","ru","run","rv","ry","s","s2","sa","said","same","saw","say","saying","says","sc","sd","se","sec","second","secondly","section","see","seeing","seem","seemed","seeming","seems","seen","self","selves","sensible","sent","serious","seriously","seven","several","sf","shall","shan","shan't","she","shed","she'd","she'll","shes","she's","should","shouldn","shouldn't","should've","show","showed","shown","showns","shows","si","side","significant","significantly","similar","similarly","since","sincere","six","sixty","sj","sl","slightly","sm","sn","so","some","somebody","somehow","someone","somethan","something","sometime","sometimes","somewhat","somewhere","soon","sorry","sp","specifically","specified","specify","specifying","sq","sr","ss","st","still","stop","strongly","sub","substantially","successfully","such","sufficiently","suggest","sup","sure","sy","system","sz","t","t1","t2","t3","take","taken","taking","tb","tc","td","te","tell","ten","tends","tf","th","than","thank","thanks","thanx","that","that'll","thats","that's","that've","the","their","theirs","them","themselves","then","thence","there","thereafter","thereby","thered","therefore","therein","there'll","thereof","therere","theres","there's","thereto","thereupon","there've","these","they","theyd","they'd","they'll","theyre","they're","they've","thickv","thin","think","third","this","thorough","thoroughly","those","thou","though","thoughh","thousand","three","throug","through","throughout","thru","thus","ti","til","tip","tj","tl","tm","tn","to","together","too","took","top","toward","towards","tp","tq","tr","tried","tries","truly","try","trying","ts","t's","tt","tv","twelve","twenty","twice","two","tx","u","u201d","ue","ui","uj","uk","um","un","under","unfortunately","unless","unlike","unlikely","until","unto","uo","up","upon","ups","ur","us","use","used","useful","usefully","usefulness","uses","using","usually","ut","v","va","value","various","vd","ve","ve","very","via","viz","vj","vo","vol","vols","volumtype","vq","vs","vt","vu","w","wa","want","wants","was","wasn","wasnt","wasn't","way","we","wed","we'd","welcome","well","we'll","well-b","went","were","we're","weren","werent","weren't","we've","what","whatever","what'll","whats","what's","when","whence","whenever","when's","where","whereafter","whereas","whereby","wherein","wheres","where's","whereupon","wherever","whether","which","while","whim","whither","who","whod","whoever","whole","who'll","whom","whomever","whos","who's","whose","why","why's","wi","widely","will","willing","wish","with","within","without","wo","won","wonder","wont","won't","words","world","would","wouldn","wouldnt","wouldn't","www","x","x1","x2","x3","xf","xi","xj","xk","xl","xn","xo","xs","xt","xv","xx","y","y2","yes","yet","yj","yl","you","youd","you'd","you'll","your","youre","you're","yours","yourself","yourselves","you've","yr","ys","yt","z","zero","zi","zz","task"]);function js(s,e){return s.filter(t=>!e.has(t))}function Hs(s){let e={};for(let t of s)e[t]=(e[t]||0)+1;return e}function me(s){let e=s.normalize("NFD");return e=e.replace(/\b(a|an|the)\b/g," "),e=e.split(/\s+/).join(" "),e=e.replace(/[!"#$%&'()*+,-./:;<=>?@[\]^_`{|}~]/g,""),e.toLowerCase()}function aa(s,e){return me(s)===me(e)?1:0}function la(s,e){let t=me(s).split(" "),n=me(e).split(" "),o=Hs(t),r=Hs(n),i=0;for(let p in o){let c=o[p]??0,u=r[p]??0;i+=Math.min(c,u)}if(i===0)return 0;let a=i/t.length,l=i/n.length;return 2*a*l/(a+l)}function pa(s,e,t,n=!1){let o=me(s).split(" "),r=me(e).split(" "),i=me(t).split(" "),a=new Set([...zs,...o]);r=js(r,a),i=js(i,a);let l=0,p=l/r.length,c=l/i.length,u=2*p*c/(p+c);return n?c:u}var Ks={emScore:aa,f1Score:la,novelF1ScoreOptimized:pa};var Mn=class{ai;program;examples;constructor({ai:e,program:t,examples:n=[]}){if(n.length===0)throw new Error("No examples found");this.ai=e,this.program=t,this.examples=n}async run(e){let t=Date.now(),n=this.examples.length,o=0;for(let i=0;i<n;i++){let a=this.examples[i];if(!a)throw new Error("Invalid example");try{let l=await this.program.forward(this.ai,a,{maxRetries:1}),p=await e({prediction:l,example:a});o+=p}catch(l){console.warn(`Program evaluation failed for example ${i}: ${l instanceof Error?l.message:"Unknown error"}`)}}let r=n>0?o/n:0;console.log(`
191
- Performance: `,o,"/",n,"Average Score: ",r,`
192
- `)}};var kn=class{rows=[];baseUrl;dataset;split;config;options;constructor({dataset:e,split:t,config:n,options:o}){this.baseUrl="https://datasets-server.huggingface.co/rows",this.dataset=e,this.split=t,this.config=n,this.options=o}async fetchDataFromAPI(e){try{let t=await fetch(e);if(!t.ok)throw new Error(`Error fetching data: ${t.statusText}`);let n=await t.json();if(!n?.rows)throw new Error("Invalid data format");return n.rows}catch(t){throw console.error("Error fetching data from API:",t),t}}async loadData(){let e=this.options?.offset??0,t=this.options?.length??100,n=encodeURIComponent(this.dataset),o=`${this.baseUrl}?dataset=${n}&config=${this.config}&split=${this.split}&offset=${e}&length=${t}`;return console.log("Downloading data from API."),this.rows=await this.fetchDataFromAPI(o),this.rows}setData(e){this.rows=e}getData(){return this.rows}async getRows({count:e,fields:t,renameMap:n}){if(this.rows.length===0)throw new Error("No data loaded, call loadData or setData first.");return this.rows.slice(0,e).map(r=>{let i={};return t.forEach(a=>{let l=a.split("."),p=r.row;for(let u of l)Object.hasOwn(p,u)&&(p=p[u]);if(!p)return;let c=n&&a in n?n[a]:a;if(!c)throw new Error(`Invalid field name: ${a}`);i[c]=p}),i}).filter(r=>Object.keys(r).length!==0)}};var Vs=s=>{console.log(s)},Ko=(s=Vs)=>{let e=new G,t=e.gray("\u2500".repeat(50)),n=e.gray("\u2501".repeat(50));return o=>{let r="";switch(o.name){case"OptimizationStart":r=`
184
+ `}),t};renderInField=(e,t,n)=>{let r=t[e.name];if(ea(e,r,n))return;e.type&&bn(e,r);let o=Zi(e,r);return(this.fieldTemplates?.[e.name]??this.defaultRenderInField)(e,o)};defaultRenderInField=(e,t)=>{if(e.type?.name==="image"){let r=i=>{if(!i)throw new Error("Image field value is required.");if(typeof i!="object")throw new Error("Image field value must be an object.");if(!("mimeType"in i))throw new Error("Image field must have mimeType");if(!("data"in i))throw new Error("Image field must have data");return i},o=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("Image field value must be an array.");o=o.concat(t.map(i=>{let a=r(i);return{type:"image",mimeType:a.mimeType,image:a.data}}))}else{let i=r(t);o.push({type:"image",mimeType:i.mimeType,image:i.data})}return o}if(e.type?.name==="audio"){let r=i=>{if(!i)throw new Error("Audio field value is required.");if(typeof i!="object")throw new Error("Audio field value must be an object.");if(!("data"in i))throw new Error("Audio field must have data");return i},o=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("Audio field value must be an array.");o=o.concat(t.map(i=>{let a=r(i);return{type:"audio",format:a.format??"wav",data:a.data}}))}else{let i=r(t);o.push({type:"audio",format:i.format??"wav",data:i.data})}return o}if(e.type?.name==="file"){let r=i=>{if(!i)throw new Error("File field value is required.");if(typeof i!="object")throw new Error("File field value must be an object.");if(!("mimeType"in i))throw new Error("File field must have mimeType");let a="data"in i,l="fileUri"in i;if(!a&&!l)throw new Error("File field must have either data or fileUri");if(a&&l)throw new Error("File field cannot have both data and fileUri");return i},o=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("File field value must be an array.");o=o.concat(t.map(i=>{let a=r(i);return"fileUri"in a?{type:"file",mimeType:a.mimeType,fileUri:a.fileUri}:{type:"file",mimeType:a.mimeType,data:a.data}}))}else{let i=r(t);o.push("fileUri"in i?{type:"file",mimeType:i.mimeType,fileUri:i.fileUri}:{type:"file",mimeType:i.mimeType,data:i.data})}return o}if(e.type?.name==="url"){let r=i=>{if(!i)throw new Error("URL field value is required.");if(typeof i=="string")return{url:i};if(typeof i!="object")throw new Error("URL field value must be a string or object.");if(!("url"in i))throw new Error("URL field must have url property");return i},o=[{type:"text",text:`${e.title}: `}];if(e.type.isArray){if(!Array.isArray(t))throw new Error("URL field value must be an array.");o=o.concat(t.map(i=>{let a=r(i);return{type:"url",url:a.url,...a.title?{title:a.title}:{},...a.description?{description:a.description}:{}}}))}else{let i=r(t);o.push({type:"url",url:i.url,...i.title?{title:i.title}:{},...i.description?{description:i.description}:{}})}return o}let n=[e.title,": "];return Array.isArray(t)?(n.push(`
185
+ `),n.push(t.map(r=>`- ${r}`).join(`
186
+ `))):n.push(t),[{type:"text",text:n.join("")}]}},Us=s=>s.map(e=>`\`${e.title}\``).join(", "),Yi=s=>s.map(t=>{let n=t.title,r=t.type?.name?qs(t.type):"string",o=t.isOptional?`This optional ${r} field may be omitted`:`A ${r} field`,i=t.description?` ${Tn(t.description)}`:"";return`${n}: (${o})${i}`.trim()}).join(`
187
+ `),Xi=s=>s.map(t=>{let n=t.title,r=t.type?.name?qs(t.type):"string",o=t.isOptional?`Only include this ${r} field if its value is available`:`This ${r} field must be included`,i="";return t.description&&t.description.length>0&&(i=` ${t.type?.name==="class"?t.description:Tn(t.description)}`),t.type?.options&&t.type.options.length>0&&(i.length>0&&(i+=". "),i+=`Allowed values: ${t.type.options.join(", ")}`),`${n}: (${o})${i}`.trim()}).join(`
188
+ `),Zi=(s,e)=>{if(s.type?.name==="date"&&e instanceof Date){let t=e.toISOString();return t.slice(0,t.indexOf("T"))}return s.type?.name==="datetime"&&e instanceof Date?vs(e):s.type?.name==="image"&&typeof e=="object"||s.type?.name==="audio"&&typeof e=="object"||s.type?.name==="file"&&typeof e=="object"||s.type?.name==="url"&&(typeof e=="string"||typeof e=="object")||typeof e=="string"?e:JSON.stringify(e,null,2)},qs=s=>{let e=(()=>{switch(s?.name){case"string":return"string";case"number":return"number";case"boolean":return"boolean (true or false)";case"date":return'date ("YYYY-MM-DD" format)';case"datetime":return'date time ("YYYY-MM-DD HH:mm Timezone" format)';case"json":return"JSON object";case"class":return"classification class";case"code":return"code";case"file":return"file (with filename, mimeType, and data)";case"url":return"URL (string or object with url, title, description)";default:return"string"}})();return s?.isArray?`json array of ${e} items`:e};function Bs(s){return(e,t)=>{if(t.type==="text"){let n=e.length>0?e[e.length-1]:null;n&&n.type==="text"?n.text+=s+t.text:e.push(t)}else e.push(t);return e}}var ea=(s,e,t)=>{if(typeof e=="boolean")return!1;if(!e||(Array.isArray(e)||typeof e=="string")&&e.length===0){if(t?.isExample||s.isOptional||s.isInternal)return!0;let n=t?.isInputField!==!1?"input":"output";throw new Error(`Value for ${n} field '${s.name}' is required.`)}return!1};function Tn(s){let e=s.trim();return e.length>0?`${e.charAt(0).toUpperCase()}${e.slice(1)}${e.endsWith(".")?"":"."}`:""}function ta(s,e){let t=s.history(0,e),n=t.some(o=>o.role==="function");return t.some(o=>o.role==="assistant"&&"functionCalls"in o&&Array.isArray(o.functionCalls)&&o.functionCalls.length>0)&&n}function na(s,e){let t=s.history(0,e),n=[],r=t.filter(i=>i.role==="assistant"&&"functionCalls"in i&&Array.isArray(i.functionCalls)&&i.functionCalls.length>0),o=t.filter(i=>i.role==="function");for(let i of r)if("functionCalls"in i&&i.functionCalls)for(let a of i.functionCalls){let l=o.find(p=>"functionId"in p&&p.functionId===a.id);l&&"result"in l&&"functionId"in l&&n.push({index:n.length,functionName:a.function.name,functionId:a.id,args:a.function.params||"",result:String(l.result),isError:"isError"in l?!!l.isError:!1})}return n}async function Cn(s,e,t,n){if(!e?.resultPicker||s.length<=1)return 0;let r=e.resultPicker;if((t?ta(t,n):!1)&&t){let l=na(t,n),p=await r({type:"function",results:l});if(p<0||p>=l.length)throw new Error(`Result picker returned invalid index: ${p}. Must be between 0 and ${l.length-1}`);return p}let i=s.map((l,p)=>({index:p,sample:l.delta})),a=await r({type:"fields",results:i});if(a<0||a>=s.length)throw new Error(`Result picker returned invalid index: ${a}. Must be between 0 and ${s.length-1}`);return a}async function zs(s,e,t){let n=s?.getLast(e);if(!n||n.role!=="assistant"||n.chat.length<=1)return 0;let r=n.chat.map(i=>({version:0,index:i.index,delta:i.value}));return await Cn(r,t,s,e)}var Rt=class{tools;logger;constructor(e,t){this.tools=new Map(e.map(n=>[n.name,n])),this.logger=t}async route(e,t){let n={},r={},o=[];for(let[a,l]of Object.entries(e)){let p=this.tools.get(this.normalizeToolName(a));if(p&&l!==void 0&&l!==null)try{let c=await this.executeTool(p,l,t);n[a]=c,o.push(p.name)}catch{r[a]=l}else r[a]=l}let i={...r,...n};return{toolResults:n,remainingFields:i}}async executeTool(e,t,n){if(!e.func)throw new Error(`Tool ${e.name} has no handler function`);let r=typeof t=="object"&&t!==null?t:{},o=e.func;return await o(r,{sessionId:n?.sessionId,traceId:n?.traceId})}normalizeToolName(e){return e.replace(/_([a-z])/g,(t,n)=>n.toUpperCase())}isToolField(e){return this.tools.has(this.normalizeToolName(e))}getToolFieldNames(){return Array.from(this.tools.keys()).map(e=>e.replace(/([A-Z])/g,"_$1").toLowerCase().replace(/^_/,""))}};var wn=class{functionCallMode;tools;router;usePromptMode=!1;constructor(e){this.functionCallMode=e.functionCallMode??"auto",this.tools=e.functions??[],this.functionCallMode==="prompt"&&this.tools.length>0&&(this.usePromptMode=!0,this.router=new Rt(this.tools))}setUsePromptMode(e){this.usePromptMode=e,e&&this.tools.length>0&&!this.router&&(this.router=new Rt(this.tools))}getMode(){return this.functionCallMode}processSignature(e){return this.usePromptMode&&this.tools.length>0?e.injectToolFields(this.tools):e}async processResults(e,t){return this.usePromptMode&&this.router?(await this.router.route(e,t)).remainingFields:e}isPromptModeEnabled(){return this.usePromptMode}isNativeModeEnabled(){return!this.usePromptMode&&this.tools.length>0}getRouter(){return this.router}};var $=class extends ie{promptTemplate;asserts;streamingAsserts;options;functions;fieldProcessors=[];streamingFieldProcessors=[];excludeContentFromTrace=!1;thoughtFieldName;signatureToolCallingManager;constructor(e,t){super(e,{description:t?.description,traceLabel:t?.traceLabel}),this.options=t,this.thoughtFieldName=t?.thoughtFieldName??"thought";let n={functions:t?.functions,thoughtFieldName:this.thoughtFieldName};this.promptTemplate=new(t?.promptTemplate??de)(this.signature,n),this.asserts=this.options?.asserts??[],this.streamingAsserts=this.options?.streamingAsserts??[],this.excludeContentFromTrace=t?.excludeContentFromTrace??!1,this.usage=[],t?.functions&&(this.functions=_r(t.functions),t?.functionCallMode&&(this.signatureToolCallingManager=new wn({functionCallMode:t.functionCallMode,functions:this.functions})))}getSignatureName(){return this.signature.getDescription()||"unknown_signature"}getMetricsInstruments(){return kr()}updateMeter(e){kr(e)}createStates(e){return Array.from({length:e},(t,n)=>({index:n,functionCalls:[],values:{},content:"",functionsExecuted:new Set,xstate:{extractedFields:[],streamedIndex:{},s:-1}}))}addAssert=(e,t)=>{this.asserts.push({fn:e,message:t})};addStreamingAssert=(e,t,n)=>{this.streamingAsserts.push({fieldName:e,fn:t,message:n})};addFieldProcessorInternal=(e,t,n=!1)=>{let r=this.signature.getOutputFields().find(o=>o.name===e);if(!r)throw new Error(`addFieldProcessor: field ${e} not found`);if(n){let o=r.type?.name;if(!(!o||o==="string"||o==="code"))throw new Error(`addFieldProcessor: field ${e} is must be a text field`);this.streamingFieldProcessors.push({field:r,process:t})}else this.fieldProcessors.push({field:r,process:t})};addStreamingFieldProcessor=(e,t)=>{this.addFieldProcessorInternal(e,t,!0)};addFieldProcessor=(e,t)=>{this.addFieldProcessorInternal(e,t,!1)};async forwardSendRequest({ai:e,mem:t,options:n,traceContext:r,functions:o,functionCall:i,stepIndex:a}){let{sessionId:l,model:p,rateLimiter:c,stream:u,thinkingTokenBudget:d,showThoughts:m}=n??{},h=await zs(t,l,{resultPicker:n?.resultPicker}),g=t?.history(h,l)??[];if(g.length===0)throw new Error("No chat prompt found");let f={...n?.modelConfig,...n?.sampleCount?{n:n.sampleCount}:{},...n?.sampleCount&&n?.modelConfig?.temperature===1?{temperature:.8}:{}},A=this.isDebug(e,n),x=a===0,y=this.getLogger(e,n);return await e.chat({chatPrompt:g,functions:o,functionCall:i,modelConfig:f,model:p},{sessionId:l,rateLimiter:c,stream:u,debug:A,debugHideSystemPrompt:!x,thinkingTokenBudget:d,showThoughts:m,traceContext:r,abortSignal:n?.abortSignal,stepIndex:a,logger:y})}async*forwardCore({ai:e,mem:t,options:n,stepIndex:r,span:o,traceContext:i}){let{sessionId:a,functions:l}=n??{},p=n?.functionCall??this.options?.functionCall,c=n?.strictMode??!1,u=n.model,d=this.createStates(n.sampleCount??1),m=this.usage,h=r===0,g=this.getLogger(e,n),{functions:f,functionCall:A}=bs(l,p,h,n),x=await this.forwardSendRequest({ai:e,mem:t,options:n,traceContext:i,functions:f,functionCall:A,stepIndex:r});x instanceof ReadableStream?yield*_s({ai:e,model:u,res:x,mem:t,sessionId:a,functions:f,strictMode:c,span:o,states:d,usage:m,asserts:this.asserts,streamingAsserts:this.streamingAsserts,fieldProcessors:this.fieldProcessors,streamingFieldProcessors:this.streamingFieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,logger:g,functionResultFormatter:n?.functionResultFormatter??this.options?.functionResultFormatter}):yield*Ds({ai:e,model:u,res:x,mem:t,sessionId:a,functions:f,span:o,strictMode:c,states:d,usage:m,asserts:this.asserts,fieldProcessors:this.fieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,logger:g,functionResultFormatter:n?.functionResultFormatter??this.options?.functionResultFormatter})}async*_forward2(e,t,n,r,o,i){let a=(r?.stopFunction??this.options?.stopFunction)?.toLowerCase(),l=r.maxRetries??this.options?.maxRetries??10,p=r.maxSteps??this.options?.maxSteps??10,c=r.mem??this.options?.mem??new Ne;if(this.signatureToolCallingManager?.getMode()==="auto"&&this.functions&&this.functions.length>0){let y=e.getFeatures(r.model).functions;this.signatureToolCallingManager.setUsePromptMode(!y)}let u;if(r?.functions&&r.functions.length>0){let x=this.options?.promptTemplate??de,y=this.signature;this.signatureToolCallingManager?.isPromptModeEnabled()&&(y=this.signatureToolCallingManager.processSignature(this.signature));let w={functions:r.functions,thoughtFieldName:this.thoughtFieldName};this.promptTemplate=new x(y,w)}let d,m=performance.now();Array.isArray(t)?(nr(t),d=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos})):d=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos});let h=performance.now()-m,g=this.getMetricsInstruments();g&&yn(g,"prompt_render",h,this.getSignatureName());let f=performance.now();c.addRequest(d,r.sessionId);let A=performance.now()-f;g&&yn(g,"memory_update",A,this.getSignatureName());e:for(let x=0;x<p;x++){for(let w=0;w<l;w++)try{let I=this.forwardCore({options:r,ai:e,mem:c,stepIndex:x,span:o,traceContext:i});for await(let v of I)v!==void 0&&(yield{version:w,index:v.index,delta:v.delta});if(Ls(c,a,n,r?.sessionId)){let v=this.getMetricsInstruments();v&&fn(v,x+1,p,this.getSignatureName());continue e}let k=this.getMetricsInstruments();if(k){fn(k,x+1,p,this.getSignatureName());let v=new Set;n.forEach(H=>{H.functionsExecuted.forEach(Y=>v.add(Y))}),v.size>0&&Fr(k,!0,v.size,!0,!1,this.getSignatureName()),ms(k,this.fieldProcessors.length,this.streamingFieldProcessors.length,this.getSignatureName())}return}catch(I){let R,k=this.isDebug(e,r),v=this.getLogger(e,r),H=this.getMetricsInstruments(),Y=this.getSignatureName(),Qe={error:I,errCount:w,logger:v,metricsInstruments:H,signatureName:Y,span:o,debug:k};if(o?.recordException(I),I instanceof D)R=xs(Qe),u=I;else if(I instanceof oe)R=fs(Qe),u=I;else if(I instanceof O)ys(Qe);else if(!(I instanceof K))throw Kr(I,e,this.signature);R&&(c.addRequest([{role:"user",content:this.promptTemplate.renderExtraFields(R)}],r.sessionId),c.addTag("error",r.sessionId))}let y=this.getMetricsInstruments();throw y&&Pr(y,l,!1,l,this.getSignatureName()),Kr(new Error(`Unable to fix validation error: ${u?.toString()}`),e,this.signature)}throw g&&fn(g,p,p,this.getSignatureName()),Kr(new Error(`Max steps reached: ${p}`),e,this.signature)}async*_forward1(e,t,n){let r=performance.now(),o=this.createStates(n.sampleCount??1),i=performance.now()-r,a=this.getMetricsInstruments();a&&yn(a,"state_creation",i,this.getSignatureName());let l=n?.tracer??this.options?.tracer??e.getOptions().tracer,p=this.functions;if(n?.functions&&(p=_r(n.functions,this.functions)),!l){yield*this._forward2(e,t,o,{...n,functions:p});return}let c=p?.map(A=>A.name).join(","),u={signature:JSON.stringify(this.signature.toJSON(),null,2),...this.examples?{examples:JSON.stringify(this.examples,null,2)}:{},...c?{provided_functions:c}:{},...n?.model?{model:n.model}:{},...n?.thinkingTokenBudget?{thinking_token_budget:n.thinkingTokenBudget}:{},...n?.showThoughts?{show_thoughts:n.showThoughts}:{},...n?.maxSteps?{max_steps:n.maxSteps}:{},...n?.maxRetries?{max_retries:n.maxRetries}:{}},d=this.traceLabel&&n.traceLabel?`${this.traceLabel} > ${n.traceLabel}`:n.traceLabel??this.traceLabel,m=d?`AxGen > ${d}`:"AxGen",h=l.startSpan(m,{kind:qe.SpanKind.SERVER,attributes:u}),g=qe.context.active(),f=qe.trace.setSpan(g,h);try{if(this.excludeContentFromTrace||h.addEvent("input",{content:JSON.stringify(t,null,2)}),yield*this._forward2(e,t,o,{...n,functions:p},h,f),!this.excludeContentFromTrace){let A=o.map(y=>y.values),x=A.length===1?A[0]:A;h.addEvent("output",{content:JSON.stringify(x,null,2)})}}finally{h.end()}}async forward(e,t,n){let r=performance.now(),o=this.getSignatureName(),i=n?.stream??!1,a=!1,l=0,p=!1,c=0,u=!1;try{let d=this.getMetricsInstruments();d&&As(d,this.signature.getInputFields().length,this.signature.getOutputFields().length,this.examples?.length??0,this.demos?.length??0,o),p=!!(n?.functions||this.functions);let m=this._forward1(e,t,n??{}),h=[],g=0,f=0;for await(let R of m)R.version!==g&&(h=[]),g=R.version,h=$r(h,R),f++;l=g;let A=performance.now();u=!!n?.resultPicker;let x=await Cn(h,{resultPicker:n?.resultPicker},n?.mem,n?.sessionId),y=performance.now()-A,I=h[x]?.delta??{};if(this.signatureToolCallingManager?.isPromptModeEnabled()&&(I=await this.signatureToolCallingManager.processResults(I,{sessionId:n?.sessionId})),this.trace={...t,...I},u&&this.isDebug(e,n)){let R=this.getLogger(e,n);fo(h.length,x,y,R)}return a=!0,d&&(hs(d,h.length,u,u?y:void 0,o),gs(d,i,f,void 0,o)),I}catch(d){throw a=!1,d}finally{let d=performance.now()-r,m=this.getMetricsInstruments();m&&(us(m,d,a,o,e.getName(),n?.model?String(n.model):void 0),p&&Fr(m,p,c,c>0,!1,o),l>0&&Pr(m,l,a,n?.maxRetries??10,o))}}async*streamingForward(e,t,n){if(!n?.resultPicker){yield*this._forward1(e,t,{...n,stream:!0});return}let r=this._forward1(e,t,{...n,stream:!0}),o=[],i=0;for await(let p of r)p.version!==i&&(o=[]),i=p.version,o=$r(o,p);let a=await Cn(o,{resultPicker:n?.resultPicker},n?.mem,n?.sessionId),l=o[a];l&&(yield{version:i,index:a,delta:l.delta})}setExamples(e,t){super.setExamples(e,t)}isDebug(e,t){return t?.debug??this.options?.debug??e.getOptions().debug??!1}getLogger(e,t){return t?.logger??this.options?.logger??M.logger??e.getLogger()}},Tt=class extends Error{details;constructor(e,t,n){super(e),this.name="AxGenerateError",this.details=t,n?.cause&&(this.cause=n.cause)}};function Kr(s,e,t){let n=s instanceof Error?s:new Error(String(s)),r=e.getLastUsedChatModel(),o=e.getLastUsedModelConfig(),i={model:r,maxTokens:o?.maxTokens,streaming:o?.stream??!1,signature:{input:t.getInputFields(),output:t.getOutputFields(),description:t.getDescription()}};return new Tt("Generate failed",i,{cause:n})}var ra=s=>s.replace(/^\W+|\W+$/g,""),oa=(s,e)=>{let t=s.search(e);if(t===-1)return[s];let n=s.match(e);if(!n)throw new Error("Match failed unexpectedly.");let r=s.substring(0,t),o=s.substring(t+n[0].length);return[r,o]},sa=s=>{let e=new Set,t=[];for(let n of s)e.has(n)||(e.add(n),t.push(n));return t},ia=s=>{let e=s.match(/^(\d+)[.,\s]+(.*)$/);if(!e||e.length<3)throw new Error('line must start with a number, a dot and then text. e.g. "1. hello"');let t=Number.parseInt(e[1],10),n=e[2].trim();return{id:t,text:n}},aa=s=>{let e=s.match(/^(\d+)[.,\s]+(.*)$/);return e&&e[2]!==void 0?e[2].trim():s},la=(s,e)=>{let t=[];for(let n=0;n<s.length;n+=e)t.push(s.slice(n,n+e));return t},Sn={trimNonAlphaNum:ra,splitIntoTwo:oa,dedup:sa,extractIdAndText:ia,extractIndexPrefixedText:aa,batchArray:la};var vn=class extends ${constructor(e){super(`"You are a re-ranker assistant tasked with evaluating a set of content items in relation to a specific question. Your role involves critically analyzing each content item to determine its relevance to the question and re-ranking them accordingly. This process includes assigning a relevance score from 0 to 10 to each content item based on how well it answers the question, its coverage of the topic, and the reliability of its information. This re-ranked list should start with the content item that is most relevant to the question and end with the least relevant. Output only the list."
189
+ query: string, items: string[] -> rankedItems: string[] "list of id, 5-words Rationale, relevance score"`,e)}forward=async(e,t,n)=>{let{rankedItems:r}=await super.forward(e,t,n),o=r.map(a=>{let{id:l}=Sn.extractIdAndText(a);return l});return{rankedItems:t.items.map((a,l)=>{let p=o[l];return p!==void 0?t.items[p]:void 0}).filter(a=>a!==void 0)}}};var On=class{tikaUrl;fetch;constructor(e){let t=e??{url:"http://localhost:9998/"};this.tikaUrl=new URL("/tika",t.url),this.fetch=t.fetch}async _convert(e,t){if(!e)throw new Error("Failed to read file data");let n=t?.format==="html"?"text/html":"text/plain";try{let r={body:e,headers:{Accept:n},method:"PUT"};typeof window>"u"&&typeof process<"u"&&(r.duplex="half");let o=await(this.fetch??fetch)(this.tikaUrl,r);if(!o.ok)throw new Error(`Failed to upload file: ${o.statusText}`);return await o.text()}catch(r){throw new Error(`Error converting file: ${r}`)}}async convert(e,t){let n=[],r=t?.batchSize??10;for(let o=0;o<e.length;o+=r){let a=e.slice(o,o+r).map(p=>this._convert(p,{format:t?.format})),l=await Promise.all(a);n.push(...l)}return n}};var js=new G,Mn=class{name;context;constructor(e,t){this.name=e,this.context=t}getName(){return this.name}getContext(){return this.context}},kn=class{ai;db;debug;constructor(e){this.db=new re,this.ai=e}getState(){return this.db.getDB()}setState(e){this.db.setDB(e)}setClasses=async(e,t)=>{for(let n of e){let r=await this.ai.embed({texts:n.getContext()},{abortSignal:t?.abortSignal});await this.db.upsert({id:n.getName(),table:"classes",values:r.embeddings[0]})}};async forward(e,t){let{embeddings:n}=await this.ai.embed({texts:[e]},{abortSignal:t?.abortSignal}),o=(await this.db.query({table:"classes",values:n[0]})).matches;if(typeof t?.cutoff=="number"){let{cutoff:a}=t;o=o.filter(l=>l.score<=a)}this.debug&&console.log(`${js.whiteBright(`query: ${e}`)}
190
+ ${js.greenBright(JSON.stringify(o.map(a=>`${a.id}, ${a.score}`)))}`);let i=o.at(0);return i?i.id:""}setOptions(e){typeof e.debug=="boolean"&&(this.debug=e.debug)}};var Hs=new Set(["0o","0s","3a","3b","3d","6b","6o","a","a1","a2","a3","a4","ab","able","about","above","abst","ac","accordance","according","accordingly","across","act","actually","ad","added","adj","ae","af","affected","affecting","affects","after","afterwards","ag","again","against","ah","ain","ain't","aj","al","all","allow","allows","almost","alone","along","already","also","although","always","am","among","amongst","amoungst","amount","an","and","announce","another","any","anybody","anyhow","anymore","anyone","anything","anyway","anyways","anywhere","ao","ap","apart","apparently","appear","appreciate","appropriate","approximately","ar","are","aren","arent","aren't","arise","around","as","a's","aside","ask","asking","associated","at","au","auth","av","available","aw","away","awfully","ax","ay","az","b","b1","b2","b3","ba","back","bc","bd","be","became","because","become","becomes","becoming","been","before","beforehand","begin","beginning","beginnings","begins","behind","being","believe","below","beside","besides","best","better","between","beyond","bi","bill","biol","bj","bk","bl","bn","both","bottom","bp","br","brief","briefly","bs","bt","bu","but","bx","by","c","c1","c2","c3","ca","call","came","can","cannot","cant","can't","cause","causes","cc","cd","ce","certain","certainly","cf","cg","ch","changes","ci","cit","cj","cl","clearly","cm","c'mon","cn","co","com","come","comes","con","concerning","consequently","consider","considering","contain","containing","contains","corresponding","could","couldn","couldnt","couldn't","course","cp","cq","cr","cry","cs","c's","ct","cu","currently","cv","cx","cy","cz","d","d2","da","date","dc","dd","de","definitely","describe","described","despite","detail","df","di","did","didn","didn't","different","dj","dk","dl","do","does","doesn","doesn't","doing","don","done","don't","down","downwards","dp","dr","ds","dt","du","due","during","dx","dy","e","e2","e3","ea","each","ec","ed","edu","ee","ef","effect","eg","ei","eight","eighty","either","ej","el","eleven","else","elsewhere","em","empty","en","end","ending","enough","entirely","eo","ep","eq","er","es","especially","est","et","et-al","etc","eu","ev","even","ever","every","everybody","everyone","everything","everywhere","ex","exactly","example","except","ey","f","f2","fa","far","fc","few","ff","fi","fifteen","fifth","fify","fill","find","fire","first","five","fix","fj","fl","fn","fo","followed","following","follows","for","former","formerly","forth","forty","found","four","fr","from","front","ft","fu","full","further","furthermore","fy","g","ga","gave","ge","get","gets","getting","gi","give","given","gives","giving","gj","gl","go","goes","going","gone","got","gotten","gr","greetings","gs","gy","h","h2","h3","had","hadn","hadn't","happens","hardly","has","hasn","hasnt","hasn't","have","haven","haven't","having","he","hed","he'd","he'll","hello","help","hence","her","here","hereafter","hereby","herein","heres","here's","hereupon","hers","herself","hes","he's","hh","hi","hid","him","himself","his","hither","hj","ho","home","hopefully","how","howbeit","however","how's","hr","hs","http","hu","hundred","hy","i","i2","i3","i4","i6","i7","i8","ia","ib","ibid","ic","id","i'd","ie","if","ig","ignored","ih","ii","ij","il","i'll","im","i'm","immediate","immediately","importance","important","in","inasmuch","inc","indeed","index","indicate","indicated","indicates","information","inner","insofar","instead","interest","into","invention","inward","io","ip","iq","ir","is","isn","isn't","it","itd","it'd","it'll","its","it's","itself","iv","i've","ix","iy","iz","j","jj","jr","js","jt","ju","just","k","ke","keep","keeps","kept","kg","kj","km","know","known","knows","ko","l","l2","la","largely","last","lately","later","latter","latterly","lb","lc","le","least","les","less","lest","let","lets","let's","lf","like","liked","likely","line","little","lj","ll","ll","ln","lo","look","looking","looks","los","lr","ls","lt","ltd","m","m2","ma","made","mainly","make","makes","many","may","maybe","me","mean","means","meantime","meanwhile","merely","mg","might","mightn","mightn't","mill","million","mine","miss","ml","mn","mo","more","moreover","most","mostly","move","mr","mrs","ms","mt","mu","much","mug","must","mustn","mustn't","my","myself","model","n","n2","na","name","namely","nay","nc","nd","ne","near","nearly","necessarily","necessary","need","needn","needn't","needs","neither","never","nevertheless","new","next","ng","ni","nine","ninety","nj","nl","nn","no","nobody","non","none","nonetheless","noone","nor","normally","nos","not","noted","nothing","novel","now","nowhere","nr","ns","nt","ny","o","oa","ob","obtain","obtained","obviously","oc","od","of","off","often","og","oh","oi","oj","ok","okay","ol","old","om","omitted","on","once","one","ones","only","onto","oo","op","oq","or","ord","os","ot","other","others","otherwise","ou","ought","our","ours","ourselves","out","outside","over","overall","ow","owing","own","ox","oz","p","p1","p2","p3","page","pagecount","pages","par","part","particular","particularly","pas","past","pc","pd","pe","per","perhaps","pf","ph","pi","pj","pk","pl","placed","please","plus","pm","pn","po","poorly","possible","possibly","potentially","pp","pq","pr","predominantly","present","presumably","previously","primarily","probably","promptly","proud","provides","ps","pt","pu","put","py","q","qj","qu","que","quickly","quite","qv","r","r2","ra","ran","rather","rc","rd","re","readily","really","reasonably","recent","recently","ref","refs","regarding","regardless","regards","related","relatively","research","research-articl","respectively","resulted","resulting","results","rf","rh","ri","right","rj","rl","rm","rn","ro","rq","rr","rs","rt","ru","run","rv","ry","s","s2","sa","said","same","saw","say","saying","says","sc","sd","se","sec","second","secondly","section","see","seeing","seem","seemed","seeming","seems","seen","self","selves","sensible","sent","serious","seriously","seven","several","sf","shall","shan","shan't","she","shed","she'd","she'll","shes","she's","should","shouldn","shouldn't","should've","show","showed","shown","showns","shows","si","side","significant","significantly","similar","similarly","since","sincere","six","sixty","sj","sl","slightly","sm","sn","so","some","somebody","somehow","someone","somethan","something","sometime","sometimes","somewhat","somewhere","soon","sorry","sp","specifically","specified","specify","specifying","sq","sr","ss","st","still","stop","strongly","sub","substantially","successfully","such","sufficiently","suggest","sup","sure","sy","system","sz","t","t1","t2","t3","take","taken","taking","tb","tc","td","te","tell","ten","tends","tf","th","than","thank","thanks","thanx","that","that'll","thats","that's","that've","the","their","theirs","them","themselves","then","thence","there","thereafter","thereby","thered","therefore","therein","there'll","thereof","therere","theres","there's","thereto","thereupon","there've","these","they","theyd","they'd","they'll","theyre","they're","they've","thickv","thin","think","third","this","thorough","thoroughly","those","thou","though","thoughh","thousand","three","throug","through","throughout","thru","thus","ti","til","tip","tj","tl","tm","tn","to","together","too","took","top","toward","towards","tp","tq","tr","tried","tries","truly","try","trying","ts","t's","tt","tv","twelve","twenty","twice","two","tx","u","u201d","ue","ui","uj","uk","um","un","under","unfortunately","unless","unlike","unlikely","until","unto","uo","up","upon","ups","ur","us","use","used","useful","usefully","usefulness","uses","using","usually","ut","v","va","value","various","vd","ve","ve","very","via","viz","vj","vo","vol","vols","volumtype","vq","vs","vt","vu","w","wa","want","wants","was","wasn","wasnt","wasn't","way","we","wed","we'd","welcome","well","we'll","well-b","went","were","we're","weren","werent","weren't","we've","what","whatever","what'll","whats","what's","when","whence","whenever","when's","where","whereafter","whereas","whereby","wherein","wheres","where's","whereupon","wherever","whether","which","while","whim","whither","who","whod","whoever","whole","who'll","whom","whomever","whos","who's","whose","why","why's","wi","widely","will","willing","wish","with","within","without","wo","won","wonder","wont","won't","words","world","would","wouldn","wouldnt","wouldn't","www","x","x1","x2","x3","xf","xi","xj","xk","xl","xn","xo","xs","xt","xv","xx","y","y2","yes","yet","yj","yl","you","youd","you'd","you'll","your","youre","you're","yours","yourself","yourselves","you've","yr","ys","yt","z","zero","zi","zz","task"]);function Ks(s,e){return s.filter(t=>!e.has(t))}function Vs(s){let e={};for(let t of s)e[t]=(e[t]||0)+1;return e}function me(s){let e=s.normalize("NFD");return e=e.replace(/\b(a|an|the)\b/g," "),e=e.split(/\s+/).join(" "),e=e.replace(/[!"#$%&'()*+,-./:;<=>?@[\]^_`{|}~]/g,""),e.toLowerCase()}function pa(s,e){return me(s)===me(e)?1:0}function ca(s,e){let t=me(s).split(" "),n=me(e).split(" "),r=Vs(t),o=Vs(n),i=0;for(let p in r){let c=r[p]??0,u=o[p]??0;i+=Math.min(c,u)}if(i===0)return 0;let a=i/t.length,l=i/n.length;return 2*a*l/(a+l)}function ua(s,e,t,n=!1){let r=me(s).split(" "),o=me(e).split(" "),i=me(t).split(" "),a=new Set([...Hs,...r]);o=Ks(o,a),i=Ks(i,a);let l=0,p=l/o.length,c=l/i.length,u=2*p*c/(p+c);return n?c:u}var Ws={emScore:pa,f1Score:ca,novelF1ScoreOptimized:ua};var En=class{ai;program;examples;constructor({ai:e,program:t,examples:n=[]}){if(n.length===0)throw new Error("No examples found");this.ai=e,this.program=t,this.examples=n}async run(e){let t=Date.now(),n=this.examples.length,r=0;for(let i=0;i<n;i++){let a=this.examples[i];if(!a)throw new Error("Invalid example");try{let l=await this.program.forward(this.ai,a,{maxRetries:1}),p=await e({prediction:l,example:a});r+=p}catch(l){console.warn(`Program evaluation failed for example ${i}: ${l instanceof Error?l.message:"Unknown error"}`)}}let o=n>0?r/n:0;console.log(`
191
+ Performance: `,r,"/",n,"Average Score: ",o,`
192
+ `)}};var Pn=class{rows=[];baseUrl;dataset;split;config;options;constructor({dataset:e,split:t,config:n,options:r}){this.baseUrl="https://datasets-server.huggingface.co/rows",this.dataset=e,this.split=t,this.config=n,this.options=r}async fetchDataFromAPI(e){try{let t=await fetch(e);if(!t.ok)throw new Error(`Error fetching data: ${t.statusText}`);let n=await t.json();if(!n?.rows)throw new Error("Invalid data format");return n.rows}catch(t){throw console.error("Error fetching data from API:",t),t}}async loadData(){let e=this.options?.offset??0,t=this.options?.length??100,n=encodeURIComponent(this.dataset),r=`${this.baseUrl}?dataset=${n}&config=${this.config}&split=${this.split}&offset=${e}&length=${t}`;return console.log("Downloading data from API."),this.rows=await this.fetchDataFromAPI(r),this.rows}setData(e){this.rows=e}getData(){return this.rows}async getRows({count:e,fields:t,renameMap:n}){if(this.rows.length===0)throw new Error("No data loaded, call loadData or setData first.");return this.rows.slice(0,e).map(o=>{let i={};return t.forEach(a=>{let l=a.split("."),p=o.row;for(let u of l)Object.hasOwn(p,u)&&(p=p[u]);if(!p)return;let c=n&&a in n?n[a]:a;if(!c)throw new Error(`Invalid field name: ${a}`);i[c]=p}),i}).filter(o=>Object.keys(o).length!==0)}};var Js=s=>{console.log(s)},Vr=(s=Js)=>{let e=new G,t=e.gray("\u2500".repeat(50)),n=e.gray("\u2501".repeat(50));return r=>{let o="";switch(r.name){case"OptimizationStart":o=`
193
193
  ${e.blueBright("\u25CF ")}${e.whiteBright("Optimization Started")}
194
194
  ${t}
195
- ${e.white("Optimizer:")} ${e.cyan(o.value.optimizerType)}
196
- ${e.white("Examples:")} ${e.green(o.value.exampleCount.toString())} training, ${e.green(o.value.validationCount.toString())} validation
197
- ${e.white("Config:")} ${e.white(JSON.stringify(o.value.config).slice(0,80))}${JSON.stringify(o.value.config).length>80?"...":""}
195
+ ${e.white("Optimizer:")} ${e.cyan(r.value.optimizerType)}
196
+ ${e.white("Examples:")} ${e.green(r.value.exampleCount.toString())} training, ${e.green(r.value.validationCount.toString())} validation
197
+ ${e.white("Config:")} ${e.white(JSON.stringify(r.value.config).slice(0,80))}${JSON.stringify(r.value.config).length>80?"...":""}
198
198
  ${n}
199
- `;break;case"RoundProgress":r=`${e.yellow("\u25CF ")}${e.whiteBright(`Round ${o.value.round}/${o.value.totalRounds}`)}
200
- ${e.white("Score:")} ${e.green(o.value.currentScore.toFixed(3))} ${e.white("(best:")} ${e.greenBright(o.value.bestScore.toFixed(3))}${e.white(")")}
201
- `;break;case"EarlyStopping":r=`
199
+ `;break;case"RoundProgress":o=`${e.yellow("\u25CF ")}${e.whiteBright(`Round ${r.value.round}/${r.value.totalRounds}`)}
200
+ ${e.white("Score:")} ${e.green(r.value.currentScore.toFixed(3))} ${e.white("(best:")} ${e.greenBright(r.value.bestScore.toFixed(3))}${e.white(")")}
201
+ `;break;case"EarlyStopping":o=`
202
202
  ${e.red("\u25CF ")}${e.whiteBright("Early Stopping")}
203
203
  ${t}
204
- ${e.white("Round:")} ${e.yellow(o.value.round.toString())}
205
- ${e.white("Reason:")} ${e.yellow(o.value.reason)}
206
- ${e.white("Final Score:")} ${e.green(o.value.finalScore.toFixed(3))}
204
+ ${e.white("Round:")} ${e.yellow(r.value.round.toString())}
205
+ ${e.white("Reason:")} ${e.yellow(r.value.reason)}
206
+ ${e.white("Final Score:")} ${e.green(r.value.finalScore.toFixed(3))}
207
207
  ${n}
208
- `;break;case"OptimizationComplete":r=`
208
+ `;break;case"OptimizationComplete":o=`
209
209
  ${e.green("\u25CF ")}${e.whiteBright("Optimization Complete")}
210
210
  ${t}
211
- ${e.white("Best Score:")} ${e.greenBright(o.value.bestScore.toFixed(3))}
212
- ${e.white("Best Config:")} ${e.cyan(JSON.stringify(o.value.bestConfiguration).slice(0,80))}${JSON.stringify(o.value.bestConfiguration).length>80?"...":""}
213
- ${e.white("Total Calls:")} ${e.white(o.value.stats.totalCalls?.toString()||"N/A")}
214
- ${e.white("Success Rate:")} ${e.green(`${((o.value.stats.successfulDemos||0)/Math.max(o.value.stats.totalCalls||1,1)*100).toFixed(1)}%`)}
211
+ ${e.white("Best Score:")} ${e.greenBright(r.value.bestScore.toFixed(3))}
212
+ ${e.white("Best Config:")} ${e.cyan(JSON.stringify(r.value.bestConfiguration).slice(0,80))}${JSON.stringify(r.value.bestConfiguration).length>80?"...":""}
213
+ ${e.white("Total Calls:")} ${e.white(r.value.stats.totalCalls?.toString()||"N/A")}
214
+ ${e.white("Success Rate:")} ${e.green(`${((r.value.stats.successfulDemos||0)/Math.max(r.value.stats.totalCalls||1,1)*100).toFixed(1)}%`)}
215
215
  ${n}
216
- `;break;case"ConfigurationProposal":r=`${e.magenta("\u25CF ")}${e.whiteBright(`${o.value.type} Proposals`)} ${e.white(`(${o.value.count})`)}
217
- ${e.white("Candidates:")} ${e.white(o.value.proposals.slice(0,2).map(i=>typeof i=="string"?`"${i.slice(0,40)}..."`:`${JSON.stringify(i).slice(0,40)}...`).join(", "))}
218
- `;break;case"BootstrappedDemos":r=`${e.cyan("\u25CF ")}${e.whiteBright("Bootstrapped Demos")} ${e.white(`(${o.value.count})`)}
219
- ${e.white("Generated:")} ${e.green(o.value.count.toString())} demonstration examples
220
- `;break;case"BestConfigFound":r=`${e.green("\u25CF ")}${e.whiteBright("Best Configuration Found")}
221
- ${e.white("Score:")} ${e.greenBright(o.value.score.toFixed(3))}
222
- ${e.white("Config:")} ${e.cyan(JSON.stringify(o.value.config).slice(0,80))}${JSON.stringify(o.value.config).length>80?"...":""}
223
- `;break;default:r=`${e.red("\u25CF ")}${e.whiteBright("Unknown Event")}
224
- ${e.white(JSON.stringify(o).slice(0,100))}${JSON.stringify(o).length>100?"...":""}
225
- `}s(r)}},Ws=(s=Vs)=>{let e="\u2500".repeat(60);return t=>{let n="";switch(t.name){case"OptimizationStart":n=`[ OPTIMIZATION START: ${t.value.optimizerType} ]
216
+ `;break;case"ConfigurationProposal":o=`${e.magenta("\u25CF ")}${e.whiteBright(`${r.value.type} Proposals`)} ${e.white(`(${r.value.count})`)}
217
+ ${e.white("Candidates:")} ${e.white(r.value.proposals.slice(0,2).map(i=>typeof i=="string"?`"${i.slice(0,40)}..."`:`${JSON.stringify(i).slice(0,40)}...`).join(", "))}
218
+ `;break;case"BootstrappedDemos":o=`${e.cyan("\u25CF ")}${e.whiteBright("Bootstrapped Demos")} ${e.white(`(${r.value.count})`)}
219
+ ${e.white("Generated:")} ${e.green(r.value.count.toString())} demonstration examples
220
+ `;break;case"BestConfigFound":o=`${e.green("\u25CF ")}${e.whiteBright("Best Configuration Found")}
221
+ ${e.white("Score:")} ${e.greenBright(r.value.score.toFixed(3))}
222
+ ${e.white("Config:")} ${e.cyan(JSON.stringify(r.value.config).slice(0,80))}${JSON.stringify(r.value.config).length>80?"...":""}
223
+ `;break;default:o=`${e.red("\u25CF ")}${e.whiteBright("Unknown Event")}
224
+ ${e.white(JSON.stringify(r).slice(0,100))}${JSON.stringify(r).length>100?"...":""}
225
+ `}s(o)}},Qs=(s=Js)=>{let e="\u2500".repeat(60);return t=>{let n="";switch(t.name){case"OptimizationStart":n=`[ OPTIMIZATION START: ${t.value.optimizerType} ]
226
226
  ${e}
227
227
  Config: ${JSON.stringify(t.value.config,null,2)}
228
228
  Examples: ${t.value.exampleCount}, Validation: ${t.value.validationCount}
@@ -248,7 +248,7 @@ Score: ${t.value.score.toFixed(3)}
248
248
  Config: ${JSON.stringify(t.value.config)}
249
249
  ${e}`;break;default:n=`[ UNKNOWN OPTIMIZER EVENT ]
250
250
  ${JSON.stringify(t)}
251
- ${e}`}s(n)}},En=Ko();var Vo={enabled:!0,enabledCategories:["optimization","convergence","resource_usage","teacher_student","checkpointing","pareto"],maxLabelLength:100,samplingRate:1},Pn,ca=s=>{if(Pn)return Pn;if(s)return Pn=ua(s),Pn};var Fn=Vo,Js=s=>{Fn={...Fn,...s}},Qs=()=>({...Fn}),ua=s=>({optimizationLatencyHistogram:s.createHistogram("ax_optimizer_optimization_duration_ms",{description:"End-to-end duration of optimization runs",unit:"ms"}),optimizationRequestsCounter:s.createCounter("ax_optimizer_optimization_requests_total",{description:"Total number of optimization requests"}),optimizationErrorsCounter:s.createCounter("ax_optimizer_optimization_errors_total",{description:"Total number of failed optimizations"}),convergenceRoundsHistogram:s.createHistogram("ax_optimizer_convergence_rounds",{description:"Number of rounds until convergence"}),convergenceScoreGauge:s.createGauge("ax_optimizer_convergence_score",{description:"Current best score during optimization"}),convergenceImprovementGauge:s.createGauge("ax_optimizer_convergence_improvement",{description:"Improvement in score from baseline"}),stagnationRoundsGauge:s.createGauge("ax_optimizer_stagnation_rounds",{description:"Number of rounds without improvement"}),earlyStoppingCounter:s.createCounter("ax_optimizer_early_stopping_total",{description:"Total number of early stopping events"}),tokenUsageCounter:s.createCounter("ax_optimizer_token_usage_total",{description:"Total tokens used during optimization"}),costUsageCounter:s.createCounter("ax_optimizer_cost_usage_total",{description:"Total cost incurred during optimization",unit:"$"}),memoryUsageGauge:s.createGauge("ax_optimizer_memory_usage_bytes",{description:"Peak memory usage during optimization",unit:"By"}),optimizationDurationHistogram:s.createHistogram("ax_optimizer_duration_ms",{description:"Duration of optimization runs",unit:"ms"}),teacherStudentUsageCounter:s.createCounter("ax_optimizer_teacher_student_usage_total",{description:"Total number of teacher-student interactions"}),teacherStudentLatencyHistogram:s.createHistogram("ax_optimizer_teacher_student_latency_ms",{description:"Latency of teacher-student interactions",unit:"ms"}),teacherStudentScoreImprovementGauge:s.createGauge("ax_optimizer_teacher_student_score_improvement",{description:"Score improvement from teacher-student interactions"}),checkpointSaveCounter:s.createCounter("ax_optimizer_checkpoint_save_total",{description:"Total number of checkpoint saves"}),checkpointLoadCounter:s.createCounter("ax_optimizer_checkpoint_load_total",{description:"Total number of checkpoint loads"}),checkpointSaveLatencyHistogram:s.createHistogram("ax_optimizer_checkpoint_save_latency_ms",{description:"Latency of checkpoint save operations",unit:"ms"}),checkpointLoadLatencyHistogram:s.createHistogram("ax_optimizer_checkpoint_load_latency_ms",{description:"Latency of checkpoint load operations",unit:"ms"}),paretoOptimizationsCounter:s.createCounter("ax_optimizer_pareto_optimizations_total",{description:"Total number of Pareto optimizations"}),paretoFrontSizeHistogram:s.createHistogram("ax_optimizer_pareto_front_size",{description:"Size of Pareto frontier"}),paretoHypervolumeGauge:s.createGauge("ax_optimizer_pareto_hypervolume",{description:"Hypervolume of Pareto frontier"}),paretoSolutionsGeneratedHistogram:s.createHistogram("ax_optimizer_pareto_solutions_generated",{description:"Number of solutions generated for Pareto optimization"}),programInputFieldsGauge:s.createGauge("ax_optimizer_program_input_fields",{description:"Number of input fields in optimized program"}),programOutputFieldsGauge:s.createGauge("ax_optimizer_program_output_fields",{description:"Number of output fields in optimized program"}),examplesCountGauge:s.createGauge("ax_optimizer_examples_count",{description:"Number of training examples used"}),validationSetSizeGauge:s.createGauge("ax_optimizer_validation_set_size",{description:"Size of validation set used"}),evaluationLatencyHistogram:s.createHistogram("ax_optimizer_evaluation_latency_ms",{description:"Latency of program evaluations",unit:"ms"}),demoGenerationLatencyHistogram:s.createHistogram("ax_optimizer_demo_generation_latency_ms",{description:"Latency of demo generation",unit:"ms"}),metricComputationLatencyHistogram:s.createHistogram("ax_optimizer_metric_computation_latency_ms",{description:"Latency of metric computation",unit:"ms"}),optimizerTypeGauge:s.createGauge("ax_optimizer_type",{description:"Type of optimizer being used"}),targetScoreGauge:s.createGauge("ax_optimizer_target_score",{description:"Target score for optimization"}),maxRoundsGauge:s.createGauge("ax_optimizer_max_rounds",{description:"Maximum rounds for optimization"})}),Q=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let o=String(n),r=Fn.maxLabelLength;e[t]=o.length>r?o.substring(0,r):o}return e},da=(s,e,t,n,o)=>{try{let r=Q({success:t.toString(),optimizer_type:n,...o?{program_signature:o}:{}});s.optimizationLatencyHistogram&&s.optimizationLatencyHistogram.record(e,r),s.optimizationRequestsCounter&&s.optimizationRequestsCounter.add(1,r),!t&&s.optimizationErrorsCounter&&s.optimizationErrorsCounter.add(1,r)}catch(r){console.warn("Failed to record optimization metric:",r)}},ma=(s,e,t,n,o,r)=>{try{let i=Q({optimizer_type:r});s.convergenceRoundsHistogram&&s.convergenceRoundsHistogram.record(e,i),s.convergenceScoreGauge&&s.convergenceScoreGauge.record(t,i),s.convergenceImprovementGauge&&s.convergenceImprovementGauge.record(n,i),s.stagnationRoundsGauge&&s.stagnationRoundsGauge.record(o,i)}catch(i){console.warn("Failed to record convergence metric:",i)}},ga=(s,e,t)=>{try{let n=Q({reason:e,optimizer_type:t});s.earlyStoppingCounter&&s.earlyStoppingCounter.add(1,n)}catch(n){console.warn("Failed to record early stopping metric:",n)}},ha=(s,e,t,n,o)=>{try{let r=Q({optimizer_type:n});s.tokenUsageCounter&&s.tokenUsageCounter.add(e,r),s.costUsageCounter&&s.costUsageCounter.add(t,r),o!==void 0&&s.memoryUsageGauge&&s.memoryUsageGauge.record(o,r)}catch(r){console.warn("Failed to record resource usage metric:",r)}},Aa=(s,e,t)=>{try{let n=Q({optimizer_type:t});s.optimizationDurationHistogram&&s.optimizationDurationHistogram.record(e,n)}catch(n){console.warn("Failed to record optimization duration metric:",n)}},xa=(s,e,t,n)=>{try{let o=Q({optimizer_type:n});s.teacherStudentUsageCounter&&s.teacherStudentUsageCounter.add(1,o),s.teacherStudentLatencyHistogram&&s.teacherStudentLatencyHistogram.record(e,o),s.teacherStudentScoreImprovementGauge&&s.teacherStudentScoreImprovementGauge.record(t,o)}catch(o){console.warn("Failed to record teacher-student metric:",o)}},fa=(s,e,t,n,o)=>{try{let r=Q({operation:e,success:n.toString(),optimizer_type:o});e==="save"?(s.checkpointSaveCounter&&s.checkpointSaveCounter.add(1,r),s.checkpointSaveLatencyHistogram&&s.checkpointSaveLatencyHistogram.record(t,r)):(s.checkpointLoadCounter&&s.checkpointLoadCounter.add(1,r),s.checkpointLoadLatencyHistogram&&s.checkpointLoadLatencyHistogram.record(t,r))}catch(r){console.warn("Failed to record checkpoint metric:",r)}},ya=(s,e,t,n,o)=>{try{let r=Q({optimizer_type:n});s.paretoOptimizationsCounter&&s.paretoOptimizationsCounter.add(1,r),s.paretoFrontSizeHistogram&&s.paretoFrontSizeHistogram.record(e,r),o!==void 0&&s.paretoHypervolumeGauge&&s.paretoHypervolumeGauge.record(o,r),s.paretoSolutionsGeneratedHistogram&&s.paretoSolutionsGeneratedHistogram.record(t,r)}catch(r){console.warn("Failed to record Pareto metric:",r)}},Ia=(s,e,t,n,o,r)=>{try{let i=Q({optimizer_type:r});s.programInputFieldsGauge&&s.programInputFieldsGauge.record(e,i),s.programOutputFieldsGauge&&s.programOutputFieldsGauge.record(t,i),s.examplesCountGauge&&s.examplesCountGauge.record(n,i),s.validationSetSizeGauge&&s.validationSetSizeGauge.record(o,i)}catch(i){console.warn("Failed to record program complexity metric:",i)}},ba=(s,e,t,n)=>{try{let o=Q({metric_type:e,optimizer_type:n});switch(e){case"evaluation":s.evaluationLatencyHistogram&&s.evaluationLatencyHistogram.record(t,o);break;case"demo_generation":s.demoGenerationLatencyHistogram&&s.demoGenerationLatencyHistogram.record(t,o);break;case"metric_computation":s.metricComputationLatencyHistogram&&s.metricComputationLatencyHistogram.record(t,o);break}}catch(o){console.warn("Failed to record optimizer performance metric:",o)}},Ra=(s,e,t,n)=>{try{let o=Q({optimizer_type:e});s.optimizerTypeGauge&&s.optimizerTypeGauge.record(1,o),t!==void 0&&s.targetScoreGauge&&s.targetScoreGauge.record(t,o),n!==void 0&&s.maxRoundsGauge&&s.maxRoundsGauge.record(n,o)}catch(o){console.warn("Failed to record optimizer configuration metric:",o)}},Ct=class{tokenUsage={};totalTokens=0;costPerModel;maxCost;maxTokens;constructor(e){this.costPerModel=e?.costPerModel??{},this.maxCost=e?.maxCost,this.maxTokens=e?.maxTokens}trackTokens(e,t){this.tokenUsage[t]=(this.tokenUsage[t]||0)+e,this.totalTokens+=e}getCurrentCost(){let e=0;for(let[t,n]of Object.entries(this.tokenUsage)){let o=this.costPerModel[t]||.001;e+=n/1e3*o}return e}getTokenUsage(){return{...this.tokenUsage}}getTotalTokens(){return this.totalTokens}isLimitReached(){return this.maxTokens!==void 0&&this.totalTokens>=this.maxTokens||this.maxCost!==void 0&&this.getCurrentCost()>=this.maxCost}reset(){this.tokenUsage={},this.totalTokens=0}},ae=class{studentAI;teacherAI;examples;validationSet;targetScore;minSuccessRate;onProgress;onEarlyStop;costTracker;seed;checkpointSave;checkpointLoad;checkpointInterval;resumeFromCheckpoint;logger;verbose;debugOptimizer;optimizerLogger;currentRound=0;scoreHistory=[];configurationHistory=[];stats;metricsInstruments;constructor(e){if(e.examples.length===0)throw new Error("No examples found");this.studentAI=e.studentAI,this.teacherAI=e.teacherAI,this.examples=e.examples,this.validationSet=e.validationSet,this.targetScore=e.targetScore,this.minSuccessRate=e.minSuccessRate,this.onProgress=e.onProgress,this.onEarlyStop=e.onEarlyStop,this.seed=e.seed,this.checkpointSave=e.checkpointSave,this.checkpointLoad=e.checkpointLoad,this.checkpointInterval=e.checkpointInterval??10,this.resumeFromCheckpoint=e.resumeFromCheckpoint,this.logger=e.logger,this.verbose=e.verbose;let t=new Ct({maxTokens:1e6});this.costTracker=e.costTracker??t,this.metricsInstruments=ca(M.meter),this.stats=this.initializeStats(),this.debugOptimizer=e.debugOptimizer??!1,this.optimizerLogger=e.optimizerLogger}initializeStats(){return{totalCalls:0,successfulDemos:0,estimatedTokenUsage:0,earlyStopped:!1,resourceUsage:{totalTokens:0,totalTime:0,avgLatencyPerEval:0,costByModel:{}},convergenceInfo:{converged:!1,finalImprovement:0,stagnationRounds:0,convergenceThreshold:.01},bestScore:0,bestConfiguration:{}}}setupRandomSeed(){this.seed!==void 0&&(Math.random=(()=>{let e=this.seed;return()=>(e=(e*9301+49297)%233280,e/233280)})())}checkCostLimits(){return this.costTracker?.isLimitReached()??!1}checkTargetScore(e){return this.targetScore!==void 0&&e>=this.targetScore}updateResourceUsage(e,t=0){this.stats.resourceUsage.totalTime=Date.now()-e,this.stats.resourceUsage.totalTokens+=t,this.stats.totalCalls>0&&(this.stats.resourceUsage.avgLatencyPerEval=this.stats.resourceUsage.totalTime/this.stats.totalCalls)}triggerEarlyStopping(e,t){this.stats.earlyStopped=!0,this.stats.earlyStopping={bestScoreRound:t,patienceExhausted:e.includes("improvement"),reason:e},this.recordEarlyStoppingMetrics(e,"unknown"),this.onEarlyStop&&this.onEarlyStop(e,this.stats),this.getOptimizerLogger()?.({name:"EarlyStopping",value:{reason:e,finalScore:this.stats.bestScore??0,round:t}})}getValidationSet(e){return e?.overrideValidationSet||this.validationSet||this.examples.slice(0,Math.floor(this.examples.length*.2))}getAIService(e=!1,t){return e&&t?.overrideTeacherAI?t.overrideTeacherAI:e&&this.teacherAI?this.teacherAI:this.studentAI}hasTeacherAI(e){return e?.overrideTeacherAI!==void 0||this.teacherAI!==void 0}getTeacherOrStudentAI(e){return e?.overrideTeacherAI||this.teacherAI||this.studentAI}async executeWithTeacher(e,t=!0,n){let o=this.getAIService(t,n);return await e(o)}async*compileStream(e,t,n){let o=Date.now(),r=this.constructor.name,i=e.getSignature().toString();this.recordOptimizationStart(r,i);let a,l=(m,h,g,f,A,x,y,S={},I)=>{this.getOptimizerLogger(I)?.({name:"RoundProgress",value:{round:m,totalRounds:I?.maxIterations??0,currentScore:h,bestScore:x,configuration:g}}),this.updateOptimizationProgress(m,h,g,f,A,x,y,S,I)},p=(m,h)=>{a=m,this.triggerEarlyStopping(m,this.currentRound)},c=m=>{this.onProgress?.(m),l(m.round,m.currentScore,m.currentConfiguration||{},r,{},m.bestScore,m.bestConfiguration,m.convergenceInfo,n)},u=await this.compile(e,t,{...n,overrideOnProgress:c,overrideOnEarlyStop:p}),d=Date.now()-o;return this.recordOptimizationComplete(d,!0,r,i),a&&this.getLogger(n)?.({name:"Notification",id:"optimization_early_stop",value:`Optimization stopped early due to ${a}`}),{demos:u.demos,stats:u.stats,bestScore:u.bestScore,finalConfiguration:u.finalConfiguration,scoreHistory:u.scoreHistory,configurationHistory:u.configurationHistory}}async compilePareto(e,t,n){let o=this.constructor.name,r=Date.now(),i=await this.generateWeightedSolutions(e,t,n),a=await this.generateConstraintSolutions(e,t,n),l=[...i,...a],p=this.findParetoFrontier(l),c=this.calculateHypervolume(p);this.updateResourceUsage(r),this.stats.convergenceInfo.converged=!0,this.recordParetoMetrics(p.length,l.length,"base_optimizer",c);let u=p.length>0?Math.max(...p.map(d=>Math.max(...Object.values(d.scores)))):0;return{demos:p.length>0?[...p[0].demos]:void 0,stats:this.stats,bestScore:u,paretoFront:p,hypervolume:c,paretoFrontSize:p.length,finalConfiguration:{paretoFrontSize:p.length,hypervolume:c,strategy:"weighted_combinations_and_constraints",numSolutions:l.length}}}async generateWeightedSolutions(e,t,n){let o=[],r=this.examples[0],i=await e.forward(this.getAIService(!1,n),r),a=await t({prediction:i,example:r}),l=Object.keys(a),p=this.generateWeightCombinations(l);for(let c=0;c<p.length;c++){let u=p[c],d=async({prediction:m,example:h})=>{let g=await t({prediction:m,example:h}),f=0;for(let[A,x]of Object.entries(g))f+=x*(u[A]||0);return f};try{let m=await this.compile(e,d,{...n,verbose:!1}),h=await this.evaluateWithMultiObjective(e,m,t);o.push({scores:h,demos:m.demos,configuration:{...m.finalConfiguration,weights:u,strategy:"weighted_combination"}})}catch{}}return o}async generateConstraintSolutions(e,t,n){let o=[],r=this.examples[0],i=await e.forward(this.getAIService(!1,n),r),a=await t({prediction:i,example:r}),l=Object.keys(a);for(let p of l){let c=async({prediction:u,example:d})=>{let m=await t({prediction:u,example:d}),h=m[p]||0,g=0;for(let[f,A]of Object.entries(m))f!==p&&A<.3&&(g+=(.3-A)*2);return h-g};try{let u=await this.compile(e,c,{...n,verbose:!1}),d=await this.evaluateWithMultiObjective(e,u,t);o.push({scores:d,demos:u.demos,configuration:{...u.finalConfiguration,primaryObjective:p,strategy:"constraint_based"}})}catch{}}return o}generateWeightCombinations(e){let t=[];for(let o of e){let r={};for(let i of e)r[i]=i===o?1:0;t.push(r)}let n={};for(let o of e)n[o]=1/e.length;if(t.push(n),e.length===2){let[o,r]=e;for(let i=.1;i<=.9;i+=.2){let a=1-i;t.push({[o]:i,[r]:a})}}if(e.length===3){let[o,r,i]=e;t.push({[o]:.5,[r]:.3,[i]:.2},{[o]:.3,[r]:.5,[i]:.2},{[o]:.2,[r]:.3,[i]:.5})}return t}async evaluateWithMultiObjective(e,t,n){let o=new N(e.getSignature());t.demos&&o.setDemos(t.demos);let r=[];for(let c of this.examples){let u=await o.forward(this.studentAI,c);r.push({prediction:u,example:c})}let i=this.getValidationSet(),a={},l=i.slice(0,Math.min(5,i.length));for(let c of l)try{let u=await o.forward(this.studentAI,c),d=await n({prediction:u,example:c});for(let[m,h]of Object.entries(d))a[m]||(a[m]=[]),a[m].push(h)}catch{}let p={};for(let[c,u]of Object.entries(a))p[c]=u.length>0?u.reduce((d,m)=>d+m,0)/u.length:0;return p}findParetoFrontier(e){let t=[];for(let n=0;n<e.length;n++){let o=e[n],r=!1,i=0;for(let a=0;a<e.length;a++){if(n===a)continue;let l=e[a];if(this.dominates(l.scores,o.scores)){r=!0;break}this.dominates(o.scores,l.scores)&&i++}r||t.push({demos:o.demos||[],scores:o.scores,configuration:o.configuration,dominatedSolutions:i})}return t}dominates(e,t){let n=Object.keys(e),o=!0,r=!1;for(let i of n){let a=e[i]||0,l=t[i]||0;if(a<l){o=!1;break}a>l&&(r=!0)}return o&&r}calculateHypervolume(e){if(e.length===0)return;let t=e[0],n=Object.keys(t.scores);if(n.length===2){let[o,r]=n,i=0,a=[...e].sort((p,c)=>(c.scores[o]||0)-(p.scores[o]||0)),l=0;for(let p of a){let c=p.scores[o]||0,u=p.scores[r]||0;i+=c*(u-l),l=Math.max(l,u)}return i}}async saveCheckpoint(e,t,n,o,r={},i){let a=i?.overrideCheckpointSave||this.checkpointSave;if(!a)return;let l=Date.now(),p=!1,c;try{let u={version:"1.0.0",timestamp:Date.now(),optimizerType:e,optimizerConfig:t,currentRound:this.currentRound,totalRounds:this.stats.resourceUsage.totalTime>0?this.currentRound:0,bestScore:n,bestConfiguration:o,scoreHistory:[...this.scoreHistory],configurationHistory:[...this.configurationHistory],stats:{...this.stats},optimizerState:r,examples:this.examples,validationSet:this.validationSet};c=await a(u),p=!0}catch(u){throw p=!1,u}finally{let u=Date.now()-l;this.recordCheckpointMetrics("save",u,p,e)}return c}async loadCheckpoint(e,t){let n=t?.overrideCheckpointLoad||this.checkpointLoad;if(!n)return null;let o=Date.now(),r=!1,i=null;try{i=await n(e),r=i!==null}catch(a){throw r=!1,a}finally{let a=Date.now()-o;this.recordCheckpointMetrics("load",a,r,"unknown")}return i}restoreFromCheckpoint(e){this.currentRound=e.currentRound,this.scoreHistory=[...e.scoreHistory],this.configurationHistory=[...e.configurationHistory],this.stats={...e.stats}}shouldSaveCheckpoint(e,t){let n=t?.overrideCheckpointInterval||this.checkpointInterval;return n!==void 0&&e%n===0}async updateOptimizationProgress(e,t,n,o,r,i,a,l={},p){this.currentRound=e,this.scoreHistory.push(t),this.configurationHistory.push(n),this.shouldSaveCheckpoint(e,p)&&await this.saveCheckpoint(o,r,i,a,l,p),this.getOptimizerLogger(p)?.({name:"RoundProgress",value:{round:e,totalRounds:p?.maxIterations??0,currentScore:t,bestScore:i,configuration:n}})}async saveFinalCheckpoint(e,t,n,o,r={},i){i?.saveCheckpointOnComplete!==!1&&await this.saveCheckpoint(e,t,n,o,{...r,final:!0},i)}getLogger(e){if(this.isLoggingEnabled(e))return this.logger?this.logger:this.studentAI.getLogger()}isLoggingEnabled(e){return e?.verbose!==void 0?e.verbose:this.verbose??!0}recordOptimizationStart(e,t){if(this.metricsInstruments){if(t){let n=(t.match(/input:/g)||[]).length,o=(t.match(/output:/g)||[]).length;Ia(this.metricsInstruments,n,o,this.examples.length,this.getValidationSet().length,e)}Ra(this.metricsInstruments,e,this.targetScore,void 0)}}recordOptimizationComplete(e,t,n,o){if(!this.metricsInstruments)return;da(this.metricsInstruments,e,t,n,o),Aa(this.metricsInstruments,e,n);let r=this.costTracker?.getCurrentCost()??0,i=this.costTracker?.getTotalTokens()??0;ha(this.metricsInstruments,i,r,n)}recordConvergenceMetrics(e,t,n,o,r){this.metricsInstruments&&ma(this.metricsInstruments,e,t,n,o,r)}recordEarlyStoppingMetrics(e,t){this.metricsInstruments&&ga(this.metricsInstruments,e,t)}recordTeacherStudentMetrics(e,t,n){this.metricsInstruments&&xa(this.metricsInstruments,e,t,n)}recordCheckpointMetrics(e,t,n,o){this.metricsInstruments&&fa(this.metricsInstruments,e,t,n,o)}recordParetoMetrics(e,t,n,o){this.metricsInstruments&&ya(this.metricsInstruments,e,t,n,o)}recordPerformanceMetrics(e,t,n){this.metricsInstruments&&ba(this.metricsInstruments,e,t,n)}isOptimizerLoggingEnabled(e){return this.debugOptimizer||(e?.verbose??this.verbose??!1)}getOptimizerLogger(e){if(this.isOptimizerLoggingEnabled(e))return this.optimizerLogger??M.optimizerLogger??En}getStats(){return{...this.stats}}reset(){this.stats=this.initializeStats(),this.costTracker?.reset(),this.currentRound=0,this.scoreHistory=[],this.configurationHistory=[]}};var ze=class extends ae{maxRounds;maxDemos;maxExamples;batchSize;earlyStoppingPatience;costMonitoring;maxTokensPerGeneration;verboseMode;debugMode;traces=[];constructor(e){super(e);let t=e.options||{};this.maxRounds=t.maxRounds??3,this.maxDemos=t.maxDemos??4,this.maxExamples=t.maxExamples??16,this.batchSize=t.batchSize??1,this.earlyStoppingPatience=t.earlyStoppingPatience??0,this.costMonitoring=t.costMonitoring??!1,this.maxTokensPerGeneration=t.maxTokensPerGeneration??0,this.verboseMode=t.verboseMode??!0,this.debugMode=t.debugMode??!1}async compileRound(e,t,n,o){let r=Date.now(),i=o?.maxDemos??this.maxDemos,a={modelConfig:{temperature:.7}};this.maxTokensPerGeneration>0&&(a.modelConfig.max_tokens=this.maxTokensPerGeneration);let l=Ta(this.examples,this.maxExamples),p=this.traces.length;for(let c=0;c<l.length;c+=this.batchSize){c>0&&(a.modelConfig.temperature=.7+.001*c);let u=l.slice(c,c+this.batchSize);for(let d of u){if(!d)continue;let m=l.filter(f=>f!==d);e.setExamples(m);let h=this.getTeacherOrStudentAI();this.stats.totalCalls++;let g;try{let f={...a,maxRetries:1};g=await e.forward(h,d,f),this.costMonitoring&&(this.stats.estimatedTokenUsage+=JSON.stringify(d).length/4+JSON.stringify(g).length/4),await n({prediction:g,example:d})>=.5&&(this.traces=[...this.traces,...e.getTraces()],this.stats.successfulDemos++)}catch(f){(this.verboseMode||this.debugMode)&&console.warn(`Student model failed during bootstrap: ${f instanceof Error?f.message:"Unknown error"}`),g={}}if(this.traces.length>=i)return}}if(this.earlyStoppingPatience>0){let u=this.traces.length-p;if(!this.stats.earlyStopping)this.stats.earlyStopping={bestScoreRound:u>0?t:0,patienceExhausted:!1,reason:"No improvement detected"};else if(u>0)this.stats.earlyStopping.bestScoreRound=t;else if(t-this.stats.earlyStopping.bestScoreRound>=this.earlyStoppingPatience){this.stats.earlyStopping.patienceExhausted=!0,this.stats.earlyStopped=!0,this.stats.earlyStopping.reason=`No improvement for ${this.earlyStoppingPatience} rounds`;return}}}async compile(e,t,n){let o=n?.maxIterations??this.maxRounds;this.traces=[],this.reset();for(let a=0;a<o&&(await this.compileRound(e,a,t,n),!this.stats.earlyStopped);a++);if(this.traces.length===0)throw new Error("No demonstrations found. Either provide more examples or improve the existing ones.");let r=Ca(this.traces),i=0;return this.traces.length>0&&(i=this.stats.successfulDemos/Math.max(1,this.stats.totalCalls)),{demos:r,stats:this.stats,bestScore:i,finalConfiguration:{maxRounds:this.maxRounds,maxDemos:this.maxDemos,batchSize:this.batchSize,successRate:i}}}};function Ca(s){let e=new Map;for(let n of s)if(e.has(n.programId)){let o=e.get(n.programId);o&&o.push(n.trace)}else e.set(n.programId,[n.trace]);let t=[];return e.forEach((n,o)=>{t.push({traces:n,programId:o})}),t}var Ta=(s,e)=>{let t=[...s];for(let n=t.length-1;n>0;n--){let o=Math.floor(Math.random()*(n+1)),r=t[n],i=t[o];if(!r||!i)throw new Error("Invalid array elements");[t[n],t[o]]=[i,r]}return t.slice(0,e)};var _n=class{endpoint;timeout;retryAttempts;retryDelay;logger;constructor(e){this.endpoint=e.endpoint.replace(/\/$/,""),this.timeout=e.timeout??3e4,this.retryAttempts=e.retryAttempts??3,this.retryDelay=e.retryDelay??1e3,this.logger=e.logger}async healthCheck(){try{return(await this.fetchWithRetry("/health",{method:"GET"})).ok}catch(e){return this.logger?.({name:"Notification",id:"health_check_failed",value:`Health check failed: ${e}`}),!1}}async createOptimizationJob(e){let t=await this.fetchWithRetry("/optimize",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)});if(!t.ok){let n=await t.text();throw new Error(`Failed to create optimization job: ${n}`)}return t.json()}async getJobStatus(e){let t=await this.fetchWithRetry(`/jobs/${e}`,{method:"GET"});if(!t.ok){let n=await t.text();throw new Error(`Failed to get job status: ${n}`)}return t.json()}async cancelJob(e){let t=await this.fetchWithRetry(`/jobs/${e}`,{method:"DELETE"});if(!t.ok){let n=await t.text();throw new Error(`Failed to cancel job: ${n}`)}}async suggestParameters(e){let t=await this.fetchWithRetry(`/studies/${e}/suggest`,{method:"POST",headers:{"Content-Type":"application/json"}});if(!t.ok){let n=await t.text();throw new Error(`Failed to suggest parameters: ${n}`)}return t.json()}async evaluateTrial(e){let t=await this.fetchWithRetry(`/studies/${e.study_name}/evaluate`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)});if(!t.ok){let n=await t.text();throw new Error(`Failed to evaluate trial: ${n}`)}}async getStudyResults(e){let t=await this.fetchWithRetry(`/studies/${e}/results`,{method:"GET"});if(!t.ok){let n=await t.text();throw new Error(`Failed to get study results: ${n}`)}return t.json()}async deleteStudy(e){let t=await this.fetchWithRetry(`/studies/${e}`,{method:"DELETE"});if(!t.ok){let n=await t.text();throw new Error(`Failed to delete study: ${n}`)}}async listStudies(){let e=await this.fetchWithRetry("/studies",{method:"GET"});if(!e.ok){let t=await e.text();throw new Error(`Failed to list studies: ${t}`)}return e.json()}async waitForJobCompletion(e,t=2e3,n=3e5){let o=Date.now();for(;Date.now()-o<n;){let r=await this.getJobStatus(e);if(["completed","failed","cancelled"].includes(r.status))return r;this.logger?.({name:"Notification",id:"job_status",value:`Job ${e} status: ${r.status}, waiting...`}),await this.sleep(t)}throw new Error(`Job ${e} did not complete within ${n}ms`)}async fetchWithRetry(e,t){let n=`${this.endpoint}${e}`,o=null;for(let r=0;r<this.retryAttempts;r++)try{let i=new AbortController,a=setTimeout(()=>i.abort(),this.timeout),l=await fetch(n,{...t,signal:i.signal});return clearTimeout(a),l}catch(i){o=i,this.logger?.({name:"Notification",id:"retry_attempt",value:`Attempt ${r+1} failed: ${i}`}),r<this.retryAttempts-1&&await this.sleep(this.retryDelay*Math.pow(2,r))}throw new Error(`Request failed after ${this.retryAttempts} attempts: ${o?.message}`)}sleep(e){return new Promise(t=>setTimeout(t,e))}};var Dn=class extends ae{maxBootstrappedDemos;maxLabeledDemos;numCandidates;initTemperature;numTrials;minibatch;minibatchSize;minibatchFullEvalSteps;programAwareProposer;dataAwareProposer;viewDataBatchSize;tipAwareProposer;fewshotAwareProposer;earlyStoppingTrials;minImprovementThreshold;bayesianOptimization;acquisitionFunction;explorationWeight;sampleCount;miproConfigHistory=[];surrogateModel=new Map;pythonClient;constructor(e){if(super(e),this.numCandidates=e.numCandidates??5,this.initTemperature=e.initTemperature??.7,this.maxBootstrappedDemos=e.maxBootstrappedDemos??3,this.maxLabeledDemos=e.maxLabeledDemos??4,this.numTrials=e.numTrials??30,this.minibatch=e.minibatch??!0,this.minibatchSize=e.minibatchSize??25,this.minibatchFullEvalSteps=e.minibatchFullEvalSteps??10,this.programAwareProposer=e.programAwareProposer??!0,this.dataAwareProposer=e.dataAwareProposer??!0,this.viewDataBatchSize=e.viewDataBatchSize??10,this.tipAwareProposer=e.tipAwareProposer??!0,this.fewshotAwareProposer=e.fewshotAwareProposer??!0,this.earlyStoppingTrials=e.earlyStoppingTrials??5,this.minImprovementThreshold=e.minImprovementThreshold??.01,this.bayesianOptimization=e.bayesianOptimization??!1,this.acquisitionFunction=e.acquisitionFunction??"expected_improvement",this.explorationWeight=e.explorationWeight??.1,this.sampleCount=e.sampleCount??1,e.optimizerEndpoint){let t={endpoint:e.optimizerEndpoint,timeout:e.optimizerTimeout??3e4,retryAttempts:e.optimizerRetries??3,logger:n=>{this.logger?.({name:"Notification",id:"python_client",value:typeof n=="string"?n:JSON.stringify(n)})}};this.pythonClient=new _n(t)}this.stats.convergenceInfo.convergenceThreshold=this.minImprovementThreshold}configureAuto(e){switch(e){case"light":this.numCandidates=3,this.numTrials=10,this.minibatch=!0,this.minibatchSize=20;break;case"medium":this.numCandidates=5,this.numTrials=20,this.minibatch=!0,this.minibatchSize=25;break;case"heavy":this.numCandidates=7,this.numTrials=30,this.minibatch=!0,this.minibatchSize=30;break}}generateTips(){return["Be very specific and detailed in your instructions.","Focus on step-by-step reasoning in your instructions.","Provide clear constraints and guidelines in your instructions.","Keep your instructions concise and to the point.","Emphasize accuracy and precision in your instructions.","Include examples of good outputs in your instructions.","Focus on handling edge cases in your instructions.","Explicitly outline the reasoning process in your instructions."]}async generateProgramSummary(e,t){let o=`
251
+ ${e}`}s(n)}},Fn=Vr();var Wr={enabled:!0,enabledCategories:["optimization","convergence","resource_usage","teacher_student","checkpointing","pareto"],maxLabelLength:100,samplingRate:1},_n,da=s=>{if(_n)return _n;if(s)return _n=ma(s),_n};var Dn=Wr,Ys=s=>{Dn={...Dn,...s}},Xs=()=>({...Dn}),ma=s=>({optimizationLatencyHistogram:s.createHistogram("ax_optimizer_optimization_duration_ms",{description:"End-to-end duration of optimization runs",unit:"ms"}),optimizationRequestsCounter:s.createCounter("ax_optimizer_optimization_requests_total",{description:"Total number of optimization requests"}),optimizationErrorsCounter:s.createCounter("ax_optimizer_optimization_errors_total",{description:"Total number of failed optimizations"}),convergenceRoundsHistogram:s.createHistogram("ax_optimizer_convergence_rounds",{description:"Number of rounds until convergence"}),convergenceScoreGauge:s.createGauge("ax_optimizer_convergence_score",{description:"Current best score during optimization"}),convergenceImprovementGauge:s.createGauge("ax_optimizer_convergence_improvement",{description:"Improvement in score from baseline"}),stagnationRoundsGauge:s.createGauge("ax_optimizer_stagnation_rounds",{description:"Number of rounds without improvement"}),earlyStoppingCounter:s.createCounter("ax_optimizer_early_stopping_total",{description:"Total number of early stopping events"}),tokenUsageCounter:s.createCounter("ax_optimizer_token_usage_total",{description:"Total tokens used during optimization"}),costUsageCounter:s.createCounter("ax_optimizer_cost_usage_total",{description:"Total cost incurred during optimization",unit:"$"}),memoryUsageGauge:s.createGauge("ax_optimizer_memory_usage_bytes",{description:"Peak memory usage during optimization",unit:"By"}),optimizationDurationHistogram:s.createHistogram("ax_optimizer_duration_ms",{description:"Duration of optimization runs",unit:"ms"}),teacherStudentUsageCounter:s.createCounter("ax_optimizer_teacher_student_usage_total",{description:"Total number of teacher-student interactions"}),teacherStudentLatencyHistogram:s.createHistogram("ax_optimizer_teacher_student_latency_ms",{description:"Latency of teacher-student interactions",unit:"ms"}),teacherStudentScoreImprovementGauge:s.createGauge("ax_optimizer_teacher_student_score_improvement",{description:"Score improvement from teacher-student interactions"}),checkpointSaveCounter:s.createCounter("ax_optimizer_checkpoint_save_total",{description:"Total number of checkpoint saves"}),checkpointLoadCounter:s.createCounter("ax_optimizer_checkpoint_load_total",{description:"Total number of checkpoint loads"}),checkpointSaveLatencyHistogram:s.createHistogram("ax_optimizer_checkpoint_save_latency_ms",{description:"Latency of checkpoint save operations",unit:"ms"}),checkpointLoadLatencyHistogram:s.createHistogram("ax_optimizer_checkpoint_load_latency_ms",{description:"Latency of checkpoint load operations",unit:"ms"}),paretoOptimizationsCounter:s.createCounter("ax_optimizer_pareto_optimizations_total",{description:"Total number of Pareto optimizations"}),paretoFrontSizeHistogram:s.createHistogram("ax_optimizer_pareto_front_size",{description:"Size of Pareto frontier"}),paretoHypervolumeGauge:s.createGauge("ax_optimizer_pareto_hypervolume",{description:"Hypervolume of Pareto frontier"}),paretoSolutionsGeneratedHistogram:s.createHistogram("ax_optimizer_pareto_solutions_generated",{description:"Number of solutions generated for Pareto optimization"}),programInputFieldsGauge:s.createGauge("ax_optimizer_program_input_fields",{description:"Number of input fields in optimized program"}),programOutputFieldsGauge:s.createGauge("ax_optimizer_program_output_fields",{description:"Number of output fields in optimized program"}),examplesCountGauge:s.createGauge("ax_optimizer_examples_count",{description:"Number of training examples used"}),validationSetSizeGauge:s.createGauge("ax_optimizer_validation_set_size",{description:"Size of validation set used"}),evaluationLatencyHistogram:s.createHistogram("ax_optimizer_evaluation_latency_ms",{description:"Latency of program evaluations",unit:"ms"}),demoGenerationLatencyHistogram:s.createHistogram("ax_optimizer_demo_generation_latency_ms",{description:"Latency of demo generation",unit:"ms"}),metricComputationLatencyHistogram:s.createHistogram("ax_optimizer_metric_computation_latency_ms",{description:"Latency of metric computation",unit:"ms"}),optimizerTypeGauge:s.createGauge("ax_optimizer_type",{description:"Type of optimizer being used"}),targetScoreGauge:s.createGauge("ax_optimizer_target_score",{description:"Target score for optimization"}),maxRoundsGauge:s.createGauge("ax_optimizer_max_rounds",{description:"Maximum rounds for optimization"})}),Q=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let r=String(n),o=Dn.maxLabelLength;e[t]=r.length>o?r.substring(0,o):r}return e},ga=(s,e,t,n,r)=>{try{let o=Q({success:t.toString(),optimizer_type:n,...r?{program_signature:r}:{}});s.optimizationLatencyHistogram&&s.optimizationLatencyHistogram.record(e,o),s.optimizationRequestsCounter&&s.optimizationRequestsCounter.add(1,o),!t&&s.optimizationErrorsCounter&&s.optimizationErrorsCounter.add(1,o)}catch(o){console.warn("Failed to record optimization metric:",o)}},ha=(s,e,t,n,r,o)=>{try{let i=Q({optimizer_type:o});s.convergenceRoundsHistogram&&s.convergenceRoundsHistogram.record(e,i),s.convergenceScoreGauge&&s.convergenceScoreGauge.record(t,i),s.convergenceImprovementGauge&&s.convergenceImprovementGauge.record(n,i),s.stagnationRoundsGauge&&s.stagnationRoundsGauge.record(r,i)}catch(i){console.warn("Failed to record convergence metric:",i)}},Aa=(s,e,t)=>{try{let n=Q({reason:e,optimizer_type:t});s.earlyStoppingCounter&&s.earlyStoppingCounter.add(1,n)}catch(n){console.warn("Failed to record early stopping metric:",n)}},xa=(s,e,t,n,r)=>{try{let o=Q({optimizer_type:n});s.tokenUsageCounter&&s.tokenUsageCounter.add(e,o),s.costUsageCounter&&s.costUsageCounter.add(t,o),r!==void 0&&s.memoryUsageGauge&&s.memoryUsageGauge.record(r,o)}catch(o){console.warn("Failed to record resource usage metric:",o)}},fa=(s,e,t)=>{try{let n=Q({optimizer_type:t});s.optimizationDurationHistogram&&s.optimizationDurationHistogram.record(e,n)}catch(n){console.warn("Failed to record optimization duration metric:",n)}},ya=(s,e,t,n)=>{try{let r=Q({optimizer_type:n});s.teacherStudentUsageCounter&&s.teacherStudentUsageCounter.add(1,r),s.teacherStudentLatencyHistogram&&s.teacherStudentLatencyHistogram.record(e,r),s.teacherStudentScoreImprovementGauge&&s.teacherStudentScoreImprovementGauge.record(t,r)}catch(r){console.warn("Failed to record teacher-student metric:",r)}},Ia=(s,e,t,n,r)=>{try{let o=Q({operation:e,success:n.toString(),optimizer_type:r});e==="save"?(s.checkpointSaveCounter&&s.checkpointSaveCounter.add(1,o),s.checkpointSaveLatencyHistogram&&s.checkpointSaveLatencyHistogram.record(t,o)):(s.checkpointLoadCounter&&s.checkpointLoadCounter.add(1,o),s.checkpointLoadLatencyHistogram&&s.checkpointLoadLatencyHistogram.record(t,o))}catch(o){console.warn("Failed to record checkpoint metric:",o)}},ba=(s,e,t,n,r)=>{try{let o=Q({optimizer_type:n});s.paretoOptimizationsCounter&&s.paretoOptimizationsCounter.add(1,o),s.paretoFrontSizeHistogram&&s.paretoFrontSizeHistogram.record(e,o),r!==void 0&&s.paretoHypervolumeGauge&&s.paretoHypervolumeGauge.record(r,o),s.paretoSolutionsGeneratedHistogram&&s.paretoSolutionsGeneratedHistogram.record(t,o)}catch(o){console.warn("Failed to record Pareto metric:",o)}},Ra=(s,e,t,n,r,o)=>{try{let i=Q({optimizer_type:o});s.programInputFieldsGauge&&s.programInputFieldsGauge.record(e,i),s.programOutputFieldsGauge&&s.programOutputFieldsGauge.record(t,i),s.examplesCountGauge&&s.examplesCountGauge.record(n,i),s.validationSetSizeGauge&&s.validationSetSizeGauge.record(r,i)}catch(i){console.warn("Failed to record program complexity metric:",i)}},Ta=(s,e,t,n)=>{try{let r=Q({metric_type:e,optimizer_type:n});switch(e){case"evaluation":s.evaluationLatencyHistogram&&s.evaluationLatencyHistogram.record(t,r);break;case"demo_generation":s.demoGenerationLatencyHistogram&&s.demoGenerationLatencyHistogram.record(t,r);break;case"metric_computation":s.metricComputationLatencyHistogram&&s.metricComputationLatencyHistogram.record(t,r);break}}catch(r){console.warn("Failed to record optimizer performance metric:",r)}},Ca=(s,e,t,n)=>{try{let r=Q({optimizer_type:e});s.optimizerTypeGauge&&s.optimizerTypeGauge.record(1,r),t!==void 0&&s.targetScoreGauge&&s.targetScoreGauge.record(t,r),n!==void 0&&s.maxRoundsGauge&&s.maxRoundsGauge.record(n,r)}catch(r){console.warn("Failed to record optimizer configuration metric:",r)}},Ct=class{tokenUsage={};totalTokens=0;costPerModel;maxCost;maxTokens;constructor(e){this.costPerModel=e?.costPerModel??{},this.maxCost=e?.maxCost,this.maxTokens=e?.maxTokens}trackTokens(e,t){this.tokenUsage[t]=(this.tokenUsage[t]||0)+e,this.totalTokens+=e}getCurrentCost(){let e=0;for(let[t,n]of Object.entries(this.tokenUsage)){let r=this.costPerModel[t]||.001;e+=n/1e3*r}return e}getTokenUsage(){return{...this.tokenUsage}}getTotalTokens(){return this.totalTokens}isLimitReached(){return this.maxTokens!==void 0&&this.totalTokens>=this.maxTokens||this.maxCost!==void 0&&this.getCurrentCost()>=this.maxCost}reset(){this.tokenUsage={},this.totalTokens=0}},ae=class{studentAI;teacherAI;examples;validationSet;targetScore;minSuccessRate;onProgress;onEarlyStop;costTracker;seed;checkpointSave;checkpointLoad;checkpointInterval;resumeFromCheckpoint;logger;verbose;debugOptimizer;optimizerLogger;currentRound=0;scoreHistory=[];configurationHistory=[];stats;metricsInstruments;constructor(e){if(e.examples.length===0)throw new Error("No examples found");this.studentAI=e.studentAI,this.teacherAI=e.teacherAI,this.examples=e.examples,this.validationSet=e.validationSet,this.targetScore=e.targetScore,this.minSuccessRate=e.minSuccessRate,this.onProgress=e.onProgress,this.onEarlyStop=e.onEarlyStop,this.seed=e.seed,this.checkpointSave=e.checkpointSave,this.checkpointLoad=e.checkpointLoad,this.checkpointInterval=e.checkpointInterval??10,this.resumeFromCheckpoint=e.resumeFromCheckpoint,this.logger=e.logger,this.verbose=e.verbose;let t=new Ct({maxTokens:1e6});this.costTracker=e.costTracker??t,this.metricsInstruments=da(M.meter),this.stats=this.initializeStats(),this.debugOptimizer=e.debugOptimizer??!1,this.optimizerLogger=e.optimizerLogger}initializeStats(){return{totalCalls:0,successfulDemos:0,estimatedTokenUsage:0,earlyStopped:!1,resourceUsage:{totalTokens:0,totalTime:0,avgLatencyPerEval:0,costByModel:{}},convergenceInfo:{converged:!1,finalImprovement:0,stagnationRounds:0,convergenceThreshold:.01},bestScore:0,bestConfiguration:{}}}setupRandomSeed(){this.seed!==void 0&&(Math.random=(()=>{let e=this.seed;return()=>(e=(e*9301+49297)%233280,e/233280)})())}checkCostLimits(){return this.costTracker?.isLimitReached()??!1}checkTargetScore(e){return this.targetScore!==void 0&&e>=this.targetScore}updateResourceUsage(e,t=0){this.stats.resourceUsage.totalTime=Date.now()-e,this.stats.resourceUsage.totalTokens+=t,this.stats.totalCalls>0&&(this.stats.resourceUsage.avgLatencyPerEval=this.stats.resourceUsage.totalTime/this.stats.totalCalls)}triggerEarlyStopping(e,t){this.stats.earlyStopped=!0,this.stats.earlyStopping={bestScoreRound:t,patienceExhausted:e.includes("improvement"),reason:e},this.recordEarlyStoppingMetrics(e,"unknown"),this.onEarlyStop&&this.onEarlyStop(e,this.stats),this.getOptimizerLogger()?.({name:"EarlyStopping",value:{reason:e,finalScore:this.stats.bestScore??0,round:t}})}getValidationSet(e){return e?.overrideValidationSet||this.validationSet||this.examples.slice(0,Math.floor(this.examples.length*.2))}getAIService(e=!1,t){return e&&t?.overrideTeacherAI?t.overrideTeacherAI:e&&this.teacherAI?this.teacherAI:this.studentAI}hasTeacherAI(e){return e?.overrideTeacherAI!==void 0||this.teacherAI!==void 0}getTeacherOrStudentAI(e){return e?.overrideTeacherAI||this.teacherAI||this.studentAI}async executeWithTeacher(e,t=!0,n){let r=this.getAIService(t,n);return await e(r)}async*compileStream(e,t,n){let r=Date.now(),o=this.constructor.name,i=e.getSignature().toString();this.recordOptimizationStart(o,i);let a,l=(m,h,g,f,A,x,y,w={},I)=>{this.getOptimizerLogger(I)?.({name:"RoundProgress",value:{round:m,totalRounds:I?.maxIterations??0,currentScore:h,bestScore:x,configuration:g}}),this.updateOptimizationProgress(m,h,g,f,A,x,y,w,I)},p=(m,h)=>{a=m,this.triggerEarlyStopping(m,this.currentRound)},c=m=>{this.onProgress?.(m),l(m.round,m.currentScore,m.currentConfiguration||{},o,{},m.bestScore,m.bestConfiguration,m.convergenceInfo,n)},u=await this.compile(e,t,{...n,overrideOnProgress:c,overrideOnEarlyStop:p}),d=Date.now()-r;return this.recordOptimizationComplete(d,!0,o,i),a&&this.getLogger(n)?.({name:"Notification",id:"optimization_early_stop",value:`Optimization stopped early due to ${a}`}),{demos:u.demos,stats:u.stats,bestScore:u.bestScore,finalConfiguration:u.finalConfiguration,scoreHistory:u.scoreHistory,configurationHistory:u.configurationHistory}}async compilePareto(e,t,n){let r=this.constructor.name,o=Date.now(),i=await this.generateWeightedSolutions(e,t,n),a=await this.generateConstraintSolutions(e,t,n),l=[...i,...a],p=this.findParetoFrontier(l),c=this.calculateHypervolume(p);this.updateResourceUsage(o),this.stats.convergenceInfo.converged=!0,this.recordParetoMetrics(p.length,l.length,"base_optimizer",c);let u=p.length>0?Math.max(...p.map(d=>Math.max(...Object.values(d.scores)))):0;return{demos:p.length>0?[...p[0].demos]:void 0,stats:this.stats,bestScore:u,paretoFront:p,hypervolume:c,paretoFrontSize:p.length,finalConfiguration:{paretoFrontSize:p.length,hypervolume:c,strategy:"weighted_combinations_and_constraints",numSolutions:l.length}}}async generateWeightedSolutions(e,t,n){let r=[],o=this.examples[0],i=await e.forward(this.getAIService(!1,n),o),a=await t({prediction:i,example:o}),l=Object.keys(a),p=this.generateWeightCombinations(l);for(let c=0;c<p.length;c++){let u=p[c],d=async({prediction:m,example:h})=>{let g=await t({prediction:m,example:h}),f=0;for(let[A,x]of Object.entries(g))f+=x*(u[A]||0);return f};try{let m=await this.compile(e,d,{...n,verbose:!1}),h=await this.evaluateWithMultiObjective(e,m,t);r.push({scores:h,demos:m.demos,configuration:{...m.finalConfiguration,weights:u,strategy:"weighted_combination"}})}catch{}}return r}async generateConstraintSolutions(e,t,n){let r=[],o=this.examples[0],i=await e.forward(this.getAIService(!1,n),o),a=await t({prediction:i,example:o}),l=Object.keys(a);for(let p of l){let c=async({prediction:u,example:d})=>{let m=await t({prediction:u,example:d}),h=m[p]||0,g=0;for(let[f,A]of Object.entries(m))f!==p&&A<.3&&(g+=(.3-A)*2);return h-g};try{let u=await this.compile(e,c,{...n,verbose:!1}),d=await this.evaluateWithMultiObjective(e,u,t);r.push({scores:d,demos:u.demos,configuration:{...u.finalConfiguration,primaryObjective:p,strategy:"constraint_based"}})}catch{}}return r}generateWeightCombinations(e){let t=[];for(let r of e){let o={};for(let i of e)o[i]=i===r?1:0;t.push(o)}let n={};for(let r of e)n[r]=1/e.length;if(t.push(n),e.length===2){let[r,o]=e;for(let i=.1;i<=.9;i+=.2){let a=1-i;t.push({[r]:i,[o]:a})}}if(e.length===3){let[r,o,i]=e;t.push({[r]:.5,[o]:.3,[i]:.2},{[r]:.3,[o]:.5,[i]:.2},{[r]:.2,[o]:.3,[i]:.5})}return t}async evaluateWithMultiObjective(e,t,n){let r=new $(e.getSignature());t.demos&&r.setDemos(t.demos);let o=[];for(let c of this.examples){let u=await r.forward(this.studentAI,c);o.push({prediction:u,example:c})}let i=this.getValidationSet(),a={},l=i.slice(0,Math.min(5,i.length));for(let c of l)try{let u=await r.forward(this.studentAI,c),d=await n({prediction:u,example:c});for(let[m,h]of Object.entries(d))a[m]||(a[m]=[]),a[m].push(h)}catch{}let p={};for(let[c,u]of Object.entries(a))p[c]=u.length>0?u.reduce((d,m)=>d+m,0)/u.length:0;return p}findParetoFrontier(e){let t=[];for(let n=0;n<e.length;n++){let r=e[n],o=!1,i=0;for(let a=0;a<e.length;a++){if(n===a)continue;let l=e[a];if(this.dominates(l.scores,r.scores)){o=!0;break}this.dominates(r.scores,l.scores)&&i++}o||t.push({demos:r.demos||[],scores:r.scores,configuration:r.configuration,dominatedSolutions:i})}return t}dominates(e,t){let n=Object.keys(e),r=!0,o=!1;for(let i of n){let a=e[i]||0,l=t[i]||0;if(a<l){r=!1;break}a>l&&(o=!0)}return r&&o}calculateHypervolume(e){if(e.length===0)return;let t=e[0],n=Object.keys(t.scores);if(n.length===2){let[r,o]=n,i=0,a=[...e].sort((p,c)=>(c.scores[r]||0)-(p.scores[r]||0)),l=0;for(let p of a){let c=p.scores[r]||0,u=p.scores[o]||0;i+=c*(u-l),l=Math.max(l,u)}return i}}async saveCheckpoint(e,t,n,r,o={},i){let a=i?.overrideCheckpointSave||this.checkpointSave;if(!a)return;let l=Date.now(),p=!1,c;try{let u={version:"1.0.0",timestamp:Date.now(),optimizerType:e,optimizerConfig:t,currentRound:this.currentRound,totalRounds:this.stats.resourceUsage.totalTime>0?this.currentRound:0,bestScore:n,bestConfiguration:r,scoreHistory:[...this.scoreHistory],configurationHistory:[...this.configurationHistory],stats:{...this.stats},optimizerState:o,examples:this.examples,validationSet:this.validationSet};c=await a(u),p=!0}catch(u){throw p=!1,u}finally{let u=Date.now()-l;this.recordCheckpointMetrics("save",u,p,e)}return c}async loadCheckpoint(e,t){let n=t?.overrideCheckpointLoad||this.checkpointLoad;if(!n)return null;let r=Date.now(),o=!1,i=null;try{i=await n(e),o=i!==null}catch(a){throw o=!1,a}finally{let a=Date.now()-r;this.recordCheckpointMetrics("load",a,o,"unknown")}return i}restoreFromCheckpoint(e){this.currentRound=e.currentRound,this.scoreHistory=[...e.scoreHistory],this.configurationHistory=[...e.configurationHistory],this.stats={...e.stats}}shouldSaveCheckpoint(e,t){let n=t?.overrideCheckpointInterval||this.checkpointInterval;return n!==void 0&&e%n===0}async updateOptimizationProgress(e,t,n,r,o,i,a,l={},p){this.currentRound=e,this.scoreHistory.push(t),this.configurationHistory.push(n),this.shouldSaveCheckpoint(e,p)&&await this.saveCheckpoint(r,o,i,a,l,p),this.getOptimizerLogger(p)?.({name:"RoundProgress",value:{round:e,totalRounds:p?.maxIterations??0,currentScore:t,bestScore:i,configuration:n}})}async saveFinalCheckpoint(e,t,n,r,o={},i){i?.saveCheckpointOnComplete!==!1&&await this.saveCheckpoint(e,t,n,r,{...o,final:!0},i)}getLogger(e){if(this.isLoggingEnabled(e))return this.logger?this.logger:this.studentAI.getLogger()}isLoggingEnabled(e){return e?.verbose!==void 0?e.verbose:this.verbose??!0}recordOptimizationStart(e,t){if(this.metricsInstruments){if(t){let n=(t.match(/input:/g)||[]).length,r=(t.match(/output:/g)||[]).length;Ra(this.metricsInstruments,n,r,this.examples.length,this.getValidationSet().length,e)}Ca(this.metricsInstruments,e,this.targetScore,void 0)}}recordOptimizationComplete(e,t,n,r){if(!this.metricsInstruments)return;ga(this.metricsInstruments,e,t,n,r),fa(this.metricsInstruments,e,n);let o=this.costTracker?.getCurrentCost()??0,i=this.costTracker?.getTotalTokens()??0;xa(this.metricsInstruments,i,o,n)}recordConvergenceMetrics(e,t,n,r,o){this.metricsInstruments&&ha(this.metricsInstruments,e,t,n,r,o)}recordEarlyStoppingMetrics(e,t){this.metricsInstruments&&Aa(this.metricsInstruments,e,t)}recordTeacherStudentMetrics(e,t,n){this.metricsInstruments&&ya(this.metricsInstruments,e,t,n)}recordCheckpointMetrics(e,t,n,r){this.metricsInstruments&&Ia(this.metricsInstruments,e,t,n,r)}recordParetoMetrics(e,t,n,r){this.metricsInstruments&&ba(this.metricsInstruments,e,t,n,r)}recordPerformanceMetrics(e,t,n){this.metricsInstruments&&Ta(this.metricsInstruments,e,t,n)}isOptimizerLoggingEnabled(e){return this.debugOptimizer||(e?.verbose??this.verbose??!1)}getOptimizerLogger(e){if(this.isOptimizerLoggingEnabled(e))return this.optimizerLogger??M.optimizerLogger??Fn}getStats(){return{...this.stats}}reset(){this.stats=this.initializeStats(),this.costTracker?.reset(),this.currentRound=0,this.scoreHistory=[],this.configurationHistory=[]}};var ze=class extends ae{maxRounds;maxDemos;maxExamples;batchSize;earlyStoppingPatience;costMonitoring;maxTokensPerGeneration;verboseMode;debugMode;traces=[];constructor(e){super(e);let t=e.options||{};this.maxRounds=t.maxRounds??3,this.maxDemos=t.maxDemos??4,this.maxExamples=t.maxExamples??16,this.batchSize=t.batchSize??1,this.earlyStoppingPatience=t.earlyStoppingPatience??0,this.costMonitoring=t.costMonitoring??!1,this.maxTokensPerGeneration=t.maxTokensPerGeneration??0,this.verboseMode=t.verboseMode??!0,this.debugMode=t.debugMode??!1}async compileRound(e,t,n,r){let o=Date.now(),i=r?.maxDemos??this.maxDemos,a={modelConfig:{temperature:.7}};this.maxTokensPerGeneration>0&&(a.modelConfig.max_tokens=this.maxTokensPerGeneration);let l=Sa(this.examples,this.maxExamples),p=this.traces.length;for(let c=0;c<l.length;c+=this.batchSize){c>0&&(a.modelConfig.temperature=.7+.001*c);let u=l.slice(c,c+this.batchSize);for(let d of u){if(!d)continue;let m=l.filter(f=>f!==d);e.setExamples(m);let h=this.getTeacherOrStudentAI();this.stats.totalCalls++;let g;try{let f={...a,maxRetries:1};g=await e.forward(h,d,f),this.costMonitoring&&(this.stats.estimatedTokenUsage+=JSON.stringify(d).length/4+JSON.stringify(g).length/4),await n({prediction:g,example:d})>=.5&&(this.traces=[...this.traces,...e.getTraces()],this.stats.successfulDemos++)}catch(f){(this.verboseMode||this.debugMode)&&console.warn(`Student model failed during bootstrap: ${f instanceof Error?f.message:"Unknown error"}`),g={}}if(this.traces.length>=i)return}}if(this.earlyStoppingPatience>0){let u=this.traces.length-p;if(!this.stats.earlyStopping)this.stats.earlyStopping={bestScoreRound:u>0?t:0,patienceExhausted:!1,reason:"No improvement detected"};else if(u>0)this.stats.earlyStopping.bestScoreRound=t;else if(t-this.stats.earlyStopping.bestScoreRound>=this.earlyStoppingPatience){this.stats.earlyStopping.patienceExhausted=!0,this.stats.earlyStopped=!0,this.stats.earlyStopping.reason=`No improvement for ${this.earlyStoppingPatience} rounds`;return}}}async compile(e,t,n){let r=n?.maxIterations??this.maxRounds;this.traces=[],this.reset();for(let a=0;a<r&&(await this.compileRound(e,a,t,n),!this.stats.earlyStopped);a++);if(this.traces.length===0)throw new Error("No demonstrations found. Either provide more examples or improve the existing ones.");let o=wa(this.traces),i=0;return this.traces.length>0&&(i=this.stats.successfulDemos/Math.max(1,this.stats.totalCalls)),{demos:o,stats:this.stats,bestScore:i,finalConfiguration:{maxRounds:this.maxRounds,maxDemos:this.maxDemos,batchSize:this.batchSize,successRate:i}}}};function wa(s){let e=new Map;for(let n of s)if(e.has(n.programId)){let r=e.get(n.programId);r&&r.push(n.trace)}else e.set(n.programId,[n.trace]);let t=[];return e.forEach((n,r)=>{t.push({traces:n,programId:r})}),t}var Sa=(s,e)=>{let t=[...s];for(let n=t.length-1;n>0;n--){let r=Math.floor(Math.random()*(n+1)),o=t[n],i=t[r];if(!o||!i)throw new Error("Invalid array elements");[t[n],t[r]]=[i,o]}return t.slice(0,e)};var Ln=class{endpoint;timeout;retryAttempts;retryDelay;logger;constructor(e){this.endpoint=e.endpoint.replace(/\/$/,""),this.timeout=e.timeout??3e4,this.retryAttempts=e.retryAttempts??3,this.retryDelay=e.retryDelay??1e3,this.logger=e.logger}async healthCheck(){try{return(await this.fetchWithRetry("/health",{method:"GET"})).ok}catch(e){return this.logger?.({name:"Notification",id:"health_check_failed",value:`Health check failed: ${e}`}),!1}}async createOptimizationJob(e){let t=await this.fetchWithRetry("/optimize",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)});if(!t.ok){let n=await t.text();throw new Error(`Failed to create optimization job: ${n}`)}return t.json()}async getJobStatus(e){let t=await this.fetchWithRetry(`/jobs/${e}`,{method:"GET"});if(!t.ok){let n=await t.text();throw new Error(`Failed to get job status: ${n}`)}return t.json()}async cancelJob(e){let t=await this.fetchWithRetry(`/jobs/${e}`,{method:"DELETE"});if(!t.ok){let n=await t.text();throw new Error(`Failed to cancel job: ${n}`)}}async suggestParameters(e){let t=await this.fetchWithRetry(`/studies/${e}/suggest`,{method:"POST",headers:{"Content-Type":"application/json"}});if(!t.ok){let n=await t.text();throw new Error(`Failed to suggest parameters: ${n}`)}return t.json()}async evaluateTrial(e){let t=await this.fetchWithRetry(`/studies/${e.study_name}/evaluate`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)});if(!t.ok){let n=await t.text();throw new Error(`Failed to evaluate trial: ${n}`)}}async getStudyResults(e){let t=await this.fetchWithRetry(`/studies/${e}/results`,{method:"GET"});if(!t.ok){let n=await t.text();throw new Error(`Failed to get study results: ${n}`)}return t.json()}async deleteStudy(e){let t=await this.fetchWithRetry(`/studies/${e}`,{method:"DELETE"});if(!t.ok){let n=await t.text();throw new Error(`Failed to delete study: ${n}`)}}async listStudies(){let e=await this.fetchWithRetry("/studies",{method:"GET"});if(!e.ok){let t=await e.text();throw new Error(`Failed to list studies: ${t}`)}return e.json()}async waitForJobCompletion(e,t=2e3,n=3e5){let r=Date.now();for(;Date.now()-r<n;){let o=await this.getJobStatus(e);if(["completed","failed","cancelled"].includes(o.status))return o;this.logger?.({name:"Notification",id:"job_status",value:`Job ${e} status: ${o.status}, waiting...`}),await this.sleep(t)}throw new Error(`Job ${e} did not complete within ${n}ms`)}async fetchWithRetry(e,t){let n=`${this.endpoint}${e}`,r=null;for(let o=0;o<this.retryAttempts;o++)try{let i=new AbortController,a=setTimeout(()=>i.abort(),this.timeout),l=await fetch(n,{...t,signal:i.signal});return clearTimeout(a),l}catch(i){r=i,this.logger?.({name:"Notification",id:"retry_attempt",value:`Attempt ${o+1} failed: ${i}`}),o<this.retryAttempts-1&&await this.sleep(this.retryDelay*Math.pow(2,o))}throw new Error(`Request failed after ${this.retryAttempts} attempts: ${r?.message}`)}sleep(e){return new Promise(t=>setTimeout(t,e))}};var Gn=class extends ae{maxBootstrappedDemos;maxLabeledDemos;numCandidates;initTemperature;numTrials;minibatch;minibatchSize;minibatchFullEvalSteps;programAwareProposer;dataAwareProposer;viewDataBatchSize;tipAwareProposer;fewshotAwareProposer;earlyStoppingTrials;minImprovementThreshold;bayesianOptimization;acquisitionFunction;explorationWeight;sampleCount;miproConfigHistory=[];surrogateModel=new Map;pythonClient;constructor(e){if(super(e),this.numCandidates=e.numCandidates??5,this.initTemperature=e.initTemperature??.7,this.maxBootstrappedDemos=e.maxBootstrappedDemos??3,this.maxLabeledDemos=e.maxLabeledDemos??4,this.numTrials=e.numTrials??30,this.minibatch=e.minibatch??!0,this.minibatchSize=e.minibatchSize??25,this.minibatchFullEvalSteps=e.minibatchFullEvalSteps??10,this.programAwareProposer=e.programAwareProposer??!0,this.dataAwareProposer=e.dataAwareProposer??!0,this.viewDataBatchSize=e.viewDataBatchSize??10,this.tipAwareProposer=e.tipAwareProposer??!0,this.fewshotAwareProposer=e.fewshotAwareProposer??!0,this.earlyStoppingTrials=e.earlyStoppingTrials??5,this.minImprovementThreshold=e.minImprovementThreshold??.01,this.bayesianOptimization=e.bayesianOptimization??!1,this.acquisitionFunction=e.acquisitionFunction??"expected_improvement",this.explorationWeight=e.explorationWeight??.1,this.sampleCount=e.sampleCount??1,e.optimizerEndpoint){let t={endpoint:e.optimizerEndpoint,timeout:e.optimizerTimeout??3e4,retryAttempts:e.optimizerRetries??3,logger:n=>{this.logger?.({name:"Notification",id:"python_client",value:typeof n=="string"?n:JSON.stringify(n)})}};this.pythonClient=new Ln(t)}this.stats.convergenceInfo.convergenceThreshold=this.minImprovementThreshold}configureAuto(e){switch(e){case"light":this.numCandidates=3,this.numTrials=10,this.minibatch=!0,this.minibatchSize=20;break;case"medium":this.numCandidates=5,this.numTrials=20,this.minibatch=!0,this.minibatchSize=25;break;case"heavy":this.numCandidates=7,this.numTrials=30,this.minibatch=!0,this.minibatchSize=30;break}}generateTips(){return["Be very specific and detailed in your instructions.","Focus on step-by-step reasoning in your instructions.","Provide clear constraints and guidelines in your instructions.","Keep your instructions concise and to the point.","Emphasize accuracy and precision in your instructions.","Include examples of good outputs in your instructions.","Focus on handling edge cases in your instructions.","Explicitly outline the reasoning process in your instructions."]}async generateProgramSummary(e,t){let r=`
252
252
  Analyze this language model program and provide a concise summary of its purpose and structure.
253
253
 
254
254
  Program Signature: ${e.getSignature()}
@@ -258,7 +258,7 @@ Provide a 2-3 sentence summary focusing on:
258
258
  2. The input-output relationship
259
259
  3. Any special constraints or requirements
260
260
 
261
- Summary:`;try{let r=await t.chat({chatPrompt:[{role:"user",content:o}]});return"results"in r&&r.results[0]?.content?.trim()||"General language model program"}catch{return"General language model program"}}async generateDatasetSummary(e,t){if(e.length===0)return"No examples available";let n=Math.min(this.viewDataBatchSize,e.length),i=`
261
+ Summary:`;try{let o=await t.chat({chatPrompt:[{role:"user",content:r}]});return"results"in o&&o.results[0]?.content?.trim()||"General language model program"}catch{return"General language model program"}}async generateDatasetSummary(e,t){if(e.length===0)return"No examples available";let n=Math.min(this.viewDataBatchSize,e.length),i=`
262
262
  Analyze this dataset and provide a concise summary of its characteristics.
263
263
 
264
264
  Sample Examples:
@@ -270,9 +270,9 @@ Provide a 2-3 sentence summary focusing on:
270
270
  2. Common patterns or structures in the examples
271
271
  3. Key challenges or requirements for processing this data
272
272
 
273
- Dataset Summary:`;try{let a=await t.chat({chatPrompt:[{role:"user",content:i}]});return"results"in a&&a.results[0]?.content?.trim()||"General dataset"}catch{return"General dataset"}}async generateInstruction({tip:e,candidateIndex:t,ai:n,programSummary:o,datasetSummary:r,previousInstructions:i=[]}){let a="";this.programAwareProposer&&o&&(a+=`
274
- Program Context: ${o}`),this.dataAwareProposer&&r&&(a+=`
275
- Dataset Context: ${r}`),this.fewshotAwareProposer&&i.length>0&&(a+=`
273
+ Dataset Summary:`;try{let a=await t.chat({chatPrompt:[{role:"user",content:i}]});return"results"in a&&a.results[0]?.content?.trim()||"General dataset"}catch{return"General dataset"}}async generateInstruction({tip:e,candidateIndex:t,ai:n,programSummary:r,datasetSummary:o,previousInstructions:i=[]}){let a="";this.programAwareProposer&&r&&(a+=`
274
+ Program Context: ${r}`),this.dataAwareProposer&&o&&(a+=`
275
+ Dataset Context: ${o}`),this.fewshotAwareProposer&&i.length>0&&(a+=`
276
276
  Previous Instructions (avoid repeating): ${i.slice(-3).join("; ")}`);let l=`
277
277
  Generate a high-quality instruction for a language model program.
278
278
 
@@ -288,88 +288,88 @@ Requirements:
288
288
  5. Keep it concise but comprehensive
289
289
 
290
290
  Generate a single, well-crafted instruction:
291
- Instruction:`;try{let u=await n.chat({chatPrompt:[{role:"user",content:l}]});if("results"in u){let d=u.results[0]?.content?.trim();if(d&&d.length>10)return d}}catch{}let p=["Analyze the input systematically and provide a precise, well-reasoned response.","Think through this step-by-step, considering all relevant factors before responding.","Examine the input carefully and generate an accurate, detailed answer.","Process the information methodically and deliver a clear, comprehensive response.","Consider the context thoroughly and provide a thoughtful, accurate answer."],c=p[t%p.length]||p[0];return e&&(c=`${c} ${e}`),c}async proposeInstructionCandidates(e,t){let n=[],o=this.getTeacherOrStudentAI(t),r,i;this.dataAwareProposer&&(i=await this.generateDatasetSummary(this.examples,o));let a=this.tipAwareProposer?this.generateTips():[];for(let l=0;l<this.numCandidates;l++){let p=a.length>0?l%a.length:-1,c=p>=0?a[p]:void 0,u=await this.generateInstruction({tip:c,candidateIndex:l,ai:o,programSummary:r,datasetSummary:i,previousInstructions:n});n.push(u)}return n}async bootstrapFewShotExamples(e,t){return(await new ze({studentAI:this.studentAI,examples:this.examples,options:{maxDemos:this.maxBootstrappedDemos,maxRounds:3,verboseMode:this.isLoggingEnabled()}}).compile(e,t,{maxDemos:this.maxBootstrappedDemos})).demos||[]}selectLabeledExamples(){let e=[],t=new Set;for(;t.size<this.maxLabeledDemos&&t.size<this.examples.length;){let n=Math.floor(Math.random()*this.examples.length);if(!t.has(n)){t.add(n);let o=this.examples[n];o&&e.push(o)}}return e}async runOptimization(e,t,n,o,r,i,a){let l={instruction:o[0]||"",bootstrappedDemos:Math.min(1,t.length),labeledExamples:Math.min(1,n.length)},p=0,c=0,u=[],d=0;if(this.resumeFromCheckpoint){let m=await this.loadCheckpoint(this.resumeFromCheckpoint,a);m&&m.optimizerType==="MiPRO"&&(this.restoreFromCheckpoint(m),d=m.currentRound,p=m.bestScore,l=m.bestConfiguration||l,c=m.stats.convergenceInfo?.stagnationRounds||0)}for(let m=d;m<this.numTrials;m++){let h;this.bayesianOptimization&&this.miproConfigHistory.length>2?h=await this.selectConfigurationViaBayesianOptimization(o,t,n):h={instruction:o[m%o.length]||o[0]||"",bootstrappedDemos:Math.min(Math.floor(Math.random()*(t.length+1)),this.maxBootstrappedDemos),labeledExamples:Math.min(Math.floor(Math.random()*(n.length+1)),this.maxLabeledDemos)};let g=await this.evaluateConfig(e,h,t,n,r,i,m+1);this.updateSurrogateModel(h,g),u.push(g);let f=g-p;if(f>this.minImprovementThreshold?(p=g,l=h,c=0):c++,await this.updateOptimizationProgress(m+1,g,h,"MiPRO",this.getConfiguration(),p,l,{stagnationRounds:c,bootstrappedDemos:t.length,labeledExamples:n.length,instructions:o.length},a),this.onProgress&&this.onProgress({round:m+1,totalRounds:this.numTrials,currentScore:g,bestScore:p,tokensUsed:this.stats.resourceUsage.totalTokens,timeElapsed:Date.now(),successfulExamples:this.stats.successfulDemos,totalExamples:this.examples.length,currentConfiguration:h,convergenceInfo:{improvement:f,stagnationRounds:c,isConverging:c<this.earlyStoppingTrials}}),this.checkCostLimits()){this.triggerEarlyStopping("Cost limit reached",m+1);break}if(c>=this.earlyStoppingTrials){this.triggerEarlyStopping(`No improvement for ${this.earlyStoppingTrials} trials`,m-c+1);break}if(this.checkTargetScore(p)){this.triggerEarlyStopping(`Target score ${this.targetScore} reached`,m+1);break}}return this.stats.convergenceInfo.stagnationRounds=c,this.stats.convergenceInfo.finalImprovement=u.length>1?p-u[0]:0,this.stats.convergenceInfo.converged=c<this.earlyStoppingTrials,{bestConfig:l,bestScore:p}}async evaluateConfig(e,t,n,o,r,i,a=0){let l=new N(e.getSignature());this.applyConfigToProgram(l,t,n,o);let p=0,c=0,u;if(this.minibatch){let h=Math.min(this.minibatchSize,r.length);a%this.minibatchFullEvalSteps===0||a>this.numTrials*.8?u=Math.min(r.length,h*2):u=Math.max(3,Math.min(h,r.length))}else u=r.length;let m=this.shuffleArray([...Array(r.length).keys()]).slice(0,u).map(h=>r[h]);for(let h of m)try{let g=this.sampleCount>1?{sampleCount:this.sampleCount,resultPicker:wa(),maxRetries:1}:{maxRetries:1},f=await l.forward(this.studentAI,h,g),A=await i({prediction:f,example:h});p+=A,c++,this.stats.totalCalls++}catch(g){this.isLoggingEnabled()&&console.warn(`Student model failed during evaluation: ${g instanceof Error?g.message:"Unknown error"}`),this.stats.totalCalls++}return c>0?p/c:0}shuffleArray(e){let t=[...e];for(let n=t.length-1;n>0;n--){let o=Math.floor(Math.random()*(n+1));[t[n],t[o]]=[t[o],t[n]]}return t}applyConfigToProgram(e,t,n,o){e.setInstruction&&e.setInstruction(t.instruction),t.bootstrappedDemos>0&&e.setDemos&&e.setDemos(n.slice(0,t.bootstrappedDemos)),t.labeledExamples>0&&e.setExamples&&e.setExamples(o.slice(0,t.labeledExamples))}async compile(e,t,n){let o=Date.now();this.setupRandomSeed();let r=n;if(r?.auto&&this.configureAuto(r.auto),this.pythonClient){if(!await this.pythonClient.healthCheck())throw new Error("Python optimizer service is not available or unhealthy");return this.logger?.({name:"Notification",id:"mipro_mode",value:"Using Python optimizer service for MiPRO optimization"}),await this.compilePython(e,t,n)}let i=this.getValidationSet(n)||(r?.validationExamples??this.examples.slice(0,Math.floor(this.examples.length*.2))),a=[];this.maxBootstrappedDemos>0&&(a=await this.bootstrapFewShotExamples(e,t));let l=[];this.maxLabeledDemos>0&&(l=this.selectLabeledExamples());let p=await this.proposeInstructionCandidates(e,n),{bestConfig:c,bestScore:u}=await this.runOptimization(e,a,l,p,i,t,n);this.checkTargetScore(u)&&this.triggerEarlyStopping(`Target score ${this.targetScore} reached with score ${u}`,this.numTrials);let d;"getSignature"in e&&typeof e.getSignature=="function"?d=e.getSignature():d="input -> output";let m=new N(d);return this.applyConfigToAxGen(m,c,a,l),this.updateResourceUsage(o),this.stats.convergenceInfo.converged=!0,this.stats.convergenceInfo.finalImprovement=u,await this.saveFinalCheckpoint("MiPRO",this.getConfiguration(),u,c,{bootstrappedDemos:a.length,labeledExamples:l.length,instructions:p.length,optimizedGen:!!m},n),{demos:a,stats:this.stats,bestScore:u,optimizedGen:m,finalConfiguration:{instruction:c.instruction,bootstrappedDemos:c.bootstrappedDemos,labeledExamples:c.labeledExamples,numCandidates:this.numCandidates,numTrials:this.numTrials,sampleCount:this.sampleCount}}}applyConfigToAxGen(e,t,n,o){"setInstruction"in e&&typeof e.setInstruction=="function"&&e.setInstruction(t.instruction),t.bootstrappedDemos>0&&e.setDemos(n.slice(0,t.bootstrappedDemos)),t.labeledExamples>0&&e.setExamples(o.slice(0,t.labeledExamples))}getConfiguration(){return{numCandidates:this.numCandidates,initTemperature:this.initTemperature,maxBootstrappedDemos:this.maxBootstrappedDemos,maxLabeledDemos:this.maxLabeledDemos,numTrials:this.numTrials,minibatch:this.minibatch,minibatchSize:this.minibatchSize,minibatchFullEvalSteps:this.minibatchFullEvalSteps,programAwareProposer:this.programAwareProposer,dataAwareProposer:this.dataAwareProposer,tipAwareProposer:this.tipAwareProposer,fewshotAwareProposer:this.fewshotAwareProposer,earlyStoppingTrials:this.earlyStoppingTrials,minImprovementThreshold:this.minImprovementThreshold,bayesianOptimization:this.bayesianOptimization,acquisitionFunction:this.acquisitionFunction,explorationWeight:this.explorationWeight,sampleCount:this.sampleCount}}updateConfiguration(e){e.numCandidates!==void 0&&(this.numCandidates=e.numCandidates),e.initTemperature!==void 0&&(this.initTemperature=e.initTemperature),e.maxBootstrappedDemos!==void 0&&(this.maxBootstrappedDemos=e.maxBootstrappedDemos),e.maxLabeledDemos!==void 0&&(this.maxLabeledDemos=e.maxLabeledDemos),e.numTrials!==void 0&&(this.numTrials=e.numTrials),e.minibatch!==void 0&&(this.minibatch=e.minibatch),e.minibatchSize!==void 0&&(this.minibatchSize=e.minibatchSize),e.earlyStoppingTrials!==void 0&&(this.earlyStoppingTrials=e.earlyStoppingTrials),e.minImprovementThreshold!==void 0&&(this.minImprovementThreshold=e.minImprovementThreshold),e.sampleCount!==void 0&&(this.sampleCount=e.sampleCount)}reset(){super.reset(),this.miproConfigHistory=[],this.surrogateModel.clear(),this.stats.convergenceInfo.convergenceThreshold=this.minImprovementThreshold}validateProgram(e){let t=[],n=[];return this.examples.length<this.maxBootstrappedDemos+this.maxLabeledDemos&&(t.push(`Not enough examples: need at least ${this.maxBootstrappedDemos+this.maxLabeledDemos}, got ${this.examples.length}`),n.push("Reduce maxBootstrappedDemos or maxLabeledDemos, or provide more examples")),this.getValidationSet().length<5&&(t.push("Validation set too small for reliable MiPRO optimization"),n.push("Provide more examples or a larger validation set")),{isValid:t.length===0,issues:t,suggestions:n}}encodeConfiguration(e){return`${this.hashString(e.instruction)}_${e.bootstrappedDemos}_${e.labeledExamples}`}hashString(e){let t=0;if(e.length===0)return t.toString();for(let n=0;n<e.length;n++){let o=e.charCodeAt(n);t=(t<<5)-t+o,t=t&t}return Math.abs(t).toString(16)}updateSurrogateModel(e,t){this.miproConfigHistory.push({config:{...e},score:t});let n=this.encodeConfiguration(e),o=this.miproConfigHistory.filter(r=>this.encodeConfiguration(r.config)===n);if(o.length>0){let r=o.map(l=>l.score),i=r.reduce((l,p)=>l+p,0)/r.length,a=r.length>1?r.reduce((l,p)=>l+(p-i)**2,0)/(r.length-1):.1;this.surrogateModel.set(n,{mean:i,variance:a})}}predictPerformance(e){let t=this.encodeConfiguration(e);if(this.surrogateModel.has(t))return this.surrogateModel.get(t);if(this.miproConfigHistory.length>0){let n=this.miproConfigHistory.map(i=>{let a=Math.abs(i.config.bootstrappedDemos-e.bootstrappedDemos)+Math.abs(i.config.labeledExamples-e.labeledExamples);return{score:i.score,similarity:1/(1+a)}}),o=n.reduce((i,a)=>i+a.similarity,0);return{mean:n.reduce((i,a)=>i+a.score*a.similarity,0)/o,variance:.2}}return{mean:.5,variance:.3}}calculateAcquisitionValue(e){let t=this.predictPerformance(e),{mean:n,variance:o}=t,r=Math.sqrt(o),i=this.miproConfigHistory.length>0?Math.max(...this.miproConfigHistory.map(a=>a.score)):0;switch(this.acquisitionFunction){case"expected_improvement":{let a=n-i;if(r===0)return Math.max(0,a);let l=a/r,p=.5*(1+this.erf(l/Math.sqrt(2))),c=Math.exp(-.5*l*l)/Math.sqrt(2*Math.PI);return a*p+r*c}case"upper_confidence_bound":return n+this.explorationWeight*r;case"probability_improvement":{let a=n-i;if(r===0)return a>0?1:0;let l=a/r;return .5*(1+this.erf(l/Math.sqrt(2)))}default:return n}}erf(e){let t=.254829592,n=-.284496736,o=1.421413741,r=-1.453152027,i=1.061405429,a=.3275911,l=e>=0?1:-1,p=Math.abs(e),c=1/(1+a*p),u=1-((((i*c+r)*c+o)*c+n)*c+t)*c*Math.exp(-p*p);return l*u}async selectConfigurationViaBayesianOptimization(e,t,n){let o=[],r=Math.min(20,e.length*3);for(let i=0;i<r;i++){let a={instruction:e[i%e.length]||e[0]||"",bootstrappedDemos:Math.min(Math.floor(Math.random()*(t.length+1)),this.maxBootstrappedDemos),labeledExamples:Math.min(Math.floor(Math.random()*(n.length+1)),this.maxLabeledDemos)},l=this.calculateAcquisitionValue(a);o.push({config:a,acquisitionValue:l})}return o.sort((i,a)=>a.acquisitionValue-i.acquisitionValue),o[0].config}async compilePython(e,t,n){if(!this.pythonClient)throw new Error("Python client not initialized");let o=`mipro_${Date.now()}_${Math.random().toString(36).substr(2,9)}`,r={study_name:o,parameters:[{name:"temperature",type:"float",low:.1,high:2},{name:"bootstrappedDemos",type:"int",low:0,high:this.maxBootstrappedDemos}],objective:{name:"score",direction:"maximize"},n_trials:this.numTrials,sampler:"TPESampler",pruner:this.minibatch?"MedianPruner":void 0},i=await this.pythonClient.createOptimizationJob(r);this.logger?.({name:"Notification",id:"python_job_started",value:`Started Python optimization job ${i.job_id}`});let a=Number.NEGATIVE_INFINITY,l=e,p=0;for(let c=0;c<this.numTrials;c++)try{let u=await this.pythonClient.suggestParameters(o),d=u.params.temperature,m=u.params.bootstrappedDemos,h=await this.evaluateConfiguration(e,t,{temperature:d,bootstrappedDemos:m},this.minibatch?this.examples.slice(0,this.minibatchSize):this.examples);p++,await this.pythonClient.evaluateTrial({study_name:o,trial_number:u.trial_number,value:h}),h>a&&(a=h),this.logger?.({name:"Notification",id:"python_trial_result",value:`Trial ${c+1}: score=${h.toFixed(3)}, temp=${d.toFixed(2)}`}),this.onProgress?.({round:c+1,totalRounds:this.numTrials,currentScore:h,bestScore:a,tokensUsed:this.stats.estimatedTokenUsage,timeElapsed:Date.now()-Date.now(),successfulExamples:p,totalExamples:this.examples.length})}catch(u){this.logger?.({name:"Notification",id:"python_trial_error",value:`Trial ${c+1} failed: ${u}`})}try{let c=await this.pythonClient.getStudyResults(o);this.logger?.({name:"Notification",id:"python_results",value:`Python optimization completed with ${c.n_trials} trials`})}catch(c){this.logger?.({name:"Notification",id:"python_results_error",value:`Failed to get study results: ${c}`})}try{await this.pythonClient.deleteStudy(o)}catch{}return this.stats.bestScore=a,this.stats.totalCalls=p,{bestScore:a,stats:this.stats,optimizedGen:l}}async evaluateConfiguration(e,t,n,o){let r=0,i=0,a=o.slice(0,Math.min(5,o.length));for(let l of a)try{let p=await e.forward(this.studentAI,l),c=await t({prediction:p,example:l});typeof c=="number"&&!Number.isNaN(c)&&(r+=c,i++)}catch{}return i>0?r/i:0}},wa=()=>async s=>{if(s.type==="fields"){let e={};for(let{index:o,sample:r}of s.results){let i=JSON.stringify(r);e[i]||(e[i]={count:0,index:o}),e[i].count+=1}let t,n=-1;for(let[o,r]of Object.entries(e))r.count>n&&(n=r.count,t=o);return e[t]?.index??0}return s.results[0]?.index??0};function Ys(s){return P.create(s)}function Ln(s,e){let t=P.create(s);return new N(t,e)}var je=class{analyzeMappingDependencies(e,t){if(!e||typeof e!="function")return[];let n=[];try{let o=e.toString(),r=Array.from(o.matchAll(/state\.(\w+)/g));for(let i of r)i[1]&&!n.includes(i[1])&&n.push(i[1]);if(n.length===0)try{let i=this.createDependencyTracker(n);e(i)}catch{}}catch(o){console.debug("Dependency analysis failed:",o)}return n}createTrackingProxy(e,t){let n=this;return new Proxy(e,{get(o,r){typeof r=="string"&&!t.includes(r)&&t.push(r);let i=o[r];return i&&typeof i=="object"?n.createTrackingProxy(i,t):i},has(o,r){return typeof r=="string"&&!t.includes(r)&&t.push(r),r in o}})}parseStaticDependencies(e){let t=[];try{let n=Array.from(e.matchAll(/state\.(\w+)/g));for(let i of n)i[1]&&!t.includes(i[1])&&t.push(i[1]);let o=Array.from(e.matchAll(/\$\{state\.(\w+)\}/g));for(let i of o)i[1]&&!t.includes(i[1])&&t.push(i[1]);let r=Array.from(e.matchAll(/\{\s*(\w+)(?:\s*,\s*(\w+))*\s*\}\s*=\s*state/g));for(let i of r)for(let a=1;a<i.length;a++)i[a]&&!t.includes(i[a])&&t.push(i[a])}catch(n){console.debug("Static dependency parsing failed:",n)}return t}createDependencyTracker(e){return new Proxy({},{get(t,n){return typeof n=="string"&&!e.includes(n)&&e.push(n),new Proxy({},{get:()=>{}})}})}};async function He(s,e,t){if(!t||t<=0||t>=s.length){let o=s.map((r,i)=>e(r,i));return Promise.all(o)}let n=new Array(s.length);for(let o=0;o<s.length;o+=t){let i=s.slice(o,o+t).map((l,p)=>{let c=o+p;return e(l,c).then(u=>({result:u,originalIndex:c}))}),a=await Promise.all(i);for(let{result:l,originalIndex:p}of a)n[p]=l}return n}var Ke=class{steps=[];parallelGroups=[];analyzer=new je;initialFields=new Set;addExecutionStep(e,t,n,o,r,i,a){let l=[],p=[],c=o||"map";if(t&&n)c="execute",l=this.analyzer.analyzeMappingDependencies(n,t),p=[`${t}Result`];else if(c==="map"&&r)p=this.analyzeMapTransformation(r),l=this.getAllProducedFields();else if(c==="parallel-map"){if(Array.isArray(r)){let d=new Set;for(let m of r)this.analyzeMapTransformation(m).forEach(g=>d.add(g));p=Array.from(d)}else r?p=this.analyzeMapTransformation(r):p=["_parallelMapResult"];l=this.getAllProducedFields()}else if(c==="merge"){if(i?.resultKey)p=[i.resultKey];else{let m=this.analyzeBranchMergeFields();p=m.length>0?m:["_mergedResult"]}e.toString().includes("_parallelResults")?l=["_parallelResults"]:l=this.getAllProducedFields()}else if(c==="parallel")p=["_parallelResults"],l=this.getAllProducedFields();else if(c==="derive")if(a?.outputFieldName&&a?.inputFieldName){p=[a.outputFieldName];let d=r?this.analyzer.analyzeMappingDependencies(r,"derive"):[];l=[a.inputFieldName,...d].filter((m,h,g)=>g.indexOf(m)===h)}else p=["_deriveResult"],l=this.getAllProducedFields();else e.toString().includes("transform(")?(c="map",l=this.getAllProducedFields(),p=["_mapResult"]):e.toString().includes("_parallelResults")&&(p=["_parallelResults"],l=this.getAllProducedFields());for(let d of l)this.getAllProducedFields().includes(d)||this.initialFields.add(d);let u={type:c,nodeName:t,dependencies:l,produces:p,stepFunction:e,stepIndex:this.steps.length};this.steps.push(u)}analyzeStepFunctionProduction(e){try{let t=this.analyzeStepFunctionSource(e);if(t.length>0&&!t.includes("_stepResult"))return t}catch(t){console.debug("Step function source analysis failed:",t)}try{let t=this.createMockState(),n=Object.keys(t),r=e(t,{mainAi:{getOptions:()=>({trace:!1}),forward:()=>Promise.resolve({text:"mock"})},mainOptions:void 0});if(r&&typeof r=="object"&&"then"in r)return this.analyzeStepFunctionSource(e);if(r&&typeof r=="object"&&!Array.isArray(r)){let a=Object.keys(r).filter(l=>!n.includes(l));if(a.length>0)return a}}catch(t){console.debug("Step function dynamic analysis failed:",t)}return this.analyzeStepFunctionSource(e)}analyzeStepFunctionSource(e){try{let t=e.toString(),n=t.match(/\{\s*\.\.\.state\s*,\s*(\w+)\s*:/g);if(n){let r=n.map(i=>{let a=i.match(/(\w+)\s*:/);return a?a[1]:null}).filter(Boolean);if(r.length>0)return r}let o=t.match(/state\.(\w+)\s*=/g);if(o){let r=o.map(i=>{let a=i.match(/state\.(\w+)\s*=/);return a?a[1]:null}).filter(Boolean);if(r.length>0)return r}}catch(t){console.debug("Step function source analysis failed:",t)}return["_stepResult"]}analyzeMapTransformation(e){try{let t=this.createMockState(),n=e(t);if(n&&typeof n=="object"&&!Array.isArray(n))return Object.keys(n)}catch(t){console.debug("Map transformation analysis failed:",t)}return["_mapResult"]}createMockState(){let e={};for(let t of this.initialFields)e[t]=this.createMockValue(t);for(let t of this.steps)for(let n of t.produces)n.endsWith("Result")?e[n]={text:"mockText",value:"mockValue",result:"mockResult",data:"mockData",processedText:"mockProcessedText",sentimentValue:"mockSentiment",confidenceScore:.8,isComplex:!1,mockValue:"mockValue",responseText:"mockResponseText",inputText:"mockInputText"}:e[n]=this.createMockValue(n);return e}createMockValue(e){return e.includes("List")||e.includes("Array")||e.endsWith("s")?["mockItem1","mockItem2"]:e.includes("count")||e.includes("Count")||e.includes("index")||e.includes("Index")?0:e.includes("is")||e.includes("has")||e.includes("can")?!1:"mockValue"}analyzeBranchMergeFields(){let e=this.steps.slice(-5).filter(t=>t.type==="execute"&&t.nodeName).flatMap(t=>t.produces);return e.length>0?e:this.steps.filter(t=>t.type==="execute"&&t.nodeName).flatMap(t=>t.produces)}setInitialFields(e){this.initialFields=new Set(e),this.rebuildParallelGroups()}rebuildParallelGroups(){this.parallelGroups=[];let e=new Set,t=new Set(this.initialFields),n=0;for(;e.size<this.steps.length;){let o=[];for(let r of this.steps){if(e.has(r.stepIndex))continue;if(r.dependencies.length===0||r.dependencies.every(a=>t.has(a))){if(r.type==="merge"&&o.length>0)continue;if(o.push(r),e.add(r.stepIndex),r.type==="merge")break}}if(o.length>0){for(let r of o)r.produces.forEach(i=>t.add(i));this.parallelGroups.push({level:n,steps:o}),n++}else{let r=this.steps.filter(i=>!e.has(i.stepIndex));if(r.length>0){let i=r[0];e.add(i.stepIndex),i.produces.forEach(a=>t.add(a)),this.parallelGroups.push({level:n,steps:[i]}),n++}else break}}}getAllProducedFields(){let e=[];for(let t of this.steps)e.push(...t.produces);return e}createOptimizedExecution(e){let t=[];for(let n of this.parallelGroups)if(n.steps.length===1){let o=n.steps[0];o&&t.push(o.stepFunction)}else if(n.steps.length>1){let o=async(r,i)=>{let a=await He(n.steps,async c=>await c.stepFunction(r,i),e);if(a.some(c=>c&&typeof c=="object"&&"_parallelResults"in c)){let c=a.find(u=>u&&typeof u=="object"&&"_parallelResults"in u);return c||r}let p=r;for(let c of a)p={...p,...c};return p};t.push(o)}return t}getOptimizedExecutionSteps(){return this.parallelGroups.length===0&&this.steps.length>0&&this.rebuildParallelGroups(),this.createOptimizedExecution()}getExecutionPlan(){return this.parallelGroups.length===0&&this.steps.length>0&&this.rebuildParallelGroups(),{totalSteps:this.steps.length,parallelGroups:this.parallelGroups.length,maxParallelism:this.steps.length===0?1:Math.max(...this.parallelGroups.map(e=>e.steps.length),0),steps:this.steps,groups:this.parallelGroups}}};var Xs=s=>{console.log(s)},Ve=(s,e=!1)=>{if(e)return"[State hidden]";let t={};for(let[n,o]of Object.entries(s))if(typeof o=="string"&&o.length>100)t[n]=`${o.substring(0,100)}...`;else if(Array.isArray(o)&&o.length>3)t[n]=[...o.slice(0,3),`... (${o.length-3} more)`];else if(typeof o=="object"&&o!==null){let r=JSON.stringify(o);r.length>200?t[n]=`${r.substring(0,200)}...`:t[n]=o}else t[n]=o;return JSON.stringify(t,null,2)},We=s=>s<1e3?`${s.toFixed(1)}ms`:s<6e4?`${(s/1e3).toFixed(2)}s`:`${(s/6e4).toFixed(2)}min`,Tt=(s=Xs)=>{let e=new G,t=e.gray(`${"\u2501".repeat(80)}
291
+ Instruction:`;try{let u=await n.chat({chatPrompt:[{role:"user",content:l}]});if("results"in u){let d=u.results[0]?.content?.trim();if(d&&d.length>10)return d}}catch{}let p=["Analyze the input systematically and provide a precise, well-reasoned response.","Think through this step-by-step, considering all relevant factors before responding.","Examine the input carefully and generate an accurate, detailed answer.","Process the information methodically and deliver a clear, comprehensive response.","Consider the context thoroughly and provide a thoughtful, accurate answer."],c=p[t%p.length]||p[0];return e&&(c=`${c} ${e}`),c}async proposeInstructionCandidates(e,t){let n=[],r=this.getTeacherOrStudentAI(t),o,i;this.dataAwareProposer&&(i=await this.generateDatasetSummary(this.examples,r));let a=this.tipAwareProposer?this.generateTips():[];for(let l=0;l<this.numCandidates;l++){let p=a.length>0?l%a.length:-1,c=p>=0?a[p]:void 0,u=await this.generateInstruction({tip:c,candidateIndex:l,ai:r,programSummary:o,datasetSummary:i,previousInstructions:n});n.push(u)}return n}async bootstrapFewShotExamples(e,t){return(await new ze({studentAI:this.studentAI,examples:this.examples,options:{maxDemos:this.maxBootstrappedDemos,maxRounds:3,verboseMode:this.isLoggingEnabled()}}).compile(e,t,{maxDemos:this.maxBootstrappedDemos})).demos||[]}selectLabeledExamples(){let e=[],t=new Set;for(;t.size<this.maxLabeledDemos&&t.size<this.examples.length;){let n=Math.floor(Math.random()*this.examples.length);if(!t.has(n)){t.add(n);let r=this.examples[n];r&&e.push(r)}}return e}async runOptimization(e,t,n,r,o,i,a){let l={instruction:r[0]||"",bootstrappedDemos:Math.min(1,t.length),labeledExamples:Math.min(1,n.length)},p=0,c=0,u=[],d=0;if(this.resumeFromCheckpoint){let m=await this.loadCheckpoint(this.resumeFromCheckpoint,a);m&&m.optimizerType==="MiPRO"&&(this.restoreFromCheckpoint(m),d=m.currentRound,p=m.bestScore,l=m.bestConfiguration||l,c=m.stats.convergenceInfo?.stagnationRounds||0)}for(let m=d;m<this.numTrials;m++){let h;this.bayesianOptimization&&this.miproConfigHistory.length>2?h=await this.selectConfigurationViaBayesianOptimization(r,t,n):h={instruction:r[m%r.length]||r[0]||"",bootstrappedDemos:Math.min(Math.floor(Math.random()*(t.length+1)),this.maxBootstrappedDemos),labeledExamples:Math.min(Math.floor(Math.random()*(n.length+1)),this.maxLabeledDemos)};let g=await this.evaluateConfig(e,h,t,n,o,i,m+1);this.updateSurrogateModel(h,g),u.push(g);let f=g-p;if(f>this.minImprovementThreshold?(p=g,l=h,c=0):c++,await this.updateOptimizationProgress(m+1,g,h,"MiPRO",this.getConfiguration(),p,l,{stagnationRounds:c,bootstrappedDemos:t.length,labeledExamples:n.length,instructions:r.length},a),this.onProgress&&this.onProgress({round:m+1,totalRounds:this.numTrials,currentScore:g,bestScore:p,tokensUsed:this.stats.resourceUsage.totalTokens,timeElapsed:Date.now(),successfulExamples:this.stats.successfulDemos,totalExamples:this.examples.length,currentConfiguration:h,convergenceInfo:{improvement:f,stagnationRounds:c,isConverging:c<this.earlyStoppingTrials}}),this.checkCostLimits()){this.triggerEarlyStopping("Cost limit reached",m+1);break}if(c>=this.earlyStoppingTrials){this.triggerEarlyStopping(`No improvement for ${this.earlyStoppingTrials} trials`,m-c+1);break}if(this.checkTargetScore(p)){this.triggerEarlyStopping(`Target score ${this.targetScore} reached`,m+1);break}}return this.stats.convergenceInfo.stagnationRounds=c,this.stats.convergenceInfo.finalImprovement=u.length>1?p-u[0]:0,this.stats.convergenceInfo.converged=c<this.earlyStoppingTrials,{bestConfig:l,bestScore:p}}async evaluateConfig(e,t,n,r,o,i,a=0){let l=new $(e.getSignature());this.applyConfigToProgram(l,t,n,r);let p=0,c=0,u;if(this.minibatch){let h=Math.min(this.minibatchSize,o.length);a%this.minibatchFullEvalSteps===0||a>this.numTrials*.8?u=Math.min(o.length,h*2):u=Math.max(3,Math.min(h,o.length))}else u=o.length;let m=this.shuffleArray([...Array(o.length).keys()]).slice(0,u).map(h=>o[h]);for(let h of m)try{let g=this.sampleCount>1?{sampleCount:this.sampleCount,resultPicker:va(),maxRetries:1}:{maxRetries:1},f=await l.forward(this.studentAI,h,g),A=await i({prediction:f,example:h});p+=A,c++,this.stats.totalCalls++}catch(g){this.isLoggingEnabled()&&console.warn(`Student model failed during evaluation: ${g instanceof Error?g.message:"Unknown error"}`),this.stats.totalCalls++}return c>0?p/c:0}shuffleArray(e){let t=[...e];for(let n=t.length-1;n>0;n--){let r=Math.floor(Math.random()*(n+1));[t[n],t[r]]=[t[r],t[n]]}return t}applyConfigToProgram(e,t,n,r){e.setInstruction&&e.setInstruction(t.instruction),t.bootstrappedDemos>0&&e.setDemos&&e.setDemos(n.slice(0,t.bootstrappedDemos)),t.labeledExamples>0&&e.setExamples&&e.setExamples(r.slice(0,t.labeledExamples))}async compile(e,t,n){let r=Date.now();this.setupRandomSeed();let o=n;if(o?.auto&&this.configureAuto(o.auto),this.pythonClient){if(!await this.pythonClient.healthCheck())throw new Error("Python optimizer service is not available or unhealthy");return this.logger?.({name:"Notification",id:"mipro_mode",value:"Using Python optimizer service for MiPRO optimization"}),await this.compilePython(e,t,n)}let i=this.getValidationSet(n)||(o?.validationExamples??this.examples.slice(0,Math.floor(this.examples.length*.2))),a=[];this.maxBootstrappedDemos>0&&(a=await this.bootstrapFewShotExamples(e,t));let l=[];this.maxLabeledDemos>0&&(l=this.selectLabeledExamples());let p=await this.proposeInstructionCandidates(e,n),{bestConfig:c,bestScore:u}=await this.runOptimization(e,a,l,p,i,t,n);this.checkTargetScore(u)&&this.triggerEarlyStopping(`Target score ${this.targetScore} reached with score ${u}`,this.numTrials);let d;"getSignature"in e&&typeof e.getSignature=="function"?d=e.getSignature():d="input -> output";let m=new $(d);return this.applyConfigToAxGen(m,c,a,l),this.updateResourceUsage(r),this.stats.convergenceInfo.converged=!0,this.stats.convergenceInfo.finalImprovement=u,await this.saveFinalCheckpoint("MiPRO",this.getConfiguration(),u,c,{bootstrappedDemos:a.length,labeledExamples:l.length,instructions:p.length,optimizedGen:!!m},n),{demos:a,stats:this.stats,bestScore:u,optimizedGen:m,finalConfiguration:{instruction:c.instruction,bootstrappedDemos:c.bootstrappedDemos,labeledExamples:c.labeledExamples,numCandidates:this.numCandidates,numTrials:this.numTrials,sampleCount:this.sampleCount}}}applyConfigToAxGen(e,t,n,r){"setInstruction"in e&&typeof e.setInstruction=="function"&&e.setInstruction(t.instruction),t.bootstrappedDemos>0&&e.setDemos(n.slice(0,t.bootstrappedDemos)),t.labeledExamples>0&&e.setExamples(r.slice(0,t.labeledExamples))}getConfiguration(){return{numCandidates:this.numCandidates,initTemperature:this.initTemperature,maxBootstrappedDemos:this.maxBootstrappedDemos,maxLabeledDemos:this.maxLabeledDemos,numTrials:this.numTrials,minibatch:this.minibatch,minibatchSize:this.minibatchSize,minibatchFullEvalSteps:this.minibatchFullEvalSteps,programAwareProposer:this.programAwareProposer,dataAwareProposer:this.dataAwareProposer,tipAwareProposer:this.tipAwareProposer,fewshotAwareProposer:this.fewshotAwareProposer,earlyStoppingTrials:this.earlyStoppingTrials,minImprovementThreshold:this.minImprovementThreshold,bayesianOptimization:this.bayesianOptimization,acquisitionFunction:this.acquisitionFunction,explorationWeight:this.explorationWeight,sampleCount:this.sampleCount}}updateConfiguration(e){e.numCandidates!==void 0&&(this.numCandidates=e.numCandidates),e.initTemperature!==void 0&&(this.initTemperature=e.initTemperature),e.maxBootstrappedDemos!==void 0&&(this.maxBootstrappedDemos=e.maxBootstrappedDemos),e.maxLabeledDemos!==void 0&&(this.maxLabeledDemos=e.maxLabeledDemos),e.numTrials!==void 0&&(this.numTrials=e.numTrials),e.minibatch!==void 0&&(this.minibatch=e.minibatch),e.minibatchSize!==void 0&&(this.minibatchSize=e.minibatchSize),e.earlyStoppingTrials!==void 0&&(this.earlyStoppingTrials=e.earlyStoppingTrials),e.minImprovementThreshold!==void 0&&(this.minImprovementThreshold=e.minImprovementThreshold),e.sampleCount!==void 0&&(this.sampleCount=e.sampleCount)}reset(){super.reset(),this.miproConfigHistory=[],this.surrogateModel.clear(),this.stats.convergenceInfo.convergenceThreshold=this.minImprovementThreshold}validateProgram(e){let t=[],n=[];return this.examples.length<this.maxBootstrappedDemos+this.maxLabeledDemos&&(t.push(`Not enough examples: need at least ${this.maxBootstrappedDemos+this.maxLabeledDemos}, got ${this.examples.length}`),n.push("Reduce maxBootstrappedDemos or maxLabeledDemos, or provide more examples")),this.getValidationSet().length<5&&(t.push("Validation set too small for reliable MiPRO optimization"),n.push("Provide more examples or a larger validation set")),{isValid:t.length===0,issues:t,suggestions:n}}encodeConfiguration(e){return`${this.hashString(e.instruction)}_${e.bootstrappedDemos}_${e.labeledExamples}`}hashString(e){let t=0;if(e.length===0)return t.toString();for(let n=0;n<e.length;n++){let r=e.charCodeAt(n);t=(t<<5)-t+r,t=t&t}return Math.abs(t).toString(16)}updateSurrogateModel(e,t){this.miproConfigHistory.push({config:{...e},score:t});let n=this.encodeConfiguration(e),r=this.miproConfigHistory.filter(o=>this.encodeConfiguration(o.config)===n);if(r.length>0){let o=r.map(l=>l.score),i=o.reduce((l,p)=>l+p,0)/o.length,a=o.length>1?o.reduce((l,p)=>l+(p-i)**2,0)/(o.length-1):.1;this.surrogateModel.set(n,{mean:i,variance:a})}}predictPerformance(e){let t=this.encodeConfiguration(e);if(this.surrogateModel.has(t))return this.surrogateModel.get(t);if(this.miproConfigHistory.length>0){let n=this.miproConfigHistory.map(i=>{let a=Math.abs(i.config.bootstrappedDemos-e.bootstrappedDemos)+Math.abs(i.config.labeledExamples-e.labeledExamples);return{score:i.score,similarity:1/(1+a)}}),r=n.reduce((i,a)=>i+a.similarity,0);return{mean:n.reduce((i,a)=>i+a.score*a.similarity,0)/r,variance:.2}}return{mean:.5,variance:.3}}calculateAcquisitionValue(e){let t=this.predictPerformance(e),{mean:n,variance:r}=t,o=Math.sqrt(r),i=this.miproConfigHistory.length>0?Math.max(...this.miproConfigHistory.map(a=>a.score)):0;switch(this.acquisitionFunction){case"expected_improvement":{let a=n-i;if(o===0)return Math.max(0,a);let l=a/o,p=.5*(1+this.erf(l/Math.sqrt(2))),c=Math.exp(-.5*l*l)/Math.sqrt(2*Math.PI);return a*p+o*c}case"upper_confidence_bound":return n+this.explorationWeight*o;case"probability_improvement":{let a=n-i;if(o===0)return a>0?1:0;let l=a/o;return .5*(1+this.erf(l/Math.sqrt(2)))}default:return n}}erf(e){let t=.254829592,n=-.284496736,r=1.421413741,o=-1.453152027,i=1.061405429,a=.3275911,l=e>=0?1:-1,p=Math.abs(e),c=1/(1+a*p),u=1-((((i*c+o)*c+r)*c+n)*c+t)*c*Math.exp(-p*p);return l*u}async selectConfigurationViaBayesianOptimization(e,t,n){let r=[],o=Math.min(20,e.length*3);for(let i=0;i<o;i++){let a={instruction:e[i%e.length]||e[0]||"",bootstrappedDemos:Math.min(Math.floor(Math.random()*(t.length+1)),this.maxBootstrappedDemos),labeledExamples:Math.min(Math.floor(Math.random()*(n.length+1)),this.maxLabeledDemos)},l=this.calculateAcquisitionValue(a);r.push({config:a,acquisitionValue:l})}return r.sort((i,a)=>a.acquisitionValue-i.acquisitionValue),r[0].config}async compilePython(e,t,n){if(!this.pythonClient)throw new Error("Python client not initialized");let r=`mipro_${Date.now()}_${Math.random().toString(36).substr(2,9)}`,o={study_name:r,parameters:[{name:"temperature",type:"float",low:.1,high:2},{name:"bootstrappedDemos",type:"int",low:0,high:this.maxBootstrappedDemos}],objective:{name:"score",direction:"maximize"},n_trials:this.numTrials,sampler:"TPESampler",pruner:this.minibatch?"MedianPruner":void 0},i=await this.pythonClient.createOptimizationJob(o);this.logger?.({name:"Notification",id:"python_job_started",value:`Started Python optimization job ${i.job_id}`});let a=Number.NEGATIVE_INFINITY,l=e,p=0;for(let c=0;c<this.numTrials;c++)try{let u=await this.pythonClient.suggestParameters(r),d=u.params.temperature,m=u.params.bootstrappedDemos,h=await this.evaluateConfiguration(e,t,{temperature:d,bootstrappedDemos:m},this.minibatch?this.examples.slice(0,this.minibatchSize):this.examples);p++,await this.pythonClient.evaluateTrial({study_name:r,trial_number:u.trial_number,value:h}),h>a&&(a=h),this.logger?.({name:"Notification",id:"python_trial_result",value:`Trial ${c+1}: score=${h.toFixed(3)}, temp=${d.toFixed(2)}`}),this.onProgress?.({round:c+1,totalRounds:this.numTrials,currentScore:h,bestScore:a,tokensUsed:this.stats.estimatedTokenUsage,timeElapsed:Date.now()-Date.now(),successfulExamples:p,totalExamples:this.examples.length})}catch(u){this.logger?.({name:"Notification",id:"python_trial_error",value:`Trial ${c+1} failed: ${u}`})}try{let c=await this.pythonClient.getStudyResults(r);this.logger?.({name:"Notification",id:"python_results",value:`Python optimization completed with ${c.n_trials} trials`})}catch(c){this.logger?.({name:"Notification",id:"python_results_error",value:`Failed to get study results: ${c}`})}try{await this.pythonClient.deleteStudy(r)}catch{}return this.stats.bestScore=a,this.stats.totalCalls=p,{bestScore:a,stats:this.stats,optimizedGen:l}}async evaluateConfiguration(e,t,n,r){let o=0,i=0,a=r.slice(0,Math.min(5,r.length));for(let l of a)try{let p=await e.forward(this.studentAI,l),c=await t({prediction:p,example:l});typeof c=="number"&&!Number.isNaN(c)&&(o+=c,i++)}catch{}return i>0?o/i:0}},va=()=>async s=>{if(s.type==="fields"){let e={};for(let{index:r,sample:o}of s.results){let i=JSON.stringify(o);e[i]||(e[i]={count:0,index:r}),e[i].count+=1}let t,n=-1;for(let[r,o]of Object.entries(e))o.count>n&&(n=o.count,t=r);return e[t]?.index??0}return s.results[0]?.index??0};function Zs(s){return P.create(s)}function $n(s,e){let t=typeof s=="string"?P.create(s):s;return new $(t,e)}var je=class{analyzeMappingDependencies(e,t){if(!e||typeof e!="function")return[];let n=[];try{let r=e.toString(),o=Array.from(r.matchAll(/state\.(\w+)/g));for(let i of o)i[1]&&!n.includes(i[1])&&n.push(i[1]);if(n.length===0)try{let i=this.createDependencyTracker(n);e(i)}catch{}}catch(r){console.debug("Dependency analysis failed:",r)}return n}createTrackingProxy(e,t){let n=this;return new Proxy(e,{get(r,o){typeof o=="string"&&!t.includes(o)&&t.push(o);let i=r[o];return i&&typeof i=="object"?n.createTrackingProxy(i,t):i},has(r,o){return typeof o=="string"&&!t.includes(o)&&t.push(o),o in r}})}parseStaticDependencies(e){let t=[];try{let n=Array.from(e.matchAll(/state\.(\w+)/g));for(let i of n)i[1]&&!t.includes(i[1])&&t.push(i[1]);let r=Array.from(e.matchAll(/\$\{state\.(\w+)\}/g));for(let i of r)i[1]&&!t.includes(i[1])&&t.push(i[1]);let o=Array.from(e.matchAll(/\{\s*(\w+)(?:\s*,\s*(\w+))*\s*\}\s*=\s*state/g));for(let i of o)for(let a=1;a<i.length;a++)i[a]&&!t.includes(i[a])&&t.push(i[a])}catch(n){console.debug("Static dependency parsing failed:",n)}return t}createDependencyTracker(e){return new Proxy({},{get(t,n){return typeof n=="string"&&!e.includes(n)&&e.push(n),new Proxy({},{get:()=>{}})}})}};async function He(s,e,t){if(!t||t<=0||t>=s.length){let r=s.map((o,i)=>e(o,i));return Promise.all(r)}let n=new Array(s.length);for(let r=0;r<s.length;r+=t){let i=s.slice(r,r+t).map((l,p)=>{let c=r+p;return e(l,c).then(u=>({result:u,originalIndex:c}))}),a=await Promise.all(i);for(let{result:l,originalIndex:p}of a)n[p]=l}return n}var Ke=class{steps=[];parallelGroups=[];analyzer=new je;initialFields=new Set;addExecutionStep(e,t,n,r,o,i,a){let l=[],p=[],c=r||"map";if(t&&n)c="execute",l=this.analyzer.analyzeMappingDependencies(n,t),p=[`${t}Result`];else if(c==="map"&&o)p=this.analyzeMapTransformation(o),l=this.getAllProducedFields();else if(c==="parallel-map"){if(Array.isArray(o)){let d=new Set;for(let m of o)this.analyzeMapTransformation(m).forEach(g=>d.add(g));p=Array.from(d)}else o?p=this.analyzeMapTransformation(o):p=["_parallelMapResult"];l=this.getAllProducedFields()}else if(c==="merge"){if(i?.resultKey)p=[i.resultKey];else{let m=this.analyzeBranchMergeFields();p=m.length>0?m:["_mergedResult"]}e.toString().includes("_parallelResults")?l=["_parallelResults"]:l=this.getAllProducedFields()}else if(c==="parallel")p=["_parallelResults"],l=this.getAllProducedFields();else if(c==="derive")if(a?.outputFieldName&&a?.inputFieldName){p=[a.outputFieldName];let d=o?this.analyzer.analyzeMappingDependencies(o,"derive"):[];l=[a.inputFieldName,...d].filter((m,h,g)=>g.indexOf(m)===h)}else p=["_deriveResult"],l=this.getAllProducedFields();else e.toString().includes("transform(")?(c="map",l=this.getAllProducedFields(),p=["_mapResult"]):e.toString().includes("_parallelResults")&&(p=["_parallelResults"],l=this.getAllProducedFields());for(let d of l)this.getAllProducedFields().includes(d)||this.initialFields.add(d);let u={type:c,nodeName:t,dependencies:l,produces:p,stepFunction:e,stepIndex:this.steps.length};this.steps.push(u)}analyzeStepFunctionProduction(e){try{let t=this.analyzeStepFunctionSource(e);if(t.length>0&&!t.includes("_stepResult"))return t}catch(t){console.debug("Step function source analysis failed:",t)}try{let t=this.createMockState(),n=Object.keys(t),o=e(t,{mainAi:{getOptions:()=>({trace:!1}),forward:()=>Promise.resolve({text:"mock"})},mainOptions:void 0});if(o&&typeof o=="object"&&"then"in o)return this.analyzeStepFunctionSource(e);if(o&&typeof o=="object"&&!Array.isArray(o)){let a=Object.keys(o).filter(l=>!n.includes(l));if(a.length>0)return a}}catch(t){console.debug("Step function dynamic analysis failed:",t)}return this.analyzeStepFunctionSource(e)}analyzeStepFunctionSource(e){try{let t=e.toString(),n=t.match(/\{\s*\.\.\.state\s*,\s*(\w+)\s*:/g);if(n){let o=n.map(i=>{let a=i.match(/(\w+)\s*:/);return a?a[1]:null}).filter(Boolean);if(o.length>0)return o}let r=t.match(/state\.(\w+)\s*=/g);if(r){let o=r.map(i=>{let a=i.match(/state\.(\w+)\s*=/);return a?a[1]:null}).filter(Boolean);if(o.length>0)return o}}catch(t){console.debug("Step function source analysis failed:",t)}return["_stepResult"]}analyzeMapTransformation(e){try{let t=this.createMockState(),n=e(t);if(n&&typeof n=="object"&&!Array.isArray(n))return Object.keys(n)}catch(t){console.debug("Map transformation analysis failed:",t)}return["_mapResult"]}createMockState(){let e={};for(let t of this.initialFields)e[t]=this.createMockValue(t);for(let t of this.steps)for(let n of t.produces)n.endsWith("Result")?e[n]={text:"mockText",value:"mockValue",result:"mockResult",data:"mockData",processedText:"mockProcessedText",sentimentValue:"mockSentiment",confidenceScore:.8,isComplex:!1,mockValue:"mockValue",responseText:"mockResponseText",inputText:"mockInputText"}:e[n]=this.createMockValue(n);return e}createMockValue(e){return e.includes("List")||e.includes("Array")||e.endsWith("s")?["mockItem1","mockItem2"]:e.includes("count")||e.includes("Count")||e.includes("index")||e.includes("Index")?0:e.includes("is")||e.includes("has")||e.includes("can")?!1:"mockValue"}analyzeBranchMergeFields(){let e=this.steps.slice(-5).filter(t=>t.type==="execute"&&t.nodeName).flatMap(t=>t.produces);return e.length>0?e:this.steps.filter(t=>t.type==="execute"&&t.nodeName).flatMap(t=>t.produces)}setInitialFields(e){this.initialFields=new Set(e),this.rebuildParallelGroups()}rebuildParallelGroups(){this.parallelGroups=[];let e=new Set,t=new Set(this.initialFields),n=0;for(;e.size<this.steps.length;){let r=[];for(let o of this.steps){if(e.has(o.stepIndex))continue;if(o.dependencies.length===0||o.dependencies.every(a=>t.has(a))){if(o.type==="merge"&&r.length>0)continue;if(r.push(o),e.add(o.stepIndex),o.type==="merge")break}}if(r.length>0){for(let o of r)o.produces.forEach(i=>t.add(i));this.parallelGroups.push({level:n,steps:r}),n++}else{let o=this.steps.filter(i=>!e.has(i.stepIndex));if(o.length>0){let i=o[0];e.add(i.stepIndex),i.produces.forEach(a=>t.add(a)),this.parallelGroups.push({level:n,steps:[i]}),n++}else break}}}getAllProducedFields(){let e=[];for(let t of this.steps)e.push(...t.produces);return e}createOptimizedExecution(e){let t=[];for(let n of this.parallelGroups)if(n.steps.length===1){let r=n.steps[0];r&&t.push(r.stepFunction)}else if(n.steps.length>1){let r=async(o,i)=>{let a=await He(n.steps,async c=>await c.stepFunction(o,i),e);if(a.some(c=>c&&typeof c=="object"&&"_parallelResults"in c)){let c=a.find(u=>u&&typeof u=="object"&&"_parallelResults"in u);return c||o}let p=o;for(let c of a)p={...p,...c};return p};t.push(r)}return t}getOptimizedExecutionSteps(){return this.parallelGroups.length===0&&this.steps.length>0&&this.rebuildParallelGroups(),this.createOptimizedExecution()}getExecutionPlan(){return this.parallelGroups.length===0&&this.steps.length>0&&this.rebuildParallelGroups(),{totalSteps:this.steps.length,parallelGroups:this.parallelGroups.length,maxParallelism:this.steps.length===0?1:Math.max(...this.parallelGroups.map(e=>e.steps.length),0),steps:this.steps,groups:this.parallelGroups}}};var ei=s=>{console.log(s)},Ve=(s,e=!1)=>{if(e)return"[State hidden]";let t={};for(let[n,r]of Object.entries(s))if(typeof r=="string"&&r.length>100)t[n]=`${r.substring(0,100)}...`;else if(Array.isArray(r)&&r.length>3)t[n]=[...r.slice(0,3),`... (${r.length-3} more)`];else if(typeof r=="object"&&r!==null){let o=JSON.stringify(r);o.length>200?t[n]=`${o.substring(0,200)}...`:t[n]=r}else t[n]=r;return JSON.stringify(t,null,2)},We=s=>s<1e3?`${s.toFixed(1)}ms`:s<6e4?`${(s/1e3).toFixed(2)}s`:`${(s/6e4).toFixed(2)}min`,wt=(s=ei)=>{let e=new G,t=e.gray(`${"\u2501".repeat(80)}
292
292
  `),n=e.gray(`${"\u2500".repeat(40)}
293
- `);return o=>{let r="";switch(o.name){case"FlowStart":r=`
293
+ `);return r=>{let o="";switch(r.name){case"FlowStart":o=`
294
294
  ${e.blueBright("\u{1F504} [ AXFLOW START ]")}
295
- ${t}`,r+=`${e.white("Input Fields:")} ${e.cyan(o.inputFields.join(", "))}
296
- `,r+=`${e.white("Total Steps:")} ${e.yellow(o.totalSteps.toString())}
297
- `,r+=`${e.white("Parallel Groups:")} ${e.yellow(o.parallelGroups.toString())}
298
- `,r+=`${e.white("Max Parallelism:")} ${e.yellow(o.maxParallelism.toString())}
299
- `,r+=`${e.white("Auto-Parallel:")} ${o.autoParallelEnabled?e.green("enabled"):e.red("disabled")}
300
- `,r+=t;break;case"StepStart":{let i=o.stepType==="execute"?"\u26A1":o.stepType==="map"?"\u{1F504}":o.stepType==="merge"?"\u{1F500}":o.stepType==="parallel"?"\u2696\uFE0F":"\u{1F4CB}";r=`${e.greenBright(`${i} [ STEP ${o.stepIndex} START ]`)} ${e.white(`(${o.stepType})`)}`,o.nodeName&&(r+=` ${e.cyanBright(`Node: ${o.nodeName}`)}`),r+=`
301
- `,o.dependencies.length>0&&(r+=`${e.white("Dependencies:")} ${e.gray(o.dependencies.join(", "))}
302
- `),o.produces.length>0&&(r+=`${e.white("Produces:")} ${e.cyan(o.produces.join(", "))}
303
- `),r+=`${e.white("State:")} ${e.gray(Ve(o.state,!0))}
304
- `,r+=n;break}case"StepComplete":{let i=(o.stepType==="execute"||o.stepType==="map"||o.stepType==="merge"||o.stepType==="parallel","\u2705");r=`${e.greenBright(`${i} [ STEP ${o.stepIndex} COMPLETE ]`)} ${e.white(`(${o.stepType})`)}`,o.nodeName&&(r+=` ${e.cyanBright(`Node: ${o.nodeName}`)}`),r+=` ${e.magenta(`in ${We(o.executionTime)}`)}
305
- `,o.newFields&&o.newFields.length>0&&(r+=`${e.white("New Fields:")} ${e.green(o.newFields.join(", "))}
306
- `),o.result&&o.nodeName&&(r+=`${e.white("Result:")} ${e.yellow(JSON.stringify(o.result,null,2))}
307
- `),r+=n;break}case"ParallelGroupStart":r=`${e.blueBright("\u2696\uFE0F [ PARALLEL GROUP START ]")} ${e.white(`Level ${o.groupLevel}`)}
308
- `,r+=`${e.white("Steps:")} ${e.yellow(o.stepsCount.toString())} ${e.gray(`(${o.stepTypes.join(", ")})`)}
309
- `,r+=n;break;case"ParallelGroupComplete":r=`${e.blueBright("\u2705 [ PARALLEL GROUP COMPLETE ]")} ${e.white(`Level ${o.groupLevel}`)}`,r+=` ${e.magenta(`in ${We(o.executionTime)}`)}
310
- `,r+=`${e.white("Steps Executed:")} ${e.yellow(o.stepsCount.toString())}
311
- `,r+=n;break;case"BranchEvaluation":r=`${e.yellow("\u{1F500} [ BRANCH EVALUATION ]")}
312
- `,r+=`${e.white("Branch Value:")} ${e.cyan(JSON.stringify(o.branchValue))}
313
- `,r+=`${e.white("Has Matching Branch:")} ${o.hasMatchingBranch?e.green("yes"):e.red("no")}
314
- `,o.hasMatchingBranch&&(r+=`${e.white("Branch Steps:")} ${e.yellow(o.branchStepsCount.toString())}
315
- `),r+=n;break;case"FlowComplete":r=`
295
+ ${t}`,o+=`${e.white("Input Fields:")} ${e.cyan(r.inputFields.join(", "))}
296
+ `,o+=`${e.white("Total Steps:")} ${e.yellow(r.totalSteps.toString())}
297
+ `,o+=`${e.white("Parallel Groups:")} ${e.yellow(r.parallelGroups.toString())}
298
+ `,o+=`${e.white("Max Parallelism:")} ${e.yellow(r.maxParallelism.toString())}
299
+ `,o+=`${e.white("Auto-Parallel:")} ${r.autoParallelEnabled?e.green("enabled"):e.red("disabled")}
300
+ `,o+=t;break;case"StepStart":{let i=r.stepType==="execute"?"\u26A1":r.stepType==="map"?"\u{1F504}":r.stepType==="merge"?"\u{1F500}":r.stepType==="parallel"?"\u2696\uFE0F":"\u{1F4CB}";o=`${e.greenBright(`${i} [ STEP ${r.stepIndex} START ]`)} ${e.white(`(${r.stepType})`)}`,r.nodeName&&(o+=` ${e.cyanBright(`Node: ${r.nodeName}`)}`),o+=`
301
+ `,r.dependencies.length>0&&(o+=`${e.white("Dependencies:")} ${e.gray(r.dependencies.join(", "))}
302
+ `),r.produces.length>0&&(o+=`${e.white("Produces:")} ${e.cyan(r.produces.join(", "))}
303
+ `),o+=`${e.white("State:")} ${e.gray(Ve(r.state,!0))}
304
+ `,o+=n;break}case"StepComplete":{let i=(r.stepType==="execute"||r.stepType==="map"||r.stepType==="merge"||r.stepType==="parallel","\u2705");o=`${e.greenBright(`${i} [ STEP ${r.stepIndex} COMPLETE ]`)} ${e.white(`(${r.stepType})`)}`,r.nodeName&&(o+=` ${e.cyanBright(`Node: ${r.nodeName}`)}`),o+=` ${e.magenta(`in ${We(r.executionTime)}`)}
305
+ `,r.newFields&&r.newFields.length>0&&(o+=`${e.white("New Fields:")} ${e.green(r.newFields.join(", "))}
306
+ `),r.result&&r.nodeName&&(o+=`${e.white("Result:")} ${e.yellow(JSON.stringify(r.result,null,2))}
307
+ `),o+=n;break}case"ParallelGroupStart":o=`${e.blueBright("\u2696\uFE0F [ PARALLEL GROUP START ]")} ${e.white(`Level ${r.groupLevel}`)}
308
+ `,o+=`${e.white("Steps:")} ${e.yellow(r.stepsCount.toString())} ${e.gray(`(${r.stepTypes.join(", ")})`)}
309
+ `,o+=n;break;case"ParallelGroupComplete":o=`${e.blueBright("\u2705 [ PARALLEL GROUP COMPLETE ]")} ${e.white(`Level ${r.groupLevel}`)}`,o+=` ${e.magenta(`in ${We(r.executionTime)}`)}
310
+ `,o+=`${e.white("Steps Executed:")} ${e.yellow(r.stepsCount.toString())}
311
+ `,o+=n;break;case"BranchEvaluation":o=`${e.yellow("\u{1F500} [ BRANCH EVALUATION ]")}
312
+ `,o+=`${e.white("Branch Value:")} ${e.cyan(JSON.stringify(r.branchValue))}
313
+ `,o+=`${e.white("Has Matching Branch:")} ${r.hasMatchingBranch?e.green("yes"):e.red("no")}
314
+ `,r.hasMatchingBranch&&(o+=`${e.white("Branch Steps:")} ${e.yellow(r.branchStepsCount.toString())}
315
+ `),o+=n;break;case"FlowComplete":o=`
316
316
  ${e.greenBright("\u2705 [ AXFLOW COMPLETE ]")}
317
- ${t}`,r+=`${e.white("Total Time:")} ${e.magenta(We(o.totalExecutionTime))}
318
- `,r+=`${e.white("Steps Executed:")} ${e.yellow(o.stepsExecuted.toString())}
319
- `,r+=`${e.white("Output Fields:")} ${e.green(o.outputFields.join(", "))}
320
- `,r+=`${e.white("Final State:")} ${e.gray(Ve(o.finalState,!0))}
321
- `,r+=t;break;case"FlowError":r=`
317
+ ${t}`,o+=`${e.white("Total Time:")} ${e.magenta(We(r.totalExecutionTime))}
318
+ `,o+=`${e.white("Steps Executed:")} ${e.yellow(r.stepsExecuted.toString())}
319
+ `,o+=`${e.white("Output Fields:")} ${e.green(r.outputFields.join(", "))}
320
+ `,o+=`${e.white("Final State:")} ${e.gray(Ve(r.finalState,!0))}
321
+ `,o+=t;break;case"FlowError":o=`
322
322
  ${e.redBright("\u274C [ AXFLOW ERROR ]")}
323
- ${t}`,o.stepIndex!==void 0&&(r+=`${e.white("Step:")} ${e.yellow(o.stepIndex.toString())}`,o.stepType&&(r+=` ${e.gray(`(${o.stepType})`)}`),o.nodeName&&(r+=` ${e.cyan(`Node: ${o.nodeName}`)}`),r+=`
324
- `),r+=`${e.white("Error:")} ${e.red(o.error)}
325
- `,o.state&&(r+=`${e.white("State:")} ${e.gray(Ve(o.state,!0))}
326
- `),r+=t;break;default:r=e.gray(JSON.stringify(o,null,2))}s(r)}},Zs=(s=Xs)=>{let e="=".repeat(80),t="-".repeat(40);return n=>{let o="";switch(n.name){case"FlowStart":o=`
323
+ ${t}`,r.stepIndex!==void 0&&(o+=`${e.white("Step:")} ${e.yellow(r.stepIndex.toString())}`,r.stepType&&(o+=` ${e.gray(`(${r.stepType})`)}`),r.nodeName&&(o+=` ${e.cyan(`Node: ${r.nodeName}`)}`),o+=`
324
+ `),o+=`${e.white("Error:")} ${e.red(r.error)}
325
+ `,r.state&&(o+=`${e.white("State:")} ${e.gray(Ve(r.state,!0))}
326
+ `),o+=t;break;default:o=e.gray(JSON.stringify(r,null,2))}s(o)}},ti=(s=ei)=>{let e="=".repeat(80),t="-".repeat(40);return n=>{let r="";switch(n.name){case"FlowStart":r=`
327
327
  [ AXFLOW START ]
328
328
  ${e}
329
- `,o+=`Input Fields: ${n.inputFields.join(", ")}
330
- `,o+=`Total Steps: ${n.totalSteps}
331
- `,o+=`Parallel Groups: ${n.parallelGroups}
332
- `,o+=`Max Parallelism: ${n.maxParallelism}
333
- `,o+=`Auto-Parallel: ${n.autoParallelEnabled?"enabled":"disabled"}
334
- `,o+=`${e}
335
- `;break;case"StepStart":o=`[ STEP ${n.stepIndex} START ] (${n.stepType})`,n.nodeName&&(o+=` Node: ${n.nodeName}`),o+=`
336
- `,n.dependencies.length>0&&(o+=`Dependencies: ${n.dependencies.join(", ")}
337
- `),n.produces.length>0&&(o+=`Produces: ${n.produces.join(", ")}
338
- `),o+=`State: ${Ve(n.state,!0)}
339
- `,o+=`${t}
340
- `;break;case"StepComplete":o=`[ STEP ${n.stepIndex} COMPLETE ] (${n.stepType})`,n.nodeName&&(o+=` Node: ${n.nodeName}`),o+=` in ${We(n.executionTime)}
341
- `,n.newFields&&n.newFields.length>0&&(o+=`New Fields: ${n.newFields.join(", ")}
342
- `),n.result&&n.nodeName&&(o+=`Result: ${JSON.stringify(n.result,null,2)}
343
- `),o+=`${t}
344
- `;break;case"ParallelGroupStart":o=`[ PARALLEL GROUP START ] Level ${n.groupLevel}
345
- `,o+=`Steps: ${n.stepsCount} (${n.stepTypes.join(", ")})
346
- `,o+=`${t}
347
- `;break;case"ParallelGroupComplete":o=`[ PARALLEL GROUP COMPLETE ] Level ${n.groupLevel} in ${We(n.executionTime)}
348
- `,o+=`Steps Executed: ${n.stepsCount}
349
- `,o+=`${t}
350
- `;break;case"BranchEvaluation":o=`[ BRANCH EVALUATION ]
351
- `,o+=`Branch Value: ${JSON.stringify(n.branchValue)}
352
- `,o+=`Has Matching Branch: ${n.hasMatchingBranch?"yes":"no"}
353
- `,n.hasMatchingBranch&&(o+=`Branch Steps: ${n.branchStepsCount}
354
- `),o+=`${t}
355
- `;break;case"FlowComplete":o=`
329
+ `,r+=`Input Fields: ${n.inputFields.join(", ")}
330
+ `,r+=`Total Steps: ${n.totalSteps}
331
+ `,r+=`Parallel Groups: ${n.parallelGroups}
332
+ `,r+=`Max Parallelism: ${n.maxParallelism}
333
+ `,r+=`Auto-Parallel: ${n.autoParallelEnabled?"enabled":"disabled"}
334
+ `,r+=`${e}
335
+ `;break;case"StepStart":r=`[ STEP ${n.stepIndex} START ] (${n.stepType})`,n.nodeName&&(r+=` Node: ${n.nodeName}`),r+=`
336
+ `,n.dependencies.length>0&&(r+=`Dependencies: ${n.dependencies.join(", ")}
337
+ `),n.produces.length>0&&(r+=`Produces: ${n.produces.join(", ")}
338
+ `),r+=`State: ${Ve(n.state,!0)}
339
+ `,r+=`${t}
340
+ `;break;case"StepComplete":r=`[ STEP ${n.stepIndex} COMPLETE ] (${n.stepType})`,n.nodeName&&(r+=` Node: ${n.nodeName}`),r+=` in ${We(n.executionTime)}
341
+ `,n.newFields&&n.newFields.length>0&&(r+=`New Fields: ${n.newFields.join(", ")}
342
+ `),n.result&&n.nodeName&&(r+=`Result: ${JSON.stringify(n.result,null,2)}
343
+ `),r+=`${t}
344
+ `;break;case"ParallelGroupStart":r=`[ PARALLEL GROUP START ] Level ${n.groupLevel}
345
+ `,r+=`Steps: ${n.stepsCount} (${n.stepTypes.join(", ")})
346
+ `,r+=`${t}
347
+ `;break;case"ParallelGroupComplete":r=`[ PARALLEL GROUP COMPLETE ] Level ${n.groupLevel} in ${We(n.executionTime)}
348
+ `,r+=`Steps Executed: ${n.stepsCount}
349
+ `,r+=`${t}
350
+ `;break;case"BranchEvaluation":r=`[ BRANCH EVALUATION ]
351
+ `,r+=`Branch Value: ${JSON.stringify(n.branchValue)}
352
+ `,r+=`Has Matching Branch: ${n.hasMatchingBranch?"yes":"no"}
353
+ `,n.hasMatchingBranch&&(r+=`Branch Steps: ${n.branchStepsCount}
354
+ `),r+=`${t}
355
+ `;break;case"FlowComplete":r=`
356
356
  [ AXFLOW COMPLETE ]
357
357
  ${e}
358
- `,o+=`Total Time: ${We(n.totalExecutionTime)}
359
- `,o+=`Steps Executed: ${n.stepsExecuted}
360
- `,o+=`Output Fields: ${n.outputFields.join(", ")}
361
- `,o+=`Final State: ${Ve(n.finalState,!0)}
362
- `,o+=`${e}
363
- `;break;case"FlowError":o=`
358
+ `,r+=`Total Time: ${We(n.totalExecutionTime)}
359
+ `,r+=`Steps Executed: ${n.stepsExecuted}
360
+ `,r+=`Output Fields: ${n.outputFields.join(", ")}
361
+ `,r+=`Final State: ${Ve(n.finalState,!0)}
362
+ `,r+=`${e}
363
+ `;break;case"FlowError":r=`
364
364
  [ AXFLOW ERROR ]
365
365
  ${e}
366
- `,n.stepIndex!==void 0&&(o+=`Step: ${n.stepIndex}`,n.stepType&&(o+=` (${n.stepType})`),n.nodeName&&(o+=` Node: ${n.nodeName}`),o+=`
367
- `),o+=`Error: ${n.error}
368
- `,n.state&&(o+=`State: ${Ve(n.state,!0)}
369
- `),o+=`${e}
370
- `;break;default:o=JSON.stringify(n,null,2)}s(o)}},ei=Tt(),ti=s=>{let e=new Map;return{logger:s,startTiming:t=>{e.set(t,Date.now())},endTiming:t=>{let n=e.get(t);if(!n)return 0;let o=Date.now()-n;return e.delete(t),o},getCurrentTime:()=>Date.now()}};var Je=class{constructor(e){this.nodeGenerators=e}steps=[];execute(e,t,n){let o=this.nodeGenerators.get(e);if(!o)throw new Error(`Node program for '${e}' not found.`);return this.steps.push(async(r,i)=>{let a=n?.ai??i.mainAi,l=n?.options??i.mainOptions,p=t(r),c=l?.traceLabel?`Node:${e} (${l.traceLabel})`:`Node:${e}`,u;if("forward"in o&&typeof o.forward=="function")u=await o.forward(a,p,{...l,traceLabel:c});else throw new Error(`Node program for '${e}' does not have a forward method`);return{...r,[`${e}Result`]:u}}),this}map(e){return this.steps.push(t=>e(t)),this}async executeSteps(e,t){let n=e;for(let o of this.steps)n=await o(n,t);return n}},Gn=class{constructor(e){this.nodeGenerators=e}steps=[];execute(e,t,n){let o=this.nodeGenerators.get(e);if(!o)throw new Error(`Node program for '${e}' not found.`);return this.steps.push(async(r,i)=>{let a=n?.ai??i.mainAi,l=n?.options??i.mainOptions,p=t(r),c=l?.traceLabel?`Node:${e} (${l.traceLabel})`:`Node:${e}`,u;if("forward"in o&&typeof o.forward=="function")u=await o.forward(a,p,{...l,traceLabel:c});else throw new Error(`Node program for '${e}' does not have a forward method`);return{...r,[`${e}Result`]:u}}),this}map(e){return this.steps.push(t=>e(t)),this}async executeSteps(e,t){let n=e;for(let o of this.steps)n=await o(n,t);return n}};var wt=class s{nodes=new Map;flowDefinition=[];nodeGenerators=new Map;loopStack=[];stepLabels=new Map;branchContext=null;autoParallelConfig;executionPlanner=new Ke;program;nodeUsage=new Map;nodeTraces=new Map;flowLogger;timingLogger;toCamelCase(e){return e.replace(/_([a-z])/g,(t,n)=>n.toUpperCase())}async executeStepsWithLogging(e,t,n,o){let r={...t},i=0;for(let a=0;a<e.length;a++){let l=e[a];if(!l)continue;let p=this.getStepType(l,a),c=this.getStepMetadata(l,a),u=Object.keys(r);this.flowLogger&&this.flowLogger({name:"StepStart",timestamp:Date.now(),stepIndex:a,stepType:p,nodeName:c.nodeName,dependencies:c.dependencies,produces:c.produces,state:{...r}});let d=Date.now();this.timingLogger?.startTiming(`step-${a}`);try{r=await l(r,n),i++;let h=this.timingLogger?.endTiming(`step-${a}`)??Date.now()-d,f=Object.keys(r).filter(x=>!u.includes(x)),A;if(p==="execute"&&c.nodeName&&f.length>0){let x=`${c.nodeName}Result`;A=r[x]}this.flowLogger&&this.flowLogger({name:"StepComplete",timestamp:Date.now(),stepIndex:a,stepType:p,nodeName:c.nodeName,executionTime:h,state:{...r},newFields:f,result:A})}catch(m){throw this.flowLogger&&this.flowLogger({name:"FlowError",timestamp:Date.now(),error:m instanceof Error?m.message:String(m),stepIndex:a,stepType:p,nodeName:c.nodeName,state:{...r}}),m}}return{finalState:r,stepsExecuted:i}}getStepType(e,t){let n=e.toString();return n.includes("nodeName")||n.includes("nodeProgram")?"execute":n.includes("_parallelResults")||n.includes("processBatches")?"parallel":n.includes("branchValue")||n.includes("branches.get")||n.includes("mergeFunction")?"merge":n.includes("transform(")||n.includes("...state,")?"map":n.includes("inputValue")&&n.includes("transformFn")?"derive":n.includes("condition(")&&n.includes("iterations")?n.includes("while")?"while":"feedback":n.includes("branchSteps")||n.includes("currentState")?"branch":"other"}getStepMetadata(e,t){let o=this.executionPlanner.getExecutionPlan().steps.find(a=>a.stepIndex===t);if(o)return{nodeName:o.nodeName,dependencies:o.dependencies,produces:o.produces};let r=e.toString();return{nodeName:this.extractNodeNameFromSource(r),dependencies:[],produces:[]}}extractNodeNameFromSource(e){let t=e.match(/nodeName['"]?\s*[=:]\s*['"](\w+)['"]/);if(t)return t[1];let n=e.match(/nodeProgram\.get\(['"](\w+)['"]\)/);if(n)return n[1]}inferSignatureFromFlow(){let e=this.executionPlanner.getExecutionPlan();if(this.nodeGenerators.size===0&&e.steps.length===0)return Be().input("userInput",Be.string("User input to the flow")).output("flowOutput",Be.string("Output from the flow")).build();let t=new Set,n=new Set;for(let c of e.steps)c.produces.forEach(u=>t.add(u)),c.dependencies.forEach(u=>n.add(u));let o=new Set;for(let c of Array.from(n))t.has(c)||o.add(c);let r=new Set,i=e.steps[e.steps.length-1];if(i&&(i.type==="map"||i.type==="merge")){if(i.produces.forEach(c=>{c.startsWith("_")||r.add(c)}),i.type==="merge"&&i.produces.includes("_mergedResult"))for(let c of e.steps)c.type==="execute"&&c.produces.length>0&&c.produces.forEach(u=>r.add(u))}else for(let c of Array.from(t)){let u=!1;for(let d of e.steps)if(d.dependencies.includes(c)){u=!0;break}if(!u)if(c.endsWith("Result")){let d=c.replace("Result",""),m=this.nodeGenerators.get(d);if(m){let g=m.getSignature().getOutputFields();for(let f of g)r.add(f.name)}else r.add(c)}else r.add(c)}if(o.size===0&&r.size===0){let c=[],u=[];for(let[m,h]of Array.from(this.nodeGenerators)){let g=h.getSignature();for(let f of g.getInputFields()){let A=this.toCamelCase(`${m}_${f.name}`);c.push({name:A,type:f.type,description:f.description,isOptional:f.isOptional,isInternal:f.isInternal})}for(let f of g.getOutputFields()){let A=this.toCamelCase(`${m}_${f.name}`);u.push({name:A,type:f.type,description:f.description,isOptional:f.isOptional,isInternal:f.isInternal})}}let d=new P;return c.length>0?d.setInputFields(c):d.addInputField({name:"userInput",type:{name:"string"},description:"User input to the flow"}),u.length>0?d.setOutputFields(u):d.addOutputField({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),d}let a=new P,l=[];for(let c of Array.from(o))l.push({name:c,type:{name:"string"},description:`Input field: ${c}`});l.length===0&&l.push({name:"userInput",type:{name:"string"},description:"User input to the flow"});let p=[];for(let c of Array.from(r))c.startsWith("_")||p.push({name:c,type:{name:"string"},description:`Output field: ${c}`});return p.length===0&&p.push({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),a.setInputFields(l),a.setOutputFields(p),a}constructor(e){this.autoParallelConfig={enabled:e?.autoParallel!==!1,batchSize:e?.batchSize||10},e?.logger?this.flowLogger=e.logger:e?.debug===!0?this.flowLogger=Tt():this.flowLogger=void 0,this.timingLogger=this.flowLogger?ti(this.flowLogger):void 0}static create(e){return new s(e)}ensureProgram(){let e=this.inferSignatureFromFlow();this.program=new ie(e)}setExamples(e,t){this.ensureProgram(),this.program.setExamples(e,t)}setId(e){this.ensureProgram(),this.program.setId(e)}setParentId(e){this.ensureProgram(),this.program.setParentId(e)}getTraces(){let e=[];for(let[t,n]of Array.from(this.nodeTraces))e.push(...n);return e}setDemos(e){this.ensureProgram(),this.program.setDemos(e)}getUsage(){let e=[];for(let[t,n]of Array.from(this.nodeUsage))e.push(...n);return yt(e)}resetUsage(){this.nodeUsage.clear();for(let[e,t]of Array.from(this.nodeGenerators))t&&"resetUsage"in t&&t.resetUsage()}resetTraces(){this.nodeTraces.clear()}getUsageReport(){let e={};for(let[t,n]of Array.from(this.nodeUsage))e[t]=yt(n);return e}getTracesReport(){let e={};for(let[t,n]of Array.from(this.nodeTraces))e[t]=n;return e}async*streamingForward(e,t,n){yield{version:1,index:0,delta:await this.forward(e,t,n)}}async forward(e,t,n){let o=Date.now();this.timingLogger?.startTiming("flow-execution");let r={};try{this.resetUsage(),this.resetTraces();let i;if(Array.isArray(t)){let c=t.filter(u=>u.role==="user").pop();if(!c)throw new Error("No user message found in values array");i=c.values}else i=t;if(this.nodeGenerators.size>0&&this.ensureProgram(),r={...i},this.flowLogger){let c=this.getExecutionPlan();this.flowLogger({name:"FlowStart",timestamp:o,inputFields:Object.keys(i),totalSteps:c.totalSteps,parallelGroups:c.parallelGroups,maxParallelism:c.maxParallelism,autoParallelEnabled:c.autoParallelEnabled})}let a={mainAi:e,mainOptions:n?{...n,model:n.model?String(n.model):void 0}:void 0},l=n?.autoParallel!==!1&&this.autoParallelConfig.enabled,p=0;if(l){this.executionPlanner.setInitialFields(Object.keys(i));let c=this.executionPlanner.createOptimizedExecution(this.autoParallelConfig.batchSize),u=await this.executeStepsWithLogging(c,r,a,!0);r=u.finalState,p=u.stepsExecuted}else{let c=await this.executeStepsWithLogging(this.flowDefinition,r,a,!1);r=c.finalState,p=c.stepsExecuted}if(this.flowLogger){let c=this.timingLogger?.endTiming("flow-execution")??Date.now()-o;this.flowLogger({name:"FlowComplete",timestamp:Date.now(),totalExecutionTime:c,finalState:r,outputFields:Object.keys(r),stepsExecuted:p})}return r}catch(i){throw this.flowLogger&&this.flowLogger({name:"FlowError",timestamp:Date.now(),error:i instanceof Error?i.message:String(i),state:r}),i}}node(e,t){if(typeof t=="string"||t instanceof P){let n=t;if(!n)throw new Error(`Invalid signature for node '${e}': signature cannot be empty`);this.nodes.set(e,{inputs:{},outputs:{}});let o=Ln(n);this.nodeGenerators.set(e,o),this.ensureProgram(),this.program.register(o)}else if(typeof t=="function"){this.nodes.set(e,{inputs:{},outputs:{}});let n=new t;this.nodeGenerators.set(e,n),this.ensureProgram(),this.program.register(n)}else if(t&&typeof t=="object"&&"forward"in t){this.nodes.set(e,{inputs:{},outputs:{}});let n=t;this.nodeGenerators.set(e,n),this.ensureProgram(),this.program.register(n)}else throw new Error(`Invalid second argument for node '${e}': expected string, AxSignature, AxProgrammable instance, or constructor function`);return this}n(e,t){return this.node(e,t)}map(e,t){if(t?.parallel){let n=Array.isArray(e)?e:[e],o=async r=>{let i=await He(n,async(a,l)=>{let p=a(r);return Promise.resolve(p)},this.autoParallelConfig.batchSize);return i[i.length-1]};if(this.branchContext?.currentBranchValue!==void 0){let r=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];r.push(o),this.branchContext.branches.set(this.branchContext.currentBranchValue,r)}else this.flowDefinition.push(o),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(o,void 0,void 0,"parallel-map")}else{let n=async o=>{if(Array.isArray(e))throw new Error("Array of transforms requires parallel: true option");let r=e(o);return Promise.resolve(r)};if(this.branchContext?.currentBranchValue!==void 0){let o=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];o.push(n),this.branchContext.branches.set(this.branchContext.currentBranchValue,o)}else this.flowDefinition.push(n),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(n,void 0,void 0,"map",e)}return this.nodeGenerators.size>0&&this.ensureProgram(),this}m(e,t){return this.map(e,t)}returns(e){let t=async n=>{let o=e(n);return Promise.resolve(o)};if(this.branchContext?.currentBranchValue!==void 0){let n=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];n.push(t),this.branchContext.branches.set(this.branchContext.currentBranchValue,n)}else this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"map",e);return this.nodeGenerators.size>0&&this.ensureProgram(),this}r(e){return this.returns(e)}label(e){if(this.branchContext?.currentBranchValue!==void 0)throw new Error("Cannot create labels inside branch blocks");return this.stepLabels.set(e,this.flowDefinition.length),this}l(e){return this.label(e)}execute(e,t,n){if(!this.nodes.has(e))throw new Error(`Node '${e}' not found. Make sure to define it with .node() first.`);let o=this.nodeGenerators.get(e);if(!o)throw new Error(`Node program for '${e}' not found.`);let r=async(i,a)=>{let l=n?.ai??a.mainAi,p=n?.options??a.mainOptions,c=t(i),u=p?.traceLabel?`Node:${e} (${p.traceLabel})`:`Node:${e}`,d;if("forward"in o&&typeof o.forward=="function"){if(d=await o.forward(l,c,{...p,traceLabel:u}),"getUsage"in o&&typeof o.getUsage=="function"){let m=o.getUsage();if(m&&m.length>0){let h=this.nodeUsage.get(e)||[];this.nodeUsage.set(e,[...h,...m])}}if("getTraces"in o&&typeof o.getTraces=="function"){let m=o.getTraces();if(m&&m.length>0){let h=this.nodeTraces.get(e)||[];this.nodeTraces.set(e,[...h,...m])}}}else throw new Error(`Node program for '${e}' does not have a forward method`);return{...i,[`${e}Result`]:d}};if(this.branchContext?.currentBranchValue!==void 0){let i=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];i.push(r),this.branchContext.branches.set(this.branchContext.currentBranchValue,i)}else this.flowDefinition.push(r),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(r,e,t);return this.ensureProgram(),this}e(e,t,n){return this.execute(e,t,n)}branch(e){if(this.branchContext)throw new Error("Nested branches are not supported");return this.branchContext={predicate:t=>e(t),branches:new Map,currentBranchValue:void 0},this}b(e){return this.branch(e)}when(e){if(!this.branchContext)throw new Error("when() called without matching branch()");return this.branchContext.currentBranchValue=e,this.branchContext.branches.set(e,[]),this}w(e){return this.when(e)}merge(){if(!this.branchContext)throw new Error("merge() called without matching branch()");let e=this.branchContext;this.branchContext=null;let t=async(n,o)=>{let r=e.predicate(n),i=e.branches.get(r);if(!i)return n;let a=n;for(let l of i)a=await l(a,o);return a};return this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"merge"),this.ensureProgram(),this}mg(){return this.merge()}parallel(e){let t=async(n,o)=>{let r=await He(e,async(i,a)=>{let l=new Je(this.nodeGenerators);return await i(l).executeSteps(n,o)},this.autoParallelConfig.batchSize);return{...n,_parallelResults:r}};return this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"parallel",void 0,void 0),this.ensureProgram(),{merge:(n,o)=>{let r=i=>{let a=i._parallelResults;if(!Array.isArray(a))throw new Error("No parallel results found for merge");let l=o(...a),p={...i};return delete p._parallelResults,p[n]=l,p};return this.flowDefinition.push(r),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(r,void 0,void 0,"merge",void 0,{resultKey:n,mergeFunction:o}),this.ensureProgram(),this}}}p(e){return this.parallel(e)}feedback(e,t,n=10){if(!this.stepLabels.has(t))throw new Error(`Label '${t}' not found. Make sure to define it with .label() before the feedback point.`);let o=this.stepLabels.get(t),r=this.flowDefinition.length;return this.flowDefinition.push(async(i,a)=>{let l=i,p=1,c=`_feedback_${t}_iterations`;for(typeof l[c]!="number"&&(l={...l,[c]:1});e(l)&&p<n;){p++,l={...l,[c]:p};for(let u=o;u<r;u++){let d=this.flowDefinition[u];d&&(l=await d(l,a))}}return l}),this.nodeGenerators.size>0&&this.ensureProgram(),this}fb(e,t,n=10){return this.feedback(e,t,n)}while(e,t=100){let n=this.flowDefinition.length;this.loopStack.push(n);let o=Object.assign(r=>r,{_condition:e,_maxIterations:t,_isLoopStart:!0});return this.flowDefinition.push(o),this.nodeGenerators.size>0&&this.ensureProgram(),this}wh(e,t=100){return this.while(e,t)}endWhile(){if(this.loopStack.length===0)throw new Error("endWhile() called without matching while()");let e=this.loopStack.pop(),t=this.flowDefinition[e];if(!t||!("_isLoopStart"in t))throw new Error("Loop start step not found or invalid");let n=t._condition,o=t._maxIterations,r=this.flowDefinition.splice(e+1);return this.flowDefinition[e]=async(i,a)=>{let l=i,p=0;for(;n(l)&&p<o;){p++;for(let c of r)l=await c(l,a)}if(p>=o&&n(l))throw new Error(`While loop exceeded maximum iterations (${o}). Consider increasing maxIterations or ensuring the loop condition eventually becomes false.`);return l},this.nodeGenerators.size>0&&this.ensureProgram(),this}end(){return this.endWhile()}derive(e,t,n,o){let r=async i=>{let a=i[t];if(a===void 0)throw new Error(`Input field '${t}' not found in state`);let l;if(Array.isArray(a))if(this.autoParallelConfig.enabled){let p=o?.batchSize||this.autoParallelConfig.batchSize;l=await He(a,async(c,u)=>n(c,u,i),p)}else l=a.map((p,c)=>n(p,c,i));else l=n(a,void 0,i);return{...i,[e]:l}};if(this.branchContext?.currentBranchValue!==void 0){let i=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];i.push(r),this.branchContext.branches.set(this.branchContext.currentBranchValue,i)}else this.flowDefinition.push(r),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(r,void 0,void 0,"derive",n,void 0,{inputFieldName:t,outputFieldName:e,batchSize:o?.batchSize});return this.ensureProgram(),this}getExecutionPlan(){let e=this.executionPlanner.getExecutionPlan();return{totalSteps:e.totalSteps,parallelGroups:e.parallelGroups,maxParallelism:e.maxParallelism,autoParallelEnabled:this.autoParallelConfig.enabled,steps:e.steps,groups:e.groups}}getSignature(){return this.ensureProgram(),this.program.getSignature()}nodeExtended(e,t,n){let r=typeof t=="string"?P.create(t):t;if(n.prependInputs)for(let i of n.prependInputs)r=r.prependInputField(i.name,i.type);if(n.appendInputs)for(let i of n.appendInputs)r=r.appendInputField(i.name,i.type);if(n.prependOutputs)for(let i of n.prependOutputs)r=r.prependOutputField(i.name,i.type);if(n.appendOutputs)for(let i of n.appendOutputs)r=r.appendOutputField(i.name,i.type);return this.node(e,r)}nx(e,t,n){return this.nodeExtended(e,t,n)}mapOutput(e){let t=async n=>{let o=e(n);return{...n,...o}};if(this.branchContext?.currentBranchValue!==void 0){let n=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];n.push(t),this.branchContext.branches.set(this.branchContext.currentBranchValue,n)}else this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"map",e);return this.nodeGenerators.size>0&&this.ensureProgram(),this}mo(e){return this.mapOutput(e)}};function Nn(s){return wt.create(s)}var $n=class{apiUrl;containerId=null;constructor(e="http://localhost:2375"){this.apiUrl=e}async pullImage(e){let t=await this.fetchDockerAPI(`/images/create?fromImage=${encodeURIComponent(e)}`,{method:"POST"});if(!t.ok)throw new Error(`Failed to pull image: ${t.statusText}`);await t.text()}async createContainer({imageName:e,volumes:t=[],doNotPullImage:n,tag:o}){let r=t.map(p=>`${p.hostPath}:${p.containerPath}`);n||await this.pullImage(e);let i={Image:e,Tty:!0,OpenStdin:!1,AttachStdin:!1,AttachStdout:!1,AttachStderr:!1,HostConfig:{Binds:r},Labels:{}};o&&(i.Labels["com.example.tag"]=o);let a=await this.fetchDockerAPI("/containers/create",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(i)});if(!a.ok)throw new Error(`Failed to create container: ${a.statusText}`);let l=await a.json();return this.containerId=l.Id,l}async findOrCreateContainer({imageName:e,volumes:t=[],doNotPullImage:n,tag:o}){let i=(await this.listContainers(!0)).filter(l=>l.Labels&&l.Labels["com.example.tag"]===o);if(i&&i.length>0){let l=Math.floor(Math.random()*i.length),p=i[l];if(p)return await this.connectToContainer(p.Id),{Id:p.Id,isNew:!1}}return{Id:(await this.createContainer({imageName:e,volumes:t,doNotPullImage:n,tag:o})).Id,isNew:!0}}async startContainer(){if(!this.containerId)throw new Error("No container created or connected");let e=await this.fetchDockerAPI(`/containers/${this.containerId}/start`,{method:"POST"});if(!e.ok)throw new Error(`Failed to start container: ${e.statusText}`)}async connectToContainer(e){let t=await this.fetchDockerAPI(`/containers/${e}/json`);if(!t.ok)throw new Error(`Failed to connect to container: ${t.statusText}`);this.containerId=e}async stopContainers({tag:e,remove:t,timeout:n=10}){let o=[],r=await this.listContainers(!0),i=e?r.filter(a=>a.Labels["com.example.tag"]===e):r;for(let a of i){if(a.State.Status==="running"){let l=await this.fetchDockerAPI(`/containers/${a.Id}/stop?t=${n}`,{method:"POST"});if(!l.ok){console.warn(`Failed to stop container ${a.Id}: ${l.statusText}`);continue}o.push({Id:a.Id,Action:"stopped"})}if(t){let l=await this.fetchDockerAPI(`/containers/${a.Id}`,{method:"DELETE"});if(!l.ok){console.warn(`Failed to remove container ${a.Id}: ${l.statusText}`);continue}o.push({Id:a.Id,Action:"removed"})}}return o}async listContainers(e=!1){return(await this.fetchDockerAPI(`/containers/json?all=${e}`,{method:"GET"})).json()}async getContainerLogs(){if(!this.containerId)throw new Error("No container created or connected");return(await this.fetchDockerAPI(`/containers/${this.containerId}/logs?stdout=true&stderr=true`,{method:"GET"})).text()}async executeCommand(e){if(console.log("Executing command:",e),!this.containerId)throw new Error("No container created or connected");(await this.getContainerInfo(this.containerId)).State.Status!=="running"&&(await this.startContainer(),await this.waitForContainerToBeRunning(this.containerId));let n=await this.fetchDockerAPI(`/containers/${this.containerId}/exec`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({Cmd:["sh","-c",e],AttachStdout:!0,AttachStderr:!0})});if(!n.ok)throw new Error(`Failed to create exec instance: ${n.statusText}`);let o=await n.json(),r=await this.fetchDockerAPI(`/exec/${o.Id}/start`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({Detach:!1,Tty:!1})});if(!r.ok)throw new Error(`Failed to start exec instance: ${r.statusText}`);return await r.text()}async getContainerInfo(e){let t=await this.fetchDockerAPI(`/containers/${e}/json`);if(!t.ok)throw new Error(`Failed to get container info: ${t.statusText}`);return t.json()}async waitForContainerToBeRunning(e,t=3e4){let n=Date.now();for(;Date.now()-n<t;){if((await this.getContainerInfo(e)).State.Status==="running")return;await new Promise(r=>setTimeout(r,1e3))}throw new Error("Timeout waiting for container to start")}async fetchDockerAPI(e,t){let n=new URL(e,this.apiUrl).toString();return await fetch(n,t)}toFunction(){return{name:"commandExecution",description:"Use this function to execute shell commands, scripts, and programs. This function enables interaction with the file system, running system utilities, and performing tasks that require a shell interface.",parameters:{type:"object",properties:{command:{type:"string",description:'Shell command to execute. eg. `ls -l` or `echo "Hello, World!"`.'}},required:["command"]},func:async({command:e})=>await this.executeCommand(e)}}};var Un=class{aiService;info;func;constructor({ai:e,info:t,func:n}){this.aiService=e,this.info=t,this.func=n}async embedAdapter(e,t){let o=(await this.aiService.embed({texts:[e]},{sessionId:t?.sessionId,abortSignal:t?.abortSignal})).embeddings.at(0);if(!o)throw new Error("Failed to embed text");return this.func.length===2?this.func(o,t):this.func(o)}toFunction(){return{name:this.info.name,description:this.info.description,parameters:{type:"object",properties:{text:{type:"string",description:this.info.argumentDescription}},required:["text"]},func:({text:e},t)=>this.embedAdapter(e,t)}}};var Bn=class{constructor(e,t={}){this.transport=e;this.options=t;this.logger=t.logger??(n=>{console.log(typeof n=="string"?n:JSON.stringify(n,null,2))})}functions=[];activeRequests=new Map;capabilities={};logger;async init(){"connect"in this.transport&&await this.transport.connect?.();let{result:e}=await this.sendRequest("initialize",{protocolVersion:"2024-11-05",capabilities:{roots:{listChanged:!0},sampling:{}},clientInfo:{name:"AxMCPClient",version:"1.0.0"}}),t="2024-11-05";if(e.protocolVersion!==t)throw new Error(`Protocol version mismatch. Expected ${t} but got ${e.protocolVersion}`);e.capabilities.tools&&(this.capabilities.tools=!0),e.capabilities.resources&&(this.capabilities.resources=!0),e.capabilities.prompts&&(this.capabilities.prompts=!0),await this.sendNotification("notifications/initialized"),await this.discoverFunctions()}async discoverFunctions(){if(!this.capabilities.tools)throw new Error("Tools are not supported");let{result:e}=await this.sendRequest("tools/list");this.functions=e.tools.map(t=>{let n=this.options.functionOverrides?.find(r=>r.name===t.name),o=t.inputSchema.properties?{properties:t.inputSchema.properties,required:t.inputSchema.required??[],type:t.inputSchema.type}:void 0;return{name:n?.updates.name??t.name,description:n?.updates.description??t.description,parameters:o,func:async r=>{let{result:i}=await this.sendRequest("tools/call",{name:t.name,arguments:r});return i}}})}async ping(e=3e3){let t=this.sendRequest("ping"),n=new Promise((i,a)=>setTimeout(()=>a(new Error("Ping response timeout exceeded")),e)),o=await Promise.race([t,n]),{result:r}=o;if(typeof r!="object"||r===null||Object.keys(r).length!==0)throw new Error(`Unexpected ping response: ${JSON.stringify(r)}`)}toFunction(){return this.functions}cancelRequest(e){if(this.activeRequests.has(e)){this.sendNotification("notifications/cancelled",{requestId:e,reason:"Client cancelled request"});let t=this.activeRequests.get(e);t&&t.reject(new Error(`Request ${e} cancelled`)),this.activeRequests.delete(e)}}async sendRequest(e,t={}){let n=$(),o={jsonrpc:"2.0",id:n,method:e,params:t},r=new Promise((a,l)=>{this.activeRequests.set(n,{reject:l}),this.transport.send(o).then(p=>{if(this.activeRequests.delete(n),p!==null&&typeof p=="object"&&"error"in p){let c=p;l(new Error(`RPC Error ${c.error.code}: ${c.error.message}`))}else p!==null&&typeof p=="object"&&"result"in p?a({result:p.result}):l(new Error("Invalid response no result or error"))}).catch(p=>{this.activeRequests.delete(n),l(p)})}),{result:i}=await r;return{id:n,result:i}}async sendNotification(e,t={}){let n={jsonrpc:"2.0",method:e,params:t},{debug:o}=this.options;if(o){let r={name:"Notification",id:"mcp_notification",value:`Sending notification: ${JSON.stringify(n,null,2)}`};this.logger(r)}await this.transport.sendNotification(n)}};var qn=class{endpoint=null;sseUrl;eventSource;constructor(e){this.sseUrl=e}async connect(){return new Promise((e,t)=>{this.eventSource=new EventSource(this.sseUrl),this.eventSource.addEventListener("endpoint",n=>{try{let r=JSON.parse(n.data);if(!r.uri)throw new Error("Endpoint URI missing in SSE event data");this.endpoint=r.uri,e()}catch(o){t(o)}}),this.eventSource.onerror=()=>{t(new Error("Failed to establish SSE connection"))}})}async send(e){if(!this.endpoint)throw new Error("HTTPTransport endpoint is not initialized. Call connect() first.");let t=await fetch(this.endpoint,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)});if(!t.ok)throw new Error(`HTTP error ${t.status}: ${t.statusText}`);return t.json()}async sendNotification(e){if(!this.endpoint)throw new Error("HTTPTransport endpoint is not initialized. Call connect() first.");await fetch(this.endpoint,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)})}},zn=class{mcpEndpoint;sessionId;eventSource;pendingRequests=new Map;messageHandler;customHeaders;constructor(e,t){this.mcpEndpoint=e,this.customHeaders={...t?.headers},t?.authorization&&(this.customHeaders.Authorization=t.authorization)}setHeaders(e){this.customHeaders={...e}}setAuthorization(e){this.customHeaders.Authorization=e}getHeaders(){return{...this.customHeaders}}buildHeaders(e){let t={...this.customHeaders,...e};return this.sessionId&&(t["Mcp-Session-Id"]=this.sessionId),t}setMessageHandler(e){this.messageHandler=e}async connect(){return Promise.resolve()}async openListeningStream(){return new Promise((e,t)=>{let n=this.buildHeaders({Accept:"text/event-stream"}),o=new URL(this.mcpEndpoint);if(Object.keys(this.customHeaders).length>0){this.openListeningStreamWithFetch(n).then(e).catch(t);return}this.eventSource=new EventSource(o.toString()),this.eventSource.onopen=()=>{e()},this.eventSource.onmessage=r=>{try{let i=JSON.parse(r.data);this.messageHandler&&this.messageHandler(i)}catch(i){console.error("Failed to parse SSE message:",i)}},this.eventSource.onerror=()=>{t(new Error("Failed to establish SSE connection"))}})}async openListeningStreamWithFetch(e){let t=await fetch(this.mcpEndpoint,{method:"GET",headers:e});if(!t.ok)throw new Error(`Failed to open SSE stream: ${t.status} ${t.statusText}`);if(!t.body)throw new Error("No response body available for SSE stream");let n=t.body.getReader(),o=new TextDecoder,r="",i=async()=>{try{let{done:a,value:l}=await n.read();if(a){n.releaseLock();return}r+=o.decode(l,{stream:!0});let p=r.split(`
371
- `);r=p.pop()||"";for(let c of p)if(c.startsWith("data: ")){let u=c.slice(6);if(u==="[DONE]")return;try{let d=JSON.parse(u);this.messageHandler&&this.messageHandler(d)}catch(d){console.error("Failed to parse SSE data:",d)}}await i()}catch(a){throw n.releaseLock(),a}};await i()}async send(e){let t=this.buildHeaders({"Content-Type":"application/json",Accept:"application/json, text/event-stream"}),n=await fetch(this.mcpEndpoint,{method:"POST",headers:t,body:JSON.stringify(e)});if(!n.ok)throw n.status===404&&this.sessionId?(this.sessionId=void 0,new Error("Session expired. Please reinitialize.")):new Error(`HTTP error ${n.status}: ${n.statusText}`);let o=n.headers.get("Mcp-Session-Id");o&&(this.sessionId=o);let r=n.headers.get("Content-Type");if(r?.includes("text/event-stream"))return this.handleSSEResponse(n,e.id);if(r?.includes("application/json"))return n.json();throw new Error(`Unexpected content type: ${r}`)}async handleSSEResponse(e,t){return new Promise((n,o)=>{let r=e.body?.getReader();if(!r){o(new Error("No response body reader available"));return}let i=new TextDecoder,a="",l=async()=>{try{let{done:p,value:c}=await r.read();if(p){r.releaseLock();return}a+=i.decode(c,{stream:!0});let u=a.split(`
372
- `);a=u.pop()||"";for(let d of u)if(d.startsWith("data: ")){let m=d.slice(6);if(m==="[DONE]")return;try{let h=JSON.parse(m);if("id"in h&&h.id===t){n(h);return}this.messageHandler&&this.messageHandler(h)}catch(h){console.error("Failed to parse SSE data:",h)}}await l()}catch(p){r.releaseLock(),o(p)}};l().catch(o)})}async sendNotification(e){let t=this.buildHeaders({"Content-Type":"application/json",Accept:"application/json, text/event-stream"}),n=await fetch(this.mcpEndpoint,{method:"POST",headers:t,body:JSON.stringify(e)});if(!n.ok)throw n.status===404&&this.sessionId?(this.sessionId=void 0,new Error("Session expired. Please reinitialize.")):new Error(`HTTP error ${n.status}: ${n.statusText}`);n.status!==202&&console.warn(`Unexpected status for notification: ${n.status}`)}async terminateSession(){if(this.sessionId)try{let e=this.buildHeaders({});(await fetch(this.mcpEndpoint,{method:"DELETE",headers:e})).status===405&&console.info("Server does not support explicit session termination")}catch(e){console.error("Failed to terminate session:",e)}finally{this.sessionId=void 0}}close(){this.eventSource&&(this.eventSource.close(),this.eventSource=void 0)}};function Sa(s,e,t,n,o){let r={...s};if(r.parameters){let i=r.parameters.properties?Object.keys(r.parameters.properties):[],l=t.filter(p=>i.includes(p)).filter(p=>p!=="model").filter(p=>!o.excludeFieldsFromPassthrough.includes(p));if(l.length>0){r.parameters=Oa(r.parameters,l);let p=r.func;r.func=async(c,u)=>{let d={};if(Array.isArray(e)){let h=e.filter(g=>g.role==="user").pop();h&&(d=ri(h.values,l))}else d=ri(e,l);let m={...c,...d};return await p(m,u)}}return r}return n&&!o.disableSmartModelRouting&&o.canConfigureSmartModelRouting&&(r.parameters=si(r.parameters,n)),r}var ni=new Error("Agent description must be at least 20 characters (explain in detail what the agent does)"),oi=new Error("Agent definition is the prompt you give to the LLM for the agent. It must be detailed and at least 100 characters"),St=class s{ai;program;functions;agents;disableSmartModelRouting;excludeFieldsFromPassthrough;debug;name;func;constructor({ai:e,name:t,description:n,definition:o,signature:r,agents:i,functions:a},l){let{disableSmartModelRouting:p,excludeFieldsFromPassthrough:c,debug:u}=l??{};if(this.ai=e,this.agents=i,this.functions=a,this.disableSmartModelRouting=p,this.excludeFieldsFromPassthrough=c??[],this.debug=u,!t||t.length<5)throw new Error("Agent name must be at least 10 characters (more descriptive)");if(!n||n.length<20)throw ni;if(o&&o.length<100)throw oi;this.program=new N(r,{...l,description:o??n});for(let m of i??[])this.program.register(m);this.name=t,this.func={name:va(this.name),description:n,parameters:this.program.getSignature().toJSONSchema(),func:()=>this.forward};let d=e?.getModelList();d&&!this.disableSmartModelRouting&&(this.func.parameters=si(this.func.parameters,d))}static create(e,t){let n=P.create(e),{ai:o,name:r,description:i,definition:a,agents:l,functions:p,...c}=t;return new s({ai:o,name:r,description:i,definition:a,signature:n,agents:l,functions:p},c)}setExamples(e,t){this.program.setExamples(e,t)}setId(e){this.program.setId(e)}setParentId(e){this.program.setParentId(e)}getTraces(){return this.program.getTraces()}setDemos(e){this.program.setDemos(e)}getUsage(){return this.program.getUsage()}resetUsage(){this.program.resetUsage()}getFunction(){let e=this.forward.bind(this),t=async(n,o)=>{let{model:r,...i}=n,a=this.ai??o?.ai;if(!a)throw new Error("AI service is required to run the agent");let l=await e(a,i,{...o,model:r}),c=this.program.getSignature().getOutputFields();return Object.keys(l).map(d=>{let m=c.find(h=>h.name===d);return m?`${m.title}: ${l[d]}`:`${d}: ${l[d]}`}).join(`
373
- `)};return{...this.func,func:t}}getFeatures(){return{canConfigureSmartModelRouting:this.ai===void 0,excludeFieldsFromPassthrough:this.excludeFieldsFromPassthrough}}init(e,t,n){let o=this.ai??e,r=o?.getModelList(),a=this.program.getSignature().getInputFields().map(u=>u.name),l=this.getDebug(o,n),p=this.agents?.map(u=>{let d=u.getFeatures(),m={debug:l,disableSmartModelRouting:!!this.disableSmartModelRouting,excludeFieldsFromPassthrough:d.excludeFieldsFromPassthrough,canConfigureSmartModelRouting:d.canConfigureSmartModelRouting};return Sa(u.getFunction(),t,a,r,m)}),c=[...n?.functions??this.functions??[],...p??[]];return{ai:o,functions:c,debug:l}}async forward(e,t,n){let{ai:o,functions:r,debug:i}=this.init(e,t,n);return await this.program.forward(o,t,{...n,debug:i,functions:r})}async*streamingForward(e,t,n){let{ai:o,functions:r,debug:i}=this.init(e,t,n);return yield*this.program.streamingForward(o,t,{...n,debug:i,functions:r})}setDescription(e){if(!e||e.length<20)throw ni;this.program.getSignature().setDescription(e),this.func.description=e}setDefinition(e){if(!e||e.length<100)throw oi;this.program.setDescription(e),this.func.description=e}getSignature(){return this.program.getSignature()}setSignature(e){this.program.setSignature(e)}getDebug(e,t){return t?.debug??this.debug??e?.getOptions()?.debug??!1}};function va(s){return s.split(/[^a-zA-Z0-9]/).map((n,o)=>{let r=n.toLowerCase();return o>0&&r&&r[0]?r[0].toUpperCase()+r.slice(1):r}).join("")}function si(s,e){let t=s?structuredClone(s):{type:"object",properties:{},required:[]};if(t.properties?.model)return t;let n={type:"string",enum:e.map(i=>i.key),description:`The AI model to use for this function call. Available options: ${e.map(i=>`\`${i.key}\` ${i.description}`).join(", ")}`},o={...t.properties??{},model:n},r=[...t.required??[],"model"];return{...t,properties:o,required:r}}function Oa(s,e){let t=structuredClone(s);if(t.properties)for(let n of e)delete t.properties[n];if(Array.isArray(t.required)){let n=t.required.filter(o=>!e.includes(o));Object.defineProperty(t,"required",{value:n,writable:!0,configurable:!0})}return t}function ri(s,e){let t={};for(let n of e)n in s&&(t[n]=s[n]);return t}function ii(s,e){let t=P.create(s),{ai:n,name:o,description:r,definition:i,agents:a,functions:l,...p}=e;return new St({ai:n,name:o,description:r,definition:i,signature:t,agents:a,functions:l},p)}var ai=(s,e)=>{let t=e?.maxHops??3,n=e?.qualityThreshold??.8,o=e?.maxIterations??2,r=e?.qualityTarget??.85,i=e?.disableQualityHealing??!1;return Nn({logger:e?.logger,debug:e?.debug}).node("queryGenerator","originalQuestion:string, previousContext?:string -> searchQuery:string, queryReasoning:string").node("contextualizer","retrievedDocument:string, accumulatedContext?:string -> enhancedContext:string").node("qualityAssessor","currentContext:string, originalQuestion:string -> completenessScore:number, missingAspects:string[]").node("questionDecomposer","complexQuestion:string -> subQuestions:string[], decompositionReason:string").node("evidenceSynthesizer","collectedEvidence:string[], originalQuestion:string -> synthesizedEvidence:string, evidenceGaps:string[]").node("gapAnalyzer","synthesizedEvidence:string, evidenceGaps:string[], originalQuestion:string -> needsMoreInfo:boolean, focusedQueries:string[]").node("answerGenerator","finalContext:string, originalQuestion:string -> comprehensiveAnswer:string, confidenceLevel:number").node("queryRefiner","originalQuestion:string, currentContext:string, missingAspects:string[] -> refinedQuery:string").node("qualityValidator","generatedAnswer:string, userQuery:string -> qualityScore:number, issues:string[]").node("answerHealer","originalAnswer:string, healingDocument:string, issues?:string[] -> healedAnswer:string").map(a=>({...a,maxHops:t,qualityThreshold:n,maxIterations:o,qualityTarget:r,disableQualityHealing:i,currentHop:0,accumulatedContext:"",retrievedContexts:[],completenessScore:0,searchQuery:a.originalQuestion,shouldContinue:!0,iteration:0,allEvidence:[],evidenceSources:[],needsMoreInfo:!0,healingAttempts:0,currentQuality:0,shouldContinueHealing:!0,currentAnswer:"",currentIssues:[]})).while(a=>a.currentHop<a.maxHops&&a.completenessScore<a.qualityThreshold&&a.shouldContinue).map(a=>({...a,currentHop:a.currentHop+1})).execute("queryGenerator",a=>({originalQuestion:a.originalQuestion,previousContext:a.accumulatedContext||void 0})).map(async a=>{let l=a.queryGeneratorResult?.searchQuery||a.searchQuery||a.originalQuestion,p=await s(l);return{...a,retrievalResult:{retrievedDocument:p,retrievalConfidence:.9}}}).execute("contextualizer",a=>({retrievedDocument:a.retrievalResult.retrievedDocument,accumulatedContext:a.accumulatedContext||void 0})).execute("qualityAssessor",a=>({currentContext:a.contextualizerResult.enhancedContext,originalQuestion:a.originalQuestion})).map(a=>({...a,accumulatedContext:a.contextualizerResult.enhancedContext,retrievedContexts:[...a.retrievedContexts,a.retrievalResult.retrievedDocument],completenessScore:a.qualityAssessorResult.completenessScore,searchQuery:a.queryGeneratorResult.searchQuery,shouldContinue:a.qualityAssessorResult.completenessScore<a.qualityThreshold})).branch(a=>a.shouldContinue&&a.currentHop<a.maxHops).when(!0).execute("queryRefiner",a=>({originalQuestion:a.originalQuestion,currentContext:a.accumulatedContext,missingAspects:a.qualityAssessorResult.missingAspects})).map(a=>({...a,searchQuery:a.queryRefinerResult?.refinedQuery||a.searchQuery})).when(!1).map(a=>a).merge().endWhile().map(a=>({...a,allEvidence:a.retrievedContexts.length>0?a.retrievedContexts:[]})).while(a=>a.iteration<a.maxIterations&&a.needsMoreInfo).map(a=>({...a,iteration:a.iteration+1})).branch(a=>a.iteration===1).when(!0).execute("questionDecomposer",a=>({complexQuestion:a.originalQuestion})).map(a=>({...a,currentQueries:a.questionDecomposerResult.subQuestions})).when(!1).map(a=>({...a,currentQueries:a.gapAnalyzerResult?.focusedQueries||[]})).merge().map(async a=>{let l=a.currentQueries||[],p=l.length>0?await Promise.all(l.filter(Boolean).map(c=>s(c))):[];return{...a,retrievalResults:p}}).execute("evidenceSynthesizer",a=>{let l=[...a.allEvidence||[],...a.retrievalResults||[]].filter(Boolean);return{collectedEvidence:l.length>0?l:["No evidence collected yet"],originalQuestion:a.originalQuestion}}).execute("gapAnalyzer",a=>({synthesizedEvidence:a.evidenceSynthesizerResult.synthesizedEvidence,evidenceGaps:a.evidenceSynthesizerResult.evidenceGaps,originalQuestion:a.originalQuestion})).map(a=>({...a,allEvidence:[...a.allEvidence,...a.retrievalResults],evidenceSources:[...a.evidenceSources,`Iteration ${a.iteration} sources`],needsMoreInfo:a.gapAnalyzerResult.needsMoreInfo,synthesizedEvidence:a.evidenceSynthesizerResult.synthesizedEvidence})).endWhile().execute("answerGenerator",a=>({finalContext:a.accumulatedContext||a.synthesizedEvidence||a.allEvidence.join(`
366
+ `,n.stepIndex!==void 0&&(r+=`Step: ${n.stepIndex}`,n.stepType&&(r+=` (${n.stepType})`),n.nodeName&&(r+=` Node: ${n.nodeName}`),r+=`
367
+ `),r+=`Error: ${n.error}
368
+ `,n.state&&(r+=`State: ${Ve(n.state,!0)}
369
+ `),r+=`${e}
370
+ `;break;default:r=JSON.stringify(n,null,2)}s(r)}},ni=wt(),ri=s=>{let e=new Map;return{logger:s,startTiming:t=>{e.set(t,Date.now())},endTiming:t=>{let n=e.get(t);if(!n)return 0;let r=Date.now()-n;return e.delete(t),r},getCurrentTime:()=>Date.now()}};var Je=class{constructor(e){this.nodeGenerators=e}steps=[];execute(e,t,n){let r=this.nodeGenerators.get(e);if(!r)throw new Error(`Node program for '${e}' not found.`);return this.steps.push(async(o,i)=>{let a=n?.ai??i.mainAi,l=n?.options??i.mainOptions,p=t(o),c=l?.traceLabel?`Node:${e} (${l.traceLabel})`:`Node:${e}`,u;if("forward"in r&&typeof r.forward=="function")u=await r.forward(a,p,{...l,traceLabel:c});else throw new Error(`Node program for '${e}' does not have a forward method`);return{...o,[`${e}Result`]:u}}),this}map(e){return this.steps.push(t=>e(t)),this}async executeSteps(e,t){let n=e;for(let r of this.steps)n=await r(n,t);return n}},Nn=class{constructor(e){this.nodeGenerators=e}steps=[];execute(e,t,n){let r=this.nodeGenerators.get(e);if(!r)throw new Error(`Node program for '${e}' not found.`);return this.steps.push(async(o,i)=>{let a=n?.ai??i.mainAi,l=n?.options??i.mainOptions,p=t(o),c=l?.traceLabel?`Node:${e} (${l.traceLabel})`:`Node:${e}`,u;if("forward"in r&&typeof r.forward=="function")u=await r.forward(a,p,{...l,traceLabel:c});else throw new Error(`Node program for '${e}' does not have a forward method`);return{...o,[`${e}Result`]:u}}),this}map(e){return this.steps.push(t=>e(t)),this}async executeSteps(e,t){let n=e;for(let r of this.steps)n=await r(n,t);return n}};var St=class s{nodes=new Map;flowDefinition=[];nodeGenerators=new Map;loopStack=[];stepLabels=new Map;branchContext=null;autoParallelConfig;executionPlanner=new Ke;program;nodeUsage=new Map;nodeTraces=new Map;flowLogger;timingLogger;toCamelCase(e){return e.replace(/_([a-z])/g,(t,n)=>n.toUpperCase())}async executeStepsWithLogging(e,t,n,r){let o={...t},i=0;for(let a=0;a<e.length;a++){let l=e[a];if(!l)continue;let p=this.getStepType(l,a),c=this.getStepMetadata(l,a),u=Object.keys(o);this.flowLogger&&this.flowLogger({name:"StepStart",timestamp:Date.now(),stepIndex:a,stepType:p,nodeName:c.nodeName,dependencies:c.dependencies,produces:c.produces,state:{...o}});let d=Date.now();this.timingLogger?.startTiming(`step-${a}`);try{o=await l(o,n),i++;let h=this.timingLogger?.endTiming(`step-${a}`)??Date.now()-d,f=Object.keys(o).filter(x=>!u.includes(x)),A;if(p==="execute"&&c.nodeName&&f.length>0){let x=`${c.nodeName}Result`;A=o[x]}this.flowLogger&&this.flowLogger({name:"StepComplete",timestamp:Date.now(),stepIndex:a,stepType:p,nodeName:c.nodeName,executionTime:h,state:{...o},newFields:f,result:A})}catch(m){throw this.flowLogger&&this.flowLogger({name:"FlowError",timestamp:Date.now(),error:m instanceof Error?m.message:String(m),stepIndex:a,stepType:p,nodeName:c.nodeName,state:{...o}}),m}}return{finalState:o,stepsExecuted:i}}getStepType(e,t){let n=e.toString();return n.includes("nodeName")||n.includes("nodeProgram")?"execute":n.includes("_parallelResults")||n.includes("processBatches")?"parallel":n.includes("branchValue")||n.includes("branches.get")||n.includes("mergeFunction")?"merge":n.includes("transform(")||n.includes("...state,")?"map":n.includes("inputValue")&&n.includes("transformFn")?"derive":n.includes("condition(")&&n.includes("iterations")?n.includes("while")?"while":"feedback":n.includes("branchSteps")||n.includes("currentState")?"branch":"other"}getStepMetadata(e,t){let r=this.executionPlanner.getExecutionPlan().steps.find(a=>a.stepIndex===t);if(r)return{nodeName:r.nodeName,dependencies:r.dependencies,produces:r.produces};let o=e.toString();return{nodeName:this.extractNodeNameFromSource(o),dependencies:[],produces:[]}}extractNodeNameFromSource(e){let t=e.match(/nodeName['"]?\s*[=:]\s*['"](\w+)['"]/);if(t)return t[1];let n=e.match(/nodeProgram\.get\(['"](\w+)['"]\)/);if(n)return n[1]}inferSignatureFromFlow(){let e=this.executionPlanner.getExecutionPlan();if(this.nodeGenerators.size===0&&e.steps.length===0)return Be().input("userInput",Be.string("User input to the flow")).output("flowOutput",Be.string("Output from the flow")).build();let t=new Set,n=new Set;for(let c of e.steps)c.produces.forEach(u=>t.add(u)),c.dependencies.forEach(u=>n.add(u));let r=new Set;for(let c of Array.from(n))t.has(c)||r.add(c);let o=new Set,i=e.steps[e.steps.length-1];if(i&&(i.type==="map"||i.type==="merge")){if(i.produces.forEach(c=>{c.startsWith("_")||o.add(c)}),i.type==="merge"&&i.produces.includes("_mergedResult"))for(let c of e.steps)c.type==="execute"&&c.produces.length>0&&c.produces.forEach(u=>o.add(u))}else for(let c of Array.from(t)){let u=!1;for(let d of e.steps)if(d.dependencies.includes(c)){u=!0;break}if(!u)if(c.endsWith("Result")){let d=c.replace("Result",""),m=this.nodeGenerators.get(d);if(m){let g=m.getSignature().getOutputFields();for(let f of g)o.add(f.name)}else o.add(c)}else o.add(c)}if(r.size===0&&o.size===0){let c=[],u=[];for(let[m,h]of Array.from(this.nodeGenerators)){let g=h.getSignature();for(let f of g.getInputFields()){let A=this.toCamelCase(`${m}_${f.name}`);c.push({name:A,type:f.type,description:f.description,isOptional:f.isOptional,isInternal:f.isInternal})}for(let f of g.getOutputFields()){let A=this.toCamelCase(`${m}_${f.name}`);u.push({name:A,type:f.type,description:f.description,isOptional:f.isOptional,isInternal:f.isInternal})}}let d=new P;return c.length>0?d.setInputFields(c):d.addInputField({name:"userInput",type:{name:"string"},description:"User input to the flow"}),u.length>0?d.setOutputFields(u):d.addOutputField({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),d}let a=new P,l=[];for(let c of Array.from(r))l.push({name:c,type:{name:"string"},description:`Input field: ${c}`});l.length===0&&l.push({name:"userInput",type:{name:"string"},description:"User input to the flow"});let p=[];for(let c of Array.from(o))c.startsWith("_")||p.push({name:c,type:{name:"string"},description:`Output field: ${c}`});return p.length===0&&p.push({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),a.setInputFields(l),a.setOutputFields(p),a}constructor(e){this.autoParallelConfig={enabled:e?.autoParallel!==!1,batchSize:e?.batchSize||10},e?.logger?this.flowLogger=e.logger:e?.debug===!0?this.flowLogger=wt():this.flowLogger=void 0,this.timingLogger=this.flowLogger?ri(this.flowLogger):void 0}static create(e){return new s(e)}ensureProgram(){let e=this.inferSignatureFromFlow();this.program=new ie(e)}setExamples(e,t){this.ensureProgram(),this.program.setExamples(e,t)}setId(e){this.ensureProgram(),this.program.setId(e)}setParentId(e){this.ensureProgram(),this.program.setParentId(e)}getTraces(){let e=[];for(let[t,n]of Array.from(this.nodeTraces))e.push(...n);return e}setDemos(e){this.ensureProgram(),this.program.setDemos(e)}getUsage(){let e=[];for(let[t,n]of Array.from(this.nodeUsage))e.push(...n);return yt(e)}resetUsage(){this.nodeUsage.clear();for(let[e,t]of Array.from(this.nodeGenerators))t&&"resetUsage"in t&&t.resetUsage()}resetTraces(){this.nodeTraces.clear()}getUsageReport(){let e={};for(let[t,n]of Array.from(this.nodeUsage))e[t]=yt(n);return e}getTracesReport(){let e={};for(let[t,n]of Array.from(this.nodeTraces))e[t]=n;return e}async*streamingForward(e,t,n){yield{version:1,index:0,delta:await this.forward(e,t,n)}}async forward(e,t,n){let r=Date.now();this.timingLogger?.startTiming("flow-execution");let o={};try{this.resetUsage(),this.resetTraces();let i;if(Array.isArray(t)){let c=t.filter(u=>u.role==="user").pop();if(!c)throw new Error("No user message found in values array");i=c.values}else i=t;if(this.nodeGenerators.size>0&&this.ensureProgram(),o={...i},this.flowLogger){let c=this.getExecutionPlan();this.flowLogger({name:"FlowStart",timestamp:r,inputFields:Object.keys(i),totalSteps:c.totalSteps,parallelGroups:c.parallelGroups,maxParallelism:c.maxParallelism,autoParallelEnabled:c.autoParallelEnabled})}let a={mainAi:e,mainOptions:n?{...n,model:n.model?String(n.model):void 0}:void 0},l=n?.autoParallel!==!1&&this.autoParallelConfig.enabled,p=0;if(l){this.executionPlanner.setInitialFields(Object.keys(i));let c=this.executionPlanner.createOptimizedExecution(this.autoParallelConfig.batchSize),u=await this.executeStepsWithLogging(c,o,a,!0);o=u.finalState,p=u.stepsExecuted}else{let c=await this.executeStepsWithLogging(this.flowDefinition,o,a,!1);o=c.finalState,p=c.stepsExecuted}if(this.flowLogger){let c=this.timingLogger?.endTiming("flow-execution")??Date.now()-r;this.flowLogger({name:"FlowComplete",timestamp:Date.now(),totalExecutionTime:c,finalState:o,outputFields:Object.keys(o),stepsExecuted:p})}return o}catch(i){throw this.flowLogger&&this.flowLogger({name:"FlowError",timestamp:Date.now(),error:i instanceof Error?i.message:String(i),state:o}),i}}node(e,t){if(typeof t=="string"||t instanceof P){let n=t;if(!n)throw new Error(`Invalid signature for node '${e}': signature cannot be empty`);this.nodes.set(e,{inputs:{},outputs:{}});let r=$n(n);this.nodeGenerators.set(e,r),this.ensureProgram(),this.program.register(r)}else if(typeof t=="function"){this.nodes.set(e,{inputs:{},outputs:{}});let n=new t;this.nodeGenerators.set(e,n),this.ensureProgram(),this.program.register(n)}else if(t&&typeof t=="object"&&"forward"in t){this.nodes.set(e,{inputs:{},outputs:{}});let n=t;this.nodeGenerators.set(e,n),this.ensureProgram(),this.program.register(n)}else throw new Error(`Invalid second argument for node '${e}': expected string, AxSignature, AxProgrammable instance, or constructor function`);return this}n(e,t){return this.node(e,t)}map(e,t){if(t?.parallel){let n=Array.isArray(e)?e:[e],r=async o=>{let i=await He(n,async(a,l)=>{let p=a(o);return Promise.resolve(p)},this.autoParallelConfig.batchSize);return i[i.length-1]};if(this.branchContext?.currentBranchValue!==void 0){let o=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];o.push(r),this.branchContext.branches.set(this.branchContext.currentBranchValue,o)}else this.flowDefinition.push(r),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(r,void 0,void 0,"parallel-map")}else{let n=async r=>{if(Array.isArray(e))throw new Error("Array of transforms requires parallel: true option");let o=e(r);return Promise.resolve(o)};if(this.branchContext?.currentBranchValue!==void 0){let r=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];r.push(n),this.branchContext.branches.set(this.branchContext.currentBranchValue,r)}else this.flowDefinition.push(n),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(n,void 0,void 0,"map",e)}return this.nodeGenerators.size>0&&this.ensureProgram(),this}m(e,t){return this.map(e,t)}returns(e){let t=async n=>{let r=e(n);return Promise.resolve(r)};if(this.branchContext?.currentBranchValue!==void 0){let n=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];n.push(t),this.branchContext.branches.set(this.branchContext.currentBranchValue,n)}else this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"map",e);return this.nodeGenerators.size>0&&this.ensureProgram(),this}r(e){return this.returns(e)}label(e){if(this.branchContext?.currentBranchValue!==void 0)throw new Error("Cannot create labels inside branch blocks");return this.stepLabels.set(e,this.flowDefinition.length),this}l(e){return this.label(e)}execute(e,t,n){if(!this.nodes.has(e))throw new Error(`Node '${e}' not found. Make sure to define it with .node() first.`);let r=this.nodeGenerators.get(e);if(!r)throw new Error(`Node program for '${e}' not found.`);let o=async(i,a)=>{let l=n?.ai??a.mainAi,p=n?.options??a.mainOptions,c=t(i),u=p?.traceLabel?`Node:${e} (${p.traceLabel})`:`Node:${e}`,d;if("forward"in r&&typeof r.forward=="function"){if(d=await r.forward(l,c,{...p,traceLabel:u}),"getUsage"in r&&typeof r.getUsage=="function"){let m=r.getUsage();if(m&&m.length>0){let h=this.nodeUsage.get(e)||[];this.nodeUsage.set(e,[...h,...m])}}if("getTraces"in r&&typeof r.getTraces=="function"){let m=r.getTraces();if(m&&m.length>0){let h=this.nodeTraces.get(e)||[];this.nodeTraces.set(e,[...h,...m])}}}else throw new Error(`Node program for '${e}' does not have a forward method`);return{...i,[`${e}Result`]:d}};if(this.branchContext?.currentBranchValue!==void 0){let i=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];i.push(o),this.branchContext.branches.set(this.branchContext.currentBranchValue,i)}else this.flowDefinition.push(o),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(o,e,t);return this.ensureProgram(),this}e(e,t,n){return this.execute(e,t,n)}branch(e){if(this.branchContext)throw new Error("Nested branches are not supported");return this.branchContext={predicate:t=>e(t),branches:new Map,currentBranchValue:void 0},this}b(e){return this.branch(e)}when(e){if(!this.branchContext)throw new Error("when() called without matching branch()");return this.branchContext.currentBranchValue=e,this.branchContext.branches.set(e,[]),this}w(e){return this.when(e)}merge(){if(!this.branchContext)throw new Error("merge() called without matching branch()");let e=this.branchContext;this.branchContext=null;let t=async(n,r)=>{let o=e.predicate(n),i=e.branches.get(o);if(!i)return n;let a=n;for(let l of i)a=await l(a,r);return a};return this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"merge"),this.ensureProgram(),this}mg(){return this.merge()}parallel(e){let t=async(n,r)=>{let o=await He(e,async(i,a)=>{let l=new Je(this.nodeGenerators);return await i(l).executeSteps(n,r)},this.autoParallelConfig.batchSize);return{...n,_parallelResults:o}};return this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"parallel",void 0,void 0),this.ensureProgram(),{merge:(n,r)=>{let o=i=>{let a=i._parallelResults;if(!Array.isArray(a))throw new Error("No parallel results found for merge");let l=r(...a),p={...i};return delete p._parallelResults,p[n]=l,p};return this.flowDefinition.push(o),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(o,void 0,void 0,"merge",void 0,{resultKey:n,mergeFunction:r}),this.ensureProgram(),this}}}p(e){return this.parallel(e)}feedback(e,t,n=10){if(!this.stepLabels.has(t))throw new Error(`Label '${t}' not found. Make sure to define it with .label() before the feedback point.`);let r=this.stepLabels.get(t),o=this.flowDefinition.length;return this.flowDefinition.push(async(i,a)=>{let l=i,p=1,c=`_feedback_${t}_iterations`;for(typeof l[c]!="number"&&(l={...l,[c]:1});e(l)&&p<n;){p++,l={...l,[c]:p};for(let u=r;u<o;u++){let d=this.flowDefinition[u];d&&(l=await d(l,a))}}return l}),this.nodeGenerators.size>0&&this.ensureProgram(),this}fb(e,t,n=10){return this.feedback(e,t,n)}while(e,t=100){let n=this.flowDefinition.length;this.loopStack.push(n);let r=Object.assign(o=>o,{_condition:e,_maxIterations:t,_isLoopStart:!0});return this.flowDefinition.push(r),this.nodeGenerators.size>0&&this.ensureProgram(),this}wh(e,t=100){return this.while(e,t)}endWhile(){if(this.loopStack.length===0)throw new Error("endWhile() called without matching while()");let e=this.loopStack.pop(),t=this.flowDefinition[e];if(!t||!("_isLoopStart"in t))throw new Error("Loop start step not found or invalid");let n=t._condition,r=t._maxIterations,o=this.flowDefinition.splice(e+1);return this.flowDefinition[e]=async(i,a)=>{let l=i,p=0;for(;n(l)&&p<r;){p++;for(let c of o)l=await c(l,a)}if(p>=r&&n(l))throw new Error(`While loop exceeded maximum iterations (${r}). Consider increasing maxIterations or ensuring the loop condition eventually becomes false.`);return l},this.nodeGenerators.size>0&&this.ensureProgram(),this}end(){return this.endWhile()}derive(e,t,n,r){let o=async i=>{let a=i[t];if(a===void 0)throw new Error(`Input field '${t}' not found in state`);let l;if(Array.isArray(a))if(this.autoParallelConfig.enabled){let p=r?.batchSize||this.autoParallelConfig.batchSize;l=await He(a,async(c,u)=>n(c,u,i),p)}else l=a.map((p,c)=>n(p,c,i));else l=n(a,void 0,i);return{...i,[e]:l}};if(this.branchContext?.currentBranchValue!==void 0){let i=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];i.push(o),this.branchContext.branches.set(this.branchContext.currentBranchValue,i)}else this.flowDefinition.push(o),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(o,void 0,void 0,"derive",n,void 0,{inputFieldName:t,outputFieldName:e,batchSize:r?.batchSize});return this.ensureProgram(),this}getExecutionPlan(){let e=this.executionPlanner.getExecutionPlan();return{totalSteps:e.totalSteps,parallelGroups:e.parallelGroups,maxParallelism:e.maxParallelism,autoParallelEnabled:this.autoParallelConfig.enabled,steps:e.steps,groups:e.groups}}getSignature(){return this.ensureProgram(),this.program.getSignature()}nodeExtended(e,t,n){let o=typeof t=="string"?P.create(t):t;if(n.prependInputs)for(let i of n.prependInputs)o=o.prependInputField(i.name,i.type);if(n.appendInputs)for(let i of n.appendInputs)o=o.appendInputField(i.name,i.type);if(n.prependOutputs)for(let i of n.prependOutputs)o=o.prependOutputField(i.name,i.type);if(n.appendOutputs)for(let i of n.appendOutputs)o=o.appendOutputField(i.name,i.type);return this.node(e,o)}nx(e,t,n){return this.nodeExtended(e,t,n)}mapOutput(e){let t=async n=>{let r=e(n);return{...n,...r}};if(this.branchContext?.currentBranchValue!==void 0){let n=this.branchContext.branches.get(this.branchContext.currentBranchValue)||[];n.push(t),this.branchContext.branches.set(this.branchContext.currentBranchValue,n)}else this.flowDefinition.push(t),this.autoParallelConfig.enabled&&this.executionPlanner.addExecutionStep(t,void 0,void 0,"map",e);return this.nodeGenerators.size>0&&this.ensureProgram(),this}mo(e){return this.mapOutput(e)}};function Un(s){return St.create(s)}var Bn=class{apiUrl;containerId=null;constructor(e="http://localhost:2375"){this.apiUrl=e}async pullImage(e){let t=await this.fetchDockerAPI(`/images/create?fromImage=${encodeURIComponent(e)}`,{method:"POST"});if(!t.ok)throw new Error(`Failed to pull image: ${t.statusText}`);await t.text()}async createContainer({imageName:e,volumes:t=[],doNotPullImage:n,tag:r}){let o=t.map(p=>`${p.hostPath}:${p.containerPath}`);n||await this.pullImage(e);let i={Image:e,Tty:!0,OpenStdin:!1,AttachStdin:!1,AttachStdout:!1,AttachStderr:!1,HostConfig:{Binds:o},Labels:{}};r&&(i.Labels["com.example.tag"]=r);let a=await this.fetchDockerAPI("/containers/create",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(i)});if(!a.ok)throw new Error(`Failed to create container: ${a.statusText}`);let l=await a.json();return this.containerId=l.Id,l}async findOrCreateContainer({imageName:e,volumes:t=[],doNotPullImage:n,tag:r}){let i=(await this.listContainers(!0)).filter(l=>l.Labels&&l.Labels["com.example.tag"]===r);if(i&&i.length>0){let l=Math.floor(Math.random()*i.length),p=i[l];if(p)return await this.connectToContainer(p.Id),{Id:p.Id,isNew:!1}}return{Id:(await this.createContainer({imageName:e,volumes:t,doNotPullImage:n,tag:r})).Id,isNew:!0}}async startContainer(){if(!this.containerId)throw new Error("No container created or connected");let e=await this.fetchDockerAPI(`/containers/${this.containerId}/start`,{method:"POST"});if(!e.ok)throw new Error(`Failed to start container: ${e.statusText}`)}async connectToContainer(e){let t=await this.fetchDockerAPI(`/containers/${e}/json`);if(!t.ok)throw new Error(`Failed to connect to container: ${t.statusText}`);this.containerId=e}async stopContainers({tag:e,remove:t,timeout:n=10}){let r=[],o=await this.listContainers(!0),i=e?o.filter(a=>a.Labels["com.example.tag"]===e):o;for(let a of i){if(a.State.Status==="running"){let l=await this.fetchDockerAPI(`/containers/${a.Id}/stop?t=${n}`,{method:"POST"});if(!l.ok){console.warn(`Failed to stop container ${a.Id}: ${l.statusText}`);continue}r.push({Id:a.Id,Action:"stopped"})}if(t){let l=await this.fetchDockerAPI(`/containers/${a.Id}`,{method:"DELETE"});if(!l.ok){console.warn(`Failed to remove container ${a.Id}: ${l.statusText}`);continue}r.push({Id:a.Id,Action:"removed"})}}return r}async listContainers(e=!1){return(await this.fetchDockerAPI(`/containers/json?all=${e}`,{method:"GET"})).json()}async getContainerLogs(){if(!this.containerId)throw new Error("No container created or connected");return(await this.fetchDockerAPI(`/containers/${this.containerId}/logs?stdout=true&stderr=true`,{method:"GET"})).text()}async executeCommand(e){if(console.log("Executing command:",e),!this.containerId)throw new Error("No container created or connected");(await this.getContainerInfo(this.containerId)).State.Status!=="running"&&(await this.startContainer(),await this.waitForContainerToBeRunning(this.containerId));let n=await this.fetchDockerAPI(`/containers/${this.containerId}/exec`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({Cmd:["sh","-c",e],AttachStdout:!0,AttachStderr:!0})});if(!n.ok)throw new Error(`Failed to create exec instance: ${n.statusText}`);let r=await n.json(),o=await this.fetchDockerAPI(`/exec/${r.Id}/start`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({Detach:!1,Tty:!1})});if(!o.ok)throw new Error(`Failed to start exec instance: ${o.statusText}`);return await o.text()}async getContainerInfo(e){let t=await this.fetchDockerAPI(`/containers/${e}/json`);if(!t.ok)throw new Error(`Failed to get container info: ${t.statusText}`);return t.json()}async waitForContainerToBeRunning(e,t=3e4){let n=Date.now();for(;Date.now()-n<t;){if((await this.getContainerInfo(e)).State.Status==="running")return;await new Promise(o=>setTimeout(o,1e3))}throw new Error("Timeout waiting for container to start")}async fetchDockerAPI(e,t){let n=new URL(e,this.apiUrl).toString();return await fetch(n,t)}toFunction(){return{name:"commandExecution",description:"Use this function to execute shell commands, scripts, and programs. This function enables interaction with the file system, running system utilities, and performing tasks that require a shell interface.",parameters:{type:"object",properties:{command:{type:"string",description:'Shell command to execute. eg. `ls -l` or `echo "Hello, World!"`.'}},required:["command"]},func:async({command:e})=>await this.executeCommand(e)}}};var qn=class{aiService;info;func;constructor({ai:e,info:t,func:n}){this.aiService=e,this.info=t,this.func=n}async embedAdapter(e,t){let r=(await this.aiService.embed({texts:[e]},{sessionId:t?.sessionId,abortSignal:t?.abortSignal})).embeddings.at(0);if(!r)throw new Error("Failed to embed text");return this.func.length===2?this.func(r,t):this.func(r)}toFunction(){return{name:this.info.name,description:this.info.description,parameters:{type:"object",properties:{text:{type:"string",description:this.info.argumentDescription}},required:["text"]},func:({text:e},t)=>this.embedAdapter(e,t)}}};var zn=class{constructor(e,t={}){this.transport=e;this.options=t;this.logger=t.logger??(n=>{console.log(typeof n=="string"?n:JSON.stringify(n,null,2))})}functions=[];activeRequests=new Map;capabilities={};logger;async init(){"connect"in this.transport&&await this.transport.connect?.();let{result:e}=await this.sendRequest("initialize",{protocolVersion:"2024-11-05",capabilities:{roots:{listChanged:!0},sampling:{}},clientInfo:{name:"AxMCPClient",version:"1.0.0"}}),t="2024-11-05";if(e.protocolVersion!==t)throw new Error(`Protocol version mismatch. Expected ${t} but got ${e.protocolVersion}`);e.capabilities.tools&&(this.capabilities.tools=!0),e.capabilities.resources&&(this.capabilities.resources=!0),e.capabilities.prompts&&(this.capabilities.prompts=!0),await this.sendNotification("notifications/initialized"),await this.discoverFunctions()}async discoverFunctions(){if(!this.capabilities.tools)throw new Error("Tools are not supported");let{result:e}=await this.sendRequest("tools/list");this.functions=e.tools.map(t=>{let n=this.options.functionOverrides?.find(o=>o.name===t.name),r=t.inputSchema.properties?{properties:t.inputSchema.properties,required:t.inputSchema.required??[],type:t.inputSchema.type}:void 0;return{name:n?.updates.name??t.name,description:n?.updates.description??t.description,parameters:r,func:async o=>{let{result:i}=await this.sendRequest("tools/call",{name:t.name,arguments:o});return i}}})}async ping(e=3e3){let t=this.sendRequest("ping"),n=new Promise((i,a)=>setTimeout(()=>a(new Error("Ping response timeout exceeded")),e)),r=await Promise.race([t,n]),{result:o}=r;if(typeof o!="object"||o===null||Object.keys(o).length!==0)throw new Error(`Unexpected ping response: ${JSON.stringify(o)}`)}toFunction(){return this.functions}cancelRequest(e){if(this.activeRequests.has(e)){this.sendNotification("notifications/cancelled",{requestId:e,reason:"Client cancelled request"});let t=this.activeRequests.get(e);t&&t.reject(new Error(`Request ${e} cancelled`)),this.activeRequests.delete(e)}}async sendRequest(e,t={}){let n=N(),r={jsonrpc:"2.0",id:n,method:e,params:t},o=new Promise((a,l)=>{this.activeRequests.set(n,{reject:l}),this.transport.send(r).then(p=>{if(this.activeRequests.delete(n),p!==null&&typeof p=="object"&&"error"in p){let c=p;l(new Error(`RPC Error ${c.error.code}: ${c.error.message}`))}else p!==null&&typeof p=="object"&&"result"in p?a({result:p.result}):l(new Error("Invalid response no result or error"))}).catch(p=>{this.activeRequests.delete(n),l(p)})}),{result:i}=await o;return{id:n,result:i}}async sendNotification(e,t={}){let n={jsonrpc:"2.0",method:e,params:t},{debug:r}=this.options;if(r){let o={name:"Notification",id:"mcp_notification",value:`Sending notification: ${JSON.stringify(n,null,2)}`};this.logger(o)}await this.transport.sendNotification(n)}};var jn=class{endpoint=null;sseUrl;eventSource;constructor(e){this.sseUrl=e}async connect(){return new Promise((e,t)=>{this.eventSource=new EventSource(this.sseUrl),this.eventSource.addEventListener("endpoint",n=>{try{let o=JSON.parse(n.data);if(!o.uri)throw new Error("Endpoint URI missing in SSE event data");this.endpoint=o.uri,e()}catch(r){t(r)}}),this.eventSource.onerror=()=>{t(new Error("Failed to establish SSE connection"))}})}async send(e){if(!this.endpoint)throw new Error("HTTPTransport endpoint is not initialized. Call connect() first.");let t=await fetch(this.endpoint,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)});if(!t.ok)throw new Error(`HTTP error ${t.status}: ${t.statusText}`);return t.json()}async sendNotification(e){if(!this.endpoint)throw new Error("HTTPTransport endpoint is not initialized. Call connect() first.");await fetch(this.endpoint,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(e)})}},Hn=class{mcpEndpoint;sessionId;eventSource;pendingRequests=new Map;messageHandler;customHeaders;constructor(e,t){this.mcpEndpoint=e,this.customHeaders={...t?.headers},t?.authorization&&(this.customHeaders.Authorization=t.authorization)}setHeaders(e){this.customHeaders={...e}}setAuthorization(e){this.customHeaders.Authorization=e}getHeaders(){return{...this.customHeaders}}buildHeaders(e){let t={...this.customHeaders,...e};return this.sessionId&&(t["Mcp-Session-Id"]=this.sessionId),t}setMessageHandler(e){this.messageHandler=e}async connect(){return Promise.resolve()}async openListeningStream(){return new Promise((e,t)=>{let n=this.buildHeaders({Accept:"text/event-stream"}),r=new URL(this.mcpEndpoint);if(Object.keys(this.customHeaders).length>0){this.openListeningStreamWithFetch(n).then(e).catch(t);return}this.eventSource=new EventSource(r.toString()),this.eventSource.onopen=()=>{e()},this.eventSource.onmessage=o=>{try{let i=JSON.parse(o.data);this.messageHandler&&this.messageHandler(i)}catch(i){console.error("Failed to parse SSE message:",i)}},this.eventSource.onerror=()=>{t(new Error("Failed to establish SSE connection"))}})}async openListeningStreamWithFetch(e){let t=await fetch(this.mcpEndpoint,{method:"GET",headers:e});if(!t.ok)throw new Error(`Failed to open SSE stream: ${t.status} ${t.statusText}`);if(!t.body)throw new Error("No response body available for SSE stream");let n=t.body.getReader(),r=new TextDecoder,o="",i=async()=>{try{let{done:a,value:l}=await n.read();if(a){n.releaseLock();return}o+=r.decode(l,{stream:!0});let p=o.split(`
371
+ `);o=p.pop()||"";for(let c of p)if(c.startsWith("data: ")){let u=c.slice(6);if(u==="[DONE]")return;try{let d=JSON.parse(u);this.messageHandler&&this.messageHandler(d)}catch(d){console.error("Failed to parse SSE data:",d)}}await i()}catch(a){throw n.releaseLock(),a}};await i()}async send(e){let t=this.buildHeaders({"Content-Type":"application/json",Accept:"application/json, text/event-stream"}),n=await fetch(this.mcpEndpoint,{method:"POST",headers:t,body:JSON.stringify(e)});if(!n.ok)throw n.status===404&&this.sessionId?(this.sessionId=void 0,new Error("Session expired. Please reinitialize.")):new Error(`HTTP error ${n.status}: ${n.statusText}`);let r=n.headers.get("Mcp-Session-Id");r&&(this.sessionId=r);let o=n.headers.get("Content-Type");if(o?.includes("text/event-stream"))return this.handleSSEResponse(n,e.id);if(o?.includes("application/json"))return n.json();throw new Error(`Unexpected content type: ${o}`)}async handleSSEResponse(e,t){return new Promise((n,r)=>{let o=e.body?.getReader();if(!o){r(new Error("No response body reader available"));return}let i=new TextDecoder,a="",l=async()=>{try{let{done:p,value:c}=await o.read();if(p){o.releaseLock();return}a+=i.decode(c,{stream:!0});let u=a.split(`
372
+ `);a=u.pop()||"";for(let d of u)if(d.startsWith("data: ")){let m=d.slice(6);if(m==="[DONE]")return;try{let h=JSON.parse(m);if("id"in h&&h.id===t){n(h);return}this.messageHandler&&this.messageHandler(h)}catch(h){console.error("Failed to parse SSE data:",h)}}await l()}catch(p){o.releaseLock(),r(p)}};l().catch(r)})}async sendNotification(e){let t=this.buildHeaders({"Content-Type":"application/json",Accept:"application/json, text/event-stream"}),n=await fetch(this.mcpEndpoint,{method:"POST",headers:t,body:JSON.stringify(e)});if(!n.ok)throw n.status===404&&this.sessionId?(this.sessionId=void 0,new Error("Session expired. Please reinitialize.")):new Error(`HTTP error ${n.status}: ${n.statusText}`);n.status!==202&&console.warn(`Unexpected status for notification: ${n.status}`)}async terminateSession(){if(this.sessionId)try{let e=this.buildHeaders({});(await fetch(this.mcpEndpoint,{method:"DELETE",headers:e})).status===405&&console.info("Server does not support explicit session termination")}catch(e){console.error("Failed to terminate session:",e)}finally{this.sessionId=void 0}}close(){this.eventSource&&(this.eventSource.close(),this.eventSource=void 0)}};function Oa(s,e,t,n,r){let o={...s};if(o.parameters){let i=o.parameters.properties?Object.keys(o.parameters.properties):[],l=t.filter(p=>i.includes(p)).filter(p=>p!=="model").filter(p=>!r.excludeFieldsFromPassthrough.includes(p));if(l.length>0){o.parameters=ka(o.parameters,l);let p=o.func;o.func=async(c,u)=>{let d={};if(Array.isArray(e)){let h=e.filter(g=>g.role==="user").pop();h&&(d=ii(h.values,l))}else d=ii(e,l);let m={...c,...d};return await p(m,u)}}return o}return n&&!r.disableSmartModelRouting&&r.canConfigureSmartModelRouting&&(o.parameters=ai(o.parameters,n)),o}var oi=new Error("Agent description must be at least 20 characters (explain in detail what the agent does)"),si=new Error("Agent definition is the prompt you give to the LLM for the agent. It must be detailed and at least 100 characters"),vt=class s{ai;program;functions;agents;disableSmartModelRouting;excludeFieldsFromPassthrough;debug;options;name;func;constructor({ai:e,name:t,description:n,definition:r,signature:o,agents:i,functions:a},l){let{disableSmartModelRouting:p,excludeFieldsFromPassthrough:c,debug:u}=l??{};if(this.ai=e,this.agents=i,this.functions=a,this.disableSmartModelRouting=p,this.excludeFieldsFromPassthrough=c??[],this.debug=u,this.options=l,!t||t.length<5)throw new Error("Agent name must be at least 10 characters (more descriptive)");if(!n||n.length<20)throw oi;if(r&&r.length<100)throw si;this.program=new $(o,{...l,description:r??n});for(let m of i??[])this.program.register(m);this.name=t,this.func={name:Ma(this.name),description:n,parameters:this.program.getSignature().toJSONSchema(),func:()=>this.forward};let d=e?.getModelList();d&&!this.disableSmartModelRouting&&(this.func.parameters=ai(this.func.parameters,d))}static create(e,t){let n=P.create(e),{ai:r,name:o,description:i,definition:a,agents:l,functions:p,...c}=t;return new s({ai:r,name:o,description:i,definition:a,signature:n,agents:l,functions:p},c)}setExamples(e,t){this.program.setExamples(e,t)}setId(e){this.program.setId(e)}setParentId(e){this.program.setParentId(e)}getTraces(){return this.program.getTraces()}setDemos(e){this.program.setDemos(e)}getUsage(){return this.program.getUsage()}resetUsage(){this.program.resetUsage()}getFunction(){let e=this.forward.bind(this),t=async(n,r)=>{let{model:o,...i}=n,a=this.ai??r?.ai;if(!a)throw new Error("AI service is required to run the agent");let l=await e(a,i,{...r,model:o}),c=this.program.getSignature().getOutputFields();return Object.keys(l).map(d=>{let m=c.find(h=>h.name===d);return m?`${m.title}: ${l[d]}`:`${d}: ${l[d]}`}).join(`
373
+ `)};return{...this.func,func:t}}getFeatures(){return{canConfigureSmartModelRouting:this.ai===void 0,excludeFieldsFromPassthrough:this.excludeFieldsFromPassthrough}}init(e,t,n){let r=this.ai??e,o=r?.getModelList(),a=this.program.getSignature().getInputFields().map(u=>u.name),l=this.getDebug(r,n),p=this.agents?.map(u=>{let d=u.getFeatures(),m={debug:l,disableSmartModelRouting:!!this.disableSmartModelRouting,excludeFieldsFromPassthrough:d.excludeFieldsFromPassthrough,canConfigureSmartModelRouting:d.canConfigureSmartModelRouting};return Oa(u.getFunction(),t,a,o,m)}),c=[...n?.functions??this.functions??[],...p??[]];return{ai:r,functions:c,debug:l}}async forward(e,t,n){let{ai:r,functions:o,debug:i}=this.init(e,t,n),a={...this.options,...n,debug:i,functions:o};return await this.program.forward(r,t,a)}async*streamingForward(e,t,n){let{ai:r,functions:o,debug:i}=this.init(e,t,n),a={...this.options,...n,debug:i,functions:o};return yield*this.program.streamingForward(r,t,a)}setDescription(e){if(!e||e.length<20)throw oi;this.program.getSignature().setDescription(e),this.func.description=e}setDefinition(e){if(!e||e.length<100)throw si;this.program.setDescription(e),this.func.description=e}getSignature(){return this.program.getSignature()}setSignature(e){this.program.setSignature(e)}getDebug(e,t){return t?.debug??this.debug??e?.getOptions()?.debug??!1}};function Ma(s){return s.split(/[^a-zA-Z0-9]/).map((n,r)=>{let o=n.toLowerCase();return r>0&&o&&o[0]?o[0].toUpperCase()+o.slice(1):o}).join("")}function ai(s,e){let t=s?structuredClone(s):{type:"object",properties:{},required:[]};if(t.properties?.model)return t;let n={type:"string",enum:e.map(i=>i.key),description:`The AI model to use for this function call. Available options: ${e.map(i=>`\`${i.key}\` ${i.description}`).join(", ")}`},r={...t.properties??{},model:n},o=[...t.required??[],"model"];return{...t,properties:r,required:o}}function ka(s,e){let t=structuredClone(s);if(t.properties)for(let n of e)delete t.properties[n];if(Array.isArray(t.required)){let n=t.required.filter(r=>!e.includes(r));Object.defineProperty(t,"required",{value:n,writable:!0,configurable:!0})}return t}function ii(s,e){let t={};for(let n of e)n in s&&(t[n]=s[n]);return t}function li(s,e){let t=typeof s=="string"?P.create(s):s,{ai:n,name:r,description:o,definition:i,agents:a,functions:l,...p}=e;return new vt({ai:n,name:r,description:o,definition:i,signature:t,agents:a,functions:l},p)}var pi=(s,e)=>{let t=e?.maxHops??3,n=e?.qualityThreshold??.8,r=e?.maxIterations??2,o=e?.qualityTarget??.85,i=e?.disableQualityHealing??!1;return Un({logger:e?.logger,debug:e?.debug}).node("queryGenerator","originalQuestion:string, previousContext?:string -> searchQuery:string, queryReasoning:string").node("contextualizer","retrievedDocument:string, accumulatedContext?:string -> enhancedContext:string").node("qualityAssessor","currentContext:string, originalQuestion:string -> completenessScore:number, missingAspects:string[]").node("questionDecomposer","complexQuestion:string -> subQuestions:string[], decompositionReason:string").node("evidenceSynthesizer","collectedEvidence:string[], originalQuestion:string -> synthesizedEvidence:string, evidenceGaps:string[]").node("gapAnalyzer","synthesizedEvidence:string, evidenceGaps:string[], originalQuestion:string -> needsMoreInfo:boolean, focusedQueries:string[]").node("answerGenerator","finalContext:string, originalQuestion:string -> comprehensiveAnswer:string, confidenceLevel:number").node("queryRefiner","originalQuestion:string, currentContext:string, missingAspects:string[] -> refinedQuery:string").node("qualityValidator","generatedAnswer:string, userQuery:string -> qualityScore:number, issues:string[]").node("answerHealer","originalAnswer:string, healingDocument:string, issues?:string[] -> healedAnswer:string").map(a=>({...a,maxHops:t,qualityThreshold:n,maxIterations:r,qualityTarget:o,disableQualityHealing:i,currentHop:0,accumulatedContext:"",retrievedContexts:[],completenessScore:0,searchQuery:a.originalQuestion,shouldContinue:!0,iteration:0,allEvidence:[],evidenceSources:[],needsMoreInfo:!0,healingAttempts:0,currentQuality:0,shouldContinueHealing:!0,currentAnswer:"",currentIssues:[]})).while(a=>a.currentHop<a.maxHops&&a.completenessScore<a.qualityThreshold&&a.shouldContinue).map(a=>({...a,currentHop:a.currentHop+1})).execute("queryGenerator",a=>({originalQuestion:a.originalQuestion,previousContext:a.accumulatedContext||void 0})).map(async a=>{let l=a.queryGeneratorResult?.searchQuery||a.searchQuery||a.originalQuestion,p=await s(l);return{...a,retrievalResult:{retrievedDocument:p,retrievalConfidence:.9}}}).execute("contextualizer",a=>({retrievedDocument:a.retrievalResult.retrievedDocument,accumulatedContext:a.accumulatedContext||void 0})).execute("qualityAssessor",a=>({currentContext:a.contextualizerResult.enhancedContext,originalQuestion:a.originalQuestion})).map(a=>({...a,accumulatedContext:a.contextualizerResult.enhancedContext,retrievedContexts:[...a.retrievedContexts,a.retrievalResult.retrievedDocument],completenessScore:a.qualityAssessorResult.completenessScore,searchQuery:a.queryGeneratorResult.searchQuery,shouldContinue:a.qualityAssessorResult.completenessScore<a.qualityThreshold})).branch(a=>a.shouldContinue&&a.currentHop<a.maxHops).when(!0).execute("queryRefiner",a=>({originalQuestion:a.originalQuestion,currentContext:a.accumulatedContext,missingAspects:a.qualityAssessorResult.missingAspects})).map(a=>({...a,searchQuery:a.queryRefinerResult?.refinedQuery||a.searchQuery})).when(!1).map(a=>a).merge().endWhile().map(a=>({...a,allEvidence:a.retrievedContexts.length>0?a.retrievedContexts:[]})).while(a=>a.iteration<a.maxIterations&&a.needsMoreInfo).map(a=>({...a,iteration:a.iteration+1})).branch(a=>a.iteration===1).when(!0).execute("questionDecomposer",a=>({complexQuestion:a.originalQuestion})).map(a=>({...a,currentQueries:a.questionDecomposerResult.subQuestions})).when(!1).map(a=>({...a,currentQueries:a.gapAnalyzerResult?.focusedQueries||[]})).merge().map(async a=>{let l=a.currentQueries||[],p=l.length>0?await Promise.all(l.filter(Boolean).map(c=>s(c))):[];return{...a,retrievalResults:p}}).execute("evidenceSynthesizer",a=>{let l=[...a.allEvidence||[],...a.retrievalResults||[]].filter(Boolean);return{collectedEvidence:l.length>0?l:["No evidence collected yet"],originalQuestion:a.originalQuestion}}).execute("gapAnalyzer",a=>({synthesizedEvidence:a.evidenceSynthesizerResult.synthesizedEvidence,evidenceGaps:a.evidenceSynthesizerResult.evidenceGaps,originalQuestion:a.originalQuestion})).map(a=>({...a,allEvidence:[...a.allEvidence,...a.retrievalResults],evidenceSources:[...a.evidenceSources,`Iteration ${a.iteration} sources`],needsMoreInfo:a.gapAnalyzerResult.needsMoreInfo,synthesizedEvidence:a.evidenceSynthesizerResult.synthesizedEvidence})).endWhile().execute("answerGenerator",a=>({finalContext:a.accumulatedContext||a.synthesizedEvidence||a.allEvidence.join(`
374
374
  `),originalQuestion:a.originalQuestion})).branch(a=>!a.disableQualityHealing).when(!0).execute("qualityValidator",a=>({generatedAnswer:a.answerGeneratorResult.comprehensiveAnswer,userQuery:a.originalQuestion})).map(a=>({...a,currentAnswer:a.answerGeneratorResult.comprehensiveAnswer,currentQuality:a.qualityValidatorResult.qualityScore,currentIssues:a.qualityValidatorResult.issues,shouldContinueHealing:a.qualityValidatorResult.qualityScore<a.qualityTarget})).while(a=>a.healingAttempts<3&&a.shouldContinueHealing).map(a=>({...a,healingAttempts:a.healingAttempts+1})).map(async a=>{let l=a.currentIssues||[],p=l.length>0?`${a.originalQuestion} addressing issues: ${l.join(", ")}`:`${a.originalQuestion} quality improvement`,c=await s(p);return{...a,healingResult:{healingDocument:c}}}).execute("answerHealer",a=>({originalAnswer:a.currentAnswer,healingDocument:a.healingResult.healingDocument,issues:a.currentIssues})).execute("qualityValidator",a=>({generatedAnswer:a.answerHealerResult.healedAnswer,userQuery:a.originalQuestion})).map(a=>({...a,currentAnswer:a.answerHealerResult.healedAnswer,currentQuality:a.qualityValidatorResult.qualityScore,currentIssues:a.qualityValidatorResult.issues,shouldContinueHealing:a.qualityValidatorResult.qualityScore<a.qualityTarget})).endWhile().when(!1).map(a=>({...a,currentAnswer:a.answerGeneratorResult.comprehensiveAnswer,currentQuality:1,currentIssues:[],shouldContinueHealing:!1})).merge().returns(a=>({finalAnswer:a.currentAnswer,totalHops:a.currentHop,retrievedContexts:a.retrievedContexts,iterationCount:a.iteration,healingAttempts:a.healingAttempts,qualityAchieved:a.currentQuality}))};
375
375
  //# sourceMappingURL=index.cjs.map