@zibby/agent-workflow 0.6.0 → 0.6.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +30 -0
- package/dist/code-generator.js +15 -15
- package/dist/exec-context.d.ts +12 -0
- package/dist/exec-context.js +1 -1
- package/dist/graph-compiler.js +21 -21
- package/dist/graph.d.ts +40 -0
- package/dist/graph.js +25 -25
- package/dist/in-process-subgraph.js +1 -1
- package/dist/index.js +30 -30
- package/dist/node-registry.js +10 -10
- package/dist/node.js +17 -17
- package/dist/strategy-registry.js +8 -8
- package/dist/sub-graph-executor.js +1 -1
- package/package.json +1 -1
package/dist/node.js
CHANGED
|
@@ -1,34 +1,34 @@
|
|
|
1
|
-
var nt=Object.defineProperty;var N=(r,t,e)=>()=>{if(e)throw e[0];try{return r&&(t=r(r=0)),t}catch(o){throw e=[o],o}};var at=(r,t)=>{for(var e in t)nt(r,e,{get:t[e],enumerable:!0})};var D,
|
|
1
|
+
var nt=Object.defineProperty;var N=(r,t,e)=>()=>{if(e)throw e[0];try{return r&&(t=r(r=0)),t}catch(o){throw e=[o],o}};var at=(r,t)=>{for(var e in t)nt(r,e,{get:t[e],enumerable:!0})};var D,ct,A,f,T=N(()=>{D=()=>{},ct={debug:D,info:D,warn:(...r)=>console.warn("[workflow]",...r),error:(...r)=>console.error("[workflow]",...r)},A={impl:ct},f={debug:(...r)=>A.impl.debug?.(...r),info:(...r)=>A.impl.info?.(...r),warn:(...r)=>A.impl.warn?.(...r),error:(...r)=>A.impl.error?.(...r)}});var z=N(()=>{});function V(r){return pt.get(r)||null}var P,C,pt,kt,q=N(()=>{P=Symbol.for("@zibby/agent-workflow.skills"),C=Symbol.for("@zibby/agent-workflow.skills.sources");globalThis[P]||(globalThis[P]=new Map);globalThis[C]||(globalThis[C]=new Map);pt=globalThis[P],kt=globalThis[C]});var tt={};at(tt,{getAgentStrategy:()=>Q,invokeAgent:()=>ht,listStrategies:()=>ft,registerStrategy:()=>dt,resolveInvocationModel:()=>X});function dt(r){if(!r||typeof r.getName!="function"||typeof r.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let t=E.findIndex(e=>e.getName()===r.getName());t>=0?E[t]=r:E.push(r)}function ft(){return E.map(r=>r.getName())}function X({config:r={},options:t={},strategyName:e,envModel:o}={}){let n=r.models||{},i=t.nodeName&&n[t.nodeName]||null,a=n.default||null,y=r.agent?.[e]?.model||null,p=(typeof o=="string"?o.trim():"")||null;return i||a||y||t.model||p||null}function Q(r={}){let{state:t={},preferredAgent:e=null}=r,o=e||t.agentType||process.env.AGENT_TYPE;if(!o){let i=E.map(a=>a.getName()).join(", ")||"none registered";throw new Error(`No agent specified. Set agentType in state or AGENT_TYPE env var. Available: ${i}`)}f.debug(`[workflow] agent selection: requested=${o}`);let n=E.find(i=>i.getName()===o);if(!n){let i=E.map(a=>a.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${o}'. Available: ${i}`)}if(!n.canHandle(r))throw new Error(`Agent '${o}' is not available in this environment. Check credentials/environment.`);return f.debug(`[workflow] using agent: ${n.getName()}`),n}async function ht(r,t={},e={}){let o=t.state&&typeof t.state.getAll=="function"?t.state.getAll():t.state||{},n={...t,state:o},i=Q(n),a=o.config||e.config||{},y=X({config:a,options:e,strategyName:i.name,envModel:process.env.MODEL}),p={...e,model:y,workspace:o.workspace||e.workspace,schema:e.schema||t.schema,images:e.images||t.images||[],skills:e.skills||t.skills||[],extraMcpServers:e.extraMcpServers||o.extraMcpServers||t.extraMcpServers||[],plugins:e.plugins||t.plugins||[],config:a},_=r,k=p.skills||[];if(k.length>0&&!e.skipPromptFragments){let m=e.connectedIntegrations;if(!m){let S=process.env.WORKFLOW_CONNECTED_INTEGRATIONS;if(typeof S=="string"&&S.trim()!==""){m={};for(let g of S.split(",").map(l=>l.trim()).filter(Boolean))m[g]=!0}}let O=S=>{let g=S&&S.requiresIntegration;return!g||!m?!0:(Array.isArray(g)?g:[g]).some(d=>m[d]===!0)},c=k.map(S=>{let g=V(S);if(!O(g))return null;let l=g?.promptFragment;return typeof l=="function"?l():l}).filter(Boolean);c.length>0&&(_+=`
|
|
2
2
|
|
|
3
|
-
${
|
|
3
|
+
${c.join(`
|
|
4
4
|
|
|
5
|
-
`)}`)}let
|
|
6
|
-
fields: ${
|
|
5
|
+
`)}`)}let s=o._currentNodeConfig?.stores;if(Array.isArray(s)&&s.length>0&&typeof s[0]=="object"){let m=s.length<=8,O=s.map(c=>{let S=c?.id??c?.storeId??"",g=(c?.name??"").toString().trim()||S,l=c?.type?` \xB7 ${c.type}`:"",d=(c?.description||"").toString().replace(/\s+/g," ").trim(),$=`- ${g} \xB7 ${d||"(no description)"}${l} (id: ${S})`;if(m&&c?.schema&&typeof c.schema=="object"){let u=c.schema.properties&&typeof c.schema.properties=="object"?Object.keys(c.schema.properties):Object.keys(c.schema);u.length&&($+=`
|
|
6
|
+
fields: ${u.join(", ")}`)}return $});_+=`
|
|
7
7
|
|
|
8
8
|
AVAILABLE STORES (pick a store by its description and pass its NAME to the store tool):
|
|
9
|
-
${
|
|
10
|
-
`)}`}let
|
|
9
|
+
${O.join(`
|
|
10
|
+
`)}`}let w=o._currentNodeConfig?.extraPromptInstructions?.trim();return w&&(_+=`
|
|
11
11
|
|
|
12
12
|
\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
|
|
13
13
|
PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PREVIOUS CONTENT
|
|
14
14
|
\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
|
|
15
15
|
|
|
16
|
-
${
|
|
17
|
-
`),
|
|
16
|
+
${w}
|
|
17
|
+
`),f.debug(`[workflow] prompt length: ${_.length} chars`),i.invoke(_,p)}var M,E,et=N(()=>{z();T();q();M=Symbol.for("@zibby/agent-workflow.strategies");globalThis[M]||(globalThis[M]=[]);E=globalThis[M]});import b from"handlebars";var W=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(n=>n[0]);for(let n of o)try{return this.validate(JSON.parse(n))}catch(i){if(!(i instanceof SyntaxError))throw i}return this.validate({result:t.trim()})}validate(t){let e=[];for(let[o,n]of Object.entries(this.schema)){if(n.required&&!(o in t)&&e.push(`Missing required field: ${o}`),o in t&&n.type){let i=typeof t[o];i!==n.type&&e.push(`Field '${o}' expected ${n.type}, got ${i}`)}if(n.validate&&o in t){let i=n.validate(t[o]);i&&e.push(`Field '${o}': ${i}`)}}if(e.length>0)throw new Error(`Output validation failed:
|
|
18
18
|
${e.join(`
|
|
19
|
-
`)}`);return t}};
|
|
20
|
-
`?(a+=
|
|
19
|
+
`)}`);return t}};T();import{writeFileSync as F,readFileSync as rt,existsSync as ot,mkdirSync as mt}from"node:fs";import{join as j,dirname as gt}from"node:path";import h from"chalk";var lt="__WORKFLOW_GRAPH_LOG__",I=h.gray("\u2502"),ut=h.gray("\u250C"),G=h.gray("\u2514"),v=h.green("\u25C6"),H=h.hex("#c084fc")("\u25C6"),J=h.hex("#2dd4bf")("\u25C6"),x=h.red("\u25C6"),U=`${I} `,B=2;function K(r){return r<1e3?`${r}ms`:`${(r/1e3).toFixed(1)}s`}function Y(r,t){return(e,o,n)=>{if(typeof e!="string")return r(e,o,n);let i=process.stdout.columns||120,a="";for(let y=0;y<e.length;y++){let p=e[y];t.lineStart&&(a+=U,t.col=B,t.lineStart=!1),p===`
|
|
20
|
+
`?(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>=i&&(a+=`
|
|
21
21
|
${U}`,t.col=B))}return r(a,o,n)}}var R=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=Y(this._origStdoutWrite,t),process.stderr.write=Y(this._origStderrWrite,e)}_stopIntercepting(){this._origStdoutWrite&&(this._outState&&!this._outState.lineStart&&this._origStdoutWrite(`
|
|
22
22
|
`),process.stdout.write=this._origStdoutWrite),this._origStderrWrite&&(this._errState&&!this._errState.lineStart&&this._origStderrWrite(`
|
|
23
23
|
`),process.stderr.write=this._origStderrWrite),this._origStdoutWrite=null,this._origStderrWrite=null}_rawWrite(t){(this._origStdoutWrite||process.stdout.write.bind(process.stdout))(`${t}
|
|
24
|
-
`)}_emitGraphLogMarker(t){if(!this._emitWorkflowGraphMarkers)return;let e=`${
|
|
24
|
+
`)}_emitGraphLogMarker(t){if(!this._emitWorkflowGraphMarkers)return;let e=`${lt}${JSON.stringify(t)}
|
|
25
25
|
`;this._origStdoutWrite?this._origStdoutWrite(e):process.stdout.write(e)}_writeDot(t,e){this._origStdoutWrite?(this._outState&&!this._outState.lineStart&&(this._origStdoutWrite(`
|
|
26
26
|
`),this._outState.lineStart=!0,this._outState.col=0),this._origStdoutWrite(`${t} ${e}
|
|
27
27
|
`)):process.stdout.write.bind(process.stdout)(`${t} ${e}
|
|
28
|
-
`)}step(t){this._origStdoutWrite?this._writeDot(
|
|
29
|
-
`)}stepInfo(t){this.step(t)}stepTool(t){this._origStdoutWrite?this._writeDot(H,t):process.stdout.write.bind(process.stdout)(`${
|
|
30
|
-
`)}stepMemory(t){let e=
|
|
31
|
-
`)}stepFail(t){this._origStdoutWrite?this._writeDot(
|
|
32
|
-
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${ut} ${t}`),this._startIntercepting()}nodeComplete(t,e={}){this._stopIntercepting();let{duration:o,details:n}=e;if(n)for(let a of n)this._rawWrite(`${
|
|
28
|
+
`)}step(t){this._origStdoutWrite?this._writeDot(v,t):process.stdout.write.bind(process.stdout)(`${I} ${v} ${t}
|
|
29
|
+
`)}stepInfo(t){this.step(t)}stepTool(t){this._origStdoutWrite?this._writeDot(H,t):process.stdout.write.bind(process.stdout)(`${I} ${H} ${t}
|
|
30
|
+
`)}stepMemory(t){let e=h.hex("#2dd4bf")(t);this._origStdoutWrite?this._writeDot(J,e):process.stdout.write.bind(process.stdout)(`${I} ${J} ${e}
|
|
31
|
+
`)}stepFail(t){this._origStdoutWrite?this._writeDot(x,h.red(t)):process.stdout.write.bind(process.stdout)(`${I} ${x} ${h.red(t)}
|
|
32
|
+
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${ut} ${t}`),this._startIntercepting()}nodeComplete(t,e={}){this._stopIntercepting();let{duration:o,details:n}=e;if(n)for(let a of n)this._rawWrite(`${v} ${a}`);let i=o?h.dim(` ${K(o)}`):"";this._rawWrite(`${G} ${h.green("done")}${i}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}nodeFailed(t,e,o={}){this._stopIntercepting();let{duration:n}=o,i=n?h.dim(` ${K(n)}`):"";this._rawWrite(`${x} ${h.red(e)}`),this._rawWrite(`${G} ${h.red("failed")}${i}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}route(t,e){this._rawWrite(h.dim(` ${t} \u2192 ${e}`)),this._rawWrite("")}graphComplete(){}},Z=new R;var L=".session-info.json";var Et=Object.freeze(["codebase-memory","code-scan","artifact"]);b.helpers.inc||b.registerHelper("inc",r=>Number(r)+1);b.helpers.json||b.registerHelper("json",r=>JSON.stringify(r,null,2));b.helpers.eq||b.registerHelper("eq",(r,t)=>r===t);var st=class{constructor(t){if(this.config=t,this.name=t.name,this.prompt=t.prompt,this.outputSchema=t.outputSchema,!this.outputSchema&&!t._isCustomCode&&!t._isRouter)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 W(t.outputSchema):null,this.retries=t.retries||0,this.onComplete=t.onComplete,this.customExecute=t.execute}async execute(t,e){if(this.config._isRouter)return f.debug(`[workflow] node '${this.name}': router passthrough (routing happens on its conditional edges)`),{success:!0,output:{},raw:null};let o=()=>e&&typeof e.getAll=="function"?e.getAll():t,n=s=>e&&typeof e.get=="function"?e.get(s):t?.[s];if(typeof this.customExecute=="function"){f.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let s=await this.customExecute(t);return typeof s=="object"&&s!==null&&s.success===!1?{success:!1,error:s.error||"Node execution failed",raw:s.raw||null}:this.isZodSchema?(f.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(s),raw:null}):{success:!0,output:s,raw:null}}catch(s){return f.error(`[workflow] node '${this.name}' failed: ${s.message}`),s.name==="ZodError"&&f.error(`Schema errors: ${JSON.stringify(s.issues||s.errors,null,2)}`),{success:!1,error:s.message,raw:null}}}let i;typeof this.prompt=="function"?i=this.prompt(o()):typeof this.prompt=="string"&&this.prompt.includes("{{")?(this._compiledPrompt||(this._compiledPrompt=b.compile(this.prompt,{noEscape:!0})),i=this._compiledPrompt(o())):i=this.prompt;let a=n("_skillHints");a&&(i=`${a}
|
|
33
33
|
|
|
34
|
-
${
|
|
34
|
+
${i}`);let y=o(),p=y.cwd||process.cwd(),_=y.sessionPath;try{if(_){let s=j(_,L);if(ot(s)){let m=JSON.parse(rt(s,"utf-8"));m.currentNode=this.name,F(s,JSON.stringify(m,null,2),"utf-8")}let w=j(_,"..",L);if(ot(w))try{let m=JSON.parse(rt(w,"utf-8"));m.currentNode=this.name,F(w,JSON.stringify(m,null,2),"utf-8")}catch{}}}catch(s){f.debug(`[workflow] could not update session info: ${s.message}`)}let k=null;for(let s=0;s<=this.retries;s++)try{f.debug(`[workflow] node '${this.name}' attempt ${s}`);let w=o().config||{},m=w.agents||{},O=this.config.agent??m[this.name]??null,c={state:o()};O&&(c.preferredAgent=O);let S={workspace:p,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],plugins:this.config.plugins||[],sessionPath:_,config:w,nodeName:this.name,timeout:this.config?.timeout||3e5},g=t?._coreInvokeAgent;g||(g=(await Promise.resolve().then(()=>(et(),tt))).invokeAgent);let l=await g(i,c,S),d,$;if(typeof l=="string"?(d=l,$=null):l.structured?(d=l.raw||JSON.stringify(l.structured,null,2),$=l.structured):(d=l.raw||JSON.stringify(l,null,2),$=l.extracted||null),_)try{let u=j(_,this.name,"raw_stream_output.txt");mt(gt(u),{recursive:!0}),F(u,typeof d=="string"?d:JSON.stringify(d),"utf-8")}catch(u){f.debug(`[workflow] could not save raw output: ${u.message}`)}if(this.isZodSchema&&$){f.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify($,null,2)}`);let u=$;if(typeof this.onComplete=="function")try{u=await this.onComplete(o(),$)}catch(it){f.warn(`[workflow] onComplete hook failed: ${it.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 f.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(u,null,2)}`),Z.step("Output parsed"),{success:!0,output:u,raw:d}}return{success:!0,output:d,raw:d}}catch(w){k=w,s<this.retries&&f.info(`[workflow] node '${this.name}' failed, retrying (${s+1}/${this.retries})\u2026`)}return{success:!1,error:k.message,raw:null}}};export{st as Node};
|
|
@@ -1,17 +1,17 @@
|
|
|
1
|
-
var
|
|
1
|
+
var O=()=>{},x={debug:O,info:O,warn:(...e)=>console.warn("[workflow]",...e),error:(...e)=>console.error("[workflow]",...e)},y={impl:x};var h={debug:(...e)=>y.impl.debug?.(...e),info:(...e)=>y.impl.info?.(...e),warn:(...e)=>y.impl.warn?.(...e),error:(...e)=>y.impl.error?.(...e)};var S=Symbol.for("@zibby/agent-workflow.skills"),E=Symbol.for("@zibby/agent-workflow.skills.sources");globalThis[S]||(globalThis[S]=new Map);globalThis[E]||(globalThis[E]=new Map);var _=globalThis[S],L=globalThis[E];function $(e){return _.get(e)||null}var v=Symbol.for("@zibby/agent-workflow.strategies");globalThis[v]||(globalThis[v]=[]);var f=globalThis[v];function V(e){if(!e||typeof e.getName!="function"||typeof e.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let t=f.findIndex(n=>n.getName()===e.getName());t>=0?f[t]=e:f.push(e)}function q(){return f.map(e=>e.getName())}function R({config:e={},options:t={},strategyName:n,envModel:r}={}){let l=e.models||{},a=t.nodeName&&l[t.nodeName]||null,c=l.default||null,w=e.agent?.[n]?.model||null,d=(typeof r=="string"?r.trim():"")||null;return a||c||w||t.model||d||null}function M(e={}){let{state:t={},preferredAgent:n=null}=e,r=n||t.agentType||process.env.AGENT_TYPE;if(!r){let a=f.map(c=>c.getName()).join(", ")||"none registered";throw new Error(`No agent specified. Set agentType in state or AGENT_TYPE env var. Available: ${a}`)}h.debug(`[workflow] agent selection: requested=${r}`);let l=f.find(a=>a.getName()===r);if(!l){let a=f.map(c=>c.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${r}'. Available: ${a}`)}if(!l.canHandle(e))throw new Error(`Agent '${r}' is not available in this environment. Check credentials/environment.`);return h.debug(`[workflow] using agent: ${l.getName()}`),l}async function z(e,t={},n={}){let r=t.state&&typeof t.state.getAll=="function"?t.state.getAll():t.state||{},l={...t,state:r},a=M(l),c=r.config||n.config||{},w=R({config:c,options:n,strategyName:a.name,envModel:process.env.MODEL}),d={...n,model:w,workspace:r.workspace||n.workspace,schema:n.schema||t.schema,images:n.images||t.images||[],skills:n.skills||t.skills||[],extraMcpServers:n.extraMcpServers||r.extraMcpServers||t.extraMcpServers||[],plugins:n.plugins||t.plugins||[],config:c},m=e,N=d.skills||[];if(N.length>0&&!n.skipPromptFragments){let p=n.connectedIntegrations;if(!p){let i=process.env.WORKFLOW_CONNECTED_INTEGRATIONS;if(typeof i=="string"&&i.trim()!==""){p={};for(let s of i.split(",").map(g=>g.trim()).filter(Boolean))p[s]=!0}}let b=i=>{let s=i&&i.requiresIntegration;return!s||!p?!0:(Array.isArray(s)?s:[s]).some(k=>p[k]===!0)},o=N.map(i=>{let s=$(i);if(!b(s))return null;let g=s?.promptFragment;return typeof g=="function"?g():g}).filter(Boolean);o.length>0&&(m+=`
|
|
2
2
|
|
|
3
|
-
${
|
|
3
|
+
${o.join(`
|
|
4
4
|
|
|
5
|
-
`)}`)}let
|
|
6
|
-
fields: ${
|
|
5
|
+
`)}`)}let u=r._currentNodeConfig?.stores;if(Array.isArray(u)&&u.length>0&&typeof u[0]=="object"){let p=u.length<=8,b=u.map(o=>{let i=o?.id??o?.storeId??"",s=(o?.name??"").toString().trim()||i,g=o?.type?` \xB7 ${o.type}`:"",k=(o?.description||"").toString().replace(/\s+/g," ").trim(),A=`- ${s} \xB7 ${k||"(no description)"}${g} (id: ${i})`;if(p&&o?.schema&&typeof o.schema=="object"){let I=o.schema.properties&&typeof o.schema.properties=="object"?Object.keys(o.schema.properties):Object.keys(o.schema);I.length&&(A+=`
|
|
6
|
+
fields: ${I.join(", ")}`)}return A});m+=`
|
|
7
7
|
|
|
8
8
|
AVAILABLE STORES (pick a store by its description and pass its NAME to the store tool):
|
|
9
|
-
${
|
|
10
|
-
`)}`}let
|
|
9
|
+
${b.join(`
|
|
10
|
+
`)}`}let T=r._currentNodeConfig?.extraPromptInstructions?.trim();return T&&(m+=`
|
|
11
11
|
|
|
12
12
|
\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
|
|
13
13
|
PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PREVIOUS CONTENT
|
|
14
14
|
\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
|
|
15
15
|
|
|
16
|
-
${
|
|
17
|
-
`),
|
|
16
|
+
${T}
|
|
17
|
+
`),h.debug(`[workflow] prompt length: ${m.length} chars`),a.invoke(m,d)}export{M as getAgentStrategy,z as invokeAgent,q as listStrategies,V as registerStrategy,R as resolveInvocationModel};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
var D=()=>{},Z={debug:D,info:D,warn:(...e)=>console.warn("[workflow]",...e),error:(...e)=>console.error("[workflow]",...e)},R={impl:Z};var h={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 W,existsSync as x,statSync as N,readdirSync as X,rmSync as ee}from"node:fs";import{spawn as z}from"node:child_process";import{join as v}from"node:path";import{pathToFileURL as te}from"node:url";import{AsyncLocalStorage as ne}from"node:async_hooks";import{AsyncLocalStorage as K}from"node:async_hooks";var k=new K;function A(){let e=k.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 r=k.getStore()||A(),n=Object.freeze({executionId:e.executionId,parentExecutionId:e.parentExecutionId??r.executionId??null,depth:(r.depth||0)+(e.executionId!==r.executionId?1:0),conversationId:e.conversationId!==void 0?e.conversationId:r.conversationId??null,dispatchMode:e.dispatchMode??null});return k.run(n,t)}var U=new Map,M=new Map,L=new Map;function H(e,t,r={}){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");U.set(e,t),M.set(e,"ready"),L.set(e,{...r,cachedAt:Date.now()})}function G(e,t){M.set(e,"failed"),L.set(e,{error:t?.message||String(t),failedAt:Date.now()}),U.delete(e)}function J(e){return M.get(e)==="ready"?U.get(e):null}var P=process.env.ZIBBY_SUBGRAPH_CACHE_DIR||"/tmp/zibby/subgraphs";function re(){return`node${(process.versions?.node||"").split(".")[0]||"unknown"}-${process.platform}-${process.arch}`}var g=class extends Error{constructor(t,r){super(`in-process sub-graph fallback: ${t}${r?` (${r})`:""}`),this.fallback=!0,this.reason=t,this.detail=r||null,this.name="SubgraphFallback"}},F=new ne,q=Promise.resolve();async function oe(e,t){let r=e&&typeof e=="object"&&!Array.isArray(e)?Object.entries(e).filter(([s,i])=>typeof s=="string"&&s&&typeof i=="string"):[];if(r.length===0)return t();let n=F.getStore()===!0,a=null;if(!n){let s=q;q=new Promise(i=>{a=i}),await s}let o=new Map;try{for(let[s,i]of r)o.set(s,Object.prototype.hasOwnProperty.call(process.env,s)?process.env[s]:void 0),process.env[s]=i;return h.debug(`[in-process subgraph] scoped ${r.length} child env var(s)${n?" (nested)":""}`),await F.run(!0,t)}finally{for(let[s,i]of o)i===void 0?delete process.env[s]:process.env[s]=i;a&&a()}}function se(){let e=(process.env.SUBGRAPH_INTERNAL_URL||"").replace(/\/$/,""),t=(process.env.PROGRESS_API_URL||"").replace(/\/executions\/?$/,""),r=e||t,n=process.env.PROJECT_ID,a=process.env.PROJECT_API_TOKEN;if(!r||!n||!a)throw new g("env","SUBGRAPH_INTERNAL_URL/PROGRESS_API_URL/PROJECT_ID/PROJECT_API_TOKEN missing");return{apiBase:r,projectId:n,authToken:a}}async function ae({apiBase:e,authToken:t,body:r}){let n;try{n=await fetch(`${e}/internal/subgraph/begin`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify(r)})}catch(o){throw new g("network",`begin fetch failed: ${o.message}`)}let a=null;try{a=await n.json()}catch{}if(!n.ok){if(n.status===404){let o=new Error(`Sub-graph child '${r.childWorkflowType}' not found in project`);throw o.code="SUBGRAPH_NOT_FOUND",o.status=404,o}if(n.status===429){let o=a?.quotaInfo||{},s=new Error(`Sub-graph blocked by quota (${o.used??"?"}/${o.limit??"?"} on ${o.planId||"plan"})`);throw s.code="SUBGRAPH_QUOTA_EXCEEDED",s.status=429,s.quotaInfo=o,s}if(n.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 ${n.status}`)}return a?.data||a}async function _({apiBase:e,authToken:t,payload:r}){try{let n=await fetch(`${e}/internal/subgraph/finalize`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify(r)});n.ok||h.warn(`[in-process subgraph] finalize returned ${n.status} for ${r.childExecutionId}`)}catch(n){h.warn(`[in-process subgraph] finalize failed: ${n.message}`)}}async function ie(e,t){let r=v(t,".ready"),n=v(t,"graph.mjs");if(x(r)&&x(n))return;W(t,{recursive:!0});let a=v(t,".lock"),o=!1;try{let{openSync:s,closeSync:i}=await import("node:fs"),y=s(a,"wx");i(y),o=!0}catch(s){if(s.code!=="EEXIST")throw s}if(!o){let s=Date.now()+3e4;for(;Date.now()<s;){if(x(r)&&x(n))return;await new Promise(i=>setTimeout(i,100))}throw new g("bundle-extract-timeout","sibling extract did not complete within 30s")}try{await new Promise((y,b)=>{let l=z("curl",["-fsSL",e],{stdio:["ignore","pipe","inherit"]}),S=z("tar",["-xzf","-","-C",t],{stdio:["pipe","inherit","inherit"]});l.stdout.pipe(S.stdin);let p,E,w=()=>{if(p!==void 0&&E!==void 0){if(p!==0)return b(new Error(`curl exited ${p}`));if(E!==0)return b(new Error(`tar exited ${E}`));y()}};l.on("close",$=>{p=$,w()}),S.on("close",$=>{E=$,w()}),l.on("error",b),S.on("error",b)});let{writeFileSync:s,unlinkSync:i}=await import("node:fs");s(r,"");try{i(a)}catch{}}catch(s){try{let{unlinkSync:i}=await import("node:fs");i(a)}catch{}throw new g("bundle-extract-failed",s.message)}}async function ce(e){let t=v(e,"graph.mjs");if(!x(t))throw new g("entry-missing",`graph.mjs missing under ${e}`);let r;try{r=await import(te(t).href)}catch(a){throw new g("import-failed",`${a?.code||a?.name||"unknown"}: ${a.message}`)}let n=r.default||Object.values(r).find(a=>typeof a=="function"&&a.prototype?.buildGraph);if(!n)throw new g("entry-class-missing","no buildGraph() class export found");return n}async function Y(e,t={}){if(!e||typeof e!="string")throw new Error("runInProcessSubgraph: workflowName (string) is required");let r=A(),n;try{n=se()}catch(c){throw c}h.debug(`[in-process subgraph] begin '${e}' parent=${r.executionId||"<root>"}`);let a=await ae({apiBase:n.apiBase,authToken:n.authToken,body:{parentExecutionId:r.executionId,childWorkflowType:e,input:t.input||{},...t.conversationId?{conversationId:t.conversationId}:{}}}),{childExecutionId:o,runtimeTag:s,bundlePresignedUrl:i,sourcesPresignedUrl:y,workflowVersion:b,workflowUuid:l,bundleReady:S,nodeConfigs:p}=a,E=re();if(s&&s!==E)throw await _({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"canceled",error:{message:`runtimeTag mismatch: parent=${E} child=${s}`,code:"RUNTIME_MISMATCH"}}}),new g("runtime-mismatch",`${E} vs ${s}`);if(!S||!i)throw await _({apiBase:n.apiBase,authToken:n.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 w=J(e);if(!w){let c=v(P,`${l}@${b||"0"}`);try{await ie(i,c);try{de()}catch{}}catch(d){throw d.fallback&&await _({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"failed",error:{message:d.message,code:d.reason}}}),d}try{w=await ce(c),H(e,w,{workflowUuid:l,version:b,runtimeTag:s,cacheDir:c})}catch(d){throw G(e,d),await _({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"failed",error:{message:d.message,code:d.reason||"IMPORT_FAILED"}}}),d.fallback?d:new g("import-failed",d.message)}}let $=Date.now(),C=a.env&&typeof a.env=="object"&&!Array.isArray(a.env)?a.env:null,m=p&&typeof p=="object"&&!Array.isArray(p)&&Object.keys(p).length>0,B={...t.input||{},...m?{nodeConfigs:p}:{}},I,u;try{I=await oe(C,async()=>{let d=await(typeof w=="function"&&w.prototype?.buildGraph?new w:w).buildGraph();return j({executionId:o,parentExecutionId:r.executionId,conversationId:t.conversationId!==void 0?t.conversationId:r.conversationId,dispatchMode:"inprocess"},()=>d.run(t.parentAgent,B,{signal:t.signal}))}),u=I&&typeof I=="object"&&"state"in I?I.state:I}catch(c){throw await _({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"failed",error:{message:c.message,code:c.code||"CHILD_THREW",stack:c.stack},durationMs:Date.now()-$}}),c}if(I&&typeof I=="object"&&I.stoppedExternally){await _({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"canceled",finalState:u,durationMs:Date.now()-$}});let c=new Error(`Sub-graph '${e}' canceled by parent abort`);throw c.code="SUBGRAPH_CANCELED",c.subgraphJobId=o,c}return await _({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"completed",finalState:u,durationMs:Date.now()-$}}),{finalState:u,executionId:o}}function ue(e){let t=0,r=[e];for(;r.length;){let n=r.pop(),a;try{a=N(n)}catch{continue}if(a.isDirectory()){let o;try{o=X(n)}catch{continue}for(let s of o)r.push(v(n,s))}else t+=a.size}return t}function de({cap:e=Number(process.env.ZIBBY_SUBGRAPH_CACHE_CAP_BYTES||2*1024*1024*1024)}={}){try{if(!x(P))return{evicted:0,freedBytes:0};let t=X(P),r=[],n=0;for(let i of t){let y=v(P,i),b;try{b=N(y)}catch{continue}let l=b.isDirectory()?ue(y):b.size;n+=l,r.push({name:i,full:y,size:l,mtimeMs:b.mtimeMs})}if(n<=e)return{evicted:0,freedBytes:0,totalBytes:n};r.sort((i,y)=>i.mtimeMs-y.mtimeMs);let a=Math.floor(e*.7),o=0,s=0;for(let i of r){if(n-o<=a)break;if(!x(v(i.full,".lock")))try{ee(i.full,{recursive:!0,force:!0}),o+=i.size,s+=1}catch(y){h.debug(`[sub-graph cache] evict skip ${i.name}: ${y.message}`)}}return s>0&&h.info(`[sub-graph cache] evicted ${s} entr(y/ies), freed ${(o/1024/1024).toFixed(1)}MB`),{evicted:s,freedBytes:o,totalBytes:n-o}}catch(t){return h.debug(`[sub-graph cache] evict failed: ${t.message}`),{evicted:0,freedBytes:0}}}var pe=2e3,le=600*1e3,he=new Set(["completed","failed","canceled","timeout"]);function fe(){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 ge(){let e=process.env.PROJECT_ID;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_ID env var.");return e}function ye(){let e=process.env.PROJECT_API_TOKEN;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_API_TOKEN env var.");return e}function be(){return process.env.EXECUTION_ID||null}function V(e,t){return t==null?e:typeof t=="function"?t(e):typeof t=="string"?t.split(".").reduce((r,n)=>r==null?r:r[n],e):e}async function Ce(e,t={}){if(!e||typeof e!="string")throw new Error("dispatchSubgraph: workflowName (string) is required");let r=A(),n=Number(process.env.ZIBBY_SUBGRAPH_MAX_DEPTH||10);if((r.depth||0)>=n)throw new Error(`dispatchSubgraph('${e}'): sub-graph depth ${r.depth} reached cap of ${n}. Restructure the graph or raise ZIBBY_SUBGRAPH_MAX_DEPTH.`);if(process.env.ZIBBY_INPROCESS_SUBGRAPH!=="0"&&!t.async)try{h.debug(`[sub-graph] trying in-process for '${e}'`);let{finalState:u}=await Y(e,{input:t.input,conversationId:t.conversationId,signal:t.signal,parentAgent:t.parentAgent}),c=V(u,t.output);return h.info(`[sub-graph] '${e}' completed in-process`),c}catch(u){if(u instanceof g||u?.fallback)h.info(`[sub-graph] in-process fallback for '${e}': ${u.reason||"unknown"} \u2014 using HTTP`);else throw u}let a=fe(),o=ge(),s=ye(),i=be(),y=`${a}/projects/${encodeURIComponent(o)}/workflows/${encodeURIComponent(e)}/trigger`,b={input:t.input||{},...i?{parentExecutionId:i}:{},...t.conversationId?{conversationId:t.conversationId}:{}};h.info(`[sub-graph] dispatching '${e}' (${t.async?"async":"sync"}) from parent ${i||"<none>"}`);let l=await fetch(y,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${s}`},body:JSON.stringify(b)});if(!l.ok){let u=null,c="";try{u=await l.json(),c=u?.error||u?.message||JSON.stringify(u)}catch{c=await l.text().catch(()=>"")}if(l.status===429){let f=u?.quotaInfo||{},T=new Error(`Sub-graph '${e}' blocked by execution quota (${f.used??"?"}/${f.limit??"?"} on plan ${f.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=e,T.quotaInfo=f,T}if(l.status===400){let f=new Error(`Sub-graph '${e}' rejected input: ${c}`);throw f.code="SUBGRAPH_INVALID_INPUT",f.status=400,f.subgraph=e,f.validationErrors=u?.validationErrors||null,f.missing=u?.missing||null,f}let d=new Error(`Sub-graph '${e}' trigger rejected (${l.status}): ${c}`);throw d.code="SUBGRAPH_TRIGGER_FAILED",d.status=l.status,d.subgraph=e,d}let S=await l.json(),p=S?.data?.jobId||S?.jobId;if(!p)throw new Error(`Sub-graph '${e}' trigger returned no jobId: ${JSON.stringify(S).slice(0,200)}`);if(t.async)return h.info(`[sub-graph] async dispatch of '${e}' \u2192 jobId=${p} (not waiting)`),{jobId:p,status:"accepted",workflow:e};let E=Number.isFinite(t.timeoutMs)?t.timeoutMs:le,w=Number.isFinite(t.pollIntervalMs)?t.pollIntervalMs:pe,$=`${a}/executions/${encodeURIComponent(p)}`,C=Date.now()+E,m="accepted",B=0;for(;Date.now()<C;){await new Promise(f=>setTimeout(f,w)),B+=1;let u=await fetch($,{headers:{Authorization:`Bearer ${s}`}});if(!u.ok){if(u.status>=500){h.warn(`[sub-graph] status poll for ${p} returned ${u.status}, will retry`);continue}throw new Error(`Sub-graph status poll failed for ${p}: ${u.status}`)}let c=await u.json(),d=c?.data||c?.execution||c;if(m=d?.status||m,he.has(m)){if(m!=="completed"){let O=new Error(`Sub-graph '${e}' (${p}) ended in status '${m}'`);throw O.subgraphJobId=p,O.subgraphStatus=m,O}let f=d?.finalState||d?.state||{},T=V(f,t.output);return h.info(`[sub-graph] '${e}' (${p}) completed after ${B} polls`),T}}let I=new Error(`Sub-graph '${e}' (${p}) timed out after ${Math.round(E/1e3)}s (last status: ${m})`);throw I.subgraphJobId=p,I.subgraphStatus=m,I}export{Ce as dispatchSubgraph};
|
|
1
|
+
var D=()=>{},Z={debug:D,info:D,warn:(...e)=>console.warn("[workflow]",...e),error:(...e)=>console.error("[workflow]",...e)},R={impl:Z};var h={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 W,existsSync as A,statSync as N,readdirSync as X,rmSync as ee}from"node:fs";import{spawn as J}from"node:child_process";import{join as $}from"node:path";import{pathToFileURL as te}from"node:url";import{AsyncLocalStorage as ne}from"node:async_hooks";import{AsyncLocalStorage as K}from"node:async_hooks";var k=new K;function x(){let e=k.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,agent:null,signal:null})}function M(e,t){let r=k.getStore()||x(),n=Object.freeze({executionId:e.executionId,parentExecutionId:e.parentExecutionId??r.executionId??null,depth:(r.depth||0)+(e.executionId!==r.executionId?1:0),conversationId:e.conversationId!==void 0?e.conversationId:r.conversationId??null,dispatchMode:e.dispatchMode??null,agent:e.agent!==void 0?e.agent:r.agent??null,signal:e.signal!==void 0?e.signal:r.signal??null});return k.run(n,t)}var U=new Map,j=new Map,L=new Map;function G(e,t,r={}){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");U.set(e,t),j.set(e,"ready"),L.set(e,{...r,cachedAt:Date.now()})}function H(e,t){j.set(e,"failed"),L.set(e,{error:t?.message||String(t),failedAt:Date.now()}),U.delete(e)}function z(e){return j.get(e)==="ready"?U.get(e):null}var P=process.env.ZIBBY_SUBGRAPH_CACHE_DIR||"/tmp/zibby/subgraphs";function re(){return`node${(process.versions?.node||"").split(".")[0]||"unknown"}-${process.platform}-${process.arch}`}var f=class extends Error{constructor(t,r){super(`in-process sub-graph fallback: ${t}${r?` (${r})`:""}`),this.fallback=!0,this.reason=t,this.detail=r||null,this.name="SubgraphFallback"}},F=new ne,q=Promise.resolve();async function oe(e,t){let r=e&&typeof e=="object"&&!Array.isArray(e)?Object.entries(e).filter(([s,i])=>typeof s=="string"&&s&&typeof i=="string"):[];if(r.length===0)return t();let n=F.getStore()===!0,a=null;if(!n){let s=q;q=new Promise(i=>{a=i}),await s}let o=new Map;try{for(let[s,i]of r)o.set(s,Object.prototype.hasOwnProperty.call(process.env,s)?process.env[s]:void 0),process.env[s]=i;return h.debug(`[in-process subgraph] scoped ${r.length} child env var(s)${n?" (nested)":""}`),await F.run(!0,t)}finally{for(let[s,i]of o)i===void 0?delete process.env[s]:process.env[s]=i;a&&a()}}function se(){let e=(process.env.SUBGRAPH_INTERNAL_URL||"").replace(/\/$/,""),t=(process.env.PROGRESS_API_URL||"").replace(/\/executions\/?$/,""),r=e||t,n=process.env.PROJECT_ID,a=process.env.PROJECT_API_TOKEN;if(!r||!n||!a)throw new f("env","SUBGRAPH_INTERNAL_URL/PROGRESS_API_URL/PROJECT_ID/PROJECT_API_TOKEN missing");return{apiBase:r,projectId:n,authToken:a}}async function ae({apiBase:e,authToken:t,body:r}){let n;try{n=await fetch(`${e}/internal/subgraph/begin`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify(r)})}catch(o){throw new f("network",`begin fetch failed: ${o.message}`)}let a=null;try{a=await n.json()}catch{}if(!n.ok){if(n.status===404){let o=new Error(`Sub-graph child '${r.childWorkflowType}' not found in project`);throw o.code="SUBGRAPH_NOT_FOUND",o.status=404,o}if(n.status===429){let o=a?.quotaInfo||{},s=new Error(`Sub-graph blocked by quota (${o.used??"?"}/${o.limit??"?"} on ${o.planId||"plan"})`);throw s.code="SUBGRAPH_QUOTA_EXCEEDED",s.status=429,s.quotaInfo=o,s}if(n.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 f("begin-status",`begin returned ${n.status}`)}return a?.data||a}async function T({apiBase:e,authToken:t,payload:r}){try{let n=await fetch(`${e}/internal/subgraph/finalize`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${t}`},body:JSON.stringify(r)});n.ok||h.warn(`[in-process subgraph] finalize returned ${n.status} for ${r.childExecutionId}`)}catch(n){h.warn(`[in-process subgraph] finalize failed: ${n.message}`)}}async function ie(e,t){let r=$(t,".ready"),n=$(t,"graph.mjs");if(A(r)&&A(n))return;W(t,{recursive:!0});let a=$(t,".lock"),o=!1;try{let{openSync:s,closeSync:i}=await import("node:fs"),y=s(a,"wx");i(y),o=!0}catch(s){if(s.code!=="EEXIST")throw s}if(!o){let s=Date.now()+3e4;for(;Date.now()<s;){if(A(r)&&A(n))return;await new Promise(i=>setTimeout(i,100))}throw new f("bundle-extract-timeout","sibling extract did not complete within 30s")}try{await new Promise((y,b)=>{let p=J("curl",["-fsSL",e],{stdio:["ignore","pipe","inherit"]}),S=J("tar",["-xzf","-","-C",t],{stdio:["pipe","inherit","inherit"]});p.stdout.pipe(S.stdin);let d,E,w=()=>{if(d!==void 0&&E!==void 0){if(d!==0)return b(new Error(`curl exited ${d}`));if(E!==0)return b(new Error(`tar exited ${E}`));y()}};p.on("close",v=>{d=v,w()}),S.on("close",v=>{E=v,w()}),p.on("error",b),S.on("error",b)});let{writeFileSync:s,unlinkSync:i}=await import("node:fs");s(r,"");try{i(a)}catch{}}catch(s){try{let{unlinkSync:i}=await import("node:fs");i(a)}catch{}throw new f("bundle-extract-failed",s.message)}}async function ce(e){let t=$(e,"graph.mjs");if(!A(t))throw new f("entry-missing",`graph.mjs missing under ${e}`);let r;try{r=await import(te(t).href)}catch(a){throw new f("import-failed",`${a?.code||a?.name||"unknown"}: ${a.message}`)}let n=r.default||Object.values(r).find(a=>typeof a=="function"&&a.prototype?.buildGraph);if(!n)throw new f("entry-class-missing","no buildGraph() class export found");return n}async function Y(e,t={}){if(!e||typeof e!="string")throw new Error("runInProcessSubgraph: workflowName (string) is required");let r=x(),n;try{n=se()}catch(c){throw c}h.debug(`[in-process subgraph] begin '${e}' parent=${r.executionId||"<root>"}`);let a=await ae({apiBase:n.apiBase,authToken:n.authToken,body:{parentExecutionId:r.executionId,childWorkflowType:e,input:t.input||{},...t.conversationId?{conversationId:t.conversationId}:{}}}),{childExecutionId:o,runtimeTag:s,bundlePresignedUrl:i,sourcesPresignedUrl:y,workflowVersion:b,workflowUuid:p,bundleReady:S,nodeConfigs:d}=a,E=re();if(s&&s!==E)throw await T({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,discard:!0}}),new f("runtime-mismatch",`${E} vs ${s}`);if(!S||!i)throw await T({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,discard:!0}}),new f("no-bundle","workflow bundle not built yet");let w=z(e);if(!w){let c=$(P,`${p}@${b||"0"}`);try{await ie(i,c);try{le()}catch{}}catch(l){throw l.fallback&&await T({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"failed",error:{message:l.message,code:l.reason}}}),l}try{w=await ce(c),G(e,w,{workflowUuid:p,version:b,runtimeTag:s,cacheDir:c})}catch(l){throw H(e,l),await T({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"failed",error:{message:l.message,code:l.reason||"IMPORT_FAILED"}}}),l.fallback?l:new f("import-failed",l.message)}}let v=Date.now(),C=a.env&&typeof a.env=="object"&&!Array.isArray(a.env)?a.env:null,m=d&&typeof d=="object"&&!Array.isArray(d)&&Object.keys(d).length>0,B={...t.input||{},...m?{nodeConfigs:d}:{}},I,u;try{I=await oe(C,async()=>{let l=await(typeof w=="function"&&w.prototype?.buildGraph?new w:w).buildGraph();return M({executionId:o,parentExecutionId:r.executionId,conversationId:t.conversationId!==void 0?t.conversationId:r.conversationId,dispatchMode:"inprocess"},()=>l.run(t.parentAgent,B,{signal:t.signal}))}),u=I&&typeof I=="object"&&"state"in I?I.state:I}catch(c){throw await T({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"failed",error:{message:c.message,code:c.code||"CHILD_THREW",stack:c.stack},durationMs:Date.now()-v}}),c}if(I&&typeof I=="object"&&I.stoppedExternally){await T({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"canceled",finalState:u,durationMs:Date.now()-v}});let c=new Error(`Sub-graph '${e}' canceled by parent abort`);throw c.code="SUBGRAPH_CANCELED",c.subgraphJobId=o,c}return await T({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:o,status:"completed",finalState:u,durationMs:Date.now()-v}}),{finalState:u,executionId:o}}function ue(e){let t=0,r=[e];for(;r.length;){let n=r.pop(),a;try{a=N(n)}catch{continue}if(a.isDirectory()){let o;try{o=X(n)}catch{continue}for(let s of o)r.push($(n,s))}else t+=a.size}return t}function le({cap:e=Number(process.env.ZIBBY_SUBGRAPH_CACHE_CAP_BYTES||2*1024*1024*1024)}={}){try{if(!A(P))return{evicted:0,freedBytes:0};let t=X(P),r=[],n=0;for(let i of t){let y=$(P,i),b;try{b=N(y)}catch{continue}let p=b.isDirectory()?ue(y):b.size;n+=p,r.push({name:i,full:y,size:p,mtimeMs:b.mtimeMs})}if(n<=e)return{evicted:0,freedBytes:0,totalBytes:n};r.sort((i,y)=>i.mtimeMs-y.mtimeMs);let a=Math.floor(e*.7),o=0,s=0;for(let i of r){if(n-o<=a)break;if(!A($(i.full,".lock")))try{ee(i.full,{recursive:!0,force:!0}),o+=i.size,s+=1}catch(y){h.debug(`[sub-graph cache] evict skip ${i.name}: ${y.message}`)}}return s>0&&h.info(`[sub-graph cache] evicted ${s} entr(y/ies), freed ${(o/1024/1024).toFixed(1)}MB`),{evicted:s,freedBytes:o,totalBytes:n-o}}catch(t){return h.debug(`[sub-graph cache] evict failed: ${t.message}`),{evicted:0,freedBytes:0}}}var de=2e3,pe=600*1e3,he=new Set(["completed","failed","canceled","timeout"]);function ge(){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 fe(){let e=process.env.PROJECT_ID;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_ID env var.");return e}function ye(){let e=process.env.PROJECT_API_TOKEN;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_API_TOKEN env var.");return e}function be(){return process.env.EXECUTION_ID||null}function V(e,t){return t==null?e:typeof t=="function"?t(e):typeof t=="string"?t.split(".").reduce((r,n)=>r==null?r:r[n],e):e}async function Ce(e,t={}){if(!e||typeof e!="string")throw new Error("dispatchSubgraph: workflowName (string) is required");let r=x();t.parentAgent==null&&r.agent&&(t.parentAgent=r.agent),t.signal==null&&r.signal&&(t.signal=r.signal);let n=Number(process.env.ZIBBY_SUBGRAPH_MAX_DEPTH||10);if((r.depth||0)>=n)throw new Error(`dispatchSubgraph('${e}'): sub-graph depth ${r.depth} reached cap of ${n}. Restructure the graph or raise ZIBBY_SUBGRAPH_MAX_DEPTH.`);if(process.env.ZIBBY_INPROCESS_SUBGRAPH!=="0"&&!t.async)try{h.debug(`[sub-graph] trying in-process for '${e}'`);let{finalState:u}=await Y(e,{input:t.input,conversationId:t.conversationId,signal:t.signal,parentAgent:t.parentAgent}),c=V(u,t.output);return h.info(`[sub-graph] '${e}' completed in-process`),c}catch(u){if(u instanceof f||u?.fallback)h.info(`[sub-graph] in-process fallback for '${e}': ${u.reason||"unknown"} \u2014 using HTTP`);else throw u}let a=ge(),o=fe(),s=ye(),i=be(),y=`${a}/projects/${encodeURIComponent(o)}/workflows/${encodeURIComponent(e)}/trigger`,b={input:t.input||{},...i?{parentExecutionId:i}:{},...t.conversationId?{conversationId:t.conversationId}:{}};h.info(`[sub-graph] dispatching '${e}' (${t.async?"async":"sync"}) from parent ${i||"<none>"}`);let p=await fetch(y,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${s}`},body:JSON.stringify(b)});if(!p.ok){let u=null,c="";try{u=await p.json(),c=u?.error||u?.message||JSON.stringify(u)}catch{c=await p.text().catch(()=>"")}if(p.status===429){let g=u?.quotaInfo||{},_=new Error(`Sub-graph '${e}' blocked by execution quota (${g.used??"?"}/${g.limit??"?"} on plan ${g.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=g,_}if(p.status===400){let g=new Error(`Sub-graph '${e}' rejected input: ${c}`);throw g.code="SUBGRAPH_INVALID_INPUT",g.status=400,g.subgraph=e,g.validationErrors=u?.validationErrors||null,g.missing=u?.missing||null,g}let l=new Error(`Sub-graph '${e}' trigger rejected (${p.status}): ${c}`);throw l.code="SUBGRAPH_TRIGGER_FAILED",l.status=p.status,l.subgraph=e,l}let S=await p.json(),d=S?.data?.jobId||S?.jobId;if(!d)throw new Error(`Sub-graph '${e}' trigger returned no jobId: ${JSON.stringify(S).slice(0,200)}`);if(t.async)return h.info(`[sub-graph] async dispatch of '${e}' \u2192 jobId=${d} (not waiting)`),{jobId:d,status:"accepted",workflow:e};let E=Number.isFinite(t.timeoutMs)?t.timeoutMs:pe,w=Number.isFinite(t.pollIntervalMs)?t.pollIntervalMs:de,v=`${a}/executions/${encodeURIComponent(d)}`,C=Date.now()+E,m="accepted",B=0;for(;Date.now()<C;){await new Promise(g=>setTimeout(g,w)),B+=1;let u=await fetch(v,{headers:{Authorization:`Bearer ${s}`}});if(!u.ok){if(u.status>=500){h.warn(`[sub-graph] status poll for ${d} returned ${u.status}, will retry`);continue}throw new Error(`Sub-graph status poll failed for ${d}: ${u.status}`)}let c=await u.json(),l=c?.data||c?.execution||c;if(m=l?.status||m,he.has(m)){if(m!=="completed"){let O=new Error(`Sub-graph '${e}' (${d}) ended in status '${m}'`);throw O.subgraphJobId=d,O.subgraphStatus=m,O}let g=l?.finalState||l?.state||{},_=V(g,t.output);return h.info(`[sub-graph] '${e}' (${d}) completed after ${B} polls`),_}}let I=new Error(`Sub-graph '${e}' (${d}) timed out after ${Math.round(E/1e3)}s (last status: ${m})`);throw I.subgraphJobId=d,I.subgraphStatus=m,I}export{Ce as dispatchSubgraph};
|