@zibby/agent-workflow 0.3.1 → 0.3.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +2 -0
- package/dist/graph-compiler.js +10 -10
- package/dist/graph.js +8 -8
- package/dist/index.js +9 -9
- package/dist/node.js +7 -7
- package/dist/timeline.js +7 -7
- package/package.json +1 -1
package/README.md
CHANGED
|
@@ -92,6 +92,8 @@ All workflow operations live under `zibby workflow <verb>` for consistency. The
|
|
|
92
92
|
|
|
93
93
|
The CLI also integrates with [Zibby Studio](https://zibby.dev) — a desktop UI for visualising live runs, pinning sessions, and stopping a workflow from a button.
|
|
94
94
|
|
|
95
|
+
> 📋 **Full CLI cheat sheet** including `zibby init`, `zibby template list/add`, `zibby memory remote/cost/pull/push` (UI agent memory + team sync), and `zibby test` is in [`@zibby/cli`'s README](https://www.npmjs.com/package/@zibby/cli). Workflow commands above are the engine-relevant subset.
|
|
96
|
+
|
|
95
97
|
---
|
|
96
98
|
|
|
97
99
|
## Use as a library
|
package/dist/graph-compiler.js
CHANGED
|
@@ -11,9 +11,9 @@ PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PRE
|
|
|
11
11
|
${m}
|
|
12
12
|
`),S.debug(`[workflow] prompt length: ${d.length} chars`),r.invoke(d,h)}var pe,R,J=V(()=>{Le();G();te();pe=Symbol.for("@zibby/agent-workflow.strategies");globalThis[pe]||(globalThis[pe]=[]);R=globalThis[pe]});var at=new Set(["__proto__","constructor","prototype"]);function ae(o){if(at.has(o))throw new Error(`Invalid state key: "${o}"`)}var q=class{constructor(e={}){this._state=Object.create(null),Object.assign(this._state,{messages:[],errors:[],artifacts:{},metadata:{},...e}),this._history=[]}get(e){return this._state[e]}set(e,t){ae(e),this._history.push({...this._state}),this._state[e]=t}update(e){let t=Object.getOwnPropertyNames(e);for(let r of t)ae(r);this._history.push({...this._state});for(let r of t)this._state[r]=e[r]}append(e,t){ae(e),this._history.push({...this._state}),Array.isArray(this._state[e])||(this._state[e]=[]),this._state[e].push(t)}getAll(){return{...this._state}}rollback(){this._history.length>0&&(this._state=this._history.pop())}};var X=class{constructor(e){this.schema=e}parse(e){let t=e.match(/```json\s*([\s\S]*?)\s*```/);if(t)return this.validate(JSON.parse(t[1]));let r=[e.match(/\{[\s\S]*?\}/),e.match(/\{[\s\S]*\}/)].filter(Boolean).map(s=>s[0]);for(let s of r)try{return this.validate(JSON.parse(s))}catch(i){if(!(i instanceof SyntaxError))throw i}return this.validate({result:e.trim()})}validate(e){let t=[];for(let[r,s]of Object.entries(this.schema)){if(s.required&&!(r in e)&&t.push(`Missing required field: ${r}`),r in e&&s.type){let i=typeof e[r];i!==s.type&&t.push(`Field '${r}' expected ${s.type}, got ${i}`)}if(s.validate&&r in e){let i=s.validate(e[r]);i&&t.push(`Field '${r}': ${i}`)}}if(t.length>0)throw new Error(`Output validation failed:
|
|
13
13
|
${t.join(`
|
|
14
|
-
`)}`);return e}};G();import{writeFileSync as fe,readFileSync as Ye,existsSync as Ge,mkdirSync as _t}from"node:fs";import{join as he,dirname as yt}from"node:path";import
|
|
14
|
+
`)}`);return e}};G();import{writeFileSync as fe,readFileSync as Ye,existsSync as Ge,mkdirSync as _t}from"node:fs";import{join as he,dirname as yt}from"node:path";import I from"chalk";var lt="__WORKFLOW_GRAPH_LOG__",Z=I.gray("\u2502"),ut=I.gray("\u250C"),xe=I.gray("\u2514"),ce=I.green("\u25C6"),Ae=I.hex("#c084fc")("\u25C6"),Oe=I.hex("#2dd4bf")("\u25C6"),le=I.red("\u25C6"),Ce=`${Z} `,Pe=2;function Re(o){return o<1e3?`${o}ms`:`${(o/1e3).toFixed(1)}s`}function Me(o,e){return(t,r,s)=>{if(typeof t!="string")return o(t,r,s);let i=process.stdout.columns||120,a="";for(let n=0;n<t.length;n++){let c=t[n];e.lineStart&&(a+=Ce,e.col=Pe,e.lineStart=!1),c===`
|
|
15
15
|
`?(a+=c,e.lineStart=!0,e.col=0,e.inEsc=!1):c==="\x1B"?(e.inEsc=!0,a+=c):e.inEsc?(a+=c,(c>="A"&&c<="Z"||c>="a"&&c<="z")&&(e.inEsc=!1)):(e.col++,a+=c,e.col>=i&&(a+=`
|
|
16
|
-
${Ce}`,e.col=Pe))}return o(a,r,s)}}var ue=class{constructor(){this._currentNode=null,this._origStdoutWrite=null,this._origStderrWrite=null,this._emitWorkflowGraphMarkers=String(process.env.ZIBBY_EMIT_GRAPH_MARKERS||"").trim()==="1"||String(process.env.ZIBBY_WORKFLOW_GRAPH_LOG_MARKERS||"").trim()==="1"}get isInsideNode(){return this._currentNode!==null}_startIntercepting(){this._origStdoutWrite=process.stdout.write.bind(process.stdout),this._origStderrWrite=process.stderr.write.bind(process.stderr);let e={lineStart:!0,col:0,inEsc:!1},t={lineStart:!0,col:0,inEsc:!1};this._outState=e,this._errState=t,process.stdout.write=
|
|
16
|
+
${Ce}`,e.col=Pe))}return o(a,r,s)}}var ue=class{constructor(){this._currentNode=null,this._origStdoutWrite=null,this._origStderrWrite=null,this._emitWorkflowGraphMarkers=String(process.env.ZIBBY_EMIT_GRAPH_MARKERS||"").trim()==="1"||String(process.env.ZIBBY_WORKFLOW_GRAPH_LOG_MARKERS||"").trim()==="1"}get isInsideNode(){return this._currentNode!==null}_startIntercepting(){this._origStdoutWrite=process.stdout.write.bind(process.stdout),this._origStderrWrite=process.stderr.write.bind(process.stderr);let e={lineStart:!0,col:0,inEsc:!1},t={lineStart:!0,col:0,inEsc:!1};this._outState=e,this._errState=t,process.stdout.write=Me(this._origStdoutWrite,e),process.stderr.write=Me(this._origStderrWrite,t)}_stopIntercepting(){this._origStdoutWrite&&(this._outState&&!this._outState.lineStart&&this._origStdoutWrite(`
|
|
17
17
|
`),process.stdout.write=this._origStdoutWrite),this._origStderrWrite&&(this._errState&&!this._errState.lineStart&&this._origStderrWrite(`
|
|
18
18
|
`),process.stderr.write=this._origStderrWrite),this._origStdoutWrite=null,this._origStderrWrite=null}_rawWrite(e){(this._origStdoutWrite||process.stdout.write.bind(process.stdout))(`${e}
|
|
19
19
|
`)}_emitGraphLogMarker(e){if(!this._emitWorkflowGraphMarkers)return;let t=`${lt}${JSON.stringify(e)}
|
|
@@ -22,19 +22,19 @@ ${Ce}`,e.col=Pe))}return o(a,r,s)}}var ue=class{constructor(){this._currentNode=
|
|
|
22
22
|
`)):process.stdout.write.bind(process.stdout)(`${e} ${t}
|
|
23
23
|
`)}step(e){this._origStdoutWrite?this._writeDot(ce,e):process.stdout.write.bind(process.stdout)(`${Z} ${ce} ${e}
|
|
24
24
|
`)}stepInfo(e){this.step(e)}stepTool(e){this._origStdoutWrite?this._writeDot(Ae,e):process.stdout.write.bind(process.stdout)(`${Z} ${Ae} ${e}
|
|
25
|
-
`)}stepMemory(e){let t=
|
|
26
|
-
`)}stepFail(e){this._origStdoutWrite?this._writeDot(le,
|
|
27
|
-
`)}nodeStart(e){this._currentNode=e,this._emitGraphLogMarker({phase:"node_begin",node:e}),this._rawWrite(`${ut} ${e}`),this._startIntercepting()}nodeComplete(e,t={}){this._stopIntercepting();let{duration:r,details:s}=t;if(s)for(let a of s)this._rawWrite(`${ce} ${a}`);let i=r?
|
|
25
|
+
`)}stepMemory(e){let t=I.hex("#2dd4bf")(e);this._origStdoutWrite?this._writeDot(Oe,t):process.stdout.write.bind(process.stdout)(`${Z} ${Oe} ${t}
|
|
26
|
+
`)}stepFail(e){this._origStdoutWrite?this._writeDot(le,I.red(e)):process.stdout.write.bind(process.stdout)(`${Z} ${le} ${I.red(e)}
|
|
27
|
+
`)}nodeStart(e){this._currentNode=e,this._emitGraphLogMarker({phase:"node_begin",node:e}),this._rawWrite(`${ut} ${e}`),this._startIntercepting()}nodeComplete(e,t={}){this._stopIntercepting();let{duration:r,details:s}=t;if(s)for(let a of s)this._rawWrite(`${ce} ${a}`);let i=r?I.dim(` ${Re(r)}`):"";this._rawWrite(`${xe} ${I.green("done")}${i}`),this._emitGraphLogMarker({phase:"node_end",node:e}),this._rawWrite("")}nodeFailed(e,t,r={}){this._stopIntercepting();let{duration:s}=r,i=s?I.dim(` ${Re(s)}`):"";this._rawWrite(`${le} ${I.red(t)}`),this._rawWrite(`${xe} ${I.red("failed")}${i}`),this._emitGraphLogMarker({phase:"node_end",node:e}),this._rawWrite("")}route(e,t){this._rawWrite(I.dim(` ${e} \u2192 ${t}`)),this._rawWrite("")}graphComplete(){}},v=new ue;var ee=".zibby/output",We="sessions",L=".session-info.json",Be=".zibby-stop";var De=["CI_JOB_ID","GITHUB_RUN_ID","CIRCLE_WORKFLOW_ID","BUILD_ID"];var j=class{constructor(e){if(this.config=e,this.name=e.name,this.prompt=e.prompt,this.outputSchema=e.outputSchema,!this.outputSchema&&!e._isCustomCode)throw new Error(`Node '${this.name}' must define outputSchema (Zod schema). This defines the contract for what the node returns to state.`);this.isZodSchema=this.outputSchema&&typeof this.outputSchema._def<"u",this.parser=e.outputSchema&&!this.isZodSchema?new X(e.outputSchema):null,this.retries=e.retries||0,this.onComplete=e.onComplete,this.customExecute=e.execute}async execute(e,t){let r=()=>t&&typeof t.getAll=="function"?t.getAll():e,s=d=>t&&typeof t.get=="function"?t.get(d):e?.[d];if(typeof this.customExecute=="function"){S.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let d=await this.customExecute(e);return typeof d=="object"&&d!==null&&d.success===!1?{success:!1,error:d.error||"Node execution failed",raw:d.raw||null}:this.isZodSchema?(S.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(d),raw:null}):{success:!0,output:d,raw:null}}catch(d){return S.error(`[workflow] node '${this.name}' failed: ${d.message}`),d.name==="ZodError"&&S.error(`Schema errors: ${JSON.stringify(d.issues||d.errors,null,2)}`),{success:!1,error:d.message,raw:null}}}let i=typeof this.prompt=="function"?this.prompt(r()):this.prompt,a=s("_skillHints");a&&(i=`${a}
|
|
28
28
|
|
|
29
|
-
${i}`);let n=r(),c=n.cwd||process.cwd(),u=n.sessionPath;try{if(u){let d=he(u,L);if(Ge(d)){let m=JSON.parse(Ye(d,"utf-8"));m.currentNode=this.name,fe(d,JSON.stringify(m,null,2),"utf-8")}let l=he(u,"..",L);if(Ge(l))try{let m=JSON.parse(Ye(l,"utf-8"));m.currentNode=this.name,fe(l,JSON.stringify(m,null,2),"utf-8")}catch{}}}catch(d){S.debug(`[workflow] could not update session info: ${d.message}`)}let h=null;for(let d=0;d<=this.retries;d++)try{S.debug(`[workflow] node '${this.name}' attempt ${d}`);let l=r().config||{},m=l.agents||{},_=this.config.agent??m[this.name]??null,w={state:r()};_&&(w.preferredAgent=_);let N={workspace:c,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:u,config:l,nodeName:this.name,timeout:this.config?.timeout||3e5},f=e?._coreInvokeAgent;f||(f=(await Promise.resolve().then(()=>(J(),H))).invokeAgent);let
|
|
29
|
+
${i}`);let n=r(),c=n.cwd||process.cwd(),u=n.sessionPath;try{if(u){let d=he(u,L);if(Ge(d)){let m=JSON.parse(Ye(d,"utf-8"));m.currentNode=this.name,fe(d,JSON.stringify(m,null,2),"utf-8")}let l=he(u,"..",L);if(Ge(l))try{let m=JSON.parse(Ye(l,"utf-8"));m.currentNode=this.name,fe(l,JSON.stringify(m,null,2),"utf-8")}catch{}}}catch(d){S.debug(`[workflow] could not update session info: ${d.message}`)}let h=null;for(let d=0;d<=this.retries;d++)try{S.debug(`[workflow] node '${this.name}' attempt ${d}`);let l=r().config||{},m=l.agents||{},_=this.config.agent??m[this.name]??null,w={state:r()};_&&(w.preferredAgent=_);let N={workspace:c,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:u,config:l,nodeName:this.name,timeout:this.config?.timeout||3e5},f=e?._coreInvokeAgent;f||(f=(await Promise.resolve().then(()=>(J(),H))).invokeAgent);let E=await f(i,w,N),y,x;if(typeof E=="string"?(y=E,x=null):E.structured?(y=E.raw||JSON.stringify(E.structured,null,2),x=E.structured):(y=E.raw||JSON.stringify(E,null,2),x=E.extracted||null),u)try{let p=he(u,this.name,"raw_stream_output.txt");_t(yt(p),{recursive:!0}),fe(p,typeof y=="string"?y:JSON.stringify(y),"utf-8")}catch(p){S.debug(`[workflow] could not save raw output: ${p.message}`)}if(this.isZodSchema&&x){S.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(x,null,2)}`);let p=x;if(typeof this.onComplete=="function")try{p=await this.onComplete(r(),x)}catch(W){S.warn(`[workflow] onComplete hook failed: ${W.message}`)}return{success:!0,output:p,raw:y}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(r(),{raw:y}),raw:y}}catch(p){throw new Error(`onComplete failed: ${p.message}`,{cause:p})}if(this.parser){let p=this.parser.parse(y);return S.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(p,null,2)}`),v.step("Output parsed"),{success:!0,output:p,raw:y}}return{success:!0,output:y,raw:y}}catch(l){h=l,d<this.retries&&S.info(`[workflow] node '${this.name}' failed, retrying (${d+1}/${this.retries})\u2026`)}return{success:!1,error:h.message,raw:null}}},oe=class extends j{constructor(e){super({...e,_isCustomCode:!0}),this.condition=e.condition}async execute(e,t){let r=t&&typeof t.getAll=="function"?t.getAll():e;return{success:!0,output:{nextNode:this.condition(r)},raw:null}}};import{existsSync as Ze,readFileSync as Et}from"node:fs";import{join as me,dirname as Ue}from"node:path";var re=class{static async loadContext(e,t,r={}){let s={},i=r.filenames||["CONTEXT.md","AGENTS.md"];if(e){let n=Ue(me(t,e));for(let c of i){let u=await this.findAndMergeContextFiles(c,n,t);if(u){let h=c.replace(/\.[^.]+$/,"").toLowerCase();s[h]=u}}}let a=r.discovery||{};for(let[n,c]of Object.entries(a))try{let u=me(t,c);Ze(u)&&(s[n]=await this.loadFile(u))}catch(u){console.warn(`[workflow] could not load context '${n}' from '${c}': ${u.message}`)}return s}static async findAndMergeContextFiles(e,t,r){let s=[],i=t;for(;i.startsWith(r);){let a=me(i,e);if(Ze(a))try{s.unshift(await this.loadFile(a))}catch(c){console.warn(`[workflow] could not load ${e} from ${a}: ${c.message}`)}let n=Ue(i);if(n===i)break;i=n}return s.length===0?null:s.every(a=>typeof a=="string")?s.join(`
|
|
30
30
|
|
|
31
31
|
---
|
|
32
32
|
|
|
33
|
-
`):s.every(a=>typeof a=="object")?Object.assign({},...s):s[s.length-1]}static async loadFile(e){let t=Et(e,"utf-8");if(e.endsWith(".json"))return JSON.parse(t);if(e.endsWith(".js")||e.endsWith(".mjs")){let{pathToFileURL:r}=await import("url"),s=await import(r(e).href);return s.default||s}return t}};import{mkdirSync as Ke,existsSync as ge,writeFileSync as He,unlinkSync as $t}from"node:fs";import{join as
|
|
33
|
+
`):s.every(a=>typeof a=="object")?Object.assign({},...s):s[s.length-1]}static async loadFile(e){let t=Et(e,"utf-8");if(e.endsWith(".json"))return JSON.parse(t);if(e.endsWith(".js")||e.endsWith(".mjs")){let{pathToFileURL:r}=await import("url"),s=await import(r(e).href);return s.default||s}return t}};import{mkdirSync as Ke,existsSync as ge,writeFileSync as He,unlinkSync as $t}from"node:fs";import{join as M,resolve as ze}from"node:path";import{config as It}from"dotenv";import{zodToJsonSchema as Je}from"zod-to-json-schema";import vt from"handlebars";function Nt({traceFrom:o,sessionId:e,sessionPath:t,idSource:r,mkdirFresh:s}){if(!(process.env.ZIBBY_SESSION_LOG==="1"||process.env.ZIBBY_SESSION_LOG==="true"))return;let a=typeof process.ppid=="number"?process.ppid:"n/a",n=`[zibby:session] from=${o} pid=${process.pid} ppid=${a} sessionId=${e} source=${r} mkdir=${s?"yes":"no"} path=${t}`;if(console.log(n),process.env.ZIBBY_TRACE_SESSION==="1"||process.env.ZIBBY_TRACE_SESSION==="true"){let h=(new Error("session trace").stack||"").split(`
|
|
34
34
|
`).slice(2,14).join(`
|
|
35
35
|
`);console.log(`[zibby:session] stack (${o}):
|
|
36
|
-
${h}`)}}function bt(){return process.env.ZIBBY_TRUST_SESSION_ENV==="1"||process.env.ZIBBY_TRUST_SESSION_ENV==="true"||process.env.ZIBBY_KEEP_SESSION_ENV==="1"||process.env.ZIBBY_KEEP_SESSION_ENV==="true"}function kt(){if(!(process.env.ZIBBY_PIN_SESSION_PATH==="1"||process.env.ZIBBY_PIN_SESSION_PATH==="true"))return;let e=process.env.ZIBBY_SESSION_PATH;if(!(e==null||String(e).trim()===""))try{return ze(String(e).trim())}catch{return String(e).trim()}}function Tt(){bt()||(delete process.env.ZIBBY_SESSION_PATH,delete process.env.ZIBBY_SESSION_ID)}function xt({sessionPath:o,sessionId:e}){o&&typeof o=="string"&&(process.env.ZIBBY_SESSION_PATH=o),e!=null&&String(e).trim()!==""&&(process.env.ZIBBY_SESSION_ID=String(e).trim())}function At(o={}){let e=De.map(i=>process.env[i]).find(Boolean),t=Math.random().toString(36).slice(2,6),r=e||`${Date.now()}_${t}`,s=o.paths?.sessionPrefix;return s?`${s}_${r}`:r}function Ot({cwd:o=process.cwd(),config:e={},initialState:t={},traceFrom:r="resolveWorkflowSession"}={}){let s=t.sessionPath,i=t.sessionTimestamp,a="initialState.sessionPath";if(!s&&process.env.ZIBBY_SESSION_PATH)try{let u=ze(String(process.env.ZIBBY_SESSION_PATH));u&&(s=u,a="ZIBBY_SESSION_PATH")}catch{}let n;if(s)n=String(s).split(/[/\\]/).filter(Boolean).pop(),i==null&&(i=Date.now());else{let u=process.env.ZIBBY_SESSION_ID&&String(process.env.ZIBBY_SESSION_ID).trim();if(u)n=u,a="ZIBBY_SESSION_ID";else{let d=e.sessionId!=null?String(e.sessionId).trim():"";d&&d!=="last"?(n=d,a="config.sessionId"):(n=At(e),a="generated")}i=i??Date.now();let h=e.paths?.output||ee;s=
|
|
37
|
-
\u{1F6D1} External stop requested \u2014 ending workflow.`),v.step("Workflow stopped externally"),{success:!0,state:f.getAll(),executionLog:
|
|
36
|
+
${h}`)}}function bt(){return process.env.ZIBBY_TRUST_SESSION_ENV==="1"||process.env.ZIBBY_TRUST_SESSION_ENV==="true"||process.env.ZIBBY_KEEP_SESSION_ENV==="1"||process.env.ZIBBY_KEEP_SESSION_ENV==="true"}function kt(){if(!(process.env.ZIBBY_PIN_SESSION_PATH==="1"||process.env.ZIBBY_PIN_SESSION_PATH==="true"))return;let e=process.env.ZIBBY_SESSION_PATH;if(!(e==null||String(e).trim()===""))try{return ze(String(e).trim())}catch{return String(e).trim()}}function Tt(){bt()||(delete process.env.ZIBBY_SESSION_PATH,delete process.env.ZIBBY_SESSION_ID)}function xt({sessionPath:o,sessionId:e}){o&&typeof o=="string"&&(process.env.ZIBBY_SESSION_PATH=o),e!=null&&String(e).trim()!==""&&(process.env.ZIBBY_SESSION_ID=String(e).trim())}function At(o={}){let e=De.map(i=>process.env[i]).find(Boolean),t=Math.random().toString(36).slice(2,6),r=e||`${Date.now()}_${t}`,s=o.paths?.sessionPrefix;return s?`${s}_${r}`:r}function Ot({cwd:o=process.cwd(),config:e={},initialState:t={},traceFrom:r="resolveWorkflowSession"}={}){let s=t.sessionPath,i=t.sessionTimestamp,a="initialState.sessionPath";if(!s&&process.env.ZIBBY_SESSION_PATH)try{let u=ze(String(process.env.ZIBBY_SESSION_PATH));u&&(s=u,a="ZIBBY_SESSION_PATH")}catch{}let n;if(s)n=String(s).split(/[/\\]/).filter(Boolean).pop(),i==null&&(i=Date.now());else{let u=process.env.ZIBBY_SESSION_ID&&String(process.env.ZIBBY_SESSION_ID).trim();if(u)n=u,a="ZIBBY_SESSION_ID";else{let d=e.sessionId!=null?String(e.sessionId).trim():"";d&&d!=="last"?(n=d,a="config.sessionId"):(n=At(e),a="generated")}i=i??Date.now();let h=e.paths?.output||ee;s=M(o,h,We,n)}let c=!ge(s);return c&&Ke(s,{recursive:!0}),(c||a!=="initialState.sessionPath")&&Nt({traceFrom:r,sessionId:n,sessionPath:s,idSource:a,mkdirFresh:c}),xt({sessionPath:s,sessionId:n}),{sessionPath:s,sessionId:n,sessionTimestamp:i}}var se=class{constructor(e={}){this.nodes=new Map,this.edges=new Map,this.entryPoint=null,this.middleware=Array.isArray(e.middleware)?[...e.middleware]:[],e.nodeMiddleware&&this.middleware.push(e.nodeMiddleware),this.nodeTypeMap=new Map,this.conditionalCodeMap=new Map,this.stateSchema=e.stateSchema||null,this.nodePrompts=new Map,this.nodeOptions=new Map,this._invokeAgent=e.invokeAgent||null,this._compiledPrompts=new Map}setStateSchema(e){return this.stateSchema=e,this}getStateSchema(){return this.stateSchema}addNode(e,t,r={}){let s=t instanceof j?t:new j(t);return s.name=e,this.nodes.set(e,s),r.prompt&&this.nodePrompts.set(e,r.prompt),Object.keys(r).length>0&&this.nodeOptions.set(e,r),this}addConditionalNode(e,t){return this.nodes.set(e,new oe({...t,name:e})),this}addEdge(e,t){return this.edges.set(e,t),this}setNodeType(e,t){return this.nodeTypeMap.set(e,t),this}addConditionalEdges(e,t,{labels:r}={}){return this.edges.set(e,{conditional:!0,routes:t,labels:r}),typeof t=="function"&&this.conditionalCodeMap.set(e,t.toString()),this}setEntryPoint(e){return this.entryPoint=e,this}use(e){return typeof e=="function"&&this.middleware.push(e),this}_composeMiddleware(e,t,r,s,i){let a=r;for(let n=e.length-1;n>=0;n--){let c=e[n],u=a;a=()=>c(t,u,s,i)}return a()}serialize(){let e=[],t={};for(let[i,a]of this.nodes){let n=this.nodeTypeMap.get(i)||i;e.push({id:i,type:n,data:{nodeType:n,label:i}});let c={};a._isCustomCode&&typeof a.execute=="function"&&(c.customCode=a.execute.toString());let u=this.nodePrompts.get(i);if(u&&(c.prompt=u),typeof a.customExecute=="function"&&(c.executeCode=a.customExecute.toString()),a.outputSchema)try{if(typeof a.outputSchema._def<"u"){let d=Je(a.outputSchema,{target:"openApi3"});c.outputSchema={jsonSchema:d,variables:this._flattenJsonSchemaToVariables(d)}}else c.outputSchema={schema:a.outputSchema}}catch(d){console.warn(`[workflow] failed to convert schema for ${i}:`,d.message)}let h=(this.resolvedToolsMap||{})[i];h?.toolIds&&(c.tools=h.toolIds),Object.keys(c).length>0&&(t[i]=c)}let r=[];for(let[i,a]of this.edges)if(typeof a=="string")r.push({source:i,target:a});else if(a.conditional){let n=this.conditionalCodeMap.get(i)||a.routes.toString(),c=this._inferConditionalTargets(a.routes),u=a.labels||{};for(let h of c){let d={source:i,target:h,data:{conditionalCode:n}};u[h]&&(d.label=u[h]),r.push(d)}}let s=null;if(this.stateSchema)try{s=Je(this.stateSchema,{target:"openApi3"})}catch{s=this.stateSchema}return{nodes:e,edges:r,nodeConfigs:t,stateSchema:s}}_inferConditionalTargets(e){let t=e.toString(),r=new Set,s=/return\s+['"]([^'"]+)['"]/g,i;for(;(i=s.exec(t))!==null;)r.add(i[1]);return[...r]}_flattenJsonSchemaToVariables(e,t=""){let r=e;if(e.$ref&&e.definitions){let s=e.$ref.replace("#/definitions/","");r=e.definitions[s]||e}return this._flattenSchema(r,t)}_flattenSchema(e,t=""){if(!e||typeof e!="object")return[];let r=[],s=e.properties||{},i=e.required||[];for(let[a,n]of Object.entries(s)){let c=t?`${t}.${a}`:a;r.push({path:c,type:n.type||"unknown",label:n.description||this._formatLabel(a),optional:!i.includes(a)}),n.type==="object"&&n.properties&&r.push(...this._flattenSchema(n,c)),n.type==="array"&&n.items?.type==="object"&&n.items.properties&&r.push(...this._flattenSchema(n.items,`${c}[]`))}return r}_formatLabel(e){return e.replace(/([A-Z])/g," $1").replace(/^./,t=>t.toUpperCase()).trim()}_summarizeNodeOutput(e,t){if(!t||typeof t!="object")return[];let r=[];t.success!==void 0&&r.push(`Result: ${t.success?"passed":"failed"}`);for(let[s,i]of Object.entries(t))if(!(s==="success"||s==="raw"||s==="nextNode")){if(typeof i=="string"&&i.length<=80)r.push(`${s}: ${i}`);else if(Array.isArray(i)){let a=i.length,n=i.filter(u=>u?.passed===!0).length,c=i.some(u=>u?.passed!==void 0);r.push(c?`${s}: ${n}/${a} passed${a-n?`, ${a-n} failed`:""}`:`${s}: ${a} items`)}if(r.length>=4)break}return r}async run(e,t={},r={}){if(!this.entryPoint)throw new Error("No entry point set for graph");let s=new AbortController;r.signal&&(r.signal.aborted?s.abort():r.signal.addEventListener("abort",()=>s.abort(),{once:!0}));let i=r.strategyAbortTimeoutMs??t.config?.strategyAbortTimeoutMs??5e3,a=t.cwd||process.cwd();It({path:M(a,".env")});let n=t.config||{};if(!n||Object.keys(n).length===0)try{let g=M(a,".zibby.config.js");ge(g)&&(n=(await import(g)).default||{})}catch{}process.env.EXECUTION_ID&&!n.agent?.strictMode&&(n.agent={...n.agent,strictMode:!0});let c=t.agentType;if(!c){let g=n?.agent;g?.provider?c=g.provider:g?.gemini?c="gemini":g?.claude?c="claude":g?.cursor?c="cursor":g?.codex?c="codex":c=process.env.AGENT_TYPE||"cursor"}let u=t.contextConfig||e?.config?.contextConfig||e?.config?.context||n?.context||{};if(this.stateSchema){let g=this.stateSchema.safeParse(t);if(!g.success){let T=g.error.issues.map(b=>`${b.path.join(".")}: ${b.message}`);throw console.error("\u274C Initial state validation failed:"),T.forEach(b=>console.error(` - ${b}`)),new Error(`State validation failed: ${T.join(", ")}`)}v.step("State validated against schema")}let h=kt(),d=t.sessionPath||h;d||Tt();let{sessionPath:l,sessionTimestamp:m,sessionId:_}=Ot({cwd:a,config:n,traceFrom:"WorkflowGraph.run",initialState:{sessionPath:d,sessionTimestamp:t.sessionTimestamp}});v.step(`Session ${_}`);let w=await re.loadContext(t.specPath||"",a,u);Object.keys(w).length>0&&v.step(`Context loaded: ${Object.keys(w).join(", ")}`);let N=t.outputPath;!N&&t.specPath&&(e?.calculateOutputPath?N=e.calculateOutputPath(t.specPath):console.warn(`\u26A0\uFE0F outputPath not resolved (specPath=${t.specPath})`));let f=new q({...t,config:n,agentType:c,outputPath:N,sessionPath:l,sessionTimestamp:m,context:w,resolvedTools:this.resolvedToolsMap||{},_signal:s.signal}),E=new Map;try{await import("@zibby/skills")}catch{}let{getSkill:y}=await Promise.resolve().then(()=>(te(),Fe)),x=new Set;for(let[,g]of this.nodes)for(let T of g.config?.skills||[])x.add(T);for(let g of x){let T=y(g);if(typeof T?.middleware=="function")try{let b=await T.middleware();typeof b=="function"&&E.set(g,b)}catch{}}let p=this.entryPoint,W=[],ye=n?.recursionLimit??100,et=0;try{for(;p&&p!=="END";){if(++et>ye)throw new Error(`Workflow exceeded recursion limit (${ye}) \u2014 likely a cyclic conditional route. Set config.recursionLimit if you need a higher cap.`);let T=M(l,Be);if(ge(T)){try{$t(T)}catch{}s.abort()}if(s.signal.aborted)return console.warn(`
|
|
37
|
+
\u{1F6D1} External stop requested \u2014 ending workflow.`),v.step("Workflow stopped externally"),{success:!0,state:f.getAll(),executionLog:W,stoppedExternally:!0};let b=this.nodes.get(p);if(!b)throw new Error(`Node '${p}' not found in graph`);let Ee=JSON.stringify({sessionPath:l,sessionTimestamp:m,currentNode:p,createdAt:new Date().toISOString(),config:f.get("config")}),tt=M(l,L);He(tt,Ee,"utf-8");let $e=f.get("config")?.paths?.output||ee,ot=M(a,$e,L);Ke(M(a,$e),{recursive:!0});try{He(ot,Ee,"utf-8")}catch{}let Ie=t.onPipelineProgress;if(typeof Ie=="function")try{Ie({cwd:a,sessionPath:l,sessionId:_,outputBase:f.get("config")?.paths?.output||ee,currentNode:p})}catch{}let rt=(this.resolvedToolsMap||{})[p]||null;f.set("_currentNodeTools",rt);let st=f.get("nodeConfigs")||{};f.set("_currentNodeConfig",st[p]||{}),v.nodeStart(p);let ve=Date.now(),K=this.nodePrompts.get(p);if(!this._invokeAgent){let A=await Promise.resolve().then(()=>(J(),H));this._invokeAgent=A.invokeAgent}let nt=this._invokeAgent,Ne=async(A,O,$={})=>{let k=nt(A,O,{...$,signal:s.signal});return k.catch(()=>{}),s.signal.aborted?k:Promise.race([k,new Promise((B,D)=>{let P=()=>{setTimeout(()=>{let Y=new Error(`Strategy ignored AbortSignal \u2014 engine deadman fired after ${i}ms`);Y.name="AbortError",D(Y)},i)};s.signal.addEventListener("abort",P,{once:!0})})])},be={state:f,invokeAgent:async(A={},O={})=>{let $=O.prompt||"";if(K){let k=this._compiledPrompts.get(p);k||(k=vt.compile(K,{noEscape:!0}),this._compiledPrompts.set(p,k));try{$=k(A)}catch(B){throw console.error(`\u274C Template rendering failed for node '${p}':`,B.message),new Error(`Template rendering failed: ${B.message}`,{cause:B})}}else if(!$)throw new Error(`No prompt template configured for node '${p}' and no prompt provided in options`);return Ne($,{state:f.getAll(),images:O.images||[]},{model:O.model||f.get("model"),workspace:f.get("workspace"),schema:O.schema,...O,signal:s.signal})},_coreInvokeAgent:Ne,agent:e,nodeId:p,promptTemplate:K,getPromptTemplate:()=>K,...f.getAll()};try{let A=(b.config?.skills||[]).map(P=>E.get(P)).filter(Boolean),O=[...this.middleware,...A],$;O.length>0?$=await this._composeMiddleware(O,p,async()=>b.execute(be,f),f.getAll(),f):$=await b.execute(be,f);let k=Date.now()-ve;if(W.push({node:p,success:$.success,duration:k,timestamp:new Date().toISOString()}),!$.success){if(s.signal.aborted)return v.step("Workflow stopped externally"),{success:!0,state:f.getAll(),executionLog:W,stoppedExternally:!0};f.append("errors",{node:p,error:$.error});let P=b.config?.retries||0,Y=`${p}_retries`,z=f.getAll()[Y]||0;if(z<P){v.stepInfo(`Retrying (attempt ${z+1}/${P})`),f.update({[Y]:z+1,[`${p}_raw`]:$.raw});continue}throw v.nodeFailed(p,$.error,{duration:k}),new Error(`Node '${p}' failed after ${z} attempts: ${$.error}`)}f.update({[p]:$.output});let B=this._summarizeNodeOutput(p,$.output);v.nodeComplete(p,{duration:k,details:B});let D=this.edges.get(p);if(!D)p="END";else if(D.conditional){let P=D.routes(f.getAll());v.route(p,P),p=P}else p=D}catch(A){throw v.isInsideNode&&v.nodeFailed(p,A.message,{duration:Date.now()-ve}),f.set("failed",!0),f.set("failedAt",p),A}}v.graphComplete();let g={success:!0,state:f.getAll(),executionLog:W};return e&&typeof e.onComplete=="function"&&await e.onComplete(g),g}finally{if(e&&typeof e.cleanup=="function")try{await e.cleanup()}catch(g){console.warn(`[workflow] agent.cleanup() failed: ${g.message}`)}}}};var we=Symbol.for("@zibby/agent-workflow.nodes");globalThis[we]||(globalThis[we]=new Map);var Se=globalThis[we];function Ct(o,e){Se.set(o,e)}function Ve(o){return Se.get(o)}function _e(o){return Se.has(o)}Ct("ai_agent",{name:"ai_agent",factory:!0,create:(o,e={})=>({name:o,_isCustomCode:!0,execute:async t=>{let r=t?._coreInvokeAgent;r||(r=(await Promise.resolve().then(()=>(J(),H))).invokeAgent);let s=e.extraPromptInstructions||"Execute the task based on the current state.",i=Pt(s,t),a=await r(i,{cwd:t.workspace||process.cwd(),model:t.model,tools:e.resolvedTools||null});return{success:!0,output:{raw:a,nodeId:o},raw:typeof a=="string"?a:a.raw}}})});function Pt(o,e){let t=/@([\w.]+)/g,r=new Set,s;for(;(s=t.exec(o))!==null;)r.add(s[1]);if(r.size===0)return o;let i=[],a=new Set;for(let n of r){let c=n.split(".")[0];if(a.has(c))continue;let u=n.split(".").reduce((l,m)=>l?.[m],e);if(u===void 0)continue;let h=typeof u=="string"?u:u?.raw??JSON.stringify(u,null,2),d=n.replace(/_/g," ").replace(/\b\w/g,l=>l.toUpperCase());i.push(`## ${d}
|
|
38
38
|
${h}`),n.includes(".")||a.add(c)}return i.length===0?o:`${o}
|
|
39
39
|
|
|
40
40
|
---
|
|
@@ -42,4 +42,4 @@ ${h}`),n.includes(".")||a.add(c)}return i.length===0?o:`${o}
|
|
|
42
42
|
|
|
43
43
|
${i.join(`
|
|
44
44
|
|
|
45
|
-
`)}`}te();G();var Rt={};function Xe(o,e){if(Array.isArray(e))return qe(e);let t=Rt[o];return!t||t.length===0?null:qe(t)}function qe(o){if(!Array.isArray(o)||o.length===0)return null;let e=[],t={},r=[];for(let s of o){let i=U(s);if(!i){S.warn(`[workflow] unknown skill "${s}" \u2014 skipping`);continue}r.push(s);for(let a of i.tools||[])e.push({name:a.name,description:a.description,input_schema:a.input_schema||{type:"object",properties:{}}});if(!t[i.serverName])if(typeof i.resolve=="function"){let a=i.resolve();a&&(t[i.serverName]={...a,toolPrefix:s})}else{let a={};for(let n of i.envKeys||[]){let c=process.env[n];c&&(a[n]=c)}t[i.serverName]={command:i.command,args:[...i.args||[]],env:a,toolPrefix:s}}}return r.length===0?null:{toolIds:r,claudeTools:e,mcpServers:t}}G();function bo(o,e={}){let{nodes:t,edges:r,nodeConfigs:s={}}=o;if(!Array.isArray(t)||t.length===0)throw new C("Graph must have at least one node");if(!Array.isArray(r))throw new C("Graph edges must be an array");let i=new se(e);e.stateSchema&&i.setStateSchema(e.stateSchema);let a=new Set,n=new Map,c={};for(let l of t){let m=ne(l);n.set(l.id,{...l,resolvedType:m}),m==="decision"&&a.add(l.id)}for(let[l,m]of n){if(a.has(l))continue;let _=m.resolvedType,w=s[l]||{},N=Xe(_,w.tools);N&&(c[l]=N);let f={};w.prompt&&(f.prompt=w.prompt);let
|
|
45
|
+
`)}`}te();G();var Rt={};function Xe(o,e){if(Array.isArray(e))return qe(e);let t=Rt[o];return!t||t.length===0?null:qe(t)}function qe(o){if(!Array.isArray(o)||o.length===0)return null;let e=[],t={},r=[];for(let s of o){let i=U(s);if(!i){S.warn(`[workflow] unknown skill "${s}" \u2014 skipping`);continue}r.push(s);for(let a of i.tools||[])e.push({name:a.name,description:a.description,input_schema:a.input_schema||{type:"object",properties:{}}});if(!t[i.serverName])if(typeof i.resolve=="function"){let a=i.resolve();a&&(t[i.serverName]={...a,toolPrefix:s})}else{let a={};for(let n of i.envKeys||[]){let c=process.env[n];c&&(a[n]=c)}t[i.serverName]={command:i.command,args:[...i.args||[]],env:a,toolPrefix:s}}}return r.length===0?null:{toolIds:r,claudeTools:e,mcpServers:t}}G();function bo(o,e={}){let{nodes:t,edges:r,nodeConfigs:s={}}=o;if(!Array.isArray(t)||t.length===0)throw new C("Graph must have at least one node");if(!Array.isArray(r))throw new C("Graph edges must be an array");let i=new se(e);e.stateSchema&&i.setStateSchema(e.stateSchema);let a=new Set,n=new Map,c={};for(let l of t){let m=ne(l);n.set(l.id,{...l,resolvedType:m}),m==="decision"&&a.add(l.id)}for(let[l,m]of n){if(a.has(l))continue;let _=m.resolvedType,w=s[l]||{},N=Xe(_,w.tools);N&&(c[l]=N);let f={};w.prompt&&(f.prompt=w.prompt);let E=_e(_);if(S.debug(`[workflow] compiler: node "${l}" type="${_}" registered=${E}`),w.customCode&&!E)i.addNode(l,Qe(l,w.customCode,w),f),i.setNodeType(l,_);else if(E){let y=Ve(_);y.factory?i.addNode(l,y.create(l,{...w,resolvedTools:N}),f):i.addNode(l,y,f),i.setNodeType(l,_)}else if(w.executeCode)i.addNode(l,Qe(l,w.executeCode,w),f),i.setNodeType(l,_);else throw new C(`Unknown node type "${_}" for node "${l}". Did you forget to register it?`)}i.resolvedToolsMap=c;let u=new Set;for(let l of r)a.has(l.target)||u.add(l.target);let h=t.find(l=>!a.has(l.id)&&!u.has(l.id));if(!h)throw new C("Could not determine entry point: no node without incoming edges found");i.setEntryPoint(h.id);let d=Mt(r,"source");for(let l of r)if(!a.has(l.source))if(a.has(l.target)){let m=l.target,_=d.get(m)||[];if(_.length===0)throw new C(`Decision node "${m}" has no outgoing edges`);let w=Wt(m,_,a);i.addConditionalEdges(l.source,w)}else i.addEdge(l.source,l.target);return i}function ko(o){let e=[];if(!o||typeof o!="object")return{valid:!1,errors:["Config must be a non-null object"]};if((!Array.isArray(o.nodes)||o.nodes.length===0)&&e.push("Graph must have at least one node"),Array.isArray(o.edges)||e.push("Graph edges must be an array"),e.length>0)return{valid:!1,errors:e};let t=o.nodeConfigs||{};for(let n of o.nodes){let c=ne(n);if(c==="decision"||_e(c))continue;let u=t[n.id]||{};u.customCode||u.executeCode||e.push(`Unknown node type "${c}" for node "${n.id}". Register it or provide customCode/executeCode.`)}let r=new Set(o.nodes.map(n=>n.id));for(let n of o.edges)r.has(n.source)||e.push(`Edge references unknown source node "${n.source}"`),r.has(n.target)||e.push(`Edge references unknown target node "${n.target}"`);let s=new Set(o.nodes.filter(n=>ne(n)==="decision").map(n=>n.id)),i=new Set;for(let n of o.edges)s.has(n.target)||i.add(n.target);let a=o.nodes.filter(n=>!s.has(n.id)&&!i.has(n.id));a.length===0?e.push("No entry point found (every node has incoming edges)"):a.length>1&&e.push(`Multiple entry points found: ${a.map(n=>n.id).join(", ")}`);for(let n of s){let c=o.edges.filter(h=>h.source===n);c.length===0&&e.push(`Decision node "${n}" has no outgoing edges`),c.some(h=>h.data?.conditionalCode||h.conditionalCode)||e.push(`Decision node "${n}" outgoing edges have no conditionalCode`)}return{valid:e.length===0,errors:e}}function To(o){return!o||!Array.isArray(o.nodes)?[]:o.nodes.filter(e=>ne(e)!=="decision").map(e=>e.id)}function ne(o){let e=o.data?.nodeType||o.data?.type||o.type;return e==="workflowNode"||e==="custom"||e==="default"?o.id:e}function Mt(o,e){let t=new Map;for(let r of o){let s=r[e];t.has(s)||t.set(s,[]),t.get(s).push(r)}return t}function Wt(o,e,t){let r=e.find(n=>n.data?.conditionalCode||n.conditionalCode);if(!r)throw new C(`Decision node "${o}" has no conditionalCode on its outgoing edges`);let s=r.data?.conditionalCode||r.conditionalCode,i=new Set(e.map(n=>n.target).filter(n=>!t.has(n))),a;try{let c=new Function(`return (${s})`)();a=u=>{let h=c(u);return i.has(h)||S.warn(`[workflow] conditional route from "${o}" returned "${h}" which is not in valid targets: ${[...i].join(", ")}`),h}}catch(n){throw new C(`Failed to compile conditionalCode for "${o}": ${n.message}`)}return a}function Qe(o,e,t={}){let r;try{r=new Function("invokeAgent","require","console",`return (${e})`)}catch(a){throw new C(`Failed to compile customCode for node "${o}": ${a.message}`)}let s=r(async(...a)=>{let{invokeAgent:n}=await Promise.resolve().then(()=>(J(),H));return n(...a)},typeof ie<"u"?ie:void 0,console),i=null;return t.outputSchema&&(i=t.outputSchema.jsonSchema||t.outputSchema),{name:o,_isCustomCode:!0,outputSchema:i,execute:async a=>{try{let n=await s(a);return typeof n=="object"&&"success"in n?n:{success:!0,output:n,raw:null}}catch(n){return{success:!1,error:n.message,raw:null}}}}}var C=class extends Error{constructor(e){super(e),this.name="CompilationError"}};export{C as CompilationError,bo as compileGraph,To as extractSteps,ko as validateGraphConfig};
|
package/dist/graph.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
var zt=Object.defineProperty;var G=(i,t)=>()=>(i&&(t=i(i=0)),t);var yt=(i,t)=>{for(var e in t)zt(i,e,{get:t[e],enumerable:!0})};var Et,Xt,K,
|
|
1
|
+
var zt=Object.defineProperty;var G=(i,t)=>()=>(i&&(t=i(i=0)),t);var yt=(i,t)=>{for(var e in t)zt(i,e,{get:t[e],enumerable:!0})};var Et,Xt,K,S,Q=G(()=>{Et=()=>{},Xt={debug:Et,info:Et,warn:(...i)=>console.warn("[workflow]",...i),error:(...i)=>console.error("[workflow]",...i)},K={impl:Xt},S={debug:(...i)=>K.impl.debug?.(...i),info:(...i)=>K.impl.info?.(...i),warn:(...i)=>K.impl.warn?.(...i),error:(...i)=>K.impl.error?.(...i)}});var Pt=G(()=>{});var Ct={};yt(Ct,{clearSkills:()=>ne,getAllSkills:()=>se,getSkill:()=>rt,hasSkill:()=>oe,listSkillIds:()=>re,registerSkill:()=>ee});function ee(i){if(!i||typeof i.id!="string")throw new Error("Skill definition must include a string id");D.set(i.id,Object.freeze({...i}))}function rt(i){return D.get(i)||null}function oe(i){return D.has(i)}function se(){return new Map(D)}function re(){return Array.from(D.keys())}function ne(){D.clear()}var st,D,nt=G(()=>{st=Symbol.for("@zibby/agent-workflow.skills");globalThis[st]||(globalThis[st]=new Map);D=globalThis[st]});var at={};yt(at,{getAgentStrategy:()=>Rt,invokeAgent:()=>ce,listStrategies:()=>ae,registerStrategy:()=>ie});function ie(i){if(!i||typeof i.getName!="function"||typeof i.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let t=C.findIndex(e=>e.getName()===i.getName());t>=0?C[t]=i:C.push(i)}function ae(){return C.map(i=>i.getName())}function Rt(i={}){let{state:t={},preferredAgent:e=null}=i,s=e||t.agentType||process.env.AGENT_TYPE;if(!s){let r=C.map(n=>n.getName()).join(", ")||"none registered";throw new Error(`No agent specified. Set agentType in state or AGENT_TYPE env var. Available: ${r}`)}S.debug(`[workflow] agent selection: requested=${s}`);let o=C.find(r=>r.getName()===s);if(!o){let r=C.map(n=>n.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${s}'. Available: ${r}`)}if(!o.canHandle(i))throw new Error(`Agent '${s}' is not available in this environment. Check credentials/environment.`);return S.debug(`[workflow] using agent: ${o.getName()}`),o}async function ce(i,t={},e={}){let s=Rt(t),o=t.state?.config||e.config||{},r=o.models||{},n=e.nodeName&&r[e.nodeName]||null,a=r.default||null,c=o.agent?.[s.name]?.model||null,d=n||a||c||e.model||null,h={...e,model:d,workspace:t.state?.workspace||e.workspace,schema:e.schema||t.schema,images:e.images||t.images||[],skills:e.skills||t.skills||[],config:o},l=i,m=h.skills||[];if(m.length>0&&!e.skipPromptFragments){let x=m.map(P=>{let T=rt(P)?.promptFragment;return typeof T=="function"?T():T}).filter(Boolean);x.length>0&&(l+=`
|
|
2
2
|
|
|
3
3
|
${x.join(`
|
|
4
4
|
|
|
@@ -9,9 +9,9 @@ PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PRE
|
|
|
9
9
|
\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501
|
|
10
10
|
|
|
11
11
|
${E}
|
|
12
|
-
`),
|
|
12
|
+
`),S.debug(`[workflow] prompt length: ${l.length} chars`),s.invoke(l,h)}var it,C,ct=G(()=>{Pt();Q();nt();it=Symbol.for("@zibby/agent-workflow.strategies");globalThis[it]||(globalThis[it]=[]);C=globalThis[it]});var qt=new Set(["__proto__","constructor","prototype"]);function X(i){if(qt.has(i))throw new Error(`Invalid state key: "${i}"`)}var H=class{constructor(t={}){this._state=Object.create(null),Object.assign(this._state,{messages:[],errors:[],artifacts:{},metadata:{},...t}),this._history=[]}get(t){return this._state[t]}set(t,e){X(t),this._history.push({...this._state}),this._state[t]=e}update(t){let e=Object.getOwnPropertyNames(t);for(let s of e)X(s);this._history.push({...this._state});for(let s of e)this._state[s]=t[s]}append(t,e){X(t),this._history.push({...this._state}),Array.isArray(this._state[t])||(this._state[t]=[]),this._state[t].push(e)}getAll(){return{...this._state}}rollback(){this._history.length>0&&(this._state=this._history.pop())}};var J=class{constructor(t){this.schema=t}parse(t){let e=t.match(/```json\s*([\s\S]*?)\s*```/);if(e)return this.validate(JSON.parse(e[1]));let s=[t.match(/\{[\s\S]*?\}/),t.match(/\{[\s\S]*\}/)].filter(Boolean).map(o=>o[0]);for(let o of s)try{return this.validate(JSON.parse(o))}catch(r){if(!(r instanceof SyntaxError))throw r}return this.validate({result:t.trim()})}validate(t){let e=[];for(let[s,o]of Object.entries(this.schema)){if(o.required&&!(s in t)&&e.push(`Missing required field: ${s}`),s in t&&o.type){let r=typeof t[s];r!==o.type&&e.push(`Field '${s}' expected ${o.type}, got ${r}`)}if(o.validate&&s in t){let r=o.validate(t[s]);r&&e.push(`Field '${s}': ${r}`)}}if(e.length>0)throw new Error(`Output validation failed:
|
|
13
13
|
${e.join(`
|
|
14
|
-
`)}`);return t}};Q();import{writeFileSync as lt,readFileSync as Wt,existsSync as Bt,mkdirSync as le}from"node:fs";import{join as ut,dirname as ue}from"node:path";import
|
|
14
|
+
`)}`);return t}};Q();import{writeFileSync as lt,readFileSync as Wt,existsSync as Bt,mkdirSync as le}from"node:fs";import{join as ut,dirname as ue}from"node:path";import _ from"chalk";var Qt="__WORKFLOW_GRAPH_LOG__",Y=_.gray("\u2502"),te=_.gray("\u250C"),It=_.gray("\u2514"),tt=_.green("\u25C6"),$t=_.hex("#c084fc")("\u25C6"),bt=_.hex("#2dd4bf")("\u25C6"),et=_.red("\u25C6"),Nt=`${Y} `,Ot=2;function kt(i){return i<1e3?`${i}ms`:`${(i/1e3).toFixed(1)}s`}function vt(i,t){return(e,s,o)=>{if(typeof e!="string")return i(e,s,o);let r=process.stdout.columns||120,n="";for(let a=0;a<e.length;a++){let c=e[a];t.lineStart&&(n+=Nt,t.col=Ot,t.lineStart=!1),c===`
|
|
15
15
|
`?(n+=c,t.lineStart=!0,t.col=0,t.inEsc=!1):c==="\x1B"?(t.inEsc=!0,n+=c):t.inEsc?(n+=c,(c>="A"&&c<="Z"||c>="a"&&c<="z")&&(t.inEsc=!1)):(t.col++,n+=c,t.col>=r&&(n+=`
|
|
16
16
|
${Nt}`,t.col=Ot))}return i(n,s,o)}}var ot=class{constructor(){this._currentNode=null,this._origStdoutWrite=null,this._origStderrWrite=null,this._emitWorkflowGraphMarkers=String(process.env.ZIBBY_EMIT_GRAPH_MARKERS||"").trim()==="1"||String(process.env.ZIBBY_WORKFLOW_GRAPH_LOG_MARKERS||"").trim()==="1"}get isInsideNode(){return this._currentNode!==null}_startIntercepting(){this._origStdoutWrite=process.stdout.write.bind(process.stdout),this._origStderrWrite=process.stderr.write.bind(process.stderr);let t={lineStart:!0,col:0,inEsc:!1},e={lineStart:!0,col:0,inEsc:!1};this._outState=t,this._errState=e,process.stdout.write=vt(this._origStdoutWrite,t),process.stderr.write=vt(this._origStderrWrite,e)}_stopIntercepting(){this._origStdoutWrite&&(this._outState&&!this._outState.lineStart&&this._origStdoutWrite(`
|
|
17
17
|
`),process.stdout.write=this._origStdoutWrite),this._origStderrWrite&&(this._errState&&!this._errState.lineStart&&this._origStderrWrite(`
|
|
@@ -22,11 +22,11 @@ ${Nt}`,t.col=Ot))}return i(n,s,o)}}var ot=class{constructor(){this._currentNode=
|
|
|
22
22
|
`)):process.stdout.write.bind(process.stdout)(`${t} ${e}
|
|
23
23
|
`)}step(t){this._origStdoutWrite?this._writeDot(tt,t):process.stdout.write.bind(process.stdout)(`${Y} ${tt} ${t}
|
|
24
24
|
`)}stepInfo(t){this.step(t)}stepTool(t){this._origStdoutWrite?this._writeDot($t,t):process.stdout.write.bind(process.stdout)(`${Y} ${$t} ${t}
|
|
25
|
-
`)}stepMemory(t){let e=
|
|
26
|
-
`)}stepFail(t){this._origStdoutWrite?this._writeDot(et,
|
|
27
|
-
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${te} ${t}`),this._startIntercepting()}nodeComplete(t,e={}){this._stopIntercepting();let{duration:s,details:o}=e;if(o)for(let n of o)this._rawWrite(`${tt} ${n}`);let r=s?
|
|
25
|
+
`)}stepMemory(t){let e=_.hex("#2dd4bf")(t);this._origStdoutWrite?this._writeDot(bt,e):process.stdout.write.bind(process.stdout)(`${Y} ${bt} ${e}
|
|
26
|
+
`)}stepFail(t){this._origStdoutWrite?this._writeDot(et,_.red(t)):process.stdout.write.bind(process.stdout)(`${Y} ${et} ${_.red(t)}
|
|
27
|
+
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${te} ${t}`),this._startIntercepting()}nodeComplete(t,e={}){this._stopIntercepting();let{duration:s,details:o}=e;if(o)for(let n of o)this._rawWrite(`${tt} ${n}`);let r=s?_.dim(` ${kt(s)}`):"";this._rawWrite(`${It} ${_.green("done")}${r}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}nodeFailed(t,e,s={}){this._stopIntercepting();let{duration:o}=s,r=o?_.dim(` ${kt(o)}`):"";this._rawWrite(`${et} ${_.red(e)}`),this._rawWrite(`${It} ${_.red("failed")}${r}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}route(t,e){this._rawWrite(_.dim(` ${t} \u2192 ${e}`)),this._rawWrite("")}graphComplete(){}},w=new ot;var V=".zibby/output",Tt="sessions",L=".session-info.json",At=".zibby-stop";var xt=["CI_JOB_ID","GITHUB_RUN_ID","CIRCLE_WORKFLOW_ID","BUILD_ID"];var F=class{constructor(t){if(this.config=t,this.name=t.name,this.prompt=t.prompt,this.outputSchema=t.outputSchema,!this.outputSchema&&!t._isCustomCode)throw new Error(`Node '${this.name}' must define outputSchema (Zod schema). This defines the contract for what the node returns to state.`);this.isZodSchema=this.outputSchema&&typeof this.outputSchema._def<"u",this.parser=t.outputSchema&&!this.isZodSchema?new J(t.outputSchema):null,this.retries=t.retries||0,this.onComplete=t.onComplete,this.customExecute=t.execute}async execute(t,e){let s=()=>e&&typeof e.getAll=="function"?e.getAll():t,o=l=>e&&typeof e.get=="function"?e.get(l):t?.[l];if(typeof this.customExecute=="function"){S.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let l=await this.customExecute(t);return typeof l=="object"&&l!==null&&l.success===!1?{success:!1,error:l.error||"Node execution failed",raw:l.raw||null}:this.isZodSchema?(S.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(l),raw:null}):{success:!0,output:l,raw:null}}catch(l){return S.error(`[workflow] node '${this.name}' failed: ${l.message}`),l.name==="ZodError"&&S.error(`Schema errors: ${JSON.stringify(l.issues||l.errors,null,2)}`),{success:!1,error:l.message,raw:null}}}let r=typeof this.prompt=="function"?this.prompt(s()):this.prompt,n=o("_skillHints");n&&(r=`${n}
|
|
28
28
|
|
|
29
|
-
${r}`);let a=s(),c=a.cwd||process.cwd(),d=a.sessionPath;try{if(d){let l=ut(d,L);if(Bt(l)){let E=JSON.parse(Wt(l,"utf-8"));E.currentNode=this.name,lt(l,JSON.stringify(E,null,2),"utf-8")}let m=ut(d,"..",L);if(Bt(m))try{let E=JSON.parse(Wt(m,"utf-8"));E.currentNode=this.name,lt(m,JSON.stringify(E,null,2),"utf-8")}catch{}}}catch(l){
|
|
29
|
+
${r}`);let a=s(),c=a.cwd||process.cwd(),d=a.sessionPath;try{if(d){let l=ut(d,L);if(Bt(l)){let E=JSON.parse(Wt(l,"utf-8"));E.currentNode=this.name,lt(l,JSON.stringify(E,null,2),"utf-8")}let m=ut(d,"..",L);if(Bt(m))try{let E=JSON.parse(Wt(m,"utf-8"));E.currentNode=this.name,lt(m,JSON.stringify(E,null,2),"utf-8")}catch{}}}catch(l){S.debug(`[workflow] could not update session info: ${l.message}`)}let h=null;for(let l=0;l<=this.retries;l++)try{S.debug(`[workflow] node '${this.name}' attempt ${l}`);let m=s().config||{},E=m.agents||{},x=this.config.agent??E[this.name]??null,P={state:s()};x&&(P.preferredAgent=x);let T={workspace:c,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:d,config:m,nodeName:this.name,timeout:this.config?.timeout||3e5},p=t?._coreInvokeAgent;p||(p=(await Promise.resolve().then(()=>(ct(),at))).invokeAgent);let $=await p(r,P,T),y,O;if(typeof $=="string"?(y=$,O=null):$.structured?(y=$.raw||JSON.stringify($.structured,null,2),O=$.structured):(y=$.raw||JSON.stringify($,null,2),O=$.extracted||null),d)try{let u=ut(d,this.name,"raw_stream_output.txt");le(ue(u),{recursive:!0}),lt(u,typeof y=="string"?y:JSON.stringify(y),"utf-8")}catch(u){S.debug(`[workflow] could not save raw output: ${u.message}`)}if(this.isZodSchema&&O){S.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(O,null,2)}`);let u=O;if(typeof this.onComplete=="function")try{u=await this.onComplete(s(),O)}catch(W){S.warn(`[workflow] onComplete hook failed: ${W.message}`)}return{success:!0,output:u,raw:y}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(s(),{raw:y}),raw:y}}catch(u){throw new Error(`onComplete failed: ${u.message}`,{cause:u})}if(this.parser){let u=this.parser.parse(y);return S.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(u,null,2)}`),w.step("Output parsed"),{success:!0,output:u,raw:y}}return{success:!0,output:y,raw:y}}catch(m){h=m,l<this.retries&&S.info(`[workflow] node '${this.name}' failed, retrying (${l+1}/${this.retries})\u2026`)}return{success:!1,error:h.message,raw:null}}},z=class extends F{constructor(t){super({...t,_isCustomCode:!0}),this.condition=t.condition}async execute(t,e){let s=e&&typeof e.getAll=="function"?e.getAll():t;return{success:!0,output:{nextNode:this.condition(s)},raw:null}}};import{existsSync as Mt,readFileSync as de}from"node:fs";import{join as dt,dirname as Lt}from"node:path";var q=class{static async loadContext(t,e,s={}){let o={},r=s.filenames||["CONTEXT.md","AGENTS.md"];if(t){let a=Lt(dt(e,t));for(let c of r){let d=await this.findAndMergeContextFiles(c,a,e);if(d){let h=c.replace(/\.[^.]+$/,"").toLowerCase();o[h]=d}}}let n=s.discovery||{};for(let[a,c]of Object.entries(n))try{let d=dt(e,c);Mt(d)&&(o[a]=await this.loadFile(d))}catch(d){console.warn(`[workflow] could not load context '${a}' from '${c}': ${d.message}`)}return o}static async findAndMergeContextFiles(t,e,s){let o=[],r=e;for(;r.startsWith(s);){let n=dt(r,t);if(Mt(n))try{o.unshift(await this.loadFile(n))}catch(c){console.warn(`[workflow] could not load ${t} from ${n}: ${c.message}`)}let a=Lt(r);if(a===r)break;r=a}return o.length===0?null:o.every(n=>typeof n=="string")?o.join(`
|
|
30
30
|
|
|
31
31
|
---
|
|
32
32
|
|
|
@@ -34,4 +34,4 @@ ${r}`);let a=s(),c=a.cwd||process.cwd(),d=a.sessionPath;try{if(d){let l=ut(d,L);
|
|
|
34
34
|
`).slice(2,14).join(`
|
|
35
35
|
`);console.log(`[zibby:session] stack (${i}):
|
|
36
36
|
${h}`)}}function ge(){return process.env.ZIBBY_TRUST_SESSION_ENV==="1"||process.env.ZIBBY_TRUST_SESSION_ENV==="true"||process.env.ZIBBY_KEEP_SESSION_ENV==="1"||process.env.ZIBBY_KEEP_SESSION_ENV==="true"}function Se(){if(!(process.env.ZIBBY_PIN_SESSION_PATH==="1"||process.env.ZIBBY_PIN_SESSION_PATH==="true"))return;let t=process.env.ZIBBY_SESSION_PATH;if(!(t==null||String(t).trim()===""))try{return Zt(String(t).trim())}catch{return String(t).trim()}}function _e(){ge()||(delete process.env.ZIBBY_SESSION_PATH,delete process.env.ZIBBY_SESSION_ID)}function we({sessionPath:i,sessionId:t}){i&&typeof i=="string"&&(process.env.ZIBBY_SESSION_PATH=i),t!=null&&String(t).trim()!==""&&(process.env.ZIBBY_SESSION_ID=String(t).trim())}function ye(i={}){let t=xt.map(r=>process.env[r]).find(Boolean),e=Math.random().toString(36).slice(2,6),s=t||`${Date.now()}_${e}`,o=i.paths?.sessionPrefix;return o?`${o}_${s}`:s}function Ee({cwd:i=process.cwd(),config:t={},initialState:e={},traceFrom:s="resolveWorkflowSession"}={}){let o=e.sessionPath,r=e.sessionTimestamp,n="initialState.sessionPath";if(!o&&process.env.ZIBBY_SESSION_PATH)try{let d=Zt(String(process.env.ZIBBY_SESSION_PATH));d&&(o=d,n="ZIBBY_SESSION_PATH")}catch{}let a;if(o)a=String(o).split(/[/\\]/).filter(Boolean).pop(),r==null&&(r=Date.now());else{let d=process.env.ZIBBY_SESSION_ID&&String(process.env.ZIBBY_SESSION_ID).trim();if(d)a=d,n="ZIBBY_SESSION_ID";else{let l=t.sessionId!=null?String(t.sessionId).trim():"";l&&l!=="last"?(a=l,n="config.sessionId"):(a=ye(t),n="generated")}r=r??Date.now();let h=t.paths?.output||V;o=R(i,h,Tt,a)}let c=!pt(o);return c&&Yt(o,{recursive:!0}),(c||n!=="initialState.sessionPath")&&me({traceFrom:s,sessionId:a,sessionPath:o,idSource:n,mkdirFresh:c}),we({sessionPath:o,sessionId:a}),{sessionPath:o,sessionId:a,sessionTimestamp:r}}var jt=class{constructor(t={}){this.nodes=new Map,this.edges=new Map,this.entryPoint=null,this.middleware=Array.isArray(t.middleware)?[...t.middleware]:[],t.nodeMiddleware&&this.middleware.push(t.nodeMiddleware),this.nodeTypeMap=new Map,this.conditionalCodeMap=new Map,this.stateSchema=t.stateSchema||null,this.nodePrompts=new Map,this.nodeOptions=new Map,this._invokeAgent=t.invokeAgent||null,this._compiledPrompts=new Map}setStateSchema(t){return this.stateSchema=t,this}getStateSchema(){return this.stateSchema}addNode(t,e,s={}){let o=e instanceof F?e:new F(e);return o.name=t,this.nodes.set(t,o),s.prompt&&this.nodePrompts.set(t,s.prompt),Object.keys(s).length>0&&this.nodeOptions.set(t,s),this}addConditionalNode(t,e){return this.nodes.set(t,new z({...e,name:t})),this}addEdge(t,e){return this.edges.set(t,e),this}setNodeType(t,e){return this.nodeTypeMap.set(t,e),this}addConditionalEdges(t,e,{labels:s}={}){return this.edges.set(t,{conditional:!0,routes:e,labels:s}),typeof e=="function"&&this.conditionalCodeMap.set(t,e.toString()),this}setEntryPoint(t){return this.entryPoint=t,this}use(t){return typeof t=="function"&&this.middleware.push(t),this}_composeMiddleware(t,e,s,o,r){let n=s;for(let a=t.length-1;a>=0;a--){let c=t[a],d=n;n=()=>c(e,d,o,r)}return n()}serialize(){let t=[],e={};for(let[r,n]of this.nodes){let a=this.nodeTypeMap.get(r)||r;t.push({id:r,type:a,data:{nodeType:a,label:r}});let c={};n._isCustomCode&&typeof n.execute=="function"&&(c.customCode=n.execute.toString());let d=this.nodePrompts.get(r);if(d&&(c.prompt=d),typeof n.customExecute=="function"&&(c.executeCode=n.customExecute.toString()),n.outputSchema)try{if(typeof n.outputSchema._def<"u"){let l=Ft(n.outputSchema,{target:"openApi3"});c.outputSchema={jsonSchema:l,variables:this._flattenJsonSchemaToVariables(l)}}else c.outputSchema={schema:n.outputSchema}}catch(l){console.warn(`[workflow] failed to convert schema for ${r}:`,l.message)}let h=(this.resolvedToolsMap||{})[r];h?.toolIds&&(c.tools=h.toolIds),Object.keys(c).length>0&&(e[r]=c)}let s=[];for(let[r,n]of this.edges)if(typeof n=="string")s.push({source:r,target:n});else if(n.conditional){let a=this.conditionalCodeMap.get(r)||n.routes.toString(),c=this._inferConditionalTargets(n.routes),d=n.labels||{};for(let h of c){let l={source:r,target:h,data:{conditionalCode:a}};d[h]&&(l.label=d[h]),s.push(l)}}let o=null;if(this.stateSchema)try{o=Ft(this.stateSchema,{target:"openApi3"})}catch{o=this.stateSchema}return{nodes:t,edges:s,nodeConfigs:e,stateSchema:o}}_inferConditionalTargets(t){let e=t.toString(),s=new Set,o=/return\s+['"]([^'"]+)['"]/g,r;for(;(r=o.exec(e))!==null;)s.add(r[1]);return[...s]}_flattenJsonSchemaToVariables(t,e=""){let s=t;if(t.$ref&&t.definitions){let o=t.$ref.replace("#/definitions/","");s=t.definitions[o]||t}return this._flattenSchema(s,e)}_flattenSchema(t,e=""){if(!t||typeof t!="object")return[];let s=[],o=t.properties||{},r=t.required||[];for(let[n,a]of Object.entries(o)){let c=e?`${e}.${n}`:n;s.push({path:c,type:a.type||"unknown",label:a.description||this._formatLabel(n),optional:!r.includes(n)}),a.type==="object"&&a.properties&&s.push(...this._flattenSchema(a,c)),a.type==="array"&&a.items?.type==="object"&&a.items.properties&&s.push(...this._flattenSchema(a.items,`${c}[]`))}return s}_formatLabel(t){return t.replace(/([A-Z])/g," $1").replace(/^./,e=>e.toUpperCase()).trim()}_summarizeNodeOutput(t,e){if(!e||typeof e!="object")return[];let s=[];e.success!==void 0&&s.push(`Result: ${e.success?"passed":"failed"}`);for(let[o,r]of Object.entries(e))if(!(o==="success"||o==="raw"||o==="nextNode")){if(typeof r=="string"&&r.length<=80)s.push(`${o}: ${r}`);else if(Array.isArray(r)){let n=r.length,a=r.filter(d=>d?.passed===!0).length,c=r.some(d=>d?.passed!==void 0);s.push(c?`${o}: ${a}/${n} passed${n-a?`, ${n-a} failed`:""}`:`${o}: ${n} items`)}if(s.length>=4)break}return s}async run(t,e={},s={}){if(!this.entryPoint)throw new Error("No entry point set for graph");let o=new AbortController;s.signal&&(s.signal.aborted?o.abort():s.signal.addEventListener("abort",()=>o.abort(),{once:!0}));let r=s.strategyAbortTimeoutMs??e.config?.strategyAbortTimeoutMs??5e3,n=e.cwd||process.cwd();fe({path:R(n,".env")});let a=e.config||{};if(!a||Object.keys(a).length===0)try{let f=R(n,".zibby.config.js");pt(f)&&(a=(await import(f)).default||{})}catch{}process.env.EXECUTION_ID&&!a.agent?.strictMode&&(a.agent={...a.agent,strictMode:!0});let c=e.agentType;if(!c){let f=a?.agent;f?.provider?c=f.provider:f?.gemini?c="gemini":f?.claude?c="claude":f?.cursor?c="cursor":f?.codex?c="codex":c=process.env.AGENT_TYPE||"cursor"}let d=e.contextConfig||t?.config?.contextConfig||t?.config?.context||a?.context||{};if(this.stateSchema){let f=this.stateSchema.safeParse(e);if(!f.success){let N=f.error.issues.map(I=>`${I.path.join(".")}: ${I.message}`);throw console.error("\u274C Initial state validation failed:"),N.forEach(I=>console.error(` - ${I}`)),new Error(`State validation failed: ${N.join(", ")}`)}w.step("State validated against schema")}let h=Se(),l=e.sessionPath||h;l||_e();let{sessionPath:m,sessionTimestamp:E,sessionId:x}=Ee({cwd:n,config:a,traceFrom:"WorkflowGraph.run",initialState:{sessionPath:l,sessionTimestamp:e.sessionTimestamp}});w.step(`Session ${x}`);let P=await q.loadContext(e.specPath||"",n,d);Object.keys(P).length>0&&w.step(`Context loaded: ${Object.keys(P).join(", ")}`);let T=e.outputPath;!T&&e.specPath&&(t?.calculateOutputPath?T=t.calculateOutputPath(e.specPath):console.warn(`\u26A0\uFE0F outputPath not resolved (specPath=${e.specPath})`));let p=new H({...e,config:a,agentType:c,outputPath:T,sessionPath:m,sessionTimestamp:E,context:P,resolvedTools:this.resolvedToolsMap||{},_signal:o.signal}),$=new Map;try{await import("@zibby/skills")}catch{}let{getSkill:y}=await Promise.resolve().then(()=>(nt(),Ct)),O=new Set;for(let[,f]of this.nodes)for(let N of f.config?.skills||[])O.add(N);for(let f of O){let N=y(f);if(typeof N?.middleware=="function")try{let I=await N.middleware();typeof I=="function"&&$.set(f,I)}catch{}}let u=this.entryPoint,W=[],ft=a?.recursionLimit??100,Ut=0;try{for(;u&&u!=="END";){if(++Ut>ft)throw new Error(`Workflow exceeded recursion limit (${ft}) \u2014 likely a cyclic conditional route. Set config.recursionLimit if you need a higher cap.`);let N=R(m,At);if(pt(N)){try{pe(N)}catch{}o.abort()}if(o.signal.aborted)return console.warn(`
|
|
37
|
-
\u{1F6D1} External stop requested \u2014 ending workflow.`),w.step("Workflow stopped externally"),{success:!0,state:p.getAll(),executionLog:W,stoppedExternally:!0};let I=this.nodes.get(u);if(!I)throw new Error(`Node '${u}' not found in graph`);let ht=JSON.stringify({sessionPath:m,sessionTimestamp:E,currentNode:u,createdAt:new Date().toISOString(),config:p.get("config")}),Gt=R(m,L);Dt(Gt,ht,"utf-8");let mt=p.get("config")?.paths?.output||V,Ht=R(n,mt,L);Yt(R(n,mt),{recursive:!0});try{Dt(Ht,ht,"utf-8")}catch{}let gt=e.onPipelineProgress;if(typeof gt=="function")try{gt({cwd:n,sessionPath:m,sessionId:x,outputBase:p.get("config")?.paths?.output||V,currentNode:u})}catch{}let Jt=(this.resolvedToolsMap||{})[u]||null;p.set("_currentNodeTools",Jt);let Kt=p.get("nodeConfigs")||{};p.set("_currentNodeConfig",Kt[u]||{}),w.nodeStart(u);let St=Date.now(),Z=this.nodePrompts.get(u);if(!this._invokeAgent){let k=await Promise.resolve().then(()=>(ct(),at));this._invokeAgent=k.invokeAgent}let Vt=this._invokeAgent,_t=async(k,v,
|
|
37
|
+
\u{1F6D1} External stop requested \u2014 ending workflow.`),w.step("Workflow stopped externally"),{success:!0,state:p.getAll(),executionLog:W,stoppedExternally:!0};let I=this.nodes.get(u);if(!I)throw new Error(`Node '${u}' not found in graph`);let ht=JSON.stringify({sessionPath:m,sessionTimestamp:E,currentNode:u,createdAt:new Date().toISOString(),config:p.get("config")}),Gt=R(m,L);Dt(Gt,ht,"utf-8");let mt=p.get("config")?.paths?.output||V,Ht=R(n,mt,L);Yt(R(n,mt),{recursive:!0});try{Dt(Ht,ht,"utf-8")}catch{}let gt=e.onPipelineProgress;if(typeof gt=="function")try{gt({cwd:n,sessionPath:m,sessionId:x,outputBase:p.get("config")?.paths?.output||V,currentNode:u})}catch{}let Jt=(this.resolvedToolsMap||{})[u]||null;p.set("_currentNodeTools",Jt);let Kt=p.get("nodeConfigs")||{};p.set("_currentNodeConfig",Kt[u]||{}),w.nodeStart(u);let St=Date.now(),Z=this.nodePrompts.get(u);if(!this._invokeAgent){let k=await Promise.resolve().then(()=>(ct(),at));this._invokeAgent=k.invokeAgent}let Vt=this._invokeAgent,_t=async(k,v,g={})=>{let b=Vt(k,v,{...g,signal:o.signal});return b.catch(()=>{}),o.signal.aborted?b:Promise.race([b,new Promise((B,M)=>{let A=()=>{setTimeout(()=>{let j=new Error(`Strategy ignored AbortSignal \u2014 engine deadman fired after ${r}ms`);j.name="AbortError",M(j)},r)};o.signal.addEventListener("abort",A,{once:!0})})])},wt={state:p,invokeAgent:async(k={},v={})=>{let g=v.prompt||"";if(Z){let b=this._compiledPrompts.get(u);b||(b=he.compile(Z,{noEscape:!0}),this._compiledPrompts.set(u,b));try{g=b(k)}catch(B){throw console.error(`\u274C Template rendering failed for node '${u}':`,B.message),new Error(`Template rendering failed: ${B.message}`,{cause:B})}}else if(!g)throw new Error(`No prompt template configured for node '${u}' and no prompt provided in options`);return _t(g,{state:p.getAll(),images:v.images||[]},{model:v.model||p.get("model"),workspace:p.get("workspace"),schema:v.schema,...v,signal:o.signal})},_coreInvokeAgent:_t,agent:t,nodeId:u,promptTemplate:Z,getPromptTemplate:()=>Z,...p.getAll()};try{let k=(I.config?.skills||[]).map(A=>$.get(A)).filter(Boolean),v=[...this.middleware,...k],g;v.length>0?g=await this._composeMiddleware(v,u,async()=>I.execute(wt,p),p.getAll(),p):g=await I.execute(wt,p);let b=Date.now()-St;if(W.push({node:u,success:g.success,duration:b,timestamp:new Date().toISOString()}),!g.success){if(o.signal.aborted)return w.step("Workflow stopped externally"),{success:!0,state:p.getAll(),executionLog:W,stoppedExternally:!0};p.append("errors",{node:u,error:g.error});let A=I.config?.retries||0,j=`${u}_retries`,U=p.getAll()[j]||0;if(U<A){w.stepInfo(`Retrying (attempt ${U+1}/${A})`),p.update({[j]:U+1,[`${u}_raw`]:g.raw});continue}throw w.nodeFailed(u,g.error,{duration:b}),new Error(`Node '${u}' failed after ${U} attempts: ${g.error}`)}p.update({[u]:g.output});let B=this._summarizeNodeOutput(u,g.output);w.nodeComplete(u,{duration:b,details:B});let M=this.edges.get(u);if(!M)u="END";else if(M.conditional){let A=M.routes(p.getAll());w.route(u,A),u=A}else u=M}catch(k){throw w.isInsideNode&&w.nodeFailed(u,k.message,{duration:Date.now()-St}),p.set("failed",!0),p.set("failedAt",u),k}}w.graphComplete();let f={success:!0,state:p.getAll(),executionLog:W};return t&&typeof t.onComplete=="function"&&await t.onComplete(f),f}finally{if(t&&typeof t.cleanup=="function")try{await t.cleanup()}catch(f){console.warn(`[workflow] agent.cleanup() failed: ${f.message}`)}}}};export{jt as WorkflowGraph,_e as clearInheritedSessionEnvForFreshRun,ye as generateWorkflowSessionId,Se as readPinnedSessionPathFromEnv,Ee as resolveWorkflowSession,ge as shouldTrustInheritedSessionEnv,we as syncProcessEnvToSession};
|
package/dist/index.js
CHANGED
|
@@ -11,7 +11,7 @@ PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PRE
|
|
|
11
11
|
${g}
|
|
12
12
|
`),_.debug(`[workflow] prompt length: ${u.length} chars`),s.invoke(u,h)}var _e,W,Y=ne(()=>{we();j();q();_e=Symbol.for("@zibby/agent-workflow.strategies");globalThis[_e]||(globalThis[_e]=[]);W=globalThis[_e]});var xt=new Set(["__proto__","constructor","prototype"]);function ue(o){if(xt.has(o))throw new Error(`Invalid state key: "${o}"`)}var J=class{constructor(e={}){this._state=Object.create(null),Object.assign(this._state,{messages:[],errors:[],artifacts:{},metadata:{},...e}),this._history=[]}get(e){return this._state[e]}set(e,t){ue(e),this._history.push({...this._state}),this._state[e]=t}update(e){let t=Object.getOwnPropertyNames(e);for(let s of t)ue(s);this._history.push({...this._state});for(let s of t)this._state[s]=e[s]}append(e,t){ue(e),this._history.push({...this._state}),Array.isArray(this._state[e])||(this._state[e]=[]),this._state[e].push(t)}getAll(){return{...this._state}}rollback(){this._history.length>0&&(this._state=this._history.pop())}};var H=class{constructor(e){this.schema=e}parse(e){let t=e.match(/```json\s*([\s\S]*?)\s*```/);if(t)return this.validate(JSON.parse(t[1]));let s=[e.match(/\{[\s\S]*?\}/),e.match(/\{[\s\S]*\}/)].filter(Boolean).map(n=>n[0]);for(let n of s)try{return this.validate(JSON.parse(n))}catch(i){if(!(i instanceof SyntaxError))throw i}return this.validate({result:e.trim()})}validate(e){let t=[];for(let[s,n]of Object.entries(this.schema)){if(n.required&&!(s in e)&&t.push(`Missing required field: ${s}`),s in e&&n.type){let i=typeof e[s];i!==n.type&&t.push(`Field '${s}' expected ${n.type}, got ${i}`)}if(n.validate&&s in e){let i=n.validate(e[s]);i&&t.push(`Field '${s}': ${i}`)}}if(t.length>0)throw new Error(`Output validation failed:
|
|
13
13
|
${t.join(`
|
|
14
|
-
`)}`);return e}},bt={string:(o=!0)=>({type:"string",required:o}),number:(o=!0)=>({type:"number",required:o}),boolean:(o=!0)=>({type:"boolean",required:o}),array:(o=!0)=>({type:"object",required:o,validate:e=>Array.isArray(e)?null:"must be an array"}),enum:(o,e=!0)=>({type:"string",required:e,validate:t=>o.includes(t)?null:`must be one of: ${o.join(", ")}`})};j();import{writeFileSync as $e,readFileSync as st,existsSync as nt,mkdirSync as Lt}from"node:fs";import{join as Te,dirname as Bt}from"node:path";import
|
|
14
|
+
`)}`);return e}},bt={string:(o=!0)=>({type:"string",required:o}),number:(o=!0)=>({type:"number",required:o}),boolean:(o=!0)=>({type:"boolean",required:o}),array:(o=!0)=>({type:"object",required:o,validate:e=>Array.isArray(e)?null:"must be an array"}),enum:(o,e=!0)=>({type:"string",required:e,validate:t=>o.includes(t)?null:`must be one of: ${o.join(", ")}`})};j();import{writeFileSync as $e,readFileSync as st,existsSync as nt,mkdirSync as Lt}from"node:fs";import{join as Te,dirname as Bt}from"node:path";import N from"chalk";var Ke="__WORKFLOW_GRAPH_LOG__",V=N.gray("\u2502"),Ot=N.gray("\u250C"),je=N.gray("\u2514"),pe=N.green("\u25C6"),Ge=N.hex("#c084fc")("\u25C6"),Ue=N.hex("#2dd4bf")("\u25C6"),fe=N.red("\u25C6"),Ye=`${V} `,Ze=2;function Je(o){return o<1e3?`${o}ms`:`${(o/1e3).toFixed(1)}s`}function He(o,e){return(t,s,n)=>{if(typeof t!="string")return o(t,s,n);let i=process.stdout.columns||120,r="";for(let a=0;a<t.length;a++){let c=t[a];e.lineStart&&(r+=Ye,e.col=Ze,e.lineStart=!1),c===`
|
|
15
15
|
`?(r+=c,e.lineStart=!0,e.col=0,e.inEsc=!1):c==="\x1B"?(e.inEsc=!0,r+=c):e.inEsc?(r+=c,(c>="A"&&c<="Z"||c>="a"&&c<="z")&&(e.inEsc=!1)):(e.col++,r+=c,e.col>=i&&(r+=`
|
|
16
16
|
${Ye}`,e.col=Ze))}return o(r,s,n)}}var ie=class{constructor(){this._currentNode=null,this._origStdoutWrite=null,this._origStderrWrite=null,this._emitWorkflowGraphMarkers=String(process.env.ZIBBY_EMIT_GRAPH_MARKERS||"").trim()==="1"||String(process.env.ZIBBY_WORKFLOW_GRAPH_LOG_MARKERS||"").trim()==="1"}get isInsideNode(){return this._currentNode!==null}_startIntercepting(){this._origStdoutWrite=process.stdout.write.bind(process.stdout),this._origStderrWrite=process.stderr.write.bind(process.stderr);let e={lineStart:!0,col:0,inEsc:!1},t={lineStart:!0,col:0,inEsc:!1};this._outState=e,this._errState=t,process.stdout.write=He(this._origStdoutWrite,e),process.stderr.write=He(this._origStderrWrite,t)}_stopIntercepting(){this._origStdoutWrite&&(this._outState&&!this._outState.lineStart&&this._origStdoutWrite(`
|
|
17
17
|
`),process.stdout.write=this._origStdoutWrite),this._origStderrWrite&&(this._errState&&!this._errState.lineStart&&this._origStderrWrite(`
|
|
@@ -22,19 +22,19 @@ ${Ye}`,e.col=Ze))}return o(r,s,n)}}var ie=class{constructor(){this._currentNode=
|
|
|
22
22
|
`)):process.stdout.write.bind(process.stdout)(`${e} ${t}
|
|
23
23
|
`)}step(e){this._origStdoutWrite?this._writeDot(pe,e):process.stdout.write.bind(process.stdout)(`${V} ${pe} ${e}
|
|
24
24
|
`)}stepInfo(e){this.step(e)}stepTool(e){this._origStdoutWrite?this._writeDot(Ge,e):process.stdout.write.bind(process.stdout)(`${V} ${Ge} ${e}
|
|
25
|
-
`)}stepMemory(e){let t
|
|
26
|
-
`)}stepFail(e){this._origStdoutWrite?this._writeDot(fe
|
|
27
|
-
`)}nodeStart(e){this._currentNode=e,this._emitGraphLogMarker({phase:"node_begin",node:e}),this._rawWrite(`${Ot} ${e}`),this._startIntercepting()}nodeComplete(e,t={}){this._stopIntercepting();let{duration:s,details:n}=t;if(n)for(let r of n)this._rawWrite(`${pe} ${r}`);let i=s
|
|
25
|
+
`)}stepMemory(e){let t=N.hex("#2dd4bf")(e);this._origStdoutWrite?this._writeDot(Ue,t):process.stdout.write.bind(process.stdout)(`${V} ${Ue} ${t}
|
|
26
|
+
`)}stepFail(e){this._origStdoutWrite?this._writeDot(fe,N.red(e)):process.stdout.write.bind(process.stdout)(`${V} ${fe} ${N.red(e)}
|
|
27
|
+
`)}nodeStart(e){this._currentNode=e,this._emitGraphLogMarker({phase:"node_begin",node:e}),this._rawWrite(`${Ot} ${e}`),this._startIntercepting()}nodeComplete(e,t={}){this._stopIntercepting();let{duration:s,details:n}=t;if(n)for(let r of n)this._rawWrite(`${pe} ${r}`);let i=s?N.dim(` ${Je(s)}`):"";this._rawWrite(`${je} ${N.green("done")}${i}`),this._emitGraphLogMarker({phase:"node_end",node:e}),this._rawWrite("")}nodeFailed(e,t,s={}){this._stopIntercepting();let{duration:n}=s,i=n?N.dim(` ${Je(n)}`):"";this._rawWrite(`${fe} ${N.red(t)}`),this._rawWrite(`${je} ${N.red("failed")}${i}`),this._emitGraphLogMarker({phase:"node_end",node:e}),this._rawWrite("")}route(e,t){this._rawWrite(N.dim(` ${e} \u2192 ${t}`)),this._rawWrite("")}graphComplete(){}},$=new ie;var z=".zibby/output",he="sessions",R=".session-info.json",me=".zibby-stop",Ct="result.json",Pt="raw_stream_output.txt",Rt="events.json",Wt={BROWSER:"browser",JIRA:"jira",GITHUB:"github",SLACK:"slack",MEMORY:"memory",CHAT_MEMORY:"chat-memory",RUNNER:"runner",SKILL_INSTALLER:"skill-installer",CORE_TOOLS:"core-tools",WORKFLOW_BUILDER:"workflow-builder"},ge=["CI_JOB_ID","GITHUB_RUN_ID","CIRCLE_WORKFLOW_ID","BUILD_ID"];var L=class{constructor(e){if(this.config=e,this.name=e.name,this.prompt=e.prompt,this.outputSchema=e.outputSchema,!this.outputSchema&&!e._isCustomCode)throw new Error(`Node '${this.name}' must define outputSchema (Zod schema). This defines the contract for what the node returns to state.`);this.isZodSchema=this.outputSchema&&typeof this.outputSchema._def<"u",this.parser=e.outputSchema&&!this.isZodSchema?new H(e.outputSchema):null,this.retries=e.retries||0,this.onComplete=e.onComplete,this.customExecute=e.execute}async execute(e,t){let s=()=>t&&typeof t.getAll=="function"?t.getAll():e,n=u=>t&&typeof t.get=="function"?t.get(u):e?.[u];if(typeof this.customExecute=="function"){_.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let u=await this.customExecute(e);return typeof u=="object"&&u!==null&&u.success===!1?{success:!1,error:u.error||"Node execution failed",raw:u.raw||null}:this.isZodSchema?(_.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(u),raw:null}):{success:!0,output:u,raw:null}}catch(u){return _.error(`[workflow] node '${this.name}' failed: ${u.message}`),u.name==="ZodError"&&_.error(`Schema errors: ${JSON.stringify(u.issues||u.errors,null,2)}`),{success:!1,error:u.message,raw:null}}}let i=typeof this.prompt=="function"?this.prompt(s()):this.prompt,r=n("_skillHints");r&&(i=`${r}
|
|
28
28
|
|
|
29
|
-
${i}`);let a=s(),c=a.cwd||process.cwd(),l=a.sessionPath;try{if(l){let u=Te(l,R);if(nt(u)){let g=JSON.parse(st(u,"utf-8"));g.currentNode=this.name,$e(u,JSON.stringify(g,null,2),"utf-8")}let d=Te(l,"..",R);if(nt(d))try{let g=JSON.parse(st(d,"utf-8"));g.currentNode=this.name,$e(d,JSON.stringify(g,null,2),"utf-8")}catch{}}}catch(u){_.debug(`[workflow] could not update session info: ${u.message}`)}let h=null;for(let u=0;u<=this.retries;u++)try{_.debug(`[workflow] node '${this.name}' attempt ${u}`);let d=s().config||{},g=d.agents||{},m=this.config.agent??g[this.name]??null,S={state:s()};m&&(S.preferredAgent=m);let w={workspace:c,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:l,config:d,nodeName:this.name,timeout:this.config?.timeout||3e5},f=e?._coreInvokeAgent;f||(f=(await Promise.resolve().then(()=>(Y(),X))).invokeAgent);let
|
|
29
|
+
${i}`);let a=s(),c=a.cwd||process.cwd(),l=a.sessionPath;try{if(l){let u=Te(l,R);if(nt(u)){let g=JSON.parse(st(u,"utf-8"));g.currentNode=this.name,$e(u,JSON.stringify(g,null,2),"utf-8")}let d=Te(l,"..",R);if(nt(d))try{let g=JSON.parse(st(d,"utf-8"));g.currentNode=this.name,$e(d,JSON.stringify(g,null,2),"utf-8")}catch{}}}catch(u){_.debug(`[workflow] could not update session info: ${u.message}`)}let h=null;for(let u=0;u<=this.retries;u++)try{_.debug(`[workflow] node '${this.name}' attempt ${u}`);let d=s().config||{},g=d.agents||{},m=this.config.agent??g[this.name]??null,S={state:s()};m&&(S.preferredAgent=m);let w={workspace:c,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:l,config:d,nodeName:this.name,timeout:this.config?.timeout||3e5},f=e?._coreInvokeAgent;f||(f=(await Promise.resolve().then(()=>(Y(),X))).invokeAgent);let T=await f(i,S,w),E,A;if(typeof T=="string"?(E=T,A=null):T.structured?(E=T.raw||JSON.stringify(T.structured,null,2),A=T.structured):(E=T.raw||JSON.stringify(T,null,2),A=T.extracted||null),l)try{let p=Te(l,this.name,"raw_stream_output.txt");Lt(Bt(p),{recursive:!0}),$e(p,typeof E=="string"?E:JSON.stringify(E),"utf-8")}catch(p){_.debug(`[workflow] could not save raw output: ${p.message}`)}if(this.isZodSchema&&A){_.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(A,null,2)}`);let p=A;if(typeof this.onComplete=="function")try{p=await this.onComplete(s(),A)}catch(F){_.warn(`[workflow] onComplete hook failed: ${F.message}`)}return{success:!0,output:p,raw:E}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(s(),{raw:E}),raw:E}}catch(p){throw new Error(`onComplete failed: ${p.message}`,{cause:p})}if(this.parser){let p=this.parser.parse(E);return _.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(p,null,2)}`),$.step("Output parsed"),{success:!0,output:p,raw:E}}return{success:!0,output:E,raw:E}}catch(d){h=d,u<this.retries&&_.info(`[workflow] node '${this.name}' failed, retrying (${u+1}/${this.retries})\u2026`)}return{success:!1,error:h.message,raw:null}}},Q=class extends L{constructor(e){super({...e,_isCustomCode:!0}),this.condition=e.condition}async execute(e,t){let s=t&&typeof t.getAll=="function"?t.getAll():e;return{success:!0,output:{nextNode:this.condition(s)},raw:null}}};import{existsSync as it,readFileSync as Ft}from"node:fs";import{join as Ie,dirname as at}from"node:path";var ee=class{static async loadContext(e,t,s={}){let n={},i=s.filenames||["CONTEXT.md","AGENTS.md"];if(e){let a=at(Ie(t,e));for(let c of i){let l=await this.findAndMergeContextFiles(c,a,t);if(l){let h=c.replace(/\.[^.]+$/,"").toLowerCase();n[h]=l}}}let r=s.discovery||{};for(let[a,c]of Object.entries(r))try{let l=Ie(t,c);it(l)&&(n[a]=await this.loadFile(l))}catch(l){console.warn(`[workflow] could not load context '${a}' from '${c}': ${l.message}`)}return n}static async findAndMergeContextFiles(e,t,s){let n=[],i=t;for(;i.startsWith(s);){let r=Ie(i,e);if(it(r))try{n.unshift(await this.loadFile(r))}catch(c){console.warn(`[workflow] could not load ${e} from ${r}: ${c.message}`)}let a=at(i);if(a===i)break;i=a}return n.length===0?null:n.every(r=>typeof r=="string")?n.join(`
|
|
30
30
|
|
|
31
31
|
---
|
|
32
32
|
|
|
33
33
|
`):n.every(r=>typeof r=="object")?Object.assign({},...n):n[n.length-1]}static async loadFile(e){let t=Ft(e,"utf-8");if(e.endsWith(".json"))return JSON.parse(t);if(e.endsWith(".js")||e.endsWith(".mjs")){let{pathToFileURL:s}=await import("url"),n=await import(s(e).href);return n.default||n}return t}};import{mkdirSync as dt,existsSync as Ne,writeFileSync as ct,unlinkSync as Mt}from"node:fs";import{join as B,resolve as ut}from"node:path";import{config as Dt}from"dotenv";import{zodToJsonSchema as lt}from"zod-to-json-schema";import jt from"handlebars";function Gt({traceFrom:o,sessionId:e,sessionPath:t,idSource:s,mkdirFresh:n}){if(!(process.env.ZIBBY_SESSION_LOG==="1"||process.env.ZIBBY_SESSION_LOG==="true"))return;let r=typeof process.ppid=="number"?process.ppid:"n/a",a=`[zibby:session] from=${o} pid=${process.pid} ppid=${r} sessionId=${e} source=${s} mkdir=${n?"yes":"no"} path=${t}`;if(console.log(a),process.env.ZIBBY_TRACE_SESSION==="1"||process.env.ZIBBY_TRACE_SESSION==="true"){let h=(new Error("session trace").stack||"").split(`
|
|
34
34
|
`).slice(2,14).join(`
|
|
35
35
|
`);console.log(`[zibby:session] stack (${o}):
|
|
36
|
-
${h}`)}}function pt(){return process.env.ZIBBY_TRUST_SESSION_ENV==="1"||process.env.ZIBBY_TRUST_SESSION_ENV==="true"||process.env.ZIBBY_KEEP_SESSION_ENV==="1"||process.env.ZIBBY_KEEP_SESSION_ENV==="true"}function ft(){if(!(process.env.ZIBBY_PIN_SESSION_PATH==="1"||process.env.ZIBBY_PIN_SESSION_PATH==="true"))return;let e=process.env.ZIBBY_SESSION_PATH;if(!(e==null||String(e).trim()===""))try{return ut(String(e).trim())}catch{return String(e).trim()}}function ht(){pt()||(delete process.env.ZIBBY_SESSION_PATH,delete process.env.ZIBBY_SESSION_ID)}function mt({sessionPath:o,sessionId:e}){o&&typeof o=="string"&&(process.env.ZIBBY_SESSION_PATH=o),e!=null&&String(e).trim()!==""&&(process.env.ZIBBY_SESSION_ID=String(e).trim())}function gt(o={}){let e=ge.map(i=>process.env[i]).find(Boolean),t=Math.random().toString(36).slice(2,6),s=e||`${Date.now()}_${t}`,n=o.paths?.sessionPrefix;return n?`${n}_${s}`:s}function St({cwd:o=process.cwd(),config:e={},initialState:t={},traceFrom:s="resolveWorkflowSession"}={}){let n=t.sessionPath,i=t.sessionTimestamp,r="initialState.sessionPath";if(!n&&process.env.ZIBBY_SESSION_PATH)try{let l=ut(String(process.env.ZIBBY_SESSION_PATH));l&&(n=l,r="ZIBBY_SESSION_PATH")}catch{}let a;if(n)a=String(n).split(/[/\\]/).filter(Boolean).pop(),i==null&&(i=Date.now());else{let l=process.env.ZIBBY_SESSION_ID&&String(process.env.ZIBBY_SESSION_ID).trim();if(l)a=l,r="ZIBBY_SESSION_ID";else{let u=e.sessionId!=null?String(e.sessionId).trim():"";u&&u!=="last"?(a=u,r="config.sessionId"):(a=gt(e),r="generated")}i=i??Date.now();let h=e.paths?.output||z;n=B(o,h,he,a)}let c=!Ne(n);return c&&dt(n,{recursive:!0}),(c||r!=="initialState.sessionPath")&&Gt({traceFrom:s,sessionId:a,sessionPath:n,idSource:r,mkdirFresh:c}),mt({sessionPath:n,sessionId:a}),{sessionPath:n,sessionId:a,sessionTimestamp:i}}var te=class{constructor(e={}){this.nodes=new Map,this.edges=new Map,this.entryPoint=null,this.middleware=Array.isArray(e.middleware)?[...e.middleware]:[],e.nodeMiddleware&&this.middleware.push(e.nodeMiddleware),this.nodeTypeMap=new Map,this.conditionalCodeMap=new Map,this.stateSchema=e.stateSchema||null,this.nodePrompts=new Map,this.nodeOptions=new Map,this._invokeAgent=e.invokeAgent||null,this._compiledPrompts=new Map}setStateSchema(e){return this.stateSchema=e,this}getStateSchema(){return this.stateSchema}addNode(e,t,s={}){let n=t instanceof L?t:new L(t);return n.name=e,this.nodes.set(e,n),s.prompt&&this.nodePrompts.set(e,s.prompt),Object.keys(s).length>0&&this.nodeOptions.set(e,s),this}addConditionalNode(e,t){return this.nodes.set(e,new Q({...t,name:e})),this}addEdge(e,t){return this.edges.set(e,t),this}setNodeType(e,t){return this.nodeTypeMap.set(e,t),this}addConditionalEdges(e,t,{labels:s}={}){return this.edges.set(e,{conditional:!0,routes:t,labels:s}),typeof t=="function"&&this.conditionalCodeMap.set(e,t.toString()),this}setEntryPoint(e){return this.entryPoint=e,this}use(e){return typeof e=="function"&&this.middleware.push(e),this}_composeMiddleware(e,t,s,n,i){let r=s;for(let a=e.length-1;a>=0;a--){let c=e[a],l=r;r=()=>c(t,l,n,i)}return r()}serialize(){let e=[],t={};for(let[i,r]of this.nodes){let a=this.nodeTypeMap.get(i)||i;e.push({id:i,type:a,data:{nodeType:a,label:i}});let c={};r._isCustomCode&&typeof r.execute=="function"&&(c.customCode=r.execute.toString());let l=this.nodePrompts.get(i);if(l&&(c.prompt=l),typeof r.customExecute=="function"&&(c.executeCode=r.customExecute.toString()),r.outputSchema)try{if(typeof r.outputSchema._def<"u"){let u=lt(r.outputSchema,{target:"openApi3"});c.outputSchema={jsonSchema:u,variables:this._flattenJsonSchemaToVariables(u)}}else c.outputSchema={schema:r.outputSchema}}catch(u){console.warn(`[workflow] failed to convert schema for ${i}:`,u.message)}let h=(this.resolvedToolsMap||{})[i];h?.toolIds&&(c.tools=h.toolIds),Object.keys(c).length>0&&(t[i]=c)}let s=[];for(let[i,r]of this.edges)if(typeof r=="string")s.push({source:i,target:r});else if(r.conditional){let a=this.conditionalCodeMap.get(i)||r.routes.toString(),c=this._inferConditionalTargets(r.routes),l=r.labels||{};for(let h of c){let u={source:i,target:h,data:{conditionalCode:a}};l[h]&&(u.label=l[h]),s.push(u)}}let n=null;if(this.stateSchema)try{n=lt(this.stateSchema,{target:"openApi3"})}catch{n=this.stateSchema}return{nodes:e,edges:s,nodeConfigs:t,stateSchema:n}}_inferConditionalTargets(e){let t=e.toString(),s=new Set,n=/return\s+['"]([^'"]+)['"]/g,i;for(;(i=n.exec(t))!==null;)s.add(i[1]);return[...s]}_flattenJsonSchemaToVariables(e,t=""){let s=e;if(e.$ref&&e.definitions){let n=e.$ref.replace("#/definitions/","");s=e.definitions[n]||e}return this._flattenSchema(s,t)}_flattenSchema(e,t=""){if(!e||typeof e!="object")return[];let s=[],n=e.properties||{},i=e.required||[];for(let[r,a]of Object.entries(n)){let c=t?`${t}.${r}`:r;s.push({path:c,type:a.type||"unknown",label:a.description||this._formatLabel(r),optional:!i.includes(r)}),a.type==="object"&&a.properties&&s.push(...this._flattenSchema(a,c)),a.type==="array"&&a.items?.type==="object"&&a.items.properties&&s.push(...this._flattenSchema(a.items,`${c}[]`))}return s}_formatLabel(e){return e.replace(/([A-Z])/g," $1").replace(/^./,t=>t.toUpperCase()).trim()}_summarizeNodeOutput(e,t){if(!t||typeof t!="object")return[];let s=[];t.success!==void 0&&s.push(`Result: ${t.success?"passed":"failed"}`);for(let[n,i]of Object.entries(t))if(!(n==="success"||n==="raw"||n==="nextNode")){if(typeof i=="string"&&i.length<=80)s.push(`${n}: ${i}`);else if(Array.isArray(i)){let r=i.length,a=i.filter(l=>l?.passed===!0).length,c=i.some(l=>l?.passed!==void 0);s.push(c?`${n}: ${a}/${r} passed${r-a?`, ${r-a} failed`:""}`:`${n}: ${r} items`)}if(s.length>=4)break}return s}async run(e,t={},s={}){if(!this.entryPoint)throw new Error("No entry point set for graph");let n=new AbortController;s.signal&&(s.signal.aborted?n.abort():s.signal.addEventListener("abort",()=>n.abort(),{once:!0}));let i=s.strategyAbortTimeoutMs??t.config?.strategyAbortTimeoutMs??5e3,r=t.cwd||process.cwd();Dt({path:B(r,".env")});let a=t.config||{};if(!a||Object.keys(a).length===0)try{let y=B(r,".zibby.config.js");Ne(y)&&(a=(await import(y)).default||{})}catch{}process.env.EXECUTION_ID&&!a.agent?.strictMode&&(a.agent={...a.agent,strictMode:!0});let c=t.agentType;if(!c){let y=a?.agent;y?.provider?c=y.provider:y?.gemini?c="gemini":y?.claude?c="claude":y?.cursor?c="cursor":y?.codex?c="codex":c=process.env.AGENT_TYPE||"cursor"}let l=t.contextConfig||e?.config?.contextConfig||e?.config?.context||a?.context||{};if(this.stateSchema){let y=this.stateSchema.safeParse(t);if(!y.success){let x=y.error.issues.map(v=>`${v.path.join(".")}: ${v.message}`);throw console.error("\u274C Initial state validation failed:"),x.forEach(v=>console.error(` - ${v}`)),new Error(`State validation failed: ${x.join(", ")}`)}
|
|
37
|
-
\u{1F6D1} External stop requested \u2014 ending workflow.`)
|
|
36
|
+
${h}`)}}function pt(){return process.env.ZIBBY_TRUST_SESSION_ENV==="1"||process.env.ZIBBY_TRUST_SESSION_ENV==="true"||process.env.ZIBBY_KEEP_SESSION_ENV==="1"||process.env.ZIBBY_KEEP_SESSION_ENV==="true"}function ft(){if(!(process.env.ZIBBY_PIN_SESSION_PATH==="1"||process.env.ZIBBY_PIN_SESSION_PATH==="true"))return;let e=process.env.ZIBBY_SESSION_PATH;if(!(e==null||String(e).trim()===""))try{return ut(String(e).trim())}catch{return String(e).trim()}}function ht(){pt()||(delete process.env.ZIBBY_SESSION_PATH,delete process.env.ZIBBY_SESSION_ID)}function mt({sessionPath:o,sessionId:e}){o&&typeof o=="string"&&(process.env.ZIBBY_SESSION_PATH=o),e!=null&&String(e).trim()!==""&&(process.env.ZIBBY_SESSION_ID=String(e).trim())}function gt(o={}){let e=ge.map(i=>process.env[i]).find(Boolean),t=Math.random().toString(36).slice(2,6),s=e||`${Date.now()}_${t}`,n=o.paths?.sessionPrefix;return n?`${n}_${s}`:s}function St({cwd:o=process.cwd(),config:e={},initialState:t={},traceFrom:s="resolveWorkflowSession"}={}){let n=t.sessionPath,i=t.sessionTimestamp,r="initialState.sessionPath";if(!n&&process.env.ZIBBY_SESSION_PATH)try{let l=ut(String(process.env.ZIBBY_SESSION_PATH));l&&(n=l,r="ZIBBY_SESSION_PATH")}catch{}let a;if(n)a=String(n).split(/[/\\]/).filter(Boolean).pop(),i==null&&(i=Date.now());else{let l=process.env.ZIBBY_SESSION_ID&&String(process.env.ZIBBY_SESSION_ID).trim();if(l)a=l,r="ZIBBY_SESSION_ID";else{let u=e.sessionId!=null?String(e.sessionId).trim():"";u&&u!=="last"?(a=u,r="config.sessionId"):(a=gt(e),r="generated")}i=i??Date.now();let h=e.paths?.output||z;n=B(o,h,he,a)}let c=!Ne(n);return c&&dt(n,{recursive:!0}),(c||r!=="initialState.sessionPath")&&Gt({traceFrom:s,sessionId:a,sessionPath:n,idSource:r,mkdirFresh:c}),mt({sessionPath:n,sessionId:a}),{sessionPath:n,sessionId:a,sessionTimestamp:i}}var te=class{constructor(e={}){this.nodes=new Map,this.edges=new Map,this.entryPoint=null,this.middleware=Array.isArray(e.middleware)?[...e.middleware]:[],e.nodeMiddleware&&this.middleware.push(e.nodeMiddleware),this.nodeTypeMap=new Map,this.conditionalCodeMap=new Map,this.stateSchema=e.stateSchema||null,this.nodePrompts=new Map,this.nodeOptions=new Map,this._invokeAgent=e.invokeAgent||null,this._compiledPrompts=new Map}setStateSchema(e){return this.stateSchema=e,this}getStateSchema(){return this.stateSchema}addNode(e,t,s={}){let n=t instanceof L?t:new L(t);return n.name=e,this.nodes.set(e,n),s.prompt&&this.nodePrompts.set(e,s.prompt),Object.keys(s).length>0&&this.nodeOptions.set(e,s),this}addConditionalNode(e,t){return this.nodes.set(e,new Q({...t,name:e})),this}addEdge(e,t){return this.edges.set(e,t),this}setNodeType(e,t){return this.nodeTypeMap.set(e,t),this}addConditionalEdges(e,t,{labels:s}={}){return this.edges.set(e,{conditional:!0,routes:t,labels:s}),typeof t=="function"&&this.conditionalCodeMap.set(e,t.toString()),this}setEntryPoint(e){return this.entryPoint=e,this}use(e){return typeof e=="function"&&this.middleware.push(e),this}_composeMiddleware(e,t,s,n,i){let r=s;for(let a=e.length-1;a>=0;a--){let c=e[a],l=r;r=()=>c(t,l,n,i)}return r()}serialize(){let e=[],t={};for(let[i,r]of this.nodes){let a=this.nodeTypeMap.get(i)||i;e.push({id:i,type:a,data:{nodeType:a,label:i}});let c={};r._isCustomCode&&typeof r.execute=="function"&&(c.customCode=r.execute.toString());let l=this.nodePrompts.get(i);if(l&&(c.prompt=l),typeof r.customExecute=="function"&&(c.executeCode=r.customExecute.toString()),r.outputSchema)try{if(typeof r.outputSchema._def<"u"){let u=lt(r.outputSchema,{target:"openApi3"});c.outputSchema={jsonSchema:u,variables:this._flattenJsonSchemaToVariables(u)}}else c.outputSchema={schema:r.outputSchema}}catch(u){console.warn(`[workflow] failed to convert schema for ${i}:`,u.message)}let h=(this.resolvedToolsMap||{})[i];h?.toolIds&&(c.tools=h.toolIds),Object.keys(c).length>0&&(t[i]=c)}let s=[];for(let[i,r]of this.edges)if(typeof r=="string")s.push({source:i,target:r});else if(r.conditional){let a=this.conditionalCodeMap.get(i)||r.routes.toString(),c=this._inferConditionalTargets(r.routes),l=r.labels||{};for(let h of c){let u={source:i,target:h,data:{conditionalCode:a}};l[h]&&(u.label=l[h]),s.push(u)}}let n=null;if(this.stateSchema)try{n=lt(this.stateSchema,{target:"openApi3"})}catch{n=this.stateSchema}return{nodes:e,edges:s,nodeConfigs:t,stateSchema:n}}_inferConditionalTargets(e){let t=e.toString(),s=new Set,n=/return\s+['"]([^'"]+)['"]/g,i;for(;(i=n.exec(t))!==null;)s.add(i[1]);return[...s]}_flattenJsonSchemaToVariables(e,t=""){let s=e;if(e.$ref&&e.definitions){let n=e.$ref.replace("#/definitions/","");s=e.definitions[n]||e}return this._flattenSchema(s,t)}_flattenSchema(e,t=""){if(!e||typeof e!="object")return[];let s=[],n=e.properties||{},i=e.required||[];for(let[r,a]of Object.entries(n)){let c=t?`${t}.${r}`:r;s.push({path:c,type:a.type||"unknown",label:a.description||this._formatLabel(r),optional:!i.includes(r)}),a.type==="object"&&a.properties&&s.push(...this._flattenSchema(a,c)),a.type==="array"&&a.items?.type==="object"&&a.items.properties&&s.push(...this._flattenSchema(a.items,`${c}[]`))}return s}_formatLabel(e){return e.replace(/([A-Z])/g," $1").replace(/^./,t=>t.toUpperCase()).trim()}_summarizeNodeOutput(e,t){if(!t||typeof t!="object")return[];let s=[];t.success!==void 0&&s.push(`Result: ${t.success?"passed":"failed"}`);for(let[n,i]of Object.entries(t))if(!(n==="success"||n==="raw"||n==="nextNode")){if(typeof i=="string"&&i.length<=80)s.push(`${n}: ${i}`);else if(Array.isArray(i)){let r=i.length,a=i.filter(l=>l?.passed===!0).length,c=i.some(l=>l?.passed!==void 0);s.push(c?`${n}: ${a}/${r} passed${r-a?`, ${r-a} failed`:""}`:`${n}: ${r} items`)}if(s.length>=4)break}return s}async run(e,t={},s={}){if(!this.entryPoint)throw new Error("No entry point set for graph");let n=new AbortController;s.signal&&(s.signal.aborted?n.abort():s.signal.addEventListener("abort",()=>n.abort(),{once:!0}));let i=s.strategyAbortTimeoutMs??t.config?.strategyAbortTimeoutMs??5e3,r=t.cwd||process.cwd();Dt({path:B(r,".env")});let a=t.config||{};if(!a||Object.keys(a).length===0)try{let y=B(r,".zibby.config.js");Ne(y)&&(a=(await import(y)).default||{})}catch{}process.env.EXECUTION_ID&&!a.agent?.strictMode&&(a.agent={...a.agent,strictMode:!0});let c=t.agentType;if(!c){let y=a?.agent;y?.provider?c=y.provider:y?.gemini?c="gemini":y?.claude?c="claude":y?.cursor?c="cursor":y?.codex?c="codex":c=process.env.AGENT_TYPE||"cursor"}let l=t.contextConfig||e?.config?.contextConfig||e?.config?.context||a?.context||{};if(this.stateSchema){let y=this.stateSchema.safeParse(t);if(!y.success){let x=y.error.issues.map(v=>`${v.path.join(".")}: ${v.message}`);throw console.error("\u274C Initial state validation failed:"),x.forEach(v=>console.error(` - ${v}`)),new Error(`State validation failed: ${x.join(", ")}`)}$.step("State validated against schema")}let h=ft(),u=t.sessionPath||h;u||ht();let{sessionPath:d,sessionTimestamp:g,sessionId:m}=St({cwd:r,config:a,traceFrom:"WorkflowGraph.run",initialState:{sessionPath:u,sessionTimestamp:t.sessionTimestamp}});$.step(`Session ${m}`);let S=await ee.loadContext(t.specPath||"",r,l);Object.keys(S).length>0&&$.step(`Context loaded: ${Object.keys(S).join(", ")}`);let w=t.outputPath;!w&&t.specPath&&(e?.calculateOutputPath?w=e.calculateOutputPath(t.specPath):console.warn(`\u26A0\uFE0F outputPath not resolved (specPath=${t.specPath})`));let f=new J({...t,config:a,agentType:c,outputPath:w,sessionPath:d,sessionTimestamp:g,context:S,resolvedTools:this.resolvedToolsMap||{},_signal:n.signal}),T=new Map;try{await import("@zibby/skills")}catch{}let{getSkill:E}=await Promise.resolve().then(()=>(q(),et)),A=new Set;for(let[,y]of this.nodes)for(let x of y.config?.skills||[])A.add(x);for(let y of A){let x=E(y);if(typeof x?.middleware=="function")try{let v=await x.middleware();typeof v=="function"&&T.set(y,v)}catch{}}let p=this.entryPoint,F=[],Oe=a?.recursionLimit??100,Et=0;try{for(;p&&p!=="END";){if(++Et>Oe)throw new Error(`Workflow exceeded recursion limit (${Oe}) \u2014 likely a cyclic conditional route. Set config.recursionLimit if you need a higher cap.`);let x=B(d,me);if(Ne(x)){try{Mt(x)}catch{}n.abort()}if(n.signal.aborted)return console.warn(`
|
|
37
|
+
\u{1F6D1} External stop requested \u2014 ending workflow.`),$.step("Workflow stopped externally"),{success:!0,state:f.getAll(),executionLog:F,stoppedExternally:!0};let v=this.nodes.get(p);if(!v)throw new Error(`Node '${p}' not found in graph`);let Ce=JSON.stringify({sessionPath:d,sessionTimestamp:g,currentNode:p,createdAt:new Date().toISOString(),config:f.get("config")}),$t=B(d,R);ct($t,Ce,"utf-8");let Pe=f.get("config")?.paths?.output||z,Tt=B(r,Pe,R);dt(B(r,Pe),{recursive:!0});try{ct(Tt,Ce,"utf-8")}catch{}let Re=t.onPipelineProgress;if(typeof Re=="function")try{Re({cwd:r,sessionPath:d,sessionId:m,outputBase:f.get("config")?.paths?.output||z,currentNode:p})}catch{}let It=(this.resolvedToolsMap||{})[p]||null;f.set("_currentNodeTools",It);let Nt=f.get("nodeConfigs")||{};f.set("_currentNodeConfig",Nt[p]||{}),$.nodeStart(p);let We=Date.now(),re=this.nodePrompts.get(p);if(!this._invokeAgent){let O=await Promise.resolve().then(()=>(Y(),X));this._invokeAgent=O.invokeAgent}let vt=this._invokeAgent,Le=async(O,C,I={})=>{let k=vt(O,C,{...I,signal:n.signal});return k.catch(()=>{}),n.signal.aborted?k:Promise.race([k,new Promise((M,D)=>{let P=()=>{setTimeout(()=>{let Z=new Error(`Strategy ignored AbortSignal \u2014 engine deadman fired after ${i}ms`);Z.name="AbortError",D(Z)},i)};n.signal.addEventListener("abort",P,{once:!0})})])},Be={state:f,invokeAgent:async(O={},C={})=>{let I=C.prompt||"";if(re){let k=this._compiledPrompts.get(p);k||(k=jt.compile(re,{noEscape:!0}),this._compiledPrompts.set(p,k));try{I=k(O)}catch(M){throw console.error(`\u274C Template rendering failed for node '${p}':`,M.message),new Error(`Template rendering failed: ${M.message}`,{cause:M})}}else if(!I)throw new Error(`No prompt template configured for node '${p}' and no prompt provided in options`);return Le(I,{state:f.getAll(),images:C.images||[]},{model:C.model||f.get("model"),workspace:f.get("workspace"),schema:C.schema,...C,signal:n.signal})},_coreInvokeAgent:Le,agent:e,nodeId:p,promptTemplate:re,getPromptTemplate:()=>re,...f.getAll()};try{let O=(v.config?.skills||[]).map(P=>T.get(P)).filter(Boolean),C=[...this.middleware,...O],I;C.length>0?I=await this._composeMiddleware(C,p,async()=>v.execute(Be,f),f.getAll(),f):I=await v.execute(Be,f);let k=Date.now()-We;if(F.push({node:p,success:I.success,duration:k,timestamp:new Date().toISOString()}),!I.success){if(n.signal.aborted)return $.step("Workflow stopped externally"),{success:!0,state:f.getAll(),executionLog:F,stoppedExternally:!0};f.append("errors",{node:p,error:I.error});let P=v.config?.retries||0,Z=`${p}_retries`,se=f.getAll()[Z]||0;if(se<P){$.stepInfo(`Retrying (attempt ${se+1}/${P})`),f.update({[Z]:se+1,[`${p}_raw`]:I.raw});continue}throw $.nodeFailed(p,I.error,{duration:k}),new Error(`Node '${p}' failed after ${se} attempts: ${I.error}`)}f.update({[p]:I.output});let M=this._summarizeNodeOutput(p,I.output);$.nodeComplete(p,{duration:k,details:M});let D=this.edges.get(p);if(!D)p="END";else if(D.conditional){let P=D.routes(f.getAll());$.route(p,P),p=P}else p=D}catch(O){throw $.isInsideNode&&$.nodeFailed(p,O.message,{duration:Date.now()-We}),f.set("failed",!0),f.set("failedAt",p),O}}$.graphComplete();let y={success:!0,state:f.getAll(),executionLog:F};return e&&typeof e.onComplete=="function"&&await e.onComplete(y),y}finally{if(e&&typeof e.cleanup=="function")try{await e.cleanup()}catch(y){console.warn(`[workflow] agent.cleanup() failed: ${y.message}`)}}}};var ve=Symbol.for("@zibby/agent-workflow.nodes");globalThis[ve]||(globalThis[ve]=new Map);var oe=globalThis[ve];function wt(o,e){oe.set(o,e)}function ke(o){return oe.get(o)}function ae(o){return oe.has(o)}function Ut(){return Array.from(oe.keys())}function xe(o){let e=oe.get(o);return e?e.factory&&typeof e.create=="function"?e.create.toString():typeof e.execute=="function"?e.execute.toString():typeof e=="function"?e.toString():null:null}wt("ai_agent",{name:"ai_agent",factory:!0,create:(o,e={})=>({name:o,_isCustomCode:!0,execute:async t=>{let s=t?._coreInvokeAgent;s||(s=(await Promise.resolve().then(()=>(Y(),X))).invokeAgent);let n=e.extraPromptInstructions||"Execute the task based on the current state.",i=Yt(n,t),r=await s(i,{cwd:t.workspace||process.cwd(),model:t.model,tools:e.resolvedTools||null});return{success:!0,output:{raw:r,nodeId:o},raw:typeof r=="string"?r:r.raw}}})});function Yt(o,e){let t=/@([\w.]+)/g,s=new Set,n;for(;(n=t.exec(o))!==null;)s.add(n[1]);if(s.size===0)return o;let i=[],r=new Set;for(let a of s){let c=a.split(".")[0];if(r.has(c))continue;let l=a.split(".").reduce((d,g)=>d?.[g],e);if(l===void 0)continue;let h=typeof l=="string"?l:l?.raw??JSON.stringify(l,null,2),u=a.replace(/_/g," ").replace(/\b\w/g,d=>d.toUpperCase());i.push(`## ${u}
|
|
38
38
|
${h}`),a.includes(".")||r.add(c)}return i.length===0?o:`${o}
|
|
39
39
|
|
|
40
40
|
---
|
|
@@ -42,7 +42,7 @@ ${h}`),a.includes(".")||r.add(c)}return i.length===0?o:`${o}
|
|
|
42
42
|
|
|
43
43
|
${i.join(`
|
|
44
44
|
|
|
45
|
-
`)}`}q();j();var ce={};function Ae(o,e){if(Array.isArray(e))return be(e);let t=ce[o];return!t||t.length===0?null:be(t)}function be(o){if(!Array.isArray(o)||o.length===0)return null;let e=[],t={},s=[];for(let n of o){let i=U(n);if(!i){_.warn(`[workflow] unknown skill "${n}" \u2014 skipping`);continue}s.push(n);for(let r of i.tools||[])e.push({name:r.name,description:r.description,input_schema:r.input_schema||{type:"object",properties:{}}});if(!t[i.serverName])if(typeof i.resolve=="function"){let r=i.resolve();r&&(t[i.serverName]={...r,toolPrefix:n})}else{let r={};for(let a of i.envKeys||[]){let c=process.env[a];c&&(r[a]=c)}t[i.serverName]={command:i.command,args:[...i.args||[]],env:r,toolPrefix:n}}}return s.length===0?null:{toolIds:s,claudeTools:e,mcpServers:t}}j();function Zt(o,e={}){let{nodes:t,edges:s,nodeConfigs:n={}}=o;if(!Array.isArray(t)||t.length===0)throw new b("Graph must have at least one node");if(!Array.isArray(s))throw new b("Graph edges must be an array");let i=new te(e);e.stateSchema&&i.setStateSchema(e.stateSchema);let r=new Set,a=new Map,c={};for(let d of t){let g=le(d);a.set(d.id,{...d,resolvedType:g}),g==="decision"&&r.add(d.id)}for(let[d,g]of a){if(r.has(d))continue;let m=g.resolvedType,S=n[d]||{},w=Ae(m,S.tools);w&&(c[d]=w);let f={};S.prompt&&(f.prompt=S.prompt);let
|
|
45
|
+
`)}`}q();j();var ce={};function Ae(o,e){if(Array.isArray(e))return be(e);let t=ce[o];return!t||t.length===0?null:be(t)}function be(o){if(!Array.isArray(o)||o.length===0)return null;let e=[],t={},s=[];for(let n of o){let i=U(n);if(!i){_.warn(`[workflow] unknown skill "${n}" \u2014 skipping`);continue}s.push(n);for(let r of i.tools||[])e.push({name:r.name,description:r.description,input_schema:r.input_schema||{type:"object",properties:{}}});if(!t[i.serverName])if(typeof i.resolve=="function"){let r=i.resolve();r&&(t[i.serverName]={...r,toolPrefix:n})}else{let r={};for(let a of i.envKeys||[]){let c=process.env[a];c&&(r[a]=c)}t[i.serverName]={command:i.command,args:[...i.args||[]],env:r,toolPrefix:n}}}return s.length===0?null:{toolIds:s,claudeTools:e,mcpServers:t}}j();function Zt(o,e={}){let{nodes:t,edges:s,nodeConfigs:n={}}=o;if(!Array.isArray(t)||t.length===0)throw new b("Graph must have at least one node");if(!Array.isArray(s))throw new b("Graph edges must be an array");let i=new te(e);e.stateSchema&&i.setStateSchema(e.stateSchema);let r=new Set,a=new Map,c={};for(let d of t){let g=le(d);a.set(d.id,{...d,resolvedType:g}),g==="decision"&&r.add(d.id)}for(let[d,g]of a){if(r.has(d))continue;let m=g.resolvedType,S=n[d]||{},w=Ae(m,S.tools);w&&(c[d]=w);let f={};S.prompt&&(f.prompt=S.prompt);let T=ae(m);if(_.debug(`[workflow] compiler: node "${d}" type="${m}" registered=${T}`),S.customCode&&!T)i.addNode(d,yt(d,S.customCode,S),f),i.setNodeType(d,m);else if(T){let E=ke(m);E.factory?i.addNode(d,E.create(d,{...S,resolvedTools:w}),f):i.addNode(d,E,f),i.setNodeType(d,m)}else if(S.executeCode)i.addNode(d,yt(d,S.executeCode,S),f),i.setNodeType(d,m);else throw new b(`Unknown node type "${m}" for node "${d}". Did you forget to register it?`)}i.resolvedToolsMap=c;let l=new Set;for(let d of s)r.has(d.target)||l.add(d.target);let h=t.find(d=>!r.has(d.id)&&!l.has(d.id));if(!h)throw new b("Could not determine entry point: no node without incoming edges found");i.setEntryPoint(h.id);let u=Kt(s,"source");for(let d of s)if(!r.has(d.source))if(r.has(d.target)){let g=d.target,m=u.get(g)||[];if(m.length===0)throw new b(`Decision node "${g}" has no outgoing edges`);let S=Vt(g,m,r);i.addConditionalEdges(d.source,S)}else i.addEdge(d.source,d.target);return i}function Jt(o){let e=[];if(!o||typeof o!="object")return{valid:!1,errors:["Config must be a non-null object"]};if((!Array.isArray(o.nodes)||o.nodes.length===0)&&e.push("Graph must have at least one node"),Array.isArray(o.edges)||e.push("Graph edges must be an array"),e.length>0)return{valid:!1,errors:e};let t=o.nodeConfigs||{};for(let a of o.nodes){let c=le(a);if(c==="decision"||ae(c))continue;let l=t[a.id]||{};l.customCode||l.executeCode||e.push(`Unknown node type "${c}" for node "${a.id}". Register it or provide customCode/executeCode.`)}let s=new Set(o.nodes.map(a=>a.id));for(let a of o.edges)s.has(a.source)||e.push(`Edge references unknown source node "${a.source}"`),s.has(a.target)||e.push(`Edge references unknown target node "${a.target}"`);let n=new Set(o.nodes.filter(a=>le(a)==="decision").map(a=>a.id)),i=new Set;for(let a of o.edges)n.has(a.target)||i.add(a.target);let r=o.nodes.filter(a=>!n.has(a.id)&&!i.has(a.id));r.length===0?e.push("No entry point found (every node has incoming edges)"):r.length>1&&e.push(`Multiple entry points found: ${r.map(a=>a.id).join(", ")}`);for(let a of n){let c=o.edges.filter(h=>h.source===a);c.length===0&&e.push(`Decision node "${a}" has no outgoing edges`),c.some(h=>h.data?.conditionalCode||h.conditionalCode)||e.push(`Decision node "${a}" outgoing edges have no conditionalCode`)}return{valid:e.length===0,errors:e}}function Ht(o){return!o||!Array.isArray(o.nodes)?[]:o.nodes.filter(e=>le(e)!=="decision").map(e=>e.id)}function le(o){let e=o.data?.nodeType||o.data?.type||o.type;return e==="workflowNode"||e==="custom"||e==="default"?o.id:e}function Kt(o,e){let t=new Map;for(let s of o){let n=s[e];t.has(n)||t.set(n,[]),t.get(n).push(s)}return t}function Vt(o,e,t){let s=e.find(a=>a.data?.conditionalCode||a.conditionalCode);if(!s)throw new b(`Decision node "${o}" has no conditionalCode on its outgoing edges`);let n=s.data?.conditionalCode||s.conditionalCode,i=new Set(e.map(a=>a.target).filter(a=>!t.has(a))),r;try{let c=new Function(`return (${n})`)();r=l=>{let h=c(l);return i.has(h)||_.warn(`[workflow] conditional route from "${o}" returned "${h}" which is not in valid targets: ${[...i].join(", ")}`),h}}catch(a){throw new b(`Failed to compile conditionalCode for "${o}": ${a.message}`)}return r}function yt(o,e,t={}){let s;try{s=new Function("invokeAgent","require","console",`return (${e})`)}catch(r){throw new b(`Failed to compile customCode for node "${o}": ${r.message}`)}let n=s(async(...r)=>{let{invokeAgent:a}=await Promise.resolve().then(()=>(Y(),X));return a(...r)},typeof de<"u"?de:void 0,console),i=null;return t.outputSchema&&(i=t.outputSchema.jsonSchema||t.outputSchema),{name:o,_isCustomCode:!0,outputSchema:i,execute:async r=>{try{let a=await n(r);return typeof a=="object"&&"success"in a?a:{success:!0,output:a,raw:null}}catch(a){return{success:!1,error:a.message,raw:null}}}}}var b=class extends Error{constructor(e){super(e),this.name="CompilationError"}};q();we();Y();function zt(o,e={}){let{nodes:t,edges:s,nodeConfigs:n={}}=o,i=new Set,r=[],a=new Map;for(let w of t){let f=w.data?.nodeType||w.type;a.set(w.id,f),f==="decision"?i.add(w.id):r.push({id:w.id,nodeType:f,label:w.data?.label||w.id})}let c=r.some(w=>{let f=n[w.id]||{};return!f.customCode&&!f.executeCode}),{toolsPerNode:l,toolIdsByVar:h}=so(r,n),{simpleEdges:u,conditionalEdges:d}=no(s,i),g=io(r,s,i),m=[],S=e.workflowType||"workflow";return m.push(Xt(e)),m.push(Qt(S,{usesRegisteredNodes:c})),m.push(eo(h)),m.push(to(S)),m.push(oo(r,n)),m.push(ro(r,g,u,d,l,S)),m.filter(Boolean).join(`
|
|
46
46
|
`)}function qt(o){let e={};for(let[t,s]of Object.entries(o)){let{tools:n,...i}=s;Object.keys(i).length>0&&(e[t]=i)}return e}function Xt(o){let e=o.workflowType||"workflow";return["// Generated workflow",`// ${o.projectId?`Project: ${o.projectId} | `:""}Type: ${e} | Version: ${o.version??0}`,`// Downloaded: ${new Date().toISOString()}`,""].join(`
|
|
47
47
|
`)}function Qt(o,{usesRegisteredNodes:e=!0}={}){let t=["import { WorkflowGraph, invokeAgent, getResolvedToolDefinitions } from '@zibby/agent-workflow';"];return e&&t.push("// import './register-nodes.js'; // register custom node types here"),t.push("import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';","import { join, dirname } from 'node:path';","import { fileURLToPath } from 'node:url';",""),t.join(`
|
|
48
48
|
`)}function eo(o){if(o.size===0)return"";let e=["// \u2500\u2500 Tool Bindings \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500"];for(let[t,s]of o)e.push(`const ${t} = getResolvedToolDefinitions(${JSON.stringify(s)}); // ${s.join(", ")}`);return e.push(""),e.join(`
|
|
@@ -51,4 +51,4 @@ ${i.join(`
|
|
|
51
51
|
`)}function ro(o,e,t,s,n,i){let r=["// \u2500\u2500 Graph Builder \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500"];r.push("export function buildGraph(options = {}) {"),r.push(" const graph = new WorkflowGraph(options);",""),r.push(" // Nodes");for(let c of o){let l=_t(c.id);r.push(` graph.addNode('${c.id}', { name: '${c.id}', execute: ${l}_execute });`),r.push(` graph.setNodeType('${c.id}', '${c.nodeType}');`)}r.push("",` graph.setEntryPoint('${e}');`,""),(t.length>0||s.length>0)&&r.push(" // Edges");for(let c of t)r.push(` graph.addEdge('${c.source}', '${c.target}');`);for(let c of s){let l=c.code.split(`
|
|
52
52
|
`).map((h,u)=>u===0?h:` ${h}`).join(`
|
|
53
53
|
`);r.push(` graph.addConditionalEdges('${c.source}', ${l});`)}let a=[];for(let c of o){let l=n.get(c.id);l&&a.push(` '${c.id}': ${l},`)}return a.length>0&&r.push(""," graph.resolvedToolsMap = {",...a," };"),r.push(""," return graph;","}",""),r.push("export { nodeConfigs };",""),r.join(`
|
|
54
|
-
`)}function so(o,e){let t=new Map,s=new Map;for(let n of o){let i=e[n.id]?.tools,r;if(Array.isArray(i)&&i.length>0)r=[...i].sort();else{let a=ce[n.nodeType];a?.length>0&&(r=[...a].sort())}if(r){let a=`${r.map(c=>c.replace(/[^a-zA-Z0-9]/g,"")).join("And")}Tools`;t.set(n.id,a),s.has(a)||s.set(a,r)}}return{toolsPerNode:t,toolIdsByVar:s}}function no(o,e){let t=[],s=[],n=new Map,i=new Set;for(let r of o)n.has(r.source)||n.set(r.source,[]),n.get(r.source).push(r);for(let r of o)if(!e.has(r.source))if(e.has(r.target)){if(i.has(r.target))continue;i.add(r.target);let c=(n.get(r.target)||[]).find(l=>l.data?.conditionalCode||l.conditionalCode);c&&s.push({source:r.source,code:c.data?.conditionalCode||c.conditionalCode})}else t.push({source:r.source,target:r.target});return{simpleEdges:t,conditionalEdges:s}}function io(o,e,t){let s=new Set;for(let i of e)t.has(i.target)||s.add(i.target);let n=o.find(i=>!s.has(i.id));return n?n.id:o[0]?.id}function _t(o){return o.replace(/[^a-zA-Z0-9]/g,"_")}j();export{Se as AgentStrategy,ge as CI_ENV_VARS,b as CompilationError,Q as ConditionalNode,ee as ContextLoader,z as DEFAULT_OUTPUT_BASE,Rt as EVENTS_FILE,ce as NODE_DEFAULT_TOOLS,L as Node,H as OutputParser,Pt as RAW_OUTPUT_FILE,Ct as RESULT_FILE,he as SESSIONS_DIR,R as SESSION_INFO_FILE,Wt as SKILLS,me as STOP_REQUEST_FILE,bt as SchemaTypes,ie as Timeline,Ke as WORKFLOW_GRAPH_LOG_MARKER_PREFIX,te as WorkflowGraph,J as WorkflowState,ht as clearInheritedSessionEnvForFreshRun,Qe as clearSkills,Zt as compileGraph,Ht as extractSteps,qt as generateNodeConfigsJson,zt as generateWorkflowCode,gt as generateWorkflowSessionId,Ee as getAgentStrategy,qe as getAllSkills,ke as getNodeImpl,xe as getNodeTemplate,be as getResolvedToolDefinitions,U as getSkill,ae as hasNode,ze as hasSkill,rt as invokeAgent,Ut as listNodeTypes,Xe as listSkillIds,ot as listStrategies,ft as readPinnedSessionPathFromEnv,wt as registerNode,Ve as registerSkill,tt as registerStrategy,Ae as resolveNodeTools,St as resolveWorkflowSession,At as setLogger,pt as shouldTrustInheritedSessionEnv,mt as syncProcessEnvToSession
|
|
54
|
+
`)}function so(o,e){let t=new Map,s=new Map;for(let n of o){let i=e[n.id]?.tools,r;if(Array.isArray(i)&&i.length>0)r=[...i].sort();else{let a=ce[n.nodeType];a?.length>0&&(r=[...a].sort())}if(r){let a=`${r.map(c=>c.replace(/[^a-zA-Z0-9]/g,"")).join("And")}Tools`;t.set(n.id,a),s.has(a)||s.set(a,r)}}return{toolsPerNode:t,toolIdsByVar:s}}function no(o,e){let t=[],s=[],n=new Map,i=new Set;for(let r of o)n.has(r.source)||n.set(r.source,[]),n.get(r.source).push(r);for(let r of o)if(!e.has(r.source))if(e.has(r.target)){if(i.has(r.target))continue;i.add(r.target);let c=(n.get(r.target)||[]).find(l=>l.data?.conditionalCode||l.conditionalCode);c&&s.push({source:r.source,code:c.data?.conditionalCode||c.conditionalCode})}else t.push({source:r.source,target:r.target});return{simpleEdges:t,conditionalEdges:s}}function io(o,e,t){let s=new Set;for(let i of e)t.has(i.target)||s.add(i.target);let n=o.find(i=>!s.has(i.id));return n?n.id:o[0]?.id}function _t(o){return o.replace(/[^a-zA-Z0-9]/g,"_")}j();export{Se as AgentStrategy,ge as CI_ENV_VARS,b as CompilationError,Q as ConditionalNode,ee as ContextLoader,z as DEFAULT_OUTPUT_BASE,Rt as EVENTS_FILE,ce as NODE_DEFAULT_TOOLS,L as Node,H as OutputParser,Pt as RAW_OUTPUT_FILE,Ct as RESULT_FILE,he as SESSIONS_DIR,R as SESSION_INFO_FILE,Wt as SKILLS,me as STOP_REQUEST_FILE,bt as SchemaTypes,ie as Timeline,Ke as WORKFLOW_GRAPH_LOG_MARKER_PREFIX,te as WorkflowGraph,J as WorkflowState,ht as clearInheritedSessionEnvForFreshRun,Qe as clearSkills,Zt as compileGraph,Ht as extractSteps,qt as generateNodeConfigsJson,zt as generateWorkflowCode,gt as generateWorkflowSessionId,Ee as getAgentStrategy,qe as getAllSkills,ke as getNodeImpl,xe as getNodeTemplate,be as getResolvedToolDefinitions,U as getSkill,ae as hasNode,ze as hasSkill,rt as invokeAgent,Ut as listNodeTypes,Xe as listSkillIds,ot as listStrategies,ft as readPinnedSessionPathFromEnv,wt as registerNode,Ve as registerSkill,tt as registerStrategy,Ae as resolveNodeTools,St as resolveWorkflowSession,At as setLogger,pt as shouldTrustInheritedSessionEnv,mt as syncProcessEnvToSession,$ as timeline,Jt as validateGraphConfig};
|
package/dist/node.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
var st=Object.defineProperty;var
|
|
1
|
+
var st=Object.defineProperty;var b=(r,t)=>()=>(r&&(t=r(r=0)),t);var it=(r,t)=>{for(var e in t)st(r,e,{get:t[e],enumerable:!0})};var D,nt,N,l,R=b(()=>{D=()=>{},nt={debug:D,info:D,warn:(...r)=>console.warn("[workflow]",...r),error:(...r)=>console.error("[workflow]",...r)},N={impl:nt},l={debug:(...r)=>N.impl.debug?.(...r),info:(...r)=>N.impl.info?.(...r),warn:(...r)=>N.impl.warn?.(...r),error:(...r)=>N.impl.error?.(...r)}});var Z=b(()=>{});function V(r){return ut.get(r)||null}var v,ut,z=b(()=>{v=Symbol.for("@zibby/agent-workflow.skills");globalThis[v]||(globalThis[v]=new Map);ut=globalThis[v]});var X={};it(X,{getAgentStrategy:()=>q,invokeAgent:()=>dt,listStrategies:()=>pt,registerStrategy:()=>ct});function ct(r){if(!r||typeof r.getName!="function"||typeof r.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let t=S.findIndex(e=>e.getName()===r.getName());t>=0?S[t]=r:S.push(r)}function pt(){return S.map(r=>r.getName())}function q(r={}){let{state:t={},preferredAgent:e=null}=r,o=e||t.agentType||process.env.AGENT_TYPE;if(!o){let n=S.map(a=>a.getName()).join(", ")||"none registered";throw new Error(`No agent specified. Set agentType in state or AGENT_TYPE env var. Available: ${n}`)}l.debug(`[workflow] agent selection: requested=${o}`);let i=S.find(n=>n.getName()===o);if(!i){let n=S.map(a=>a.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${o}'. Available: ${n}`)}if(!i.canHandle(r))throw new Error(`Agent '${o}' is not available in this environment. Check credentials/environment.`);return l.debug(`[workflow] using agent: ${i.getName()}`),i}async function dt(r,t={},e={}){let o=q(t),i=t.state?.config||e.config||{},n=i.models||{},a=e.nodeName&&n[e.nodeName]||null,g=n.default||null,p=i.agent?.[o.name]?.model||null,w=a||g||p||e.model||null,y={...e,model:w,workspace:t.state?.workspace||e.workspace,schema:e.schema||t.schema,images:e.images||t.images||[],skills:e.skills||t.skills||[],config:i},s=r,h=y.skills||[];if(h.length>0&&!e.skipPromptFragments){let E=h.map(I=>{let $=V(I)?.promptFragment;return typeof $=="function"?$():$}).filter(Boolean);E.length>0&&(s+=`
|
|
2
2
|
|
|
3
3
|
${E.join(`
|
|
4
4
|
|
|
@@ -9,9 +9,9 @@ PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PRE
|
|
|
9
9
|
\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501\u2501
|
|
10
10
|
|
|
11
11
|
${f}
|
|
12
|
-
`),
|
|
12
|
+
`),l.debug(`[workflow] prompt length: ${s.length} chars`),o.invoke(s,y)}var P,S,Q=b(()=>{Z();R();z();P=Symbol.for("@zibby/agent-workflow.strategies");globalThis[P]||(globalThis[P]=[]);S=globalThis[P]});var k=class{constructor(t){this.schema=t}parse(t){let e=t.match(/```json\s*([\s\S]*?)\s*```/);if(e)return this.validate(JSON.parse(e[1]));let o=[t.match(/\{[\s\S]*?\}/),t.match(/\{[\s\S]*\}/)].filter(Boolean).map(i=>i[0]);for(let i of o)try{return this.validate(JSON.parse(i))}catch(n){if(!(n instanceof SyntaxError))throw n}return this.validate({result:t.trim()})}validate(t){let e=[];for(let[o,i]of Object.entries(this.schema)){if(i.required&&!(o in t)&&e.push(`Missing required field: ${o}`),o in t&&i.type){let n=typeof t[o];n!==i.type&&e.push(`Field '${o}' expected ${i.type}, got ${n}`)}if(i.validate&&o in t){let n=i.validate(t[o]);n&&e.push(`Field '${o}': ${n}`)}}if(e.length>0)throw new Error(`Output validation failed:
|
|
13
13
|
${e.join(`
|
|
14
|
-
`)}`);return t}};R();import{writeFileSync as C,readFileSync as tt,existsSync as et,mkdirSync as ht}from"node:fs";import{join as F,dirname as ft}from"node:path";import
|
|
14
|
+
`)}`);return t}};R();import{writeFileSync as C,readFileSync as tt,existsSync as et,mkdirSync as ht}from"node:fs";import{join as F,dirname as ft}from"node:path";import u from"chalk";var at="__WORKFLOW_GRAPH_LOG__",O=u.gray("\u2502"),lt=u.gray("\u250C"),G=u.gray("\u2514"),x=u.green("\u25C6"),U=u.hex("#c084fc")("\u25C6"),j=u.hex("#2dd4bf")("\u25C6"),T=u.red("\u25C6"),J=`${O} `,H=2;function K(r){return r<1e3?`${r}ms`:`${(r/1e3).toFixed(1)}s`}function B(r,t){return(e,o,i)=>{if(typeof e!="string")return r(e,o,i);let n=process.stdout.columns||120,a="";for(let g=0;g<e.length;g++){let p=e[g];t.lineStart&&(a+=J,t.col=H,t.lineStart=!1),p===`
|
|
15
15
|
`?(a+=p,t.lineStart=!0,t.col=0,t.inEsc=!1):p==="\x1B"?(t.inEsc=!0,a+=p):t.inEsc?(a+=p,(p>="A"&&p<="Z"||p>="a"&&p<="z")&&(t.inEsc=!1)):(t.col++,a+=p,t.col>=n&&(a+=`
|
|
16
16
|
${J}`,t.col=H))}return r(a,o,i)}}var A=class{constructor(){this._currentNode=null,this._origStdoutWrite=null,this._origStderrWrite=null,this._emitWorkflowGraphMarkers=String(process.env.ZIBBY_EMIT_GRAPH_MARKERS||"").trim()==="1"||String(process.env.ZIBBY_WORKFLOW_GRAPH_LOG_MARKERS||"").trim()==="1"}get isInsideNode(){return this._currentNode!==null}_startIntercepting(){this._origStdoutWrite=process.stdout.write.bind(process.stdout),this._origStderrWrite=process.stderr.write.bind(process.stderr);let t={lineStart:!0,col:0,inEsc:!1},e={lineStart:!0,col:0,inEsc:!1};this._outState=t,this._errState=e,process.stdout.write=B(this._origStdoutWrite,t),process.stderr.write=B(this._origStderrWrite,e)}_stopIntercepting(){this._origStdoutWrite&&(this._outState&&!this._outState.lineStart&&this._origStdoutWrite(`
|
|
17
17
|
`),process.stdout.write=this._origStdoutWrite),this._origStderrWrite&&(this._errState&&!this._errState.lineStart&&this._origStderrWrite(`
|
|
@@ -22,8 +22,8 @@ ${J}`,t.col=H))}return r(a,o,i)}}var A=class{constructor(){this._currentNode=nul
|
|
|
22
22
|
`)):process.stdout.write.bind(process.stdout)(`${t} ${e}
|
|
23
23
|
`)}step(t){this._origStdoutWrite?this._writeDot(x,t):process.stdout.write.bind(process.stdout)(`${O} ${x} ${t}
|
|
24
24
|
`)}stepInfo(t){this.step(t)}stepTool(t){this._origStdoutWrite?this._writeDot(U,t):process.stdout.write.bind(process.stdout)(`${O} ${U} ${t}
|
|
25
|
-
`)}stepMemory(t){let e=
|
|
26
|
-
`)}stepFail(t){this._origStdoutWrite?this._writeDot(T,
|
|
27
|
-
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${lt} ${t}`),this._startIntercepting()}nodeComplete(t,e={}){this._stopIntercepting();let{duration:o,details:i}=e;if(i)for(let a of i)this._rawWrite(`${x} ${a}`);let n=o?
|
|
25
|
+
`)}stepMemory(t){let e=u.hex("#2dd4bf")(t);this._origStdoutWrite?this._writeDot(j,e):process.stdout.write.bind(process.stdout)(`${O} ${j} ${e}
|
|
26
|
+
`)}stepFail(t){this._origStdoutWrite?this._writeDot(T,u.red(t)):process.stdout.write.bind(process.stdout)(`${O} ${T} ${u.red(t)}
|
|
27
|
+
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${lt} ${t}`),this._startIntercepting()}nodeComplete(t,e={}){this._stopIntercepting();let{duration:o,details:i}=e;if(i)for(let a of i)this._rawWrite(`${x} ${a}`);let n=o?u.dim(` ${K(o)}`):"";this._rawWrite(`${G} ${u.green("done")}${n}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}nodeFailed(t,e,o={}){this._stopIntercepting();let{duration:i}=o,n=i?u.dim(` ${K(i)}`):"";this._rawWrite(`${T} ${u.red(e)}`),this._rawWrite(`${G} ${u.red("failed")}${n}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}route(t,e){this._rawWrite(u.dim(` ${t} \u2192 ${e}`)),this._rawWrite("")}graphComplete(){}},Y=new A;var L=".session-info.json";var M=class{constructor(t){if(this.config=t,this.name=t.name,this.prompt=t.prompt,this.outputSchema=t.outputSchema,!this.outputSchema&&!t._isCustomCode)throw new Error(`Node '${this.name}' must define outputSchema (Zod schema). This defines the contract for what the node returns to state.`);this.isZodSchema=this.outputSchema&&typeof this.outputSchema._def<"u",this.parser=t.outputSchema&&!this.isZodSchema?new k(t.outputSchema):null,this.retries=t.retries||0,this.onComplete=t.onComplete,this.customExecute=t.execute}async execute(t,e){let o=()=>e&&typeof e.getAll=="function"?e.getAll():t,i=s=>e&&typeof e.get=="function"?e.get(s):t?.[s];if(typeof this.customExecute=="function"){l.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let s=await this.customExecute(t);return typeof s=="object"&&s!==null&&s.success===!1?{success:!1,error:s.error||"Node execution failed",raw:s.raw||null}:this.isZodSchema?(l.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(s),raw:null}):{success:!0,output:s,raw:null}}catch(s){return l.error(`[workflow] node '${this.name}' failed: ${s.message}`),s.name==="ZodError"&&l.error(`Schema errors: ${JSON.stringify(s.issues||s.errors,null,2)}`),{success:!1,error:s.message,raw:null}}}let n=typeof this.prompt=="function"?this.prompt(o()):this.prompt,a=i("_skillHints");a&&(n=`${a}
|
|
28
28
|
|
|
29
|
-
${n}`);let g=o(),p=g.cwd||process.cwd(),w=g.sessionPath;try{if(w){let s=F(w,L);if(et(s)){let f=JSON.parse(tt(s,"utf-8"));f.currentNode=this.name,C(s,JSON.stringify(f,null,2),"utf-8")}let h=F(w,"..",L);if(et(h))try{let f=JSON.parse(tt(h,"utf-8"));f.currentNode=this.name,C(h,JSON.stringify(f,null,2),"utf-8")}catch{}}}catch(s){
|
|
29
|
+
${n}`);let g=o(),p=g.cwd||process.cwd(),w=g.sessionPath;try{if(w){let s=F(w,L);if(et(s)){let f=JSON.parse(tt(s,"utf-8"));f.currentNode=this.name,C(s,JSON.stringify(f,null,2),"utf-8")}let h=F(w,"..",L);if(et(h))try{let f=JSON.parse(tt(h,"utf-8"));f.currentNode=this.name,C(h,JSON.stringify(f,null,2),"utf-8")}catch{}}}catch(s){l.debug(`[workflow] could not update session info: ${s.message}`)}let y=null;for(let s=0;s<=this.retries;s++)try{l.debug(`[workflow] node '${this.name}' attempt ${s}`);let h=o().config||{},f=h.agents||{},E=this.config.agent??f[this.name]??null,I={state:o()};E&&(I.preferredAgent=E);let $={workspace:p,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:w,config:h,nodeName:this.name,timeout:this.config?.timeout||3e5},W=t?._coreInvokeAgent;W||(W=(await Promise.resolve().then(()=>(Q(),X))).invokeAgent);let m=await W(n,I,$),d,_;if(typeof m=="string"?(d=m,_=null):m.structured?(d=m.raw||JSON.stringify(m.structured,null,2),_=m.structured):(d=m.raw||JSON.stringify(m,null,2),_=m.extracted||null),w)try{let c=F(w,this.name,"raw_stream_output.txt");ht(ft(c),{recursive:!0}),C(c,typeof d=="string"?d:JSON.stringify(d),"utf-8")}catch(c){l.debug(`[workflow] could not save raw output: ${c.message}`)}if(this.isZodSchema&&_){l.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(_,null,2)}`);let c=_;if(typeof this.onComplete=="function")try{c=await this.onComplete(o(),_)}catch(ot){l.warn(`[workflow] onComplete hook failed: ${ot.message}`)}return{success:!0,output:c,raw:d}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(o(),{raw:d}),raw:d}}catch(c){throw new Error(`onComplete failed: ${c.message}`,{cause:c})}if(this.parser){let c=this.parser.parse(d);return l.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(c,null,2)}`),Y.step("Output parsed"),{success:!0,output:c,raw:d}}return{success:!0,output:d,raw:d}}catch(h){y=h,s<this.retries&&l.info(`[workflow] node '${this.name}' failed, retrying (${s+1}/${this.retries})\u2026`)}return{success:!1,error:y.message,raw:null}}},rt=class extends M{constructor(t){super({...t,_isCustomCode:!0}),this.condition=t.condition}async execute(t,e){let o=e&&typeof e.getAll=="function"?e.getAll():t;return{success:!0,output:{nextNode:this.condition(o)},raw:null}}};export{rt as ConditionalNode,M as Node};
|
package/dist/timeline.js
CHANGED
|
@@ -1,14 +1,14 @@
|
|
|
1
|
-
import
|
|
2
|
-
`?(s+=
|
|
3
|
-
${
|
|
1
|
+
import i from"chalk";var I="__WORKFLOW_GRAPH_LOG__",c=i.gray("\u2502"),E=i.gray("\u250C"),a=i.gray("\u2514"),h=i.green("\u25C6"),S=i.hex("#c084fc")("\u25C6"),W=i.hex("#2dd4bf")("\u25C6"),l=i.red("\u25C6"),f=`${c} `,w=2;function g(d){return d<1e3?`${d}ms`:`${(d/1e3).toFixed(1)}s`}function $(d,t){return(r,o,n)=>{if(typeof r!="string")return d(r,o,n);let _=process.stdout.columns||120,s="";for(let u=0;u<r.length;u++){let e=r[u];t.lineStart&&(s+=f,t.col=w,t.lineStart=!1),e===`
|
|
2
|
+
`?(s+=e,t.lineStart=!0,t.col=0,t.inEsc=!1):e==="\x1B"?(t.inEsc=!0,s+=e):t.inEsc?(s+=e,(e>="A"&&e<="Z"||e>="a"&&e<="z")&&(t.inEsc=!1)):(t.col++,s+=e,t.col>=_&&(s+=`
|
|
3
|
+
${f}`,t.col=w))}return d(s,o,n)}}var p=class{constructor(){this._currentNode=null,this._origStdoutWrite=null,this._origStderrWrite=null,this._emitWorkflowGraphMarkers=String(process.env.ZIBBY_EMIT_GRAPH_MARKERS||"").trim()==="1"||String(process.env.ZIBBY_WORKFLOW_GRAPH_LOG_MARKERS||"").trim()==="1"}get isInsideNode(){return this._currentNode!==null}_startIntercepting(){this._origStdoutWrite=process.stdout.write.bind(process.stdout),this._origStderrWrite=process.stderr.write.bind(process.stderr);let t={lineStart:!0,col:0,inEsc:!1},r={lineStart:!0,col:0,inEsc:!1};this._outState=t,this._errState=r,process.stdout.write=$(this._origStdoutWrite,t),process.stderr.write=$(this._origStderrWrite,r)}_stopIntercepting(){this._origStdoutWrite&&(this._outState&&!this._outState.lineStart&&this._origStdoutWrite(`
|
|
4
4
|
`),process.stdout.write=this._origStdoutWrite),this._origStderrWrite&&(this._errState&&!this._errState.lineStart&&this._origStderrWrite(`
|
|
5
5
|
`),process.stderr.write=this._origStderrWrite),this._origStdoutWrite=null,this._origStderrWrite=null}_rawWrite(t){(this._origStdoutWrite||process.stdout.write.bind(process.stdout))(`${t}
|
|
6
6
|
`)}_emitGraphLogMarker(t){if(!this._emitWorkflowGraphMarkers)return;let r=`${I}${JSON.stringify(t)}
|
|
7
7
|
`;this._origStdoutWrite?this._origStdoutWrite(r):process.stdout.write(r)}_writeDot(t,r){this._origStdoutWrite?(this._outState&&!this._outState.lineStart&&(this._origStdoutWrite(`
|
|
8
8
|
`),this._outState.lineStart=!0,this._outState.col=0),this._origStdoutWrite(`${t} ${r}
|
|
9
9
|
`)):process.stdout.write.bind(process.stdout)(`${t} ${r}
|
|
10
|
-
`)}step(t){this._origStdoutWrite?this._writeDot(
|
|
10
|
+
`)}step(t){this._origStdoutWrite?this._writeDot(h,t):process.stdout.write.bind(process.stdout)(`${c} ${h} ${t}
|
|
11
11
|
`)}stepInfo(t){this.step(t)}stepTool(t){this._origStdoutWrite?this._writeDot(S,t):process.stdout.write.bind(process.stdout)(`${c} ${S} ${t}
|
|
12
|
-
`)}stepMemory(t){let r=
|
|
13
|
-
`)}stepFail(t){this._origStdoutWrite?this._writeDot(l,
|
|
14
|
-
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${E} ${t}`),this._startIntercepting()}nodeComplete(t,r={}){this._stopIntercepting();let{duration:o,details:n}=r;if(n)for(let s of n)this._rawWrite(`${
|
|
12
|
+
`)}stepMemory(t){let r=i.hex("#2dd4bf")(t);this._origStdoutWrite?this._writeDot(W,r):process.stdout.write.bind(process.stdout)(`${c} ${W} ${r}
|
|
13
|
+
`)}stepFail(t){this._origStdoutWrite?this._writeDot(l,i.red(t)):process.stdout.write.bind(process.stdout)(`${c} ${l} ${i.red(t)}
|
|
14
|
+
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${E} ${t}`),this._startIntercepting()}nodeComplete(t,r={}){this._stopIntercepting();let{duration:o,details:n}=r;if(n)for(let s of n)this._rawWrite(`${h} ${s}`);let _=o?i.dim(` ${g(o)}`):"";this._rawWrite(`${a} ${i.green("done")}${_}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}nodeFailed(t,r,o={}){this._stopIntercepting();let{duration:n}=o,_=n?i.dim(` ${g(n)}`):"";this._rawWrite(`${l} ${i.red(r)}`),this._rawWrite(`${a} ${i.red("failed")}${_}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}route(t,r){this._rawWrite(i.dim(` ${t} \u2192 ${r}`)),this._rawWrite("")}graphComplete(){}},O=new p;var P=O;export{p as Timeline,I as WORKFLOW_GRAPH_LOG_MARKER_PREFIX,P as default,O as timeline};
|