@zibby/agent-workflow 0.4.5 → 0.4.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/constants.d.ts +5 -0
- package/dist/constants.js +1 -1
- package/dist/graph-compiler.js +20 -20
- package/dist/graph.js +16 -16
- package/dist/in-process-subgraph.d.ts +25 -2
- package/dist/in-process-subgraph.js +1 -1
- package/dist/index.js +31 -31
- package/dist/node.js +10 -10
- package/dist/sub-graph-executor.js +1 -1
- package/package.json +1 -1
package/dist/index.js
CHANGED
|
@@ -1,54 +1,54 @@
|
|
|
1
|
-
var
|
|
1
|
+
var Qt=Object.defineProperty;var Ie=(t=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(t,{get:(e,o)=>(typeof require<"u"?require:e)[o]}):t)(function(t){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+t+'" is not supported')});var he=(t,e)=>()=>(t&&(e=t(t=0)),e);var Qe=(t,e)=>{for(var o in e)Qt(t,o,{get:e[o],enumerable:!0})};function oo(t){te.impl={...tt,...t}}var et,tt,te,_,F=he(()=>{et=()=>{},tt={debug:et,info:et,warn:(...t)=>console.warn("[workflow]",...t),error:(...t)=>console.error("[workflow]",...t)},te={impl:tt};_={debug:(...t)=>te.impl.debug?.(...t),info:(...t)=>te.impl.info?.(...t),warn:(...t)=>te.impl.warn?.(...t),error:(...t)=>te.impl.error?.(...t)}});var ke,Ae=he(()=>{ke=class{constructor(e,o,n=0){this.name=e,this.description=o,this.priority=n}async invoke(e,o={}){throw new Error(`${this.constructor.name}.invoke() must be implemented`)}canHandle(e){throw new Error(`${this.constructor.name}.canHandle() must be implemented`)}getName(){return this.name}getDescription(){return this.description}getPriority(){return this.priority}}});var gt={};Qe(gt,{clearSkills:()=>ht,getAllSkills:()=>pt,getSkill:()=>V,hasSkill:()=>dt,listSkillIds:()=>ft,registerSkill:()=>ut});function ut(t){if(!t||typeof t.id!="string")throw new Error("Skill definition must include a string id");K.set(t.id,Object.freeze({...t}))}function V(t){return K.get(t)||null}function dt(t){return K.has(t)}function pt(){return new Map(K)}function ft(){return Array.from(K.keys())}function ht(){K.clear()}var Ne,K,re=he(()=>{Ne=Symbol.for("@zibby/agent-workflow.skills");globalThis[Ne]||(globalThis[Ne]=new Map);K=globalThis[Ne]});var se={};Qe(se,{getAgentStrategy:()=>Ce,invokeAgent:()=>wt,listStrategies:()=>St,registerStrategy:()=>mt});function mt(t){if(!t||typeof t.getName!="function"||typeof t.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let e=G.findIndex(o=>o.getName()===t.getName());e>=0?G[e]=t:G.push(t)}function St(){return G.map(t=>t.getName())}function Ce(t={}){let{state:e={},preferredAgent:o=null}=t,n=o||e.agentType||process.env.AGENT_TYPE;if(!n){let r=G.map(i=>i.getName()).join(", ")||"none registered";throw new Error(`No agent specified. Set agentType in state or AGENT_TYPE env var. Available: ${r}`)}_.debug(`[workflow] agent selection: requested=${n}`);let s=G.find(r=>r.getName()===n);if(!s){let r=G.map(i=>i.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${n}'. Available: ${r}`)}if(!s.canHandle(t))throw new Error(`Agent '${n}' is not available in this environment. Check credentials/environment.`);return _.debug(`[workflow] using agent: ${s.getName()}`),s}async function wt(t,e={},o={}){let n=Ce(e),s=e.state?.config||o.config||{},r=s.models||{},i=o.nodeName&&r[o.nodeName]||null,a=r.default||null,c=s.agent?.[n.name]?.model||null,l=i||a||c||o.model||null,d={...o,model:l,workspace:e.state?.workspace||o.workspace,schema:o.schema||e.schema,images:o.images||e.images||[],skills:o.skills||e.skills||[],config:s},p=t,u=d.skills||[];if(u.length>0&&!o.skipPromptFragments){let m=u.map(w=>{let I=V(w)?.promptFragment;return typeof I=="function"?I():I}).filter(Boolean);m.length>0&&(p+=`
|
|
2
2
|
|
|
3
|
-
${
|
|
3
|
+
${m.join(`
|
|
4
4
|
|
|
5
|
-
`)}`)}let
|
|
5
|
+
`)}`)}let h=e.state?._currentNodeConfig?.extraPromptInstructions?.trim();return h&&(p+=`
|
|
6
6
|
|
|
7
7
|
\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
|
|
8
8
|
PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PREVIOUS CONTENT
|
|
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
|
-
${
|
|
12
|
-
`),_.debug(`[workflow] prompt length: ${
|
|
11
|
+
${h}
|
|
12
|
+
`),_.debug(`[workflow] prompt length: ${p.length} chars`),n.invoke(p,d)}var Oe,G,q=he(()=>{Ae();F();re();Oe=Symbol.for("@zibby/agent-workflow.strategies");globalThis[Oe]||(globalThis[Oe]=[]);G=globalThis[Oe]});var eo=new Set(["__proto__","constructor","prototype"]);function Ee(t){if(eo.has(t))throw new Error(`Invalid state key: "${t}"`)}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,o){Ee(e),this._history.push({...this._state}),this._state[e]=o}update(e){let o=Object.getOwnPropertyNames(e);for(let n of o)Ee(n);this._history.push({...this._state});for(let n of o)this._state[n]=e[n]}append(e,o){Ee(e),this._history.push({...this._state}),Array.isArray(this._state[e])||(this._state[e]=[]),this._state[e].push(o)}getAll(){return{...this._state}}rollback(){this._history.length>0&&(this._state=this._history.pop())}};var ee=class{constructor(e){this.schema=e}parse(e){let o=e.match(/```json\s*([\s\S]*?)\s*```/);if(o)return this.validate(JSON.parse(o[1]));let n=[e.match(/\{[\s\S]*?\}/),e.match(/\{[\s\S]*\}/)].filter(Boolean).map(s=>s[0]);for(let s of n)try{return this.validate(JSON.parse(s))}catch(r){if(!(r instanceof SyntaxError))throw r}return this.validate({result:e.trim()})}validate(e){let o=[];for(let[n,s]of Object.entries(this.schema)){if(s.required&&!(n in e)&&o.push(`Missing required field: ${n}`),n in e&&s.type){let r=typeof e[n];r!==s.type&&o.push(`Field '${n}' expected ${s.type}, got ${r}`)}if(s.validate&&n in e){let r=s.validate(e[n]);r&&o.push(`Field '${n}': ${r}`)}}if(o.length>0)throw new Error(`Output validation failed:
|
|
13
13
|
${o.join(`
|
|
14
|
-
`)}`);return e}},
|
|
15
|
-
`?(
|
|
16
|
-
${
|
|
14
|
+
`)}`);return e}},to={string:(t=!0)=>({type:"string",required:t}),number:(t=!0)=>({type:"number",required:t}),boolean:(t=!0)=>({type:"boolean",required:t}),array:(t=!0)=>({type:"object",required:t,validate:e=>Array.isArray(e)?null:"must be an array"}),enum:(t,e=!0)=>({type:"string",required:e,validate:o=>t.includes(o)?null:`must be one of: ${t.join(", ")}`})};F();import{writeFileSync as Pe,readFileSync as yt,existsSync as _t,mkdirSync as co}from"node:fs";import{join as Re,dirname as lo}from"node:path";import k from"chalk";var lt="__WORKFLOW_GRAPH_LOG__",oe=k.gray("\u2502"),no=k.gray("\u250C"),ot=k.gray("\u2514"),$e=k.green("\u25C6"),nt=k.hex("#c084fc")("\u25C6"),rt=k.hex("#2dd4bf")("\u25C6"),be=k.red("\u25C6"),st=`${oe} `,it=2;function at(t){return t<1e3?`${t}ms`:`${(t/1e3).toFixed(1)}s`}function ct(t,e){return(o,n,s)=>{if(typeof o!="string")return t(o,n,s);let r=process.stdout.columns||120,i="";for(let a=0;a<o.length;a++){let c=o[a];e.lineStart&&(i+=st,e.col=it,e.lineStart=!1),c===`
|
|
15
|
+
`?(i+=c,e.lineStart=!0,e.col=0,e.inEsc=!1):c==="\x1B"?(e.inEsc=!0,i+=c):e.inEsc?(i+=c,(c>="A"&&c<="Z"||c>="a"&&c<="z")&&(e.inEsc=!1)):(e.col++,i+=c,e.col>=r&&(i+=`
|
|
16
|
+
${st}`,e.col=it))}return t(i,n,s)}}var ge=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},o={lineStart:!0,col:0,inEsc:!1};this._outState=e,this._errState=o,process.stdout.write=ct(this._origStdoutWrite,e),process.stderr.write=ct(this._origStderrWrite,o)}_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
|
-
`)}_emitGraphLogMarker(e){if(!this._emitWorkflowGraphMarkers)return;let o=`${
|
|
19
|
+
`)}_emitGraphLogMarker(e){if(!this._emitWorkflowGraphMarkers)return;let o=`${lt}${JSON.stringify(e)}
|
|
20
20
|
`;this._origStdoutWrite?this._origStdoutWrite(o):process.stdout.write(o)}_writeDot(e,o){this._origStdoutWrite?(this._outState&&!this._outState.lineStart&&(this._origStdoutWrite(`
|
|
21
21
|
`),this._outState.lineStart=!0,this._outState.col=0),this._origStdoutWrite(`${e} ${o}
|
|
22
22
|
`)):process.stdout.write.bind(process.stdout)(`${e} ${o}
|
|
23
|
-
`)}step(e){this._origStdoutWrite?this._writeDot(
|
|
24
|
-
`)}stepInfo(e){this.step(e)}stepTool(e){this._origStdoutWrite?this._writeDot(
|
|
25
|
-
`)}stepMemory(e){let o=k.hex("#2dd4bf")(e);this._origStdoutWrite?this._writeDot(
|
|
26
|
-
`)}stepFail(e){this._origStdoutWrite?this._writeDot(
|
|
27
|
-
`)}nodeStart(e){this._currentNode=e,this._emitGraphLogMarker({phase:"node_begin",node:e}),this._rawWrite(`${
|
|
23
|
+
`)}step(e){this._origStdoutWrite?this._writeDot($e,e):process.stdout.write.bind(process.stdout)(`${oe} ${$e} ${e}
|
|
24
|
+
`)}stepInfo(e){this.step(e)}stepTool(e){this._origStdoutWrite?this._writeDot(nt,e):process.stdout.write.bind(process.stdout)(`${oe} ${nt} ${e}
|
|
25
|
+
`)}stepMemory(e){let o=k.hex("#2dd4bf")(e);this._origStdoutWrite?this._writeDot(rt,o):process.stdout.write.bind(process.stdout)(`${oe} ${rt} ${o}
|
|
26
|
+
`)}stepFail(e){this._origStdoutWrite?this._writeDot(be,k.red(e)):process.stdout.write.bind(process.stdout)(`${oe} ${be} ${k.red(e)}
|
|
27
|
+
`)}nodeStart(e){this._currentNode=e,this._emitGraphLogMarker({phase:"node_begin",node:e}),this._rawWrite(`${no} ${e}`),this._startIntercepting()}nodeComplete(e,o={}){this._stopIntercepting();let{duration:n,details:s}=o;if(s)for(let i of s)this._rawWrite(`${$e} ${i}`);let r=n?k.dim(` ${at(n)}`):"";this._rawWrite(`${ot} ${k.green("done")}${r}`),this._emitGraphLogMarker({phase:"node_end",node:e}),this._rawWrite("")}nodeFailed(e,o,n={}){this._stopIntercepting();let{duration:s}=n,r=s?k.dim(` ${at(s)}`):"";this._rawWrite(`${be} ${k.red(o)}`),this._rawWrite(`${ot} ${k.red("failed")}${r}`),this._emitGraphLogMarker({phase:"node_end",node:e}),this._rawWrite("")}route(e,o){this._rawWrite(k.dim(` ${e} \u2192 ${o}`)),this._rawWrite("")}graphComplete(){}},x=new ge;var ne=".zibby/output",Te="sessions",W=".session-info.json",ve=".zibby-stop",ro="result.json",so="raw_stream_output.txt",io="events.json",ao={BROWSER:"browser",JIRA:"jira",GITHUB:"github",GITLAB:"gitlab",GIT:"git",SLACK:"slack",LARK:"lark",CHAT_NOTIFY:"chat_notify",SENTRY:"sentry",MEMORY:"memory",CHAT_MEMORY:"chat-memory",RUNNER:"runner",SKILL_INSTALLER:"skill-installer",CORE_TOOLS:"core-tools",WORKFLOW_BUILDER:"workflow-builder",SESSION:"session",OPENAI_BILLING:"openai_billing",ANTHROPIC_BILLING:"anthropic_billing",CURSOR_ADMIN:"cursor_admin",NOTION:"notion"},xe=["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 ee(e.outputSchema):null,this.retries=e.retries||0,this.onComplete=e.onComplete,this.customExecute=e.execute}async execute(e,o){let n=()=>o&&typeof o.getAll=="function"?o.getAll():e,s=p=>o&&typeof o.get=="function"?o.get(p):e?.[p];if(typeof this.customExecute=="function"){_.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let p=await this.customExecute(e);return typeof p=="object"&&p!==null&&p.success===!1?{success:!1,error:p.error||"Node execution failed",raw:p.raw||null}:this.isZodSchema?(_.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(p),raw:null}):{success:!0,output:p,raw:null}}catch(p){return _.error(`[workflow] node '${this.name}' failed: ${p.message}`),p.name==="ZodError"&&_.error(`Schema errors: ${JSON.stringify(p.issues||p.errors,null,2)}`),{success:!1,error:p.message,raw:null}}}let r=typeof this.prompt=="function"?this.prompt(n()):this.prompt,i=s("_skillHints");i&&(r=`${i}
|
|
28
28
|
|
|
29
|
-
${i}`);let a=r(),c=a.cwd||process.cwd(),l=a.sessionPath;try{if(l){let f=Oe(l,L);if(yt(f)){let S=JSON.parse(wt(f,"utf-8"));S.currentNode=this.name,Ne(f,JSON.stringify(S,null,2),"utf-8")}let d=Oe(l,"..",L);if(yt(d))try{let S=JSON.parse(wt(d,"utf-8"));S.currentNode=this.name,Ne(d,JSON.stringify(S,null,2),"utf-8")}catch{}}}catch(f){_.debug(`[workflow] could not update session info: ${f.message}`)}let u=null;for(let f=0;f<=this.retries;f++)try{_.debug(`[workflow] node '${this.name}' attempt ${f}`);let d=r().config||{},S=d.agents||{},h=this.config.agent??S[this.name]??null,g={state:r()};h&&(g.preferredAgent=h);let I={workspace:c,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:l,config:d,nodeName:this.name,timeout:this.config?.timeout||3e5},$=e?._coreInvokeAgent;$||($=(await Promise.resolve().then(()=>(z(),oe))).invokeAgent);let p=await $(i,g,I),w,b;if(typeof p=="string"?(w=p,b=null):p.structured?(w=p.raw||JSON.stringify(p.structured,null,2),b=p.structured):(w=p.raw||JSON.stringify(p,null,2),b=p.extracted||null),l)try{let m=Oe(l,this.name,"raw_stream_output.txt");so(io(m),{recursive:!0}),Ne(m,typeof w=="string"?w:JSON.stringify(w),"utf-8")}catch(m){_.debug(`[workflow] could not save raw output: ${m.message}`)}if(this.isZodSchema&&b){_.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(b,null,2)}`);let m=b;if(typeof this.onComplete=="function")try{m=await this.onComplete(r(),b)}catch(T){_.warn(`[workflow] onComplete hook failed: ${T.message}`)}return{success:!0,output:m,raw:w}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(r(),{raw:w}),raw:w}}catch(m){throw new Error(`onComplete failed: ${m.message}`,{cause:m})}if(this.parser){let m=this.parser.parse(w);return _.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(m,null,2)}`),x.step("Output parsed"),{success:!0,output:m,raw:w}}return{success:!0,output:w,raw:w}}catch(d){u=d,f<this.retries&&_.info(`[workflow] node '${this.name}' failed, retrying (${f+1}/${this.retries})\u2026`)}return{success:!1,error:u.message,raw:null}}},ne=class extends D{constructor(e){super({...e,_isCustomCode:!0}),this.condition=e.condition}async execute(e,o){let r=o&&typeof o.getAll=="function"?o.getAll():e;return{success:!0,output:{nextNode:this.condition(r)},raw:null}}};j();j();import{mkdirSync as lo,existsSync as re,statSync as yn}from"node:fs";import{spawn as bt}from"node:child_process";import{join as se}from"node:path";import{pathToFileURL as uo}from"node:url";import{AsyncLocalStorage as ao}from"node:async_hooks";var Ce=new ao;function Pe(){let t=Ce.getStore();return t||Object.freeze({executionId:process.env.EXECUTION_ID||null,parentExecutionId:process.env.PARENT_EXECUTION_ID||null,depth:0,conversationId:process.env.ZIBBY_CONVERSATION_ID||null,dispatchMode:process.env.DISPATCH_MODE||null})}function _t(t,e){let o=Ce.getStore()||Pe(),r=Object.freeze({executionId:t.executionId,parentExecutionId:t.parentExecutionId??o.executionId??null,depth:(o.depth||0)+(t.executionId!==o.executionId?1:0),conversationId:t.conversationId!==void 0?t.conversationId:o.conversationId??null,dispatchMode:t.dispatchMode??null});return Ce.run(r,e)}var Re=new Map,Be=new Map,It=new Map;function Et(t,e,o={}){if(!t||typeof t!="string")throw new Error("subgraph-registry.register: name required");if(typeof e!="function")throw new Error("subgraph-registry.register: factory must be a function");Re.set(t,e),Be.set(t,"ready"),It.set(t,{...o,cachedAt:Date.now()})}function $t(t,e){Be.set(t,"failed"),It.set(t,{error:e?.message||String(e),failedAt:Date.now()}),Re.delete(t)}function Tt(t){return Be.get(t)==="ready"?Re.get(t):null}var po=process.env.ZIBBY_SUBGRAPH_CACHE_DIR||"/tmp/zibby/subgraphs";function fo(){return Number(process.env.ZIBBY_SUBGRAPH_MAX_DEPTH||10)}function ho(){return`node${(process.versions?.node||"").split(".")[0]||"unknown"}-${process.platform}-${process.arch}`}var A=class extends Error{constructor(e,o){super(`in-process sub-graph fallback: ${e}${o?` (${o})`:""}`),this.fallback=!0,this.reason=e,this.detail=o||null,this.name="SubgraphFallback"}};function go(){let t=(process.env.SUBGRAPH_INTERNAL_URL||"").replace(/\/$/,""),e=(process.env.PROGRESS_API_URL||"").replace(/\/executions\/?$/,""),o=t||e,r=process.env.PROJECT_ID,n=process.env.PROJECT_API_TOKEN;if(!o||!r||!n)throw new A("env","SUBGRAPH_INTERNAL_URL/PROGRESS_API_URL/PROJECT_ID/PROJECT_API_TOKEN missing");return{apiBase:o,projectId:r,authToken:n}}async function mo({apiBase:t,authToken:e,body:o}){let r;try{r=await fetch(`${t}/internal/subgraph/begin`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify(o)})}catch(i){throw new A("network",`begin fetch failed: ${i.message}`)}let n=null;try{n=await r.json()}catch{}if(!r.ok){if(r.status===404){let i=new Error(`Sub-graph child '${o.childWorkflowType}' not found in project`);throw i.code="SUBGRAPH_NOT_FOUND",i.status=404,i}if(r.status===429){let i=n?.quotaInfo||{},s=new Error(`Sub-graph blocked by quota (${i.used??"?"}/${i.limit??"?"} on ${i.planId||"plan"})`);throw s.code="SUBGRAPH_QUOTA_EXCEEDED",s.status=429,s.quotaInfo=i,s}if(r.status===400&&n?.validationErrors){let i=new Error(`Sub-graph rejected input: ${n?.error||n?.message||"validation failed"}`);throw i.code="SUBGRAPH_INVALID_INPUT",i.status=400,i.validationErrors=n.validationErrors,i.missing=n.missing,i}throw new A("begin-status",`begin returned ${r.status}`)}return n?.data||n}async function U({apiBase:t,authToken:e,payload:o}){try{let r=await fetch(`${t}/internal/subgraph/finalize`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify(o)});r.ok||_.warn(`[in-process subgraph] finalize returned ${r.status} for ${o.childExecutionId}`)}catch(r){_.warn(`[in-process subgraph] finalize failed: ${r.message}`)}}async function So(t,e){let o=se(e,".ready"),r=se(e,"graph.mjs");if(re(o)&&re(r))return;lo(e,{recursive:!0});let n=se(e,".lock"),i=!1;try{let{openSync:s,closeSync:a}=await import("node:fs"),c=s(n,"wx");a(c),i=!0}catch(s){if(s.code!=="EEXIST")throw s}if(!i){let s=Date.now()+3e4;for(;Date.now()<s;){if(re(o)&&re(r))return;await new Promise(a=>setTimeout(a,100))}throw new A("bundle-extract-timeout","sibling extract did not complete within 30s")}try{await new Promise((c,l)=>{let u=bt("curl",["-fsSL",t],{stdio:["ignore","pipe","inherit"]}),f=bt("tar",["-xzf","-","-C",e],{stdio:["pipe","inherit","inherit"]});u.stdout.pipe(f.stdin);let d,S,h=()=>{if(d!==void 0&&S!==void 0){if(d!==0)return l(new Error(`curl exited ${d}`));if(S!==0)return l(new Error(`tar exited ${S}`));c()}};u.on("close",g=>{d=g,h()}),f.on("close",g=>{S=g,h()}),u.on("error",l),f.on("error",l)});let{writeFileSync:s,unlinkSync:a}=await import("node:fs");s(o,"");try{a(n)}catch{}}catch(s){try{let{unlinkSync:a}=await import("node:fs");a(n)}catch{}throw new A("bundle-extract-failed",s.message)}}async function wo(t){let e=se(t,"graph.mjs");if(!re(e))throw new A("entry-missing",`graph.mjs missing under ${t}`);let o;try{o=await import(uo(e).href)}catch(n){throw new A("import-failed",`${n?.code||n?.name||"unknown"}: ${n.message}`)}let r=o.default||Object.values(o).find(n=>typeof n=="function"&&n.prototype?.buildGraph);if(!r)throw new A("entry-class-missing","no buildGraph() class export found");return r}async function vt(t,e={}){if(!t||typeof t!="string")throw new Error("runInProcessSubgraph: workflowName (string) is required");let o=Pe(),r=fo();if((o.depth||0)>=r)throw new A("depth-exceeded",`depth ${o.depth} \u2265 MAX_DEPTH ${r}`);let n;try{n=go()}catch(m){throw m}_.debug(`[in-process subgraph] begin '${t}' parent=${o.executionId||"<root>"}`);let i=await mo({apiBase:n.apiBase,authToken:n.authToken,body:{parentExecutionId:o.executionId,childWorkflowType:t,input:e.input||{},...e.conversationId?{conversationId:e.conversationId}:{}}}),{childExecutionId:s,runtimeTag:a,bundlePresignedUrl:c,sourcesPresignedUrl:l,workflowVersion:u,workflowUuid:f,bundleReady:d}=i,S=ho();if(a&&a!==S)throw await U({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"canceled",error:{message:`runtimeTag mismatch: parent=${S} child=${a}`,code:"RUNTIME_MISMATCH"}}}),new A("runtime-mismatch",`${S} vs ${a}`);if(!d||!c)throw await U({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"canceled",error:{message:"bundle not ready for in-process; falling back to HTTP",code:"NO_BUNDLE"}}}),new A("no-bundle","workflow bundle not built yet");let h=Tt(t);if(!h){let m=se(po,`${f}@${u||"0"}`);try{await So(c,m)}catch(T){throw T.fallback&&await U({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"failed",error:{message:T.message,code:T.reason}}}),T}try{h=await wo(m),Et(t,h,{workflowUuid:f,version:u,runtimeTag:a,cacheDir:m})}catch(T){throw $t(t,T),await U({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"failed",error:{message:T.message,code:T.reason||"IMPORT_FAILED"}}}),T.fallback?T:new A("import-failed",T.message)}}let g=Date.now(),$=await(typeof h=="function"&&h.prototype?.buildGraph?new h:h).buildGraph(),p={...e.input||{}},w,b;try{w=await _t({executionId:s,parentExecutionId:o.executionId,conversationId:e.conversationId!==void 0?e.conversationId:o.conversationId,dispatchMode:"inprocess"},()=>$.run(e.parentAgent,p,{signal:e.signal})),b=w&&typeof w=="object"&&"state"in w?w.state:w}catch(m){throw await U({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"failed",error:{message:m.message,code:m.code||"CHILD_THREW",stack:m.stack},durationMs:Date.now()-g}}),m}if(w&&typeof w=="object"&&w.stoppedExternally){await U({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"canceled",finalState:b,durationMs:Date.now()-g}});let m=new Error(`Sub-graph '${t}' canceled by parent abort`);throw m.code="SUBGRAPH_CANCELED",m.subgraphJobId=s,m}return await U({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"completed",finalState:b,durationMs:Date.now()-g}}),{finalState:b,executionId:s}}var yo=2e3,_o=600*1e3,Io=new Set(["completed","failed","canceled","timeout"]);function Eo(){let t=process.env.PROGRESS_API_URL;if(!t)throw new Error("Sub-graph dispatch requires PROGRESS_API_URL env var (set automatically on cloud runs). Sub-graphs are not supported in local in-process runs yet \u2014 deploy the parent and child to cloud.");return t.replace(/\/executions\/?$/,"")}function $o(){let t=process.env.PROJECT_ID;if(!t)throw new Error("Sub-graph dispatch requires PROJECT_ID env var.");return t}function To(){let t=process.env.PROJECT_API_TOKEN;if(!t)throw new Error("Sub-graph dispatch requires PROJECT_API_TOKEN env var.");return t}function bo(){return process.env.EXECUTION_ID||null}function xt(t,e){return e==null?t:typeof e=="function"?e(t):typeof e=="string"?e.split(".").reduce((o,r)=>o==null?o:o[r],t):t}async function Me(t,e={}){if(!t||typeof t!="string")throw new Error("dispatchSubgraph: workflowName (string) is required");if(process.env.ZIBBY_INPROCESS_SUBGRAPH!=="0"&&!e.async)try{_.debug(`[sub-graph] trying in-process for '${t}'`);let{finalState:p}=await vt(t,{input:e.input,conversationId:e.conversationId,signal:e.signal,parentAgent:e.parentAgent}),w=xt(p,e.output);return _.info(`[sub-graph] '${t}' completed in-process`),w}catch(p){if(p instanceof A||p?.fallback)_.info(`[sub-graph] in-process fallback for '${t}': ${p.reason||"unknown"} \u2014 using HTTP`);else throw p}let o=Eo(),r=$o(),n=To(),i=bo(),s=`${o}/projects/${encodeURIComponent(r)}/workflows/${encodeURIComponent(t)}/trigger`,a={input:e.input||{},...i?{parentExecutionId:i}:{},...e.conversationId?{conversationId:e.conversationId}:{}};_.info(`[sub-graph] dispatching '${t}' (${e.async?"async":"sync"}) from parent ${i||"<none>"}`);let c=await fetch(s,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${n}`},body:JSON.stringify(a)});if(!c.ok){let p=null,w="";try{p=await c.json(),w=p?.error||p?.message||JSON.stringify(p)}catch{w=await c.text().catch(()=>"")}if(c.status===429){let m=p?.quotaInfo||{},T=new Error(`Sub-graph '${t}' blocked by execution quota (${m.used??"?"}/${m.limit??"?"} on plan ${m.planId||"unknown"}). Sub-workflow runs count toward the same monthly cap as user-triggered runs.`);throw T.code="SUBGRAPH_QUOTA_EXCEEDED",T.status=429,T.subgraph=t,T.quotaInfo=m,T}if(c.status===400){let m=new Error(`Sub-graph '${t}' rejected input: ${w}`);throw m.code="SUBGRAPH_INVALID_INPUT",m.status=400,m.subgraph=t,m.validationErrors=p?.validationErrors||null,m.missing=p?.missing||null,m}let b=new Error(`Sub-graph '${t}' trigger rejected (${c.status}): ${w}`);throw b.code="SUBGRAPH_TRIGGER_FAILED",b.status=c.status,b.subgraph=t,b}let l=await c.json(),u=l?.data?.jobId||l?.jobId;if(!u)throw new Error(`Sub-graph '${t}' trigger returned no jobId: ${JSON.stringify(l).slice(0,200)}`);if(e.async)return _.info(`[sub-graph] async dispatch of '${t}' \u2192 jobId=${u} (not waiting)`),{jobId:u,status:"accepted",workflow:t};let f=Number.isFinite(e.timeoutMs)?e.timeoutMs:_o,d=Number.isFinite(e.pollIntervalMs)?e.pollIntervalMs:yo,S=`${o}/executions/${encodeURIComponent(u)}`,h=Date.now()+f,g="accepted",I=0;for(;Date.now()<h;){await new Promise(m=>setTimeout(m,d)),I+=1;let p=await fetch(S,{headers:{Authorization:`Bearer ${n}`}});if(!p.ok){if(p.status>=500){_.warn(`[sub-graph] status poll for ${u} returned ${p.status}, will retry`);continue}throw new Error(`Sub-graph status poll failed for ${u}: ${p.status}`)}let w=await p.json(),b=w?.data||w?.execution||w;if(g=b?.status||g,Io.has(g)){if(g!=="completed"){let G=new Error(`Sub-graph '${t}' (${u}) ended in status '${g}'`);throw G.subgraphJobId=u,G.subgraphStatus=g,G}let m=b?.finalState||b?.state||{},T=xt(m,e.output);return _.info(`[sub-graph] '${t}' (${u}) completed after ${I} polls`),T}}let $=new Error(`Sub-graph '${t}' (${u}) timed out after ${Math.round(f/1e3)}s (last status: ${g})`);throw $.subgraphJobId=u,$.subgraphStatus=g,$}import{existsSync as kt,readFileSync as vo}from"node:fs";import{join as De,dirname as At}from"node:path";var ie=class{static async loadContext(e,o,r={}){let n={},i=r.filenames||["CONTEXT.md","AGENTS.md"];if(e){let a=At(De(o,e));for(let c of i){let l=await this.findAndMergeContextFiles(c,a,o);if(l){let u=c.replace(/\.[^.]+$/,"").toLowerCase();n[u]=l}}}let s=r.discovery||{};for(let[a,c]of Object.entries(s))try{let l=De(o,c);kt(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,o,r){let n=[],i=o;for(;i.startsWith(r);){let s=De(i,e);if(kt(s))try{n.unshift(await this.loadFile(s))}catch(c){console.warn(`[workflow] could not load ${e} from ${s}: ${c.message}`)}let a=At(i);if(a===i)break;i=a}return n.length===0?null:n.every(s=>typeof s=="string")?n.join(`
|
|
29
|
+
${r}`);let a=n(),c=a.cwd||process.cwd(),l=a.sessionPath;try{if(l){let p=Re(l,W);if(_t(p)){let h=JSON.parse(yt(p,"utf-8"));h.currentNode=this.name,Pe(p,JSON.stringify(h,null,2),"utf-8")}let u=Re(l,"..",W);if(_t(u))try{let h=JSON.parse(yt(u,"utf-8"));h.currentNode=this.name,Pe(u,JSON.stringify(h,null,2),"utf-8")}catch{}}}catch(p){_.debug(`[workflow] could not update session info: ${p.message}`)}let d=null;for(let p=0;p<=this.retries;p++)try{_.debug(`[workflow] node '${this.name}' attempt ${p}`);let u=n().config||{},h=u.agents||{},m=this.config.agent??h[this.name]??null,w={state:n()};m&&(w.preferredAgent=m);let I={workspace:c,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:l,config:u,nodeName:this.name,timeout:this.config?.timeout||3e5},E=e?._coreInvokeAgent;E||(E=(await Promise.resolve().then(()=>(q(),se))).invokeAgent);let f=await E(r,w,I),$,g;if(typeof f=="string"?($=f,g=null):f.structured?($=f.raw||JSON.stringify(f.structured,null,2),g=f.structured):($=f.raw||JSON.stringify(f,null,2),g=f.extracted||null),l)try{let S=Re(l,this.name,"raw_stream_output.txt");co(lo(S),{recursive:!0}),Pe(S,typeof $=="string"?$:JSON.stringify($),"utf-8")}catch(S){_.debug(`[workflow] could not save raw output: ${S.message}`)}if(this.isZodSchema&&g){_.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(g,null,2)}`);let S=g;if(typeof this.onComplete=="function")try{S=await this.onComplete(n(),g)}catch(R){_.warn(`[workflow] onComplete hook failed: ${R.message}`)}return{success:!0,output:S,raw:$}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(n(),{raw:$}),raw:$}}catch(S){throw new Error(`onComplete failed: ${S.message}`,{cause:S})}if(this.parser){let S=this.parser.parse($);return _.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(S,null,2)}`),x.step("Output parsed"),{success:!0,output:S,raw:$}}return{success:!0,output:$,raw:$}}catch(u){d=u,p<this.retries&&_.info(`[workflow] node '${this.name}' failed, retrying (${p+1}/${this.retries})\u2026`)}return{success:!1,error:d.message,raw:null}}},ie=class extends j{constructor(e){super({...e,_isCustomCode:!0}),this.condition=e.condition}async execute(e,o){let n=o&&typeof o.getAll=="function"?o.getAll():e;return{success:!0,output:{nextNode:this.condition(n)},raw:null}}};F();F();import{mkdirSync as fo,existsSync as J,statSync as xt,readdirSync as kt,rmSync as ho}from"node:fs";import{spawn as vt}from"node:child_process";import{join as U}from"node:path";import{pathToFileURL as go}from"node:url";import{AsyncLocalStorage as uo}from"node:async_hooks";var Be=new uo;function ae(){let t=Be.getStore();return t||Object.freeze({executionId:process.env.EXECUTION_ID||null,parentExecutionId:process.env.PARENT_EXECUTION_ID||null,depth:0,conversationId:process.env.ZIBBY_CONVERSATION_ID||null,dispatchMode:process.env.DISPATCH_MODE||null})}function It(t,e){let o=Be.getStore()||ae(),n=Object.freeze({executionId:t.executionId,parentExecutionId:t.parentExecutionId??o.executionId??null,depth:(o.depth||0)+(t.executionId!==o.executionId?1:0),conversationId:t.conversationId!==void 0?t.conversationId:o.conversationId??null,dispatchMode:t.dispatchMode??null});return Be.run(n,e)}var Me=new Map,De=new Map,Et=new Map;function $t(t,e,o={}){if(!t||typeof t!="string")throw new Error("subgraph-registry.register: name required");if(typeof e!="function")throw new Error("subgraph-registry.register: factory must be a function");Me.set(t,e),De.set(t,"ready"),Et.set(t,{...o,cachedAt:Date.now()})}function bt(t,e){De.set(t,"failed"),Et.set(t,{error:e?.message||String(e),failedAt:Date.now()}),Me.delete(t)}function Tt(t){return De.get(t)==="ready"?Me.get(t):null}var me=process.env.ZIBBY_SUBGRAPH_CACHE_DIR||"/tmp/zibby/subgraphs";function mo(){return`node${(process.versions?.node||"").split(".")[0]||"unknown"}-${process.platform}-${process.arch}`}var A=class extends Error{constructor(e,o){super(`in-process sub-graph fallback: ${e}${o?` (${o})`:""}`),this.fallback=!0,this.reason=e,this.detail=o||null,this.name="SubgraphFallback"}};function So(){let t=(process.env.SUBGRAPH_INTERNAL_URL||"").replace(/\/$/,""),e=(process.env.PROGRESS_API_URL||"").replace(/\/executions\/?$/,""),o=t||e,n=process.env.PROJECT_ID,s=process.env.PROJECT_API_TOKEN;if(!o||!n||!s)throw new A("env","SUBGRAPH_INTERNAL_URL/PROGRESS_API_URL/PROJECT_ID/PROJECT_API_TOKEN missing");return{apiBase:o,projectId:n,authToken:s}}async function wo({apiBase:t,authToken:e,body:o}){let n;try{n=await fetch(`${t}/internal/subgraph/begin`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify(o)})}catch(r){throw new A("network",`begin fetch failed: ${r.message}`)}let s=null;try{s=await n.json()}catch{}if(!n.ok){if(n.status===404){let r=new Error(`Sub-graph child '${o.childWorkflowType}' not found in project`);throw r.code="SUBGRAPH_NOT_FOUND",r.status=404,r}if(n.status===429){let r=s?.quotaInfo||{},i=new Error(`Sub-graph blocked by quota (${r.used??"?"}/${r.limit??"?"} on ${r.planId||"plan"})`);throw i.code="SUBGRAPH_QUOTA_EXCEEDED",i.status=429,i.quotaInfo=r,i}if(n.status===400&&s?.validationErrors){let r=new Error(`Sub-graph rejected input: ${s?.error||s?.message||"validation failed"}`);throw r.code="SUBGRAPH_INVALID_INPUT",r.status=400,r.validationErrors=s.validationErrors,r.missing=s.missing,r}throw new A("begin-status",`begin returned ${n.status}`)}return s?.data||s}async function H({apiBase:t,authToken:e,payload:o}){try{let n=await fetch(`${t}/internal/subgraph/finalize`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify(o)});n.ok||_.warn(`[in-process subgraph] finalize returned ${n.status} for ${o.childExecutionId}`)}catch(n){_.warn(`[in-process subgraph] finalize failed: ${n.message}`)}}async function yo(t,e){let o=U(e,".ready"),n=U(e,"graph.mjs");if(J(o)&&J(n))return;fo(e,{recursive:!0});let s=U(e,".lock"),r=!1;try{let{openSync:i,closeSync:a}=await import("node:fs"),c=i(s,"wx");a(c),r=!0}catch(i){if(i.code!=="EEXIST")throw i}if(!r){let i=Date.now()+3e4;for(;Date.now()<i;){if(J(o)&&J(n))return;await new Promise(a=>setTimeout(a,100))}throw new A("bundle-extract-timeout","sibling extract did not complete within 30s")}try{await new Promise((c,l)=>{let d=vt("curl",["-fsSL",t],{stdio:["ignore","pipe","inherit"]}),p=vt("tar",["-xzf","-","-C",e],{stdio:["pipe","inherit","inherit"]});d.stdout.pipe(p.stdin);let u,h,m=()=>{if(u!==void 0&&h!==void 0){if(u!==0)return l(new Error(`curl exited ${u}`));if(h!==0)return l(new Error(`tar exited ${h}`));c()}};d.on("close",w=>{u=w,m()}),p.on("close",w=>{h=w,m()}),d.on("error",l),p.on("error",l)});let{writeFileSync:i,unlinkSync:a}=await import("node:fs");i(o,"");try{a(s)}catch{}}catch(i){try{let{unlinkSync:a}=await import("node:fs");a(s)}catch{}throw new A("bundle-extract-failed",i.message)}}async function _o(t){let e=U(t,"graph.mjs");if(!J(e))throw new A("entry-missing",`graph.mjs missing under ${t}`);let o;try{o=await import(go(e).href)}catch(s){throw new A("import-failed",`${s?.code||s?.name||"unknown"}: ${s.message}`)}let n=o.default||Object.values(o).find(s=>typeof s=="function"&&s.prototype?.buildGraph);if(!n)throw new A("entry-class-missing","no buildGraph() class export found");return n}async function At(t,e={}){if(!t||typeof t!="string")throw new Error("runInProcessSubgraph: workflowName (string) is required");let o=ae(),n;try{n=So()}catch(g){throw g}_.debug(`[in-process subgraph] begin '${t}' parent=${o.executionId||"<root>"}`);let s=await wo({apiBase:n.apiBase,authToken:n.authToken,body:{parentExecutionId:o.executionId,childWorkflowType:t,input:e.input||{},...e.conversationId?{conversationId:e.conversationId}:{}}}),{childExecutionId:r,runtimeTag:i,bundlePresignedUrl:a,sourcesPresignedUrl:c,workflowVersion:l,workflowUuid:d,bundleReady:p}=s,u=mo();if(i&&i!==u)throw await H({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:r,status:"canceled",error:{message:`runtimeTag mismatch: parent=${u} child=${i}`,code:"RUNTIME_MISMATCH"}}}),new A("runtime-mismatch",`${u} vs ${i}`);if(!p||!a)throw await H({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:r,status:"canceled",error:{message:"bundle not ready for in-process; falling back to HTTP",code:"NO_BUNDLE"}}}),new A("no-bundle","workflow bundle not built yet");let h=Tt(t);if(!h){let g=U(me,`${d}@${l||"0"}`);try{await yo(a,g);try{Eo()}catch{}}catch(S){throw S.fallback&&await H({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:r,status:"failed",error:{message:S.message,code:S.reason}}}),S}try{h=await _o(g),$t(t,h,{workflowUuid:d,version:l,runtimeTag:i,cacheDir:g})}catch(S){throw bt(t,S),await H({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:r,status:"failed",error:{message:S.message,code:S.reason||"IMPORT_FAILED"}}}),S.fallback?S:new A("import-failed",S.message)}}let m=Date.now(),I=await(typeof h=="function"&&h.prototype?.buildGraph?new h:h).buildGraph(),E={...e.input||{}},f,$;try{f=await It({executionId:r,parentExecutionId:o.executionId,conversationId:e.conversationId!==void 0?e.conversationId:o.conversationId,dispatchMode:"inprocess"},()=>I.run(e.parentAgent,E,{signal:e.signal})),$=f&&typeof f=="object"&&"state"in f?f.state:f}catch(g){throw await H({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:r,status:"failed",error:{message:g.message,code:g.code||"CHILD_THREW",stack:g.stack},durationMs:Date.now()-m}}),g}if(f&&typeof f=="object"&&f.stoppedExternally){await H({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:r,status:"canceled",finalState:$,durationMs:Date.now()-m}});let g=new Error(`Sub-graph '${t}' canceled by parent abort`);throw g.code="SUBGRAPH_CANCELED",g.subgraphJobId=r,g}return await H({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:r,status:"completed",finalState:$,durationMs:Date.now()-m}}),{finalState:$,executionId:r}}function Io(t){let e=0,o=[t];for(;o.length;){let n=o.pop(),s;try{s=xt(n)}catch{continue}if(s.isDirectory()){let r;try{r=kt(n)}catch{continue}for(let i of r)o.push(U(n,i))}else e+=s.size}return e}function Eo({cap:t=Number(process.env.ZIBBY_SUBGRAPH_CACHE_CAP_BYTES||2*1024*1024*1024)}={}){try{if(!J(me))return{evicted:0,freedBytes:0};let e=kt(me),o=[],n=0;for(let a of e){let c=U(me,a),l;try{l=xt(c)}catch{continue}let d=l.isDirectory()?Io(c):l.size;n+=d,o.push({name:a,full:c,size:d,mtimeMs:l.mtimeMs})}if(n<=t)return{evicted:0,freedBytes:0,totalBytes:n};o.sort((a,c)=>a.mtimeMs-c.mtimeMs);let s=Math.floor(t*.7),r=0,i=0;for(let a of o){if(n-r<=s)break;if(!J(U(a.full,".lock")))try{ho(a.full,{recursive:!0,force:!0}),r+=a.size,i+=1}catch(c){_.debug(`[sub-graph cache] evict skip ${a.name}: ${c.message}`)}}return i>0&&_.info(`[sub-graph cache] evicted ${i} entr(y/ies), freed ${(r/1024/1024).toFixed(1)}MB`),{evicted:i,freedBytes:r,totalBytes:n-r}}catch(e){return _.debug(`[sub-graph cache] evict failed: ${e.message}`),{evicted:0,freedBytes:0}}}var $o=2e3,bo=600*1e3,To=new Set(["completed","failed","canceled","timeout"]);function vo(){let t=process.env.PROGRESS_API_URL;if(!t)throw new Error("Sub-graph dispatch requires PROGRESS_API_URL env var (set automatically on cloud runs). Sub-graphs are not supported in local in-process runs yet \u2014 deploy the parent and child to cloud.");return t.replace(/\/executions\/?$/,"")}function xo(){let t=process.env.PROJECT_ID;if(!t)throw new Error("Sub-graph dispatch requires PROJECT_ID env var.");return t}function ko(){let t=process.env.PROJECT_API_TOKEN;if(!t)throw new Error("Sub-graph dispatch requires PROJECT_API_TOKEN env var.");return t}function Ao(){return process.env.EXECUTION_ID||null}function Nt(t,e){return e==null?t:typeof e=="function"?e(t):typeof e=="string"?e.split(".").reduce((o,n)=>o==null?o:o[n],t):t}async function Le(t,e={}){if(!t||typeof t!="string")throw new Error("dispatchSubgraph: workflowName (string) is required");let o=ae(),n=Number(process.env.ZIBBY_SUBGRAPH_MAX_DEPTH||10);if((o.depth||0)>=n)throw new Error(`dispatchSubgraph('${t}'): sub-graph depth ${o.depth} reached cap of ${n}. Restructure the graph or raise ZIBBY_SUBGRAPH_MAX_DEPTH.`);if(process.env.ZIBBY_INPROCESS_SUBGRAPH!=="0"&&!e.async)try{_.debug(`[sub-graph] trying in-process for '${t}'`);let{finalState:g}=await At(t,{input:e.input,conversationId:e.conversationId,signal:e.signal,parentAgent:e.parentAgent}),S=Nt(g,e.output);return _.info(`[sub-graph] '${t}' completed in-process`),S}catch(g){if(g instanceof A||g?.fallback)_.info(`[sub-graph] in-process fallback for '${t}': ${g.reason||"unknown"} \u2014 using HTTP`);else throw g}let s=vo(),r=xo(),i=ko(),a=Ao(),c=`${s}/projects/${encodeURIComponent(r)}/workflows/${encodeURIComponent(t)}/trigger`,l={input:e.input||{},...a?{parentExecutionId:a}:{},...e.conversationId?{conversationId:e.conversationId}:{}};_.info(`[sub-graph] dispatching '${t}' (${e.async?"async":"sync"}) from parent ${a||"<none>"}`);let d=await fetch(c,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${i}`},body:JSON.stringify(l)});if(!d.ok){let g=null,S="";try{g=await d.json(),S=g?.error||g?.message||JSON.stringify(g)}catch{S=await d.text().catch(()=>"")}if(d.status===429){let v=g?.quotaInfo||{},D=new Error(`Sub-graph '${t}' blocked by execution quota (${v.used??"?"}/${v.limit??"?"} on plan ${v.planId||"unknown"}). Sub-workflow runs count toward the same monthly cap as user-triggered runs.`);throw D.code="SUBGRAPH_QUOTA_EXCEEDED",D.status=429,D.subgraph=t,D.quotaInfo=v,D}if(d.status===400){let v=new Error(`Sub-graph '${t}' rejected input: ${S}`);throw v.code="SUBGRAPH_INVALID_INPUT",v.status=400,v.subgraph=t,v.validationErrors=g?.validationErrors||null,v.missing=g?.missing||null,v}let R=new Error(`Sub-graph '${t}' trigger rejected (${d.status}): ${S}`);throw R.code="SUBGRAPH_TRIGGER_FAILED",R.status=d.status,R.subgraph=t,R}let p=await d.json(),u=p?.data?.jobId||p?.jobId;if(!u)throw new Error(`Sub-graph '${t}' trigger returned no jobId: ${JSON.stringify(p).slice(0,200)}`);if(e.async)return _.info(`[sub-graph] async dispatch of '${t}' \u2192 jobId=${u} (not waiting)`),{jobId:u,status:"accepted",workflow:t};let h=Number.isFinite(e.timeoutMs)?e.timeoutMs:bo,m=Number.isFinite(e.pollIntervalMs)?e.pollIntervalMs:$o,w=`${s}/executions/${encodeURIComponent(u)}`,I=Date.now()+h,E="accepted",f=0;for(;Date.now()<I;){await new Promise(v=>setTimeout(v,m)),f+=1;let g=await fetch(w,{headers:{Authorization:`Bearer ${i}`}});if(!g.ok){if(g.status>=500){_.warn(`[sub-graph] status poll for ${u} returned ${g.status}, will retry`);continue}throw new Error(`Sub-graph status poll failed for ${u}: ${g.status}`)}let S=await g.json(),R=S?.data||S?.execution||S;if(E=R?.status||E,To.has(E)){if(E!=="completed"){let y=new Error(`Sub-graph '${t}' (${u}) ended in status '${E}'`);throw y.subgraphJobId=u,y.subgraphStatus=E,y}let v=R?.finalState||R?.state||{},D=Nt(v,e.output);return _.info(`[sub-graph] '${t}' (${u}) completed after ${f} polls`),D}}let $=new Error(`Sub-graph '${t}' (${u}) timed out after ${Math.round(h/1e3)}s (last status: ${E})`);throw $.subgraphJobId=u,$.subgraphStatus=E,$}import{existsSync as Ot,readFileSync as No}from"node:fs";import{join as je,dirname as Ct}from"node:path";var ce=class{static async loadContext(e,o,n={}){let s={},r=n.filenames||["CONTEXT.md","AGENTS.md"];if(e){let a=Ct(je(o,e));for(let c of r){let l=await this.findAndMergeContextFiles(c,a,o);if(l){let d=c.replace(/\.[^.]+$/,"").toLowerCase();s[d]=l}}}let i=n.discovery||{};for(let[a,c]of Object.entries(i))try{let l=je(o,c);Ot(l)&&(s[a]=await this.loadFile(l))}catch(l){console.warn(`[workflow] could not load context '${a}' from '${c}': ${l.message}`)}return s}static async findAndMergeContextFiles(e,o,n){let s=[],r=o;for(;r.startsWith(n);){let i=je(r,e);if(Ot(i))try{s.unshift(await this.loadFile(i))}catch(c){console.warn(`[workflow] could not load ${e} from ${i}: ${c.message}`)}let a=Ct(r);if(a===r)break;r=a}return s.length===0?null:s.every(i=>typeof i=="string")?s.join(`
|
|
30
30
|
|
|
31
31
|
---
|
|
32
32
|
|
|
33
|
-
`):
|
|
33
|
+
`):s.every(i=>typeof i=="object")?Object.assign({},...s):s[s.length-1]}static async loadFile(e){let o=No(e,"utf-8");if(e.endsWith(".json"))return JSON.parse(o);if(e.endsWith(".js")||e.endsWith(".mjs")){let{pathToFileURL:n}=await import("url"),s=await import(n(e).href);return s.default||s}return o}};import{mkdirSync as Bt,existsSync as Fe,writeFileSync as Pt,unlinkSync as Oo}from"node:fs";import{join as Y,resolve as Mt}from"node:path";import{config as Co}from"dotenv";import{zodToJsonSchema as Rt}from"zod-to-json-schema";import Po from"handlebars";function Ro({traceFrom:t,sessionId:e,sessionPath:o,idSource:n,mkdirFresh:s}){if(!(process.env.ZIBBY_SESSION_LOG==="1"||process.env.ZIBBY_SESSION_LOG==="true"))return;let i=typeof process.ppid=="number"?process.ppid:"n/a",a=`[zibby:session] from=${t} pid=${process.pid} ppid=${i} sessionId=${e} source=${n} mkdir=${s?"yes":"no"} path=${o}`;if(console.log(a),process.env.ZIBBY_TRACE_SESSION==="1"||process.env.ZIBBY_TRACE_SESSION==="true"){let d=(new Error("session trace").stack||"").split(`
|
|
34
34
|
`).slice(2,14).join(`
|
|
35
35
|
`);console.log(`[zibby:session] stack (${t}):
|
|
36
|
-
${
|
|
37
|
-
\u{1F6D1} External stop requested \u2014 ending workflow.`),x.step("Workflow stopped externally"),{success:!0,state:
|
|
38
|
-
${
|
|
36
|
+
${d}`)}}function Dt(){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 Lt(){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 Mt(String(e).trim())}catch{return String(e).trim()}}function jt(){Dt()||(delete process.env.ZIBBY_SESSION_PATH,delete process.env.ZIBBY_SESSION_ID)}function Ft({sessionPath:t,sessionId:e}){t&&typeof t=="string"&&(process.env.ZIBBY_SESSION_PATH=t),e!=null&&String(e).trim()!==""&&(process.env.ZIBBY_SESSION_ID=String(e).trim())}function Ut(t={}){let e=xe.map(r=>process.env[r]).find(Boolean),o=Math.random().toString(36).slice(2,6),n=e||`${Date.now()}_${o}`,s=t.paths?.sessionPrefix;return s?`${s}_${n}`:n}function Wt({cwd:t=process.cwd(),config:e={},initialState:o={},traceFrom:n="resolveWorkflowSession"}={}){let s=o.sessionPath,r=o.sessionTimestamp,i="initialState.sessionPath";if(!s&&process.env.ZIBBY_SESSION_PATH)try{let l=Mt(String(process.env.ZIBBY_SESSION_PATH));l&&(s=l,i="ZIBBY_SESSION_PATH")}catch{}let a;if(s)a=String(s).split(/[/\\]/).filter(Boolean).pop(),r==null&&(r=Date.now());else{let l=process.env.ZIBBY_SESSION_ID&&String(process.env.ZIBBY_SESSION_ID).trim();if(l)a=l,i="ZIBBY_SESSION_ID";else{let p=e.sessionId!=null?String(e.sessionId).trim():"";p&&p!=="last"?(a=p,i="config.sessionId"):(a=Ut(e),i="generated")}r=r??Date.now();let d=e.paths?.output||ne;s=Y(t,d,Te,a)}let c=!Fe(s);return c&&Bt(s,{recursive:!0}),(c||i!=="initialState.sessionPath")&&Ro({traceFrom:n,sessionId:a,sessionPath:s,idSource:i,mkdirFresh:c}),Ft({sessionPath:s,sessionId:a}),{sessionPath:s,sessionId:a,sessionTimestamp:r}}var le=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.inputSchema=e.inputSchema||null,this.contextSchema=e.contextSchema||null,this.nodePrompts=new Map,this.nodeOptions=new Map,this._invokeAgent=e.invokeAgent||null,this._compiledPrompts=new Map}setInputSchema(e){return this.inputSchema=e,this}setContextSchema(e){return this.contextSchema=e,this}setStateSchema(e){return this.stateSchema=e,this}getInputSchema(){return this.inputSchema}getContextSchema(){return this.contextSchema}getStateSchema(){return this.stateSchema}_runtimeSchema(){if(this.inputSchema&&this.contextSchema)try{return this.inputSchema.merge(this.contextSchema)}catch{}return this.inputSchema&&!this.contextSchema?this.inputSchema:this.stateSchema}addNode(e,o,n={}){if(!(o instanceof j)&&o&&typeof o=="object"&&typeof o.workflow=="string"){let r=o,i={name:e,_isCustomCode:!0,retries:r.retries,onComplete:r.onComplete,execute:async c=>{let l=c?.state&&typeof c.state.getAll=="function"?c.state.getAll():c,d;return typeof r.input=="function"?d=r.input(l):r.input&&typeof r.input=="object"?d=r.input:d={},Le(r.workflow,{input:d,async:r.async===!0,conversationId:typeof r.conversationId=="function"?r.conversationId(l):r.conversationId,output:r.output,timeoutMs:r.timeoutMs,pollIntervalMs:r.pollIntervalMs,signal:l?._signal,parentAgent:c?.agent})}},a=new j(i);return a.name=e,this.nodes.set(e,a),n.prompt&&this.nodePrompts.set(e,n.prompt),Object.keys(n).length>0&&this.nodeOptions.set(e,n),this}let s=o instanceof j?o:new j(o);return s.name=e,this.nodes.set(e,s),n.prompt&&this.nodePrompts.set(e,n.prompt),Object.keys(n).length>0&&this.nodeOptions.set(e,n),this}addConditionalNode(e,o){return this.nodes.set(e,new ie({...o,name:e})),this}addEdge(e,o){return this.edges.set(e,o),this}setNodeType(e,o){return this.nodeTypeMap.set(e,o),this}addConditionalEdges(e,o,{labels:n}={}){return this.edges.set(e,{conditional:!0,routes:o,labels:n}),typeof o=="function"&&this.conditionalCodeMap.set(e,o.toString()),this}setEntryPoint(e){return this.entryPoint=e,this}use(e){return typeof e=="function"&&this.middleware.push(e),this}_composeMiddleware(e,o,n,s,r){let i=n;for(let a=e.length-1;a>=0;a--){let c=e[a],l=i;i=()=>c(o,l,s,r)}return i()}serialize(){let e=[],o={};for(let[l,d]of this.nodes){let p=this.nodeTypeMap.get(l)||l;e.push({id:l,type:p,data:{nodeType:p,label:l}});let u={};d._isCustomCode&&typeof d.execute=="function"&&(u.customCode=d.execute.toString());let h=this.nodePrompts.get(l);if(h&&(u.prompt=h),typeof d.customExecute=="function"&&(u.executeCode=d.customExecute.toString()),d.outputSchema)try{if(typeof d.outputSchema._def<"u"){let I=Rt(d.outputSchema,{target:"openApi3"});u.outputSchema={jsonSchema:I,variables:this._flattenJsonSchemaToVariables(I)}}else u.outputSchema={schema:d.outputSchema}}catch(I){console.warn(`[workflow] failed to convert schema for ${l}:`,I.message)}let m=(this.resolvedToolsMap||{})[l];m?.toolIds&&(u.tools=m.toolIds);let w=Array.isArray(d?.config?.skills)?d.config.skills:Array.isArray(d?.skills)?d.skills:null;w&&w.length>0&&(u.skills=[...w]),Object.keys(u).length>0&&(o[l]=u)}let n=[];for(let[l,d]of this.edges)if(typeof d=="string")n.push({source:l,target:d});else if(d.conditional){let p=this.conditionalCodeMap.get(l)||d.routes.toString(),u=this._inferConditionalTargets(d.routes),h=d.labels||{};for(let m of u){let w={source:l,target:m,data:{conditionalCode:p}};h[m]&&(w.label=h[m]),n.push(w)}}let s=l=>{if(!l)return null;try{return Rt(l,{target:"openApi3"})}catch{return null}},r=this._runtimeSchema(),i=s(r||this.stateSchema),a=s(this.inputSchema),c=s(this.contextSchema);return{nodes:e,edges:n,nodeConfigs:o,stateSchema:i,inputSchema:a,contextSchema:c}}_inferConditionalTargets(e){let o=e.toString(),n=new Set,s=/return\s+['"]([^'"]+)['"]/g,r;for(;(r=s.exec(o))!==null;)n.add(r[1]);return[...n]}_flattenJsonSchemaToVariables(e,o=""){let n=e;if(e.$ref&&e.definitions){let s=e.$ref.replace("#/definitions/","");n=e.definitions[s]||e}return this._flattenSchema(n,o)}_flattenSchema(e,o=""){if(!e||typeof e!="object")return[];let n=[],s=e.properties||{},r=e.required||[];for(let[i,a]of Object.entries(s)){let c=o?`${o}.${i}`:i;n.push({path:c,type:a.type||"unknown",label:a.description||this._formatLabel(i),optional:!r.includes(i)}),a.type==="object"&&a.properties&&n.push(...this._flattenSchema(a,c)),a.type==="array"&&a.items?.type==="object"&&a.items.properties&&n.push(...this._flattenSchema(a.items,`${c}[]`))}return n}_formatLabel(e){return e.replace(/([A-Z])/g," $1").replace(/^./,o=>o.toUpperCase()).trim()}_summarizeNodeOutput(e,o){if(!o||typeof o!="object")return[];let n=[];o.success!==void 0&&n.push(`Result: ${o.success?"passed":"failed"}`);for(let[s,r]of Object.entries(o))if(!(s==="success"||s==="raw"||s==="nextNode")){if(typeof r=="string"&&r.length<=80)n.push(`${s}: ${r}`);else if(Array.isArray(r)){let i=r.length,a=r.filter(l=>l?.passed===!0).length,c=r.some(l=>l?.passed!==void 0);n.push(c?`${s}: ${a}/${i} passed${i-a?`, ${i-a} failed`:""}`:`${s}: ${i} items`)}if(n.length>=4)break}return n}async run(e,o={},n={}){if(!this.entryPoint)throw new Error("No entry point set for graph");let s=new AbortController;n.signal&&(n.signal.aborted?s.abort():n.signal.addEventListener("abort",()=>s.abort(),{once:!0}));let r=n.strategyAbortTimeoutMs??o.config?.strategyAbortTimeoutMs??5e3,i=o.cwd||process.cwd();Co({path:Y(i,".env")});let a=o.config||{};if(!a||Object.keys(a).length===0)try{let b=Y(i,".zibby.config.js");Fe(b)&&(a=(await import(b)).default||{})}catch{}process.env.EXECUTION_ID&&!a.agent?.strictMode&&(a.agent={...a.agent,strictMode:!0});let c=o.agentType;if(!c){let b=a?.agent;b?.provider?c=b.provider:b?.gemini?c="gemini":b?.claude?c="claude":b?.cursor?c="cursor":b?.codex?c="codex":c=process.env.AGENT_TYPE||"cursor"}let l=o.contextConfig||e?.config?.contextConfig||e?.config?.context||a?.context||{},d=this._runtimeSchema();if(d){let b=d.safeParse(o);if(!b.success){let N=b.error.issues.map(O=>`${O.path.join(".")}: ${O.message}`);throw console.error("\u274C Initial state validation failed:"),N.forEach(O=>console.error(` - ${O}`)),new Error(`State validation failed: ${N.join(", ")}`)}x.step("State validated against schema")}let p=Lt(),u=o.sessionPath||p;u||jt();let{sessionPath:h,sessionTimestamp:m,sessionId:w}=Wt({cwd:i,config:a,traceFrom:"WorkflowGraph.run",initialState:{sessionPath:u,sessionTimestamp:o.sessionTimestamp}});x.step(`Session ${w}`);let I=await ce.loadContext(o.specPath||"",i,l);Object.keys(I).length>0&&x.step(`Context loaded: ${Object.keys(I).join(", ")}`);let E=o.outputPath;!E&&o.specPath&&(e?.calculateOutputPath?E=e.calculateOutputPath(o.specPath):console.warn(`\u26A0\uFE0F outputPath not resolved (specPath=${o.specPath})`));let f=new Q({...o,config:a,agentType:c,outputPath:E,sessionPath:h,sessionTimestamp:m,context:I,resolvedTools:this.resolvedToolsMap||{},_signal:s.signal}),$=new Map;try{await import("@zibby/skills")}catch{}let{getSkill:g}=await Promise.resolve().then(()=>(re(),gt)),S=a.skills&&typeof a.skills=="object"?a.skills:{},R=Object.values(S).filter(b=>b&&typeof b=="object"&&typeof b.id=="string"),v=b=>{for(let N of R)if(N.id===b)return N;return g(b)},D=new Set;for(let[,b]of this.nodes)for(let N of b.config?.skills||[])D.add(N);for(let b of D){let N=v(b);if(typeof N?.middleware=="function")try{let O=await N.middleware();typeof O=="function"&&$.set(b,O)}catch{}}let y=this.entryPoint,de=[],Ye=a?.recursionLimit??100,Yt=0;try{for(;y&&y!=="END";){if(++Yt>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 N=Y(h,ve);if(Fe(N)){try{Oo(N)}catch{}s.abort()}if(s.signal.aborted)return console.warn(`
|
|
37
|
+
\u{1F6D1} External stop requested \u2014 ending workflow.`),x.step("Workflow stopped externally"),{success:!0,state:f.getAll(),executionLog:de,stoppedExternally:!0};let O=this.nodes.get(y);if(!O)throw new Error(`Node '${y}' not found in graph`);let Ze=JSON.stringify({sessionPath:h,sessionTimestamp:m,currentNode:y,createdAt:new Date().toISOString(),config:f.get("config")}),Zt=Y(h,W);Pt(Zt,Ze,"utf-8");let ze=f.get("config")?.paths?.output||ne,zt=Y(i,ze,W);Bt(Y(i,ze),{recursive:!0});try{Pt(zt,Ze,"utf-8")}catch{}let Ke=o.onPipelineProgress;if(typeof Ke=="function")try{Ke({cwd:i,sessionPath:h,sessionId:w,outputBase:f.get("config")?.paths?.output||ne,currentNode:y})}catch{}let Kt=(this.resolvedToolsMap||{})[y]||null;f.set("_currentNodeTools",Kt);let Vt=f.get("nodeConfigs")||{};f.set("_currentNodeConfig",Vt[y]||{}),x.nodeStart(y);let Ve=Date.now(),pe=this.nodePrompts.get(y);if(!this._invokeAgent){let C=await Promise.resolve().then(()=>(q(),se));this._invokeAgent=C.invokeAgent}let qt=this._invokeAgent,_e={},Xt=O.config?.skills||[];for(let C of Xt){let P=v(C);if(typeof P?.invokeAgentOptions=="function")try{let T=P.invokeAgentOptions(f.getAll(),{agentType:f.get("agentType"),nodeName:y});T&&typeof T=="object"&&(_e={..._e,...T})}catch(T){console.warn(`[graph] skill '${C}' invokeAgentOptions threw: ${T.message}`)}}let qe=async(C,P,T={})=>{let B=qt(C,P,{..._e,...T,signal:s.signal});return B.catch(()=>{}),s.signal.aborted?B:Promise.race([B,new Promise((Z,z)=>{let L=()=>{setTimeout(()=>{let X=new Error(`Strategy ignored AbortSignal \u2014 engine deadman fired after ${r}ms`);X.name="AbortError",z(X)},r)};s.signal.addEventListener("abort",L,{once:!0})})])},Xe={state:f,invokeAgent:async(C={},P={})=>{let T=P.prompt||"";if(pe){let B=this._compiledPrompts.get(y);B||(B=Po.compile(pe,{noEscape:!0}),this._compiledPrompts.set(y,B));try{T=B(C)}catch(Z){throw console.error(`\u274C Template rendering failed for node '${y}':`,Z.message),new Error(`Template rendering failed: ${Z.message}`,{cause:Z})}}else if(!T)throw new Error(`No prompt template configured for node '${y}' and no prompt provided in options`);return qe(T,{state:f.getAll(),images:P.images||[]},{model:P.model||f.get("model"),workspace:f.get("workspace"),schema:P.schema,...P,signal:s.signal})},_coreInvokeAgent:qe,agent:e,nodeId:y,promptTemplate:pe,getPromptTemplate:()=>pe,...f.getAll()};try{let C=(O.config?.skills||[]).map(L=>$.get(L)).filter(Boolean),P=[...this.middleware,...C],T;P.length>0?T=await this._composeMiddleware(P,y,async()=>O.execute(Xe,f),f.getAll(),f):T=await O.execute(Xe,f);let B=Date.now()-Ve;if(de.push({node:y,success:T.success,duration:B,timestamp:new Date().toISOString()}),!T.success){if(s.signal.aborted)return x.step("Workflow stopped externally"),{success:!0,state:f.getAll(),executionLog:de,stoppedExternally:!0};f.append("errors",{node:y,error:T.error});let L=O.config?.retries||0,X=`${y}_retries`,fe=f.getAll()[X]||0;if(fe<L){x.stepInfo(`Retrying (attempt ${fe+1}/${L})`),f.update({[X]:fe+1,[`${y}_raw`]:T.raw});continue}throw x.nodeFailed(y,T.error,{duration:B}),new Error(`Node '${y}' failed after ${fe} attempts: ${T.error}`)}f.update({[y]:T.output});let Z=this._summarizeNodeOutput(y,T.output);x.nodeComplete(y,{duration:B,details:Z});let z=this.edges.get(y);if(!z)y="END";else if(z.conditional){let L=z.routes(f.getAll());x.route(y,L),y=L}else y=z}catch(C){throw x.isInsideNode&&x.nodeFailed(y,C.message,{duration:Date.now()-Ve}),f.set("failed",!0),f.set("failedAt",y),C}}x.graphComplete();let b={success:!0,state:f.getAll(),executionLog:de};return e&&typeof e.onComplete=="function"&&await e.onComplete(b),b}finally{if(e&&typeof e.cleanup=="function")try{await e.cleanup()}catch(b){console.warn(`[workflow] agent.cleanup() failed: ${b.message}`)}}}};var Ue=Symbol.for("@zibby/agent-workflow.nodes");globalThis[Ue]||(globalThis[Ue]=new Map);var ue=globalThis[Ue];function Gt(t,e){ue.set(t,e)}function We(t){return ue.get(t)}function Se(t){return ue.has(t)}function Bo(){return Array.from(ue.keys())}function Ge(t){let e=ue.get(t);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}Gt("ai_agent",{name:"ai_agent",factory:!0,create:(t,e={})=>({name:t,_isCustomCode:!0,execute:async o=>{let n=o?._coreInvokeAgent;n||(n=(await Promise.resolve().then(()=>(q(),se))).invokeAgent);let s=e.extraPromptInstructions||"Execute the task based on the current state.",r=Mo(s,o),i=await n(r,{cwd:o.workspace||process.cwd(),model:o.model,tools:e.resolvedTools||null});return{success:!0,output:{raw:i,nodeId:t},raw:typeof i=="string"?i:i.raw}}})});function Mo(t,e){let o=/@([\w.]+)/g,n=new Set,s;for(;(s=o.exec(t))!==null;)n.add(s[1]);if(n.size===0)return t;let r=[],i=new Set;for(let a of n){let c=a.split(".")[0];if(i.has(c))continue;let l=a.split(".").reduce((u,h)=>u?.[h],e);if(l===void 0)continue;let d=typeof l=="string"?l:l?.raw??JSON.stringify(l,null,2),p=a.replace(/_/g," ").replace(/\b\w/g,u=>u.toUpperCase());r.push(`## ${p}
|
|
38
|
+
${d}`),a.includes(".")||i.add(c)}return r.length===0?t:`${t}
|
|
39
39
|
|
|
40
40
|
---
|
|
41
41
|
# Referenced Context
|
|
42
42
|
|
|
43
|
-
${
|
|
44
|
-
|
|
45
|
-
`)}`}
|
|
46
|
-
`)}function
|
|
47
|
-
`)}function
|
|
48
|
-
`)}function
|
|
49
|
-
`)}function
|
|
50
|
-
`)}function
|
|
51
|
-
`)}function
|
|
52
|
-
`).map((
|
|
53
|
-
`);
|
|
54
|
-
`)}function
|
|
43
|
+
${r.join(`
|
|
44
|
+
|
|
45
|
+
`)}`}re();F();var we={};function Je(t,e){if(Array.isArray(e))return He(e);let o=we[t];return!o||o.length===0?null:He(o)}function He(t){if(!Array.isArray(t)||t.length===0)return null;let e=[],o={},n=[];for(let s of t){let r=V(s);if(!r){_.warn(`[workflow] unknown skill "${s}" \u2014 skipping`);continue}n.push(s);for(let i of r.tools||[])e.push({name:i.name,description:i.description,input_schema:i.input_schema||{type:"object",properties:{}}});if(!o[r.serverName])if(typeof r.resolve=="function"){let i=r.resolve();i&&(o[r.serverName]={...i,toolPrefix:s})}else{let i={};for(let a of r.envKeys||[]){let c=process.env[a];c&&(i[a]=c)}o[r.serverName]={command:r.command,args:[...r.args||[]],env:i,toolPrefix:s}}}return n.length===0?null:{toolIds:n,claudeTools:e,mcpServers:o}}F();function Do(t,e={}){let{nodes:o,edges:n,nodeConfigs:s={}}=t;if(!Array.isArray(o)||o.length===0)throw new M("Graph must have at least one node");if(!Array.isArray(n))throw new M("Graph edges must be an array");let r=new le(e);e.stateSchema&&r.setStateSchema(e.stateSchema);let i=new Set,a=new Map,c={};for(let u of o){let h=ye(u);a.set(u.id,{...u,resolvedType:h}),h==="decision"&&i.add(u.id)}for(let[u,h]of a){if(i.has(u))continue;let m=h.resolvedType,w=s[u]||{},I=Je(m,w.tools);I&&(c[u]=I);let E={};w.prompt&&(E.prompt=w.prompt);let f=Se(m);if(_.debug(`[workflow] compiler: node "${u}" type="${m}" registered=${f}`),w.customCode&&!f)r.addNode(u,Ht(u,w.customCode,w),E),r.setNodeType(u,m);else if(f){let $=We(m);$.factory?r.addNode(u,$.create(u,{...w,resolvedTools:I}),E):r.addNode(u,$,E),r.setNodeType(u,m)}else if(w.executeCode)r.addNode(u,Ht(u,w.executeCode,w),E),r.setNodeType(u,m);else throw new M(`Unknown node type "${m}" for node "${u}". Did you forget to register it?`)}r.resolvedToolsMap=c;let l=new Set;for(let u of n)i.has(u.target)||l.add(u.target);let d=o.find(u=>!i.has(u.id)&&!l.has(u.id));if(!d)throw new M("Could not determine entry point: no node without incoming edges found");r.setEntryPoint(d.id);let p=Fo(n,"source");for(let u of n)if(!i.has(u.source))if(i.has(u.target)){let h=u.target,m=p.get(h)||[];if(m.length===0)throw new M(`Decision node "${h}" has no outgoing edges`);let w=Uo(h,m,i);r.addConditionalEdges(u.source,w)}else r.addEdge(u.source,u.target);return r}function Lo(t){let e=[];if(!t||typeof t!="object")return{valid:!1,errors:["Config must be a non-null object"]};if((!Array.isArray(t.nodes)||t.nodes.length===0)&&e.push("Graph must have at least one node"),Array.isArray(t.edges)||e.push("Graph edges must be an array"),e.length>0)return{valid:!1,errors:e};let o=t.nodeConfigs||{};for(let a of t.nodes){let c=ye(a);if(c==="decision"||Se(c))continue;let l=o[a.id]||{};l.customCode||l.executeCode||e.push(`Unknown node type "${c}" for node "${a.id}". Register it or provide customCode/executeCode.`)}let n=new Set(t.nodes.map(a=>a.id));for(let a of t.edges)n.has(a.source)||e.push(`Edge references unknown source node "${a.source}"`),n.has(a.target)||e.push(`Edge references unknown target node "${a.target}"`);let s=new Set(t.nodes.filter(a=>ye(a)==="decision").map(a=>a.id)),r=new Set;for(let a of t.edges)s.has(a.target)||r.add(a.target);let i=t.nodes.filter(a=>!s.has(a.id)&&!r.has(a.id));i.length===0?e.push("No entry point found (every node has incoming edges)"):i.length>1&&e.push(`Multiple entry points found: ${i.map(a=>a.id).join(", ")}`);for(let a of s){let c=t.edges.filter(d=>d.source===a);c.length===0&&e.push(`Decision node "${a}" has no outgoing edges`),c.some(d=>d.data?.conditionalCode||d.conditionalCode)||e.push(`Decision node "${a}" outgoing edges have no conditionalCode`)}return{valid:e.length===0,errors:e}}function jo(t){return!t||!Array.isArray(t.nodes)?[]:t.nodes.filter(e=>ye(e)!=="decision").map(e=>e.id)}function ye(t){let e=t.data?.nodeType||t.data?.type||t.type;return e==="workflowNode"||e==="custom"||e==="default"?t.id:e}function Fo(t,e){let o=new Map;for(let n of t){let s=n[e];o.has(s)||o.set(s,[]),o.get(s).push(n)}return o}function Uo(t,e,o){let n=e.find(a=>a.data?.conditionalCode||a.conditionalCode);if(!n)throw new M(`Decision node "${t}" has no conditionalCode on its outgoing edges`);let s=n.data?.conditionalCode||n.conditionalCode,r=new Set(e.map(a=>a.target).filter(a=>!o.has(a))),i;try{let c=new Function(`return (${s})`)();i=l=>{let d=c(l);return r.has(d)||_.warn(`[workflow] conditional route from "${t}" returned "${d}" which is not in valid targets: ${[...r].join(", ")}`),d}}catch(a){throw new M(`Failed to compile conditionalCode for "${t}": ${a.message}`)}return i}function Ht(t,e,o={}){let n;try{n=new Function("invokeAgent","require","console",`return (${e})`)}catch(i){throw new M(`Failed to compile customCode for node "${t}": ${i.message}`)}let s=n(async(...i)=>{let{invokeAgent:a}=await Promise.resolve().then(()=>(q(),se));return a(...i)},typeof Ie<"u"?Ie:void 0,console),r=null;return o.outputSchema&&(r=o.outputSchema.jsonSchema||o.outputSchema),{name:t,_isCustomCode:!0,outputSchema:r,execute:async i=>{try{let a=await s(i);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 M=class extends Error{constructor(e){super(e),this.name="CompilationError"}};re();Ae();q();function Wo(t,e={}){let{nodes:o,edges:n,nodeConfigs:s={}}=t,r=new Set,i=[],a=new Map;for(let I of o){let E=I.data?.nodeType||I.type;a.set(I.id,E),E==="decision"?r.add(I.id):i.push({id:I.id,nodeType:E,label:I.data?.label||I.id})}let c=i.some(I=>{let E=s[I.id]||{};return!E.customCode&&!E.executeCode}),{toolsPerNode:l,toolIdsByVar:d}=Vo(i,s),{simpleEdges:p,conditionalEdges:u}=qo(n,r),h=Xo(i,n,r),m=[],w=e.workflowType||"workflow";return m.push(Ho(e)),m.push(Jo(w,{usesRegisteredNodes:c})),m.push(Yo(d)),m.push(Zo(w)),m.push(zo(i,s)),m.push(Ko(i,h,p,u,l,w)),m.filter(Boolean).join(`
|
|
46
|
+
`)}function Go(t){let e={};for(let[o,n]of Object.entries(t)){let{tools:s,...r}=n;Object.keys(r).length>0&&(e[o]=r)}return e}function Ho(t){let e=t.workflowType||"workflow";return["// Generated workflow",`// ${t.projectId?`Project: ${t.projectId} | `:""}Type: ${e} | Version: ${t.version??0}`,`// Downloaded: ${new Date().toISOString()}`,""].join(`
|
|
47
|
+
`)}function Jo(t,{usesRegisteredNodes:e=!0}={}){let o=["import { WorkflowGraph, invokeAgent, getResolvedToolDefinitions } from '@zibby/agent-workflow';"];return e&&o.push("// import './register-nodes.js'; // register custom node types here"),o.push("import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';","import { join, dirname } from 'node:path';","import { fileURLToPath } from 'node:url';",""),o.join(`
|
|
48
|
+
`)}function Yo(t){if(t.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[o,n]of t)e.push(`const ${o} = getResolvedToolDefinitions(${JSON.stringify(n)}); // ${n.join(", ")}`);return e.push(""),e.join(`
|
|
49
|
+
`)}function Zo(t){return["// \u2500\u2500 Node Configs \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\u2500","const __filename = fileURLToPath(import.meta.url);","const __dirname = dirname(__filename);",`const configPath = join(__dirname, 'workflow-${t}.config.json');`,"const nodeConfigs = existsSync(configPath) ? JSON.parse(readFileSync(configPath, 'utf-8')) : {};",""].join(`
|
|
50
|
+
`)}function zo(t,e){let o=["// \u2500\u2500 Node Implementations \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 n of t){let s=Jt(n.id),r=e[n.id]?.customCode;if(r)o.push(`// @custom \u2014 modified from default "${n.nodeType}" template`),o.push(`const ${s}_execute = ${r};`);else{let i=Ge(n.nodeType);i?(o.push(`// Default "${n.nodeType}" implementation`),o.push(`const ${s}_execute = ${i};`)):(o.push(`// No template for "${n.nodeType}" \u2014 passthrough`),o.push(`const ${s}_execute = async (state) => ({ success: true, output: {}, raw: null });`))}o.push("")}return o.join(`
|
|
51
|
+
`)}function Ko(t,e,o,n,s,r){let i=["// \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"];i.push("export function buildGraph(options = {}) {"),i.push(" const graph = new WorkflowGraph(options);",""),i.push(" // Nodes");for(let c of t){let l=Jt(c.id);i.push(` graph.addNode('${c.id}', { name: '${c.id}', execute: ${l}_execute });`),i.push(` graph.setNodeType('${c.id}', '${c.nodeType}');`)}i.push("",` graph.setEntryPoint('${e}');`,""),(o.length>0||n.length>0)&&i.push(" // Edges");for(let c of o)i.push(` graph.addEdge('${c.source}', '${c.target}');`);for(let c of n){let l=c.code.split(`
|
|
52
|
+
`).map((d,p)=>p===0?d:` ${d}`).join(`
|
|
53
|
+
`);i.push(` graph.addConditionalEdges('${c.source}', ${l});`)}let a=[];for(let c of t){let l=s.get(c.id);l&&a.push(` '${c.id}': ${l},`)}return a.length>0&&i.push(""," graph.resolvedToolsMap = {",...a," };"),i.push(""," return graph;","}",""),i.push("export { nodeConfigs };",""),i.join(`
|
|
54
|
+
`)}function Vo(t,e){let o=new Map,n=new Map;for(let s of t){let r=e[s.id]?.tools,i;if(Array.isArray(r)&&r.length>0)i=[...r].sort();else{let a=we[s.nodeType];a?.length>0&&(i=[...a].sort())}if(i){let a=`${i.map(c=>c.replace(/[^a-zA-Z0-9]/g,"")).join("And")}Tools`;o.set(s.id,a),n.has(a)||n.set(a,i)}}return{toolsPerNode:o,toolIdsByVar:n}}function qo(t,e){let o=[],n=[],s=new Map,r=new Set;for(let i of t)s.has(i.source)||s.set(i.source,[]),s.get(i.source).push(i);for(let i of t)if(!e.has(i.source))if(e.has(i.target)){if(r.has(i.target))continue;r.add(i.target);let c=(s.get(i.target)||[]).find(l=>l.data?.conditionalCode||l.conditionalCode);c&&n.push({source:i.source,code:c.data?.conditionalCode||c.conditionalCode})}else o.push({source:i.source,target:i.target});return{simpleEdges:o,conditionalEdges:n}}function Xo(t,e,o){let n=new Set;for(let r of e)o.has(r.target)||n.add(r.target);let s=t.find(r=>!n.has(r.id));return s?s.id:t[0]?.id}function Jt(t){return t.replace(/[^a-zA-Z0-9]/g,"_")}F();export{ke as AgentStrategy,xe as CI_ENV_VARS,M as CompilationError,ie as ConditionalNode,ce as ContextLoader,ne as DEFAULT_OUTPUT_BASE,io as EVENTS_FILE,we as NODE_DEFAULT_TOOLS,j as Node,ee as OutputParser,so as RAW_OUTPUT_FILE,ro as RESULT_FILE,Te as SESSIONS_DIR,W as SESSION_INFO_FILE,ao as SKILLS,ve as STOP_REQUEST_FILE,to as SchemaTypes,ge as Timeline,lt as WORKFLOW_GRAPH_LOG_MARKER_PREFIX,le as WorkflowGraph,Q as WorkflowState,jt as clearInheritedSessionEnvForFreshRun,ht as clearSkills,Do as compileGraph,Le as dispatchSubgraph,jo as extractSteps,Go as generateNodeConfigsJson,Wo as generateWorkflowCode,Ut as generateWorkflowSessionId,Ce as getAgentStrategy,pt as getAllSkills,We as getNodeImpl,Ge as getNodeTemplate,He as getResolvedToolDefinitions,V as getSkill,Se as hasNode,dt as hasSkill,wt as invokeAgent,Bo as listNodeTypes,ft as listSkillIds,St as listStrategies,Lt as readPinnedSessionPathFromEnv,Gt as registerNode,ut as registerSkill,mt as registerStrategy,Je as resolveNodeTools,Wt as resolveWorkflowSession,oo as setLogger,Dt as shouldTrustInheritedSessionEnv,Ft as syncProcessEnvToSession,x as timeline,Lo as validateGraphConfig};
|
package/dist/node.js
CHANGED
|
@@ -1,29 +1,29 @@
|
|
|
1
|
-
var
|
|
1
|
+
var it=Object.defineProperty;var N=(r,t)=>()=>(r&&(t=r(r=0)),t);var st=(r,t)=>{for(var e in t)it(r,e,{get:t[e],enumerable:!0})};var D,nt,k,l,T=N(()=>{D=()=>{},nt={debug:D,info:D,warn:(...r)=>console.warn("[workflow]",...r),error:(...r)=>console.error("[workflow]",...r)},k={impl:nt},l={debug:(...r)=>k.impl.debug?.(...r),info:(...r)=>k.impl.info?.(...r),warn:(...r)=>k.impl.warn?.(...r),error:(...r)=>k.impl.error?.(...r)}});var Z=N(()=>{});function V(r){return ct.get(r)||null}var v,ct,z=N(()=>{v=Symbol.for("@zibby/agent-workflow.skills");globalThis[v]||(globalThis[v]=new Map);ct=globalThis[v]});var X={};st(X,{getAgentStrategy:()=>q,invokeAgent:()=>dt,listStrategies:()=>pt,registerStrategy:()=>ut});function ut(r){if(!r||typeof r.getName!="function"||typeof r.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let t=w.findIndex(e=>e.getName()===r.getName());t>=0?w[t]=r:w.push(r)}function pt(){return w.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=w.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 s=w.find(n=>n.getName()===o);if(!s){let n=w.map(a=>a.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${o}'. Available: ${n}`)}if(!s.canHandle(r))throw new Error(`Agent '${o}' is not available in this environment. Check credentials/environment.`);return l.debug(`[workflow] using agent: ${s.getName()}`),s}async function dt(r,t={},e={}){let o=q(t),s=t.state?.config||e.config||{},n=s.models||{},a=e.nodeName&&n[e.nodeName]||null,g=n.default||null,p=s.agent?.[o.name]?.model||null,_=a||g||p||e.model||null,E={...e,model:_,workspace:t.state?.workspace||e.workspace,schema:e.schema||t.schema,images:e.images||t.images||[],skills:e.skills||t.skills||[],config:s},i=r,h=E.skills||[];if(h.length>0&&!e.skipPromptFragments){let y=h.map(O=>{let $=V(O)?.promptFragment;return typeof $=="function"?$():$}).filter(Boolean);y.length>0&&(i+=`
|
|
2
2
|
|
|
3
3
|
${y.join(`
|
|
4
4
|
|
|
5
|
-
`)}`)}let f=t.state?._currentNodeConfig?.extraPromptInstructions?.trim();return f&&(
|
|
5
|
+
`)}`)}let f=t.state?._currentNodeConfig?.extraPromptInstructions?.trim();return f&&(i+=`
|
|
6
6
|
|
|
7
7
|
\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
|
|
8
8
|
PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PREVIOUS CONTENT
|
|
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
|
-
`),l.debug(`[workflow] prompt length: ${
|
|
12
|
+
`),l.debug(`[workflow] prompt length: ${i.length} chars`),o.invoke(i,E)}var P,w,Q=N(()=>{Z();T();z();P=Symbol.for("@zibby/agent-workflow.strategies");globalThis[P]||(globalThis[P]=[]);w=globalThis[P]});var b=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(s=>s[0]);for(let s of o)try{return this.validate(JSON.parse(s))}catch(n){if(!(n instanceof SyntaxError))throw n}return this.validate({result:t.trim()})}validate(t){let e=[];for(let[o,s]of Object.entries(this.schema)){if(s.required&&!(o in t)&&e.push(`Missing required field: ${o}`),o in t&&s.type){let n=typeof t[o];n!==s.type&&e.push(`Field '${o}' expected ${s.type}, got ${n}`)}if(s.validate&&o in t){let n=s.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}};
|
|
14
|
+
`)}`);return t}};T();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 c from"chalk";var at="__WORKFLOW_GRAPH_LOG__",I=c.gray("\u2502"),lt=c.gray("\u250C"),G=c.gray("\u2514"),W=c.green("\u25C6"),U=c.hex("#c084fc")("\u25C6"),j=c.hex("#2dd4bf")("\u25C6"),A=c.red("\u25C6"),B=`${I} `,H=2;function J(r){return r<1e3?`${r}ms`:`${(r/1e3).toFixed(1)}s`}function K(r,t){return(e,o,s)=>{if(typeof e!="string")return r(e,o,s);let n=process.stdout.columns||120,a="";for(let g=0;g<e.length;g++){let p=e[g];t.lineStart&&(a+=B,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
|
+
${B}`,t.col=H))}return r(a,o,s)}}var x=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=K(this._origStdoutWrite,t),process.stderr.write=K(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(`
|
|
18
18
|
`),process.stderr.write=this._origStderrWrite),this._origStdoutWrite=null,this._origStderrWrite=null}_rawWrite(t){(this._origStdoutWrite||process.stdout.write.bind(process.stdout))(`${t}
|
|
19
19
|
`)}_emitGraphLogMarker(t){if(!this._emitWorkflowGraphMarkers)return;let e=`${at}${JSON.stringify(t)}
|
|
20
20
|
`;this._origStdoutWrite?this._origStdoutWrite(e):process.stdout.write(e)}_writeDot(t,e){this._origStdoutWrite?(this._outState&&!this._outState.lineStart&&(this._origStdoutWrite(`
|
|
21
21
|
`),this._outState.lineStart=!0,this._outState.col=0),this._origStdoutWrite(`${t} ${e}
|
|
22
22
|
`)):process.stdout.write.bind(process.stdout)(`${t} ${e}
|
|
23
|
-
`)}step(t){this._origStdoutWrite?this._writeDot(
|
|
23
|
+
`)}step(t){this._origStdoutWrite?this._writeDot(W,t):process.stdout.write.bind(process.stdout)(`${I} ${W} ${t}
|
|
24
24
|
`)}stepInfo(t){this.step(t)}stepTool(t){this._origStdoutWrite?this._writeDot(U,t):process.stdout.write.bind(process.stdout)(`${I} ${U} ${t}
|
|
25
|
-
`)}stepMemory(t){let e=
|
|
26
|
-
`)}stepFail(t){this._origStdoutWrite?this._writeDot(
|
|
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:
|
|
25
|
+
`)}stepMemory(t){let e=c.hex("#2dd4bf")(t);this._origStdoutWrite?this._writeDot(j,e):process.stdout.write.bind(process.stdout)(`${I} ${j} ${e}
|
|
26
|
+
`)}stepFail(t){this._origStdoutWrite?this._writeDot(A,c.red(t)):process.stdout.write.bind(process.stdout)(`${I} ${A} ${c.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:s}=e;if(s)for(let a of s)this._rawWrite(`${W} ${a}`);let n=o?c.dim(` ${J(o)}`):"";this._rawWrite(`${G} ${c.green("done")}${n}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}nodeFailed(t,e,o={}){this._stopIntercepting();let{duration:s}=o,n=s?c.dim(` ${J(s)}`):"";this._rawWrite(`${A} ${c.red(e)}`),this._rawWrite(`${G} ${c.red("failed")}${n}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}route(t,e){this._rawWrite(c.dim(` ${t} \u2192 ${e}`)),this._rawWrite("")}graphComplete(){}},Y=new x;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 b(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,s=i=>e&&typeof e.get=="function"?e.get(i):t?.[i];if(typeof this.customExecute=="function"){l.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let i=await this.customExecute(t);return typeof i=="object"&&i!==null&&i.success===!1?{success:!1,error:i.error||"Node execution failed",raw:i.raw||null}:this.isZodSchema?(l.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(i),raw:null}):{success:!0,output:i,raw:null}}catch(i){return l.error(`[workflow] node '${this.name}' failed: ${i.message}`),i.name==="ZodError"&&l.error(`Schema errors: ${JSON.stringify(i.issues||i.errors,null,2)}`),{success:!1,error:i.message,raw:null}}}let n=typeof this.prompt=="function"?this.prompt(o()):this.prompt,a=s("_skillHints");a&&(n=`${a}
|
|
28
28
|
|
|
29
|
-
${n}`);let g=o(),p=g.cwd||process.cwd(),
|
|
29
|
+
${n}`);let g=o(),p=g.cwd||process.cwd(),_=g.sessionPath;try{if(_){let i=F(_,L);if(et(i)){let f=JSON.parse(tt(i,"utf-8"));f.currentNode=this.name,C(i,JSON.stringify(f,null,2),"utf-8")}let h=F(_,"..",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(i){l.debug(`[workflow] could not update session info: ${i.message}`)}let E=null;for(let i=0;i<=this.retries;i++)try{l.debug(`[workflow] node '${this.name}' attempt ${i}`);let h=o().config||{},f=h.agents||{},y=this.config.agent??f[this.name]??null,O={state:o()};y&&(O.preferredAgent=y);let $={workspace:p,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],sessionPath:_,config:h,nodeName:this.name,timeout:this.config?.timeout||3e5},R=t?._coreInvokeAgent;R||(R=(await Promise.resolve().then(()=>(Q(),X))).invokeAgent);let m=await R(n,O,$),d,S;if(typeof m=="string"?(d=m,S=null):m.structured?(d=m.raw||JSON.stringify(m.structured,null,2),S=m.structured):(d=m.raw||JSON.stringify(m,null,2),S=m.extracted||null),_)try{let u=F(_,this.name,"raw_stream_output.txt");ht(ft(u),{recursive:!0}),C(u,typeof d=="string"?d:JSON.stringify(d),"utf-8")}catch(u){l.debug(`[workflow] could not save raw output: ${u.message}`)}if(this.isZodSchema&&S){l.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(S,null,2)}`);let u=S;if(typeof this.onComplete=="function")try{u=await this.onComplete(o(),S)}catch(ot){l.warn(`[workflow] onComplete hook failed: ${ot.message}`)}return{success:!0,output:u,raw:d}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(o(),{raw:d}),raw:d}}catch(u){throw new Error(`onComplete failed: ${u.message}`,{cause:u})}if(this.parser){let u=this.parser.parse(d);return l.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(u,null,2)}`),Y.step("Output parsed"),{success:!0,output:u,raw:d}}return{success:!0,output:d,raw:d}}catch(h){E=h,i<this.retries&&l.info(`[workflow] node '${this.name}' failed, retrying (${i+1}/${this.retries})\u2026`)}return{success:!1,error:E.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};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
var U=()=>{},q={debug:U,info:U,warn:(...t)=>console.warn("[workflow]",...t),error:(...t)=>console.error("[workflow]",...t)},A={impl:q};var w={debug:(...t)=>A.impl.debug?.(...t),info:(...t)=>A.impl.info?.(...t),warn:(...t)=>A.impl.warn?.(...t),error:(...t)=>A.impl.error?.(...t)};import{mkdirSync as N,existsSync as _,statSync as lt}from"node:fs";import{spawn as G}from"node:child_process";import{join as x}from"node:path";import{pathToFileURL as X}from"node:url";import{AsyncLocalStorage as F}from"node:async_hooks";var P=new F;function B(){let t=P.getStore();return t||Object.freeze({executionId:process.env.EXECUTION_ID||null,parentExecutionId:process.env.PARENT_EXECUTION_ID||null,depth:0,conversationId:process.env.ZIBBY_CONVERSATION_ID||null,dispatchMode:process.env.DISPATCH_MODE||null})}function k(t,e){let r=P.getStore()||B(),o=Object.freeze({executionId:t.executionId,parentExecutionId:t.parentExecutionId??r.executionId??null,depth:(r.depth||0)+(t.executionId!==r.executionId?1:0),conversationId:t.conversationId!==void 0?t.conversationId:r.conversationId??null,dispatchMode:t.dispatchMode??null});return P.run(o,e)}var O=new Map,C=new Map,D=new Map;function M(t,e,r={}){if(!t||typeof t!="string")throw new Error("subgraph-registry.register: name required");if(typeof e!="function")throw new Error("subgraph-registry.register: factory must be a function");O.set(t,e),C.set(t,"ready"),D.set(t,{...r,cachedAt:Date.now()})}function j(t,e){C.set(t,"failed"),D.set(t,{error:e?.message||String(e),failedAt:Date.now()}),O.delete(t)}function L(t){return C.get(t)==="ready"?O.get(t):null}var V=process.env.ZIBBY_SUBGRAPH_CACHE_DIR||"/tmp/zibby/subgraphs";function K(){return Number(process.env.ZIBBY_SUBGRAPH_MAX_DEPTH||10)}function Y(){return`node${(process.versions?.node||"").split(".")[0]||"unknown"}-${process.platform}-${process.arch}`}var h=class extends Error{constructor(e,r){super(`in-process sub-graph fallback: ${e}${r?` (${r})`:""}`),this.fallback=!0,this.reason=e,this.detail=r||null,this.name="SubgraphFallback"}};function Z(){let t=(process.env.SUBGRAPH_INTERNAL_URL||"").replace(/\/$/,""),e=(process.env.PROGRESS_API_URL||"").replace(/\/executions\/?$/,""),r=t||e,o=process.env.PROJECT_ID,n=process.env.PROJECT_API_TOKEN;if(!r||!o||!n)throw new h("env","SUBGRAPH_INTERNAL_URL/PROGRESS_API_URL/PROJECT_ID/PROJECT_API_TOKEN missing");return{apiBase:r,projectId:o,authToken:n}}async function W({apiBase:t,authToken:e,body:r}){let o;try{o=await fetch(`${t}/internal/subgraph/begin`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify(r)})}catch(i){throw new h("network",`begin fetch failed: ${i.message}`)}let n=null;try{n=await o.json()}catch{}if(!o.ok){if(o.status===404){let i=new Error(`Sub-graph child '${r.childWorkflowType}' not found in project`);throw i.code="SUBGRAPH_NOT_FOUND",i.status=404,i}if(o.status===429){let i=n?.quotaInfo||{},a=new Error(`Sub-graph blocked by quota (${i.used??"?"}/${i.limit??"?"} on ${i.planId||"plan"})`);throw a.code="SUBGRAPH_QUOTA_EXCEEDED",a.status=429,a.quotaInfo=i,a}if(o.status===400&&n?.validationErrors){let i=new Error(`Sub-graph rejected input: ${n?.error||n?.message||"validation failed"}`);throw i.code="SUBGRAPH_INVALID_INPUT",i.status=400,i.validationErrors=n.validationErrors,i.missing=n.missing,i}throw new h("begin-status",`begin returned ${o.status}`)}return n?.data||n}async function $({apiBase:t,authToken:e,payload:r}){try{let o=await fetch(`${t}/internal/subgraph/finalize`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${e}`},body:JSON.stringify(r)});o.ok||w.warn(`[in-process subgraph] finalize returned ${o.status} for ${r.childExecutionId}`)}catch(o){w.warn(`[in-process subgraph] finalize failed: ${o.message}`)}}async function Q(t,e){let r=x(e,".ready"),o=x(e,"graph.mjs");if(_(r)&&_(o))return;N(e,{recursive:!0});let n=x(e,".lock"),i=!1;try{let{openSync:a,closeSync:g}=await import("node:fs"),f=a(n,"wx");g(f),i=!0}catch(a){if(a.code!=="EEXIST")throw a}if(!i){let a=Date.now()+3e4;for(;Date.now()<a;){if(_(r)&&_(o))return;await new Promise(g=>setTimeout(g,100))}throw new h("bundle-extract-timeout","sibling extract did not complete within 30s")}try{await new Promise((f,y)=>{let p=G("curl",["-fsSL",t],{stdio:["ignore","pipe","inherit"]}),m=G("tar",["-xzf","-","-C",e],{stdio:["pipe","inherit","inherit"]});p.stdout.pipe(m.stdin);let S,E,b=()=>{if(S!==void 0&&E!==void 0){if(S!==0)return y(new Error(`curl exited ${S}`));if(E!==0)return y(new Error(`tar exited ${E}`));f()}};p.on("close",l=>{S=l,b()}),m.on("close",l=>{E=l,b()}),p.on("error",y),m.on("error",y)});let{writeFileSync:a,unlinkSync:g}=await import("node:fs");a(r,"");try{g(n)}catch{}}catch(a){try{let{unlinkSync:g}=await import("node:fs");g(n)}catch{}throw new h("bundle-extract-failed",a.message)}}async function tt(t){let e=x(t,"graph.mjs");if(!_(e))throw new h("entry-missing",`graph.mjs missing under ${t}`);let r;try{r=await import(X(e).href)}catch(n){throw new h("import-failed",`${n?.code||n?.name||"unknown"}: ${n.message}`)}let o=r.default||Object.values(r).find(n=>typeof n=="function"&&n.prototype?.buildGraph);if(!o)throw new h("entry-class-missing","no buildGraph() class export found");return o}async function H(t,e={}){if(!t||typeof t!="string")throw new Error("runInProcessSubgraph: workflowName (string) is required");let r=B(),o=K();if((r.depth||0)>=o)throw new h("depth-exceeded",`depth ${r.depth} \u2265 MAX_DEPTH ${o}`);let n;try{n=Z()}catch(s){throw s}w.debug(`[in-process subgraph] begin '${t}' parent=${r.executionId||"<root>"}`);let i=await W({apiBase:n.apiBase,authToken:n.authToken,body:{parentExecutionId:r.executionId,childWorkflowType:t,input:e.input||{},...e.conversationId?{conversationId:e.conversationId}:{}}}),{childExecutionId:a,runtimeTag:g,bundlePresignedUrl:f,sourcesPresignedUrl:y,workflowVersion:p,workflowUuid:m,bundleReady:S}=i,E=Y();if(g&&g!==E)throw await $({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:a,status:"canceled",error:{message:`runtimeTag mismatch: parent=${E} child=${g}`,code:"RUNTIME_MISMATCH"}}}),new h("runtime-mismatch",`${E} vs ${g}`);if(!S||!f)throw await $({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:a,status:"canceled",error:{message:"bundle not ready for in-process; falling back to HTTP",code:"NO_BUNDLE"}}}),new h("no-bundle","workflow bundle not built yet");let b=L(t);if(!b){let s=x(V,`${m}@${p||"0"}`);try{await Q(f,s)}catch(d){throw d.fallback&&await $({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:a,status:"failed",error:{message:d.message,code:d.reason}}}),d}try{b=await tt(s),M(t,b,{workflowUuid:m,version:p,runtimeTag:g,cacheDir:s})}catch(d){throw j(t,d),await $({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:a,status:"failed",error:{message:d.message,code:d.reason||"IMPORT_FAILED"}}}),d.fallback?d:new h("import-failed",d.message)}}let l=Date.now(),T=await(typeof b=="function"&&b.prototype?.buildGraph?new b:b).buildGraph(),c={...e.input||{}},u,I;try{u=await k({executionId:a,parentExecutionId:r.executionId,conversationId:e.conversationId!==void 0?e.conversationId:r.conversationId,dispatchMode:"inprocess"},()=>T.run(e.parentAgent,c,{signal:e.signal})),I=u&&typeof u=="object"&&"state"in u?u.state:u}catch(s){throw await $({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:a,status:"failed",error:{message:s.message,code:s.code||"CHILD_THREW",stack:s.stack},durationMs:Date.now()-l}}),s}if(u&&typeof u=="object"&&u.stoppedExternally){await $({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:a,status:"canceled",finalState:I,durationMs:Date.now()-l}});let s=new Error(`Sub-graph '${t}' canceled by parent abort`);throw s.code="SUBGRAPH_CANCELED",s.subgraphJobId=a,s}return await $({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:a,status:"completed",finalState:I,durationMs:Date.now()-l}}),{finalState:I,executionId:a}}var et=2e3,nt=600*1e3,rt=new Set(["completed","failed","canceled","timeout"]);function ot(){let t=process.env.PROGRESS_API_URL;if(!t)throw new Error("Sub-graph dispatch requires PROGRESS_API_URL env var (set automatically on cloud runs). Sub-graphs are not supported in local in-process runs yet \u2014 deploy the parent and child to cloud.");return t.replace(/\/executions\/?$/,"")}function st(){let t=process.env.PROJECT_ID;if(!t)throw new Error("Sub-graph dispatch requires PROJECT_ID env var.");return t}function at(){let t=process.env.PROJECT_API_TOKEN;if(!t)throw new Error("Sub-graph dispatch requires PROJECT_API_TOKEN env var.");return t}function it(){return process.env.EXECUTION_ID||null}function J(t,e){return e==null?t:typeof e=="function"?e(t):typeof e=="string"?e.split(".").reduce((r,o)=>r==null?r:r[o],t):t}async function mt(t,e={}){if(!t||typeof t!="string")throw new Error("dispatchSubgraph: workflowName (string) is required");if(process.env.ZIBBY_INPROCESS_SUBGRAPH!=="0"&&!e.async)try{w.debug(`[sub-graph] trying in-process for '${t}'`);let{finalState:c}=await H(t,{input:e.input,conversationId:e.conversationId,signal:e.signal,parentAgent:e.parentAgent}),u=J(c,e.output);return w.info(`[sub-graph] '${t}' completed in-process`),u}catch(c){if(c instanceof h||c?.fallback)w.info(`[sub-graph] in-process fallback for '${t}': ${c.reason||"unknown"} \u2014 using HTTP`);else throw c}let r=ot(),o=st(),n=at(),i=it(),a=`${r}/projects/${encodeURIComponent(o)}/workflows/${encodeURIComponent(t)}/trigger`,g={input:e.input||{},...i?{parentExecutionId:i}:{},...e.conversationId?{conversationId:e.conversationId}:{}};w.info(`[sub-graph] dispatching '${t}' (${e.async?"async":"sync"}) from parent ${i||"<none>"}`);let f=await fetch(a,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${n}`},body:JSON.stringify(g)});if(!f.ok){let c=null,u="";try{c=await f.json(),u=c?.error||c?.message||JSON.stringify(c)}catch{u=await f.text().catch(()=>"")}if(f.status===429){let s=c?.quotaInfo||{},d=new Error(`Sub-graph '${t}' blocked by execution quota (${s.used??"?"}/${s.limit??"?"} on plan ${s.planId||"unknown"}). Sub-workflow runs count toward the same monthly cap as user-triggered runs.`);throw d.code="SUBGRAPH_QUOTA_EXCEEDED",d.status=429,d.subgraph=t,d.quotaInfo=s,d}if(f.status===400){let s=new Error(`Sub-graph '${t}' rejected input: ${u}`);throw s.code="SUBGRAPH_INVALID_INPUT",s.status=400,s.subgraph=t,s.validationErrors=c?.validationErrors||null,s.missing=c?.missing||null,s}let I=new Error(`Sub-graph '${t}' trigger rejected (${f.status}): ${u}`);throw I.code="SUBGRAPH_TRIGGER_FAILED",I.status=f.status,I.subgraph=t,I}let y=await f.json(),p=y?.data?.jobId||y?.jobId;if(!p)throw new Error(`Sub-graph '${t}' trigger returned no jobId: ${JSON.stringify(y).slice(0,200)}`);if(e.async)return w.info(`[sub-graph] async dispatch of '${t}' \u2192 jobId=${p} (not waiting)`),{jobId:p,status:"accepted",workflow:t};let m=Number.isFinite(e.timeoutMs)?e.timeoutMs:nt,S=Number.isFinite(e.pollIntervalMs)?e.pollIntervalMs:et,E=`${r}/executions/${encodeURIComponent(p)}`,b=Date.now()+m,l="accepted",R=0;for(;Date.now()<b;){await new Promise(s=>setTimeout(s,S)),R+=1;let c=await fetch(E,{headers:{Authorization:`Bearer ${n}`}});if(!c.ok){if(c.status>=500){w.warn(`[sub-graph] status poll for ${p} returned ${c.status}, will retry`);continue}throw new Error(`Sub-graph status poll failed for ${p}: ${c.status}`)}let u=await c.json(),I=u?.data||u?.execution||u;if(l=I?.status||l,rt.has(l)){if(l!=="completed"){let v=new Error(`Sub-graph '${t}' (${p}) ended in status '${l}'`);throw v.subgraphJobId=p,v.subgraphStatus=l,v}let s=I?.finalState||I?.state||{},d=J(s,e.output);return w.info(`[sub-graph] '${t}' (${p}) completed after ${R} polls`),d}}let T=new Error(`Sub-graph '${t}' (${p}) timed out after ${Math.round(m/1e3)}s (last status: ${l})`);throw T.subgraphJobId=p,T.subgraphStatus=l,T}export{mt as dispatchSubgraph};
|
|
1
|
+
var M=()=>{},Y={debug:M,info:M,warn:(...e)=>console.warn("[workflow]",...e),error:(...e)=>console.error("[workflow]",...e)},R={impl:Y};var f={debug:(...e)=>R.impl.debug?.(...e),info:(...e)=>R.impl.info?.(...e),warn:(...e)=>R.impl.warn?.(...e),error:(...e)=>R.impl.error?.(...e)};import{mkdirSync as K,existsSync as v,statSync as F,readdirSync as q,rmSync as W}from"node:fs";import{spawn as z}from"node:child_process";import{join as T}from"node:path";import{pathToFileURL as Q}from"node:url";import{AsyncLocalStorage as Z}from"node:async_hooks";var U=new Z;function A(){let e=U.getStore();return e||Object.freeze({executionId:process.env.EXECUTION_ID||null,parentExecutionId:process.env.PARENT_EXECUTION_ID||null,depth:0,conversationId:process.env.ZIBBY_CONVERSATION_ID||null,dispatchMode:process.env.DISPATCH_MODE||null})}function j(e,t){let n=U.getStore()||A(),r=Object.freeze({executionId:e.executionId,parentExecutionId:e.parentExecutionId??n.executionId??null,depth:(n.depth||0)+(e.executionId!==n.executionId?1:0),conversationId:e.conversationId!==void 0?e.conversationId:n.conversationId??null,dispatchMode:e.dispatchMode??null});return U.run(r,t)}var k=new Map,D=new Map,L=new Map;function G(e,t,n={}){if(!e||typeof e!="string")throw new Error("subgraph-registry.register: name required");if(typeof t!="function")throw new Error("subgraph-registry.register: factory must be a function");k.set(e,t),D.set(e,"ready"),L.set(e,{...n,cachedAt:Date.now()})}function H(e,t){D.set(e,"failed"),L.set(e,{error:t?.message||String(t),failedAt:Date.now()}),k.delete(e)}function J(e){return D.get(e)==="ready"?k.get(e):null}var P=process.env.ZIBBY_SUBGRAPH_CACHE_DIR||"/tmp/zibby/subgraphs";function ee(){return`node${(process.versions?.node||"").split(".")[0]||"unknown"}-${process.platform}-${process.arch}`}var g=class extends Error{constructor(t,n){super(`in-process sub-graph fallback: ${t}${n?` (${n})`:""}`),this.fallback=!0,this.reason=t,this.detail=n||null,this.name="SubgraphFallback"}};function te(){let e=(process.env.SUBGRAPH_INTERNAL_URL||"").replace(/\/$/,""),t=(process.env.PROGRESS_API_URL||"").replace(/\/executions\/?$/,""),n=e||t,r=process.env.PROJECT_ID,a=process.env.PROJECT_API_TOKEN;if(!n||!r||!a)throw new g("env","SUBGRAPH_INTERNAL_URL/PROGRESS_API_URL/PROJECT_ID/PROJECT_API_TOKEN missing");return{apiBase:n,projectId:r,authToken:a}}async function re({apiBase:e,authToken:t,body:n}){let r;try{r=await fetch(`${e}/internal/subgraph/begin`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify(n)})}catch(o){throw new g("network",`begin fetch failed: ${o.message}`)}let a=null;try{a=await r.json()}catch{}if(!r.ok){if(r.status===404){let o=new Error(`Sub-graph child '${n.childWorkflowType}' not found in project`);throw o.code="SUBGRAPH_NOT_FOUND",o.status=404,o}if(r.status===429){let o=a?.quotaInfo||{},i=new Error(`Sub-graph blocked by quota (${o.used??"?"}/${o.limit??"?"} on ${o.planId||"plan"})`);throw i.code="SUBGRAPH_QUOTA_EXCEEDED",i.status=429,i.quotaInfo=o,i}if(r.status===400&&a?.validationErrors){let o=new Error(`Sub-graph rejected input: ${a?.error||a?.message||"validation failed"}`);throw o.code="SUBGRAPH_INVALID_INPUT",o.status=400,o.validationErrors=a.validationErrors,o.missing=a.missing,o}throw new g("begin-status",`begin returned ${r.status}`)}return a?.data||a}async function x({apiBase:e,authToken:t,payload:n}){try{let r=await fetch(`${e}/internal/subgraph/finalize`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify(n)});r.ok||f.warn(`[in-process subgraph] finalize returned ${r.status} for ${n.childExecutionId}`)}catch(r){f.warn(`[in-process subgraph] finalize failed: ${r.message}`)}}async function ne(e,t){let n=T(t,".ready"),r=T(t,"graph.mjs");if(v(n)&&v(r))return;K(t,{recursive:!0});let a=T(t,".lock"),o=!1;try{let{openSync:i,closeSync:c}=await import("node:fs"),I=i(a,"wx");c(I),o=!0}catch(i){if(i.code!=="EEXIST")throw i}if(!o){let i=Date.now()+3e4;for(;Date.now()<i;){if(v(n)&&v(r))return;await new Promise(c=>setTimeout(c,100))}throw new g("bundle-extract-timeout","sibling extract did not complete within 30s")}try{await new Promise((I,b)=>{let p=z("curl",["-fsSL",e],{stdio:["ignore","pipe","inherit"]}),E=z("tar",["-xzf","-","-C",t],{stdio:["pipe","inherit","inherit"]});p.stdout.pipe(E.stdin);let d,l,m=()=>{if(d!==void 0&&l!==void 0){if(d!==0)return b(new Error(`curl exited ${d}`));if(l!==0)return b(new Error(`tar exited ${l}`));I()}};p.on("close",B=>{d=B,m()}),E.on("close",B=>{l=B,m()}),p.on("error",b),E.on("error",b)});let{writeFileSync:i,unlinkSync:c}=await import("node:fs");i(n,"");try{c(a)}catch{}}catch(i){try{let{unlinkSync:c}=await import("node:fs");c(a)}catch{}throw new g("bundle-extract-failed",i.message)}}async function oe(e){let t=T(e,"graph.mjs");if(!v(t))throw new g("entry-missing",`graph.mjs missing under ${e}`);let n;try{n=await import(Q(t).href)}catch(a){throw new g("import-failed",`${a?.code||a?.name||"unknown"}: ${a.message}`)}let r=n.default||Object.values(n).find(a=>typeof a=="function"&&a.prototype?.buildGraph);if(!r)throw new g("entry-class-missing","no buildGraph() class export found");return r}async function N(e,t={}){if(!e||typeof e!="string")throw new Error("runInProcessSubgraph: workflowName (string) is required");let n=A(),r;try{r=te()}catch(s){throw s}f.debug(`[in-process subgraph] begin '${e}' parent=${n.executionId||"<root>"}`);let a=await re({apiBase:r.apiBase,authToken:r.authToken,body:{parentExecutionId:n.executionId,childWorkflowType:e,input:t.input||{},...t.conversationId?{conversationId:t.conversationId}:{}}}),{childExecutionId:o,runtimeTag:i,bundlePresignedUrl:c,sourcesPresignedUrl:I,workflowVersion:b,workflowUuid:p,bundleReady:E}=a,d=ee();if(i&&i!==d)throw await x({apiBase:r.apiBase,authToken:r.authToken,payload:{childExecutionId:o,status:"canceled",error:{message:`runtimeTag mismatch: parent=${d} child=${i}`,code:"RUNTIME_MISMATCH"}}}),new g("runtime-mismatch",`${d} vs ${i}`);if(!E||!c)throw await x({apiBase:r.apiBase,authToken:r.authToken,payload:{childExecutionId:o,status:"canceled",error:{message:"bundle not ready for in-process; falling back to HTTP",code:"NO_BUNDLE"}}}),new g("no-bundle","workflow bundle not built yet");let l=J(e);if(!l){let s=T(P,`${p}@${b||"0"}`);try{await ne(c,s);try{ae()}catch{}}catch(u){throw u.fallback&&await x({apiBase:r.apiBase,authToken:r.authToken,payload:{childExecutionId:o,status:"failed",error:{message:u.message,code:u.reason}}}),u}try{l=await oe(s),G(e,l,{workflowUuid:p,version:b,runtimeTag:i,cacheDir:s})}catch(u){throw H(e,u),await x({apiBase:r.apiBase,authToken:r.authToken,payload:{childExecutionId:o,status:"failed",error:{message:u.message,code:u.reason||"IMPORT_FAILED"}}}),u.fallback?u:new g("import-failed",u.message)}}let m=Date.now(),C=await(typeof l=="function"&&l.prototype?.buildGraph?new l:l).buildGraph(),w={...t.input||{}},y,S;try{y=await j({executionId:o,parentExecutionId:n.executionId,conversationId:t.conversationId!==void 0?t.conversationId:n.conversationId,dispatchMode:"inprocess"},()=>C.run(t.parentAgent,w,{signal:t.signal})),S=y&&typeof y=="object"&&"state"in y?y.state:y}catch(s){throw await x({apiBase:r.apiBase,authToken:r.authToken,payload:{childExecutionId:o,status:"failed",error:{message:s.message,code:s.code||"CHILD_THREW",stack:s.stack},durationMs:Date.now()-m}}),s}if(y&&typeof y=="object"&&y.stoppedExternally){await x({apiBase:r.apiBase,authToken:r.authToken,payload:{childExecutionId:o,status:"canceled",finalState:S,durationMs:Date.now()-m}});let s=new Error(`Sub-graph '${e}' canceled by parent abort`);throw s.code="SUBGRAPH_CANCELED",s.subgraphJobId=o,s}return await x({apiBase:r.apiBase,authToken:r.authToken,payload:{childExecutionId:o,status:"completed",finalState:S,durationMs:Date.now()-m}}),{finalState:S,executionId:o}}function se(e){let t=0,n=[e];for(;n.length;){let r=n.pop(),a;try{a=F(r)}catch{continue}if(a.isDirectory()){let o;try{o=q(r)}catch{continue}for(let i of o)n.push(T(r,i))}else t+=a.size}return t}function ae({cap:e=Number(process.env.ZIBBY_SUBGRAPH_CACHE_CAP_BYTES||2*1024*1024*1024)}={}){try{if(!v(P))return{evicted:0,freedBytes:0};let t=q(P),n=[],r=0;for(let c of t){let I=T(P,c),b;try{b=F(I)}catch{continue}let p=b.isDirectory()?se(I):b.size;r+=p,n.push({name:c,full:I,size:p,mtimeMs:b.mtimeMs})}if(r<=e)return{evicted:0,freedBytes:0,totalBytes:r};n.sort((c,I)=>c.mtimeMs-I.mtimeMs);let a=Math.floor(e*.7),o=0,i=0;for(let c of n){if(r-o<=a)break;if(!v(T(c.full,".lock")))try{W(c.full,{recursive:!0,force:!0}),o+=c.size,i+=1}catch(I){f.debug(`[sub-graph cache] evict skip ${c.name}: ${I.message}`)}}return i>0&&f.info(`[sub-graph cache] evicted ${i} entr(y/ies), freed ${(o/1024/1024).toFixed(1)}MB`),{evicted:i,freedBytes:o,totalBytes:r-o}}catch(t){return f.debug(`[sub-graph cache] evict failed: ${t.message}`),{evicted:0,freedBytes:0}}}var ie=2e3,ce=600*1e3,ue=new Set(["completed","failed","canceled","timeout"]);function de(){let e=process.env.PROGRESS_API_URL;if(!e)throw new Error("Sub-graph dispatch requires PROGRESS_API_URL env var (set automatically on cloud runs). Sub-graphs are not supported in local in-process runs yet \u2014 deploy the parent and child to cloud.");return e.replace(/\/executions\/?$/,"")}function pe(){let e=process.env.PROJECT_ID;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_ID env var.");return e}function le(){let e=process.env.PROJECT_API_TOKEN;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_API_TOKEN env var.");return e}function he(){return process.env.EXECUTION_ID||null}function X(e,t){return t==null?e:typeof t=="function"?t(e):typeof t=="string"?t.split(".").reduce((n,r)=>n==null?n:n[r],e):e}async function ve(e,t={}){if(!e||typeof e!="string")throw new Error("dispatchSubgraph: workflowName (string) is required");let n=A(),r=Number(process.env.ZIBBY_SUBGRAPH_MAX_DEPTH||10);if((n.depth||0)>=r)throw new Error(`dispatchSubgraph('${e}'): sub-graph depth ${n.depth} reached cap of ${r}. Restructure the graph or raise ZIBBY_SUBGRAPH_MAX_DEPTH.`);if(process.env.ZIBBY_INPROCESS_SUBGRAPH!=="0"&&!t.async)try{f.debug(`[sub-graph] trying in-process for '${e}'`);let{finalState:s}=await N(e,{input:t.input,conversationId:t.conversationId,signal:t.signal,parentAgent:t.parentAgent}),u=X(s,t.output);return f.info(`[sub-graph] '${e}' completed in-process`),u}catch(s){if(s instanceof g||s?.fallback)f.info(`[sub-graph] in-process fallback for '${e}': ${s.reason||"unknown"} \u2014 using HTTP`);else throw s}let a=de(),o=pe(),i=le(),c=he(),I=`${a}/projects/${encodeURIComponent(o)}/workflows/${encodeURIComponent(e)}/trigger`,b={input:t.input||{},...c?{parentExecutionId:c}:{},...t.conversationId?{conversationId:t.conversationId}:{}};f.info(`[sub-graph] dispatching '${e}' (${t.async?"async":"sync"}) from parent ${c||"<none>"}`);let p=await fetch(I,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${i}`},body:JSON.stringify(b)});if(!p.ok){let s=null,u="";try{s=await p.json(),u=s?.error||s?.message||JSON.stringify(s)}catch{u=await p.text().catch(()=>"")}if(p.status===429){let h=s?.quotaInfo||{},_=new Error(`Sub-graph '${e}' blocked by execution quota (${h.used??"?"}/${h.limit??"?"} on plan ${h.planId||"unknown"}). Sub-workflow runs count toward the same monthly cap as user-triggered runs.`);throw _.code="SUBGRAPH_QUOTA_EXCEEDED",_.status=429,_.subgraph=e,_.quotaInfo=h,_}if(p.status===400){let h=new Error(`Sub-graph '${e}' rejected input: ${u}`);throw h.code="SUBGRAPH_INVALID_INPUT",h.status=400,h.subgraph=e,h.validationErrors=s?.validationErrors||null,h.missing=s?.missing||null,h}let $=new Error(`Sub-graph '${e}' trigger rejected (${p.status}): ${u}`);throw $.code="SUBGRAPH_TRIGGER_FAILED",$.status=p.status,$.subgraph=e,$}let E=await p.json(),d=E?.data?.jobId||E?.jobId;if(!d)throw new Error(`Sub-graph '${e}' trigger returned no jobId: ${JSON.stringify(E).slice(0,200)}`);if(t.async)return f.info(`[sub-graph] async dispatch of '${e}' \u2192 jobId=${d} (not waiting)`),{jobId:d,status:"accepted",workflow:e};let l=Number.isFinite(t.timeoutMs)?t.timeoutMs:ce,m=Number.isFinite(t.pollIntervalMs)?t.pollIntervalMs:ie,B=`${a}/executions/${encodeURIComponent(d)}`,C=Date.now()+l,w="accepted",y=0;for(;Date.now()<C;){await new Promise(h=>setTimeout(h,m)),y+=1;let s=await fetch(B,{headers:{Authorization:`Bearer ${i}`}});if(!s.ok){if(s.status>=500){f.warn(`[sub-graph] status poll for ${d} returned ${s.status}, will retry`);continue}throw new Error(`Sub-graph status poll failed for ${d}: ${s.status}`)}let u=await s.json(),$=u?.data||u?.execution||u;if(w=$?.status||w,ue.has(w)){if(w!=="completed"){let O=new Error(`Sub-graph '${e}' (${d}) ended in status '${w}'`);throw O.subgraphJobId=d,O.subgraphStatus=w,O}let h=$?.finalState||$?.state||{},_=X(h,t.output);return f.info(`[sub-graph] '${e}' (${d}) completed after ${y} polls`),_}}let S=new Error(`Sub-graph '${e}' (${d}) timed out after ${Math.round(l/1e3)}s (last status: ${w})`);throw S.subgraphJobId=d,S.subgraphStatus=w,S}export{ve as dispatchSubgraph};
|