@fifthrevision/axle-cli 0.19.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Chong Han Chua
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/cli.d.ts

@@ -0,0 +1 @@
+export { };

package/dist/cli.js

@@ -0,0 +1,6 @@
+#!/usr/bin/env node
+import{t as e}from"./memory-BmDsikXS.js";import{t}from"./store-BFxiiKeG.js";import{a as n,i as r,n as i,r as a,t as o}from"./tools-Cp5kiYmz.js";import{Command as s}from"@commander-js/extra-typings";import{Agent as c,Anthropic as l,Gemini as u,Instruct as d,MCP as f,OpenAI as p,SimpleWriter as m,Tracer as h,addStats as g,anthropic as _,chatCompletions as v,createAgentConfig as y,createStats as ee,gemini as te,loadFileContent as b,openai as x}from"@fifthrevision/axle";import{createHash as ne}from"node:crypto";import{z as S}from"zod";import{appendFile as re,mkdir as ie,readFile as C}from"node:fs/promises";import{basename as ae,dirname as oe,extname as w}from"node:path";import{config as se}from"dotenv";import T from"yaml";import{glob as ce}from"glob";import{createInterface as le}from"node:readline";var ue=`0.19.0`;async function de(e,t){let n=[];for(let r of e){let e=new f(r);await e.connect({tracer:t}),n.push(e)}return n}async function fe(e,t){for(let n of e)try{await n.close({tracer:t})}catch{}}function pe(e,t){switch(e){case`calculator`:return n;case`exec`:{let e=t?.exec;return e&&r.configure(e),r}case`patch-file`:return a;case`read-file`:return i;case`write-file`:return o;default:throw Error(`Unknown tool: ${e}`)}}function E(e,t){return e.map(e=>pe(e,t))}function D(e,t){let n=e.config??{};switch(e.type){case`openai`:{let e={...t.openai,...n},r=O(e);if(!r)throw Error(`The provider openai is not configured. Please check your configuration.`);return{provider:x(r),model:e.model||p.DefaultModel}}case`anthropic`:{let e={...t.anthropic,...n},r=O(e);if(!r)throw Error(`The provider anthropic is not configured. Please check your configuration.`);return{provider:_(r),model:e.model||l.DefaultModel}}case`gemini`:{let e={...t.gemini,...n},r=O(e);if(!r)throw Error(`The provider gemini is not configured. Please check your configuration.`);return{provider:te(r),model:e.model||u.DefaultModel}}case`chatcompletions`:{let e={...t.chatcompletions,...n},r=e[`base-url`],i=e.model;if(!r||!i)throw Error(`The provider chatcompletions is not configured. Please check your configuration.`);return{provider:v(r,O(e)),model:i}}default:throw Error(`Unknown provider type: ${e.type}`)}}function O(e){let t=e.apiKeyEnv??e[`api-key-env`];return typeof t==`string`&&t.length>0?process.env[t]:e[`api-key`]}function k(e){let{type:t,...n}=e.provider,r=Object.keys(n).length>0?{type:t,config:n}:{type:t};return{version:1,name:e.name,provider:r,tools:e.tools?.map(e=>({name:e})),providerTools:e.provider_tools?.map(e=>({name:e})),mcps:e.mcps}}async function A(n,r,i){let a=k(n),o=a.mcps?.length?await de(a.mcps,i):[],s=await y(a,e=>{let t=D(e.provider,r);return{provider:t.provider,model:t.model,tools:e.tools?.length?E(e.tools.map(e=>e.name)):void 0,mcps:o.length>0?o:void 0}});return{agentConfig:{...s,memory:new e({provider:s.provider,model:s.model,store:new t(`.axle`)})},mcps:o}}S.object({value:S.string()}),S.object({timeout:S.number().optional(),maxBuffer:S.number().optional(),cwd:S.string().optional()});const j={"api-key":S.string().optional(),apiKeyEnv:S.string().optional(),"api-key-env":S.string().optional()},M=S.strictObject({type:S.literal(`chatcompletions`),"base-url":S.string().optional(),model:S.string().optional(),...j}),N=S.strictObject({type:S.literal(`anthropic`),model:S.string().optional(),...j}),P=S.strictObject({type:S.literal(`openai`),model:S.string().optional(),...j}),F=S.strictObject({type:S.literal(`gemini`),model:S.string().optional(),...j}),I=S.discriminatedUnion(`type`,[M,N,P,F]),L=S.object({transport:S.literal(`stdio`),name:S.string().optional(),command:S.string(),args:S.array(S.string()).optional(),env:S.record(S.string(),S.string()).optional()}),R=S.object({transport:S.literal(`http`),name:S.string().optional(),url:S.string(),headers:S.record(S.string(),S.string()).optional()}),z=S.discriminatedUnion(`transport`,[L,R]),B=S.object({files:S.string(),resume:S.boolean().default(!1),concurrency:S.number().int().positive().default(3)}),me=S.object({name:S.string().optional(),provider:I,task:S.string(),tools:S.array(S.string()).optional(),provider_tools:S.array(S.string()).optional(),files:S.array(S.string()).optional(),mcps:S.array(z).optional(),batch:B.optional()});async function he(e,t){let{tracer:n}=t,r=w(e).slice(1);if(r!==`yaml`&&r!==`yml`)throw Error(`Invalid job file format. Expected .yaml or .yml`);let i;try{i=await C(e,{encoding:`utf-8`})}catch{throw Error(`Job File not found, see --help for details`)}let a=T.parse(i);n?.debug(`Job config: `+JSON.stringify(a,null,2));let o=me.safeParse(a);if(!o.success)throw Error(`The job file is not valid:\n${be(o.error)}`);return o.data.name||(o.data.name=ae(e,w(e))),o.data}async function ge(e){let{tracer:t}=e;se({quiet:!0});let n=_e();return t?.debug(`Service config: `+JSON.stringify(V(n),null,2)),n}function _e(){return ve({openai:process.env.OPENAI_API_KEY?{"api-key":process.env.OPENAI_API_KEY,model:process.env.OPENAI_MODEL}:void 0,anthropic:process.env.ANTHROPIC_API_KEY?{"api-key":process.env.ANTHROPIC_API_KEY,model:process.env.ANTHROPIC_MODEL}:void 0,gemini:process.env.GEMINI_API_KEY?{"api-key":process.env.GEMINI_API_KEY,model:process.env.GEMINI_MODEL}:void 0,chatcompletions:process.env.CHATCOMPLETIONS_BASE_URL&&process.env.CHATCOMPLETIONS_MODEL?{"base-url":process.env.CHATCOMPLETIONS_BASE_URL,model:process.env.CHATCOMPLETIONS_MODEL,"api-key":process.env.CHATCOMPLETIONS_API_KEY}:void 0})}function ve(e){return Object.fromEntries(Object.entries(e).filter(([,e])=>e&&Object.keys(e).length>0))}function V(e){return Array.isArray(e)?e.map(e=>V(e)):!e||typeof e!=`object`?e:Object.fromEntries(Object.entries(e).map(([e,t])=>[e,ye(e)&&t?`[redacted]`:V(t)]))}function ye(e){return e===`api-key`||e.toLowerCase().includes(`secret`)}function be(e){return e.issues.map(e=>`  - ${e.path.join(`.`)||`root`}: ${e.message}`).join(`
+`)}const H=`.axle/batch.jsonl`;function xe(e,t){let n=ne(`sha256`);return n.update(e),n.update(`\0`),n.update(t),n.digest(`hex`)}async function Se(e=H){let t=new Map,n;try{n=await C(e,`utf-8`)}catch{return t}for(let e of n.split(`
+`)){let n=e.trim();if(n)try{let e=JSON.parse(n);e.file&&e.hash&&t.set(e.file,e)}catch{}}return t}async function Ce(e,t=H){await ie(oe(t),{recursive:!0}),await re(t,JSON.stringify(e)+`
+`,`utf-8`)}async function we(e,t,n,r,i,a){let o=new d({prompt:e.task});if(e.files)for(let t of e.files)o.addFile(await b(t));let s=a.startSpan(`job`,{type:`workflow`}),l=new c({...t,tracer:s});try{let e=await l.send(o.withInputs(n)).final;if(g(i,e.usage),e.response){let t=e.response;a.info(t,{markdown:!0})}r.interactive&&await Te(l,i,a),s.end()}catch(e){let t=e instanceof Error?e.message:String(e);throw s.error(t),s.end(`error`),e}}async function Te(e,t,n){let r=le({input:process.stdin,output:process.stdout});r.on(`SIGINT`,()=>{r.close()});let i=e=>new Promise(t=>{r.question(e,t),r.once(`close`,()=>t(null))});try{for(;;){let r=await i(`
+> `);if(r===null||r.trim()===``)break;try{let i=await e.send(r.trim()).final;if(g(t,i.usage),i.response){let e=i.response;n.info(e,{markdown:!0})}}catch(e){let t=e instanceof Error?e.message:String(e);n.error(t)}}}finally{r.close()}}async function Ee(e,t,n,r,i,a,o){let s=await ce(t.files);if(s.length===0){o.warn(`No files matched pattern: ${t.files}`);return}o.info(`Batch: ${s.length} file(s) matched "${t.files}"`);let l=t.resume?await Se():new Map,u=e.files?await Promise.all(e.files.map(e=>b(e))):[],f=0,p=0,m=0;await De(t.concurrency??3,s,async i=>{let s=o.startSpan(`batch:${i}`,{type:`workflow`});try{let o=await C(i),m=xe(e.task,o),h=l.get(i);if(t.resume&&h&&h.hash===m){s.info(`Skipped (already completed)`),s.end(),p++;return}let _=new d({prompt:e.task});for(let e of u)_.addFile(e);_.addFile(await b(i));let v={...r,file:i};g(a,(await new c({...n,tracer:s}).send(_.withInputs(v)).final).usage),await Ce({file:i,hash:m,timestamp:Date.now()}),s.end(),f++}catch(e){let t=e instanceof Error?e.message:String(e);s.error(`Failed: ${t}`),s.end(`error`),m++}}),o.info(`Batch complete: ${f} completed, ${p} skipped, ${m} failed`)}async function De(e,t,n){let r=0;async function i(){for(;r<t.length;)await n(t[r++])}let a=Array.from({length:Math.min(e,t.length)},()=>i());await Promise.all(a)}const U=new s().name(`axle`).description(`Axle is a CLI tool for running AI workflows`).version(ue).requiredOption(`-j, --job <path>`,`Path to the YAML job file`).option(`--no-log`,`Do not write the output to a log file`).option(`-d, --debug`,`Print additional debug information`).option(`-i, --interactive`,`Continue the conversation interactively after the initial task`).option(`--args <args...>`,`Additional arguments in the form key=value`);U.parse(process.argv);const W=U.opts(),G={date:new Date().toISOString().split(`T`)[0],datetime:new Date().toISOString(),cwd:process.cwd()};W.args&&W.args.forEach(e=>{let[t,n]=e.split(`=`);t&&n&&(G[t.trim()]=n.trim())});const K=new h;W.debug&&(K.minLevel=`debug`);const Oe=new m({minLevel:W.debug?`debug`:`info`,showInternal:W.debug,showTimestamp:!0,markdown:!0});if(K.addWriter(Oe),W.log){let e=new m({minLevel:`debug`,showInternal:!0,showTimestamp:!0,output:e=>{}});K.addWriter(e)}const q=K.startSpan(`cli`,{type:`root`});process.on(`uncaughtException`,async e=>{console.error(`Uncaught exception:`),console.error(e),q.error(`Uncaught exception:`),q.error(e.message),q.error(e.stack||``),q.end(`error`),await K.flush(),process.exit(1)}),W.debug&&(q.debug(`Options: `+JSON.stringify(W,null,2)),q.debug(`Additional Arguments: `+JSON.stringify(G,null,2)));let J,Y;try{J=await ge({tracer:q}),Y=await he(W.job,{tracer:q})}catch(e){let t=e instanceof Error?e:Error(String(e));q.error(t.message),q.debug(t.stack??``),q.end(`error`),await K.flush(),U.outputHelp(),process.exit(1)}let X=[],Z;try{let e=await A(Y,J,q);Z=e.agentConfig,X=e.mcps}catch(e){let t=e instanceof Error?e:Error(String(e));q.error(t.message),q.error(t.stack??``),q.end(`error`),await K.flush(),U.outputHelp(),process.exit(1)}if(!Z)throw Error(`Failed to create agent config.`);q.info(`All systems operational. Running job...`);const Q=ee(),ke=performance.now(),$={task:Y.task,files:Y.files};try{Y.batch?await Ee($,Y.batch,Z,G,W,Q,q):await we($,Z,G,W,Q,q)}catch(e){let t=e instanceof Error?e:Error(String(e));q.error(t.message),q.debug(t.stack??``)}finally{X.length>0&&await fe(X,q)}const Ae=performance.now()-ke;q.info(`Total run time: ${Math.round(Ae)}ms`),q.info(`Input tokens: ${Q.in}`),q.info(`Output tokens: ${Q.out}`),Q.cachedIn!==void 0&&q.info(`Cached input tokens: ${Q.cachedIn}`),Q.cacheWriteIn!==void 0&&q.info(`Cache write input tokens: ${Q.cacheWriteIn}`),Q.reasoningOut!==void 0&&q.info(`Reasoning output tokens: ${Q.reasoningOut}`),q.info(`Complete. Goodbye`),q.end(),await K.flush();export{};

package/dist/memory/index.d.ts

@@ -0,0 +1,30 @@
+import { AIProvider, AgentMemory, ExecutableTool, FileStore, MemoryContext, RecallResult } from "@fifthrevision/axle";
+
+//#region src/memory/ProceduralMemory.d.ts
+interface ProceduralMemoryConfig {
+  provider: AIProvider;
+  model: string;
+  enableTools?: boolean;
+  store: FileStore;
+  /** Memory-owned partition metadata, such as user or workspace ids. */
+  scope?: Record<string, string>;
+}
+declare class ProceduralMemory implements AgentMemory {
+  private provider;
+  private model;
+  private enableTools;
+  private store;
+  private scope?;
+  private lastAgentName?;
+  constructor(config: ProceduralMemoryConfig);
+  recall(context: MemoryContext): Promise<RecallResult>;
+  record(context: MemoryContext): Promise<void>;
+  tools(): ExecutableTool[];
+  private formatMessages;
+  private parseInstructions;
+  private getStorePath;
+  private loadStore;
+  private saveStore;
+}
+//#endregion
+export { ProceduralMemory, type ProceduralMemoryConfig };

package/dist/memory/index.js

@@ -0,0 +1 @@
+import{t as e}from"../memory-BmDsikXS.js";export{e as ProceduralMemory};

package/dist/memory-BmDsikXS.js

@@ -0,0 +1,20 @@
+import{generate as e}from"@fifthrevision/axle";import t from"node:crypto";import{z as n}from"zod";var r=class{provider;model;enableTools;store;scope;lastAgentName;constructor(e){this.provider=e.provider,this.model=e.model,this.enableTools=e.enableTools??!1,this.store=e.store,this.scope=e.scope}async recall(e){let t=e.tracer?.startSpan(`memory.recall`,{type:`internal`});this.lastAgentName=e.agentName;let n=await this.loadStore(e.agentName);if(n.instructions.length===0)return t?.info(`no stored instructions`),t?.end(),{};t?.info(`loaded instructions`,{count:n.instructions.length});let r=n.instructions.map((e,t)=>`${t+1}. ${e}`).join(`
+`);return t?.end(),{systemSuffix:`## Learned Instructions\n\n${r}`}}async record(t){if(!t.newMessages||t.newMessages.length===0)return;let n=t.tracer?.startSpan(`memory.record`,{type:`internal`}),r=this.formatMessages(t.newMessages);if(!r.trim()){n?.info(`no text content to extract from`),n?.end();return}let a=n?.startSpan(`memory.extract`,{type:`llm`}),o=await e({provider:this.provider,model:this.model,messages:[{role:`user`,content:r}],system:`You are a memory extraction system. Your job is to extract learnings from a conversation that should be remembered for future runs.
+
+Only extract:
+- Explicit user corrections (e.g., "No, always use bullet points")
+- Stated preferences (e.g., "I prefer concise summaries")
+- Patterns that clearly emerged from user feedback
+
+Do NOT extract:
+- General knowledge or facts from the conversation content
+- Inferences or speculation about what the user might want
+- Task-specific details that won't apply to future runs
+
+Respond with a JSON array of instruction strings. Each instruction should be a clear, actionable directive.
+If there are no learnings to extract, respond with an empty array: []
+
+Example response:
+["Always use bullet points for lists", "Keep summaries under 3 sentences"]`,tracer:a});if(!o.ok){n?.warn(`extraction failed`,{error:o.error}),n?.end();return}let s=i(o.final.content);if(!s){n?.end();return}let c=this.parseInstructions(s);if(c.length===0){n?.info(`no instructions extracted`),n?.end();return}let l=await this.loadStore(t.agentName);l.instructions.push(...c),await this.saveStore(t.agentName,l),n?.info(`saved instructions`,{count:c.length}),n?.end()}tools(){if(!this.enableTools)return[];let e=this;return[{name:`add_instruction`,description:`Save a learned instruction for future runs. Use this when the user explicitly asks you to remember something.`,schema:n.object({instruction:n.string().describe(`The instruction to remember`)}),async execute(t){if(!e.lastAgentName)return`Error: memory not initialized (no recall has been called yet)`;let n=await e.loadStore(e.lastAgentName);return n.instructions.push(t.instruction),await e.saveStore(e.lastAgentName,n),`Instruction saved: "${t.instruction}"`}}]}formatMessages(e){let t=[];for(let n of e)if(n.role===`user`){let e=typeof n.content==`string`?n.content:i(n.content);e&&t.push(`User: ${e}`)}else if(n.role===`assistant`){let e=i(n.content);e&&t.push(`Assistant: ${e}`)}return t.join(`
+
+`)}parseInstructions(e){let t=e.trim(),n=t.match(/^```(?:json)?\s*\n?([\s\S]*?)\n?\s*```$/);n&&(t=n[1].trim());try{let e=JSON.parse(t);if(Array.isArray(e)&&e.every(e=>typeof e==`string`))return e}catch{}return[]}getStorePath(e,n){let r=e??`default`,i=r;if(n&&Object.keys(n).length>0){let e=Object.entries(n).sort(([e],[t])=>e.localeCompare(t)).map(([e,t])=>`${e}=${t}`).join(`&`);i=`${r}-${t.createHash(`sha256`).update(e).digest(`hex`).slice(0,8)}`}return`memory/procedural/${i}.json`}async loadStore(e){let t=this.getStorePath(e,this.scope),n=await this.store.read(t);if(n)try{let e=JSON.parse(n);if(e&&Array.isArray(e.instructions))return{instructions:e.instructions}}catch{}return{instructions:[]}}async saveStore(e,t){if(!t)return;let n=this.getStorePath(e,this.scope);await this.store.write(n,JSON.stringify(t,null,2))}};function i(e){return typeof e==`string`?e:e.filter(e=>e.type===`text`||e.type===`thinking`).map(e=>e.text).join(``)}export{r as t};

package/dist/store/index.d.ts

@@ -0,0 +1,11 @@
+import { FileStore } from "@fifthrevision/axle";
+
+//#region src/store/LocalFileStore.d.ts
+declare class LocalFileStore implements FileStore {
+  readonly rootPath: string;
+  constructor(rootPath: string);
+  read(path: string): Promise<string | null>;
+  write(path: string, content: string): Promise<void>;
+}
+//#endregion
+export { LocalFileStore };

package/dist/store/index.js

@@ -0,0 +1 @@
+import{t as e}from"../store-BFxiiKeG.js";export{e as LocalFileStore};

package/dist/store-BFxiiKeG.js

@@ -0,0 +1 @@
+import e from"node:fs/promises";import t from"node:path";var n=class{rootPath;constructor(e){this.rootPath=e}async read(n){let r=t.join(this.rootPath,n);try{return await e.readFile(r,`utf-8`)}catch{return null}}async write(n,r){let i=t.join(this.rootPath,n);await e.mkdir(t.dirname(i),{recursive:!0}),await e.writeFile(i,r,`utf-8`)}};export{n as t};

package/dist/tools/index.d.ts

@@ -0,0 +1,71 @@
+import { ExecutableTool, ProviderTool, ToolContext, ToolDefinition } from "@fifthrevision/axle";
+import * as z$1 from "zod";
+import { z } from "zod";
+
+//#region src/tools/types.d.ts
+interface ExecProviderConfig {
+  timeout?: number;
+  maxBuffer?: number;
+  cwd?: string;
+}
+interface ToolProviderConfig {
+  exec?: ExecProviderConfig;
+}
+//#endregion
+//#region src/tools/calculator.d.ts
+declare const calculatorSchema: z.ZodObject<{
+  operation: z.ZodEnum<{
+    add: "add";
+    subtract: "subtract";
+    multiply: "multiply";
+    divide: "divide";
+  }>;
+  a: z.ZodNumber;
+  b: z.ZodNumber;
+}, z.core.$strip>;
+declare const calculatorTool: ExecutableTool<typeof calculatorSchema>;
+//#endregion
+//#region src/tools/exec/index.d.ts
+declare const execSchema: z$1.ZodObject<{
+  command: z$1.ZodString;
+}, z$1.core.$strip>;
+declare class ExecTool implements ExecutableTool<typeof execSchema> {
+  name: string;
+  description: string;
+  schema: z$1.ZodObject<{
+    command: z$1.ZodString;
+  }, z$1.core.$strip>;
+  private timeout;
+  private maxBuffer;
+  private cwd?;
+  constructor(config?: ExecProviderConfig);
+  configure(config: ExecProviderConfig): void;
+  summarize(params: z$1.infer<typeof execSchema>): string;
+  execute(params: z$1.infer<typeof execSchema>, ctx: ToolContext): Promise<string>;
+}
+declare const execTool: ExecTool;
+//#endregion
+//#region src/tools/patch-file.d.ts
+declare const patchFileSchema: z.ZodObject<{
+  path: z.ZodString;
+  old_string: z.ZodString;
+  new_string: z.ZodString;
+  start_line: z.ZodNumber;
+  end_line: z.ZodNumber;
+}, z.core.$strip>;
+declare const patchFileTool: ExecutableTool<typeof patchFileSchema>;
+//#endregion
+//#region src/tools/read-file.d.ts
+declare const readFileSchema: z.ZodObject<{
+  path: z.ZodString;
+}, z.core.$strip>;
+declare const readFileTool: ExecutableTool<typeof readFileSchema>;
+//#endregion
+//#region src/tools/write-file.d.ts
+declare const writeFileSchema: z.ZodObject<{
+  path: z.ZodString;
+  content: z.ZodString;
+}, z.core.$strip>;
+declare const writeFileTool: ExecutableTool<typeof writeFileSchema>;
+//#endregion
+export { type ExecProviderConfig, type ExecutableTool, type ProviderTool, type ToolContext, type ToolDefinition, type ToolProviderConfig, calculatorTool, execTool, patchFileTool, readFileTool, writeFileTool };

package/dist/tools/index.js

@@ -0,0 +1 @@
+import{a as e,i as t,n,r,t as i}from"../tools-Cp5kiYmz.js";export{e as calculatorTool,t as execTool,r as patchFileTool,n as readFileTool,i as writeFileTool};

package/dist/tools-Cp5kiYmz.js

@@ -0,0 +1,5 @@
+import*as e from"zod";import{z as t}from"zod";import{mkdir as n,readFile as r,writeFile as i}from"node:fs/promises";import{dirname as a}from"node:path";import{spawn as o}from"node:child_process";const s={name:`calculator`,description:`Performs basic arithmetic operations`,schema:t.object({operation:t.enum([`add`,`subtract`,`multiply`,`divide`]).describe(`The operation to perform (add, subtract, multiply, divide)`),a:t.number().describe(`First operand`),b:t.number().describe(`Second operand`)}),execute:async({operation:e,a:t,b:n})=>{switch(e){case`add`:return`${t} + ${n} = ${t+n}`;case`subtract`:return`${t} - ${n} = ${t-n}`;case`multiply`:return`${t} * ${n} = ${t*n}`;case`divide`:if(n===0)throw Error(`Cannot divide by zero`);return`${t} / ${n} = ${t/n}`;default:throw Error(`Unknown operation: ${e}`)}}};async function c(e,t={}){let n=t.timeout??3e4,r=t.maxBuffer??1048576;return new Promise((i,a)=>{let s=o(e,[],{shell:!0,cwd:t.cwd}),c=``,l=``,u=!1,d=!1,f=!1,p=setTimeout(()=>{u=!0,s.kill(`SIGTERM`)},n),m=()=>{d=!0,s.kill(`SIGTERM`)};t.signal?.addEventListener(`abort`,m);let h=()=>{clearTimeout(p),t.signal?.removeEventListener(`abort`,m)};s.stdout?.setEncoding(`utf-8`),s.stderr?.setEncoding(`utf-8`),s.stdout?.on(`data`,e=>{if(c+=e,c.length+l.length>r){f=!0,s.kill(`SIGTERM`);return}t.onChunk?.(e)}),s.stderr?.on(`data`,e=>{if(l+=e,c.length+l.length>r){f=!0,s.kill(`SIGTERM`);return}t.onChunk?.(e)}),s.on(`error`,e=>{h(),a(e)}),s.on(`close`,e=>{if(h(),u){let e=Error(`Command timed out after ${n}ms`);e.stdout=c,e.stderr=l,a(e);return}if(d){let e=Error(`Command aborted`);e.stdout=c,e.stderr=l,a(e);return}if(f){let e=Error(`Command output exceeded maxBuffer (${r} bytes)`);e.stdout=c,e.stderr=l,a(e);return}if(e!==0){let t=Error(`Command failed with exit code ${e}`);t.stdout=c,t.stderr=l,t.code=e??-1,a(t);return}i({stdout:c,stderr:l})})})}function l(e){if(e instanceof Error){let t=e,n=`Error executing command: ${e.message}`;return t.stdout&&(n+=`\n[stdout]: ${t.stdout}`),t.stderr&&(n+=`\n[stderr]: ${t.stderr}`),n}return`Error executing command: ${String(e)}`}function u(e,t){return t&&t.trim()?`${e}\n[stderr]: ${t}`:e}const d=e.object({command:e.string().describe(`The shell command to execute`)}),f=new class{name=`exec`;description=`Execute a shell command and return the output.`;schema=d;timeout=3e4;maxBuffer=1024*1024;cwd;constructor(e){e&&this.configure(e)}configure(e){this.timeout=e.timeout??3e4,this.maxBuffer=e.maxBuffer??1024*1024,this.cwd=e.cwd}summarize(e){return e.command}async execute(e,t){let{command:n}=e;try{let e=await c(n,{timeout:this.timeout,maxBuffer:this.maxBuffer,cwd:this.cwd,signal:t.signal,onChunk:e=>t.emit(e)});return u(e.stdout,e.stderr)}catch(e){return l(e)}}},p={name:`patch-file`,description:`Patch a file by replacing an exact string match within a specified line range`,schema:t.object({path:t.string().describe(`The file path to patch`),old_string:t.string().describe(`The exact text to find and replace`),new_string:t.string().describe(`The replacement text`),start_line:t.number().int().positive().describe(`1-indexed start line of the region to match within`),end_line:t.number().int().positive().describe(`1-indexed end line (inclusive) of the region to match within`)}),summarize:({path:e,start_line:t,end_line:n})=>`${e}:${t}:${n}`,execute:async({path:e,old_string:t,new_string:n,start_line:a,end_line:o})=>{if(o<a)throw Error(`end_line (${o}) must be >= start_line (${a})`);let s;try{s=await r(e,`utf-8`)}catch(t){throw t instanceof Error?Error(`Failed to read file "${e}": ${t.message}`):t}let c=s.split(`
+`);if(a>c.length)throw Error(`start_line (${a}) exceeds file length (${c.length} lines)`);if(o>c.length)throw Error(`end_line (${o}) exceeds file length (${c.length} lines)`);let l=c.slice(a-1,o).join(`
+`),u=l.indexOf(t);if(u===-1)throw Error(`old_string not found within lines ${a}-${o} of "${e}"`);if(l.indexOf(t,u+1)!==-1)throw Error(`old_string matches multiple times within lines ${a}-${o} of "${e}"`);let d=l.replace(t,n),f=[...c.slice(0,a-1),...d.split(`
+`),...c.slice(o)].join(`
+`);try{await i(e,f,`utf-8`)}catch(t){throw t instanceof Error?Error(`Failed to write file "${e}": ${t.message}`):t}return`Successfully patched "${e}" (lines ${a}-${o})`}},m={name:`read-file`,description:`Read the contents of a file from disk`,schema:t.object({path:t.string().describe(`The file path to read from`)}),summarize:({path:e})=>e,execute:async({path:e})=>{try{return await r(e,`utf-8`)}catch(t){throw t instanceof Error?Error(`Failed to read file "${e}": ${t.message}`):t}}},h={name:`write-file`,description:`Write content to a file on disk, creating directories if needed`,schema:t.object({path:t.string().describe(`The file path to write to`),content:t.string().describe(`The content to write to the file`)}),summarize:({path:e})=>e,execute:async({path:e,content:t})=>{try{return await n(a(e),{recursive:!0}),await i(e,t,`utf-8`),`Successfully wrote ${t.length} characters to "${e}"`}catch(t){throw t instanceof Error?Error(`Failed to write file "${e}": ${t.message}`):t}}};export{s as a,f as i,m as n,p as r,h as t};

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "@fifthrevision/axle-cli",
+  "version": "0.19.0",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/johncch/axle.git"
+  },
+  "description": "Axle command line workflow runner.",
+  "type": "module",
+  "bin": {
+    "axle": "dist/cli.js"
+  },
+  "exports": {
+    "./tools": {
+      "types": "./dist/tools/index.d.ts",
+      "import": "./dist/tools/index.js"
+    },
+    "./memory": {
+      "types": "./dist/memory/index.d.ts",
+      "import": "./dist/memory/index.js"
+    },
+    "./store": {
+      "types": "./dist/store/index.d.ts",
+      "import": "./dist/store/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "author": "Chong Han Chua",
+  "license": "ISC",
+  "dependencies": {
+    "@commander-js/extra-typings": "^14.0.0",
+    "commander": "^14.0.3",
+    "dotenv": "^17.4.2",
+    "glob": "^13.0.6",
+    "yaml": "^2.9.0",
+    "zod": "^4.4.3",
+    "@fifthrevision/axle": "0.19.0"
+  },
+  "devDependencies": {
+    "@arethetypeswrong/core": "^0.18.2",
+    "publint": "^0.3.21",
+    "tsdown": "^0.22.0",
+    "tsx": "^4.22.3",
+    "typescript": "^6.0.3"
+  },
+  "scripts": {
+    "start": "tsx ./src/cli.ts",
+    "build": "tsdown --clean --minify",
+    "build:watch": "tsdown --watch",
+    "build-dev": "tsdown"
+  }
+}