@fifthrevision/axle 0.4.1 → 0.5.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cli.js +3 -11
- package/dist/consoleWriter-CtNNCSna.js +38 -0
- package/dist/index.d.ts +106 -65
- package/dist/index.js +2 -2
- package/package.json +3 -2
- package/dist/dag-DM0XblG3.js +0 -19
package/dist/cli.js
CHANGED
|
@@ -1,12 +1,4 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
|
-
var
|
|
3
|
-
`)
|
|
4
|
-
`}
|
|
5
|
-
`}this.tasks.delete(a.id)}console.log(n)}for(const n of this.entries){const{level:a,time:f,kind:c,payload:h}=n;c==="heading"?it(a,h,{truncate:this.truncate}):nt(a,h,{truncate:this.truncate})}this.entries=[];let o="";for(const n of this.tasks.values())if(n.status===u.Running){const a=r.cyan(m.spinning[n.frameIndex]);n.frameIndex=(n.frameIndex+1)%m.spinning.length,o+=`${a} ${n.text}
|
|
6
|
-
`}else if(n.status===u.Success){const a=r.green(m.success);o+=`${a} ${n.text}
|
|
7
|
-
`}else if(n.status===u.Fail){const a=r.red(m.fail);o+=`${a} ${n.text}
|
|
8
|
-
`}this.lastRender=o,process.stdout.write(o),this.isRendering=!1}handleEvent(t){const{level:e,time:s,payload:o}=t;if(o.length>0&&et(o[0])){const a=o[0],{id:f,message:c,status:h}=a;if(h===u.Running)this.tasks.set(f,{id:f,text:c,status:h,frameIndex:0});else if((h===u.Success||h===u.Fail)&&this.tasks.has(f)){const j=this.tasks.get(f);j.status=h,j.text=c}}else this.entries.push(t);this.renderTasks();const n=[...this.tasks.values()].some(a=>a.status===u.Running);n&&this.intervalId===null?this.startSpinner():!n&&this.intervalId!==null&&this.stopSpinner()}destroy(){this.stopSpinner()}}function et(i){if(typeof i!="object"||i===null)return!1;const t=i;if(t.type!=="task"||typeof t.id!="string"||typeof t.message!="string")return!1;switch(t.status){case u.Running:case u.Success:case u.PartialSuccess:case u.Fail:return!0;default:return!1}}g(et,"isTask");function it(i,t,e){let s,o;i===p.Error?(s=r.red,o=r.redBright.bold):i===p.Warn?(s=r.yellow,o=r.yellowBright.bold):i>=p.Info?(s=r.blue,o=r.whiteBright.bold):(s=r.gray,o=r.white);const{message:n,data:a}=D(t);console.log(`${s("==>")} ${o(n)}`),S(i,a,e)}g(it,"heading");function nt(i,t,e){let s;i===p.Error?s=r.red:i===p.Warn?s=r.yellow:i>=p.Info?s=r.white:s=r.gray;const{message:o,data:n}=D(t);o&&console.log(s(o)),S(i,n,e)}g(nt,"body");const $=" ";function S(i,t,e){let s;i===p.Error?(s=r.red,e.truncate=0):i==p.Warn?s=r.yellow:i>=p.Info?s=r.white:s=r.gray,t.forEach(o=>{if(typeof o=="string"){console.log(s(`${$}${o}`));return}for(const[n,a]of Object.entries(o)){let f=JSON.stringify(a,st(e.truncate)," ");const c=`${n}: ${f}`.split(`
|
|
9
|
-
`).map(h=>$+h).join(`
|
|
10
|
-
`);console.log(s(c))}})}g(S,"values");function D(i){const[t,...e]=i;let s="",o=e;if(t){let{message:n,...a}=t;s=n&&typeof n=="string"?n:"",Object.keys(a).length>0&&(o=[a,...o])}return{message:s,data:o}}g(D,"toMsgData");function st(i){return i===0?null:(t,e)=>typeof e=="string"&&e.length>i?e.slice(0,i)+"<...>":e}g(st,"truncator");const y="./logs/",ot="~/.axle/logs/";class at{static{g(this,"LogWriter")}time;initialized=!1;logDir=y;pendingWrites=[];constructor(){this.time=new Date().toISOString()}get filename(){return`${this.logDir}${this.time}.log`}async initialize(){try{await T(y),this.logDir=y}catch{const s=ot.replace("~",z());try{await T(s),this.logDir=s}catch{await P(s,{recursive:!0}),this.logDir=s}}const t=M(this.filename,`AXLE: New run at ${this.time}
|
|
11
|
-
`);this.pendingWrites.push(t);try{await t,this.initialized=!0}finally{const e=this.pendingWrites.indexOf(t);e!==-1&&this.pendingWrites.splice(e,1)}}async writeToLog(t){const{time:e,level:s,payload:o}=t;this.initialized||await this.initialize();const n=o.map(c=>typeof c=="string"?c:JSON.stringify(c)),a=`${p[s]} ${new Date(e).toISOString()} > ${n.join(" >> ")}
|
|
12
|
-
`,f=U(this.filename,a).catch(c=>{console.error(`Failed to write to log file: ${c}`)});this.pendingWrites.push(f);try{await f}finally{const c=this.pendingWrites.indexOf(f);c!==-1&&this.pendingWrites.splice(c,1)}}async handleEvent(t){await this.writeToLog(t)}async flush(){this.pendingWrites.length>0&&await Promise.all(this.pendingWrites)}}const b=new A().name("axle").description("Axle is a CLI tool for running AI workflows").version(G.version).option("--dry-run","Run the application without executing against the AI providers").option("-c, --config <path>","Path to the config file").option("-j, --job <path>","Path to the job file").option("--no-log","Do not write the output to a log file").option("--no-warn-unused","Do not warn about unused variables").option("--no-inline","Do not inline the console output").option("-d, --debug","Print additional debug information").option("--truncate <num>","Truncate printed strings to a certain number of characters, 0 to disable",parseInt,100).option("--args <args...>","Additional arguments in the form key=value");b.parse(process.argv);const d=b.opts(),v={};d.args&&d.args.forEach(i=>{const[t,e]=i.split("=");t&&e&&(v[t.trim()]=e.trim())}),process.on("uncaughtException",async i=>{console.error("Uncaught exception:"),console.error(i),l&&(l.error?.log("Uncaught exception:"),l.error?.log(i.message),l.error?.log(i.stack||""),await l.shutdown()),process.exit(1)});const l=new W;d.debug&&(l.level=p.Debug);const lt=new tt(d);if(l.subscribe(lt),d.log){const i=new at;await i.initialize(),l.subscribe(i)}d.debug&&(l.debug?.heading.log("Options"),l.debug?.log(d),l.debug?.heading.log("Additional Arguments:"),l.debug?.log(v));let w,k;try{w=await Q(d.config??null,{recorder:l}),k=await V(d.job??null,{recorder:l})}catch(i){l.error.log(i.message),l.debug?.log(i.stack),await l.shutdown(),b.outputHelp(),process.exit(1)}let E;try{const{engine:i,...t}=k.using,e={...w[i],...t};E=_(i,e)}catch(i){l.error.log(i.message),l.error.log(i.stack),await l.shutdown(),b.outputHelp(),process.exit(1)}J().setConfig(w),l.info?.heading.log("All systems operational. Running job...");const rt=Date.now();d.dryRun&&l.info?.log("Dry run mode enabled. No API calls will be made.");const C={in:0,out:0},O=await N(k.jobs).execute({provider:E,variables:v,options:d,stats:C,recorder:l});O&&(l.info?.heading.log("Response"),l.info.log(O)),l.info?.heading.log("Usage"),l.info?.log(`Total run time: ${Date.now()-rt}ms`),l.info?.log(`Input tokens: ${C.in} `),l.info?.log(`Output tokens: ${C.out} `),l.info?.heading.log("Complete. Goodbye"),await l.shutdown();
|
|
2
|
+
var D=Object.defineProperty;var u=(i,e)=>D(i,"name",{value:e,configurable:!0});import{Command as L}from"@commander-js/extra-typings";import{e as y,f as k,L as v,R as x,C as O,g as E,h as T,d as F}from"./consoleWriter-CtNNCSna.js";import w from"yaml";import{access as C,mkdir as I,writeFile as W,appendFile as R}from"node:fs/promises";import{homedir as S}from"node:os";import"@anthropic-ai/sdk";import"@google/genai";import"openai";import"serialize-error";import"fs/promises";import"glob";import"node:path";import"zod/v4";import"chalk";import"node:readline";var _="0.5.0",$={version:_};function J(i,e){if(typeof i!="object"||i===null)return e&&(e.value="Config: must be a non-null object"),!1;if("openai"in i){const t=i.openai;if(typeof t!="object"||t===null)return e&&(e.value="Config: openai must be an object"),!1;if(typeof t["api-key"]!="string")return e&&(e.value="Config: openai.api-key must be a string"),!1;if("model"in t&&typeof t.model!="string")return e&&(e.value="Config: openai.model must be a string"),!1}if("anthropic"in i){const t=i.anthropic;if(typeof t!="object"||t===null)return e&&(e.value="Config: anthropic must be an object"),!1;if(typeof t["api-key"]!="string")return e&&(e.value="Config: anthropic.api-key must be a string"),!1;if("model"in t&&typeof t.model!="string")return e&&(e.value="Config: anthropic.model must be a string"),!1}if("ollama"in i){const t=i.ollama;if(typeof t!="object"||t===null)return e&&(e.value="Config: ollama must be an object"),!1;if("url"in t&&typeof t.url!="string")return e&&(e.value="Config: ollama.url must be a string"),!1;if("model"in t&&typeof t.model!="string")return e&&(e.value="Config: ollama.model must be a string"),!1}if("googleai"in i){const t=i.googleai;if(typeof t!="object"||t===null)return e&&(e.value="Config: googleai must be an object"),!1;if(typeof t["api-key"]!="string")return e&&(e.value="Config: googleai.api-key must be a string"),!1;if("model"in t&&typeof t.model!="string")return e&&(e.value="Config: googleai.model must be a string"),!1}if("brave"in i){const t=i.brave;if(typeof t!="object"||t===null)return e&&(e.value="Config: brave must be an object"),!1;if(typeof t["api-key"]!="string")return e&&(e.value="Config: brave.api-key must be a string"),!1;if("rateLimit"in t&&typeof t.rateLimit!="number")return e&&(e.value="Config: brave.rateLimit must be a number"),!1}return!0}u(J,"isServiceConfig");const N="ax.job",P=["yaml","yml","json"];async function U(i,e){const{recorder:t}=e,{content:a,format:l}=await y(i,{defaults:{name:N,formats:P},tag:"Job File"});let n=null;if(l==="json")n=JSON.parse(a);else if(l==="yaml"||l==="yml")n=w.parse(a);else throw new Error("Invalid job file format");t?.debug?.heading.log("The Job Object"),t?.debug?.log(n);const g={value:""};if(k(n,g))return n;throw new Error(`The job file is not valid: ${g.value}`)}u(U,"getJobConfig");const z="ax.config",M=["yaml","yml","json"];async function G(i,e){const{recorder:t}=e,{content:a,format:l}=await y(i,{defaults:{name:z,formats:M},tag:"Config File"});let n=null;if(l==="json")n=JSON.parse(a);else if(l==="yaml"||l==="yml")n=w.parse(a);else throw new Error("Invalid config file format");t?.debug?.heading.log("The Config Object"),t?.debug?.log(n);const g={value:""};if(J(n,g))return n;throw new Error(g.value)}u(G,"getServiceConfig");const p="./logs/",B="~/.axle/logs/";class H{static{u(this,"LogWriter")}time;initialized=!1;logDir=p;pendingWrites=[];constructor(){this.time=new Date().toISOString()}get filename(){return`${this.logDir}${this.time}.log`}async initialize(){try{await C(p),this.logDir=p}catch{const a=B.replace("~",S());try{await C(a),this.logDir=a}catch{await I(a,{recursive:!0}),this.logDir=a}}const e=W(this.filename,`AXLE: New run at ${this.time}
|
|
3
|
+
`);this.pendingWrites.push(e);try{await e,this.initialized=!0}finally{const t=this.pendingWrites.indexOf(e);t!==-1&&this.pendingWrites.splice(t,1)}}async writeToLog(e){const{time:t,level:a,payload:l}=e;this.initialized||await this.initialize();const n=l.map(r=>typeof r=="string"?r:JSON.stringify(r)),g=`${v[a]} ${new Date(t).toISOString()} > ${n.join(" >> ")}
|
|
4
|
+
`,c=R(this.filename,g).catch(r=>{console.error(`Failed to write to log file: ${r}`)});this.pendingWrites.push(c);try{await c}finally{const r=this.pendingWrites.indexOf(c);r!==-1&&this.pendingWrites.splice(r,1)}}async handleEvent(e){await this.writeToLog(e)}async flush(){this.pendingWrites.length>0&&await Promise.all(this.pendingWrites)}}const f=new L().name("axle").description("Axle is a CLI tool for running AI workflows").version($.version).option("--dry-run","Run the application without executing against the AI providers").option("-c, --config <path>","Path to the config file").option("-j, --job <path>","Path to the job file").option("--no-log","Do not write the output to a log file").option("--no-warn-unused","Do not warn about unused variables").option("--no-inline","Do not inline the console output").option("-d, --debug","Print additional debug information").option("--truncate <num>","Truncate printed strings to a certain number of characters, 0 to disable",parseInt,100).option("--args <args...>","Additional arguments in the form key=value");f.parse(process.argv);const s=f.opts(),m={};s.args&&s.args.forEach(i=>{const[e,t]=i.split("=");e&&t&&(m[e.trim()]=t.trim())}),process.on("uncaughtException",async i=>{console.error("Uncaught exception:"),console.error(i),o&&(o.error?.log("Uncaught exception:"),o.error?.log(i.message),o.error?.log(i.stack||""),await o.shutdown()),process.exit(1)});const o=new x;s.debug&&(o.level=v.Debug);const X=new O(s);if(o.subscribe(X),s.log){const i=new H;await i.initialize(),o.subscribe(i)}s.debug&&(o.debug?.heading.log("Options"),o.debug?.log(s),o.debug?.heading.log("Additional Arguments:"),o.debug?.log(m));let d,b;try{d=await G(s.config??null,{recorder:o}),b=await U(s.job??null,{recorder:o})}catch(i){o.error.log(i.message),o.debug?.log(i.stack),await o.shutdown(),f.outputHelp(),process.exit(1)}let j;try{const{engine:i,...e}=b.using,t={...d[i],...e};j=E(i,t)}catch(i){o.error.log(i.message),o.error.log(i.stack),await o.shutdown(),f.outputHelp(),process.exit(1)}T().setConfig(d),o.info?.heading.log("All systems operational. Running job...");const K=Date.now();s.dryRun&&o.info?.log("Dry run mode enabled. No API calls will be made.");const h={in:0,out:0},A=await F(b.jobs).execute({provider:j,variables:m,options:s,stats:h,recorder:o});A&&(o.info?.heading.log("Response"),o.info.log(A)),o.info?.heading.log("Usage"),o.info?.log(`Total run time: ${Date.now()-K}ms`),o.info?.log(`Input tokens: ${h.in} `),o.info?.log(`Output tokens: ${h.out} `),o.info?.heading.log("Complete. Goodbye"),await o.shutdown();
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
var Ne=Object.defineProperty;var u=(t,e)=>Ne(t,"name",{value:e,configurable:!0});import Re from"@anthropic-ai/sdk";import{GoogleGenAI as Oe,Type as be,FinishReason as P}from"@google/genai";import ke from"openai";import{serializeError as Me}from"serialize-error";import{readFile as $e,access as Ge,constants as xe}from"fs/promises";import{glob as ae}from"glob";import{readFile as C,access as ce,stat as Le,writeFile as Ce,mkdir as Fe}from"node:fs/promises";import{resolve as ee,extname as ue,dirname as De}from"node:path";import*as I from"zod/v4";import w from"chalk";import le from"node:readline";class A extends Error{static{u(this,"AxleError")}code;id;details;constructor(e,s){super(e,{cause:s?.cause}),this.name=this.constructor.name,this.code=s?.code||"AXLE_ERROR",this.id=s?.id,this.details=s?.details,Object.setPrototypeOf(this,A.prototype)}}const O={CLAUDE_OPUS_4_20250514:"claude-opus-4-20250514",CLAUDE_OPUS_4_LATEST:"claude-opus-4-0",CLAUDE_SONNET_4_20250514:"claude-sonnet-4-20250514",CLAUDE_SONNET_4_LATEST:"claude-sonnet-4-0",CLAUDE_3_7_SONNET_20250219:"claude-3-7-sonnet-20250219",CLAUDE_3_7_SONNET_LATEST:"claude-3-7-sonnet-latest",CLAUDE_3_5_HAIKU_20241022:"claude-3-5-haiku-20241022",CLAUDE_3_5_HAIKU_LATEST:"claude-3-5-haiku-latest"},fe=[O.CLAUDE_3_7_SONNET_LATEST,O.CLAUDE_3_7_SONNET_20250219,O.CLAUDE_3_5_HAIKU_LATEST,O.CLAUDE_3_5_HAIKU_20241022,O.CLAUDE_SONNET_4_LATEST,O.CLAUDE_SONNET_4_20250514,O.CLAUDE_OPUS_4_LATEST,O.CLAUDE_OPUS_4_20250514];class Ue{static{u(this,"Chat")}system;messages=[];tools=[];setToolSchemas(e){this.tools=e}addSystem(e){this.system=e}addUser(e,s,n){let o,r=[];if(typeof s=="string"?(o=s,r=n||[]):Array.isArray(s)&&(r=s),!o&&r.length===0){this.messages.push({role:"user",content:e});return}const i=[{type:"text",text:e}];o&&i.push({type:"instructions",instructions:o});for(const a of r)i.push({type:"file",file:a});this.messages.push({role:"user",content:i})}addAssistant(e,s){this.messages.push({role:"assistant",content:e,...s&&{toolCalls:s}})}addTools(e){this.messages.push({role:"tool",content:e})}hasFiles(){return this.messages.some(e=>Array.isArray(e.content)&&e.content.some(s=>s.type==="file"))}latest(){return this.messages[this.messages.length-1]}toString(){return JSON.stringify({system:this.system,messages:this.messages,tools:this.tools})}}function H(t,e=`
|
|
2
|
+
|
|
3
|
+
`){if(typeof t=="string")return t;const s=t.filter(o=>o.type==="text").map(o=>o.text),n=t.filter(o=>o.type==="instructions").map(o=>o.instructions);return s.length===0&&n.length===0?null:[...s,...n].join(e)}u(H,"getTextAndInstructions");function He(t){return typeof t=="string"?t:t.filter(e=>e.type==="text").map(e=>e.text).join(`
|
|
4
|
+
|
|
5
|
+
`)}u(He,"getTextContent");function We(t){if(typeof t=="string")return null;const e=t.filter(s=>s.type==="instructions").map(s=>s.instructions);return e.length>0?e.join(`
|
|
6
|
+
|
|
7
|
+
`):null}u(We,"getInstructions");function W(t){return typeof t=="string"?[]:t.filter(e=>e.type==="file"&&e.file.type==="document").map(e=>e.file)}u(W,"getDocuments");function F(t){return typeof t=="string"?[]:t.filter(e=>e.type==="file"&&e.file.type==="image").map(e=>e.file)}u(F,"getImages");var E=(t=>(t[t.Stop=0]="Stop",t[t.Length=1]="Length",t[t.FunctionCall=2]="FunctionCall",t[t.Error=3]="Error",t))(E||{});const Be=O.CLAUDE_SONNET_4_LATEST;class Je{static{u(this,"AnthropicProvider")}name="Anthropic";client;model;constructor(e,s){this.model=s??Be,this.client=new Re({apiKey:e})}createChatRequest(e,s={}){const{recorder:n}=s;return e.hasFiles()&&!fe.includes(this.model)&&n?.warn?.log(`Model ${this.model} may not support multimodal content. Use one of: ${fe.join(", ")}`),new Xe(this,e)}}class Xe{static{u(this,"AnthropicChatRequest")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:n,model:o}=this.provider,r={model:o,max_tokens:4096,...qe(this.chat)};s?.debug?.log(r);let i;try{const a=await n.messages.create(r);i=Ke(a)}catch(a){i={type:"error",error:{type:a.error.error.type??"Undetermined",message:a.error.error.message??"Unexpected error from Anthropic"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function pe(t){switch(t){case"max_tokens":return E.Length;case"end_turn":return E.Stop;case"stop_sequence":return E.Stop;case"tool_use":return E.FunctionCall;default:return E.Error}}u(pe,"getStopReason$2");function qe(t){const e=t.messages.map(n=>{if(n.role==="assistant"){const o=[];return o.push({type:"text",text:n.content}),n.toolCalls&&o.push(...n.toolCalls.map(r=>({type:"tool_use",id:r.id,name:r.name,input:r.arguments}))),{role:"assistant",content:o}}if(n.role==="tool")return{role:"user",content:n.content.map(o=>({type:"tool_result",tool_use_id:o.id,content:o.content}))};if(typeof n.content=="string")return{role:"user",content:n.content};{const o=[],r=H(n.content);r&&o.push({type:"text",text:r});const i=F(n.content);i.length>0&&o.push(...i.map(c=>({type:"image",source:{type:"base64",media_type:c.mimeType,data:c.base64}})));const a=W(n.content);return a.length>0&&o.push(...a.filter(c=>c.mimeType==="application/pdf").map(c=>({type:"document",source:{type:"base64",media_type:"application/pdf",data:c.base64}}))),{role:"user",content:o}}}),s=t.tools.map(n=>({name:n.name,description:n.description,input_schema:n.parameters}));return{system:t.system,messages:e,tools:s}}u(qe,"prepareRequest$4");function Ke(t){const e=pe(t.stop_reason);if(e===E.Error)return{type:"error",error:{type:"Uncaught error",message:"Stop reason is not recognized."},usage:{in:t.usage.input_tokens,out:t.usage.output_tokens},raw:t};if(e===E.FunctionCall){const s=t.content[0],n=s.type==="text"?s.text:"",o=t.content.slice(1).map(r=>{if(r.type==="tool_use")return{id:r.id,name:r.name,arguments:r.input}}).filter(r=>r!==null);return{type:"success",id:t.id,model:t.model,reason:E.FunctionCall,message:{role:t.role,content:n,toolCalls:o},usage:{in:t.usage.input_tokens,out:t.usage.output_tokens},raw:t}}if(t.type=="message"){const s=t.content[0];if(s.type=="text")return{type:"success",id:t.id,model:t.model,reason:pe(t.stop_reason),message:{role:t.role,content:s.text},usage:{in:t.usage.input_tokens,out:t.usage.output_tokens},raw:t}}}u(Ke,"translate");const p={GEMINI_1_0_PRO_VISION_LATEST:"gemini-1.0-pro-vision-latest",GEMINI_PRO_VISION:"gemini-pro-vision",GEMINI_1_5_PRO_LATEST:"gemini-1.5-pro-latest",GEMINI_1_5_PRO_001:"gemini-1.5-pro-001",GEMINI_1_5_PRO_002:"gemini-1.5-pro-002",GEMINI_1_5_PRO:"gemini-1.5-pro",GEMINI_1_5_FLASH_LATEST:"gemini-1.5-flash-latest",GEMINI_1_5_FLASH_001:"gemini-1.5-flash-001",GEMINI_1_5_FLASH_001_TUNING:"gemini-1.5-flash-001-tuning",GEMINI_1_5_FLASH:"gemini-1.5-flash",GEMINI_1_5_FLASH_002:"gemini-1.5-flash-002",GEMINI_1_5_FLASH_8B:"gemini-1.5-flash-8b",GEMINI_1_5_FLASH_8B_001:"gemini-1.5-flash-8b-001",GEMINI_1_5_FLASH_8B_LATEST:"gemini-1.5-flash-8b-latest",GEMINI_1_5_FLASH_8B_EXP_0827:"gemini-1.5-flash-8b-exp-0827",GEMINI_1_5_FLASH_8B_EXP_0924:"gemini-1.5-flash-8b-exp-0924",GEMINI_2_5_PRO_EXP_03_25:"gemini-2.5-pro-exp-03-25",GEMINI_2_5_PRO_PREVIEW_03_25:"gemini-2.5-pro-preview-03-25",GEMINI_2_5_FLASH_PREVIEW_04_17:"gemini-2.5-flash-preview-04-17",GEMINI_2_5_FLASH_PREVIEW_05_20:"gemini-2.5-flash-preview-05-20",GEMINI_2_5_FLASH_PREVIEW_04_17_THINKING:"gemini-2.5-flash-preview-04-17-thinking",GEMINI_2_5_PRO_PREVIEW_05_06:"gemini-2.5-pro-preview-05-06",GEMINI_2_5_PRO_PREVIEW_06_05:"gemini-2.5-pro-preview-06-05",GEMINI_2_0_FLASH_EXP:"gemini-2.0-flash-exp",GEMINI_2_0_FLASH:"gemini-2.0-flash",GEMINI_2_0_FLASH_001:"gemini-2.0-flash-001",GEMINI_2_0_FLASH_EXP_IMAGE_GENERATION:"gemini-2.0-flash-exp-image-generation",GEMINI_2_0_FLASH_LITE_001:"gemini-2.0-flash-lite-001",GEMINI_2_0_FLASH_LITE:"gemini-2.0-flash-lite",GEMINI_2_0_FLASH_PREVIEW_IMAGE_GENERATION:"gemini-2.0-flash-preview-image-generation",GEMINI_2_0_FLASH_LITE_PREVIEW_02_05:"gemini-2.0-flash-lite-preview-02-05",GEMINI_2_0_FLASH_LITE_PREVIEW:"gemini-2.0-flash-lite-preview",GEMINI_2_0_PRO_EXP:"gemini-2.0-pro-exp",GEMINI_2_0_PRO_EXP_02_05:"gemini-2.0-pro-exp-02-05",GEMINI_EXP_1206:"gemini-exp-1206",GEMINI_2_0_FLASH_THINKING_EXP_01_21:"gemini-2.0-flash-thinking-exp-01-21",GEMINI_2_0_FLASH_THINKING_EXP:"gemini-2.0-flash-thinking-exp",GEMINI_2_0_FLASH_THINKING_EXP_1219:"gemini-2.0-flash-thinking-exp-1219",GEMINI_2_5_FLASH_PREVIEW_TTS:"gemini-2.5-flash-preview-tts",GEMINI_2_5_PRO_PREVIEW_TTS:"gemini-2.5-pro-preview-tts",LEARNLM_2_0_FLASH_EXPERIMENTAL:"learnlm-2.0-flash-experimental",GEMMA_3_1B_IT:"gemma-3-1b-it",GEMMA_3_4B_IT:"gemma-3-4b-it",GEMMA_3_12B_IT:"gemma-3-12b-it",GEMMA_3_27B_IT:"gemma-3-27b-it",GEMMA_3N_E4B_IT:"gemma-3n-e4b-it"},_e=[p.GEMINI_1_0_PRO_VISION_LATEST,p.GEMINI_PRO_VISION,p.GEMINI_1_5_PRO_LATEST,p.GEMINI_1_5_PRO_001,p.GEMINI_1_5_PRO_002,p.GEMINI_1_5_PRO,p.GEMINI_1_5_FLASH_LATEST,p.GEMINI_1_5_FLASH_001,p.GEMINI_1_5_FLASH_001_TUNING,p.GEMINI_1_5_FLASH,p.GEMINI_1_5_FLASH_002,p.GEMINI_1_5_FLASH_8B,p.GEMINI_1_5_FLASH_8B_001,p.GEMINI_1_5_FLASH_8B_LATEST,p.GEMINI_1_5_FLASH_8B_EXP_0827,p.GEMINI_1_5_FLASH_8B_EXP_0924,p.GEMINI_2_5_PRO_EXP_03_25,p.GEMINI_2_5_PRO_PREVIEW_03_25,p.GEMINI_2_5_FLASH_PREVIEW_04_17,p.GEMINI_2_5_FLASH_PREVIEW_05_20,p.GEMINI_2_5_FLASH_PREVIEW_04_17_THINKING,p.GEMINI_2_5_PRO_PREVIEW_05_06,p.GEMINI_2_5_PRO_PREVIEW_06_05,p.GEMINI_2_0_FLASH_EXP,p.GEMINI_2_0_FLASH,p.GEMINI_2_0_FLASH_001,p.GEMINI_2_0_FLASH_EXP_IMAGE_GENERATION,p.GEMINI_2_0_FLASH_LITE_001,p.GEMINI_2_0_FLASH_LITE,p.GEMINI_2_0_FLASH_PREVIEW_IMAGE_GENERATION,p.GEMINI_2_0_FLASH_LITE_PREVIEW_02_05,p.GEMINI_2_0_FLASH_LITE_PREVIEW,p.GEMINI_2_0_PRO_EXP,p.GEMINI_2_0_PRO_EXP_02_05,p.GEMINI_EXP_1206,p.GEMINI_2_0_FLASH_THINKING_EXP_01_21,p.GEMINI_2_0_FLASH_THINKING_EXP,p.GEMINI_2_0_FLASH_THINKING_EXP_1219,p.GEMINI_2_5_FLASH_PREVIEW_TTS,p.GEMINI_2_5_PRO_PREVIEW_TTS,p.LEARNLM_2_0_FLASH_EXPERIMENTAL,p.GEMMA_3_1B_IT,p.GEMMA_3_4B_IT,p.GEMMA_3_12B_IT,p.GEMMA_3_27B_IT,p.GEMMA_3N_E4B_IT],ze=p.GEMINI_2_5_FLASH_PREVIEW_05_20;class Ze{static{u(this,"GoogleAIProvider")}name="GoogleAI";client;model;constructor(e,s){this.model=s??ze,this.client=new Oe({apiKey:e})}createChatRequest(e,s={}){const{recorder:n}=s;return e.hasFiles()&&!_e.includes(this.model)&&n?.warn.log(`Model ${this.model} does not support multimodal content. Use one of: ${_e.join(", ")}`),new je(this,e)}}class je{static{u(this,"GoogleAIChatRequest")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:n,model:o}=this.provider,r=Ye(this.chat);s?.debug?.log(r);let i;try{const a=await n.models.generateContent({model:o,...r});i=Qe(a,e)}catch(a){s?.error?.log(a),i={type:"error",error:{type:a.name??"Undetermined",message:a.message??"Unexpected error from Google AI"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function Ye(t){let e;t.messages.length===1&&t.messages[0].role=="user"&&typeof t.messages[0].content=="string"?e=t.messages[0].content:e=t.messages.map(n=>{if(n.role==="user"){if(typeof n.content=="string")return{role:"user",parts:[{text:n.content}]};{const o=[],r=H(n.content);r&&o.push({text:r});const i=F(n.content);i.length>0&&o.push(...i.map(c=>({inlineData:{mimeType:c.mimeType,data:c.base64}})));const a=W(n.content);return a.length>0&&o.push(...a.map(c=>({inlineData:{mimeType:c.mimeType,data:c.base64}}))),{role:"user",parts:o}}}else if(n.role==="assistant"){const o={role:"assistant",parts:[]};return n.content!==void 0&&o.parts.push({text:n.content}),n.toolCalls&&(o.parts=o.parts.concat(n.toolCalls.map(r=>{let i;return typeof r.arguments=="string"?i=JSON.parse(r.arguments):i=r.arguments,{functionCall:{id:r.id??void 0,name:r.name,args:i}}}))),o}else if(n.role==="tool")return{role:"user",parts:n.content.map(o=>({functionResponse:{id:o.id??void 0,name:o.name,response:{output:o.content}}}))}});const s={};return t.system&&(s.systemInstruction=t.system),t.tools.length>0&&(s.tools=t.tools.map(n=>({functionDeclarations:[{name:n.name,description:n.description,parameters:{...n.parameters,type:be.OBJECT}}]}))),{contents:e,config:s}}u(Ye,"prepareRequest$3");function Qe(t,e){const{recorder:s}=e,n=t.usageMetadata.promptTokenCount,o=t.usageMetadata.totalTokenCount-n,r={in:n,out:o};if(!t)return{type:"error",error:{type:"InvalidResponse",message:"Invalid or empty response from Google AI"},usage:{in:0,out:0},raw:t};if(t.promptFeedback&&t.promptFeedback.blockReason)return{type:"error",error:{type:"Blocked",message:`Response blocked by Google AI: ${t.promptFeedback.blockReason}, ${t.promptFeedback.blockReasonMessage}`},usage:r,raw:t};if(!t.candidates||t.candidates.length===0)return{type:"error",error:{type:"InvalidResponse",message:"Invalid or empty response from Google AI"},usage:{in:0,out:0},raw:t};t.candidates.length>1&&s?.warn?.log(`We received ${t.candidates.length} response candidates`);const i=t.candidates[0],c=(i.content?.parts||[]).map(_=>_.text).filter(_=>_!==void 0).join(""),[l,f]=Ve(i.finishReason);if(l){let _;return t.functionCalls&&(_=t.functionCalls.map(g=>({id:g.id,name:g.name,arguments:JSON.stringify(g.args)}))),{type:"success",id:t.responseId,model:t.modelVersion,reason:t.functionCalls?E.FunctionCall:f,message:{role:"assistant",...c?{content:c}:{},..._?{toolCalls:_}:{}},usage:r,raw:t}}else return{type:"error",error:{type:"Undetermined",message:`Unexpected stop reason: ${f}`},usage:r,raw:t}}u(Qe,"translateResponse$2");function Ve(t){switch(t){case P.STOP:return[!0,E.Stop];case P.MAX_TOKENS:return[!0,E.Length];case P.FINISH_REASON_UNSPECIFIED:case P.SAFETY:case P.RECITATION:case P.LANGUAGE:case P.OTHER:case P.BLOCKLIST:case P.PROHIBITED_CONTENT:case P.SPII:case P.MALFORMED_FUNCTION_CALL:case P.IMAGE_SAFETY:return[!1,E.Error]}}u(Ve,"getStopReason$1");const et="http://localhost:11434";class tt{static{u(this,"OllamaProvider")}name="Ollama";url;model;recorder;constructor(e,s){this.url=s||et,this.model=e}createChatRequest(e,s={}){const{recorder:n}=s;return e.hasFiles()&&n?.warn?.log(`Ollama model ${this.model} multimodal support depends on the specific model. Ensure you're using a vision-capable model like llava.`),new st(this.url,this.model,e)}}class st{static{u(this,"OllamaChatCompletionRequest")}chat;url;model;constructor(e,s,n){this.url=e,this.model=s,this.chat=n}async execute(e){const{recorder:s}=e,n={stream:!1,options:{temperature:.7},...nt(this.chat,this.model)};s?.debug?.log(n);let o;try{const r=await fetch(`${this.url}/api/chat`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(n)});if(!r.ok)throw console.log(r),new Error(`HTTP error! status: ${r.status}`);const i=await r.json();o=rt(i)}catch(r){s?.error?.log("Error fetching Ollama response:",r),o={type:"error",error:{type:"OllamaError",message:r.message||"Unexpected error from Ollama"},usage:{in:0,out:0},raw:JSON.stringify(r)}}return s?.debug?.log(o),o}}function nt(t,e){const s=[];t.system&&s.push({role:"system",content:t.system});let n;t.tools.length>0&&(n=t.tools.map(r=>({type:"function",function:r})));const o=t.messages.map(r=>{if(r.role==="tool")return r.content.map(i=>({role:"tool",tool_call_id:i.id,content:i.content}));if(r.role==="assistant"){const i=r.toolCalls?.map(a=>{const c=a.id;return{type:"function",function:{name:a.name,arguments:a.arguments},...c&&{id:c}}});return{role:r.role,content:r.content,...i&&{toolCalls:i}}}if(typeof r.content=="string")return{role:r.role,content:r.content};{const i=H(r.content),a=F(r.content).map(c=>c.base64);return{role:r.role,content:i,...a.length>0&&{images:a}}}}).flat(1);return{model:e,messages:[...s,...o],...n&&{tools:n}}}u(nt,"prepareRequest$2");function rt(t){if(t.done_reason==="stop"&&t.message){const e=t.message.content,s=[];if(t.message.tool_calls)for(const o of t.message.tool_calls)s.push({id:o.id,name:o.function.name,arguments:o.function.arguments});const n=s.length>0;return{type:"success",id:`ollama-${Date.now()}`,model:t.model,reason:n?E.FunctionCall:E.Stop,message:{role:"assistant",content:e,...n&&{toolCalls:s}},usage:{in:t.prompt_eval_count||0,out:t.eval_count||0},raw:t}}return{type:"error",error:{type:"OllamaError",message:"Unexpected error from Ollama"},usage:{in:0,out:0},raw:t}}u(rt,"translateResponse$1");const d={GPT_4_1:"gpt-4.1",GPT_4_1_2025_04_14:"gpt-4.1-2025-04-14",GPT_4_1_MINI:"gpt-4.1-mini",GPT_4_1_MINI_2025_04_14:"gpt-4.1-mini-2025-04-14",GPT_4_1_NANO:"gpt-4.1-nano",GPT_4_1_NANO_2025_04_14:"gpt-4.1-nano-2025-04-14",GPT_4O:"gpt-4o",GPT_4O_2024_05_13:"gpt-4o-2024-05-13",GPT_4O_2024_08_06:"gpt-4o-2024-08-06",GPT_4O_2024_11_20:"gpt-4o-2024-11-20",GPT_4O_AUDIO_PREVIEW:"gpt-4o-audio-preview",GPT_4O_AUDIO_PREVIEW_2024_10_01:"gpt-4o-audio-preview-2024-10-01",GPT_4O_AUDIO_PREVIEW_2024_12_17:"gpt-4o-audio-preview-2024-12-17",GPT_4O_AUDIO_PREVIEW_2025_06_03:"gpt-4o-audio-preview-2025-06-03",GPT_4O_MINI:"gpt-4o-mini",GPT_4O_MINI_2024_07_18:"gpt-4o-mini-2024-07-18",GPT_4O_MINI_AUDIO_PREVIEW:"gpt-4o-mini-audio-preview",GPT_4O_MINI_AUDIO_PREVIEW_2024_12_17:"gpt-4o-mini-audio-preview-2024-12-17",GPT_4O_MINI_REALTIME_PREVIEW:"gpt-4o-mini-realtime-preview",GPT_4O_MINI_REALTIME_PREVIEW_2024_12_17:"gpt-4o-mini-realtime-preview-2024-12-17",GPT_4O_MINI_SEARCH_PREVIEW:"gpt-4o-mini-search-preview",GPT_4O_MINI_SEARCH_PREVIEW_2025_03_11:"gpt-4o-mini-search-preview-2025-03-11",GPT_4O_MINI_TRANSCRIBE:"gpt-4o-mini-transcribe",GPT_4O_MINI_TTS:"gpt-4o-mini-tts",GPT_4O_REALTIME_PREVIEW:"gpt-4o-realtime-preview",GPT_4O_REALTIME_PREVIEW_2024_10_01:"gpt-4o-realtime-preview-2024-10-01",GPT_4O_REALTIME_PREVIEW_2024_12_17:"gpt-4o-realtime-preview-2024-12-17",GPT_4O_REALTIME_PREVIEW_2025_06_03:"gpt-4o-realtime-preview-2025-06-03",GPT_4O_SEARCH_PREVIEW:"gpt-4o-search-preview",GPT_4O_SEARCH_PREVIEW_2025_03_11:"gpt-4o-search-preview-2025-03-11",GPT_4O_TRANSCRIBE:"gpt-4o-transcribe",O3_MINI:"o3-mini",O3_MINI_2025_01_31:"o3-mini-2025-01-31",O4_MINI:"o4-mini",O4_MINI_2025_04_16:"o4-mini-2025-04-16"},ot=[d.GPT_4_1,d.GPT_4_1_2025_04_14,d.GPT_4_1_MINI,d.GPT_4_1_MINI_2025_04_14,d.GPT_4_1_NANO,d.GPT_4_1_NANO_2025_04_14,d.GPT_4O,d.GPT_4O_2024_05_13,d.GPT_4O_2024_08_06,d.GPT_4O_2024_11_20,d.GPT_4O_AUDIO_PREVIEW,d.GPT_4O_AUDIO_PREVIEW_2024_10_01,d.GPT_4O_AUDIO_PREVIEW_2024_12_17,d.GPT_4O_AUDIO_PREVIEW_2025_06_03,d.GPT_4O_MINI,d.GPT_4O_MINI_2024_07_18,d.GPT_4O_MINI_AUDIO_PREVIEW,d.GPT_4O_MINI_AUDIO_PREVIEW_2024_12_17,d.GPT_4O_MINI_REALTIME_PREVIEW,d.GPT_4O_MINI_REALTIME_PREVIEW_2024_12_17,d.GPT_4O_MINI_SEARCH_PREVIEW,d.GPT_4O_MINI_SEARCH_PREVIEW_2025_03_11,d.GPT_4O_MINI_TRANSCRIBE,d.GPT_4O_MINI_TTS,d.GPT_4O_REALTIME_PREVIEW,d.GPT_4O_REALTIME_PREVIEW_2024_10_01,d.GPT_4O_REALTIME_PREVIEW_2024_12_17,d.GPT_4O_REALTIME_PREVIEW_2025_06_03,d.GPT_4O_SEARCH_PREVIEW,d.GPT_4O_SEARCH_PREVIEW_2025_03_11,d.GPT_4O_TRANSCRIBE,d.O3_MINI,d.O3_MINI_2025_01_31,d.O4_MINI,d.O4_MINI_2025_04_16];class it{static{u(this,"OpenAIChatCompletionRequest")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:n,model:o}=this.provider,r=ct(this.chat,o);s?.debug?.heading.log("[Open AI Provider] Using the ChatCompletion API"),s?.debug?.log(r);let i;try{const a=await n.chat.completions.create(r);i=ut(a)}catch(a){s?.error?.log(a),i={type:"error",error:{type:a.type??"Undetermined",message:a.message??"Unexpected error from OpenAI"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function at(t){switch(t){case"length":return E.Length;case"stop":return E.Stop;case"tool_calls":return E.FunctionCall;default:return E.Error}}u(at,"getStopReason");function ct(t,e){const s=[];t.system&&s.push({role:"system",content:t.system});let n;t.tools.length>0&&(n=t.tools.map(r=>({type:"function",function:r})));const o=t.messages.map(r=>{if(r.role==="tool")return r.content.map(i=>({role:"tool",tool_call_id:i.id,content:i.content}));if(r.role==="assistant"){const i=r.toolCalls?.map(a=>{const c=a.id;return{type:"function",function:{name:a.name,arguments:typeof a.arguments=="string"?a.arguments:JSON.stringify(a.arguments)},...c&&{id:c}}});return{role:r.role,content:r.content,...i&&{toolCalls:i}}}if(typeof r.content=="string")return{role:r.role,content:r.content};{const i=[],a=H(r.content);a&&i.push({type:"text",text:a});const c=F(r.content);c.length>0&&i.push(...c.map(f=>({type:"image_url",image_url:{url:`data:${f.mimeType};base64,${f.base64}`}})));const l=W(r.content);return l.length>0&&i.push(...l.map(f=>({type:"file",file:{filename:f.name,file_data:`data:${f.mimeType};base64,${f.base64}`}}))),{role:r.role,content:i}}}).flat(1);return{model:e,messages:[...s,...o],...n&&{tools:n}}}u(ct,"prepareRequest$1");function ut(t){if(t.choices.length>0){const e=t.choices[0],s=e.message.tool_calls?.map(n=>({id:n.id,name:n.function.name,arguments:n.function.arguments}));return{type:"success",id:t.id,model:t.model,reason:at(e.finish_reason),message:{content:e.message.content??"",role:e.message.role,toolCalls:s},usage:{in:t.usage?.prompt_tokens??0,out:t.usage?.completion_tokens??0},raw:t}}return{type:"error",error:{type:"undetermined",message:"Unexpected response from OpenAI"},usage:{in:t.usage?.prompt_tokens??0,out:t.usage?.completion_tokens??0},raw:t}}u(ut,"translateResponse");class lt{static{u(this,"OpenAIResponsesAPI")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:n,model:o}=this.provider,r=ft(this.chat,o);s?.debug?.heading.log("[Open AI Provider] Using the Responses API"),s?.debug?.log(r);let i;try{const a=await n.responses.create(r);i=pt(a)}catch(a){s?.error?.log(a),i={type:"error",error:{type:a.type??"Undetermined",message:a.message??"Unexpected error from OpenAI"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function ft(t,e){const s=t.messages.map(r=>{if(r.role==="tool")return r.content.map(i=>({type:"function_call_output",call_id:i.id,output:i.content}));if(r.role==="assistant"){const i=r.toolCalls?.map(a=>{const c=a.id;return{type:"function",function:{name:a.name,arguments:typeof a.arguments=="string"?a.arguments:JSON.stringify(a.arguments)},...c&&{id:c}}});return{role:r.role,content:r.content,...i&&{toolCalls:i}}}if(typeof r.content=="string")return{role:r.role,content:r.content};{const i=[],a=He(r.content);a&&i.push({type:"input_text",text:a});const c=F(r.content);c.length>0&&i.push(...c.map(f=>({type:"input_image",image_url:`data:${f.mimeType};base64,${f.base64}`})));const l=W(r.content);return l.length>0&&i.push(...l.map(f=>({type:"input_file",filename:f.path,file_data:`data:${f.mimeType};base64,${f.base64}`}))),{role:r.role,content:i}}}).flat(1),n={model:e,input:s},o=t.latest();if(o&&o.role==="user"){let r="";const i=We(o.content);t.system&&(r=t.system),i&&(r=r?`${r}
|
|
8
|
+
|
|
9
|
+
${i}`:i),r&&(n.instructions=r)}return t.tools.length>0&&(n.tools=t.tools.map(r=>({type:"function",strict:!0,...r}))),n}u(ft,"prepareRequest");function pt(t){if(t.error)return{type:"error",error:{type:t.error.code||"undetermined",message:t.error.message||"Response generation failed"},usage:{in:t.usage?.input_tokens??0,out:t.usage?.output_tokens??0},raw:t};const e=t.output?.filter(s=>s.type==="function_call")?.map(s=>({id:s.id||"",name:s.function?.name||"",arguments:s.function?.arguments||""}));return{type:"success",id:t.id,model:t.model||"",reason:t.incomplete_details?E.Error:E.Stop,message:{content:t.output_text||"",role:"assistant",...e?.length&&{toolCalls:e}},usage:{in:t.usage?.input_tokens??0,out:t.usage?.output_tokens??0},raw:t}}u(pt,"translateResponseToAIResponse");const _t=d.GPT_4_1;class dt{static{u(this,"OpenAIProvider")}name="OpenAI";client;model;constructor(e,s){this.model=s||_t,this.client=new ke({apiKey:e})}createChatRequest(e,s={}){const{recorder:n}=s;return ot.includes(this.model)?new lt(this,e):new it(this,e)}}function gt(t,e){if(!e||Object.keys(e).length===0)throw new A(`The provider ${t} is not configured. Please check your configuration.`);switch(t){case"openai":return new dt(e["api-key"],e.model);case"anthropic":return new Je(e["api-key"],e.model);case"googleai":return new Ze(e["api-key"],e.model);case"ollama":{const s=e;return new tt(s.model,s.url)}default:throw new A("The provider is unsupported")}}u(gt,"getProvider");class oe extends A{static{u(this,"TaskError")}constructor(e,s){super(e,{code:"TASK_ERROR",id:s?.id,details:{taskType:s?.taskType,taskIndex:s?.taskIndex,...s?.details},cause:s?.cause}),Object.setPrototypeOf(this,oe.prototype)}}const m={Running:"running",Success:"success",PartialSuccess:"partialSuccess",Fail:"fail"};var T=(t=>(t[t.Trace=10]="Trace",t[t.Debug=20]="Debug",t[t.Info=30]="Info",t[t.Warn=40]="Warn",t[t.Error=50]="Error",t[t.Fatal=60]="Fatal",t))(T||{});class mt{static{u(this,"Recorder")}instanceId=crypto.randomUUID();currentLevel=T.Info;logs=[];writers=[];_debug;_info;_warn;_error;constructor(){this.buildMethods()}buildMethods(){this._debug=T.Debug>=this.currentLevel?this.createLoggingFunction(T.Debug):null,this._info=T.Info>=this.currentLevel?this.createLoggingFunction(T.Info):null,this._warn=T.Warn>=this.currentLevel?this.createLoggingFunction(T.Warn):null,this._error=this.createLoggingFunction(T.Error)}set level(e){this.currentLevel=e,this.buildMethods()}get level(){return this.currentLevel}get info(){return this._info}get warn(){return this._warn}get error(){return this._error}get debug(){return this._debug}subscribe(e){this.writers.includes(e)||this.writers.push(e)}unsubscribe(e){const s=this.writers.indexOf(e);s!==-1&&this.writers.splice(s,1)}publish(e){this.logs.push(e);for(const s of this.writers)s.handleEvent(e)}logFunction(e,s,...n){let o=n.map(r=>typeof r=="string"?{message:r}:r instanceof Error?Me(r):r);this.publish({level:e,time:Date.now(),kind:s,payload:o})}createLoggingFunction(e){return{log:this.logFunction.bind(this,e,"body"),heading:{log:this.logFunction.bind(this,e,"heading")}}}getLogs(e=T.Info){return this.logs.filter(s=>s.level>=e)}async shutdown(){for(const e of this.writers)typeof e.flush=="function"&&await e.flush()}}async function ht(t,e){const{defaults:s,tag:n}=e;let o=null,r="";if(t)try{r=ee(t),o=await C(r,{encoding:"utf-8"})}catch{throw new Error(`${n} not found, see --help for details`)}else{for(const i of s.formats)try{r=ee(s.name+"."+i),o=await C(r,{encoding:"utf-8"});break}catch{continue}if(o===null)throw new Error(`${n} not found, see --help for details`)}return{content:o,format:r.split(".").pop()??""}}u(ht,"searchAndLoadFile");async function yt(t,e){let s="";for(const n of t){const o=await ae(n);e?.debug?.log(`many-files parser. For glob "${n}", found ${o.length} files.`);const r=await Promise.all(o.map(async i=>{const a=await C(i,"utf-8");return i+`:
|
|
10
|
+
`+a}));s+=r.join(`
|
|
11
|
+
`)}return s}u(yt,"loadManyFiles");function Et(t,e){t=t.replace("**/*","**");const s=/(?<asterisks>\*{1,2})(?<extension>\.[^\\/]+)?/,n=t.match(s);if(n){let o="";return n.groups?.asterisks.length==1?o+=e.stem:o+=e.dir+e.stem,n.groups?.extension?o+=n.groups.extension:o+=e.ext,t.replace(n[0],o)}return t}u(Et,"replaceFilePattern");function It(t){const e=/(?<name>[^\\/]+)(?<extension>\.[^\\/]+)$/,s=t.match(e);return s&&s.length>0&&s.groups?{abs:t,dir:t.replace(s[0],""),ext:s.groups.extension,stem:s.groups.name,name:s[0]}:null}u(It,"pathToComponents");async function de(t){const e=De(t);try{await ce(e)}catch{await Fe(e),await de(e)}}u(de,"ensureDirectoryExistence");async function wt({filePath:t,content:e}){await de(t),await Ce(t,e)}u(wt,"writeFileWithDirectories");const B=[".jpg",".jpeg",".png",".gif",".webp",".bmp",".tiff"],J=[".pdf"],X=[".txt",".md",".markdown"],ge=20*1024*1024;function Tt(t){return t.type==="text"}u(Tt,"isTextFileInfo");function At(t){return t.type==="image"||t.type==="document"}u(At,"isBase64FileInfo");function Pt(t){const e=ue(t).toLowerCase();if(X.includes(e))return"utf-8";if(B.includes(e)||J.includes(e))return"base64";{const s=[...X,...B,...J];throw new Error(`Unsupported file type: ${e}. Supported types: ${s.join(", ")}`)}}u(Pt,"getEncodingForFile");async function q(t,e){const s=ee(t);try{await ce(s)}catch{throw new Error(`File not found: ${t}`)}const n=await Le(s);if(n.size>ge)throw new Error(`File too large: ${n.size} bytes. Maximum allowed: ${ge} bytes`);const o=ue(s).toLowerCase(),r=s.split("/").pop()||"";if((e||Pt(s))==="utf-8"){if(!X.includes(o))throw new Error(`Unsupported text file type: ${o}. Supported types: ${X.join(", ")}`);let a;switch(o){case".txt":a="text/plain";break;case".md":case".markdown":a="text/markdown";break;default:a="text/plain"}const c=await C(s,"utf-8");return{path:s,content:c,mimeType:a,size:n.size,name:r,type:"text"}}else{let a,c;if(B.includes(o))switch(a="image",o){case".jpg":case".jpeg":c="image/jpeg";break;case".png":c="image/png";break;case".gif":c="image/gif";break;case".webp":c="image/webp";break;case".bmp":c="image/bmp";break;case".tiff":c="image/tiff";break;default:c="image/jpeg"}else if(J.includes(o))a="document",c="application/pdf";else throw new Error(`Unsupported file type: ${o}. Supported types: ${[...B,...J].join(", ")}`);const f=(await C(s)).toString("base64");return{path:s,base64:f,mimeType:c,size:n.size,name:r,type:a}}}u(q,"loadFileContent");function vt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):St(t.using,e)?!t.jobs||typeof t.jobs!="object"?(e&&(e.value="Missing or invalid 'jobs' property"),!1):me(t.jobs,e)?!0:(e&&(e.value=`Invalid 'jobs' property: ${e?.value}`),!1):(e&&(e.value=`Invalid 'using' property: ${e?.value}`),!1)}u(vt,"isJobConfig");function St(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(typeof t.engine!="string")return e&&(e.value="Missing or invalid 'engine' property"),!1;if(!["openai","anthropic","ollama","googleai"].includes(t.engine))return e&&(e.value="Invalid provider type. Must be 'openai', 'anthropic', 'googleai', or 'ollama'"),!1;switch(t.engine){case"ollama":if("model"in t&&typeof t.model!="string")return e&&(e.value="Property 'model' must be a string"),!1;if("url"in t&&typeof t.url!="string")return e&&(e.value="Property 'url' must be a string"),!1;break;case"googleai":case"anthropic":case"openai":if("api-key"in t&&typeof t["api-key"]!="string")return e&&(e.value="Property 'api-key' must be a string"),!1;if("model"in t&&typeof t.model!="string")return e&&(e.value="Property 'model' must be a string"),!1;break}return!0}u(St,"isUsing");function me(t,e){for(const[s,n]of Object.entries(t))if(!Nt(n,e))return e&&(e.value=`Invalid job '${s}': ${e?.value}`),!1;return!0}u(me,"isDAGJob");function Nt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(!Rt(t,e))return!1;if("dependsOn"in t&&t.dependsOn!==void 0){const s=t.dependsOn;if(typeof s!="string")if(Array.isArray(s)){for(let n=0;n<s.length;n++)if(typeof s[n]!="string")return e&&(e.value=`Dependency at index ${n} must be a string`),!1}else return e&&(e.value="Property 'dependsOn' must be a string or array of strings"),!1}return!0}u(Nt,"isDAGJobValue");function Rt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):"batch"in t?bt(t,e):Ot(t,e)}u(Rt,"isJob");function Ot(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if("batch"in t)return e&&(e.value="Serial job should not have a batch property"),!1;if(t.tools!==void 0){if(!Array.isArray(t.tools))return e&&(e.value="Property 'tools' must be an array"),!1;for(const s of t.tools)if(typeof s!="string")return e&&(e.value="All tools must be strings"),!1}if(!Array.isArray(t.steps))return e&&(e.value="Property 'steps' must be an array"),!1;for(let s=0;s<t.steps.length;s++)if(!he(t.steps[s],e))return e&&(e.value=`Invalid step at index ${s}: ${e?.value}`),!1;return!0}u(Ot,"isSerialJob");function bt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(t.tools!==void 0){if(!Array.isArray(t.tools))return e&&(e.value="Property 'tools' must be an array"),!1;for(const s of t.tools)if(typeof s!="string")return e&&(e.value="All tools must be strings"),!1}if(!Array.isArray(t.batch))return e&&(e.value="Property 'batch' must be an array"),!1;for(let s=0;s<t.batch.length;s++)if(!kt(t.batch[s],e))return e&&(e.value=`Invalid batch item at index ${s}: ${e?.value}`),!1;if(!Array.isArray(t.steps))return e&&(e.value="Property 'steps' must be an array"),!1;for(let s=0;s<t.steps.length;s++)if(!he(t.steps[s],e))return e&&(e.value=`Invalid step at index ${s}: ${e?.value}`),!1;return!0}u(bt,"isBatchJob");function kt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(t.type!=="files")return e&&(e.value="Property 'type' must be 'files'"),!1;if(typeof t.source!="string")return e&&(e.value="Property 'source' must be a string"),!1;if(typeof t.bind!="string")return e&&(e.value="Property 'bind' must be a string"),!1;if(t["skip-if"]!==void 0){if(!Array.isArray(t["skip-if"]))return e&&(e.value="Property 'skip-if' must be an array"),!1;for(let s=0;s<t["skip-if"].length;s++)if(!Mt(t["skip-if"][s],e))return e&&(e.value=`Invalid skip condition at index ${s}: ${e?.value}`),!1}return!0}u(kt,"isBatchOptions");function Mt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):t.type!=="file-exist"?(e&&(e.value="Property 'type' must be 'file-exist'"),!1):typeof t.pattern!="string"?(e&&(e.value="Property 'pattern' must be a string"),!1):!0}u(Mt,"isSkipOptions");function he(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):!t.uses||typeof t.uses!="string"?(e&&(e.value="Step must have a string 'uses' property"),!1):t.uses==="chat"?$t(t,e):t.uses==="write-to-disk"?Gt(t,e):(e&&(e.value=`Unknown uses type: ${t.uses}`),!1)}u(he,"isStep");function $t(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(t.uses!=="chat")return e&&(e.value="Uses must be 'chat'"),!1;if(typeof t.message!="string")return e&&(e.value="Property 'message' must be a string"),!1;if(t.output!==void 0){if(!t.output||typeof t.output!="object"||Array.isArray(t.output))return e&&(e.value="Property 'output' must be an object"),!1;const s=["string","string[]","number","boolean"];for(const[n,o]of Object.entries(t.output))if(typeof n!="string"||typeof o!="string"||!s.includes(o))return e&&(e.value="Property 'output' must be a Record<string, ResTypeStrings> where ResTypeStrings is 'string' | 'string[]' | 'number' | 'boolean'"),!1}if(t.system!==void 0&&typeof t.system!="string")return e&&(e.value="Property 'system' must be a string"),!1;if(t.replace!==void 0){if(!Array.isArray(t.replace))return e&&(e.value="Property 'replace' must be an array"),!1;for(let s=0;s<t.replace.length;s++)if(!xt(t.replace[s],e))return e&&(e.value=`Invalid replace at index ${s}: ${e?.value}`),!1}if(t.tools!==void 0){if(!Array.isArray(t.tools))return e&&(e.value="Property 'tools' must be an array"),!1;for(const s of t.tools)if(typeof s!="string")return e&&(e.value="All tools must be strings"),!1}if(t.images!==void 0){if(!Array.isArray(t.images))return e&&(e.value="Property 'images' must be an array"),!1;for(let s=0;s<t.images.length;s++)if(!Lt(t.images[s],e))return e&&(e.value=`Invalid image at index ${s}: ${e?.value}`),!1}if(t.documents!==void 0){if(!Array.isArray(t.documents))return e&&(e.value="Property 'documents' must be an array"),!1;for(let s=0;s<t.documents.length;s++)if(!Ct(t.documents[s],e))return e&&(e.value=`Invalid document at index ${s}: ${e?.value}`),!1}if(t.references!==void 0){if(!Array.isArray(t.references))return e&&(e.value="Property 'references' must be an array"),!1;for(let s=0;s<t.references.length;s++)if(!Ft(t.references[s],e))return e&&(e.value=`Invalid reference at index ${s}: ${e?.value}`),!1}return!0}u($t,"isChatStep");function Gt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(t.uses!=="write-to-disk")return e&&(e.value="Uses must be 'write-to-disk'"),!1;if(typeof t.output!="string")return e&&(e.value="Property 'output' must be a string"),!1;if(t.keys!==void 0&&typeof t.keys!="string")if(Array.isArray(t.keys)){for(let s=0;s<t.keys.length;s++)if(typeof t.keys[s]!="string")return e&&(e.value=`Key at index ${s} must be a string`),!1}else return e&&(e.value="Property 'keys' must be a string or array of strings"),!1;return!0}u(Gt,"isWriteToDiskStep");function xt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(typeof t.pattern!="string")return e&&(e.value="Property 'pattern' must be a string"),!1;if(t.source!=="file")return e&&(e.value="Property 'source' must be 'file'"),!1;if(typeof t.files!="string"&&!Array.isArray(t.files))return e&&(e.value="Property 'files' must be a string or an array of strings"),!1;if(Array.isArray(t.files)){for(let s=0;s<t.files.length;s++)if(typeof t.files[s]!="string")return e&&(e.value=`Files entry at index ${s} must be a string`),!1}return!0}u(xt,"isReplace");function Lt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):typeof t.file!="string"?(e&&(e.value="Property 'file' must be a string"),!1):!0}u(Lt,"isImageReference");function Ct(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):typeof t.file!="string"?(e&&(e.value="Property 'file' must be a string"),!1):!0}u(Ct,"isDocumentReference");function Ft(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):typeof t.file!="string"?(e&&(e.value="Property 'file' must be a string"),!1):!0}u(Ft,"isTextFileReference");class Dt{static{u(this,"FileRunPlanner")}constructor(e,s,n=[]){this.source=e,this.bind=s,this.skipConditions=n}async plan(e){const s=[],n=await ae(this.source,{withFileTypes:!0});for(const o of n){const r=o.fullpath(),i=It(r);let a=!1;for(const c of this.skipConditions)if(a=await c.eval({components:i}),a)break;if(!a){const c=await $e(r,"utf-8"),l={variables:{[this.bind]:c,...i},tasks:e};s.push(l)}}return s}}class Ut{static{u(this,"MultiPlanner")}planners;constructor(e){this.planners=e}async plan(e){const s=this.planners.map(async o=>await o.plan(e));return(await Promise.all(s)).flat()}}function te(t,e,s="{{}}"){const n=s==="{{}}"?/\{\{(.*?)\}\}/g:/\{(.*?)\}/g;return t=t.replace(n,(o,r)=>{if(r=r.trim(),Object.prototype.hasOwnProperty.call(e,r)){const i=e[r];return i==null?"":String(i)}return o}),t}u(te,"replaceVariables");class Ht{static{u(this,"FileExistSkipCondition")}constructor(e){this.pattern=e}type="file-exist";async eval(e){const s=te(this.pattern,e.components,"{}");try{return await Ge(s,xe.F_OK),!0}catch{return!1}}}function K(t,e=0){const s={};for(const[n,o]of Object.entries(t))if(typeof o=="string")switch(o){case"string":s[n]=e===0?I.string():I.string().optional();break;case"number":s[n]=e===0?I.number():I.number().optional();break;case"boolean":s[n]=e===0?I.boolean():I.boolean().optional();break;case"string[]":s[n]=I.array(I.string());break;default:throw new Error(`Unsupported declarative type: ${o}`)}else if(Array.isArray(o))if(o.length===1&&typeof o[0]=="object"){const r=K(o[0],e+1);s[n]=I.array(I.object(r))}else throw new Error(`Unsupported array format for key ${n}. Expected [DeclarativeSchema].`);else s[n]=I.object(K(o,e+1));return s}u(K,"declarativeToOutputSchema");function z(t){if(t instanceof I.ZodString)return["string","Your answer"];if(t instanceof I.ZodNumber)return["number",42];if(t instanceof I.ZodBoolean)return["boolean",!0];if(t instanceof I.ZodArray){const e=t.element;if(e instanceof I.ZodString)return["string array",["answer 1","answer 2","third answer"]];if(e instanceof I.ZodNumber)return["number array",[42,59,3.14]];if(e instanceof I.ZodBoolean)return["boolean array",[!0,!1,!1]];if(e instanceof I.ZodObject){const[s,n]=z(e);return["object array",[n,n]]}return["array",[]]}if(t instanceof I.ZodObject){const e=t.shape,s={};for(const[n,o]of Object.entries(e)){const[r,i]=z(o);s[n]=i}return["JSON object",s]}if(t instanceof I.ZodOptional){const e=t.unwrap(),[s,n]=z(e);return[`${s} | undefined`,n]}}u(z,"zodToExample");function ye(t){return typeof t=="object"&&Object.values(t).every(e=>e&&typeof e=="object"&&"_def"in e)}u(ye,"isOutputSchema");class Ee{static{u(this,"AbstractInstruct")}type="instruct";prompt;system=null;inputs={};tools={};files=[];textReferences=[];instructions=[];schema;rawResponse;_taggedSections=void 0;_result=void 0;constructor(e,s){this.prompt=e,this.schema=s}setInputs(e){this.inputs=e}addInput(e,s){this.inputs[e]=s}addTools(e){for(const s of e)this.tools[s.name]=s}addTool(e){this.tools[e.name]=e}addImage(e){if(e.type!=="image")throw new Error(`Expected image file, got ${e.type}`);const s=e;this.files.push(s)}addDocument(e){if(e.type!=="document")throw new Error(`Expected document file, got ${e.type}`);const s=e;this.files.push(s)}addFile(e){if(!At(e))throw new Error(`Expected image or document file, got ${e.type}`);this.files.push(e)}addReference(e,s){if(typeof e=="string"){this.textReferences.push({content:e,name:s?.name});return}if(Tt(e))this.textReferences.push({content:e.content,name:s?.name??e.name});else throw new Error(`Expected text file, got ${e.type}`)}addInstructions(e){if(typeof e!="string"||e.trim()==="")throw new Error("Instruction must be a non-empty string");this.instructions.push(e)}hasTools(){return Object.keys(this.tools).length>0}hasFiles(){return this.files.length>0}get result(){return this._result}compile(e,s={}){const n=this.createUserMessage(e,s),o=this.createInstructions();return{message:n,instructions:o}}createUserMessage(e,s={}){const{recorder:n,options:o}=s,r={...e,...this.inputs};let i=te(this.prompt,r);if(this.textReferences.length>0)for(const[a,c]of this.textReferences.entries()){const l=c.name?`: ${c.name}`:"";i+=`
|
|
12
|
+
|
|
13
|
+
## Reference ${a+1}${l}
|
|
14
|
+
|
|
15
|
+
\`\`\`${c.content}'''`}if(o?.warnUnused){const a=i.match(/\{\{(.*?)\}\}/g);if(a)throw n?.error.log(`Warning unused variables ${a.join(", ")}`),new Error(`Unused variables: ${a.join(", ")}`)}return i}createInstructions(e=""){if(e=`# Instructions
|
|
16
|
+
|
|
17
|
+
`+e,Object.keys(this.schema).length>0){e+=`## Output Format Instructions
|
|
18
|
+
`,e+=`
|
|
19
|
+
Here is how you should format your output. Follow the instructions strictly.
|
|
20
|
+
`;for(const[n,o]of Object.entries(this.schema)){const r=this.generateFieldInstructions(n,o);e+=r}}if(this.instructions.length>0){e+=`
|
|
21
|
+
## Additional Instructions
|
|
22
|
+
|
|
23
|
+
`;for(const n of this.instructions)e+=`- ${n}
|
|
24
|
+
`}return e}generateFieldInstructions(e,s){const[n,o]=z(s);return`
|
|
25
|
+
- Use <${e}></${e}> tags to indicate the answer for ${e}. The answer must be a ${n}.
|
|
26
|
+
Example: <${e}>${JSON.stringify(o)}</${e}>
|
|
27
|
+
`}finalize(e,s={}){const{recorder:n}=s;if(this.rawResponse=e,Object.keys(this.schema).length===0){if(e.trim()==="{}"||e.trim()==="")return this._result={},this._result;throw new Error("Schema is empty, but rawValue is not an empty object representation or empty string.")}this._taggedSections=this._taggedSections||this.parseTaggedSections(e);const r={};for(const[i,a]of Object.entries(this.schema)){const c=this._taggedSections.tags[i];if(c!==void 0)r[i]=this.preprocessValue(a,c);else if(a.def.type!=="optional")throw new Error(`Expected results with tag ${i} but it does not exist`)}try{const i={};for(const[a,c]of Object.entries(this.schema))a in r&&(i[a]=c.parse(r[a]));return this._result=i,this._result}catch(i){if(i&&typeof i=="object"&&"issues"in i){const a=i.issues.map(c=>`${c.path.join(".")}: ${c.message}`).join(", ");throw new Error(`Validation failed: ${a}`)}throw i}}preprocessValue(e,s){switch(s=s.trim(),e.def.type){case"string":try{return JSON.parse(s)}catch{if(typeof s=="string")return s;throw new Error(`Cannot parse '${s}' as string. Ensure it is a valid JSON string or a plain string.`)}case"number":{const n=parseFloat(s);if(isNaN(n))throw new Error(`Cannot parse '${s}' as number`);return n}case"boolean":{const n=s.toLowerCase();if(n==="true")return!0;if(n==="false")return!1;throw new Error(`Cannot parse '${s}' as boolean. Expected 'true' or 'false'`)}case"array":{if(s==="")return[];try{const n=JSON.parse(s);if(Array.isArray(n))return n}catch{}if(s.includes(","))return s.split(",").map(n=>{const o=n.trim();try{return JSON.parse(o)}catch{return o}}).filter(n=>n!=="")}case"object":{s.includes("```json")&&(s=s.replace(/```json/g,"").replace(/```/g,""));try{return JSON.parse(s)}catch(n){throw new Error(`Cannot parse object as JSON: ${n.message}`)}}case"optional":{const n=e.def.innerType;return this.preprocessValue(n,s)}default:return s}}parseTaggedSections(e){e.trim().startsWith("```json")&&e.trim().endsWith("```")&&(e=e.trim().slice(7,-3).trim());const s=/<(\w+)>(.*?)<\/\1>/gs,n={};let o=e;o=o.replace(s,(i,a,c)=>(n[a]=c,""));const r=/<(\w+)>(.*?)(?:<\/?\w+>|$)/gs;return o=o.replace(r,(i,a,c)=>(n[a]=c,"")),{tags:n,remaining:o.trim()}}}class $ extends Ee{static{u(this,"Instruct")}constructor(e,s){super(e,s)}static with(e,s){if(!s)return new $(e,{response:I.string()});if(ye(s))return new $(e,s);{const n=K(s);return new $(e,n)}}}function D(t){return Array.isArray(t)?t:[t]}u(D,"arrayify");function S(t,e){return e?`${e}:${t.slice(0,8)}`:t.slice(0,8)}u(S,"friendly");function Wt(t){return new Promise(e=>setTimeout(e,t))}u(Wt,"delay");const Bt={name:"brave",description:"Perform a search using the Brave search engine",parameters:{type:"object",properties:{searchTerm:{type:"string",description:"The search term to query"}},required:["searchTerm"]}};class Jt{static{u(this,"BraveSearchTool")}name="brave";schema=Bt;apiKey;throttle;lastExecTime=0;constructor(e){e&&this.setConfig(e)}setConfig(e){const{rateLimit:s}=e;this.apiKey=e["api-key"],this.throttle=s?1100/s:void 0}async execute(e,s={}){const{searchTerm:n}=e,{recorder:o}=s;if(o?.debug?.heading.log(`Brave: searching for ${n}`),this.throttle){for(;Date.now()-this.lastExecTime<this.throttle;)await Wt(this.throttle-(Date.now()-this.lastExecTime));this.lastExecTime=Date.now()}try{const r=this.apiKey,i="https://api.search.brave.com/res/v1/web/search",a=new URL(i);a.searchParams.append("q",n),a.searchParams.append("format","json");const c=await fetch(a.toString(),{method:"GET",headers:{Accept:"application/json","X-Subscription-Token":r}});if(!c.ok)throw new Error(`[Brave] HTTP error ${c.status}: ${c.statusText}`);return await c.json()}catch(r){throw o?.error.log("[Brave] Error fetching search results:",r),r}}}const Xt=new Jt,qt={name:"calculator",description:"Performs basic arithmetic operations",parameters:{type:"object",properties:{operation:{type:"string",description:"The operation to perform (add, subtract, multiply, divide)",enum:["add","subtract","multiply","divide"]},a:{type:"number",description:"First operand"},b:{type:"number",description:"Second operand"}},required:["operation","a","b"]}},Kt={name:"calculator",schema:qt,execute:u(async t=>{const{operation:e,a:s,b:n}=t;switch(e){case"add":return`${s} + ${n} = ${s+n}`;case"subtract":return`${s} - ${n} = ${s-n}`;case"multiply":return`${s} * ${n} = ${s*n}`;case"divide":if(n===0)throw new Error("Cannot divide by zero");return`${s} / ${n} = ${s/n}`;default:throw new Error(`Unknown operation: ${e}`)}},"execute")};class zt{static{u(this,"ToolRegistry")}executables={};config;setConfig(e){this.config=e}register(e){if(this.executables[e.name])throw new Error(`Tool with name '${e.name}' is already registered`);this.executables[e.name]=e}get(e){const s=this.executables[e];if(!s)throw new Error(`Tool '${e}' is not registered`);return s.setConfig?.(this.config[e]),s}}let U;function Ie(){return U||(U=new zt,U.register(Kt),U.register(Xt)),U}u(Ie,"getToolRegistry");const Zt={async convert(t,e){const{recorder:s,toolNames:n}=e,{message:o,system:r,replace:i}=t;let a;t.output?a=$.with(o,t.output):a=$.with(o),r&&(a.system=r);const c=[...new Set([...n??[],...t.tools??[]])];for(const l of c){const f=Ie().get(l);a.addTool(f)}if(i){for(const l of i)if(l.source==="file"){const f=D(l.files),_=await yt(f,s);a.addInput(l.pattern,_)}}if(t.images)for(const l of t.images)try{const f=await q(l.file,"base64");a.addFile(f)}catch(f){throw new Error(`Failed to load image '${l.file}': ${f.message}`)}if(t.documents)for(const l of t.documents)try{const f=await q(l.file,"base64");a.addFile(f)}catch(f){throw new Error(`Failed to load document '${l.file}': ${f.message}`)}if(t.references)for(const l of t.references)try{const f=await q(l.file,"utf-8");a.addReference(f)}catch(f){throw new Error(`Failed to load reference file '${l.file}': ${f.message}`)}return a}};class jt{static{u(this,"StepToClassRegistry")}converters=new Map;get(e){const s=this.converters.get(e);if(!s)throw new Error(`No converter registered for step: ${e}`);return s}register(e,s){this.converters.set(e,s)}}class se{static{u(this,"WriteOutputTask")}constructor(e,s=["response"]){this.output=e,this.keys=s}type="write-to-disk"}const Yt={async convert(t){if(t.keys){const e=D(t.keys);return new se(t.output,e)}return new se(t.output)}},ne=new jt;ne.register("write-to-disk",Yt),ne.register("chat",Zt);async function Z(t,e){const{recorder:s}=e,n=t.tools??void 0,o=t.steps.map(async r=>(r.uses,await ne.get(r.uses).convert(r,{recorder:s,toolNames:n})));return Promise.all(o)}u(Z,"configToTasks");async function we(t,e){const{batch:s}=t;return s.length===1?Te(s[0]):new Ut(s.map(n=>Te(n)))}u(we,"configToPlanner");function Te(t){switch(t.type){case"files":let e;return t["skip-if"]&&(e=t["skip-if"].map(n=>Qt(n))),new Dt(t.source,t.bind,e)}}u(Te,"batchOptionsToPlanner");function Qt(t){switch(t.type){case"file-exist":return new Ht(t.pattern)}}u(Qt,"skipOptionsToSkipConditions");function Vt(t){return t.success===!1&&t.error!==void 0}u(Vt,"isErrorResult");function j(t,e){return{response:t,stats:e,success:!0}}u(j,"createResult");function Y(t,e,s){return{response:e,stats:s,error:t,success:!1}}u(Y,"createErrorResult");class es{static{u(this,"WriteToDiskTaskHandler")}taskType="write-to-disk";canHandle(e){return e&&typeof e=="object"&&"type"in e&&e.type==="write-to-disk"}async execute(e){const{task:s,variables:n,options:o={},recorder:r}=e,i=s.output,a=s.keys??[];if(o?.warnUnused){const f=a.filter(_=>!(_ in n));f.length>0&&r?.warn?.log(`[Write To Disk] The following keys were not found in the variables: ${f.join(", ")}`)}let c="";if(a.length===1?c=n[a[0]]??"<not found>":c=a.map(f=>`[${f}]:
|
|
28
|
+
${n[f]??"<not found>"}
|
|
29
|
+
`).join(`
|
|
30
|
+
`),o?.dryRun){r?.info?.log("[Dry run] Write to Disk is not executed.");return}let l="";i.includes("*")?l=Et(i,n.file):l=te(i,n,"{}"),await wt({filePath:l,content:c})}}var Q=(t=>(t.LastResult="lastResult",t))(Q||{});function ts(t,e,s){const{options:n,recorder:o}=s,r=n?.warnUnused??!0;for(const[i,a]of Object.entries(t))r&&e[i]&&o?.warn?.log(`Warning: Variable "${i}" is being overwritten. Previous value: ${e[i]}, new value: ${a}`),e[i]=a}u(ts,"setResultsIntoVariables");class ss{static{u(this,"ChatTaskHandler")}taskType="instruct";canHandle(e){return e&&typeof e=="object"&&"type"in e&&e.type==="instruct"}async execute(e){const{task:s,...n}=e;await ns({instruct:s,...n})}}async function ns(t){const{instruct:e,chat:s,provider:n,stats:o,variables:r,options:i,recorder:a}=t;e.system&&s.addSystem(e.system);const{message:c,instructions:l}=e.compile(r,{recorder:a,options:i});if(e.hasFiles()?s.addUser(c,l,e.files):s.addUser(c,l),e.hasTools()){const _=os(e.tools);s.setToolSchemas(_)}if(i?.dryRun)return a?.debug?.log(s),{action:"complete"};let f=!0;for(;f;){const g=await n.createChatRequest(s,{recorder:a}).execute({recorder:a});if(o.in+=g.usage.in,o.out+=g.usage.out,g.type==="error")throw new Error(JSON.stringify(g.error));if(g.type==="success")switch(g.reason){case E.Stop:{if(g.message.content){const y=g.message.content;s.addAssistant(y);const h=e.finalize(y,{recorder:a});ts(h,r,{options:i,recorder:a}),r[Q.LastResult]=h}return f=!1,{action:"continue"}}case E.Length:throw new Error("Incomplete model output due to `max_tokens` parameter or token limit");case E.FunctionCall:{let y=g.message;if(g.message&&s.addAssistant(y.content,y.toolCalls),y.toolCalls&&y.toolCalls.length>0){const h=await rs(y.toolCalls,e,{recorder:a});a?.debug?.log(h),s.addTools(h),f=!0}else f=!1;break}}if(g.type!=="success")throw a?.debug?.log(g),new Error("Unexpected response type")}return{action:"continue"}}u(ns,"executeChatAction");async function rs(t,e,s={}){const{recorder:n}=s,o=[];for(const r of t)o.push(new Promise((i,a)=>{const c=e.tools[r.name];if(!c){a(`Tool not found: ${r.name}`);return}n?.debug?.heading.log(`Executing tool ${c.name}`);let l={};try{l=typeof r.arguments=="string"?JSON.parse(r.arguments):r.arguments}catch{a(`argument for tool ${r.name} is not valid: ${JSON.stringify(r.arguments)}`)}c.execute(l).then(f=>{n?.debug?.log(`Complete tool ${c.name}: ${r.id}`),i({id:r.id,name:r.name,content:JSON.stringify(f)})}).catch(a)}));return Promise.all(o)}u(rs,"executeToolCalls");function os(t){const e=[];for(const[s,n]of Object.entries(t))e.push(n.schema);return e}u(os,"getToolSchemas");class is{static{u(this,"TaskRegistry")}handlers=new Map;register(e){this.handlers.set(e.taskType,e)}getHandler(e){return this.handlers.get(e.type)}hasHandler(e){return this.handlers.has(e.type)}async executeTask(e){const{task:s}=e,n=s.type,o=this.getHandler(s);if(!o)throw new Error(`No handler registered for action type: ${n}`);if(!o.canHandle(s))throw new Error(`Handler found but action does not match expected format: ${n}`);await o.execute(e)}}function as(){const t=new is;return t.register(new ss),t}u(as,"createBaseRegistry");function cs(){const t=as();return t.register(new es),t}u(cs,"createNodeRegistry");const re=u((t,...e)=>{const s=u(async o=>{const{recorder:r}=o;let i=[];return"steps"in t?i=await Z(t,{recorder:r}):i=[t,...e],i},"prepare");return{execute:u(async o=>{const{provider:r,variables:i,options:a,stats:c,recorder:l,name:f}=o,_=crypto.randomUUID(),g=cs();l?.info?.log({type:"task",id:_,status:m.Running,message:`[${S(_,f)}] Starting job`});try{const y=await s({recorder:l}),h=new Ue;for(const[b,v]of y.entries()){l?.info?.log({type:"task",id:_,status:m.Running,message:`[${S(_,f)}] Processing step ${b+1}: ${v.type}`});try{await g.executeTask({task:v,chat:h,provider:r,variables:i,options:a,stats:c,recorder:l})}catch(N){throw N instanceof A?N:new oe(`Error executing task ${v.type}`,{id:_,taskType:v.type,taskIndex:b,cause:N instanceof Error?N:new Error(String(N))})}}return l?.info?.log({type:"task",status:m.Success,id:_,message:`[${S(_,f)}] Completed ${y.length} steps`}),j(i[Q.LastResult],c)}catch(y){const h=y instanceof A?y:new A("Serial workflow execution failed",{id:_,cause:y instanceof Error?y:new Error(String(y))});return l?.info?.log({type:"task",status:m.Fail,id:_,message:`[${S(_,f)}] Failed: ${h.message}`}),l?.error.log(h),Y(h,i[Q.LastResult],c)}},"execute")}},"serialWorkflow"),Ae=u((t,...e)=>{const s=u(async o=>{const{recorder:r}=o;let i=[],a=null;if("batch"in t){const c=t;a=await we(c),i=await Z(c,{recorder:r})}else a=t,i=[...e];return[a,i]},"prepare");return{execute:u(async o=>{const{provider:r,variables:i,options:a,stats:c,recorder:l,name:f}=o,_=crypto.randomUUID();try{const[g,y]=await s({recorder:l}),h=await g.plan(y);if(l?.debug?.heading.log("Runs",h),h.length===0)return l?.info?.log("No runs to execute"),j([],c);let b=0;l?.info?.log({type:"task",status:m.Running,id:_,message:`[${S(_,"CRW")}] Working on 0/${h.length}`});const v=u(async(R,V)=>{try{return await re(...R.tasks).execute({provider:r,variables:{...R.variables,...i},options:a,stats:c,recorder:l,name:`${f}-${V}`})}catch(k){const ie=k instanceof A?k:new A("Error executing run",{cause:k instanceof Error?k:new Error(String(k))});return l?.error?.log(ie),Y(ie,null,c)}finally{b++,l?.info?.log({type:"task",status:m.Running,id:_,message:`[${S(_,"CRW")}] Working on ${b}/${h.length}`})}},"executeRun"),N=5;let G=[];for(let R=0;R<h.length;R+=N){const V=h.slice(R,R+N),k=await Promise.all(V.map(v));G=G.concat(k)}const M=G.some(Vt);l?.info?.log({type:"task",status:M?m.PartialSuccess:m.Success,id:_,message:`[${S(_,"CRW")}] All jobs (${h.length}) completed${M?" with some errors":""}`});const L=G.map(R=>R.response);return j(L,c)}catch(g){const y=g instanceof A?g:new A("Concurrent workflow execution failed",{id:_,cause:g instanceof Error?g:new Error(String(g))});return l?.error?.log(y),Y(y,null,c)}},"execute")}},"concurrentWorkflow");class us{static{u(this,"DAGParser")}static parse(e){const s=new Map;for(const[o,r]of Object.entries(e)){const i=this.parseNodeDefinition(o,r);s.set(o,i)}return this.validateDependencies(s),this.checkForCycles(s),{stages:this.createExecutionStages(s),nodes:s}}static parseNodeDefinition(e,s){if(this.isSimpleTask(s))return{id:e,tasks:Array.isArray(s)?s:[s],dependencies:[],executionType:"serial"};if(this.isConcurrentNodeDefinition(s)){const i=s,a=i.dependsOn?D(i.dependsOn):[];return{id:e,tasks:i.tasks,dependencies:a,planner:i.planner,executionType:"concurrent"}}const n=s,o=n.dependsOn?D(n.dependsOn):[],r=D(n.task);return{id:e,tasks:r,dependencies:o,executionType:"serial"}}static isSimpleTask(e){return e.type||Array.isArray(e)}static isConcurrentNodeDefinition(e){return e&&typeof e=="object"&&"planner"in e}static validateDependencies(e){for(const s of e.values())for(const n of s.dependencies)if(!e.has(n))throw new A(`Node "${s.id}" depends on non-existent node "${n}"`)}static checkForCycles(e){const s=new Set,n=new Set,o=u(r=>{if(n.has(r))return!0;if(s.has(r))return!1;s.add(r),n.add(r);const i=e.get(r);for(const a of i.dependencies)if(o(a))return!0;return n.delete(r),!1},"hasCycle");for(const r of e.keys())if(o(r))throw new A(`Circular dependency detected involving node "${r}"`)}static createExecutionStages(e){const s=[],n=new Set,o=new Set(e.keys());for(;o.size>0;){const r=[];for(const i of o)e.get(i).dependencies.every(l=>n.has(l))&&r.push(i);if(r.length===0)throw new A("Unable to resolve DAG dependencies - possible circular reference");s.push(r),r.forEach(i=>{n.add(i),o.delete(i)})}return s}}class ls{static{u(this,"DAGJobToDefinition")}static async convert(e,s){const{recorder:n}=s,o={};for(const[r,i]of Object.entries(e)){const{dependsOn:a,...c}=i;if("batch"in c){const l=c,f=await we(l),_=await Z(l,{recorder:n}),g={planner:f,tasks:_,...a?{dependsOn:a}:{}};o[r]=g}else{const l=await Z(c,{recorder:n});if(a){const f={task:l,dependsOn:a};o[r]=f}else o[r]=l}}return o}}async function fs(t,e,s,n={}){const{variables:o}=s,r=e.nodes.get(t);try{let i;if(r.executionType==="concurrent"&&r.planner?i=await Ae(r.planner,...r.tasks).execute({...s,variables:o,name:t}):i=await re(...r.tasks).execute({...s,variables:o,name:t}),!i.success)throw new A(`Node "${t}" failed: ${i.error?.message}`);return i.response}catch(i){if(!n.continueOnError)throw i;return null}}u(fs,"executeNode");const ps=u((t,e={})=>{const s=u(async(o,r)=>{const{recorder:i}=r,a={value:""};return me(o,a)?await ls.convert(o,r):(i?.warn?.log(a),o)},"prepare");return{execute:u(async o=>{const{stats:r,recorder:i}=o,{maxConcurrency:a=3}=e,c=crypto.randomUUID();try{const l=await s(t,{recorder:i});i?.debug?.log(l);const f=us.parse(l),_=new Map;i?.info?.log({type:"task",id:c,status:m.Running,message:`[${S(c)}] Starting workflow execution with ${f.stages.length} stages`});for(const[y,h]of f.stages.entries()){i?.info?.log({type:"task",id:c,status:m.Running,message:`[${S(c)}] Stage ${y+1}/${f.stages.length}, executing ${h.length} nodes: ${h.join(", ")}`});const b=Math.min(h.length,a);for(let v=0;v<h.length;v+=b){const N=h.slice(v,v+b);(await Promise.all(N.map(async M=>{const L=await fs(M,f,o,e);return{nodeId:M,result:L}}))).forEach(({nodeId:M,result:L})=>{_.set(M,L)})}}i?.info?.log({type:"task",status:m.Success,id:c,message:`[${S(c)}] Workflow execution completed successfully`});const g=Object.fromEntries(_);return j(g,r)}catch(l){const f=l instanceof A?l:new A("DAG workflow execution failed",{id:c,cause:l instanceof Error?l:new Error(String(l))});return i?.info?.log({type:"task",status:m.Fail,id:c,message:`[${S(c)}] Workflow execution failed: ${f.message}`}),i?.error?.log(f),Y(f,null,r)}},"execute")}},"dagWorkflow"),_s=["\u280B","\u2819","\u2839","\u2838","\u283C","\u2834","\u2826","\u2827","\u2807","\u280F"],x={success:"\u2713",fail:"\u2717",spinning:_s};class ds{static{u(this,"ConsoleWriter")}tasks=new Map;entries=[];truncate=0;intervalId=null;spinnerInterval=80;lastRender="";isRendering=!1;inline=!0;constructor(e={}){this.truncate=e.truncate??0,this.inline=e.inline??!0}startSpinner(){this.intervalId===null&&(this.intervalId=setInterval(()=>{[...this.tasks.values()].some(s=>s.status===m.Running)&&this.renderTasks()},this.spinnerInterval))}stopSpinner(){this.intervalId!==null&&(clearInterval(this.intervalId),this.intervalId=null)}renderTasks(){if(this.isRendering)return;if(this.isRendering=!0,this.inline&&this.lastRender){const r=this.lastRender.split(`
|
|
31
|
+
`).length;le.moveCursor(process.stdout,0,-r+1),le.clearScreenDown(process.stdout)}const e=[...this.tasks.values()],s=e.filter(r=>r.status===m.Running),n=e.filter(r=>r.status===m.Success||r.status===m.Fail);if(s.length===0&&n.length>0){let r="";for(const i of n){if(i.status===m.Success){const a=w.green(x.success);r+=`${a} ${i.text}
|
|
32
|
+
`}else if(i.status===m.Fail){const a=w.red(x.fail);r+=`${a} ${i.text}
|
|
33
|
+
`}this.tasks.delete(i.id)}console.log(r)}for(const r of this.entries){const{level:i,time:a,kind:c,payload:l}=r;c==="heading"?ms(i,l,{truncate:this.truncate}):hs(i,l,{truncate:this.truncate})}this.entries=[];let o="";for(const r of this.tasks.values())if(r.status===m.Running){const i=w.cyan(x.spinning[r.frameIndex]);r.frameIndex=(r.frameIndex+1)%x.spinning.length,o+=`${i} ${r.text}
|
|
34
|
+
`}else if(r.status===m.Success){const i=w.green(x.success);o+=`${i} ${r.text}
|
|
35
|
+
`}else if(r.status===m.Fail){const i=w.red(x.fail);o+=`${i} ${r.text}
|
|
36
|
+
`}this.lastRender=o,process.stdout.write(o),this.isRendering=!1}handleEvent(e){const{level:s,time:n,payload:o}=e;if(o.length>0&&gs(o[0])){const i=o[0],{id:a,message:c,status:l}=i;if(l===m.Running)this.tasks.set(a,{id:a,text:c,status:l,frameIndex:0});else if((l===m.Success||l===m.Fail)&&this.tasks.has(a)){const f=this.tasks.get(a);f.status=l,f.text=c}}else this.entries.push(e);this.renderTasks();const r=[...this.tasks.values()].some(i=>i.status===m.Running);r&&this.intervalId===null?this.startSpinner():!r&&this.intervalId!==null&&this.stopSpinner()}destroy(){this.stopSpinner()}}function gs(t){if(typeof t!="object"||t===null)return!1;const e=t;if(e.type!=="task"||typeof e.id!="string"||typeof e.message!="string")return!1;switch(e.status){case m.Running:case m.Success:case m.PartialSuccess:case m.Fail:return!0;default:return!1}}u(gs,"isTask");function ms(t,e,s){let n,o;t===T.Error?(n=w.red,o=w.redBright.bold):t===T.Warn?(n=w.yellow,o=w.yellowBright.bold):t>=T.Info?(n=w.blue,o=w.whiteBright.bold):(n=w.gray,o=w.white);const{message:r,data:i}=Se(e);console.log(`${n("==>")} ${o(r)}`),ve(t,i,s)}u(ms,"heading");function hs(t,e,s){let n;t===T.Error?n=w.red:t===T.Warn?n=w.yellow:t>=T.Info?n=w.white:n=w.gray;const{message:o,data:r}=Se(e);o&&console.log(n(o)),ve(t,r,s)}u(hs,"body");const Pe=" ";function ve(t,e,s){let n;t===T.Error?(n=w.red,s.truncate=0):t==T.Warn?n=w.yellow:t>=T.Info?n=w.white:n=w.gray,e.forEach(o=>{if(typeof o=="string"){console.log(n(`${Pe}${o}`));return}for(const[r,i]of Object.entries(o)){let a=JSON.stringify(i,ys(s.truncate)," ");const c=`${r}: ${a}`.split(`
|
|
37
|
+
`).map(l=>Pe+l).join(`
|
|
38
|
+
`);console.log(n(c))}})}u(ve,"values");function Se(t){const[e,...s]=t;let n="",o=s;if(e){let{message:r,...i}=e;n=r&&typeof r=="string"?r:"",Object.keys(i).length>0&&(o=[i,...o])}return{message:n,data:o}}u(Se,"toMsgData");function ys(t){return t===0?null:(e,s)=>typeof s=="string"&&s.length>t?s.slice(0,t)+"<...>":s}u(ys,"truncator");export{A,ds as C,$ as I,T as L,mt as R,se as W,Ee as a,K as b,Ae as c,ps as d,ht as e,vt as f,gt as g,Ie as h,ye as i,q as l,re as s};
|
package/dist/index.d.ts
CHANGED
|
@@ -1,3 +1,6 @@
|
|
|
1
|
+
import * as z from 'zod/v4';
|
|
2
|
+
import z__default from 'zod/v4';
|
|
3
|
+
|
|
1
4
|
type PlainObject = Record<string, unknown>;
|
|
2
5
|
type ProgramOptions = {
|
|
3
6
|
dryRun?: boolean;
|
|
@@ -74,18 +77,23 @@ declare class Recorder {
|
|
|
74
77
|
|
|
75
78
|
interface FileInfo {
|
|
76
79
|
path: string;
|
|
77
|
-
base64
|
|
80
|
+
base64?: string;
|
|
81
|
+
content?: string;
|
|
78
82
|
mimeType: string;
|
|
79
83
|
size: number;
|
|
80
84
|
name: string;
|
|
81
|
-
type: "image" | "document";
|
|
85
|
+
type: "image" | "document" | "text";
|
|
82
86
|
}
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
87
|
+
type TextFileInfo = FileInfo & {
|
|
88
|
+
content: string;
|
|
89
|
+
base64?: never;
|
|
90
|
+
type: "text";
|
|
91
|
+
};
|
|
92
|
+
type Base64FileInfo = FileInfo & {
|
|
93
|
+
base64: string;
|
|
94
|
+
content?: never;
|
|
95
|
+
type: "image" | "document";
|
|
96
|
+
};
|
|
89
97
|
|
|
90
98
|
interface ToolSchema {
|
|
91
99
|
name: string;
|
|
@@ -147,6 +155,8 @@ interface AIProviderConfig {
|
|
|
147
155
|
googleai: GoogleAIProviderConfig;
|
|
148
156
|
}
|
|
149
157
|
interface AIProvider {
|
|
158
|
+
get name(): string;
|
|
159
|
+
get model(): string;
|
|
150
160
|
createChatRequest(chat: Chat, context: {
|
|
151
161
|
recorder?: Recorder;
|
|
152
162
|
}): AIRequest;
|
|
@@ -279,7 +289,7 @@ interface DAGWorkflowOptions {
|
|
|
279
289
|
}
|
|
280
290
|
|
|
281
291
|
declare class Axle {
|
|
282
|
-
|
|
292
|
+
provider: AIProvider;
|
|
283
293
|
private stats;
|
|
284
294
|
private variables;
|
|
285
295
|
recorder: Recorder;
|
|
@@ -301,46 +311,65 @@ declare class Axle {
|
|
|
301
311
|
executeDAG(dagDefinition: DAGDefinition, variables?: Record<string, any>, options?: DAGWorkflowOptions): Promise<WorkflowResult>;
|
|
302
312
|
get logs(): RecorderEntry[];
|
|
303
313
|
/**
|
|
304
|
-
* Load a file
|
|
305
|
-
* @param filePath - Path to the
|
|
306
|
-
* @
|
|
314
|
+
* Load a file with the specified encoding or auto-detect based on file extension
|
|
315
|
+
* @param filePath - Path to the file
|
|
316
|
+
* @param encoding - How to load the file: "utf-8" for text, "base64" for binary, or omit for auto-detection
|
|
317
|
+
* @returns FileInfo object with appropriate content based on encoding
|
|
307
318
|
*/
|
|
308
|
-
static
|
|
319
|
+
static loadFileContent(filePath: string): Promise<FileInfo>;
|
|
320
|
+
static loadFileContent(filePath: string, encoding: "utf-8"): Promise<TextFileInfo>;
|
|
321
|
+
static loadFileContent(filePath: string, encoding: "base64"): Promise<Base64FileInfo>;
|
|
309
322
|
}
|
|
310
323
|
|
|
311
|
-
declare enum
|
|
324
|
+
declare enum ResultType {
|
|
312
325
|
String = "string",
|
|
313
326
|
List = "string[]",
|
|
314
327
|
Number = "number",
|
|
315
328
|
Boolean = "boolean"
|
|
316
329
|
}
|
|
317
|
-
type
|
|
318
|
-
type
|
|
319
|
-
|
|
320
|
-
|
|
330
|
+
type ResultTypeUnion = `${ResultType}`;
|
|
331
|
+
type DeclarativeSchema = {
|
|
332
|
+
[key: string]: ResultTypeUnion | DeclarativeSchema | DeclarativeSchema[];
|
|
333
|
+
};
|
|
334
|
+
type OutputSchema = Record<string, z__default.ZodTypeAny>;
|
|
335
|
+
type InferedOutputSchema<T extends OutputSchema> = {
|
|
336
|
+
[K in keyof T]: z__default.output<T[K]>;
|
|
321
337
|
};
|
|
322
338
|
|
|
323
|
-
declare abstract class AbstractInstruct<
|
|
339
|
+
declare abstract class AbstractInstruct<T extends OutputSchema> implements Task {
|
|
324
340
|
readonly type = "instruct";
|
|
325
|
-
protected _result: StructuredOutput<O> | undefined;
|
|
326
341
|
prompt: string;
|
|
327
342
|
system: string | null;
|
|
328
343
|
inputs: Record<string, string>;
|
|
329
344
|
tools: Record<string, ToolExecutable>;
|
|
330
|
-
files:
|
|
331
|
-
|
|
345
|
+
files: Base64FileInfo[];
|
|
346
|
+
textReferences: Array<{
|
|
347
|
+
content: string;
|
|
348
|
+
name?: string;
|
|
349
|
+
}>;
|
|
350
|
+
instructions: string[];
|
|
351
|
+
schema: T;
|
|
332
352
|
rawResponse: string;
|
|
333
|
-
|
|
334
|
-
|
|
353
|
+
protected _taggedSections: {
|
|
354
|
+
tags: Record<string, string>;
|
|
355
|
+
remaining: string;
|
|
356
|
+
} | undefined;
|
|
357
|
+
protected _result: InferedOutputSchema<T> | undefined;
|
|
358
|
+
protected constructor(prompt: string, schema: T);
|
|
335
359
|
setInputs(inputs: Record<string, string>): void;
|
|
336
360
|
addInput(name: string, value: string): void;
|
|
337
361
|
addTools(tools: ToolExecutable[]): void;
|
|
338
362
|
addTool(tool: ToolExecutable): void;
|
|
339
363
|
addImage(file: FileInfo): void;
|
|
364
|
+
addDocument(file: FileInfo): void;
|
|
340
365
|
addFile(file: FileInfo): void;
|
|
366
|
+
addReference(textFile: FileInfo | TextFileInfo | string, options?: {
|
|
367
|
+
name?: string;
|
|
368
|
+
}): void;
|
|
369
|
+
addInstructions(instruction: string): void;
|
|
341
370
|
hasTools(): boolean;
|
|
342
371
|
hasFiles(): boolean;
|
|
343
|
-
get result():
|
|
372
|
+
get result(): InferedOutputSchema<T> | undefined;
|
|
344
373
|
compile(variables: Record<string, string>, runtime?: {
|
|
345
374
|
recorder?: Recorder;
|
|
346
375
|
options?: {
|
|
@@ -350,57 +379,45 @@ declare abstract class AbstractInstruct<O extends Record<string, ResTypeStrings>
|
|
|
350
379
|
message: string;
|
|
351
380
|
instructions: string;
|
|
352
381
|
};
|
|
353
|
-
protected
|
|
382
|
+
protected createUserMessage(variables: Record<string, string>, runtime?: {
|
|
354
383
|
recorder?: Recorder;
|
|
355
384
|
options?: {
|
|
356
385
|
warnUnused?: boolean;
|
|
357
386
|
};
|
|
358
387
|
}): string;
|
|
359
|
-
protected
|
|
360
|
-
|
|
361
|
-
|
|
362
|
-
|
|
363
|
-
|
|
364
|
-
|
|
365
|
-
*/
|
|
366
|
-
finalize(rawValue: string, taggedSections?: {
|
|
367
|
-
tags: Record<string, string>;
|
|
368
|
-
remaining: string;
|
|
369
|
-
}): StructuredOutput<O>;
|
|
388
|
+
protected createInstructions(instructions?: string): string;
|
|
389
|
+
protected generateFieldInstructions(key: string, schema: z.ZodTypeAny): string;
|
|
390
|
+
finalize(rawValue: string, runtime?: {
|
|
391
|
+
recorder?: Recorder;
|
|
392
|
+
}): InferedOutputSchema<T>;
|
|
393
|
+
private preprocessValue;
|
|
370
394
|
protected parseTaggedSections(input: string): {
|
|
371
395
|
tags: Record<string, string>;
|
|
372
396
|
remaining: string;
|
|
373
397
|
};
|
|
374
|
-
protected typeResponses(typeString: ResTypeStrings, rawValue: string): StringToType<ResTypes>;
|
|
375
398
|
}
|
|
376
399
|
|
|
377
|
-
|
|
378
|
-
|
|
379
|
-
|
|
380
|
-
|
|
381
|
-
|
|
382
|
-
|
|
383
|
-
|
|
400
|
+
declare class Instruct<T extends OutputSchema> extends AbstractInstruct<T> {
|
|
401
|
+
constructor(prompt: string, schema: T);
|
|
402
|
+
static with<T extends OutputSchema>(prompt: string, schema: T): Instruct<T>;
|
|
403
|
+
static with<T extends DeclarativeSchema>(prompt: string, schema: T): Instruct<OutputSchema>;
|
|
404
|
+
static with(prompt: string): Instruct<{
|
|
405
|
+
response: z.ZodString;
|
|
406
|
+
}>;
|
|
384
407
|
}
|
|
385
408
|
|
|
386
|
-
|
|
387
|
-
|
|
388
|
-
|
|
389
|
-
|
|
390
|
-
|
|
391
|
-
|
|
392
|
-
|
|
393
|
-
|
|
409
|
+
declare class ChainOfThought<T extends OutputSchema> extends AbstractInstruct<T> {
|
|
410
|
+
constructor(prompt: string, schema: T);
|
|
411
|
+
static with<T extends OutputSchema>(prompt: string, schema: T): ChainOfThought<T>;
|
|
412
|
+
static with<T extends DeclarativeSchema>(prompt: string, schema: T): ChainOfThought<OutputSchema>;
|
|
413
|
+
static with(prompt: string): ChainOfThought<{
|
|
414
|
+
response: z.ZodString;
|
|
415
|
+
}>;
|
|
416
|
+
createInstructions(instructions?: string): string;
|
|
417
|
+
finalize(rawValue: string, runtime?: {
|
|
394
418
|
recorder?: Recorder;
|
|
395
|
-
|
|
396
|
-
|
|
397
|
-
};
|
|
398
|
-
}): {
|
|
399
|
-
message: string;
|
|
400
|
-
instructions: string;
|
|
401
|
-
};
|
|
402
|
-
finalize(rawValue: string): StructuredOutput<O> & {
|
|
403
|
-
thinking: any;
|
|
419
|
+
}): InferedOutputSchema<T> & {
|
|
420
|
+
thinking: string;
|
|
404
421
|
};
|
|
405
422
|
}
|
|
406
423
|
|
|
@@ -449,16 +466,17 @@ interface ChatStep extends StepBase {
|
|
|
449
466
|
uses: "chat";
|
|
450
467
|
system?: string;
|
|
451
468
|
message: string;
|
|
452
|
-
output?: Record<string,
|
|
469
|
+
output?: Record<string, ResultTypeUnion>;
|
|
453
470
|
replace?: Replace[];
|
|
454
471
|
tools?: string[];
|
|
455
472
|
images?: ImageReference[];
|
|
456
473
|
documents?: DocumentReference[];
|
|
474
|
+
references?: TextFileReference[];
|
|
457
475
|
}
|
|
458
476
|
interface WriteToDiskStep extends StepBase {
|
|
459
477
|
uses: "write-to-disk";
|
|
460
478
|
output: string;
|
|
461
|
-
keys
|
|
479
|
+
keys?: string | string[];
|
|
462
480
|
}
|
|
463
481
|
interface Replace {
|
|
464
482
|
source: "file";
|
|
@@ -471,6 +489,9 @@ interface ImageReference {
|
|
|
471
489
|
interface DocumentReference {
|
|
472
490
|
file: string;
|
|
473
491
|
}
|
|
492
|
+
interface TextFileReference {
|
|
493
|
+
file: string;
|
|
494
|
+
}
|
|
474
495
|
|
|
475
496
|
interface ConcurrentWorkflow {
|
|
476
497
|
(jobConfig: BatchJob): WorkflowExecutable;
|
|
@@ -489,4 +510,24 @@ interface SerialWorkflow {
|
|
|
489
510
|
}
|
|
490
511
|
declare const serialWorkflow: SerialWorkflow;
|
|
491
512
|
|
|
492
|
-
|
|
513
|
+
declare class ConsoleWriter implements RecorderWriter {
|
|
514
|
+
private tasks;
|
|
515
|
+
private entries;
|
|
516
|
+
private truncate;
|
|
517
|
+
private intervalId;
|
|
518
|
+
private spinnerInterval;
|
|
519
|
+
private lastRender;
|
|
520
|
+
private isRendering;
|
|
521
|
+
private inline;
|
|
522
|
+
constructor(options?: {
|
|
523
|
+
truncate?: number;
|
|
524
|
+
inline?: boolean;
|
|
525
|
+
});
|
|
526
|
+
private startSpinner;
|
|
527
|
+
private stopSpinner;
|
|
528
|
+
private renderTasks;
|
|
529
|
+
handleEvent(event: RecorderEntry): void;
|
|
530
|
+
destroy(): void;
|
|
531
|
+
}
|
|
532
|
+
|
|
533
|
+
export { type AIProvider, Axle, ChainOfThought, ConsoleWriter, type DAGDefinition, type DAGWorkflowOptions, type FileInfo, Instruct, LogLevel, type SerializedExecutionResponse, WriteOutputTask, concurrentWorkflow, dagWorkflow, serialWorkflow };
|
package/dist/index.js
CHANGED
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
var
|
|
1
|
+
var h=Object.defineProperty;var a=(u,e)=>h(u,"name",{value:e,configurable:!0});import{R as p,A as o,g,s as l,d,l as c,a as f,i as w,b as k}from"./consoleWriter-CtNNCSna.js";import{C,I as D,L as F,W as P,c as j}from"./consoleWriter-CtNNCSna.js";import*as m from"zod/v4";import"@anthropic-ai/sdk";import"@google/genai";import"openai";import"serialize-error";import"fs/promises";import"glob";import"node:fs/promises";import"node:path";import"chalk";import"node:readline";class v{static{a(this,"Axle")}provider;stats={in:0,out:0};variables={};recorder=new p;constructor(e){if(Object.entries(e).length!==1)throw new o("Must have exactly one config");try{const r=Object.keys(e)[0],t=e[r];this.provider=g(r,t)}catch(r){throw r instanceof o?r:new o("Failed to initialize provider",{code:"PROVIDER_INIT_ERROR",cause:r instanceof Error?r:new Error(String(r))})}}addWriter(e){this.recorder.subscribe(e)}async execute(...e){try{let r;return r=await l(...e).execute({provider:this.provider,variables:this.variables,stats:this.stats,recorder:this.recorder}),r}catch(r){const t=r instanceof o?r:new o("Execution failed",{cause:r instanceof Error?r:new Error(String(r))});return this.recorder.error?.log(t),{response:null,error:t,success:!1}}}async executeDAG(e,r={},t){try{return await d(e,t).execute({provider:this.provider,variables:{...this.variables,...r},stats:this.stats,recorder:this.recorder})}catch(s){const n=s instanceof o?s:new o("DAG execution failed",{cause:s instanceof Error?s:new Error(String(s))});return this.recorder.error?.log(n),{response:null,error:n,success:!1}}}get logs(){return this.recorder.getLogs()}static async loadFileContent(e,r){return r==="utf-8"?c(e,"utf-8"):r==="base64"?c(e,"base64"):c(e)}}class i extends f{static{a(this,"ChainOfThought")}constructor(e,r){super(e,r)}static with(e,r){if(!r)return new i(e,{response:m.string()});if(w(r))return new i(e,r);{const t=k(r);return new i(e,t)}}createInstructions(e=""){return super.createInstructions(`Let's think step by step. Use <thinking></thinking> tags to show your reasoning and thought process.
|
|
2
2
|
|
|
3
|
-
|
|
3
|
+
`)}finalize(e,r={}){const t=super.finalize(e,r),s=this.parseTaggedSections(e);let n="thinking";return"thinking"in s.tags||("think"in s.tags?(n="think",r.recorder?.warn?.log("No <thinking> section found in the response but found <think> instead. This may be a limitation of the model or prompt.")):r.recorder?.warn?.log("No <thinking> section found in the response. Please ensure your response includes a <thinking> tag.")),{...t,thinking:s.tags[n]||""}}}export{v as Axle,i as ChainOfThought,C as ConsoleWriter,D as Instruct,F as LogLevel,P as WriteOutputTask,j as concurrentWorkflow,d as dagWorkflow,l as serialWorkflow};
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@fifthrevision/axle",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.5.0",
|
|
4
4
|
"repository": {
|
|
5
5
|
"type": "git",
|
|
6
6
|
"url": "https://github.com/johncch/axle.git"
|
|
@@ -41,7 +41,8 @@
|
|
|
41
41
|
"glob": "^11.0.0",
|
|
42
42
|
"openai": "^5.0.2",
|
|
43
43
|
"serialize-error": "^12.0.0",
|
|
44
|
-
"yaml": "^2.5.0"
|
|
44
|
+
"yaml": "^2.5.0",
|
|
45
|
+
"zod": "^3.25.67"
|
|
45
46
|
},
|
|
46
47
|
"devDependencies": {
|
|
47
48
|
"@types/jest": "^29.5.12",
|
package/dist/dag-DM0XblG3.js
DELETED
|
@@ -1,19 +0,0 @@
|
|
|
1
|
-
var he=Object.defineProperty;var c=(t,e)=>he(t,"name",{value:e,configurable:!0});import ye from"@anthropic-ai/sdk";import{GoogleGenAI as Ee,Type as Ie,FinishReason as w}from"@google/genai";import we from"openai";import{serializeError as Te}from"serialize-error";import{readFile as Ae,access as Pe,constants as ve}from"fs/promises";import{glob as ee}from"glob";import{readFile as C,access as te,stat as Ne,writeFile as Oe,mkdir as Re}from"node:fs/promises";import{resolve as K,extname as Se,dirname as be}from"node:path";class E extends Error{static{c(this,"AxleError")}code;id;details;constructor(e,s){super(e,{cause:s?.cause}),this.name=this.constructor.name,this.code=s?.code||"AXLE_ERROR",this.id=s?.id,this.details=s?.details,Object.setPrototypeOf(this,E.prototype)}}const R={CLAUDE_OPUS_4_20250514:"claude-opus-4-20250514",CLAUDE_OPUS_4_LATEST:"claude-opus-4-0",CLAUDE_SONNET_4_20250514:"claude-sonnet-4-20250514",CLAUDE_SONNET_4_LATEST:"claude-sonnet-4-0",CLAUDE_3_7_SONNET_20250219:"claude-3-7-sonnet-20250219",CLAUDE_3_7_SONNET_LATEST:"claude-3-7-sonnet-latest",CLAUDE_3_5_HAIKU_20241022:"claude-3-5-haiku-20241022",CLAUDE_3_5_HAIKU_LATEST:"claude-3-5-haiku-latest"},se=[R.CLAUDE_3_7_SONNET_LATEST,R.CLAUDE_3_7_SONNET_20250219,R.CLAUDE_3_5_HAIKU_LATEST,R.CLAUDE_3_5_HAIKU_20241022,R.CLAUDE_SONNET_4_LATEST,R.CLAUDE_SONNET_4_20250514,R.CLAUDE_OPUS_4_LATEST,R.CLAUDE_OPUS_4_20250514];class Me{static{c(this,"Chat")}system;messages=[];tools=[];setToolSchemas(e){this.tools=e}addSystem(e){this.system=e}addUser(e,s,r){let o,n=[];if(typeof s=="string"?(o=s,n=r||[]):Array.isArray(s)&&(n=s),!o&&n.length===0){this.messages.push({role:"user",content:e});return}const i=[{type:"text",text:e}];o&&i.push({type:"instructions",instructions:o});for(const a of n)i.push({type:"file",file:a});this.messages.push({role:"user",content:i})}addAssistant(e,s){this.messages.push({role:"assistant",content:e,...s&&{toolCalls:s}})}addTools(e){this.messages.push({role:"tool",content:e})}hasFiles(){return this.messages.some(e=>Array.isArray(e.content)&&e.content.some(s=>s.type==="file"))}latest(){return this.messages[this.messages.length-1]}toString(){return JSON.stringify({system:this.system,messages:this.messages,tools:this.tools})}}function U(t,e=`
|
|
2
|
-
|
|
3
|
-
`){if(typeof t=="string")return t;const s=t.filter(o=>o.type==="text").map(o=>o.text),r=t.filter(o=>o.type==="instructions").map(o=>o.instructions);return s.length===0&&r.length===0?null:[...s,...r].join(e)}c(U,"getTextAndInstructions");function Ge(t){return typeof t=="string"?t:t.filter(e=>e.type==="text").map(e=>e.text).join(`
|
|
4
|
-
|
|
5
|
-
`)}c(Ge,"getTextContent");function ke(t){if(typeof t=="string")return null;const e=t.filter(s=>s.type==="instructions").map(s=>s.instructions);return e.length>0?e.join(`
|
|
6
|
-
|
|
7
|
-
`):null}c(ke,"getInstructions");function D(t){return typeof t=="string"?[]:t.filter(e=>e.type==="file"&&e.file.type==="document").map(e=>e.file)}c(D,"getDocuments");function x(t){return typeof t=="string"?[]:t.filter(e=>e.type==="file"&&e.file.type==="image").map(e=>e.file)}c(x,"getImages");var y=(t=>(t[t.Stop=0]="Stop",t[t.Length=1]="Length",t[t.FunctionCall=2]="FunctionCall",t[t.Error=3]="Error",t))(y||{});const $e=R.CLAUDE_SONNET_4_LATEST;class xe{static{c(this,"AnthropicProvider")}name="Anthropic";client;model;constructor(e,s){this.model=s??$e,this.client=new ye({apiKey:e})}createChatRequest(e,s={}){const{recorder:r}=s;return e.hasFiles()&&!se.includes(this.model)&&r?.warn?.log(`Model ${this.model} may not support multimodal content. Use one of: ${se.join(", ")}`),new Le(this,e)}}class Le{static{c(this,"AnthropicChatRequest")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:r,model:o}=this.provider,n={model:o,max_tokens:4096,...Fe(this.chat)};s?.debug?.log(n);let i;try{const a=await r.messages.create(n);i=Ce(a)}catch(a){i={type:"error",error:{type:a.error.error.type??"Undetermined",message:a.error.error.message??"Unexpected error from Anthropic"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function ne(t){switch(t){case"max_tokens":return y.Length;case"end_turn":return y.Stop;case"stop_sequence":return y.Stop;case"tool_use":return y.FunctionCall;default:return y.Error}}c(ne,"getStopReason$2");function Fe(t){const e=t.messages.map(r=>{if(r.role==="assistant"){const o=[];return o.push({type:"text",text:r.content}),r.toolCalls&&o.push(...r.toolCalls.map(n=>({type:"tool_use",id:n.id,name:n.name,input:n.arguments}))),{role:"assistant",content:o}}if(r.role==="tool")return{role:"user",content:r.content.map(o=>({type:"tool_result",tool_use_id:o.id,content:o.content}))};if(typeof r.content=="string")return{role:"user",content:r.content};{const o=[],n=U(r.content);n&&o.push({type:"text",text:n});const i=x(r.content);i.length>0&&o.push(...i.map(u=>({type:"image",source:{type:"base64",media_type:u.mimeType,data:u.base64}})));const a=D(r.content);return a.length>0&&o.push(...a.filter(u=>u.mimeType==="application/pdf").map(u=>({type:"document",source:{type:"base64",media_type:"application/pdf",data:u.base64}}))),{role:"user",content:o}}}),s=t.tools.map(r=>({name:r.name,description:r.description,input_schema:r.parameters}));return{system:t.system,messages:e,tools:s}}c(Fe,"prepareRequest$4");function Ce(t){const e=ne(t.stop_reason);if(e===y.Error)return{type:"error",error:{type:"Uncaught error",message:"Stop reason is not recognized."},usage:{in:t.usage.input_tokens,out:t.usage.output_tokens},raw:t};if(e===y.FunctionCall){const s=t.content[0],r=s.type==="text"?s.text:"",o=t.content.slice(1).map(n=>{if(n.type==="tool_use")return{id:n.id,name:n.name,arguments:n.input}}).filter(n=>n!==null);return{type:"success",id:t.id,model:t.model,reason:y.FunctionCall,message:{role:t.role,content:r,toolCalls:o},usage:{in:t.usage.input_tokens,out:t.usage.output_tokens},raw:t}}if(t.type=="message"){const s=t.content[0];if(s.type=="text")return{type:"success",id:t.id,model:t.model,reason:ne(t.stop_reason),message:{role:t.role,content:s.text},usage:{in:t.usage.input_tokens,out:t.usage.output_tokens},raw:t}}}c(Ce,"translate");const _={GEMINI_1_0_PRO_VISION_LATEST:"gemini-1.0-pro-vision-latest",GEMINI_PRO_VISION:"gemini-pro-vision",GEMINI_1_5_PRO_LATEST:"gemini-1.5-pro-latest",GEMINI_1_5_PRO_001:"gemini-1.5-pro-001",GEMINI_1_5_PRO_002:"gemini-1.5-pro-002",GEMINI_1_5_PRO:"gemini-1.5-pro",GEMINI_1_5_FLASH_LATEST:"gemini-1.5-flash-latest",GEMINI_1_5_FLASH_001:"gemini-1.5-flash-001",GEMINI_1_5_FLASH_001_TUNING:"gemini-1.5-flash-001-tuning",GEMINI_1_5_FLASH:"gemini-1.5-flash",GEMINI_1_5_FLASH_002:"gemini-1.5-flash-002",GEMINI_1_5_FLASH_8B:"gemini-1.5-flash-8b",GEMINI_1_5_FLASH_8B_001:"gemini-1.5-flash-8b-001",GEMINI_1_5_FLASH_8B_LATEST:"gemini-1.5-flash-8b-latest",GEMINI_1_5_FLASH_8B_EXP_0827:"gemini-1.5-flash-8b-exp-0827",GEMINI_1_5_FLASH_8B_EXP_0924:"gemini-1.5-flash-8b-exp-0924",GEMINI_2_5_PRO_EXP_03_25:"gemini-2.5-pro-exp-03-25",GEMINI_2_5_PRO_PREVIEW_03_25:"gemini-2.5-pro-preview-03-25",GEMINI_2_5_FLASH_PREVIEW_04_17:"gemini-2.5-flash-preview-04-17",GEMINI_2_5_FLASH_PREVIEW_05_20:"gemini-2.5-flash-preview-05-20",GEMINI_2_5_FLASH_PREVIEW_04_17_THINKING:"gemini-2.5-flash-preview-04-17-thinking",GEMINI_2_5_PRO_PREVIEW_05_06:"gemini-2.5-pro-preview-05-06",GEMINI_2_5_PRO_PREVIEW_06_05:"gemini-2.5-pro-preview-06-05",GEMINI_2_0_FLASH_EXP:"gemini-2.0-flash-exp",GEMINI_2_0_FLASH:"gemini-2.0-flash",GEMINI_2_0_FLASH_001:"gemini-2.0-flash-001",GEMINI_2_0_FLASH_EXP_IMAGE_GENERATION:"gemini-2.0-flash-exp-image-generation",GEMINI_2_0_FLASH_LITE_001:"gemini-2.0-flash-lite-001",GEMINI_2_0_FLASH_LITE:"gemini-2.0-flash-lite",GEMINI_2_0_FLASH_PREVIEW_IMAGE_GENERATION:"gemini-2.0-flash-preview-image-generation",GEMINI_2_0_FLASH_LITE_PREVIEW_02_05:"gemini-2.0-flash-lite-preview-02-05",GEMINI_2_0_FLASH_LITE_PREVIEW:"gemini-2.0-flash-lite-preview",GEMINI_2_0_PRO_EXP:"gemini-2.0-pro-exp",GEMINI_2_0_PRO_EXP_02_05:"gemini-2.0-pro-exp-02-05",GEMINI_EXP_1206:"gemini-exp-1206",GEMINI_2_0_FLASH_THINKING_EXP_01_21:"gemini-2.0-flash-thinking-exp-01-21",GEMINI_2_0_FLASH_THINKING_EXP:"gemini-2.0-flash-thinking-exp",GEMINI_2_0_FLASH_THINKING_EXP_1219:"gemini-2.0-flash-thinking-exp-1219",GEMINI_2_5_FLASH_PREVIEW_TTS:"gemini-2.5-flash-preview-tts",GEMINI_2_5_PRO_PREVIEW_TTS:"gemini-2.5-pro-preview-tts",LEARNLM_2_0_FLASH_EXPERIMENTAL:"learnlm-2.0-flash-experimental",GEMMA_3_1B_IT:"gemma-3-1b-it",GEMMA_3_4B_IT:"gemma-3-4b-it",GEMMA_3_12B_IT:"gemma-3-12b-it",GEMMA_3_27B_IT:"gemma-3-27b-it",GEMMA_3N_E4B_IT:"gemma-3n-e4b-it"},re=[_.GEMINI_1_0_PRO_VISION_LATEST,_.GEMINI_PRO_VISION,_.GEMINI_1_5_PRO_LATEST,_.GEMINI_1_5_PRO_001,_.GEMINI_1_5_PRO_002,_.GEMINI_1_5_PRO,_.GEMINI_1_5_FLASH_LATEST,_.GEMINI_1_5_FLASH_001,_.GEMINI_1_5_FLASH_001_TUNING,_.GEMINI_1_5_FLASH,_.GEMINI_1_5_FLASH_002,_.GEMINI_1_5_FLASH_8B,_.GEMINI_1_5_FLASH_8B_001,_.GEMINI_1_5_FLASH_8B_LATEST,_.GEMINI_1_5_FLASH_8B_EXP_0827,_.GEMINI_1_5_FLASH_8B_EXP_0924,_.GEMINI_2_5_PRO_EXP_03_25,_.GEMINI_2_5_PRO_PREVIEW_03_25,_.GEMINI_2_5_FLASH_PREVIEW_04_17,_.GEMINI_2_5_FLASH_PREVIEW_05_20,_.GEMINI_2_5_FLASH_PREVIEW_04_17_THINKING,_.GEMINI_2_5_PRO_PREVIEW_05_06,_.GEMINI_2_5_PRO_PREVIEW_06_05,_.GEMINI_2_0_FLASH_EXP,_.GEMINI_2_0_FLASH,_.GEMINI_2_0_FLASH_001,_.GEMINI_2_0_FLASH_EXP_IMAGE_GENERATION,_.GEMINI_2_0_FLASH_LITE_001,_.GEMINI_2_0_FLASH_LITE,_.GEMINI_2_0_FLASH_PREVIEW_IMAGE_GENERATION,_.GEMINI_2_0_FLASH_LITE_PREVIEW_02_05,_.GEMINI_2_0_FLASH_LITE_PREVIEW,_.GEMINI_2_0_PRO_EXP,_.GEMINI_2_0_PRO_EXP_02_05,_.GEMINI_EXP_1206,_.GEMINI_2_0_FLASH_THINKING_EXP_01_21,_.GEMINI_2_0_FLASH_THINKING_EXP,_.GEMINI_2_0_FLASH_THINKING_EXP_1219,_.GEMINI_2_5_FLASH_PREVIEW_TTS,_.GEMINI_2_5_PRO_PREVIEW_TTS,_.LEARNLM_2_0_FLASH_EXPERIMENTAL,_.GEMMA_3_1B_IT,_.GEMMA_3_4B_IT,_.GEMMA_3_12B_IT,_.GEMMA_3_27B_IT,_.GEMMA_3N_E4B_IT],Ue=_.GEMINI_2_5_FLASH_PREVIEW_05_20;class De{static{c(this,"GoogleAIProvider")}name="GoogleAI";client;model;constructor(e,s){this.model=s??Ue,this.client=new Ee({apiKey:e})}createChatRequest(e,s={}){const{recorder:r}=s;return e.hasFiles()&&!re.includes(this.model)&&r?.warn.log(`Model ${this.model} does not support multimodal content. Use one of: ${re.join(", ")}`),new He(this,e)}}class He{static{c(this,"GoogleAIChatRequest")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:r,model:o}=this.provider,n=We(this.chat);s?.debug?.log(n);let i;try{const a=await r.models.generateContent({model:o,...n});i=Be(a,e)}catch(a){s?.error?.log(a),i={type:"error",error:{type:a.name??"Undetermined",message:a.message??"Unexpected error from Google AI"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function We(t){let e;t.messages.length===1&&t.messages[0].role=="user"&&typeof t.messages[0].content=="string"?e=t.messages[0].content:e=t.messages.map(r=>{if(r.role==="user"){if(typeof r.content=="string")return{role:"user",parts:[{text:r.content}]};{const o=[],n=U(r.content);n&&o.push({text:n});const i=x(r.content);i.length>0&&o.push(...i.map(u=>({inlineData:{mimeType:u.mimeType,data:u.base64}})));const a=D(r.content);return a.length>0&&o.push(...a.map(u=>({inlineData:{mimeType:u.mimeType,data:u.base64}}))),{role:"user",parts:o}}}else if(r.role==="assistant"){const o={role:"assistant",parts:[]};return r.content!==void 0&&o.parts.push({text:r.content}),r.toolCalls&&(o.parts=o.parts.concat(r.toolCalls.map(n=>{let i;return typeof n.arguments=="string"?i=JSON.parse(n.arguments):i=n.arguments,{functionCall:{id:n.id??void 0,name:n.name,args:i}}}))),o}else if(r.role==="tool")return{role:"user",parts:r.content.map(o=>({functionResponse:{id:o.id??void 0,name:o.name,response:{output:o.content}}}))}});const s={};return t.system&&(s.systemInstruction=t.system),t.tools.length>0&&(s.tools=t.tools.map(r=>({functionDeclarations:[{name:r.name,description:r.description,parameters:{...r.parameters,type:Ie.OBJECT}}]}))),{contents:e,config:s}}c(We,"prepareRequest$3");function Be(t,e){const{recorder:s}=e,r=t.usageMetadata.promptTokenCount,o=t.usageMetadata.totalTokenCount-r,n={in:r,out:o};if(!t)return{type:"error",error:{type:"InvalidResponse",message:"Invalid or empty response from Google AI"},usage:{in:0,out:0},raw:t};if(t.promptFeedback&&t.promptFeedback.blockReason)return{type:"error",error:{type:"Blocked",message:`Response blocked by Google AI: ${t.promptFeedback.blockReason}, ${t.promptFeedback.blockReasonMessage}`},usage:n,raw:t};if(!t.candidates||t.candidates.length===0)return{type:"error",error:{type:"InvalidResponse",message:"Invalid or empty response from Google AI"},usage:{in:0,out:0},raw:t};t.candidates.length>1&&s?.warn?.log(`We received ${t.candidates.length} response candidates`);const i=t.candidates[0],u=(i.content?.parts||[]).map(p=>p.text).filter(p=>p!==void 0).join(""),[l,f]=Xe(i.finishReason);if(l){let p;return t.functionCalls&&(p=t.functionCalls.map(d=>({id:d.id,name:d.name,arguments:JSON.stringify(d.args)}))),{type:"success",id:t.responseId,model:t.modelVersion,reason:t.functionCalls?y.FunctionCall:f,message:{role:"assistant",...u?{content:u}:{},...p?{toolCalls:p}:{}},usage:n,raw:t}}else return{type:"error",error:{type:"Undetermined",message:`Unexpected stop reason: ${f}`},usage:n,raw:t}}c(Be,"translateResponse$2");function Xe(t){switch(t){case w.STOP:return[!0,y.Stop];case w.MAX_TOKENS:return[!0,y.Length];case w.FINISH_REASON_UNSPECIFIED:case w.SAFETY:case w.RECITATION:case w.LANGUAGE:case w.OTHER:case w.BLOCKLIST:case w.PROHIBITED_CONTENT:case w.SPII:case w.MALFORMED_FUNCTION_CALL:case w.IMAGE_SAFETY:return[!1,y.Error]}}c(Xe,"getStopReason$1");const qe="http://localhost:11434";class Ke{static{c(this,"OllamaProvider")}name="Ollama";url;model;recorder;constructor(e,s){this.url=s||qe,this.model=e}createChatRequest(e,s={}){const{recorder:r}=s;return e.hasFiles()&&r?.warn?.log(`Ollama model ${this.model} multimodal support depends on the specific model. Ensure you're using a vision-capable model like llava.`),new Je(this.url,this.model,e)}}class Je{static{c(this,"OllamaChatCompletionRequest")}chat;url;model;constructor(e,s,r){this.url=e,this.model=s,this.chat=r}async execute(e){const{recorder:s}=e,r={stream:!1,options:{temperature:.7},...ze(this.chat,this.model)};s?.debug?.log(r);let o;try{const n=await fetch(`${this.url}/api/chat`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(r)});if(!n.ok)throw console.log(n),new Error(`HTTP error! status: ${n.status}`);const i=await n.json();o=Ye(i)}catch(n){s?.error?.log("Error fetching Ollama response:",n),o={type:"error",error:{type:"OllamaError",message:n.message||"Unexpected error from Ollama"},usage:{in:0,out:0},raw:JSON.stringify(n)}}return s?.debug?.log(o),o}}function ze(t,e){const s=[];t.system&&s.push({role:"system",content:t.system});let r;t.tools.length>0&&(r=t.tools.map(n=>({type:"function",function:n})));const o=t.messages.map(n=>{if(n.role==="tool")return n.content.map(i=>({role:"tool",tool_call_id:i.id,content:i.content}));if(n.role==="assistant"){const i=n.toolCalls?.map(a=>{const u=a.id;return{type:"function",function:{name:a.name,arguments:a.arguments},...u&&{id:u}}});return{role:n.role,content:n.content,...i&&{toolCalls:i}}}if(typeof n.content=="string")return{role:n.role,content:n.content};{const i=U(n.content),a=x(n.content).map(u=>u.base64);return{role:n.role,content:i,...a.length>0&&{images:a}}}}).flat(1);return{model:e,messages:[...s,...o],...r&&{tools:r}}}c(ze,"prepareRequest$2");function Ye(t){if(t.done_reason==="stop"&&t.message){const e=t.message.content,s=[];if(t.message.tool_calls)for(const o of t.message.tool_calls)s.push({id:o.id,name:o.function.name,arguments:o.function.arguments});const r=s.length>0;return{type:"success",id:`ollama-${Date.now()}`,model:t.model,reason:r?y.FunctionCall:y.Stop,message:{role:"assistant",content:e,...r&&{toolCalls:s}},usage:{in:t.prompt_eval_count||0,out:t.eval_count||0},raw:t}}return{type:"error",error:{type:"OllamaError",message:"Unexpected error from Ollama"},usage:{in:0,out:0},raw:t}}c(Ye,"translateResponse$1");const g={GPT_4_1:"gpt-4.1",GPT_4_1_2025_04_14:"gpt-4.1-2025-04-14",GPT_4_1_MINI:"gpt-4.1-mini",GPT_4_1_MINI_2025_04_14:"gpt-4.1-mini-2025-04-14",GPT_4_1_NANO:"gpt-4.1-nano",GPT_4_1_NANO_2025_04_14:"gpt-4.1-nano-2025-04-14",GPT_4O:"gpt-4o",GPT_4O_2024_05_13:"gpt-4o-2024-05-13",GPT_4O_2024_08_06:"gpt-4o-2024-08-06",GPT_4O_2024_11_20:"gpt-4o-2024-11-20",GPT_4O_AUDIO_PREVIEW:"gpt-4o-audio-preview",GPT_4O_AUDIO_PREVIEW_2024_10_01:"gpt-4o-audio-preview-2024-10-01",GPT_4O_AUDIO_PREVIEW_2024_12_17:"gpt-4o-audio-preview-2024-12-17",GPT_4O_AUDIO_PREVIEW_2025_06_03:"gpt-4o-audio-preview-2025-06-03",GPT_4O_MINI:"gpt-4o-mini",GPT_4O_MINI_2024_07_18:"gpt-4o-mini-2024-07-18",GPT_4O_MINI_AUDIO_PREVIEW:"gpt-4o-mini-audio-preview",GPT_4O_MINI_AUDIO_PREVIEW_2024_12_17:"gpt-4o-mini-audio-preview-2024-12-17",GPT_4O_MINI_REALTIME_PREVIEW:"gpt-4o-mini-realtime-preview",GPT_4O_MINI_REALTIME_PREVIEW_2024_12_17:"gpt-4o-mini-realtime-preview-2024-12-17",GPT_4O_MINI_SEARCH_PREVIEW:"gpt-4o-mini-search-preview",GPT_4O_MINI_SEARCH_PREVIEW_2025_03_11:"gpt-4o-mini-search-preview-2025-03-11",GPT_4O_MINI_TRANSCRIBE:"gpt-4o-mini-transcribe",GPT_4O_MINI_TTS:"gpt-4o-mini-tts",GPT_4O_REALTIME_PREVIEW:"gpt-4o-realtime-preview",GPT_4O_REALTIME_PREVIEW_2024_10_01:"gpt-4o-realtime-preview-2024-10-01",GPT_4O_REALTIME_PREVIEW_2024_12_17:"gpt-4o-realtime-preview-2024-12-17",GPT_4O_REALTIME_PREVIEW_2025_06_03:"gpt-4o-realtime-preview-2025-06-03",GPT_4O_SEARCH_PREVIEW:"gpt-4o-search-preview",GPT_4O_SEARCH_PREVIEW_2025_03_11:"gpt-4o-search-preview-2025-03-11",GPT_4O_TRANSCRIBE:"gpt-4o-transcribe",O3_MINI:"o3-mini",O3_MINI_2025_01_31:"o3-mini-2025-01-31",O4_MINI:"o4-mini",O4_MINI_2025_04_16:"o4-mini-2025-04-16"},Ze=[g.GPT_4_1,g.GPT_4_1_2025_04_14,g.GPT_4_1_MINI,g.GPT_4_1_MINI_2025_04_14,g.GPT_4_1_NANO,g.GPT_4_1_NANO_2025_04_14,g.GPT_4O,g.GPT_4O_2024_05_13,g.GPT_4O_2024_08_06,g.GPT_4O_2024_11_20,g.GPT_4O_AUDIO_PREVIEW,g.GPT_4O_AUDIO_PREVIEW_2024_10_01,g.GPT_4O_AUDIO_PREVIEW_2024_12_17,g.GPT_4O_AUDIO_PREVIEW_2025_06_03,g.GPT_4O_MINI,g.GPT_4O_MINI_2024_07_18,g.GPT_4O_MINI_AUDIO_PREVIEW,g.GPT_4O_MINI_AUDIO_PREVIEW_2024_12_17,g.GPT_4O_MINI_REALTIME_PREVIEW,g.GPT_4O_MINI_REALTIME_PREVIEW_2024_12_17,g.GPT_4O_MINI_SEARCH_PREVIEW,g.GPT_4O_MINI_SEARCH_PREVIEW_2025_03_11,g.GPT_4O_MINI_TRANSCRIBE,g.GPT_4O_MINI_TTS,g.GPT_4O_REALTIME_PREVIEW,g.GPT_4O_REALTIME_PREVIEW_2024_10_01,g.GPT_4O_REALTIME_PREVIEW_2024_12_17,g.GPT_4O_REALTIME_PREVIEW_2025_06_03,g.GPT_4O_SEARCH_PREVIEW,g.GPT_4O_SEARCH_PREVIEW_2025_03_11,g.GPT_4O_TRANSCRIBE,g.O3_MINI,g.O3_MINI_2025_01_31,g.O4_MINI,g.O4_MINI_2025_04_16];class Qe{static{c(this,"OpenAIChatCompletionRequest")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:r,model:o}=this.provider,n=Ve(this.chat,o);s?.debug?.heading.log("[Open AI Provider] Using the ChatCompletion API"),s?.debug?.log(n);let i;try{const a=await r.chat.completions.create(n);i=et(a)}catch(a){s?.error?.log(a),i={type:"error",error:{type:a.type??"Undetermined",message:a.message??"Unexpected error from OpenAI"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function je(t){switch(t){case"length":return y.Length;case"stop":return y.Stop;case"tool_calls":return y.FunctionCall;default:return y.Error}}c(je,"getStopReason");function Ve(t,e){const s=[];t.system&&s.push({role:"system",content:t.system});let r;t.tools.length>0&&(r=t.tools.map(n=>({type:"function",function:n})));const o=t.messages.map(n=>{if(n.role==="tool")return n.content.map(i=>({role:"tool",tool_call_id:i.id,content:i.content}));if(n.role==="assistant"){const i=n.toolCalls?.map(a=>{const u=a.id;return{type:"function",function:{name:a.name,arguments:typeof a.arguments=="string"?a.arguments:JSON.stringify(a.arguments)},...u&&{id:u}}});return{role:n.role,content:n.content,...i&&{toolCalls:i}}}if(typeof n.content=="string")return{role:n.role,content:n.content};{const i=[],a=U(n.content);a&&i.push({type:"text",text:a});const u=x(n.content);u.length>0&&i.push(...u.map(f=>({type:"image_url",image_url:{url:`data:${f.mimeType};base64,${f.base64}`}})));const l=D(n.content);return l.length>0&&i.push(...l.map(f=>({type:"file",file:{filename:f.name,file_data:`data:${f.mimeType};base64,${f.base64}`}}))),{role:n.role,content:i}}}).flat(1);return{model:e,messages:[...s,...o],...r&&{tools:r}}}c(Ve,"prepareRequest$1");function et(t){if(t.choices.length>0){const e=t.choices[0],s=e.message.tool_calls?.map(r=>({id:r.id,name:r.function.name,arguments:r.function.arguments}));return{type:"success",id:t.id,model:t.model,reason:je(e.finish_reason),message:{content:e.message.content??"",role:e.message.role,toolCalls:s},usage:{in:t.usage?.prompt_tokens??0,out:t.usage?.completion_tokens??0},raw:t}}return{type:"error",error:{type:"undetermined",message:"Unexpected response from OpenAI"},usage:{in:t.usage?.prompt_tokens??0,out:t.usage?.completion_tokens??0},raw:t}}c(et,"translateResponse");class tt{static{c(this,"OpenAIResponsesAPI")}constructor(e,s){this.provider=e,this.chat=s}async execute(e){const{recorder:s}=e,{client:r,model:o}=this.provider,n=st(this.chat,o);s?.debug?.heading.log("[Open AI Provider] Using the Responses API"),s?.debug?.log(n);let i;try{const a=await r.responses.create(n);i=nt(a)}catch(a){s?.error?.log(a),i={type:"error",error:{type:a.type??"Undetermined",message:a.message??"Unexpected error from OpenAI"},usage:{in:0,out:0},raw:a}}return s?.debug?.log(i),i}}function st(t,e){const s=t.messages.map(n=>{if(n.role==="tool")return n.content.map(i=>({type:"function_call_output",call_id:i.id,output:i.content}));if(n.role==="assistant"){const i=n.toolCalls?.map(a=>{const u=a.id;return{type:"function",function:{name:a.name,arguments:typeof a.arguments=="string"?a.arguments:JSON.stringify(a.arguments)},...u&&{id:u}}});return{role:n.role,content:n.content,...i&&{toolCalls:i}}}if(typeof n.content=="string")return{role:n.role,content:n.content};{const i=[],a=Ge(n.content);a&&i.push({type:"input_text",text:a});const u=x(n.content);u.length>0&&i.push(...u.map(f=>({type:"input_image",image_url:`data:${f.mimeType};base64,${f.base64}`})));const l=D(n.content);return l.length>0&&i.push(...l.map(f=>({type:"input_file",filename:f.path,file_data:`data:${f.mimeType};base64,${f.base64}`}))),{role:n.role,content:i}}}).flat(1),r={model:e,input:s},o=t.latest();if(o&&o.role==="user"){let n="";const i=ke(o.content);t.system&&(n=t.system),i&&(n=n?`${n}
|
|
8
|
-
|
|
9
|
-
${i}`:i),n&&(r.instructions=n)}return t.tools.length>0&&(r.tools=t.tools.map(n=>({type:"function",strict:!0,...n}))),r}c(st,"prepareRequest");function nt(t){if(t.error)return{type:"error",error:{type:t.error.code||"undetermined",message:t.error.message||"Response generation failed"},usage:{in:t.usage?.input_tokens??0,out:t.usage?.output_tokens??0},raw:t};const e=t.output?.filter(s=>s.type==="function_call")?.map(s=>({id:s.id||"",name:s.function?.name||"",arguments:s.function?.arguments||""}));return{type:"success",id:t.id,model:t.model||"",reason:t.incomplete_details?y.Error:y.Stop,message:{content:t.output_text||"",role:"assistant",...e?.length&&{toolCalls:e}},usage:{in:t.usage?.input_tokens??0,out:t.usage?.output_tokens??0},raw:t}}c(nt,"translateResponseToAIResponse");const rt=g.GPT_4_1;class ot{static{c(this,"OpenAIProvider")}name="OpenAI";client;model;constructor(e,s){this.model=s||rt,this.client=new we({apiKey:e})}createChatRequest(e,s={}){const{recorder:r}=s;return Ze.includes(this.model)?new tt(this,e):new Qe(this,e)}}function it(t,e){if(!e||Object.keys(e).length===0)throw new E(`The provider ${t} is not configured. Please check your configuration.`);switch(t){case"openai":return new ot(e["api-key"],e.model);case"anthropic":return new xe(e["api-key"],e.model);case"googleai":return new De(e["api-key"],e.model);case"ollama":{const s=e;return new Ke(s.model,s.url)}default:throw new E("The provider is unsupported")}}c(it,"getProvider");class j extends E{static{c(this,"TaskError")}constructor(e,s){super(e,{code:"TASK_ERROR",id:s?.id,details:{taskType:s?.taskType,taskIndex:s?.taskIndex,...s?.details},cause:s?.cause}),Object.setPrototypeOf(this,j.prototype)}}const I={Running:"running",Success:"success",PartialSuccess:"partialSuccess",Fail:"fail"};var A=(t=>(t[t.Trace=10]="Trace",t[t.Debug=20]="Debug",t[t.Info=30]="Info",t[t.Warn=40]="Warn",t[t.Error=50]="Error",t[t.Fatal=60]="Fatal",t))(A||{});class at{static{c(this,"Recorder")}instanceId=crypto.randomUUID();currentLevel=A.Info;logs=[];writers=[];_debug;_info;_warn;_error;constructor(){this.buildMethods()}buildMethods(){this._debug=A.Debug>=this.currentLevel?this.createLoggingFunction(A.Debug):null,this._info=A.Info>=this.currentLevel?this.createLoggingFunction(A.Info):null,this._warn=A.Warn>=this.currentLevel?this.createLoggingFunction(A.Warn):null,this._error=this.createLoggingFunction(A.Error)}set level(e){this.currentLevel=e,this.buildMethods()}get level(){return this.currentLevel}get info(){return this._info}get warn(){return this._warn}get error(){return this._error}get debug(){return this._debug}subscribe(e){this.writers.includes(e)||this.writers.push(e)}unsubscribe(e){const s=this.writers.indexOf(e);s!==-1&&this.writers.splice(s,1)}publish(e){this.logs.push(e);for(const s of this.writers)s.handleEvent(e)}logFunction(e,s,...r){let o=r.map(n=>typeof n=="string"?{message:n}:n instanceof Error?Te(n):n);this.publish({level:e,time:Date.now(),kind:s,payload:o})}createLoggingFunction(e){return{log:this.logFunction.bind(this,e,"body"),heading:{log:this.logFunction.bind(this,e,"heading")}}}getLogs(e=A.Info){return this.logs.filter(s=>s.level>=e)}async shutdown(){for(const e of this.writers)typeof e.flush=="function"&&await e.flush()}}async function ct({path:t,defaults:e,loader:s="File"}){let r=null,o="";if(t)try{o=K(t),r=await C(o,{encoding:"utf-8"})}catch{throw new Error(`${s} not found, see --help for details`)}else{for(const n of e.formats)try{o=K(e.name+"."+n),r=await C(o,{encoding:"utf-8"});break}catch{continue}if(r===null)throw new Error(`${s} not found, see --help for details`)}return{content:r,format:o.split(".").pop()??""}}c(ct,"loadFile");async function ut(t,e){let s="";for(const r of t){const o=await ee(r);e?.debug?.log(`many-files parser. For glob "${r}", found ${o.length} files.`);const n=await Promise.all(o.map(async i=>{const a=await C(i,"utf-8");return i+`:
|
|
10
|
-
`+a}));s+=n.join(`
|
|
11
|
-
`)}return s}c(ut,"loadManyFiles");function lt(t,e){t=t.replace("**/*","**");const s=/(?<asterisks>\*{1,2})(?<extension>\.[^\\/]+)?/,r=t.match(s);if(r){let o="";return r.groups?.asterisks.length==1?o+=e.stem:o+=e.dir+e.stem,r.groups?.extension?o+=r.groups.extension:o+=e.ext,t.replace(r[0],o)}return t}c(lt,"replaceFilePattern");function ft(t){const e=/(?<name>[^\\/]+)(?<extension>\.[^\\/]+)$/,s=t.match(e);return s&&s.length>0&&s.groups?{abs:t,dir:t.replace(s[0],""),ext:s.groups.extension,stem:s.groups.name,name:s[0]}:null}c(ft,"pathToComponents");async function oe(t){const e=be(t);try{await te(e)}catch{await Re(e),await oe(e)}}c(oe,"ensureDirectoryExistence");async function _t({filePath:t,content:e}){await oe(t),await Oe(t,e)}c(_t,"writeFileWithDirectories");const ie=[".jpg",".jpeg",".png",".gif",".webp",".bmp",".tiff"],ae=[".pdf"],ce=20*1024*1024;async function J(t){const e=K(t);try{await te(e)}catch{throw new Error(`File not found: ${t}`)}const s=await Ne(e);if(s.size>ce)throw new Error(`File too large: ${s.size} bytes. Maximum allowed: ${ce} bytes`);const r=Se(e).toLowerCase();let o,n;if(ie.includes(r))switch(o="image",r){case".jpg":case".jpeg":n="image/jpeg";break;case".png":n="image/png";break;case".gif":n="image/gif";break;case".webp":n="image/webp";break;case".bmp":n="image/bmp";break;case".tiff":n="image/tiff";break;default:n="image/jpeg"}else if(ae.includes(r))o="document",n="application/pdf";else throw new Error(`Unsupported file type: ${r}. Supported types: ${[...ie,...ae].join(", ")}`);const a=(await C(e)).toString("base64");return{path:e,base64:a,mimeType:n,size:s.size,name:e.split("/").pop()||"",type:o}}c(J,"loadFileAsBase64");function pt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):gt(t.using,e)?!t.jobs||typeof t.jobs!="object"?(e&&(e.value="Missing or invalid 'jobs' property"),!1):ue(t.jobs,e)?!0:(e&&(e.value=`Invalid 'jobs' property: ${e?.value}`),!1):(e&&(e.value=`Invalid 'using' property: ${e?.value}`),!1)}c(pt,"isJobConfig");function gt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(typeof t.engine!="string")return e&&(e.value="Missing or invalid 'engine' property"),!1;if(!["openai","anthropic","ollama","googleai"].includes(t.engine))return e&&(e.value="Invalid provider type. Must be 'openai', 'anthropic', 'googleai', or 'ollama'"),!1;switch(t.engine){case"ollama":if("model"in t&&typeof t.model!="string")return e&&(e.value="Property 'model' must be a string"),!1;if("url"in t&&typeof t.url!="string")return e&&(e.value="Property 'url' must be a string"),!1;break;case"googleai":case"anthropic":case"openai":if("api-key"in t&&typeof t["api-key"]!="string")return e&&(e.value="Property 'api-key' must be a string"),!1;if("model"in t&&typeof t.model!="string")return e&&(e.value="Property 'model' must be a string"),!1;break}return!0}c(gt,"isUsing");function ue(t,e){for(const[s,r]of Object.entries(t))if(!dt(r,e))return e&&(e.value=`Invalid job '${s}': ${e?.value}`),!1;return!0}c(ue,"isDAGJob");function dt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(!mt(t,e))return!1;if("dependsOn"in t&&t.dependsOn!==void 0){const s=t.dependsOn;if(typeof s!="string")if(Array.isArray(s)){for(let r=0;r<s.length;r++)if(typeof s[r]!="string")return e&&(e.value=`Dependency at index ${r} must be a string`),!1}else return e&&(e.value="Property 'dependsOn' must be a string or array of strings"),!1}return!0}c(dt,"isDAGJobValue");function mt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):"batch"in t?yt(t,e):ht(t,e)}c(mt,"isJob");function ht(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if("batch"in t)return e&&(e.value="Serial job should not have a batch property"),!1;if(t.tools!==void 0){if(!Array.isArray(t.tools))return e&&(e.value="Property 'tools' must be an array"),!1;for(const s of t.tools)if(typeof s!="string")return e&&(e.value="All tools must be strings"),!1}if(!Array.isArray(t.steps))return e&&(e.value="Property 'steps' must be an array"),!1;for(let s=0;s<t.steps.length;s++)if(!le(t.steps[s],e))return e&&(e.value=`Invalid step at index ${s}: ${e?.value}`),!1;return!0}c(ht,"isSerialJob");function yt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(t.tools!==void 0){if(!Array.isArray(t.tools))return e&&(e.value="Property 'tools' must be an array"),!1;for(const s of t.tools)if(typeof s!="string")return e&&(e.value="All tools must be strings"),!1}if(!Array.isArray(t.batch))return e&&(e.value="Property 'batch' must be an array"),!1;for(let s=0;s<t.batch.length;s++){const r=t.batch[s];if(!r||typeof r!="object")return e&&(e.value=`Batch item at index ${s} must be an object`),!1;if(r.type!=="files")return e&&(e.value=`Batch item at index ${s} must have type 'files'`),!1;if(typeof r.source!="string")return e&&(e.value=`Batch item at index ${s} must have a string 'source' property`),!1;if(typeof r.bind!="string")return e&&(e.value=`Batch item at index ${s} must have a string 'bind' property`),!1;if(r["skip-if"]!==void 0){if(!Array.isArray(r["skip-if"]))return e&&(e.value=`Batch item at index ${s} must have an array 'skip-if' property`),!1;for(let o=0;o<r["skip-if"].length;o++)if(!Et(r["skip-if"][o],e))return e&&(e.value=`Invalid skip condition at index ${o} in batch item ${s}: ${e?.value}`),!1}}if(!Array.isArray(t.steps))return e&&(e.value="Property 'steps' must be an array"),!1;for(let s=0;s<t.steps.length;s++)if(!le(t.steps[s],e))return e&&(e.value=`Invalid step at index ${s}: ${e?.value}`),!1;return!0}c(yt,"isBatchJob");function Et(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):t.type!=="file-exist"?(e&&(e.value="Property 'type' must be 'file-exist'"),!1):typeof t.pattern!="string"?(e&&(e.value="Property 'pattern' must be a string"),!1):!0}c(Et,"isSkipOptions");function le(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):!t.uses||typeof t.uses!="string"?(e&&(e.value="Step must have a string 'uses' property"),!1):t.uses==="chat"?It(t,e):t.uses==="write-to-disk"?wt(t,e):(e&&(e.value=`Unknown uses type: ${t.uses}`),!1)}c(le,"isStep");function It(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(t.uses!=="chat")return e&&(e.value="Uses must be 'chat'"),!1;if(typeof t.message!="string")return e&&(e.value="Property 'message' must be a string"),!1;if(t.output!==void 0){if(!t.output||typeof t.output!="object"||Array.isArray(t.output))return e&&(e.value="Property 'output' must be an object"),!1;const s=["string","string[]","number","boolean"];for(const[r,o]of Object.entries(t.output))if(typeof r!="string"||typeof o!="string"||!s.includes(o))return e&&(e.value="Property 'output' must be a Record<string, ResTypeStrings> where ResTypeStrings is 'string' | 'string[]' | 'number' | 'boolean'"),!1}if(t.system!==void 0&&typeof t.system!="string")return e&&(e.value="Property 'system' must be a string"),!1;if(t.replace!==void 0){if(!Array.isArray(t.replace))return e&&(e.value="Property 'replace' must be an array"),!1;for(let s=0;s<t.replace.length;s++)if(!Tt(t.replace[s],e))return e&&(e.value=`Invalid replace at index ${s}: ${e?.value}`),!1}if(t.tools!==void 0){if(!Array.isArray(t.tools))return e&&(e.value="Property 'tools' must be an array"),!1;for(const s of t.tools)if(typeof s!="string")return e&&(e.value="All tools must be strings"),!1}if(t.images!==void 0){if(!Array.isArray(t.images))return e&&(e.value="Property 'images' must be an array"),!1;for(let s=0;s<t.images.length;s++)if(!At(t.images[s],e))return e&&(e.value=`Invalid image at index ${s}: ${e?.value}`),!1}if(t.documents!==void 0){if(!Array.isArray(t.documents))return e&&(e.value="Property 'documents' must be an array"),!1;for(let s=0;s<t.documents.length;s++)if(!Pt(t.documents[s],e))return e&&(e.value=`Invalid document at index ${s}: ${e?.value}`),!1}return!0}c(It,"isChatStep");function wt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):t.uses!=="write-to-disk"?(e&&(e.value="Uses must be 'write-to-disk'"),!1):typeof t.output!="string"?(e&&(e.value="Property 'output' must be a string"),!1):!0}c(wt,"isWriteToDiskStep");function Tt(t,e){if(!t||typeof t!="object")return e&&(e.value="Not an object"),!1;if(typeof t.pattern!="string")return e&&(e.value="Property 'pattern' must be a string"),!1;if(t.source!=="file")return e&&(e.value="Property 'source' must be 'file'"),!1;if(typeof t.files!="string"&&!Array.isArray(t.files))return e&&(e.value="Property 'files' must be a string or an array of strings"),!1;if(Array.isArray(t.files)){for(let s=0;s<t.files.length;s++)if(typeof t.files[s]!="string")return e&&(e.value=`Files entry at index ${s} must be a string`),!1}return!0}c(Tt,"isReplace");function At(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):typeof t.file!="string"?(e&&(e.value="Property 'file' must be a string"),!1):!0}c(At,"isImageReference");function Pt(t,e){return!t||typeof t!="object"?(e&&(e.value="Not an object"),!1):typeof t.file!="string"?(e&&(e.value="Property 'file' must be a string"),!1):!0}c(Pt,"isDocumentReference");class vt{static{c(this,"FileRunPlanner")}constructor(e,s,r=[]){this.source=e,this.bind=s,this.skipConditions=r}async plan(e){const s=[],r=await ee(this.source,{withFileTypes:!0});for(const o of r){const n=o.fullpath(),i=ft(n);let a=!1;for(const u of this.skipConditions)if(a=await u.eval({components:i}),a)break;if(!a){const u=await Ae(n,"utf-8"),l={variables:{[this.bind]:u,...i},tasks:e};s.push(l)}}return s}}class Nt{static{c(this,"MultiPlanner")}planners;constructor(e){this.planners=e}async plan(e){const s=this.planners.map(async o=>await o.plan(e));return(await Promise.all(s)).flat()}}function z(t,e,s="{{}}"){const r=s==="{{}}"?/\{\{(.*?)\}\}/g:/\{(.*?)\}/g;return t=t.replace(r,(o,n)=>{if(Object.prototype.hasOwnProperty.call(e,n)){const i=e[n];return i==null?"":String(i)}return o}),t}c(z,"replaceVariables");class Ot{static{c(this,"FileExistSkipCondition")}constructor(e){this.pattern=e}type="file-exist";async eval(e){const s=z(this.pattern,e.components,"{}");try{return await Pe(s,ve.F_OK),!0}catch{return!1}}}var N=(t=>(t.String="string",t.List="string[]",t.Number="number",t.Boolean="boolean",t))(N||{});const fe={response:N.String};class _e{static{c(this,"AbstractInstruct")}type="instruct";_result=void 0;prompt;system=null;inputs={};tools={};files=[];resFormat;rawResponse;finalPrompt;constructor(e,s){this.prompt=e,this.resFormat=s}setInputs(e){this.inputs=e}addInput(e,s){this.inputs[e]=s}addTools(e){for(const s of e)this.tools[s.name]=s}addTool(e){this.tools[e.name]=e}addImage(e){if(e.type!=="image")throw new Error(`Expected image file, got ${e.type}`);this.files.push(e)}addFile(e){this.files.push(e)}hasTools(){return Object.keys(this.tools).length>0}hasFiles(){return this.files.length>0}get result(){return this._result}compile(e,s={}){const r=this.getFinalUserPrompt(e,s),o=this.getFormatInstructions();return{message:r,instructions:o}}getFinalUserPrompt(e,s={}){const{recorder:r,options:o}=s,n={...e,...this.inputs};let i=z(this.prompt,n);if(o?.warnUnused){const a=i.match(/\{\{(.*?)\}\}/g);if(a)throw r?.error.log(`Warning unused variables ${a.join(", ")}`),new Error(`Unused variables: ${a.join(", ")}`)}return i}getFormatInstructions(){let e="";for(const[s,r]of Object.entries(this.resFormat))switch(r){case N.String:e+=`Use <${s}></${s}> to indicate the answer for ${s}. The answer must be a string.
|
|
12
|
-
`;break;case N.Number:e+=`Use <${s}></${s}> to indicate the answer for ${s}. the answer must be a number.
|
|
13
|
-
`;break;case N.Boolean:e+=`Use <${s}></${s}> to indicate the answer for ${s}. The answer must be a true/false.
|
|
14
|
-
`;break;case N.List:e+=`Use <${s}></${s}> to indicate the answer for ${s}. The answer must be a list of strings. Each string should be in a new line.
|
|
15
|
-
`;break}return e}finalize(e,s){this.rawResponse=e;const r={},o=Object.keys(this.resFormat);if(o.length===0){if(e.trim()==="{}"||e.trim()==="")return{};throw new Error("Output format is empty, but rawValue is not an empty object representation or empty string.")}s=s||this.parseTaggedSections(e);for(const n of o){const i=n;let a;const u=s.tags[i];if(u)a=u;else throw new Error(`Expected results with tag ${i} but it does not exist`);const l=this.resFormat[n];try{const f=this.typeResponses(l,a);r[n]=f}catch(f){throw new Error(`Cannot convert value of key ${i} to ${l}: ${f.message}`)}}return this._result=r,r}parseTaggedSections(e){const s=/<(\w+)>(.*?)<\/\1>/gs,r={};let o=e;return o=o.replace(s,(n,i,a)=>(r[i]=a,"")),{tags:r,remaining:o.trim()}}typeResponses(e,s){let r;switch(e){case N.String:r=s;break;case N.Number:if(r=parseFloat(s),isNaN(r))throw new Error(`Cannot parse '${s}' as number. Expected a numeric string.`);break;case N.Boolean:const o=s.toLowerCase();if(o==="true")r=!0;else if(o==="false")r=!1;else throw new Error(`Cannot parse '${s}' as boolean. Expected 'true' or 'false'.`);break;case N.List:s===""?r=[]:r=s.split(`
|
|
16
|
-
`).map(n=>n.trim()).filter(n=>n.length>0);break}return r}}class k extends _e{static{c(this,"Instruct")}constructor(e,s){super(e,s)}static with(e,s){return s?new k(e,s):new k(e,fe)}}function L(t){return Array.isArray(t)?t:[t]}c(L,"arrayify");function P(t,e){return e?`${e}:${t.slice(0,8)}`:t.slice(0,8)}c(P,"friendly");function Rt(t){return new Promise(e=>setTimeout(e,t))}c(Rt,"delay");const St={name:"brave",description:"Perform a search using the Brave search engine",parameters:{type:"object",properties:{searchTerm:{type:"string",description:"The search term to query"}},required:["searchTerm"]}};class bt{static{c(this,"BraveSearchTool")}name="brave";schema=St;apiKey;throttle;lastExecTime=0;constructor(e){e&&this.setConfig(e)}setConfig(e){const{rateLimit:s}=e;this.apiKey=e["api-key"],this.throttle=s?1100/s:void 0}async execute(e,s={}){const{searchTerm:r}=e,{recorder:o}=s;if(o?.debug?.heading.log(`Brave: searching for ${r}`),this.throttle){for(;Date.now()-this.lastExecTime<this.throttle;)await Rt(this.throttle-(Date.now()-this.lastExecTime));this.lastExecTime=Date.now()}try{const n=this.apiKey,i="https://api.search.brave.com/res/v1/web/search",a=new URL(i);a.searchParams.append("q",r),a.searchParams.append("format","json");const u=await fetch(a.toString(),{method:"GET",headers:{Accept:"application/json","X-Subscription-Token":n}});if(!u.ok)throw new Error(`[Brave] HTTP error ${u.status}: ${u.statusText}`);return await u.json()}catch(n){throw o?.error.log("[Brave] Error fetching search results:",n),n}}}const Mt=new bt,Gt={name:"calculator",description:"Performs basic arithmetic operations",parameters:{type:"object",properties:{operation:{type:"string",description:"The operation to perform (add, subtract, multiply, divide)",enum:["add","subtract","multiply","divide"]},a:{type:"number",description:"First operand"},b:{type:"number",description:"Second operand"}},required:["operation","a","b"]}},kt={name:"calculator",schema:Gt,execute:c(async t=>{const{operation:e,a:s,b:r}=t;switch(e){case"add":return`${s} + ${r} = ${s+r}`;case"subtract":return`${s} - ${r} = ${s-r}`;case"multiply":return`${s} * ${r} = ${s*r}`;case"divide":if(r===0)throw new Error("Cannot divide by zero");return`${s} / ${r} = ${s/r}`;default:throw new Error(`Unknown operation: ${e}`)}},"execute")};class $t{static{c(this,"ToolRegistry")}executables={};config;setConfig(e){this.config=e}register(e){if(this.executables[e.name])throw new Error(`Tool with name '${e.name}' is already registered`);this.executables[e.name]=e}get(e){const s=this.executables[e];if(!s)throw new Error(`Tool '${e}' is not registered`);return s.setConfig?.(this.config[e]),s}}let F;function pe(){return F||(F=new $t,F.register(kt),F.register(Mt)),F}c(pe,"getToolRegistry");const xt={async convert(t,e){const{recorder:s,toolNames:r}=e,{message:o,system:n,replace:i}=t;let a;t.output?a=k.with(o,t.output):a=k.with(o),n&&(a.system=n);const u=[...new Set([...r??[],...t.tools??[]])];for(const l of u){const f=pe().get(l);a.addTool(f)}if(i){for(const l of i)if(l.source==="file"){const f=L(l.files),p=await ut(f,s);a.addInput(l.pattern,p)}}if(t.images)for(const l of t.images)try{const f=await J(l.file);a.addFile(f)}catch(f){throw new Error(`Failed to load image '${l.file}': ${f.message}`)}if(t.documents)for(const l of t.documents)try{const f=await J(l.file);a.addFile(f)}catch(f){throw new Error(`Failed to load document '${l.file}': ${f.message}`)}return a}};class Lt{static{c(this,"StepToClassRegistry")}converters=new Map;get(e){const s=this.converters.get(e);if(!s)throw new Error(`No converter registered for step: ${e}`);return s}register(e,s){this.converters.set(e,s)}}class Y{static{c(this,"WriteOutputTask")}constructor(e,s=["response"]){this.output=e,this.keys=s}type="write-to-disk"}const Ft={async convert(t){if(t.keys){const e=L(t.keys);return new Y(t.output,e)}return new Y(t.output)}},Z=new Lt;Z.register("write-to-disk",Ft),Z.register("chat",xt);async function H(t,e){const{recorder:s}=e,r=t.tools??void 0,o=t.steps.map(async n=>(n.uses,await Z.get(n.uses).convert(n,{recorder:s,toolNames:r})));return Promise.all(o)}c(H,"configToTasks");async function ge(t,e){const{batch:s}=t;return s.length===1?de(s[0]):new Nt(s.map(r=>de(r)))}c(ge,"configToPlanner");function de(t){switch(t.type){case"files":let e;return t["skip-if"]&&(e=t["skip-if"].map(r=>Ct(r))),new vt(t.source,t.bind,e)}}c(de,"batchOptionsToPlanner");function Ct(t){switch(t.type){case"file-exist":return new Ot(t.pattern)}}c(Ct,"skipOptionsToSkipConditions");function Ut(t){return t.success===!1&&t.error!==void 0}c(Ut,"isErrorResult");function W(t,e){return{response:t,stats:e,success:!0}}c(W,"createResult");function B(t,e,s){return{response:e,stats:s,error:t,success:!1}}c(B,"createErrorResult");class Dt{static{c(this,"WriteToDiskTaskHandler")}taskType="write-to-disk";canHandle(e){return e&&typeof e=="object"&&"type"in e&&e.type==="write-to-disk"}async execute(e){const{task:s,variables:r,options:o={},recorder:n}=e,i=s.output,a=s.keys??[];if(o?.warnUnused){const f=a.filter(p=>!(p in r));f.length>0&&n?.warn?.log(`[Write To Disk] The following keys were not found in the variables: ${f.join(", ")}`)}let u="";if(a.length===1?u=r[a[0]]??"<not found>":u=a.map(f=>`[${f}]:
|
|
17
|
-
${r[f]??"<not found>"}
|
|
18
|
-
`).join(`
|
|
19
|
-
`),o?.dryRun){n?.info?.log("[Dry run] Write to Disk is not executed.");return}let l="";i.includes("*")?l=lt(i,r.file):l=z(i,r,"{}"),await _t({filePath:l,content:u})}}var X=(t=>(t.LastResult="lastResult",t))(X||{});function Ht(t,e,s){const{options:r,recorder:o}=s,n=r?.warnUnused??!0;for(const[i,a]of Object.entries(t))n&&e[i]&&o?.warn?.log(`Warning: Variable "${i}" is being overwritten. Previous value: ${e[i]}, new value: ${a}`),e[i]=a}c(Ht,"setResultsIntoVariables");class Wt{static{c(this,"ChatTaskHandler")}taskType="instruct";canHandle(e){return e&&typeof e=="object"&&"type"in e&&e.type==="instruct"}async execute(e){const{task:s,...r}=e;await Bt({instruct:s,...r})}}async function Bt(t){const{instruct:e,chat:s,provider:r,stats:o,variables:n,options:i,recorder:a}=t;e.system&&s.addSystem(e.system);const{message:u,instructions:l}=e.compile(n,{recorder:a,options:i});if(e.hasFiles()?s.addUser(u,l,e.files):s.addUser(u,l),e.hasTools()){const p=qt(e.tools);s.setToolSchemas(p)}if(i?.dryRun)return a?.debug?.log(s),{action:"complete"};let f=!0;for(;f;){const d=await r.createChatRequest(s,{recorder:a}).execute({recorder:a});if(o.in+=d.usage.in,o.out+=d.usage.out,d.type==="error")throw new Error(JSON.stringify(d.error));if(d.type==="success")switch(d.reason){case y.Stop:{if(d.message.content){const h=d.message.content;s.addAssistant(h);const m=e.finalize(h);Ht(m,n,{options:i,recorder:a}),n[X.LastResult]=m}return f=!1,{action:"continue"}}case y.Length:throw new Error("Incomplete model output due to `max_tokens` parameter or token limit");case y.FunctionCall:{let h=d.message;if(d.message&&s.addAssistant(h.content,h.toolCalls),h.toolCalls&&h.toolCalls.length>0){const m=await Xt(h.toolCalls,e,{recorder:a});a?.debug?.log(m),s.addTools(m),f=!0}else f=!1;break}}if(d.type!=="success")throw a?.debug?.log(d),new Error("Unexpected response type")}return{action:"continue"}}c(Bt,"executeChatAction");async function Xt(t,e,s={}){const{recorder:r}=s,o=[];for(const n of t)o.push(new Promise((i,a)=>{const u=e.tools[n.name];if(!u){a(`Tool not found: ${n.name}`);return}r?.debug?.heading.log(`Executing tool ${u.name}`);let l={};try{l=typeof n.arguments=="string"?JSON.parse(n.arguments):n.arguments}catch{a(`argument for tool ${n.name} is not valid: ${JSON.stringify(n.arguments)}`)}u.execute(l).then(f=>{r?.debug?.log(`Complete tool ${u.name}: ${n.id}`),i({id:n.id,name:n.name,content:JSON.stringify(f)})}).catch(a)}));return Promise.all(o)}c(Xt,"executeToolCalls");function qt(t){const e=[];for(const[s,r]of Object.entries(t))e.push(r.schema);return e}c(qt,"getToolSchemas");class Kt{static{c(this,"TaskRegistry")}handlers=new Map;register(e){this.handlers.set(e.taskType,e)}getHandler(e){return this.handlers.get(e.type)}hasHandler(e){return this.handlers.has(e.type)}async executeTask(e){const{task:s}=e,r=s.type,o=this.getHandler(s);if(!o)throw new Error(`No handler registered for action type: ${r}`);if(!o.canHandle(s))throw new Error(`Handler found but action does not match expected format: ${r}`);await o.execute(e)}}function Jt(){const t=new Kt;return t.register(new Wt),t}c(Jt,"createBaseRegistry");function zt(){const t=Jt();return t.register(new Dt),t}c(zt,"createNodeRegistry");const Q=c((t,...e)=>{const s=c(async o=>{const{recorder:n}=o;let i=[];return"steps"in t?i=await H(t,{recorder:n}):i=[t,...e],i},"prepare");return{execute:c(async o=>{const{provider:n,variables:i,options:a,stats:u,recorder:l,name:f}=o,p=crypto.randomUUID(),d=zt();l?.info?.log({type:"task",id:p,status:I.Running,message:`[${P(p,f)}] Starting job`});try{const h=await s({recorder:l}),m=new Me;for(const[S,T]of h.entries()){l?.info?.log({type:"task",id:p,status:I.Running,message:`[${P(p,f)}] Processing step ${S+1}: ${T.type}`});try{await d.executeTask({task:T,chat:m,provider:n,variables:i,options:a,stats:u,recorder:l})}catch(v){throw v instanceof E?v:new j(`Error executing task ${T.type}`,{id:p,taskType:T.type,taskIndex:S,cause:v instanceof Error?v:new Error(String(v))})}}return l?.info?.log({type:"task",status:I.Success,id:p,message:`[${P(p,f)}] Completed ${h.length} steps`}),W(i[X.LastResult],u)}catch(h){const m=h instanceof E?h:new E("Serial workflow execution failed",{id:p,cause:h instanceof Error?h:new Error(String(h))});return l?.info?.log({type:"task",status:I.Fail,id:p,message:`[${P(p,f)}] Failed: ${m.message}`}),l?.error.log(m),B(m,i[X.LastResult],u)}},"execute")}},"serialWorkflow"),me=c((t,...e)=>{const s=c(async o=>{const{recorder:n}=o;let i=[],a=null;if("batch"in t){const u=t;a=await ge(u),i=await H(u,{recorder:n})}else a=t,i=[...e];return[a,i]},"prepare");return{execute:c(async o=>{const{provider:n,variables:i,options:a,stats:u,recorder:l,name:f}=o,p=crypto.randomUUID();try{const[d,h]=await s({recorder:l}),m=await d.plan(h);if(l?.debug?.heading.log("Runs",m),m.length===0)return l?.info?.log("No runs to execute"),W([],u);let S=0;l?.info?.log({type:"task",status:I.Running,id:p,message:`[${P(p,"CRW")}] Working on 0/${m.length}`});const T=c(async(O,q)=>{try{return await Q(...O.tasks).execute({provider:n,variables:{...O.variables,...i},options:a,stats:u,recorder:l,name:`${f}-${q}`})}catch(b){const V=b instanceof E?b:new E("Error executing run",{cause:b instanceof Error?b:new Error(String(b))});return l?.error?.log(V),B(V,null,u)}finally{S++,l?.info?.log({type:"task",status:I.Running,id:p,message:`[${P(p,"CRW")}] Working on ${S}/${m.length}`})}},"executeRun"),v=5;let G=[];for(let O=0;O<m.length;O+=v){const q=m.slice(O,O+v),b=await Promise.all(q.map(T));G=G.concat(b)}const M=G.some(Ut);l?.info?.log({type:"task",status:M?I.PartialSuccess:I.Success,id:p,message:`[${P(p,"CRW")}] All jobs (${m.length}) completed${M?" with some errors":""}`});const $=G.map(O=>O.response);return W($,u)}catch(d){const h=d instanceof E?d:new E("Concurrent workflow execution failed",{id:p,cause:d instanceof Error?d:new Error(String(d))});return l?.error?.log(h),B(h,null,u)}},"execute")}},"concurrentWorkflow");class Yt{static{c(this,"DAGParser")}static parse(e){const s=new Map;for(const[o,n]of Object.entries(e)){const i=this.parseNodeDefinition(o,n);s.set(o,i)}return this.validateDependencies(s),this.checkForCycles(s),{stages:this.createExecutionStages(s),nodes:s}}static parseNodeDefinition(e,s){if(this.isSimpleTask(s))return{id:e,tasks:Array.isArray(s)?s:[s],dependencies:[],executionType:"serial"};if(this.isConcurrentNodeDefinition(s)){const i=s,a=i.dependsOn?L(i.dependsOn):[];return{id:e,tasks:i.tasks,dependencies:a,planner:i.planner,executionType:"concurrent"}}const r=s,o=r.dependsOn?L(r.dependsOn):[],n=L(r.task);return{id:e,tasks:n,dependencies:o,executionType:"serial"}}static isSimpleTask(e){return e.type||Array.isArray(e)}static isConcurrentNodeDefinition(e){return e&&typeof e=="object"&&"planner"in e}static validateDependencies(e){for(const s of e.values())for(const r of s.dependencies)if(!e.has(r))throw new E(`Node "${s.id}" depends on non-existent node "${r}"`)}static checkForCycles(e){const s=new Set,r=new Set,o=c(n=>{if(r.has(n))return!0;if(s.has(n))return!1;s.add(n),r.add(n);const i=e.get(n);for(const a of i.dependencies)if(o(a))return!0;return r.delete(n),!1},"hasCycle");for(const n of e.keys())if(o(n))throw new E(`Circular dependency detected involving node "${n}"`)}static createExecutionStages(e){const s=[],r=new Set,o=new Set(e.keys());for(;o.size>0;){const n=[];for(const i of o)e.get(i).dependencies.every(l=>r.has(l))&&n.push(i);if(n.length===0)throw new E("Unable to resolve DAG dependencies - possible circular reference");s.push(n),n.forEach(i=>{r.add(i),o.delete(i)})}return s}}class Zt{static{c(this,"DAGJobToDefinition")}static async convert(e,s){const{recorder:r}=s,o={};for(const[n,i]of Object.entries(e)){const{dependsOn:a,...u}=i;if("batch"in u){const l=u,f=await ge(l),p=await H(l,{recorder:r}),d={planner:f,tasks:p,...a?{dependsOn:a}:{}};o[n]=d}else{const l=await H(u,{recorder:r});if(a){const f={task:l,dependsOn:a};o[n]=f}else o[n]=l}}return o}}async function Qt(t,e,s,r={}){const{variables:o}=s,n=e.nodes.get(t);try{let i;if(n.executionType==="concurrent"&&n.planner?i=await me(n.planner,...n.tasks).execute({...s,variables:o,name:t}):i=await Q(...n.tasks).execute({...s,variables:o,name:t}),!i.success)throw new E(`Node "${t}" failed: ${i.error?.message}`);return i.response}catch(i){if(!r.continueOnError)throw i;return null}}c(Qt,"executeNode");const jt=c((t,e={})=>{const s=c(async(o,n)=>{const{recorder:i}=n,a={value:""};return ue(o,a)?await Zt.convert(o,n):(i?.warn?.log(a),o)},"prepare");return{execute:c(async o=>{const{stats:n,recorder:i}=o,{maxConcurrency:a=3}=e,u=crypto.randomUUID();try{const l=await s(t,{recorder:i});i?.debug?.log(l);const f=Yt.parse(l),p=new Map;i?.info?.log({type:"task",id:u,status:I.Running,message:`[${P(u)}] Starting workflow execution with ${f.stages.length} stages`});for(const[h,m]of f.stages.entries()){i?.info?.log({type:"task",id:u,status:I.Running,message:`[${P(u)}] Stage ${h+1}/${f.stages.length}, executing ${m.length} nodes: ${m.join(", ")}`});const S=Math.min(m.length,a);for(let T=0;T<m.length;T+=S){const v=m.slice(T,T+S);(await Promise.all(v.map(async M=>{const $=await Qt(M,f,o,e);return{nodeId:M,result:$}}))).forEach(({nodeId:M,result:$})=>{p.set(M,$)})}}i?.info?.log({type:"task",status:I.Success,id:u,message:`[${P(u)}] Workflow execution completed successfully`});const d=Object.fromEntries(p);return W(d,n)}catch(l){const f=l instanceof E?l:new E("DAG workflow execution failed",{id:u,cause:l instanceof Error?l:new Error(String(l))});return i?.info?.log({type:"task",status:I.Fail,id:u,message:`[${P(u)}] Workflow execution failed: ${f.message}`}),i?.error?.log(f),B(f,null,n)}},"execute")}},"dagWorkflow");export{E as A,fe as D,k as I,A as L,at as R,I as T,Y as W,_e as a,ct as b,me as c,jt as d,pe as e,it as g,pt as i,J as l,Q as s};
|