@ax-llm/ax 13.0.13 → 13.0.15
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 +11 -11
- package/index.cjs.map +1 -1
- package/index.d.cts +5 -3
- package/index.d.ts +5 -3
- package/index.js +10 -10
- package/index.js.map +1 -1
- package/package.json +1 -1
package/index.js
CHANGED
|
@@ -6,7 +6,7 @@ function B({model:s,modelInfo:e,models:t}){let n=t?.find(l=>l.key===s),o=n&&"mod
|
|
|
6
6
|
`)?`
|
|
7
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 kt=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)}},Ve=class extends TransformStream{constructor(){super(new kt)}};var Zr={maxRetries:3,initialDelayMs:1e3,maxDelayMs:6e4,backoffFactor:2,retryableStatusCodes:[500,408,429,502,503,504]},es=3e4,ts=globalThis.TextDecoderStream??Ve,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=q(),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
8
|
`)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},re=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}},J=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}},W=class extends U{constructor(e,t,n,o){super(e,t,n,void 0,o),this.name=this.constructor.name}},V=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}},se=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}},pe=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}},Q=class extends U{constructor(e,t,n,o){super("Authentication failed",e,t,n,o),this.name=this.constructor.name}},S=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=q()}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()}};async function ao(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 lo(s,e){return Math.min(e.maxDelayMs,e.initialDelayMs*e.backoffFactor**s)*(.75+Math.random()*.5)}function ns(){return{startTime:Date.now(),retryCount:0}}function po(s){s.retryCount++,s.lastRetryTime=Date.now()}function co(s,e,t,n){return t>=n.maxRetries?!1:e&&n.retryableStatusCodes.includes(e)?!0:s instanceof J&&!(s instanceof Q)}var D=async(s,e)=>{let t={...Zr,...s.retry},n=s.timeout??es,o=ns(),r,i=new URL(s.url),a=`${[i.pathname,s.name].filter(Boolean).join("/").replace(/\/+/g,"/")}${i.search}`,l=new URL(a,i),p=q();if(s.validateRequest&&!await s.validateRequest(e))throw new W("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 d=new AbortController;if(s.abortSignal){if(s.abortSignal.aborted)throw new pe(l.href,s.abortSignal.reason,e,{metrics:o});let u=()=>{d.abort(s.abortSignal.reason||"User aborted request")};s.abortSignal.addEventListener("abort",u,{once:!0});let m=d.abort.bind(d);d.abort=g=>{s.abortSignal.removeEventListener("abort",u),m(g)}}r=setTimeout(()=>{d.abort("Request timeout")},n);try{let u=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:d.signal});if(clearTimeout(r),u.status===401||u.status===403){let h=await ao(u);throw new Q(l.href,e,h,{metrics:o})}if(u.status>=400&&co(new Error,u.status,c,t)){let h=lo(c,t);c++,po(o),s.span?.addEvent("retry",{attempt:c,delay:h,status:u.status,"metrics.startTime":o.startTime,"metrics.retryCount":o.retryCount,"metrics.lastRetryTime":o.lastRetryTime}),await new Promise(y=>setTimeout(y,h));continue}if(u.status>=400){let h=await ao(u);throw new re(u.status,u.statusText,l.href,e,h,{metrics:o})}if(!s.stream){let h=await u.json();if(s.validateResponse&&!await s.validateResponse(h))throw new W("Invalid response data",l.href,e,{validation:"response"});return s.span?.setAttributes({"response.time":Date.now()-o.startTime,"response.retries":o.retryCount}),h}if(!u.body)throw new W("Response body is null",l.href,e,{metrics:o});let m,g=0,A=new TransformStream({transform(h,y){m=h,g++,o.streamChunks=g,o.lastChunkTime=Date.now(),y.enqueue(h),s.span?.addEvent("stream.chunk",{"stream.chunks":g,"stream.duration":Date.now()-o.startTime,"response.retries":o.retryCount})}}),x=!1;return new ReadableStream({start(h){let y=u.body.pipeThrough(new ts).pipeThrough(new Ke).pipeThrough(A).getReader();async function f(){try{for(;;){let{done:R,value:b}=await y.read();if(R){x||(x=!0,h.close());break}if(x)break;h.enqueue(b)}}catch(R){let b=R,w={...o,streamDuration:Date.now()-o.startTime};throw b.name==="AbortError"||b.message?.includes("aborted")?h.error(new V(l.href,e,m,{streamMetrics:w})):b instanceof TypeError&&b.message.includes("cancelled")?h.error(new V(l.href,e,m,{streamMetrics:w,cancelReason:"Stream cancelled by client"})):h.error(new J(b,l.href,e,"[ReadableStream - consumed during streaming]",{streamMetrics:w})),b}finally{clearTimeout(r),y.releaseLock()}}f()},cancel(){x=!0}})}catch(u){if(u instanceof Error&&u.name==="AbortError")throw s.abortSignal?.aborted?new pe(l.href,s.abortSignal.reason,e,{metrics:o}):new se(l.href,n,e,{metrics:o});if(s.span?.isRecording()&&(s.span.recordException(u),s.span.setAttributes({"error.time":Date.now()-o.startTime,"error.retries":o.retryCount})),u instanceof J&&co(u,void 0,c,t)){let m=lo(c,t);c++,po(o),s.span?.addEvent("retry",{attempt:c,delay:m,error:u.message,"metrics.startTime":o.startTime,"metrics.retryCount":o.retryCount,"metrics.lastRetryTime":o.lastRetryTime}),await new Promise(g=>setTimeout(g,m));continue}throw u instanceof U&&(u.context.metrics=o),u}finally{r!==void 0&&clearTimeout(r)}}};import{context as jo,SpanKind as Ho}from"@opentelemetry/api";var
|
|
9
|
+
`)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}};async function ao(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 lo(s,e){return Math.min(e.maxDelayMs,e.initialDelayMs*e.backoffFactor**s)*(.75+Math.random()*.5)}function ns(){return{startTime:Date.now(),retryCount:0}}function po(s){s.retryCount++,s.lastRetryTime=Date.now()}function co(s,e,t,n){return t>=n.maxRetries?!1:e&&n.retryableStatusCodes.includes(e)?!0:s instanceof J&&!(s instanceof Q)}var D=async(s,e)=>{let t={...Zr,...s.retry},n=s.timeout??es,o=ns(),r,i=new URL(s.url),a=`${[i.pathname,s.name].filter(Boolean).join("/").replace(/\/+/g,"/")}${i.search}`,l=new URL(a,i),p=q();if(s.validateRequest&&!await s.validateRequest(e))throw new W("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 d=new AbortController;if(s.abortSignal){if(s.abortSignal.aborted)throw new pe(l.href,s.abortSignal.reason,e,{metrics:o});let u=()=>{d.abort(s.abortSignal.reason||"User aborted request")};s.abortSignal.addEventListener("abort",u,{once:!0});let m=d.abort.bind(d);d.abort=g=>{s.abortSignal.removeEventListener("abort",u),m(g)}}r=setTimeout(()=>{d.abort("Request timeout")},n);try{let u=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:d.signal});if(clearTimeout(r),u.status===401||u.status===403){let h=await ao(u);throw new Q(l.href,e,h,{metrics:o})}if(u.status>=400&&co(new Error,u.status,c,t)){let h=lo(c,t);c++,po(o),s.span?.addEvent("retry",{attempt:c,delay:h,status:u.status,"metrics.startTime":o.startTime,"metrics.retryCount":o.retryCount,"metrics.lastRetryTime":o.lastRetryTime}),await new Promise(y=>setTimeout(y,h));continue}if(u.status>=400){let h=await ao(u);throw new re(u.status,u.statusText,l.href,e,h,{metrics:o})}if(!s.stream){let h=await u.json();if(s.validateResponse&&!await s.validateResponse(h))throw new W("Invalid response data",l.href,e,{validation:"response"});return s.span?.setAttributes({"response.time":Date.now()-o.startTime,"response.retries":o.retryCount}),h}if(!u.body)throw new W("Response body is null",l.href,e,{metrics:o});let m,g=0,A=new TransformStream({transform(h,y){m=h,g++,o.streamChunks=g,o.lastChunkTime=Date.now(),y.enqueue(h),s.span?.addEvent("stream.chunk",{"stream.chunks":g,"stream.duration":Date.now()-o.startTime,"response.retries":o.retryCount})}}),x=!1;return new ReadableStream({start(h){let y=u.body.pipeThrough(new ts).pipeThrough(new Ke).pipeThrough(A).getReader();async function f(){try{for(;;){let{done:R,value:b}=await y.read();if(R){x||(x=!0,h.close());break}if(x)break;h.enqueue(b)}}catch(R){let b=R,w={...o,streamDuration:Date.now()-o.startTime};throw b.name==="AbortError"||b.message?.includes("aborted")?h.error(new V(l.href,e,m,{streamMetrics:w})):b instanceof TypeError&&b.message.includes("cancelled")?h.error(new V(l.href,e,m,{streamMetrics:w,cancelReason:"Stream cancelled by client"})):h.error(new J(b,l.href,e,"[ReadableStream - consumed during streaming]",{streamMetrics:w})),b}finally{clearTimeout(r),y.releaseLock()}}f()},cancel(){x=!0}})}catch(u){if(u instanceof Error&&u.name==="AbortError")throw s.abortSignal?.aborted?new pe(l.href,s.abortSignal.reason,e,{metrics:o}):new se(l.href,n,e,{metrics:o});if(s.span?.isRecording()&&(s.span.recordException(u),s.span.setAttributes({"error.time":Date.now()-o.startTime,"error.retries":o.retryCount})),u instanceof J&&co(u,void 0,c,t)){let m=lo(c,t);c++,po(o),s.span?.addEvent("retry",{attempt:c,delay:m,error:u.message,"metrics.startTime":o.startTime,"metrics.retryCount":o.retryCount,"metrics.lastRetryTime":o.lastRetryTime}),await new Promise(g=>setTimeout(g,m));continue}throw u instanceof U&&(u.context.metrics=o),u}finally{r!==void 0&&clearTimeout(r)}}};import{context as jo,SpanKind as Ho}from"@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 N=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 oa=new N,uo=s=>{console.log(s)},mo=(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")}
|
|
10
10
|
${n(s.content,"magenta")}`;case"function":return`${n("[ FUNCTION RESULT ]","yellow")}
|
|
11
11
|
${n(s.result??"[No result]","yellowDim")}`;case"user":{let o=`${n("[ USER ]","greenBright")}
|
|
12
12
|
`;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(`
|
|
@@ -101,7 +101,7 @@ ${e}
|
|
|
101
101
|
${e}
|
|
102
102
|
`,n.value.forEach((r,i)=>{o+=`Embedding ${i+1}: [${r.sample.join(", ")}${r.truncated?", ...":""}] (length: ${r.length})`,i<n.value.length-1&&(o+=`
|
|
103
103
|
${e}
|
|
104
|
-
`)});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"},K={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"},Ao=(o=>(o.COMPLETION="completion",o.CHAT="chat",o.RERANK="rerank",o.UNKNOWN="unknown",o))(Ao||{}),ho=(r=>(r.WORKFLOW="workflow",r.TASK="task",r.AGENT="agent",r.TOOL="tool",r.UNKNOWN="unknown",r))(ho||{});var Et=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()}},We=class extends TransformStream{constructor(e,t){super(new Et(e,t))}};function Je(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 xo=(s,e,t,n=_)=>{let o=t?s.filter(i=>i.role!=="system"):[...s];n({name:"ChatRequestChatPrompt",step:e,value:o})};var fo=(s,e=_)=>{if(!s.results)return;let t={name:"ChatResponseResults",value:s.results};e(t)},yo=(s,e,t=_)=>{t({name:"ChatResponseStreamingResult",index:e,value:s})};function Io(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?Je(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 bo=(s,e=_)=>{e({name:"FunctionResults",value:s})},Ro=(s,e,t,n=_)=>{n({name:"FunctionError",index:e,fixingInstructions:t,error:s})},Co=(s,e,t,n=_)=>{n({name:"ValidationError",index:e,fixingInstructions:t,error:s})},To=(s,e,t,n=_)=>{n({name:"AssertionError",index:e,fixingInstructions:t,error:s})},wo=(s,e,t=_)=>{t({name:"RefusalError",index:e,error:s})};var vo=(s,e,t=_)=>{t({name:"EmbedRequest",embedModel:e,value:s})},So=(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)},Oo=(s,e,t,n=_)=>{n({name:"ResultPickerUsed",sampleCount:s,selectedIndex:e,latency:t})};var Pt=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},Ye,Mo=s=>{if(Ye)return Ye;if(s)return Ye=rs(s),Ye};var rs=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)"})}),ko=(s,e,t,n,o)=>{try{if(s.latencyHistogram){let r=Pt({operation:e,ai_service:n,...o?{model:o}:{}});s.latencyHistogram.record(t,r)}}catch(r){console.warn("Failed to record latency metric:",r)}},Eo=(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)},Po=(s,e,t,n)=>{try{if(s.errorCounter){let o=Pt({operation:e,ai_service:t,...n?{model:n}:{}});s.errorCounter.add(1,o)}}catch(o){console.warn("Failed to record error metric:",o)}},_o=(s,e,t,n,o)=>{s.errorRateGauge&&s.errorRateGauge.record(t*100,{operation:e,ai_service:n,...o?{model:o}:{}})},Fo=(s,e,t,n)=>{s.requestCounter&&s.requestCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},ce=(s,e,t,n,o)=>{try{let r=Pt({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)}},Go=(s,e,t,n,o)=>{t&&s.streamingRequestsCounter&&s.streamingRequestsCounter.add(1,{operation:e,ai_service:n,...o?{model:o}:{}})},Do=(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)},_t=(s,e,t,n,o)=>{s.requestSizeHistogram&&s.requestSizeHistogram.record(t,{operation:e,ai_service:n,...o?{model:o}:{}})},Ft=(s,e,t,n,o)=>{s.responseSizeHistogram&&s.responseSizeHistogram.record(t,{operation:e,ai_service:n,...o?{model:o}:{}})},Uo=(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)},Gt=(s,e,t,n,o)=>{s.estimatedCostCounter&&s.estimatedCostCounter.add(t,{operation:e,ai_service:n,...o?{model:o}:{}})},No=(s,e,t,n)=>{s.promptLengthHistogram&&s.promptLengthHistogram.record(e,{ai_service:t,...n?{model:n}:{}})},$o=(s,e,t,n)=>{s.contextWindowUsageGauge&&s.contextWindowUsageGauge.record(e,{ai_service:t,...n?{model:n}:{}})},Lo=(s,e,t,n)=>{s.timeoutsCounter&&s.timeoutsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},Bo=(s,e,t,n)=>{s.abortsCounter&&s.abortsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},qo=(s,e,t,n)=>{s.thinkingBudgetUsageCounter&&s.thinkingBudgetUsageCounter.add(e,{ai_service:t,...n?{model:n}:{}})},zo=(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 v=()=>structuredClone({temperature:0,topK:40,topP:.9}),G=()=>structuredClone({temperature:.4,topP:.7,frequencyPenalty:.2}),P=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??k.tracer,this.meter=a.meter??k.meter,this.modelInfo=r,this.models=p,this.id=q();let c=this.getModel(i.model)??i.model,d=this.getEmbedModel(i.embedModel)??i.embedModel;if(this.defaults={model:c,embedModel:d},!i.model||typeof i.model!="string"||i.model==="")throw new Error("No model defined");this.setOptions(a),p&&is(p)}debug=!1;rt;fetch;tracer;meter;timeout;excludeContentFromTrace;models;abortSignal;logger=_;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 Mo(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??k.tracer,this.meter=e.meter??k.meter,this.excludeContentFromTrace=e.excludeContentFromTrace,this.abortSignal=e.abortSignal,this.logger=e.logger??_}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}}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;ko(o,e,t,this.name,r),Eo(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;Fo(o,e,this.name,r),t&&Po(o,e,this.name,r),_o(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&&ce(t,"input",n,this.name,e.model),o&&ce(t,"output",o,this.name,e.model),r&&ce(t,"total",r,this.name,e.model),i&&ce(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&&Do(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;Lo(t,e,this.name,n)}}recordAbortMetric(e){let t=this.getMetricsInstruments();if(t){let n=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Bo(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;Go(o,"chat",a,this.name,r);let{hasImages:l,hasAudio:p}=this.detectMultimodalContent(e);zo(o,l,p,this.name,r);let c=this.calculatePromptLength(e);No(o,c,this.name,r),Uo(o,i?.temperature,i?.maxTokens,this.name,r),t?.thinkingTokenBudget&&this.modelUsage?.tokens?.thoughtsTokens&&qo(o,this.modelUsage.tokens.thoughtsTokens,this.name,r);let d=this.calculateRequestSize(e);if(_t(o,"chat",d,this.name,r),n&&!a){let u=n,m=this.calculateResponseSize(u);if(Ft(o,"chat",m,this.name,r),u.results)for(let x of u.results)x.functionCalls&&this.recordFunctionCallMetrics(x.functionCalls,this.lastUsedChatModel);let g=this.calculateContextWindowUsage(this.lastUsedChatModel,u.modelUsage);g>0&&$o(o,g,this.name,r);let A=this.estimateCost(this.lastUsedChatModel,u.modelUsage);A>0&&Gt(o,"chat",A,this.name,r)}}recordEmbedMetrics(e,t){let n=this.getMetricsInstruments();if(!n)return;let o=this.lastUsedEmbedModel,r=this.calculateRequestSize(e);_t(n,"embed",r,this.name,o);let i=this.calculateResponseSize(t);Ft(n,"embed",i,this.name,o);let a=this.estimateCostByName(o,t.modelUsage);a>0&&Gt(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)&&Dt(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:Ho.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??jo.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,g]=await this.aiImpl.createChatReq(l,o);return r?.isRecording()&&ss(n,r,this.excludeContentFromTrace),await D({name:m.name,url:this.apiURL,headers:await this.buildHeaders(m.headers),stream:t.stream,timeout:this.timeout,debug:i,fetch:this.fetch,span:r,abortSignal:o?.abortSignal??this.abortSignal},g)};i&&xo(l.chatPrompt,o?.stepIndex??0,o?.debugHideSystemPrompt,o?.logger??this.logger);let c=o?.rateLimiter??this.rt,d=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),g=h=>y=>{let f=m(y,h);if(f.sessionId=o?.sessionId,!f.modelUsage){let R=this.aiImpl.getTokenUsage();R&&(f.modelUsage={ai:this.name,model:e,tokens:R})}if(this.modelUsage=f.modelUsage,this.recordTokenUsage(f.modelUsage),r?.isRecording()&&Ko(f,r,this.excludeContentFromTrace),i)for(let R of f.results)yo(R,R.index,o?.logger??this.logger);return f},A=async h=>{r?.isRecording()&&r.end(),i&&Io(h,o?.logger??this.logger)};return d.pipeThrough(new We(g({}),A))}if(!this.aiImpl.createChatResp)throw new Error("generateChatResp not implemented");let u=this.aiImpl.createChatResp(d);if(u.sessionId=o?.sessionId,!u.modelUsage){let m=this.aiImpl.getTokenUsage();m&&(u.modelUsage={ai:this.name,model:e,tokens:m})}return u.modelUsage&&(this.modelUsage=u.modelUsage,this.recordTokenUsage(u.modelUsage)),r?.isRecording()&&(Ko(u,r,this.excludeContentFromTrace),r.end()),i&&fo(u,o?.logger??this.logger),u}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||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:Ho.SERVER,attributes:{[T.LLM_SYSTEM]:this.name,[T.LLM_OPERATION_NAME]:"embeddings",[T.LLM_REQUEST_MODEL]:n}},t?.traceContext??jo.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=n?.debug??this.debug,i={...t,embedModel:e};this.lastUsedEmbedModel=e,r&&vo(i.texts??[],e,n?.logger??this.logger);let a=async()=>{let[c,d]=await this.aiImpl.createEmbedReq(i);return await D({name:c.name,url:this.apiURL,headers:await this.buildHeaders(c.headers),debug:r,fetch:this.fetch,timeout:this.timeout,span:o,abortSignal:n?.abortSignal??this.abortSignal},d)},l=this.rt?await this.rt(a,{modelUsage:this.embedModelUsage}):await a(),p=this.aiImpl.createEmbedResp(l);if(p.sessionId=n?.sessionId,!p.modelUsage){let c=this.aiImpl.getTokenUsage();c&&(p.modelUsage={ai:this.name,model:e,tokens:c})}return this.embedModelUsage=p.modelUsage,this.recordTokenUsage(p.modelUsage),o?.isRecording()&&p.modelUsage?.tokens&&o.addEvent(K.GEN_AI_USAGE,{[T.LLM_USAGE_INPUT_TOKENS]:p.modelUsage.tokens.promptTokens,[T.LLM_USAGE_OUTPUT_TOKENS]:p.modelUsage.tokens.completionTokens??0,[T.LLM_USAGE_TOTAL_TOKENS]:p.modelUsage.tokens.totalTokens}),r&&So(p.embeddings,n?.logger??this.logger),o?.end(),p}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 ss(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(K.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(K.GEN_AI_ASSISTANT_MESSAGE,a)}else if(r.content){let a={};t||(a.content=r.content),e.addEvent(K.GEN_AI_ASSISTANT_MESSAGE,a)}break}case"function":{let i={id:r.functionId};t||(i.content=r.result),e.addEvent(K.GEN_AI_TOOL_MESSAGE,i);break}}let o={};t||(o.content=n.join(`
|
|
104
|
+
`)});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"},K={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"},Ao=(o=>(o.COMPLETION="completion",o.CHAT="chat",o.RERANK="rerank",o.UNKNOWN="unknown",o))(Ao||{}),ho=(r=>(r.WORKFLOW="workflow",r.TASK="task",r.AGENT="agent",r.TOOL="tool",r.UNKNOWN="unknown",r))(ho||{});var Et=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()}},We=class extends TransformStream{constructor(e,t){super(new Et(e,t))}};function Je(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 xo=(s,e,t,n=_)=>{let o=t?s.filter(i=>i.role!=="system"):[...s];n({name:"ChatRequestChatPrompt",step:e,value:o})};var fo=(s,e=_)=>{if(!s.results)return;let t={name:"ChatResponseResults",value:s.results};e(t)},yo=(s,e,t=_)=>{t({name:"ChatResponseStreamingResult",index:e,value:s})};function Io(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?Je(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 bo=(s,e=_)=>{e({name:"FunctionResults",value:s})},Ro=(s,e,t,n=_)=>{n({name:"FunctionError",index:e,fixingInstructions:t,error:s})},Co=(s,e,t,n=_)=>{n({name:"ValidationError",index:e,fixingInstructions:t,error:s})},To=(s,e,t,n=_)=>{n({name:"AssertionError",index:e,fixingInstructions:t,error:s})},wo=(s,e,t=_)=>{t({name:"RefusalError",index:e,error:s})};var vo=(s,e,t=_)=>{t({name:"EmbedRequest",embedModel:e,value:s})},So=(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)},Oo=(s,e,t,n=_)=>{n({name:"ResultPickerUsed",sampleCount:s,selectedIndex:e,latency:t})};var Pt=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},Ye,Mo=s=>{if(Ye)return Ye;if(s)return Ye=rs(s),Ye};var rs=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)"})}),ko=(s,e,t,n,o)=>{try{if(s.latencyHistogram){let r=Pt({operation:e,ai_service:n,...o?{model:o}:{}});s.latencyHistogram.record(t,r)}}catch(r){console.warn("Failed to record latency metric:",r)}},Eo=(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)},Po=(s,e,t,n)=>{try{if(s.errorCounter){let o=Pt({operation:e,ai_service:t,...n?{model:n}:{}});s.errorCounter.add(1,o)}}catch(o){console.warn("Failed to record error metric:",o)}},_o=(s,e,t,n,o)=>{s.errorRateGauge&&s.errorRateGauge.record(t*100,{operation:e,ai_service:n,...o?{model:o}:{}})},Fo=(s,e,t,n)=>{s.requestCounter&&s.requestCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},ce=(s,e,t,n,o)=>{try{let r=Pt({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)}},Go=(s,e,t,n,o)=>{t&&s.streamingRequestsCounter&&s.streamingRequestsCounter.add(1,{operation:e,ai_service:n,...o?{model:o}:{}})},Do=(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)},_t=(s,e,t,n,o)=>{s.requestSizeHistogram&&s.requestSizeHistogram.record(t,{operation:e,ai_service:n,...o?{model:o}:{}})},Ft=(s,e,t,n,o)=>{s.responseSizeHistogram&&s.responseSizeHistogram.record(t,{operation:e,ai_service:n,...o?{model:o}:{}})},Uo=(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)},Gt=(s,e,t,n,o)=>{s.estimatedCostCounter&&s.estimatedCostCounter.add(t,{operation:e,ai_service:n,...o?{model:o}:{}})},No=(s,e,t,n)=>{s.promptLengthHistogram&&s.promptLengthHistogram.record(e,{ai_service:t,...n?{model:n}:{}})},$o=(s,e,t,n)=>{s.contextWindowUsageGauge&&s.contextWindowUsageGauge.record(e,{ai_service:t,...n?{model:n}:{}})},Lo=(s,e,t,n)=>{s.timeoutsCounter&&s.timeoutsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},Bo=(s,e,t,n)=>{s.abortsCounter&&s.abortsCounter.add(1,{operation:e,ai_service:t,...n?{model:n}:{}})},qo=(s,e,t,n)=>{s.thinkingBudgetUsageCounter&&s.thinkingBudgetUsageCounter.add(e,{ai_service:t,...n?{model:n}:{}})},zo=(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 v=()=>structuredClone({temperature:0,topK:40,topP:.9}),G=()=>structuredClone({temperature:.4,topP:.7,frequencyPenalty:.2}),P=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=q();let c=this.getModel(i.model)??i.model,d=this.getEmbedModel(i.embedModel)??i.embedModel;if(this.defaults={model:c,embedModel:d},!i.model||typeof i.model!="string"||i.model==="")throw new Error("No model defined");this.setOptions(a),p&&is(p)}debug=!1;rt;fetch;tracer;meter;timeout;excludeContentFromTrace;models;abortSignal;logger=_;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 Mo(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??_}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}}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;ko(o,e,t,this.name,r),Eo(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;Fo(o,e,this.name,r),t&&Po(o,e,this.name,r),_o(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&&ce(t,"input",n,this.name,e.model),o&&ce(t,"output",o,this.name,e.model),r&&ce(t,"total",r,this.name,e.model),i&&ce(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&&Do(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;Lo(t,e,this.name,n)}}recordAbortMetric(e){let t=this.getMetricsInstruments();if(t){let n=e==="chat"?this.lastUsedChatModel:this.lastUsedEmbedModel;Bo(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;Go(o,"chat",a,this.name,r);let{hasImages:l,hasAudio:p}=this.detectMultimodalContent(e);zo(o,l,p,this.name,r);let c=this.calculatePromptLength(e);No(o,c,this.name,r),Uo(o,i?.temperature,i?.maxTokens,this.name,r),t?.thinkingTokenBudget&&this.modelUsage?.tokens?.thoughtsTokens&&qo(o,this.modelUsage.tokens.thoughtsTokens,this.name,r);let d=this.calculateRequestSize(e);if(_t(o,"chat",d,this.name,r),n&&!a){let u=n,m=this.calculateResponseSize(u);if(Ft(o,"chat",m,this.name,r),u.results)for(let x of u.results)x.functionCalls&&this.recordFunctionCallMetrics(x.functionCalls,this.lastUsedChatModel);let g=this.calculateContextWindowUsage(this.lastUsedChatModel,u.modelUsage);g>0&&$o(o,g,this.name,r);let A=this.estimateCost(this.lastUsedChatModel,u.modelUsage);A>0&&Gt(o,"chat",A,this.name,r)}}recordEmbedMetrics(e,t){let n=this.getMetricsInstruments();if(!n)return;let o=this.lastUsedEmbedModel,r=this.calculateRequestSize(e);_t(n,"embed",r,this.name,o);let i=this.calculateResponseSize(t);Ft(n,"embed",i,this.name,o);let a=this.estimateCostByName(o,t.modelUsage);a>0&&Gt(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)&&Dt(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:Ho.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??jo.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,g]=await this.aiImpl.createChatReq(l,o);return r?.isRecording()&&ss(n,r,this.excludeContentFromTrace),await D({name:m.name,url:this.apiURL,headers:await this.buildHeaders(m.headers),stream:t.stream,timeout:this.timeout,debug:i,fetch:this.fetch,span:r,abortSignal:o?.abortSignal??this.abortSignal},g)};i&&xo(l.chatPrompt,o?.stepIndex??0,o?.debugHideSystemPrompt,o?.logger??this.logger);let c=o?.rateLimiter??this.rt,d=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),g=h=>y=>{let f=m(y,h);if(f.sessionId=o?.sessionId,!f.modelUsage){let R=this.aiImpl.getTokenUsage();R&&(f.modelUsage={ai:this.name,model:e,tokens:R})}if(this.modelUsage=f.modelUsage,this.recordTokenUsage(f.modelUsage),r?.isRecording()&&Ko(f,r,this.excludeContentFromTrace),i)for(let R of f.results)yo(R,R.index,o?.logger??this.logger);return f},A=async h=>{r?.isRecording()&&r.end(),i&&Io(h,o?.logger??this.logger)};return d.pipeThrough(new We(g({}),A))}if(!this.aiImpl.createChatResp)throw new Error("generateChatResp not implemented");let u=this.aiImpl.createChatResp(d);if(u.sessionId=o?.sessionId,!u.modelUsage){let m=this.aiImpl.getTokenUsage();m&&(u.modelUsage={ai:this.name,model:e,tokens:m})}return u.modelUsage&&(this.modelUsage=u.modelUsage,this.recordTokenUsage(u.modelUsage)),r?.isRecording()&&(Ko(u,r,this.excludeContentFromTrace),r.end()),i&&fo(u,o?.logger??this.logger),u}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||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:Ho.SERVER,attributes:{[T.LLM_SYSTEM]:this.name,[T.LLM_OPERATION_NAME]:"embeddings",[T.LLM_REQUEST_MODEL]:n}},t?.traceContext??jo.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=n?.debug??this.debug,i={...t,embedModel:e};this.lastUsedEmbedModel=e,r&&vo(i.texts??[],e,n?.logger??this.logger);let a=async()=>{let[c,d]=await this.aiImpl.createEmbedReq(i);return await D({name:c.name,url:this.apiURL,headers:await this.buildHeaders(c.headers),debug:r,fetch:this.fetch,timeout:this.timeout,span:o,abortSignal:n?.abortSignal??this.abortSignal},d)},l=this.rt?await this.rt(a,{modelUsage:this.embedModelUsage}):await a(),p=this.aiImpl.createEmbedResp(l);if(p.sessionId=n?.sessionId,!p.modelUsage){let c=this.aiImpl.getTokenUsage();c&&(p.modelUsage={ai:this.name,model:e,tokens:c})}return this.embedModelUsage=p.modelUsage,this.recordTokenUsage(p.modelUsage),o?.isRecording()&&p.modelUsage?.tokens&&o.addEvent(K.GEN_AI_USAGE,{[T.LLM_USAGE_INPUT_TOKENS]:p.modelUsage.tokens.promptTokens,[T.LLM_USAGE_OUTPUT_TOKENS]:p.modelUsage.tokens.completionTokens??0,[T.LLM_USAGE_TOTAL_TOKENS]:p.modelUsage.tokens.totalTokens}),r&&So(p.embeddings,n?.logger??this.logger),o?.end(),p}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 ss(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(K.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(K.GEN_AI_ASSISTANT_MESSAGE,a)}else if(r.content){let a={};t||(a.content=r.content),e.addEvent(K.GEN_AI_ASSISTANT_MESSAGE,a)}break}case"function":{let i={id:r.functionId};t||(i.content=r.result),e.addEvent(K.GEN_AI_TOOL_MESSAGE,i);break}}let o={};t||(o.content=n.join(`
|
|
105
105
|
`)),e.addEvent(K.GEN_AI_USER_MESSAGE,o)}function Ko(s,e,t){if(s.modelUsage?.tokens){let n=s.modelUsage.tokens.thoughtsTokens?{[T.LLM_USAGE_THOUGHTS_TOKENS]:s.modelUsage.tokens.thoughtsTokens}:{};e.addEvent(K.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(K.GEN_AI_CHOICE,{finish_reason:o.finishReason,index:n,message:JSON.stringify(i,null,2)})}}function Dt(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 is(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 Qe=(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))(Qe||{}),Ut=(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))(Ut||{});var Xe=[{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 Wo=()=>structuredClone({model:"claude-3-7-sonnet-latest",maxTokens:4e4,thinkingTokenBudgetLevels:{minimal:1024,low:5e3,medium:1e4,high:2e4,highest:32e3},...v()}),as=()=>structuredClone({model:"claude-3-7-sonnet",maxTokens:4e4,thinkingTokenBudgetLevels:{minimal:1024,low:5e3,medium:1e4,high:2e4,highest:32e3},...v()}),Nt=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(f=>f.role==="system").map(f=>({type:"text",text:f.content,...f.cache?{cache:{type:"ephemeral"}}:{}})),l=e.chatPrompt.filter(f=>f.role!=="system"),p=ls(l),c=e.functions?.map(f=>({name:f.name,description:f.description,input_schema:f.parameters})),d=e.modelConfig?.maxTokens??this.config.maxTokens,u=e.modelConfig?.stopSequences??this.config.stopSequences,m=e.modelConfig?.temperature??this.config.temperature,g=e.modelConfig?.topP??this.config.topP,A=e.modelConfig?.topK??this.config.topK,x=e.modelConfig?.n??this.config.n;if(x&&x>1)throw new Error("Anthropic does not support sampling (n > 1)");let h;if(this.config.thinking?.budget_tokens&&(h=this.config.thinking),t?.thinkingTokenBudget){let f=this.config.thinkingTokenBudgetLevels;switch(t.thinkingTokenBudget){case"none":h=void 0;break;case"minimal":h={type:"enabled",budget_tokens:f?.minimal??1024};break;case"low":h={type:"enabled",budget_tokens:f?.low??5e3};break;case"medium":h={type:"enabled",budget_tokens:f?.medium??1e4};break;case"high":h={type:"enabled",budget_tokens:f?.high??2e4};break;case"highest":h={type:"enabled",budget_tokens:f?.highest??32e3};break}}let y={...this.isVertex?{anthropic_version:"vertex-2023-10-16"}:{model:n},...d?{max_tokens:d}:{},...u&&u.length>0?{stop_sequences:u}:{},...m&&!h?{temperature:m}:{},...g&&(!h||g>=.95)?{top_p:g}:{},...A&&!h?{top_k:A}:{},...i,...c&&c.length>0?{tools:c}:{},...o?{stream:!0}:{},...a?{system:a}:{},...h?{thinking:h}:{},messages:p};return[r,y]};createChatResp=e=>{if(e.type==="error")throw new S(e.error.message,void 0,void 0);let t=Vo(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 S(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:Vo(r.stop_reason)}]}}return{results:[{index:o,content:""}]}}},ue=class s extends P{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={...Wo(),...o},d=new Nt(c,a),u=m=>{let g=B({model:m,modelInfo:Xe,models:i});return{functions:!0,streaming:!0,hasThinkingBudget:g?.hasThinkingBudget??!1,hasShowThoughts:g?.hasShowThoughts??!1,functionCot:!0}};super(d,{name:"Anthropic",apiURL:l,headers:p,modelInfo:Xe,defaults:{model:c.model},options:r,supportFor:u,models:i})}};function ls(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 ps(e)}function ps(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 Vo(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 Ze=(h=>(h.GPT4="gpt-4",h.GPT41="gpt-4.1",h.GPT41Mini="gpt-4.1-mini",h.GPT4O="gpt-4o",h.GPT4OMini="gpt-4o-mini",h.GPT4ChatGPT4O="chatgpt-4o-latest",h.GPT4Turbo="gpt-4-turbo",h.GPT35Turbo="gpt-3.5-turbo",h.GPT35TurboInstruct="gpt-3.5-turbo-instruct",h.GPT35TextDavinci002="text-davinci-002",h.GPT3TextBabbage002="text-babbage-002",h.GPT3TextAda001="text-ada-001",h.O1="o1",h.O1Mini="o1-mini",h.O3="o3",h.O3Mini="o3-mini",h.O4Mini="o4-mini",h))(Ze||{}),de=(n=>(n.TextEmbeddingAda002="text-embedding-ada-002",n.TextEmbedding3Small="text-embedding-3-small",n.TextEmbedding3Large="text-embedding-3-large",n))(de||{});var me=(f=>(f.GPT4="gpt-4",f.GPT41="gpt-4.1",f.GPT41Mini="gpt-4.1-mini",f.GPT4O="gpt-4o",f.GPT4OMini="gpt-4o-mini",f.GPT4ChatGPT4O="chatgpt-4o-latest",f.GPT4Turbo="gpt-4-turbo",f.GPT35Turbo="gpt-3.5-turbo",f.GPT35TurboInstruct="gpt-3.5-turbo-instruct",f.GPT35TextDavinci002="text-davinci-002",f.GPT3TextBabbage002="text-babbage-002",f.GPT3TextAda001="text-ada-001",f.O1Pro="o1-pro",f.O1="o1",f.O1Mini="o1-mini",f.O3Pro="o3-pro",f.O3="o3",f.O3Mini="o3-mini",f.O4Mini="o4-mini",f))(me||{});var ge=[{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}],$t=[{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 cs=s=>{let e=["o1","o1-mini","o3","o3-mini","o4-mini","o1-pro","o3-pro"];return e.includes(s)||e.includes(s)},ie=()=>structuredClone({model:"gpt-4.1",embedModel:"text-embedding-3-small",...v()}),Bt=()=>structuredClone({...ie(),model:"gpt-4.1"}),qt=()=>structuredClone({model:"gpt-4.1",embedModel:"text-embedding-3-small",...G()}),zt=()=>({...ie(),model:"gpt-4.1-mini"}),Lt=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=us(e),l=e.modelConfig?.frequencyPenalty??this.config.frequencyPenalty,p=e.modelConfig?.stream??this.config.stream,c=this.config.store,d=cs(n),u={model:n,messages:a,response_format:this.config?.responseFormat?{type:this.config.responseFormat}:void 0,tools:r,tool_choice:i,...d?{}:{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&&(u.reasoning_effort=this.config.reasoningEffort),this.config.webSearchOptions&&(u.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":u.reasoning_effort=void 0;break;case"minimal":u.reasoning_effort="low";break;case"low":u.reasoning_effort="medium";break;case"medium":u.reasoning_effort="high";break;case"high":u.reasoning_effort="high";break;case"highest":u.reasoning_effort="high";break}return this.chatReqUpdater&&(u=this.chatReqUpdater(u)),[o,u]}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 S(a.message.refusal,e.model,e.id);let l=Jo(a.finish_reason),p=a.message.tool_calls?.map(({id:c,function:{arguments:d,name:u}})=>({id:c,type:"function",function:{name:u,params:d}}));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:d,tool_calls:u,reasoning_content:m,annotations:g},finish_reason:A})=>{if(d)throw new S(d,void 0,n);let x=Jo(A),h=u?.map(({id:y,index:f,function:{name:R,arguments:b}})=>{typeof y=="string"&&typeof f=="number"&&!i.indexIdMap[f]&&(i.indexIdMap[f]=y);let w=i.indexIdMap[f];return w?{id:w,type:"function",function:{name:R,params:b}}:null}).filter(y=>y!==null);return{index:l,content:p??void 0,role:c,thought:m,annotations:g,functionCalls:h,finishReason:x,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)}}},Jo=s=>{switch(s){case"stop":return"stop";case"length":return"length";case"content_filter":return"error";case"tool_calls":return"function_call"}};function us(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"}};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 E=class extends P{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 Lt(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})}},Ae=class extends E{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("OpenAI API key not set");r=[...ge,...r??[]];let i=a=>{let l=B({model:a,modelInfo:r,models:o});return{functions:!0,streaming:!0,hasThinkingBudget:l?.hasThinkingBudget??!1,hasShowThoughts:l?.hasShowThoughts??!1}};super({apiKey:e,config:{...ie(),...t},options:n,modelInfo:r,models:o,supportFor:i}),super.setName("OpenAI")}};var Yo=ie,ds=qt,ms=zt,gs=Bt,he=class extends E{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={...Yo(),...r};l=[...ge,...l??[]];let c=u=>{let m=B({model:u,modelInfo:l,models:a});return{functions:!0,streaming:!0,hasThinkingBudget:m?.hasThinkingBudget??!1,hasShowThoughts:m?.hasShowThoughts??!1}};super({apiKey:e,config:p,options:i,models:a,modelInfo:l,supportFor:c});let d=t.includes("://")?t:`https://${t}.openai.azure.com/`;super.setName("Azure OpenAI"),super.setAPIURL(new URL(`/openai/deployments/${n}?api-version=${o}`,d).href),super.setHeaders(async()=>({"api-key":e}))}};var jt=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.");As(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 Q:throw n;case re:break;case J:break;case W:break;case V:break;case se: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 As(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}"`)}}var et=(o=>(o.CommandRPlus="command-r-plus",o.CommandR="command-r",o.Command="command",o.CommandLight="command-light",o))(et||{}),tt=(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))(tt||{});var Ht=[{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 Xo=()=>structuredClone({model:"command-r-plus",embedModel:"embed-english-v3.0",...v()}),hs=()=>structuredClone({model:"command-r",embedModel:"embed-english-v3.0",...G()}),Kt=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=xs(o),a=e.functions?.map(d=>{let u={};if(d.parameters?.properties)for(let[m,g]of Object.entries(d.parameters.properties))u[m]={description:g.description,type:g.type,required:d.parameters.required?.includes(m)??!1};return{name:d.name,description:d.description,parameter_definitions:u}}),l=e.chatPrompt.filter(d=>d.role==="function").map(d=>{let u=a?.find(m=>m.name===d.functionId);if(!u)throw new Error("Function not found");return{call:{name:u.name,parameters:u.parameter_definitions},outputs:[{result:d.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}}},xe=class extends P{constructor({apiKey:e,config:t,options:n,models:o}){if(!e||e==="")throw new Error("Cohere API key not set");let r={...Xo(),...t},i=new Kt(r);super(i,{name:"Cohere",apiURL:"https://api.cohere.ai/v1",headers:async()=>({Authorization:`Bearer ${e}`}),modelInfo:Ht,defaults:{model:r.model},supportFor:{functions:!0,streaming:!0},options:n,models:o})}};function xs(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=Qo(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=Qo(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 Qo(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 nt=(n=>(n.DeepSeekChat="deepseek-chat",n.DeepSeekCoder="deepseek-coder",n.DeepSeekReasoner="deepseek-reasoner",n))(nt||{});var Vt=[{name:"deepseek-chat",currency:"USD",promptTokenCostPer1M:.27,completionTokenCostPer1M:1.1},{name:"deepseek-reasoner",currency:"USD",promptTokenCostPer1M:.55,completionTokenCostPer1M:2.19}];var Zo=()=>structuredClone({model:"deepseek-chat",...v()}),fs=()=>structuredClone({model:"deepseek-coder",...G()}),fe=class extends E{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("DeepSeek API key not set");let i={...Zo(),...t};r=[...Vt,...r??[]],super({apiKey:e,config:i,options:n,apiURL:"https://api.deepseek.com",modelInfo:r,supportFor:{functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1},models:o}),super.setName("DeepSeek")}};var ot=(c=>(c.Gemini25Pro="gemini-2.5-pro",c.Gemini25Flash="gemini-2.5-flash",c.Gemini25FlashLite="gemini-2.5-flash-lite-preview-06-17",c.Gemini20Flash="gemini-2.0-flash",c.Gemini20FlashLite="gemini-2.0-flash-lite-preview-02-05",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))(ot||{}),Wt=(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))(Wt||{}),Jt=(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))(Jt||{}),Yt=(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))(Yt||{}),er=(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))(er||{});var Qt=[{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-preview-06-17",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-preview-02-05",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 tr=[{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"}],nr=()=>structuredClone({model:"gemini-2.5-flash",embedModel:"text-embedding-005",safetySettings:tr,thinkingTokenBudgetLevels:{minimal:200,low:800,medium:5e3,high:1e4,highest:24500},...v()}),ys=()=>structuredClone({model:"gemini-2.0-flash",embedModel:"text-embedding-005",safetySettings:tr,thinkingTokenBudgetLevels:{minimal:200,low:800,medium:5e3,high:1e4,highest:24500},...G()}),Xt=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 A=o?"&":"?",x=typeof this.apiKey=="function"?await this.apiKey():this.apiKey;r.name+=`${A}key=${x}`}let i=e.chatPrompt.filter(A=>A.role==="system").map(A=>A.content),a=i.length>0?{role:"user",parts:[{text:i.join(" ")}]}:void 0,l=e.chatPrompt.filter(A=>A.role!=="system").map((A,x)=>{switch(A.role){case"user":return{role:"user",parts:Array.isArray(A.content)?A.content.map((y,f)=>{switch(y.type){case"text":return{text:y.text};case"image":return{inlineData:{mimeType:y.mimeType,data:y.image}};default:throw new Error(`Chat prompt content type not supported (index: ${f})`)}}):[{text:A.content}]};case"assistant":{let h=[];if(A.functionCalls){if(h=A.functionCalls.map(y=>{let f=typeof y.function.params=="string"?JSON.parse(y.function.params):y.function.params;return{functionCall:{name:y.function.name,args:f}}}),!h)throw new Error("Function call is empty");return{role:"model",parts:h}}if(!A.content)throw new Error("Assistant content is empty");return h=[{text:A.content}],{role:"model",parts:h}}case"function":{if(!("functionId"in A))throw new Error(`Chat prompt functionId is empty (index: ${x})`);return{role:"user",parts:[{functionResponse:{name:A.functionId,response:{result:A.result}}}]}}default:throw new Error(`Invalid role: ${JSON.stringify(A)} (index: ${x})`)}}),p=[];e.functions&&e.functions.length>0&&p.push({function_declarations:e.functions}),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 A=e.functionCall.function?.name?{allowedFunctionNames:[e.functionCall.function.name]}:{};c={function_calling_config:{mode:"ANY"},...A}}else p&&p.length>0&&(c={function_calling_config:{mode:"AUTO"}});let d={};if(this.config.thinking?.includeThoughts&&(d.includeThoughts=!0),this.config.thinking?.thinkingTokenBudget&&(d.thinkingBudget=this.config.thinking.thinkingTokenBudget),t?.thinkingTokenBudget){let A=this.config.thinkingTokenBudgetLevels;switch(t.thinkingTokenBudget){case"none":d.thinkingBudget=0,d.includeThoughts=!1;break;case"minimal":d.thinkingBudget=A?.minimal??200;break;case"low":d.thinkingBudget=A?.low??800;break;case"medium":d.thinkingBudget=A?.medium??5e3;break;case"high":d.thinkingBudget=A?.high??1e4;break;case"highest":d.thinkingBudget=A?.highest??24500;break}}t?.showThoughts!==void 0&&t?.thinkingTokenBudget!=="none"&&(d.includeThoughts=t.showThoughts);let u={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(d).length>0?{thinkingConfig:d}:{}},m=this.config.safetySettings;return[r,{contents:l,tools:p,toolConfig:c,systemInstruction:a,generationConfig:u,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 S("Content was blocked due to safety settings",void 0,void 0);case"RECITATION":throw new S("Content was blocked due to recitation policy",void 0,void 0);case"MALFORMED_FUNCTION_CALL":throw new S("Function call was malformed and blocked",void 0,void 0);case"UNEXPECTED_TOOL_CALL":throw new S("Unexpected tool call",void 0,void 0);case"FINISH_REASON_UNSPECIFIED":throw new S("Finish reason unspecified",void 0,void 0);case"BLOCKLIST":throw new S("Content was blocked due to blocklist",void 0,void 0);case"PROHIBITED_CONTENT":throw new S("Content was blocked due to prohibited content",void 0,void 0);case"SPII":throw new S("Content was blocked due to SPII",void 0,void 0);case"OTHER":throw new S("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:q(),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}}},ye=class s extends P{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,d;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 A;o?A="endpoints":A="publishers/google",c=`https://${n==="global"?"aiplatform":`${n}-aiplatform`}.googleapis.com/v1/projects/${t}/locations/${n}/${A}`,d=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",d=async()=>({})}let u={...nr(),...r},m=new Xt(u,p,o,e,i);l=[...Qt,...l??[]];let g=A=>{let x=B({model:A,modelInfo:l,models:a});return{functions:!0,streaming:!0,hasThinkingBudget:x?.hasThinkingBudget??!1,hasShowThoughts:x?.hasShowThoughts??!1,functionCot:!1}};super(m,{name:"GoogleGeminiAI",apiURL:c,headers:d,modelInfo:l,defaults:{model:u.model,embedModel:u.embedModel},options:i,supportFor:g,models:a})}};var Is=new N,Ie=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(Is.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 rt=(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))(rt||{});var Zt=[{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 bs=()=>structuredClone({model:"llama-3.3-70b-versatile",...v()}),be=class extends E{constructor({apiKey:e,config:t,options:n,models:o,modelInfo:r}){if(!e||e==="")throw new Error("Groq API key not set");let i={...bs(),...t},a={...n,streamingUsage:!1};r=[...Zt,...r??[]];let l={functions:!0,streaming:!0,hasThinkingBudget:!1,hasShowThoughts:!1};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 Ie(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 en=[];var tn=(e=>(e.MetaLlama270BChatHF="meta-llama/Llama-2-70b-chat-hf",e))(tn||{});var or=()=>structuredClone({model:"meta-llama/Llama-2-70b-chat-hf",...v()}),Rs=()=>structuredClone({model:"meta-llama/Llama-2-70b-chat-hf",...G()}),nn=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:
|
|
106
106
|
${JSON.stringify(e.functions,null,2)}
|
|
107
107
|
`:"",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 d=typeof c.function.params=="string"?c.function.params:JSON.stringify(c.function.params);return`${c.function.name}(${d})`}).join(`
|
|
@@ -135,20 +135,20 @@ ${p}`:`Assistant: ${l.content}`}default:throw new Error("Unknown role")}}).join(
|
|
|
135
135
|
|
|
136
136
|
`,r+=p.length)}else o=`${i}
|
|
137
137
|
|
|
138
|
-
`,r=l}),(r>t||n.length===0)&&n.push(o.trim()),n},Ds=(s,e=.1)=>{let t=[...s].sort((o,r)=>o.score-r.score),n=Math.ceil(t.length*e);return t.slice(0,n)};import{context as li,SpanKind as pi,trace as ci}from"@opentelemetry/api";var _e=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=[]}},Fe=class{memories=new Map;defaultMemory;constructor(){this.defaultMemory=new _e}getMemory(e){return e?(this.memories.has(e)||this.memories.set(e,new _e),this.memories.get(e)):this.defaultMemory}addRequest(e,t){for(let n of e)dn(n);this.getMemory(t).addRequest(e,0)}addResponse(e,t){mn(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 _e):this.defaultMemory.reset()}};var Z=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()}},ut=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 Z({message:o}):new Error("Assertion Failed: No message provided for assertion")}},Rn=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 Z({message:a})}};var pr={enabled:!0,enabledCategories:["generation","streaming","functions","errors","performance"],maxLabelLength:100,samplingRate:1},Ge,Cn=s=>{if(Ge)return Ge;let e=s??k.meter;if(e)return Ge=Ns(e),Ge};var Us=()=>{let s=[];return k.meter||s.push("Global meter not initialized"),!Ge&&k.meter&&s.push("Metrics instruments not created despite available meter"),{healthy:s.length===0,issues:s}},Ns=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"})}),dt=pr,$s=s=>{dt={...dt,...s}},Ls=()=>({...dt}),j=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let o=String(n),r=dt.maxLabelLength;e[t]=o.length>r?o.substring(0,r):o}return e},cr=(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)}},mt=(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)}},Tn=(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)}},ur=(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)}},wn=(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)}},vn=(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)}},dr=(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)}},mr=(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)}},gr=(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)}},Ar=(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)}},gt=(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 F=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(`
|
|
138
|
+
`,r=l}),(r>t||n.length===0)&&n.push(o.trim()),n},Ds=(s,e=.1)=>{let t=[...s].sort((o,r)=>o.score-r.score),n=Math.ceil(t.length*e);return t.slice(0,n)};import{context as li,SpanKind as pi,trace as ci}from"@opentelemetry/api";var _e=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=[]}},Fe=class{memories=new Map;defaultMemory;constructor(){this.defaultMemory=new _e}getMemory(e){return e?(this.memories.has(e)||this.memories.set(e,new _e),this.memories.get(e)):this.defaultMemory}addRequest(e,t){for(let n of e)dn(n);this.getMemory(t).addRequest(e,0)}addResponse(e,t){mn(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 _e):this.defaultMemory.reset()}};var Z=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()}},ut=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 Z({message:o}):new Error("Assertion Failed: No message provided for assertion")}},Rn=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 Z({message:a})}};var pr={enabled:!0,enabledCategories:["generation","streaming","functions","errors","performance"],maxLabelLength:100,samplingRate:1},Ge,Cn=s=>{if(Ge)return Ge;let e=s??M.meter;if(e)return Ge=Ns(e),Ge};var Us=()=>{let s=[];return M.meter||s.push("Global meter not initialized"),!Ge&&M.meter&&s.push("Metrics instruments not created despite available meter"),{healthy:s.length===0,issues:s}},Ns=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"})}),dt=pr,$s=s=>{dt={...dt,...s}},Ls=()=>({...dt}),j=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let o=String(n),r=dt.maxLabelLength;e[t]=o.length>r?o.substring(0,r):o}return e},cr=(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)}},mt=(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)}},Tn=(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)}},ur=(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)}},wn=(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)}},vn=(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)}},dr=(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)}},mr=(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)}},gr=(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)}},Ar=(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)}},gt=(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 F=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(`
|
|
139
139
|
`)}[Symbol.for("nodejs.util.inspect.custom")](e,t){return this.toString()}},hr=({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(", ")??"";Co(s,e,l,n)}return o&&Tn(o,"validation",r),i&&i.addEvent("validation.error",{message:s.toString(),fixing_instructions:a?.map(l=>l.title).join(", ")??""}),a},xr=({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(", ")??"";To(s,e,l,n)}return o&&Tn(o,"assertion",r),i&&i.addEvent("assertion.error",{message:s.toString(),fixing_instructions:a?.map(l=>l.title).join(", ")??""}),a},fr=({error:s,errCount:e,debug:t,logger:n,metricsInstruments:o,signatureName:r,span:i})=>{t&&n&&wo(s,e,n),o&&ur(o,r),i&&i.addEvent("refusal.error",{message:s.toString()})};var yr=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(`
|
|
140
140
|
`)}),"","Please fix these issues and try again."].join(`
|
|
141
141
|
`);throw new Error(n)}};var At=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(`
|
|
142
142
|
`)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},ht=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}'
|
|
143
143
|
${t.join(`
|
|
144
144
|
`)}`};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(`
|
|
145
|
-
`)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},xt=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??
|
|
145
|
+
`)}[Symbol.for("nodejs.util.inspect.custom")](t,n){return this.toString()}},xt=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 At?new ht(o.getFields(),n,e.id):o}}},Sn=(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&&yr(n.parameters);return[...e??[],...t]},On=async({ai:s,functionList:e,functionCalls:t,mem:n,sessionId:o,span:r,excludeContentFromTrace:i,index:a,functionResultFormatter:l})=>{let p=new xt(e),c=new Set,d=t.map(g=>{if(!g.id)throw new Error(`Function ${g.name} did not return an ID`);return p.execute(g,{sessionId:o,ai:s,functionResultFormatter:l}).then(x=>{if(c.add(g.name.toLowerCase()),r){let h={name:g.name};i||(h.args=g.args,h.result=x??""),r.addEvent("function.call",h)}return{result:x??"",role:"function",functionId:g.id,index:a}}).catch(x=>{if(!(x instanceof ht))throw x;let h=x.getFixingInstructions();if(r){let y={name:g.name,message:x.toString()};i||(y.args=g.args,y.fixing_instructions=h),r.addEvent("function.error",y)}if(s.getOptions().debug){let y=s.getLogger();Ro(x,a,h,y)}return{functionId:g.id,isError:!0,index:a,result:h,role:"function"}})}),m=(await Promise.all(d)).filter(g=>g!==void 0);if(n.addFunctionResults(m,o),s.getOptions().debug){let g=m.filter(A=>!A.isError);if(g.length>0){let A=s.getLogger();bo(g,A)}}return m.some(g=>g.isError)&&n.addTag("error",o),c};function Mn(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 Ir(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}}import ee from"dayjs";import Bs from"dayjs/plugin/customParseFormat.js";import qs from"dayjs/plugin/timezone.js";import zs from"dayjs/plugin/utc.js";ee.extend(zs);ee.extend(qs);ee.extend(Bs);function br(s,e,t=!1){try{return js(e)}catch(n){if(s.isOptional&&!t)return;let o=n.message;throw new F({fields:[s],message:o,value:e})}}function js(s){if(!ee(s,"YYYY-MM-DD",!0).isValid())throw new Error('Invalid date format. Please provide the date in "YYYY-MM-DD" format.');return ee.utc(s,"YYYY-MM-DD").startOf("day").toDate()}function Rr(s,e,t=!1){try{return Hs(e)}catch(n){if(s.isOptional&&!t)return;let o=n.message;throw new F({fields:[s],message:o,value:e})}}function Hs(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=ee.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 Cr=s=>ee(s).utc().format("YYYY-MM-DD HH:mm:ss [UTC]");var kc=new N;var ft=(s,e)=>{let t=s.type??{name:"string",isArray:!1},n=(a,l)=>{switch(a){case"class":return typeof l=="string";case"code":return typeof l=="string";case"string":return typeof l=="string";case"number":return typeof l=="number";case"boolean":return typeof l=="boolean";case"date":return l instanceof Date||typeof l=="string";case"datetime":return l instanceof Date||typeof l=="string";case"json":return typeof l=="object"||typeof l=="string";default:return!1}},o=a=>!(!a||typeof a!="object"||!("mimeType"in a)||!("data"in a));if(s.type?.name==="image"){let a;if(Array.isArray(e)){for(let l of e)if(!o(l)){a="object ({ mimeType: string; data: string })";break}}else o(e)||(a="object ({ mimeType: string; data: string })");if(a)throw new Error(`Validation failed: Expected '${s.name}' to be type '${a}' instead got '${e}'`);return}let r=a=>!(!a||typeof a!="object"||!("data"in a));if(s.type?.name==="audio"){let a;if(Array.isArray(e)){for(let l of e)if(!r(l)){a="object ({ data: string; format?: string })";break}}else r(e)||(a="object ({ data: string; format?: string })");if(a)throw new Error(`Validation failed: Expected '${s.name}' to be type '${a}' instead got '${e}'`);return}let i=!0;if(t.isArray){if(!Array.isArray(e))i=!1;else for(let a of e)if(!n(t.name,a)){i=!1;break}}else i=n(t.name,e);if(!i){let a=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 '${a}' (${JSON.stringify(e)})`)}};function De(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 Tr=s=>{if(!s.trim())return[];let e=new Set(["-","*","+"]),t=/^\d+[\s]*[.)\]]\s*/,n=s.split(`
|
|
146
146
|
`),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 En(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 kn=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)}},Ks=new kn(500);function Pn(s,e,t=0,n=Ks){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 Sr=(s,e,t,n=!1)=>{let o={extractedFields:[],streamedIndex:{},s:-1};_n(s,e,o,t,{strictMode:n}),Fn(s,e,o,t,n);for(let r of s.getOutputFields())r.isInternal&&delete e[r.name]},Vs=(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 F({message:`Required ${n.length===1?"field":"fields"} not found`,fields:n})},_n=(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 d=`${(t.extractedFields.length===0?"":`
|
|
147
147
|
`)+p.title}:`,u=Pn(n,d,t.s),m=d.length;switch(u){case-1:if(r)continue;if(!o&&i.length===1&&t.currField===void 0){t.inAssumedField=!0,a=p,m=0,u=0;break}if(t.currField===void 0&&t.extractedFields.length===0){if(o&&!p.isOptional)throw new F({message:"Expected (Required) field not found",fields:[p]});if(!o){let g=!1;for(let A=l;A<i.length;A++){let x=i[A];if(!x)continue;let h=`${(t.extractedFields.length===0?"":`
|
|
148
148
|
`)+x.title}:`;if(Pn(n,h,t.s)>=0){g=!0;break}}if(!g){t.inAssumedField=!0,a=p,m=0,u=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 F({message:"Expected (Required) field not found",fields:[a]});if(t.currField!==void 0&&t.inAssumedField){let g=n.substring(0,u).trim();if(g&&t.currField.name===p.name){let A=Ue(t.currField,g);A!==void 0&&(e[t.currField.name]=A)}else if(g){let A=Ue(t.currField,g);A!==void 0&&(e[t.currField.name]=A)}t.inAssumedField=!1,t.streamedIndex[t.currField.name]=0,t.currField=void 0}if(t.currField){let g=n.substring(t.s,u).trim(),A=Ue(t.currField,g);A!==void 0&&(e[t.currField.name]=A),t.prevFields?t.prevFields?.push({field:t.currField,s:t.s,e:u}):t.prevFields=[{field:t.currField,s:t.s,e:u}]}t.s=u+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)}},Fn=(s,e,t,n,o=!1)=>{if(t.currField){let r=n.substring(t.s).trim(),i=Ue(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 F({message:"Expected field not found",fields:[a]})}Ws(s,e,n),Vs(t,e,s.getOutputFields())},Ws=(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 d=`${c.title}:`,u=t.indexOf(d,a);u!==-1&&u<l&&(l=u)}let p=t.substring(a,l).trim();if(p)try{let c=Ue(o,p);c!==void 0&&(e[o.name]=c)}catch{}}},wr=(s,e,t=!1)=>{switch(s.type?.name){case"code":return Or(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 br(s,e,t);case"datetime":return Rr(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*vr(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,d=c===0,u=s.substring(t+c,n);if(u.length===0)return;let m=u.replace(/\s+$/,"");o.currField?.type?.name==="code"&&(m=m.replace(/\s*```\s*$/,""));let g=d?m.trimStart():m;o.currField?.type?.name==="code"&&(g=g.replace(/^[ ]*```[a-zA-Z0-9]*\n\s*/,"")),g.length>0&&(yield{index:r,delta:{[i]:g}},o.streamedIndex[i]=c+m.length)}function*Gn(s,e,t,n,o){for(let i of n.prevFields??[]){let{field:a,s:l,e:p}=i;yield*vr(e,a,l,p,n,o)}if(n.prevFields=void 0,!n.currField||n.currField.isInternal)return;yield*vr(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 Ue(s,e){if(!e||e===""||/^(null|undefined)\s*$/i.test(e)){if(s.isOptional)return;throw new F({message:"Required field is missing",fields:[s],value:e})}let t;if(s.type?.name==="json")try{let n=Or(e);return t=JSON.parse(n),t}catch(n){throw new F({message:`Invalid JSON: ${n.message}`,fields:[s],value:e})}if(s.type?.isArray)try{try{t=JSON.parse(e)}catch{t=Tr(e)}if(!Array.isArray(t))throw new Error("Expected an array")}catch(n){throw new F({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]=wr(s,r,!0)}}else t=wr(s,e)}catch(n){throw new F({message:n.message,fields:[s],value:e})}if(!(typeof t=="string"&&t===""))return t}var Or=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 Dn(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});Mr(o.field,t,i,n)}}async function Un(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});Mr(t.currField,n,c,r)}}var Mr=(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=Js(s,o);e.addRequest([{role:"user",content:[{type:"text",text:r}]}],n),e.addTag("processor",n)};function Js(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*kr({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 d=t.find(u=>u.index===c.index);if(!d)throw new Error(`No state found for result (index: ${c.index})`);yield*Ys({...n,result:c,skipEarlyFail:o,state:d})}}}finally{i.releaseLock()}for(let a of t)yield*Qs({...n,state:a})}async function*Ys({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)Je(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),_n(i,r.values,r.xstate,r.content,{strictMode:n,skipEarlyFail:o}))return;p.length!==0&&await Rn(p,r.xstate,r.content),a.length!==0&&await Un(a,r.content,r.xstate,e,r.values,t),yield*Gn(i,r.content,r.values,r.xstate,s.index),await ut(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
|
|
149
149
|
Content: ${r.content}`)}async function*Qs({state:s,signature:e,ai:t,model:n,functions:o,mem:r,sessionId:i,traceId:a,span:l,strictMode:p,excludeContentFromTrace:c,streamingAsserts:d,asserts:u,fieldProcessors:m,streamingFieldProcessors:g,functionResultFormatter:A}){let x=Mn(t,s.functionCalls,s.values,n);if(x){if(!o)throw new Error("Functions are not defined");let h=await On({ai:t,functionList:o,functionCalls:x,mem:r,sessionId:i,traceId:a,span:l,index:s.index,excludeContentFromTrace:c,functionResultFormatter:A});s.functionsExecuted=new Set([...s.functionsExecuted,...h])}else Fn(e,s.values,s.xstate,s.content,p),await Rn(d,s.xstate,s.content,!0),await ut(u,s.values),m.length&&await Dn(m,s.values,r,i),g.length!==0&&await Un(g,s.content,s.xstate,r,s.values,i,!0),yield*Gn(e,s.content,s.values,s.xstate,s.index)}async function*Er({ai:s,res:e,mem:t,sessionId:n,traceId:o,functions:r,span:i,strictMode:a,states:l,usage:p,excludeContentFromTrace:c,asserts:d,fieldProcessors:u,thoughtFieldName:m,signature:g,functionResultFormatter:A}){let x=e.results??[];t.addResponse(x,n);for(let R of x){let b=l[R.index];if(!b)throw new Error(`No state found for result (index: ${R.index})`);if(e.modelUsage&&p.push(e.modelUsage),R.functionCalls?.length){let w=Mn(s,R.functionCalls,b.values);if(w){if(!r)throw new Error("Functions are not defined");let $=await On({ai:s,functionList:r,functionCalls:w,mem:t,sessionId:n,traceId:o,span:i,excludeContentFromTrace:c,index:R.index,functionResultFormatter:A});b.functionsExecuted=new Set([...b.functionsExecuted,...$])}}else R.content&&(R.thought&&R.thought.length>0&&(b.values[m]=R.thought),Sr(g,b.values,R.content,a),await ut(d,b.values),u.length&&await Dn(u,b.values,t,n));if(R.finishReason==="length")throw new Error(`Max tokens reached before completion
|
|
150
150
|
Content: ${R.content}`)}let h=l.map(R=>R.values);for(let R of h)for(let b of g.getOutputFields())b.isInternal&&delete R[b.name];let y=g.getOutputFields(),f=h.map((R,b)=>{let w={};for(let $ of y)$.isInternal||(w[$.name]=R[$.name]);return R[m]!==void 0&&(w[m]=R[m]),{index:b,delta:w}});for(let R of f)yield R}function Pr(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(d=>d==="processor"):!1;if(p&&e&&a||!(p||c))return!1}return!0}var Ne=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"}},Nn=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(`
|
|
151
|
-
`)}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(k.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","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 _r(s){return new Nn(s).parse()}var I=class extends Error{constructor(t,n,o){super(t);this.fieldName=n;this.suggestion=o;this.name="AxSignatureValidationError"}},O=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=_r(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 I(`Invalid Signature: ${n.message}`,void 0,o)}throw new I(`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 I("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 I("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 I?t:new I(`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 I("Invalid signature argument type",void 0,"Signature must be a string, another AxSignature instance, or an object with inputs and outputs arrays")}parseParsedField=e=>{if(!e.name||e.name.length===0)throw new I("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 I("Field type name is required",e.name,"Specify a valid type. Available types: string, number, boolean, json, image, audio, date, datetime, class, code");return{...e,title:t}};setDescription=e=>{if(typeof e!="string")throw new I("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);Y(t,"input");for(let n of this.inputFields)if(n.name===t.name)throw new I(`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 I(`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 I?t:new I(`Failed to add input field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};addOutputField=e=>{try{let t=this.parseField(e);Y(t,"output");for(let n of this.outputFields)if(n.name===t.name)throw new I(`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 I(`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 I?t:new I(`Failed to add output field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};setInputFields=e=>{if(!Array.isArray(e))throw new I("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 Y(o,"input"),o});this.inputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof I?t:new I(`Failed to set input fields: ${t instanceof Error?t.message:"Unknown error"}`)}};setOutputFields=e=>{if(!Array.isArray(e))throw new I("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 Y(o,"output"),o});this.outputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof I?t:new I(`Failed to set output fields: ${t instanceof Error?t.message:"Unknown error"}`)}};getInputFields=()=>this.inputFields;getOutputFields=()=>this.outputFields;getDescription=()=>this.description;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=>{Y(e,"input")}),this.getOutputFields().forEach(e=>{Y(e,"output")}),this.sigHash=He("sha256").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Gr(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof I?e:new I(`Signature validation failed: ${e instanceof Error?e.message:"Unknown error"}`)}};updateHash=()=>{try{return this.getInputFields().forEach(e=>{Y(e,"input")}),this.getOutputFields().forEach(e=>{Y(e,"output")}),this.validateSignatureConsistency(),this.sigHash=He("sha256").update(this.description??"").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Gr(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof I?e:new I(`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 I(`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 I(`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 I(`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 I("Signature must have at least one input field",void 0,'Add an input field. Example: "userInput:string -> ..."');if(this.outputFields.length===0)throw new I("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 Fr(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 Gr(s,e,t){let n=s?`"${s}" `:"",o=e.map(Fr).join(", "),r=t.map(Fr).join(", ");return`${n}${o} -> ${r}`}function Xs(s){let e=/^[a-z][a-zA-Z0-9]*$/,t=/^[a-z]+(_[a-z0-9]+)*$/;return e.test(s)||t.test(s)}function Y(s,e){if(!s.name||s.name.length===0)throw new I("Field name cannot be blank",s.name,"Every field must have a descriptive name");if(!Xs(s.name))throw new I(`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(k.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 I(`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 I(`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 I(`Field name '${s.name}' is too long (${s.name.length} characters)`,s.name,"Field names should be 50 characters or less");s.type&&Zs(s,e)}function Zs(s,e){if(!s.type)return;let{type:t}=s;if(t.name==="image"||t.name==="audio"){if(e==="output")throw new I(`${t.name} type is not supported in output fields`,s.name,`${t.name} types can only be used in input fields`);if(t.isArray)throw new I(`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 I("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 I("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 I("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 I(`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 I("Duplicate class options found",s.name,"Each class option must be unique (case-insensitive)")}if(t.name==="code"&&t.isArray)throw new I("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 I("Internal marker (!) is not allowed on input fields",s.name,"Internal markers are only allowed on output fields")}var te=class{signature;sigHash;examples;examplesOptions;demos;trace;usage=[];traceLabel;key;children;constructor(e,t){this.signature=new O(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 Ne,this.key={id:this.signature.hash()}}getSignature(){return new O(this.signature)}setSignature(e){this.signature=new O(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&&(ft(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 De(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 ei=`
|
|
151
|
+
`)}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","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 _r(s){return new Nn(s).parse()}var I=class extends Error{constructor(t,n,o){super(t);this.fieldName=n;this.suggestion=o;this.name="AxSignatureValidationError"}},O=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=_r(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 I(`Invalid Signature: ${n.message}`,void 0,o)}throw new I(`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 I("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 I("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 I?t:new I(`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 I("Invalid signature argument type",void 0,"Signature must be a string, another AxSignature instance, or an object with inputs and outputs arrays")}parseParsedField=e=>{if(!e.name||e.name.length===0)throw new I("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 I("Field type name is required",e.name,"Specify a valid type. Available types: string, number, boolean, json, image, audio, date, datetime, class, code");return{...e,title:t}};setDescription=e=>{if(typeof e!="string")throw new I("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);Y(t,"input");for(let n of this.inputFields)if(n.name===t.name)throw new I(`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 I(`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 I?t:new I(`Failed to add input field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};addOutputField=e=>{try{let t=this.parseField(e);Y(t,"output");for(let n of this.outputFields)if(n.name===t.name)throw new I(`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 I(`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 I?t:new I(`Failed to add output field "${e.name}": ${t instanceof Error?t.message:"Unknown error"}`,e.name)}};setInputFields=e=>{if(!Array.isArray(e))throw new I("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 Y(o,"input"),o});this.inputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof I?t:new I(`Failed to set input fields: ${t instanceof Error?t.message:"Unknown error"}`)}};setOutputFields=e=>{if(!Array.isArray(e))throw new I("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 Y(o,"output"),o});this.outputFields=t,this.invalidateValidationCache(),this.updateHashLight()}catch(t){throw t instanceof I?t:new I(`Failed to set output fields: ${t instanceof Error?t.message:"Unknown error"}`)}};getInputFields=()=>this.inputFields;getOutputFields=()=>this.outputFields;getDescription=()=>this.description;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=>{Y(e,"input")}),this.getOutputFields().forEach(e=>{Y(e,"output")}),this.sigHash=He("sha256").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Gr(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof I?e:new I(`Signature validation failed: ${e instanceof Error?e.message:"Unknown error"}`)}};updateHash=()=>{try{return this.getInputFields().forEach(e=>{Y(e,"input")}),this.getOutputFields().forEach(e=>{Y(e,"output")}),this.validateSignatureConsistency(),this.sigHash=He("sha256").update(this.description??"").update(JSON.stringify(this.inputFields)).update(JSON.stringify(this.outputFields)).digest("hex"),this.sigString=Gr(this.description,this.inputFields,this.outputFields),[this.sigHash,this.sigString]}catch(e){throw e instanceof I?e:new I(`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 I(`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 I(`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 I(`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 I("Signature must have at least one input field",void 0,'Add an input field. Example: "userInput:string -> ..."');if(this.outputFields.length===0)throw new I("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 Fr(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 Gr(s,e,t){let n=s?`"${s}" `:"",o=e.map(Fr).join(", "),r=t.map(Fr).join(", ");return`${n}${o} -> ${r}`}function Xs(s){let e=/^[a-z][a-zA-Z0-9]*$/,t=/^[a-z]+(_[a-z0-9]+)*$/;return e.test(s)||t.test(s)}function Y(s,e){if(!s.name||s.name.length===0)throw new I("Field name cannot be blank",s.name,"Every field must have a descriptive name");if(!Xs(s.name))throw new I(`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 I(`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 I(`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 I(`Field name '${s.name}' is too long (${s.name.length} characters)`,s.name,"Field names should be 50 characters or less");s.type&&Zs(s,e)}function Zs(s,e){if(!s.type)return;let{type:t}=s;if(t.name==="image"||t.name==="audio"){if(e==="output")throw new I(`${t.name} type is not supported in output fields`,s.name,`${t.name} types can only be used in input fields`);if(t.isArray)throw new I(`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 I("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 I("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 I("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 I(`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 I("Duplicate class options found",s.name,"Each class option must be unique (case-insensitive)")}if(t.name==="code"&&t.isArray)throw new I("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 I("Internal marker (!) is not allowed on input fields",s.name,"Internal markers are only allowed on output fields")}var te=class{signature;sigHash;examples;examplesOptions;demos;trace;usage=[];traceLabel;key;children;constructor(e,t){this.signature=new O(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 Ne,this.key={id:this.signature.hash()}}getSignature(){return new O(this.signature)}setSignature(e){this.signature=new O(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&&(ft(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 De(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 ei=`
|
|
152
152
|
## Function Call Instructions
|
|
153
153
|
- Complete the task, using the functions defined earlier in this prompt.
|
|
154
154
|
- Output fields should only be generated after all functions have been called.
|
|
@@ -179,7 +179,7 @@ ${c}`),l&&l.length>0&&o.push(ei.trim()),o.push(ti.trim());let d=this.sig.getDesc
|
|
|
179
179
|
`),n.push(t.map(o=>`- ${o}`).join(`
|
|
180
180
|
`))):n.push(t),[{type:"text",text:n.join("")}]}},Dr=s=>s.map(e=>`\`${e.title}\``).join(", "),ni=s=>s.map(t=>{let n=t.title,o=t.type?.name?Nr(t.type):"string",r=t.isOptional?`This optional ${o} field may be omitted`:`A ${o} field`,i=t.description?` ${yt(t.description)}`:"";return`${n}: (${r})${i}`.trim()}).join(`
|
|
181
181
|
`),oi=s=>s.map(t=>{let n=t.title,o=t.type?.name?Nr(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:yt(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(`
|
|
182
|
-
`),ri=(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?Cr(e):s.type?.name==="image"&&typeof e=="object"||s.type?.name==="audio"&&typeof e=="object"||typeof e=="string"?e:JSON.stringify(e,null,2)},Nr=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";default:return"string"}})();return s?.isArray?`json array of ${e} items`:e};function Ur(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 si=(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 yt(s){let e=s.trim();return e.length>0?`${e.charAt(0).toUpperCase()}${e.slice(1)}${e.endsWith(".")?"":"."}`:""}function ii(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 ai(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 It(s,e,t,n){if(!e?.resultPicker||s.length<=1)return 0;let o=e.resultPicker;if((t?ii(t,n):!1)&&t){let l=ai(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 $r(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 It(o,t,s,e)}var M=class extends te{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??ae)(this.signature,n),this.asserts=this.options?.asserts??[],this.streamingAsserts=this.options?.streamingAsserts??[],this.excludeContentFromTrace=t?.excludeContentFromTrace??!1,this.usage=[],t?.functions&&(this.functions=Sn(t.functions))}getSignatureName(){return this.signature.getDescription()||"unknown_signature"}getMetricsInstruments(){return Cn()}updateMeter(e){Cn(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:d,thinkingTokenBudget:u,showThoughts:m}=n??{},g=await $r(t,l,{resultPicker:n?.resultPicker}),A=t?.history(g,l)??[];if(A.length===0)throw new Error("No chat prompt found");let x={...n?.modelConfig,...n?.sampleCount?{n:n.sampleCount}:{},...n?.sampleCount&&n?.modelConfig?.temperature===1?{temperature:.8}:{}},h=this.isDebug(e,n),y=a===0;return await e.chat({chatPrompt:A,functions:r,functionCall:i,modelConfig:x,model:p},{sessionId:l,rateLimiter:c,stream:d,debug:h,debugHideSystemPrompt:!y,thinkingTokenBudget:u,showThoughts:m,traceContext:o,abortSignal:n?.abortSignal,stepIndex:a})}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,d=n.model,u=this.createStates(n.sampleCount??1),m=this.usage,g=o===0,{functions:A,functionCall:x}=Ir(l,p,g),h=await this.forwardSendRequest({ai:e,mem:t,options:n,traceContext:i,functions:A,functionCall:x,stepIndex:o});h instanceof ReadableStream?yield*kr({ai:e,model:d,res:h,mem:t,sessionId:a,functions:A,strictMode:c,span:r,states:u,usage:m,asserts:this.asserts,streamingAsserts:this.streamingAsserts,fieldProcessors:this.fieldProcessors,streamingFieldProcessors:this.streamingFieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,functionResultFormatter:n?.functionResultFormatter??this.options?.functionResultFormatter}):yield*Er({ai:e,model:d,res:h,mem:t,sessionId:a,functions:A,span:r,strictMode:c,states:u,usage:m,asserts:this.asserts,fieldProcessors:this.fieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,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 Fe,d;if(o?.functions&&o.functions.length>0){let y=this.options?.promptTemplate??ae,f={functions:o.functions,thoughtFieldName:this.thoughtFieldName};this.promptTemplate=new y(this.signature,f)}let u,m=performance.now();Array.isArray(t)?(Dt(t),u=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos})):u=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos});let g=performance.now()-m,A=this.getMetricsInstruments();A&>(A,"prompt_render",g,this.getSignatureName());let x=performance.now();c.addRequest(u,o.sessionId);let h=performance.now()-x;A&>(A,"memory_update",h,this.getSignatureName());e:for(let y=0;y<p;y++){for(let R=0;R<l;R++)try{let b=this.forwardCore({options:o,ai:e,mem:c,stepIndex:y,span:r,traceContext:i});for await(let L of b)L!==void 0&&(yield{version:R,index:L.index,delta:L.delta});if(Pr(c,a,n,o?.sessionId)){let L=this.getMetricsInstruments();L&&mt(L,y+1,p,this.getSignatureName());continue e}let $=this.getMetricsInstruments();if($){mt($,y+1,p,this.getSignatureName());let L=new Set;n.forEach(vt=>{vt.functionsExecuted.forEach(St=>L.add(St))}),L.size>0&&vn($,!0,L.size,!0,!1,this.getSignatureName()),dr($,this.fieldProcessors.length,this.streamingFieldProcessors.length,this.getSignatureName())}return}catch(b){let w,$=this.isDebug(e,o),L=this.getLogger(e,o),vt=this.getMetricsInstruments(),St=this.getSignatureName(),Ot={error:b,errCount:R,logger:L,metricsInstruments:vt,signatureName:St,span:r,debug:$};if(r?.recordException(b),b instanceof F)w=hr(Ot),d=b;else if(b instanceof Z)w=xr(Ot),d=b;else if(b instanceof S)fr(Ot);else if(!(b instanceof V))throw $n(b,e,this.signature);w&&(c.addRequest([{role:"user",content:this.promptTemplate.renderExtraFields(w)}],o.sessionId),c.addTag("error",o.sessionId))}let f=this.getMetricsInstruments();throw f&&wn(f,l,!1,l,this.getSignatureName()),$n(new Error(`Unable to fix validation error: ${d?.toString()}`),e,this.signature)}throw A&&mt(A,p,p,this.getSignatureName()),$n(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&>(a,"state_creation",i,this.getSignatureName());let l=n?.tracer??this.options?.tracer??e.getOptions().tracer,p=this.functions;if(n?.functions&&(p=Sn(n.functions,this.functions)),!l){yield*this._forward2(e,t,r,{...n,functions:p});return}let c=p?.map(h=>h.name).join(","),d={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}:{}},u=this.traceLabel&&n.traceLabel?`${this.traceLabel} > ${n.traceLabel}`:n.traceLabel??this.traceLabel,m=u?`AxGen > ${u}`:"AxGen",g=l.startSpan(m,{kind:pi.SERVER,attributes:d}),A=li.active(),x=ci.setSpan(A,g);try{if(this.excludeContentFromTrace||g.addEvent("input",{content:JSON.stringify(t,null,2)}),yield*this._forward2(e,t,r,{...n,functions:p},g,x),!this.excludeContentFromTrace){let h=r.map(f=>f.values),y=h.length===1?h[0]:h;g.addEvent("output",{content:JSON.stringify(y,null,2)})}}finally{g.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,d=!1;try{let u=this.getMetricsInstruments();u&&Ar(u,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??{}),g=[],A=0,x=0;for await(let w of m)w.version!==A&&(g=[]),A=w.version,g=En(g,w),x++;l=A;let h=performance.now();d=!!n?.resultPicker;let y=await It(g,{resultPicker:n?.resultPicker},n?.mem,n?.sessionId),f=performance.now()-h,b=g[y]?.delta??{};if(this.trace={...t,...b},d&&this.isDebug(e,n)){let w=this.getLogger(e,n);Oo(g.length,y,f,w)}return a=!0,u&&(gr(u,g.length,d,d?f:void 0,r),mr(u,i,x,void 0,r)),b}catch(u){throw a=!1,u}finally{let u=performance.now()-o,m=this.getMetricsInstruments();m&&(cr(m,u,a,r,e.getName(),n?.model?String(n.model):void 0),p&&vn(m,p,c,c>0,!1,r),l>0&&wn(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=En(r,p);let a=await It(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??e.getLogger()}},bt=class extends Error{details;constructor(e,t,n){super(e),this.name="AxGenerateError",this.details=t,n?.cause&&(this.cause=n.cause)}};function $n(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 bt("Generate failed",i,{cause:n})}var ui=s=>s.replace(/^\W+|\W+$/g,""),di=(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]},mi=s=>{let e=new Set,t=[];for(let n of s)e.has(n)||(e.add(n),t.push(n));return t},gi=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}},Ai=s=>{let e=s.match(/^(\d+)[.,\s]+(.*)$/);return e&&e[2]!==void 0?e[2].trim():s},hi=(s,e)=>{let t=[];for(let n=0;n<s.length;n+=e)t.push(s.slice(n,n+e));return t},$e={trimNonAlphaNum:ui,splitIntoTwo:di,dedup:mi,extractIdAndText:gi,extractIndexPrefixedText:Ai,batchArray:hi};var Ln=class extends M{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."
|
|
182
|
+
`),ri=(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?Cr(e):s.type?.name==="image"&&typeof e=="object"||s.type?.name==="audio"&&typeof e=="object"||typeof e=="string"?e:JSON.stringify(e,null,2)},Nr=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";default:return"string"}})();return s?.isArray?`json array of ${e} items`:e};function Ur(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 si=(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 yt(s){let e=s.trim();return e.length>0?`${e.charAt(0).toUpperCase()}${e.slice(1)}${e.endsWith(".")?"":"."}`:""}function ii(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 ai(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 It(s,e,t,n){if(!e?.resultPicker||s.length<=1)return 0;let o=e.resultPicker;if((t?ii(t,n):!1)&&t){let l=ai(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 $r(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 It(o,t,s,e)}var k=class extends te{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??ae)(this.signature,n),this.asserts=this.options?.asserts??[],this.streamingAsserts=this.options?.streamingAsserts??[],this.excludeContentFromTrace=t?.excludeContentFromTrace??!1,this.usage=[],t?.functions&&(this.functions=Sn(t.functions))}getSignatureName(){return this.signature.getDescription()||"unknown_signature"}getMetricsInstruments(){return Cn()}updateMeter(e){Cn(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:d,thinkingTokenBudget:u,showThoughts:m}=n??{},g=await $r(t,l,{resultPicker:n?.resultPicker}),A=t?.history(g,l)??[];if(A.length===0)throw new Error("No chat prompt found");let x={...n?.modelConfig,...n?.sampleCount?{n:n.sampleCount}:{},...n?.sampleCount&&n?.modelConfig?.temperature===1?{temperature:.8}:{}},h=this.isDebug(e,n),y=a===0;return await e.chat({chatPrompt:A,functions:r,functionCall:i,modelConfig:x,model:p},{sessionId:l,rateLimiter:c,stream:d,debug:h,debugHideSystemPrompt:!y,thinkingTokenBudget:u,showThoughts:m,traceContext:o,abortSignal:n?.abortSignal,stepIndex:a})}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,d=n.model,u=this.createStates(n.sampleCount??1),m=this.usage,g=o===0,{functions:A,functionCall:x}=Ir(l,p,g),h=await this.forwardSendRequest({ai:e,mem:t,options:n,traceContext:i,functions:A,functionCall:x,stepIndex:o});h instanceof ReadableStream?yield*kr({ai:e,model:d,res:h,mem:t,sessionId:a,functions:A,strictMode:c,span:r,states:u,usage:m,asserts:this.asserts,streamingAsserts:this.streamingAsserts,fieldProcessors:this.fieldProcessors,streamingFieldProcessors:this.streamingFieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,functionResultFormatter:n?.functionResultFormatter??this.options?.functionResultFormatter}):yield*Er({ai:e,model:d,res:h,mem:t,sessionId:a,functions:A,span:r,strictMode:c,states:u,usage:m,asserts:this.asserts,fieldProcessors:this.fieldProcessors,thoughtFieldName:this.thoughtFieldName,excludeContentFromTrace:this.excludeContentFromTrace,signature:this.signature,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 Fe,d;if(o?.functions&&o.functions.length>0){let y=this.options?.promptTemplate??ae,f={functions:o.functions,thoughtFieldName:this.thoughtFieldName};this.promptTemplate=new y(this.signature,f)}let u,m=performance.now();Array.isArray(t)?(Dt(t),u=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos})):u=this.promptTemplate.render(t,{examples:this.examples,demos:this.demos});let g=performance.now()-m,A=this.getMetricsInstruments();A&>(A,"prompt_render",g,this.getSignatureName());let x=performance.now();c.addRequest(u,o.sessionId);let h=performance.now()-x;A&>(A,"memory_update",h,this.getSignatureName());e:for(let y=0;y<p;y++){for(let R=0;R<l;R++)try{let b=this.forwardCore({options:o,ai:e,mem:c,stepIndex:y,span:r,traceContext:i});for await(let L of b)L!==void 0&&(yield{version:R,index:L.index,delta:L.delta});if(Pr(c,a,n,o?.sessionId)){let L=this.getMetricsInstruments();L&&mt(L,y+1,p,this.getSignatureName());continue e}let $=this.getMetricsInstruments();if($){mt($,y+1,p,this.getSignatureName());let L=new Set;n.forEach(vt=>{vt.functionsExecuted.forEach(St=>L.add(St))}),L.size>0&&vn($,!0,L.size,!0,!1,this.getSignatureName()),dr($,this.fieldProcessors.length,this.streamingFieldProcessors.length,this.getSignatureName())}return}catch(b){let w,$=this.isDebug(e,o),L=this.getLogger(e,o),vt=this.getMetricsInstruments(),St=this.getSignatureName(),Ot={error:b,errCount:R,logger:L,metricsInstruments:vt,signatureName:St,span:r,debug:$};if(r?.recordException(b),b instanceof F)w=hr(Ot),d=b;else if(b instanceof Z)w=xr(Ot),d=b;else if(b instanceof S)fr(Ot);else if(!(b instanceof V))throw $n(b,e,this.signature);w&&(c.addRequest([{role:"user",content:this.promptTemplate.renderExtraFields(w)}],o.sessionId),c.addTag("error",o.sessionId))}let f=this.getMetricsInstruments();throw f&&wn(f,l,!1,l,this.getSignatureName()),$n(new Error(`Unable to fix validation error: ${d?.toString()}`),e,this.signature)}throw A&&mt(A,p,p,this.getSignatureName()),$n(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&>(a,"state_creation",i,this.getSignatureName());let l=n?.tracer??this.options?.tracer??e.getOptions().tracer,p=this.functions;if(n?.functions&&(p=Sn(n.functions,this.functions)),!l){yield*this._forward2(e,t,r,{...n,functions:p});return}let c=p?.map(h=>h.name).join(","),d={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}:{}},u=this.traceLabel&&n.traceLabel?`${this.traceLabel} > ${n.traceLabel}`:n.traceLabel??this.traceLabel,m=u?`AxGen > ${u}`:"AxGen",g=l.startSpan(m,{kind:pi.SERVER,attributes:d}),A=li.active(),x=ci.setSpan(A,g);try{if(this.excludeContentFromTrace||g.addEvent("input",{content:JSON.stringify(t,null,2)}),yield*this._forward2(e,t,r,{...n,functions:p},g,x),!this.excludeContentFromTrace){let h=r.map(f=>f.values),y=h.length===1?h[0]:h;g.addEvent("output",{content:JSON.stringify(y,null,2)})}}finally{g.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,d=!1;try{let u=this.getMetricsInstruments();u&&Ar(u,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??{}),g=[],A=0,x=0;for await(let w of m)w.version!==A&&(g=[]),A=w.version,g=En(g,w),x++;l=A;let h=performance.now();d=!!n?.resultPicker;let y=await It(g,{resultPicker:n?.resultPicker},n?.mem,n?.sessionId),f=performance.now()-h,b=g[y]?.delta??{};if(this.trace={...t,...b},d&&this.isDebug(e,n)){let w=this.getLogger(e,n);Oo(g.length,y,f,w)}return a=!0,u&&(gr(u,g.length,d,d?f:void 0,r),mr(u,i,x,void 0,r)),b}catch(u){throw a=!1,u}finally{let u=performance.now()-o,m=this.getMetricsInstruments();m&&(cr(m,u,a,r,e.getName(),n?.model?String(n.model):void 0),p&&vn(m,p,c,c>0,!1,r),l>0&&wn(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=En(r,p);let a=await It(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()}},bt=class extends Error{details;constructor(e,t,n){super(e),this.name="AxGenerateError",this.details=t,n?.cause&&(this.cause=n.cause)}};function $n(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 bt("Generate failed",i,{cause:n})}var ui=s=>s.replace(/^\W+|\W+$/g,""),di=(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]},mi=s=>{let e=new Set,t=[];for(let n of s)e.has(n)||(e.add(n),t.push(n));return t},gi=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}},Ai=s=>{let e=s.match(/^(\d+)[.,\s]+(.*)$/);return e&&e[2]!==void 0?e[2].trim():s},hi=(s,e)=>{let t=[];for(let n=0;n<s.length;n+=e)t.push(s.slice(n,n+e));return t},$e={trimNonAlphaNum:ui,splitIntoTwo:di,dedup:mi,extractIdAndText:gi,extractIndexPrefixedText:Ai,batchArray:hi};var Ln=class extends k{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."
|
|
183
183
|
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}=$e.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 Bn=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 Lr=new N,qn=class{name;context;constructor(e,t){this.name=e,this.context=t}getName(){return this.name}getContext(){return this.context}},zn=class{ai;db;debug;constructor(e){this.db=new X,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(`${Lr.whiteBright(`query: ${e}`)}
|
|
184
184
|
${Lr.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 Br=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 qr(s,e){return s.filter(t=>!e.has(t))}function zr(s){let e={};for(let t of s)e[t]=(e[t]||0)+1;return e}function ne(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 xi(s,e){return ne(s)===ne(e)?1:0}function fi(s,e){let t=ne(s).split(" "),n=ne(e).split(" "),o=zr(t),r=zr(n),i=0;for(let p in o){let c=o[p]??0,d=r[p]??0;i+=Math.min(c,d)}if(i===0)return 0;let a=i/t.length,l=i/n.length;return 2*a*l/(a+l)}function yi(s,e,t,n=!1){let o=ne(s).split(" "),r=ne(e).split(" "),i=ne(t).split(" "),a=new Set([...Br,...o]);r=qr(r,a),i=qr(i,a);let l=0,p=l/r.length,c=l/i.length,d=2*p*c/(p+c);return n?c:d}var Ii={emScore:xi,f1Score:fi,novelF1ScoreOptimized:yi};var jn=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");let l=await this.program.forward(this.ai,a),p=await e({prediction:l,example:a});o+=p}let r=n>0?o/n:0;console.log(`
|
|
185
185
|
Performance: `,o,"/",n,"Average Score: ",r,`
|
|
@@ -242,7 +242,7 @@ Score: ${t.value.score.toFixed(3)}
|
|
|
242
242
|
Config: ${JSON.stringify(t.value.config)}
|
|
243
243
|
${e}`;break;default:n=`[ UNKNOWN OPTIMIZER EVENT ]
|
|
244
244
|
${JSON.stringify(t)}
|
|
245
|
-
${e}`}s(n)}},Kn=Hr();var Kr={enabled:!0,enabledCategories:["optimization","convergence","resource_usage","teacher_student","checkpointing","pareto"],maxLabelLength:100,samplingRate:1},Rt,Ri=s=>{if(Rt)return Rt;if(s)return Rt=wi(s),Rt};var Ct=Kr,Ci=s=>{Ct={...Ct,...s}},Ti=()=>({...Ct}),wi=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"})}),H=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let o=String(n),r=Ct.maxLabelLength;e[t]=o.length>r?o.substring(0,r):o}return e},vi=(s,e,t,n,o)=>{try{let r=H({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)}},Si=(s,e,t,n,o,r)=>{try{let i=H({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)}},Oi=(s,e,t)=>{try{let n=H({reason:e,optimizer_type:t});s.earlyStoppingCounter&&s.earlyStoppingCounter.add(1,n)}catch(n){console.warn("Failed to record early stopping metric:",n)}},Mi=(s,e,t,n,o)=>{try{let r=H({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)}},ki=(s,e,t)=>{try{let n=H({optimizer_type:t});s.optimizationDurationHistogram&&s.optimizationDurationHistogram.record(e,n)}catch(n){console.warn("Failed to record optimization duration metric:",n)}},Ei=(s,e,t,n)=>{try{let o=H({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)}},Pi=(s,e,t,n,o)=>{try{let r=H({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)}},_i=(s,e,t,n,o)=>{try{let r=H({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)}},Fi=(s,e,t,n,o,r)=>{try{let i=H({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)}},Gi=(s,e,t,n)=>{try{let o=H({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)}},Di=(s,e,t,n)=>{try{let o=H({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)}},Tt=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}},oe=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 Tt({maxTokens:1e6});this.costTracker=e.costTracker??t,this.metricsInstruments=Ri(k.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,g,A,x,h,y,f,R={},b)=>{this.getOptimizerLogger(b)?.({name:"RoundProgress",value:{round:m,totalRounds:b?.maxIterations??0,currentScore:g,bestScore:y,configuration:A}}),this.updateOptimizationProgress(m,g,A,x,h,y,f,R,b)},p=(m,g)=>{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)},d=await this.compile(e,t,{...n,overrideOnProgress:c,overrideOnEarlyStop:p}),u=Date.now()-o;return this.recordOptimizationComplete(u,!0,r,i),a&&this.getLogger(n)?.({name:"Notification",id:"optimization_early_stop",value:`Optimization stopped early due to ${a}`}),{demos:d.demos,stats:d.stats,bestScore:d.bestScore,finalConfiguration:d.finalConfiguration,scoreHistory:d.scoreHistory,configurationHistory:d.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 d=p.length>0?Math.max(...p.map(u=>Math.max(...Object.values(u.scores)))):0;return{demos:p.length>0?[...p[0].demos]:void 0,stats:this.stats,bestScore:d,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 d=p[c],u=async({prediction:m,example:g})=>{let A=await t({prediction:m,example:g}),x=0;for(let[h,y]of Object.entries(A))x+=y*(d[h]||0);return x};try{let m=await this.compile(e,u,{...n,verbose:!1}),g=await this.evaluateWithMultiObjective(e,m,t);o.push({scores:g,demos:m.demos,configuration:{...m.finalConfiguration,weights:d,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:d,example:u})=>{let m=await t({prediction:d,example:u}),g=m[p]||0,A=0;for(let[x,h]of Object.entries(m))x!==p&&h<.3&&(A+=(.3-h)*2);return g-A};try{let d=await this.compile(e,c,{...n,verbose:!1}),u=await this.evaluateWithMultiObjective(e,d,t);o.push({scores:u,demos:d.demos,configuration:{...d.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 M(e.getSignature());t.demos&&o.setDemos(t.demos);let r=[];for(let c of this.examples){let d=await o.forward(this.studentAI,c);r.push({prediction:d,example:c})}let i=this.getValidationSet(),a={},l=i.slice(0,Math.min(5,i.length));for(let c of l)try{let d=await o.forward(this.studentAI,c),u=await n({prediction:d,example:c});for(let[m,g]of Object.entries(u))a[m]||(a[m]=[]),a[m].push(g)}catch{}let p={};for(let[c,d]of Object.entries(a))p[c]=d.length>0?d.reduce((u,m)=>u+m,0)/d.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,d=p.scores[r]||0;i+=c*(d-l),l=Math.max(l,d)}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 d={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(d),p=!0}catch(d){throw p=!1,d}finally{let d=Date.now()-l;this.recordCheckpointMetrics("save",d,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;Fi(this.metricsInstruments,n,o,this.examples.length,this.getValidationSet().length,e)}Di(this.metricsInstruments,e,this.targetScore,void 0)}}recordOptimizationComplete(e,t,n,o){if(!this.metricsInstruments)return;vi(this.metricsInstruments,e,t,n,o),ki(this.metricsInstruments,e,n);let r=this.costTracker?.getCurrentCost()??0,i=this.costTracker?.getTotalTokens()??0;Mi(this.metricsInstruments,i,r,n)}recordConvergenceMetrics(e,t,n,o,r){this.metricsInstruments&&Si(this.metricsInstruments,e,t,n,o,r)}recordEarlyStoppingMetrics(e,t){this.metricsInstruments&&Oi(this.metricsInstruments,e,t)}recordTeacherStudentMetrics(e,t,n){this.metricsInstruments&&Ei(this.metricsInstruments,e,t,n)}recordCheckpointMetrics(e,t,n,o){this.metricsInstruments&&Pi(this.metricsInstruments,e,t,n,o)}recordParetoMetrics(e,t,n,o){this.metricsInstruments&&_i(this.metricsInstruments,e,t,n,o)}recordPerformanceMetrics(e,t,n){this.metricsInstruments&&Gi(this.metricsInstruments,e,t,n)}isOptimizerLoggingEnabled(e){return this.debugOptimizer||(e?.verbose??this.verbose??!1)}getOptimizerLogger(e){if(this.isOptimizerLoggingEnabled(e))return this.optimizerLogger??Kn}getStats(){return{...this.stats}}reset(){this.stats=this.initializeStats(),this.costTracker?.reset(),this.currentRound=0,this.scoreHistory=[],this.configurationHistory=[]}};var Le=class extends oe{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=Ni(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 d=l.slice(c,c+this.batchSize);for(let u of d){if(!u)continue;let m=l.filter(x=>x!==u);e.setExamples(m);let g=this.getTeacherOrStudentAI();this.stats.totalCalls++;let A;try{A=await e.forward(g,u,a),this.costMonitoring&&(this.stats.estimatedTokenUsage+=JSON.stringify(u).length/4+JSON.stringify(A).length/4),await n({prediction:A,example:u})>=.5&&(this.traces=[...this.traces,...e.getTraces()],this.stats.successfulDemos++)}catch{A={}}if(this.traces.length>=i)return}}if(this.earlyStoppingPatience>0){let d=this.traces.length-p;if(!this.stats.earlyStopping)this.stats.earlyStopping={bestScoreRound:d>0?t:0,patienceExhausted:!1,reason:"No improvement detected"};else if(d>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=Ui(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 Ui(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 Ni=(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 Vn=class extends oe{maxBootstrappedDemos;maxLabeledDemos;numCandidates;initTemperature;numTrials;minibatch;minibatchSize;minibatchFullEvalSteps;programAwareProposer;dataAwareProposer;viewDataBatchSize;tipAwareProposer;fewshotAwareProposer;earlyStoppingTrials;minImprovementThreshold;bayesianOptimization;acquisitionFunction;explorationWeight;sampleCount;miproConfigHistory=[];surrogateModel=new Map;constructor(e){super(e);let t=e.options||{};this.numCandidates=t.numCandidates??5,this.initTemperature=t.initTemperature??.7,this.maxBootstrappedDemos=t.maxBootstrappedDemos??3,this.maxLabeledDemos=t.maxLabeledDemos??4,this.numTrials=t.numTrials??30,this.minibatch=t.minibatch??!0,this.minibatchSize=t.minibatchSize??25,this.minibatchFullEvalSteps=t.minibatchFullEvalSteps??10,this.programAwareProposer=t.programAwareProposer??!0,this.dataAwareProposer=t.dataAwareProposer??!0,this.viewDataBatchSize=t.viewDataBatchSize??10,this.tipAwareProposer=t.tipAwareProposer??!0,this.fewshotAwareProposer=t.fewshotAwareProposer??!0,this.earlyStoppingTrials=t.earlyStoppingTrials??5,this.minImprovementThreshold=t.minImprovementThreshold??.01,this.bayesianOptimization=t.bayesianOptimization??!1,this.acquisitionFunction=t.acquisitionFunction??"expected_improvement",this.explorationWeight=t.explorationWeight??.1,this.sampleCount=t.sampleCount??1,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=`
|
|
245
|
+
${e}`}s(n)}},Kn=Hr();var Kr={enabled:!0,enabledCategories:["optimization","convergence","resource_usage","teacher_student","checkpointing","pareto"],maxLabelLength:100,samplingRate:1},Rt,Ri=s=>{if(Rt)return Rt;if(s)return Rt=wi(s),Rt};var Ct=Kr,Ci=s=>{Ct={...Ct,...s}},Ti=()=>({...Ct}),wi=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"})}),H=s=>{let e={};for(let[t,n]of Object.entries(s))if(n!=null){let o=String(n),r=Ct.maxLabelLength;e[t]=o.length>r?o.substring(0,r):o}return e},vi=(s,e,t,n,o)=>{try{let r=H({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)}},Si=(s,e,t,n,o,r)=>{try{let i=H({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)}},Oi=(s,e,t)=>{try{let n=H({reason:e,optimizer_type:t});s.earlyStoppingCounter&&s.earlyStoppingCounter.add(1,n)}catch(n){console.warn("Failed to record early stopping metric:",n)}},Mi=(s,e,t,n,o)=>{try{let r=H({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)}},ki=(s,e,t)=>{try{let n=H({optimizer_type:t});s.optimizationDurationHistogram&&s.optimizationDurationHistogram.record(e,n)}catch(n){console.warn("Failed to record optimization duration metric:",n)}},Ei=(s,e,t,n)=>{try{let o=H({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)}},Pi=(s,e,t,n,o)=>{try{let r=H({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)}},_i=(s,e,t,n,o)=>{try{let r=H({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)}},Fi=(s,e,t,n,o,r)=>{try{let i=H({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)}},Gi=(s,e,t,n)=>{try{let o=H({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)}},Di=(s,e,t,n)=>{try{let o=H({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)}},Tt=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}},oe=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 Tt({maxTokens:1e6});this.costTracker=e.costTracker??t,this.metricsInstruments=Ri(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,g,A,x,h,y,f,R={},b)=>{this.getOptimizerLogger(b)?.({name:"RoundProgress",value:{round:m,totalRounds:b?.maxIterations??0,currentScore:g,bestScore:y,configuration:A}}),this.updateOptimizationProgress(m,g,A,x,h,y,f,R,b)},p=(m,g)=>{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)},d=await this.compile(e,t,{...n,overrideOnProgress:c,overrideOnEarlyStop:p}),u=Date.now()-o;return this.recordOptimizationComplete(u,!0,r,i),a&&this.getLogger(n)?.({name:"Notification",id:"optimization_early_stop",value:`Optimization stopped early due to ${a}`}),{demos:d.demos,stats:d.stats,bestScore:d.bestScore,finalConfiguration:d.finalConfiguration,scoreHistory:d.scoreHistory,configurationHistory:d.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 d=p.length>0?Math.max(...p.map(u=>Math.max(...Object.values(u.scores)))):0;return{demos:p.length>0?[...p[0].demos]:void 0,stats:this.stats,bestScore:d,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 d=p[c],u=async({prediction:m,example:g})=>{let A=await t({prediction:m,example:g}),x=0;for(let[h,y]of Object.entries(A))x+=y*(d[h]||0);return x};try{let m=await this.compile(e,u,{...n,verbose:!1}),g=await this.evaluateWithMultiObjective(e,m,t);o.push({scores:g,demos:m.demos,configuration:{...m.finalConfiguration,weights:d,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:d,example:u})=>{let m=await t({prediction:d,example:u}),g=m[p]||0,A=0;for(let[x,h]of Object.entries(m))x!==p&&h<.3&&(A+=(.3-h)*2);return g-A};try{let d=await this.compile(e,c,{...n,verbose:!1}),u=await this.evaluateWithMultiObjective(e,d,t);o.push({scores:u,demos:d.demos,configuration:{...d.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 k(e.getSignature());t.demos&&o.setDemos(t.demos);let r=[];for(let c of this.examples){let d=await o.forward(this.studentAI,c);r.push({prediction:d,example:c})}let i=this.getValidationSet(),a={},l=i.slice(0,Math.min(5,i.length));for(let c of l)try{let d=await o.forward(this.studentAI,c),u=await n({prediction:d,example:c});for(let[m,g]of Object.entries(u))a[m]||(a[m]=[]),a[m].push(g)}catch{}let p={};for(let[c,d]of Object.entries(a))p[c]=d.length>0?d.reduce((u,m)=>u+m,0)/d.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,d=p.scores[r]||0;i+=c*(d-l),l=Math.max(l,d)}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 d={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(d),p=!0}catch(d){throw p=!1,d}finally{let d=Date.now()-l;this.recordCheckpointMetrics("save",d,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;Fi(this.metricsInstruments,n,o,this.examples.length,this.getValidationSet().length,e)}Di(this.metricsInstruments,e,this.targetScore,void 0)}}recordOptimizationComplete(e,t,n,o){if(!this.metricsInstruments)return;vi(this.metricsInstruments,e,t,n,o),ki(this.metricsInstruments,e,n);let r=this.costTracker?.getCurrentCost()??0,i=this.costTracker?.getTotalTokens()??0;Mi(this.metricsInstruments,i,r,n)}recordConvergenceMetrics(e,t,n,o,r){this.metricsInstruments&&Si(this.metricsInstruments,e,t,n,o,r)}recordEarlyStoppingMetrics(e,t){this.metricsInstruments&&Oi(this.metricsInstruments,e,t)}recordTeacherStudentMetrics(e,t,n){this.metricsInstruments&&Ei(this.metricsInstruments,e,t,n)}recordCheckpointMetrics(e,t,n,o){this.metricsInstruments&&Pi(this.metricsInstruments,e,t,n,o)}recordParetoMetrics(e,t,n,o){this.metricsInstruments&&_i(this.metricsInstruments,e,t,n,o)}recordPerformanceMetrics(e,t,n){this.metricsInstruments&&Gi(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??Kn}getStats(){return{...this.stats}}reset(){this.stats=this.initializeStats(),this.costTracker?.reset(),this.currentRound=0,this.scoreHistory=[],this.configurationHistory=[]}};var Le=class extends oe{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=Ni(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 d=l.slice(c,c+this.batchSize);for(let u of d){if(!u)continue;let m=l.filter(x=>x!==u);e.setExamples(m);let g=this.getTeacherOrStudentAI();this.stats.totalCalls++;let A;try{A=await e.forward(g,u,a),this.costMonitoring&&(this.stats.estimatedTokenUsage+=JSON.stringify(u).length/4+JSON.stringify(A).length/4),await n({prediction:A,example:u})>=.5&&(this.traces=[...this.traces,...e.getTraces()],this.stats.successfulDemos++)}catch{A={}}if(this.traces.length>=i)return}}if(this.earlyStoppingPatience>0){let d=this.traces.length-p;if(!this.stats.earlyStopping)this.stats.earlyStopping={bestScoreRound:d>0?t:0,patienceExhausted:!1,reason:"No improvement detected"};else if(d>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=Ui(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 Ui(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 Ni=(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 Vn=class extends oe{maxBootstrappedDemos;maxLabeledDemos;numCandidates;initTemperature;numTrials;minibatch;minibatchSize;minibatchFullEvalSteps;programAwareProposer;dataAwareProposer;viewDataBatchSize;tipAwareProposer;fewshotAwareProposer;earlyStoppingTrials;minImprovementThreshold;bayesianOptimization;acquisitionFunction;explorationWeight;sampleCount;miproConfigHistory=[];surrogateModel=new Map;constructor(e){super(e);let t=e.options||{};this.numCandidates=t.numCandidates??5,this.initTemperature=t.initTemperature??.7,this.maxBootstrappedDemos=t.maxBootstrappedDemos??3,this.maxLabeledDemos=t.maxLabeledDemos??4,this.numTrials=t.numTrials??30,this.minibatch=t.minibatch??!0,this.minibatchSize=t.minibatchSize??25,this.minibatchFullEvalSteps=t.minibatchFullEvalSteps??10,this.programAwareProposer=t.programAwareProposer??!0,this.dataAwareProposer=t.dataAwareProposer??!0,this.viewDataBatchSize=t.viewDataBatchSize??10,this.tipAwareProposer=t.tipAwareProposer??!0,this.fewshotAwareProposer=t.fewshotAwareProposer??!0,this.earlyStoppingTrials=t.earlyStoppingTrials??5,this.minImprovementThreshold=t.minImprovementThreshold??.01,this.bayesianOptimization=t.bayesianOptimization??!1,this.acquisitionFunction=t.acquisitionFunction??"expected_improvement",this.explorationWeight=t.explorationWeight??.1,this.sampleCount=t.sampleCount??1,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=`
|
|
246
246
|
Analyze this language model program and provide a concise summary of its purpose and structure.
|
|
247
247
|
|
|
248
248
|
Program Signature: ${e.getSignature()}
|
|
@@ -282,8 +282,8 @@ Requirements:
|
|
|
282
282
|
5. Keep it concise but comprehensive
|
|
283
283
|
|
|
284
284
|
Generate a single, well-crafted instruction:
|
|
285
|
-
Instruction:`;try{let d=await n.chat({chatPrompt:[{role:"user",content:l}]});if("results"in d){let u=d.results[0]?.content?.trim();if(u&&u.length>10)return u}}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,d=await this.generateInstruction({tip:c,candidateIndex:l,ai:o,programSummary:r,datasetSummary:i,previousInstructions:n});n.push(d)}return n}async bootstrapFewShotExamples(e,t){return(await new Le({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,d=[],u=0;if(this.resumeFromCheckpoint){let m=await this.loadCheckpoint(this.resumeFromCheckpoint,a);m&&m.optimizerType==="MiPRO"&&(this.restoreFromCheckpoint(m),u=m.currentRound,p=m.bestScore,l=m.bestConfiguration||l,c=m.stats.convergenceInfo?.stagnationRounds||0)}for(let m=u;m<this.numTrials;m++){let g;this.bayesianOptimization&&this.miproConfigHistory.length>2?g=await this.selectConfigurationViaBayesianOptimization(o,t,n):g={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 A=await this.evaluateConfig(e,g,t,n,r,i,m+1);this.updateSurrogateModel(g,A),d.push(A);let x=A-p;if(x>this.minImprovementThreshold?(p=A,l=g,c=0):c++,await this.updateOptimizationProgress(m+1,A,g,"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:A,bestScore:p,tokensUsed:this.stats.resourceUsage.totalTokens,timeElapsed:Date.now(),successfulExamples:this.stats.successfulDemos,totalExamples:this.examples.length,currentConfiguration:g,convergenceInfo:{improvement:x,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=d.length>1?p-d[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 M(e.getSignature());this.applyConfigToProgram(l,t,n,o);let p=0,c=0,d;if(this.minibatch){let g=Math.min(this.minibatchSize,r.length);a%this.minibatchFullEvalSteps===0||a>this.numTrials*.8?d=Math.min(r.length,g*2):d=Math.max(3,Math.min(g,r.length))}else d=r.length;let m=this.shuffleArray([...Array(r.length).keys()]).slice(0,d).map(g=>r[g]);for(let g of m)try{let A=await l.forward(this.studentAI,g,this.sampleCount>1?{sampleCount:this.sampleCount,resultPicker:$i()}:void 0),x=await i({prediction:A,example:g});p+=x,c++,this.stats.totalCalls++}catch{}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;r?.auto&&this.configureAuto(r.auto);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:d}=await this.runOptimization(e,a,l,p,i,t,n);this.checkTargetScore(d)&&this.triggerEarlyStopping(`Target score ${this.targetScore} reached with score ${d}`,this.numTrials);let u;"getSignature"in e&&typeof e.getSignature=="function"?u=e.getSignature():u="input -> output";let m=new M(u);return this.applyConfigToAxGen(m,c,a,l),this.updateResourceUsage(o),this.stats.convergenceInfo.converged=!0,this.stats.convergenceInfo.finalImprovement=d,await this.saveFinalCheckpoint("MiPRO",this.getConfiguration(),d,c,{bootstrappedDemos:a.length,labeledExamples:l.length,instructions:p.length,optimizedGen:!!m},n),{demos:a,stats:this.stats,bestScore:d,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`${e.instruction.length}_${e.bootstrappedDemos}_${e.labeledExamples}`}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),d=1-((((i*c+r)*c+o)*c+n)*c+t)*c*Math.exp(-p*p);return l*d}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}},$i=()=>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 Li(s,...e){let t="";for(let n=0;n<s.length;n++)if(t+=s[n]??"",n<e.length){let o=e[n];if(Jn(o)){let r=t.match(/(\w+)\s*:\s*$/);if(r&&(o.isOptional||o.isInternal)){let c=r[1];o.isOptional&&(c+="?"),o.isInternal&&(c+="!"),t=t.replace(/(\w+)(\s*:\s*)$/,`${c}$2`)}let{isOptional:i,isInternal:a,...l}=o;t+=wt(l)}else if(Yn(o))t+=Wn(o);else if(typeof o=="string"||o instanceof O)t+=Vr(o);else throw new Error("Unsupported template interpolation value")}return new O(t)}function Bi(s,...e){let t="";for(let n=0;n<s.length;n++)if(t+=s[n]??"",n<e.length){let o=e[n];if(Jn(o)){let r=t.match(/(\w+)\s*:\s*$/);if(r&&(o.isOptional||o.isInternal)){let c=r[1];o.isOptional&&(c+="?"),o.isInternal&&(c+="!"),t=t.replace(/(\w+)(\s*:\s*)$/,`${c}$2`)}let{isOptional:i,isInternal:a,...l}=o;t+=wt(l)}else if(Yn(o))t+=Wn(o);else if(typeof o=="string"||o instanceof O)t+=Vr(o);else throw new Error("Unsupported template interpolation value")}return new M(t)}function Vr(s){if(typeof s=="string")return s;if(Jn(s))return wt(s);if(Yn(s))return Wn(s);if(s instanceof O){let e=s.toString(),t=e.indexOf(" -> ");return t!==-1?e.substring(t+4):e}throw new Error(`Unsupported template value type: ${typeof s}`)}function wt(s){let e=s.type;return s.isArray&&(e+="[]"),s.options&&s.options.length>0&&s.type==="class"&&(e+=` "${s.options.join(", ")}"`),s.description&&(e+=` "${s.description}"`),e}function Wn(s){let e=s.name;return s.isOptional&&(e+="?"),s.isInternal&&(e+="!"),s.type&&(e+=`:${wt(s.type)}`),s.description&&!s.type?.description&&(e+=` "${s.description}"`),e}function Jn(s){return s!==null&&typeof s=="object"&&s!==void 0&&"type"in s&&typeof s.type=="string"}function Yn(s){return s!==null&&typeof s=="object"&&s!==void 0&&"name"in s&&typeof s.name=="string"}var qi={string:s=>({type:"string",description:s}),number:s=>({type:"number",description:s}),boolean:s=>({type:"boolean",description:s}),date:s=>({type:"date",description:s}),datetime:s=>({type:"datetime",description:s}),json:s=>({type:"json",description:s}),image:s=>({type:"image",description:s}),audio:s=>({type:"audio",description:s}),class:(s,e)=>({type:"class",options:s,description:e}),code:(s,e)=>({type:"code",options:[s],description:e}),array:s=>({...s,isArray:!0}),optional:s=>({...s,isOptional:!0}),internal:s=>({...s,isInternal:!0})};var Be=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 le(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(d=>({result:d,originalIndex:c}))}),a=await Promise.all(i);for(let{result:l,originalIndex:p}of a)n[p]=l}return n}var qe=class{steps=[];parallelGroups=[];analyzer=new Be;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 u=new Set;for(let m of r)this.analyzeMapTransformation(m).forEach(A=>u.add(A));p=Array.from(u)}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 u=r?this.analyzer.analyzeMappingDependencies(r,"derive"):[];l=[a.inputFieldName,...u].filter((m,g,A)=>A.indexOf(m)===g)}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 u of l)this.getAllProducedFields().includes(u)||this.initialFields.add(u);let d={type:c,nodeName:t,dependencies:l,produces:p,stepFunction:e,stepIndex:this.steps.length};this.steps.push(d)}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 le(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(d=>d&&typeof d=="object"&&"_parallelResults"in d);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 ze=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}`,d;if("forward"in o&&typeof o.forward=="function")d=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`]:d}}),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}},Qn=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}`,d;if("forward"in o&&typeof o.forward=="function")d=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`]:d}}),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 Xn=class{nodes=new Map;flowDefinition=[];nodeGenerators=new Map;loopStack=[];stepLabels=new Map;branchContext=null;autoParallelConfig;executionPlanner=new qe;program;nodeUsage=new Map;nodeTraces=new Map;toCamelCase(e){return e.replace(/_([a-z])/g,(t,n)=>n.toUpperCase())}inferSignatureFromFlow(){let e=this.executionPlanner.getExecutionPlan();if(this.nodeGenerators.size===0&&e.steps.length===0){let c=new O;return c.addInputField({name:"userInput",type:{name:"string"},description:"User input to the flow"}),c.addOutputField({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),c}let t=new Set,n=new Set;for(let c of e.steps)c.produces.forEach(d=>t.add(d)),c.dependencies.forEach(d=>n.add(d));let o=new Set;for(let c of 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(d=>r.add(d))}else for(let c of t){let d=!1;for(let u of e.steps)if(u.dependencies.includes(c)){d=!0;break}if(!d)if(c.endsWith("Result")){let u=c.replace("Result",""),m=this.nodeGenerators.get(u);if(m){let g=m.getSignature(),x=new O(g).getOutputFields();for(let h of x)r.add(h.name)}else r.add(c)}else r.add(c)}if(o.size===0&&r.size===0){let c=[],d=[];for(let[m,g]of this.nodeGenerators){let A=g.getSignature(),x=new O(A);for(let h of x.getInputFields()){let y=this.toCamelCase(`${m}_${h.name}`);c.push({name:y,type:h.type,description:h.description,isOptional:h.isOptional,isInternal:h.isInternal})}for(let h of x.getOutputFields()){let y=this.toCamelCase(`${m}_${h.name}`);d.push({name:y,type:h.type,description:h.description,isOptional:h.isOptional,isInternal:h.isInternal})}}let u=new O;return c.length>0?u.setInputFields(c):u.addInputField({name:"userInput",type:{name:"string"},description:"User input to the flow"}),d.length>0?u.setOutputFields(d):u.addOutputField({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),u}let a=new O,l=[];for(let c of 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 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}}ensureProgram(){let e=this.inferSignatureFromFlow();this.program=new te(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 this.nodeTraces)e.push(...n);return e}setDemos(e){this.ensureProgram(),this.program.setDemos(e)}getUsage(){let e=[];for(let[t,n]of this.nodeUsage)e.push(...n);return De(e)}resetUsage(){this.nodeUsage.clear();for(let[e,t]of this.nodeGenerators)t&&"resetUsage"in t&&t.resetUsage()}resetTraces(){this.nodeTraces.clear()}getUsageReport(){let e={};for(let[t,n]of this.nodeUsage)e[t]=De(n);return e}getTracesReport(){let e={};for(let[t,n]of 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){this.resetUsage(),this.resetTraces();let o;if(Array.isArray(t)){let l=t.filter(p=>p.role==="user").pop();if(!l)throw new Error("No user message found in values array");o=l.values}else o=t;this.nodeGenerators.size>0&&this.ensureProgram();let r={...o},i={mainAi:e,mainOptions:n?{...n,model:n.model?String(n.model):void 0}:void 0};if(n?.autoParallel!==!1&&this.autoParallelConfig.enabled){this.executionPlanner.setInitialFields(Object.keys(o));let l=this.executionPlanner.createOptimizedExecution(this.autoParallelConfig.batchSize);for(let p of l)r=await p(r,i)}else for(let l of this.flowDefinition)r=await l(r,i);return r}node(e,t){if(typeof t=="string"||t instanceof O){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=new M(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 le(n,async(a,l)=>a(r),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=o=>{if(Array.isArray(e))throw new Error("Array of transforms requires parallel: true option");return e(o)};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)}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),d=p?.traceLabel?`Node:${e} (${p.traceLabel})`:`Node:${e}`,u;if("forward"in o&&typeof o.forward=="function"){if(u=await o.forward(l,c,{...p,traceLabel:d}),"getUsage"in o&&typeof o.getUsage=="function"){let m=o.getUsage();if(m&&m.length>0){let g=this.nodeUsage.get(e)||[];this.nodeUsage.set(e,[...g,...m])}}if("getTraces"in o&&typeof o.getTraces=="function"){let m=o.getTraces();if(m&&m.length>0){let g=this.nodeTraces.get(e)||[];this.nodeTraces.set(e,[...g,...m])}}}else throw new Error(`Node program for '${e}' does not have a forward method`);return{...i,[`${e}Result`]:u}};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 le(e,async(i,a)=>{let l=new ze(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 d=o;d<r;d++){let u=this.flowDefinition[d];u&&(l=await u(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 le(a,async(c,d)=>n(c,d,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()}};var Zn=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 eo=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 to=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=q(),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 no=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)})}},oo=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(`
|
|
285
|
+
Instruction:`;try{let d=await n.chat({chatPrompt:[{role:"user",content:l}]});if("results"in d){let u=d.results[0]?.content?.trim();if(u&&u.length>10)return u}}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,d=await this.generateInstruction({tip:c,candidateIndex:l,ai:o,programSummary:r,datasetSummary:i,previousInstructions:n});n.push(d)}return n}async bootstrapFewShotExamples(e,t){return(await new Le({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,d=[],u=0;if(this.resumeFromCheckpoint){let m=await this.loadCheckpoint(this.resumeFromCheckpoint,a);m&&m.optimizerType==="MiPRO"&&(this.restoreFromCheckpoint(m),u=m.currentRound,p=m.bestScore,l=m.bestConfiguration||l,c=m.stats.convergenceInfo?.stagnationRounds||0)}for(let m=u;m<this.numTrials;m++){let g;this.bayesianOptimization&&this.miproConfigHistory.length>2?g=await this.selectConfigurationViaBayesianOptimization(o,t,n):g={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 A=await this.evaluateConfig(e,g,t,n,r,i,m+1);this.updateSurrogateModel(g,A),d.push(A);let x=A-p;if(x>this.minImprovementThreshold?(p=A,l=g,c=0):c++,await this.updateOptimizationProgress(m+1,A,g,"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:A,bestScore:p,tokensUsed:this.stats.resourceUsage.totalTokens,timeElapsed:Date.now(),successfulExamples:this.stats.successfulDemos,totalExamples:this.examples.length,currentConfiguration:g,convergenceInfo:{improvement:x,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=d.length>1?p-d[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 k(e.getSignature());this.applyConfigToProgram(l,t,n,o);let p=0,c=0,d;if(this.minibatch){let g=Math.min(this.minibatchSize,r.length);a%this.minibatchFullEvalSteps===0||a>this.numTrials*.8?d=Math.min(r.length,g*2):d=Math.max(3,Math.min(g,r.length))}else d=r.length;let m=this.shuffleArray([...Array(r.length).keys()]).slice(0,d).map(g=>r[g]);for(let g of m)try{let A=await l.forward(this.studentAI,g,this.sampleCount>1?{sampleCount:this.sampleCount,resultPicker:$i()}:void 0),x=await i({prediction:A,example:g});p+=x,c++,this.stats.totalCalls++}catch{}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;r?.auto&&this.configureAuto(r.auto);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:d}=await this.runOptimization(e,a,l,p,i,t,n);this.checkTargetScore(d)&&this.triggerEarlyStopping(`Target score ${this.targetScore} reached with score ${d}`,this.numTrials);let u;"getSignature"in e&&typeof e.getSignature=="function"?u=e.getSignature():u="input -> output";let m=new k(u);return this.applyConfigToAxGen(m,c,a,l),this.updateResourceUsage(o),this.stats.convergenceInfo.converged=!0,this.stats.convergenceInfo.finalImprovement=d,await this.saveFinalCheckpoint("MiPRO",this.getConfiguration(),d,c,{bootstrappedDemos:a.length,labeledExamples:l.length,instructions:p.length,optimizedGen:!!m},n),{demos:a,stats:this.stats,bestScore:d,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`${e.instruction.length}_${e.bootstrappedDemos}_${e.labeledExamples}`}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),d=1-((((i*c+r)*c+o)*c+n)*c+t)*c*Math.exp(-p*p);return l*d}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}},$i=()=>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 Li(s,...e){let t="";for(let n=0;n<s.length;n++)if(t+=s[n]??"",n<e.length){let o=e[n];if(Jn(o)){let r=t.match(/(\w+)\s*:\s*$/);if(r&&(o.isOptional||o.isInternal)){let c=r[1];o.isOptional&&(c+="?"),o.isInternal&&(c+="!"),t=t.replace(/(\w+)(\s*:\s*)$/,`${c}$2`)}let{isOptional:i,isInternal:a,...l}=o;t+=wt(l)}else if(Yn(o))t+=Wn(o);else if(typeof o=="string"||o instanceof O)t+=Vr(o);else throw new Error("Unsupported template interpolation value")}return new O(t)}function Bi(s,...e){let t="";for(let n=0;n<s.length;n++)if(t+=s[n]??"",n<e.length){let o=e[n];if(Jn(o)){let r=t.match(/(\w+)\s*:\s*$/);if(r&&(o.isOptional||o.isInternal)){let c=r[1];o.isOptional&&(c+="?"),o.isInternal&&(c+="!"),t=t.replace(/(\w+)(\s*:\s*)$/,`${c}$2`)}let{isOptional:i,isInternal:a,...l}=o;t+=wt(l)}else if(Yn(o))t+=Wn(o);else if(typeof o=="string"||o instanceof O)t+=Vr(o);else throw new Error("Unsupported template interpolation value")}return new k(t)}function Vr(s){if(typeof s=="string")return s;if(Jn(s))return wt(s);if(Yn(s))return Wn(s);if(s instanceof O){let e=s.toString(),t=e.indexOf(" -> ");return t!==-1?e.substring(t+4):e}throw new Error(`Unsupported template value type: ${typeof s}`)}function wt(s){let e=s.type;return s.isArray&&(e+="[]"),s.options&&s.options.length>0&&s.type==="class"&&(e+=` "${s.options.join(", ")}"`),s.description&&(e+=` "${s.description}"`),e}function Wn(s){let e=s.name;return s.isOptional&&(e+="?"),s.isInternal&&(e+="!"),s.type&&(e+=`:${wt(s.type)}`),s.description&&!s.type?.description&&(e+=` "${s.description}"`),e}function Jn(s){return s!==null&&typeof s=="object"&&s!==void 0&&"type"in s&&typeof s.type=="string"}function Yn(s){return s!==null&&typeof s=="object"&&s!==void 0&&"name"in s&&typeof s.name=="string"}var qi={string:s=>({type:"string",description:s}),number:s=>({type:"number",description:s}),boolean:s=>({type:"boolean",description:s}),date:s=>({type:"date",description:s}),datetime:s=>({type:"datetime",description:s}),json:s=>({type:"json",description:s}),image:s=>({type:"image",description:s}),audio:s=>({type:"audio",description:s}),class:(s,e)=>({type:"class",options:s,description:e}),code:(s,e)=>({type:"code",options:[s],description:e}),array:s=>({...s,isArray:!0}),optional:s=>({...s,isOptional:!0}),internal:s=>({...s,isInternal:!0})};var Be=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 le(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(d=>({result:d,originalIndex:c}))}),a=await Promise.all(i);for(let{result:l,originalIndex:p}of a)n[p]=l}return n}var qe=class{steps=[];parallelGroups=[];analyzer=new Be;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 u=new Set;for(let m of r)this.analyzeMapTransformation(m).forEach(A=>u.add(A));p=Array.from(u)}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 u=r?this.analyzer.analyzeMappingDependencies(r,"derive"):[];l=[a.inputFieldName,...u].filter((m,g,A)=>A.indexOf(m)===g)}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 u of l)this.getAllProducedFields().includes(u)||this.initialFields.add(u);let d={type:c,nodeName:t,dependencies:l,produces:p,stepFunction:e,stepIndex:this.steps.length};this.steps.push(d)}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 le(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(d=>d&&typeof d=="object"&&"_parallelResults"in d);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 ze=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}`,d;if("forward"in o&&typeof o.forward=="function")d=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`]:d}}),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}},Qn=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}`,d;if("forward"in o&&typeof o.forward=="function")d=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`]:d}}),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 Xn=class{nodes=new Map;flowDefinition=[];nodeGenerators=new Map;loopStack=[];stepLabels=new Map;branchContext=null;autoParallelConfig;executionPlanner=new qe;program;nodeUsage=new Map;nodeTraces=new Map;toCamelCase(e){return e.replace(/_([a-z])/g,(t,n)=>n.toUpperCase())}inferSignatureFromFlow(){let e=this.executionPlanner.getExecutionPlan();if(this.nodeGenerators.size===0&&e.steps.length===0){let c=new O;return c.addInputField({name:"userInput",type:{name:"string"},description:"User input to the flow"}),c.addOutputField({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),c}let t=new Set,n=new Set;for(let c of e.steps)c.produces.forEach(d=>t.add(d)),c.dependencies.forEach(d=>n.add(d));let o=new Set;for(let c of 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(d=>r.add(d))}else for(let c of t){let d=!1;for(let u of e.steps)if(u.dependencies.includes(c)){d=!0;break}if(!d)if(c.endsWith("Result")){let u=c.replace("Result",""),m=this.nodeGenerators.get(u);if(m){let g=m.getSignature(),x=new O(g).getOutputFields();for(let h of x)r.add(h.name)}else r.add(c)}else r.add(c)}if(o.size===0&&r.size===0){let c=[],d=[];for(let[m,g]of this.nodeGenerators){let A=g.getSignature(),x=new O(A);for(let h of x.getInputFields()){let y=this.toCamelCase(`${m}_${h.name}`);c.push({name:y,type:h.type,description:h.description,isOptional:h.isOptional,isInternal:h.isInternal})}for(let h of x.getOutputFields()){let y=this.toCamelCase(`${m}_${h.name}`);d.push({name:y,type:h.type,description:h.description,isOptional:h.isOptional,isInternal:h.isInternal})}}let u=new O;return c.length>0?u.setInputFields(c):u.addInputField({name:"userInput",type:{name:"string"},description:"User input to the flow"}),d.length>0?u.setOutputFields(d):u.addOutputField({name:"flowOutput",type:{name:"string"},description:"Output from the flow"}),u}let a=new O,l=[];for(let c of 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 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}}ensureProgram(){let e=this.inferSignatureFromFlow();this.program=new te(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 this.nodeTraces)e.push(...n);return e}setDemos(e){this.ensureProgram(),this.program.setDemos(e)}getUsage(){let e=[];for(let[t,n]of this.nodeUsage)e.push(...n);return De(e)}resetUsage(){this.nodeUsage.clear();for(let[e,t]of this.nodeGenerators)t&&"resetUsage"in t&&t.resetUsage()}resetTraces(){this.nodeTraces.clear()}getUsageReport(){let e={};for(let[t,n]of this.nodeUsage)e[t]=De(n);return e}getTracesReport(){let e={};for(let[t,n]of 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){this.resetUsage(),this.resetTraces();let o;if(Array.isArray(t)){let l=t.filter(p=>p.role==="user").pop();if(!l)throw new Error("No user message found in values array");o=l.values}else o=t;this.nodeGenerators.size>0&&this.ensureProgram();let r={...o},i={mainAi:e,mainOptions:n?{...n,model:n.model?String(n.model):void 0}:void 0};if(n?.autoParallel!==!1&&this.autoParallelConfig.enabled){this.executionPlanner.setInitialFields(Object.keys(o));let l=this.executionPlanner.createOptimizedExecution(this.autoParallelConfig.batchSize);for(let p of l)r=await p(r,i)}else for(let l of this.flowDefinition)r=await l(r,i);return r}node(e,t){if(typeof t=="string"||t instanceof O){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=new k(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 le(n,async(a,l)=>a(r),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=o=>{if(Array.isArray(e))throw new Error("Array of transforms requires parallel: true option");return e(o)};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)}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),d=p?.traceLabel?`Node:${e} (${p.traceLabel})`:`Node:${e}`,u;if("forward"in o&&typeof o.forward=="function"){if(u=await o.forward(l,c,{...p,traceLabel:d}),"getUsage"in o&&typeof o.getUsage=="function"){let m=o.getUsage();if(m&&m.length>0){let g=this.nodeUsage.get(e)||[];this.nodeUsage.set(e,[...g,...m])}}if("getTraces"in o&&typeof o.getTraces=="function"){let m=o.getTraces();if(m&&m.length>0){let g=this.nodeTraces.get(e)||[];this.nodeTraces.set(e,[...g,...m])}}}else throw new Error(`Node program for '${e}' does not have a forward method`);return{...i,[`${e}Result`]:u}};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 le(e,async(i,a)=>{let l=new ze(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 d=o;d<r;d++){let u=this.flowDefinition[d];u&&(l=await u(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 le(a,async(c,d)=>n(c,d,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()}};var Zn=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 eo=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 to=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=q(),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 no=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)})}},oo=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(`
|
|
286
286
|
`);r=p.pop()||"";for(let c of p)if(c.startsWith("data: ")){let d=c.slice(6);if(d==="[DONE]")return;try{let u=JSON.parse(d);this.messageHandler&&this.messageHandler(u)}catch(u){console.error("Failed to parse SSE data:",u)}}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 d=a.split(`
|
|
287
|
-
`);a=d.pop()||"";for(let u of d)if(u.startsWith("data: ")){let m=u.slice(6);if(m==="[DONE]")return;try{let g=JSON.parse(m);if("id"in g&&g.id===t){n(g);return}this.messageHandler&&this.messageHandler(g)}catch(g){console.error("Failed to parse SSE data:",g)}}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 zi(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=Hi(r.parameters,l);let p=r.func;r.func=async(c,d)=>{let u={};if(Array.isArray(e)){let g=e.filter(A=>A.role==="user").pop();g&&(u=Yr(g.values,l))}else u=Yr(e,l);let m={...c,...u};return await p(m,d)}}return r}return n&&!o.disableSmartModelRouting&&o.canConfigureSmartModelRouting&&(r.parameters=Qr(r.parameters,n)),r}var Wr=new Error("Agent description must be at least 20 characters (explain in detail what the agent does)"),Jr=new Error("Agent definition is the prompt you give to the LLM for the agent. It must be detailed and at least 100 characters"),ro=class{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:d}=l??{};if(this.ai=e,this.agents=i,this.functions=a,this.disableSmartModelRouting=p,this.excludeFieldsFromPassthrough=c??[],this.debug=d,!t||t.length<5)throw new Error("Agent name must be at least 10 characters (more descriptive)");if(!n||n.length<20)throw Wr;if(o&&o.length<100)throw Jr;this.program=new
|
|
288
|
-
`)};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(d=>d.name),l=this.getDebug(o,n),p=this.agents?.map(d=>{let u=d.getFeatures(),m={debug:l,disableSmartModelRouting:!!this.disableSmartModelRouting,excludeFieldsFromPassthrough:u.excludeFieldsFromPassthrough,canConfigureSmartModelRouting:u.canConfigureSmartModelRouting};return zi(d.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 Wr;this.program.getSignature().setDescription(e),this.func.description=e}setDefinition(e){if(!e||e.length<100)throw Jr;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 ji(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 Qr(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 Hi(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 Yr(s,e){let t={};for(let n of e)n in s&&(t[n]=s[n]);return t}var je=class extends
|
|
287
|
+
`);a=d.pop()||"";for(let u of d)if(u.startsWith("data: ")){let m=u.slice(6);if(m==="[DONE]")return;try{let g=JSON.parse(m);if("id"in g&&g.id===t){n(g);return}this.messageHandler&&this.messageHandler(g)}catch(g){console.error("Failed to parse SSE data:",g)}}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 zi(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=Hi(r.parameters,l);let p=r.func;r.func=async(c,d)=>{let u={};if(Array.isArray(e)){let g=e.filter(A=>A.role==="user").pop();g&&(u=Yr(g.values,l))}else u=Yr(e,l);let m={...c,...u};return await p(m,d)}}return r}return n&&!o.disableSmartModelRouting&&o.canConfigureSmartModelRouting&&(r.parameters=Qr(r.parameters,n)),r}var Wr=new Error("Agent description must be at least 20 characters (explain in detail what the agent does)"),Jr=new Error("Agent definition is the prompt you give to the LLM for the agent. It must be detailed and at least 100 characters"),ro=class{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:d}=l??{};if(this.ai=e,this.agents=i,this.functions=a,this.disableSmartModelRouting=p,this.excludeFieldsFromPassthrough=c??[],this.debug=d,!t||t.length<5)throw new Error("Agent name must be at least 10 characters (more descriptive)");if(!n||n.length<20)throw Wr;if(o&&o.length<100)throw Jr;this.program=new k(r,{...l,description:o??n});for(let m of i??[])this.program.register(m);this.name=t,this.func={name:ji(this.name),description:n,parameters:this.program.getSignature().toJSONSchema(),func:()=>this.forward};let u=e?.getModelList();u&&!this.disableSmartModelRouting&&(this.func.parameters=Qr(this.func.parameters,u))}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(u=>{let m=c.find(g=>g.name===u);return m?`${m.title}: ${l[u]}`:`${u}: ${l[u]}`}).join(`
|
|
288
|
+
`)};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(d=>d.name),l=this.getDebug(o,n),p=this.agents?.map(d=>{let u=d.getFeatures(),m={debug:l,disableSmartModelRouting:!!this.disableSmartModelRouting,excludeFieldsFromPassthrough:u.excludeFieldsFromPassthrough,canConfigureSmartModelRouting:u.canConfigureSmartModelRouting};return zi(d.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 Wr;this.program.getSignature().setDescription(e),this.func.description=e}setDefinition(e){if(!e||e.length<100)throw Jr;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 ji(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 Qr(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 Hi(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 Yr(s,e){let t={};for(let n of e)n in s&&(t[n]=s[n]);return t}var je=class extends k{constructor(e,t){let n=new O(e);n.setOutputFields([{name:"reason",description:"Let's work this out in a step by step way in order to ensure we have the right answer.",isInternal:t?.setVisibleReasoning!==!0},...n.getOutputFields()]),super(n,t)}};var so=class extends je{genQuery;queryFn;maxHops;constructor(e,t){super('"Answer questions with short factoid answers." context:string[] "may contain relevant facts", question -> answer',t),this.maxHops=t?.maxHops??3;let o=new O('"Write a simple search query that will help answer a complex question." context?:string[] "may contain relevant facts", question -> query "question to further our understanding"');this.genQuery=new k(o),this.queryFn=e}async forward(e,t,n){let o;if(Array.isArray(t)){let l=t.filter(p=>p.role==="user").pop();if(!l)throw new Error("No user message found in values array");o=l.values.question}else o=t.question;let r=0,i=[];for(;r<this.maxHops;){let l=await this.genQuery.forward(e,{context:i,question:o}),p=await this.queryFn(l.query);i=$e.dedup([...i,p]),r++}return await super.forward(e,{context:i,question:o},n)}};export{hn as AxAI,ue as AxAIAnthropic,Qe as AxAIAnthropicModel,Ut as AxAIAnthropicVertexModel,he as AxAIAzureOpenAI,xe as AxAICohere,tt as AxAICohereEmbedModel,et as AxAICohereModel,fe as AxAIDeepSeek,nt as AxAIDeepSeekModel,ye as AxAIGoogleGemini,Wt as AxAIGoogleGeminiEmbedModel,er as AxAIGoogleGeminiEmbedTypes,ot as AxAIGoogleGeminiModel,Jt as AxAIGoogleGeminiSafetyCategory,Yt as AxAIGoogleGeminiSafetyThreshold,Me as AxAIGrok,lr as AxAIGrokEmbedModels,ct as AxAIGrokModel,be as AxAIGroq,rt as AxAIGroqModel,Re as AxAIHuggingFace,tn as AxAIHuggingFaceModel,Ce as AxAIMistral,rr as AxAIMistralEmbedModels,st as AxAIMistralModel,Te as AxAIOllama,Ae as AxAIOpenAI,E as AxAIOpenAIBase,de as AxAIOpenAIEmbedModel,Ze as AxAIOpenAIModel,ve as AxAIOpenAIResponses,it as AxAIOpenAIResponsesBase,we as AxAIOpenAIResponsesImpl,me as AxAIOpenAIResponsesModel,S as AxAIRefusalError,Se as AxAIReka,lt as AxAIRekaModel,pe as AxAIServiceAbortedError,Q as AxAIServiceAuthenticationError,U as AxAIServiceError,J as AxAIServiceNetworkError,W as AxAIServiceResponseError,re as AxAIServiceStatusError,V as AxAIServiceStreamTerminatedError,se as AxAIServiceTimeoutError,Oe as AxAITogether,ro as AxAgent,Bn as AxApacheTika,Z as AxAssertionError,jt as AxBalancer,P as AxBaseAI,oe as AxBaseOptimizer,Le as AxBootstrapFewShot,je as AxChainOfThought,yn as AxDB,z as AxDBBase,ke as AxDBCloudflare,bn as AxDBManager,X as AxDBMemory,Ee as AxDBPinecone,Pe as AxDBWeaviate,Tt as AxDefaultCostTracker,Ln as AxDefaultResultReranker,Zn as AxDockerSession,eo as AxEmbeddingAdapter,Ii as AxEvalUtil,Xn as AxFlow,Be as AxFlowDependencyAnalyzer,qe as AxFlowExecutionPlanner,ze as AxFlowSubContextImpl,Qn as AxFlowTypedSubContextImpl,At as AxFunctionError,xt as AxFunctionProcessor,k as AxGen,bt as AxGenerateError,Hn as AxHFDataLoader,Ne as AxInstanceRegistry,Ao as AxLLMRequestTypeValues,to as AxMCPClient,no as AxMCPHTTPSSETransport,oo as AxMCPStreambleHTTPTransport,Fe as AxMemory,Vn as AxMiPRO,sn as AxMockAIService,an as AxMultiServiceRouter,te as AxProgram,ae as AxPromptTemplate,so as AxRAG,Ie as AxRateLimiterTokenUsage,O as AxSignature,zn as AxSimpleClassifier,qn as AxSimpleClassifierClass,ho as AxSpanKindValues,$e as AxStringUtil,jn as AxTestPrompt,Bi as ax,Wo as axAIAnthropicDefaultConfig,as as axAIAnthropicVertexDefaultConfig,gs as axAIAzureOpenAIBestConfig,ds as axAIAzureOpenAICreativeConfig,Yo as axAIAzureOpenAIDefaultConfig,ms as axAIAzureOpenAIFastConfig,hs as axAICohereCreativeConfig,Xo as axAICohereDefaultConfig,fs as axAIDeepSeekCodeConfig,Zo as axAIDeepSeekDefaultConfig,nr as axAIGoogleGeminiDefaultConfig,ys as axAIGoogleGeminiDefaultCreativeConfig,Ps as axAIGrokBestConfig,An as axAIGrokDefaultConfig,Rs as axAIHuggingFaceCreativeConfig,or as axAIHuggingFaceDefaultConfig,Cs as axAIMistralBestConfig,rn as axAIMistralDefaultConfig,sr as axAIOllamaDefaultConfig,Ts as axAIOllamaDefaultCreativeConfig,Bt as axAIOpenAIBestConfig,qt as axAIOpenAICreativeConfig,ie as axAIOpenAIDefaultConfig,zt as axAIOpenAIFastConfig,vs as axAIOpenAIResponsesBestConfig,Ss as axAIOpenAIResponsesCreativeConfig,at as axAIOpenAIResponsesDefaultConfig,Os as axAIRekaBestConfig,Ms as axAIRekaCreativeConfig,pt as axAIRekaDefaultConfig,ks as axAIRekaFastConfig,ar as axAITogetherDefaultConfig,v as axBaseAIDefaultConfig,G as axBaseAIDefaultCreativeConfig,Us as axCheckMetricsHealth,go as axCreateDefaultColorLogger,Hr as axCreateDefaultOptimizerColorLogger,bi as axCreateDefaultOptimizerTextLogger,os as axCreateDefaultTextLogger,pr as axDefaultMetricsConfig,Kn as axDefaultOptimizerLogger,Kr as axDefaultOptimizerMetricsConfig,Ls as axGetMetricsConfig,Ti as axGetOptimizerMetricsConfig,M as axGlobals,Xe as axModelInfoAnthropic,Ht as axModelInfoCohere,Vt as axModelInfoDeepSeek,Qt as axModelInfoGoogleGemini,gn as axModelInfoGrok,Zt as axModelInfoGroq,en as axModelInfoHuggingFace,on as axModelInfoMistral,ge as axModelInfoOpenAI,$t as axModelInfoOpenAIResponses,pn as axModelInfoReka,un as axModelInfoTogether,T as axSpanAttributes,K as axSpanEvents,$s as axUpdateMetricsConfig,Ci as axUpdateOptimizerMetricsConfig,dn as axValidateChatRequestMessage,mn as axValidateChatResponseResult,qi as f,Li as s};
|
|
289
289
|
//# sourceMappingURL=index.js.map
|