@zibby/agent-workflow 0.6.6 → 0.6.9
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/code-generator.js +10 -10
- package/dist/graph-compiler.js +21 -21
- package/dist/graph.js +18 -18
- package/dist/index.d.ts +1 -1
- package/dist/index.js +26 -26
- package/dist/node-registry.js +7 -7
- package/dist/node.js +8 -8
- package/dist/strategy-registry.d.ts +25 -0
- package/dist/strategy-registry.js +6 -6
- package/dist/sub-graph-executor.js +1 -1
- package/package.json +1 -1
package/dist/node.js
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
1
|
-
var
|
|
1
|
+
var ct=Object.defineProperty;var W=(r,t,e)=>()=>{if(e)throw e[0];try{return r&&(t=r(r=0)),t}catch(o){throw e=[o],o}};var ut=(r,t)=>{for(var e in t)ct(r,e,{get:t[e],enumerable:!0})};var J,pt,x,d,R=W(()=>{J=()=>{},pt={debug:J,info:J,warn:(...r)=>console.warn("[workflow]",...r),error:(...r)=>console.error("[workflow]",...r)},x={impl:pt},d={debug:(...r)=>x.impl.debug?.(...r),info:(...r)=>x.impl.info?.(...r),warn:(...r)=>x.impl.warn?.(...r),error:(...r)=>x.impl.error?.(...r)}});var X=W(()=>{});function Q(r){return ht.get(r)||null}var F,j,ht,Tt,tt=W(()=>{F=Symbol.for("@zibby/agent-workflow.skills"),j=Symbol.for("@zibby/agent-workflow.skills.sources");globalThis[F]||(globalThis[F]=new Map);globalThis[j]||(globalThis[j]=new Map);ht=globalThis[F],Tt=globalThis[j]});var st={};ut(st,{getAgentStrategy:()=>ot,invokeAgent:()=>St,listStrategies:()=>gt,registerStrategy:()=>mt,resolveInvocationExtras:()=>rt,resolveInvocationModel:()=>et});function mt(r){if(!r||typeof r.getName!="function"||typeof r.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let t=O.findIndex(e=>e.getName()===r.getName());t>=0?O[t]=r:O.push(r)}function gt(){return O.map(r=>r.getName())}function et({config:r={},options:t={},strategyName:e,envModel:o,nodeConfigModel:i}={}){let s=r.models||{},c=typeof i=="string"&&i.trim()&&i.trim()!=="auto"?i.trim():null,g=t.nodeName&&s[t.nodeName]||null,p=s.default||null,_=r.agent?.[e]?.model||null,k=(typeof o=="string"?o.trim():"")||null;return c||g||p||_||t.model||k||null}function rt({options:r={},stateView:t={},context:e={}}={}){return{workspace:t.workspace||r.workspace,schema:r.schema||e.schema,images:r.images||e.images||[],skills:r.skills||e.skills||[],extraMcpServers:r.extraMcpServers||t.extraMcpServers||e.extraMcpServers||[],plugins:r.plugins||e.plugins||[]}}function ot(r={}){let{state:t={},preferredAgent:e=null}=r,o=e||t.agentType||process.env.AGENT_TYPE;if(!o){let s=O.map(c=>c.getName()).join(", ")||"none registered";throw new Error(`No agent specified. Set agentType in state or AGENT_TYPE env var. Available: ${s}`)}d.debug(`[workflow] agent selection: requested=${o}`);let i=O.find(s=>s.getName()===o);if(!i){let s=O.map(c=>c.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${o}'. Available: ${s}`)}if(!i.canHandle(r))throw new Error(`Agent '${o}' is not available in this environment. Check credentials/environment.`);return d.debug(`[workflow] using agent: ${i.getName()}`),i}async function St(r,t={},e={}){let o=t.state&&typeof t.state.getAll=="function"?t.state.getAll():t.state||{},i={...t,state:o},s=ot(i),c=o.config||e.config||{},g=o._currentNodeConfig||{},p=typeof g.agent=="string"?g.agent.trim():"",_=!p||p===s.name?g.model:null,k=et({config:c,options:e,strategyName:s.name,envModel:process.env.MODEL,nodeConfigModel:_}),n={...e,model:k,...rt({options:e,stateView:o,context:t}),config:c},h=r,y=n.skills||[];if(y.length>0&&!e.skipPromptFragments){let w=e.connectedIntegrations;if(!w){let a=process.env.WORKFLOW_CONNECTED_INTEGRATIONS;if(typeof a=="string"&&a.trim()!==""){w={};for(let m of a.split(",").map($=>$.trim()).filter(Boolean))w[m]=!0}}let b=a=>{let m=a&&a.requiresIntegration;return!m||!w?!0:(Array.isArray(m)?m:[m]).some(A=>w[A]===!0)},l=s.name==="assistant",u=a=>a?.inProcessOnly!==!0||l,S=y.map(a=>{let m=Q(a);if(!b(m)||!u(m))return null;let $=m?.promptFragment;return typeof $=="function"?$():$}).filter(Boolean);S.length>0&&(h+=`
|
|
2
2
|
|
|
3
3
|
${S.join(`
|
|
4
4
|
|
|
5
|
-
`)}`)}let E=o._currentNodeConfig?.stores;if(Array.isArray(E)&&E.length>0&&typeof E[0]=="object"){let
|
|
5
|
+
`)}`)}let E=o._currentNodeConfig?.stores;if(Array.isArray(E)&&E.length>0&&typeof E[0]=="object"){let w=E.length<=8,b=E.map(l=>{let u=l?.id??l?.storeId??"",S=(l?.name??"").toString().trim()||u,a=l?.type?` \xB7 ${l.type}`:"",m=(l?.description||"").toString().replace(/\s+/g," ").trim(),$=`- ${S} \xB7 ${m||"(no description)"}${a} (id: ${u})`;if(w&&l?.schema&&typeof l.schema=="object"){let A=l.schema.properties&&typeof l.schema.properties=="object"?Object.keys(l.schema.properties):Object.keys(l.schema);A.length&&($+=`
|
|
6
6
|
fields: ${A.join(", ")}`)}return $});h+=`
|
|
7
7
|
|
|
8
8
|
AVAILABLE STORES (pick a store by its description and pass its NAME to the store tool):
|
|
@@ -14,14 +14,14 @@ PRIORITY OVERRIDE \u2014 THE FOLLOWING INSTRUCTIONS TAKE PRECEDENCE OVER ALL PRE
|
|
|
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
16
|
${N}
|
|
17
|
-
`),d.debug(`[workflow] prompt length: ${h.length} chars`),s.invoke(h,n)}var D,O,
|
|
17
|
+
`),d.debug(`[workflow] prompt length: ${h.length} chars`),s.invoke(h,n)}var D,O,it=W(()=>{X();R();tt();D=Symbol.for("@zibby/agent-workflow.strategies");globalThis[D]||(globalThis[D]=[]);O=globalThis[D]});import I from"handlebars";var v=class{constructor(t){this.schema=t}parse(t){let e=t.match(/```json\s*([\s\S]*?)\s*```/);if(e)return this.validate(JSON.parse(e[1]));let o=[t.match(/\{[\s\S]*?\}/),t.match(/\{[\s\S]*\}/)].filter(Boolean).map(i=>i[0]);for(let i of o)try{return this.validate(JSON.parse(i))}catch(s){if(!(s instanceof SyntaxError))throw s}return this.validate({result:t.trim()})}validate(t){let e=[];for(let[o,i]of Object.entries(this.schema)){if(i.required&&!(o in t)&&e.push(`Missing required field: ${o}`),o in t&&i.type){let s=typeof t[o];s!==i.type&&e.push(`Field '${o}' expected ${i.type}, got ${s}`)}if(i.validate&&o in t){let s=i.validate(t[o]);s&&e.push(`Field '${o}': ${s}`)}}if(e.length>0)throw new Error(`Output validation failed:
|
|
18
18
|
${e.join(`
|
|
19
|
-
`)}`);return t}};R();import{writeFileSync as G,readFileSync as
|
|
19
|
+
`)}`);return t}};R();import{writeFileSync as G,readFileSync as nt,existsSync as at,mkdirSync as _t}from"node:fs";import{join as H,dirname as yt}from"node:path";import f from"chalk";var dt="__WORKFLOW_GRAPH_LOG__",T=f.gray("\u2502"),ft=f.gray("\u250C"),U=f.gray("\u2514"),P=f.green("\u25C6"),B=f.hex("#c084fc")("\u25C6"),K=f.hex("#2dd4bf")("\u25C6"),L=f.red("\u25C6"),Y=`${T} `,Z=2;function z(r){return r<1e3?`${r}ms`:`${(r/1e3).toFixed(1)}s`}function q(r,t){return(e,o,i)=>{if(typeof e!="string")return r(e,o,i);let s=process.stdout.columns||120,c="";for(let g=0;g<e.length;g++){let p=e[g];t.lineStart&&(c+=Y,t.col=Z,t.lineStart=!1),p===`
|
|
20
20
|
`?(c+=p,t.lineStart=!0,t.col=0,t.inEsc=!1):p==="\x1B"?(t.inEsc=!0,c+=p):t.inEsc?(c+=p,(p>="A"&&p<="Z"||p>="a"&&p<="z")&&(t.inEsc=!1)):(t.col++,c+=p,t.col>=s&&(c+=`
|
|
21
|
-
${Y}`,t.col=Z))}return r(c,o,i)}}var M=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=
|
|
21
|
+
${Y}`,t.col=Z))}return r(c,o,i)}}var M=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=q(this._origStdoutWrite,t),process.stderr.write=q(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=`${dt}${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}
|
|
@@ -29,6 +29,6 @@ ${Y}`,t.col=Z))}return r(c,o,i)}}var M=class{constructor(){this._currentNode=nul
|
|
|
29
29
|
`)}stepInfo(t){this.step(t)}stepTool(t){this._origStdoutWrite?this._writeDot(B,t):process.stdout.write.bind(process.stdout)(`${T} ${B} ${t}
|
|
30
30
|
`)}stepMemory(t){let e=f.hex("#2dd4bf")(t);this._origStdoutWrite?this._writeDot(K,e):process.stdout.write.bind(process.stdout)(`${T} ${K} ${e}
|
|
31
31
|
`)}stepFail(t){this._origStdoutWrite?this._writeDot(L,f.red(t)):process.stdout.write.bind(process.stdout)(`${T} ${L} ${f.red(t)}
|
|
32
|
-
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${
|
|
32
|
+
`)}nodeStart(t){this._currentNode=t,this._emitGraphLogMarker({phase:"node_begin",node:t}),this._rawWrite(`${ft} ${t}`),this._startIntercepting()}nodeComplete(t,e={}){this._stopIntercepting();let{duration:o,details:i}=e;if(i)for(let c of i)this._rawWrite(`${P} ${c}`);let s=o?f.dim(` ${z(o)}`):"";this._rawWrite(`${U} ${f.green("done")}${s}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}nodeFailed(t,e,o={}){this._stopIntercepting();let{duration:i}=o,s=i?f.dim(` ${z(i)}`):"";this._rawWrite(`${L} ${f.red(e)}`),this._rawWrite(`${U} ${f.red("failed")}${s}`),this._emitGraphLogMarker({phase:"node_end",node:t}),this._rawWrite("")}route(t,e){this._rawWrite(f.dim(` ${t} \u2192 ${e}`)),this._rawWrite("")}graphComplete(){}},V=new M;var C=".session-info.json";var It=Object.freeze(["codebase-memory","code-scan","artifact"]);I.helpers.inc||I.registerHelper("inc",r=>Number(r)+1);I.helpers.json||I.registerHelper("json",r=>JSON.stringify(r,null,2));I.helpers.eq||I.registerHelper("eq",(r,t)=>r===t);var lt=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 v(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 d.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,i=n=>e&&typeof e.get=="function"?e.get(n):t?.[n];if(typeof this.customExecute=="function"){d.debug(`[workflow] node '${this.name}': custom execute (skipping LLM)`);try{let n=await this.customExecute(t);return typeof n=="object"&&n!==null&&n.success===!1?{success:!1,error:n.error||"Node execution failed",raw:n.raw||null}:this.isZodSchema?(d.debug(`[workflow] node '${this.name}': validating output schema`),{success:!0,output:this.outputSchema.parse(n),raw:null}):{success:!0,output:n,raw:null}}catch(n){return d.error(`[workflow] node '${this.name}' failed: ${n.message}`),n.name==="ZodError"&&d.error(`Schema errors: ${JSON.stringify(n.issues||n.errors,null,2)}`),{success:!1,error:n.message,raw:null}}}let s;typeof this.prompt=="function"?s=this.prompt(o()):typeof this.prompt=="string"&&this.prompt.includes("{{")?(this._compiledPrompt||(this._compiledPrompt=I.compile(this.prompt,{noEscape:!0})),s=this._compiledPrompt(o())):s=this.prompt;let c=i("_skillHints");c&&(s=`${c}
|
|
33
33
|
|
|
34
|
-
${s}`);let g=o(),p=g.cwd||process.cwd(),_=g.sessionPath;try{if(_){let n=H(_,C);if(
|
|
34
|
+
${s}`);let g=o(),p=g.cwd||process.cwd(),_=g.sessionPath;try{if(_){let n=H(_,C);if(at(n)){let y=JSON.parse(nt(n,"utf-8"));y.currentNode=this.name,G(n,JSON.stringify(y,null,2),"utf-8")}let h=H(_,"..",C);if(at(h))try{let y=JSON.parse(nt(h,"utf-8"));y.currentNode=this.name,G(h,JSON.stringify(y,null,2),"utf-8")}catch{}}}catch(n){d.debug(`[workflow] could not update session info: ${n.message}`)}let k=null;for(let n=0;n<=this.retries;n++)try{d.debug(`[workflow] node '${this.name}' attempt ${n}`);let h=o().config||{},y=h.agents||{},E=this.config.agent??y[this.name]??null,N={state:o()};E&&(N.preferredAgent=E);let w={workspace:p,schema:this.isZodSchema?this.outputSchema:null,skills:this.config.skills||[],disallowedTools:this.config.disallowedTools||[],plugins:this.config.plugins||[],sessionPath:_,config:h,nodeName:this.name,timeout:this.config?.timeout||3e5},b=t?._coreInvokeAgent;b||(b=(await Promise.resolve().then(()=>(it(),st))).invokeAgent);let l=await b(s,N,w),u,S;if(typeof l=="string"?(u=l,S=null):l.structured?(u=l.raw||JSON.stringify(l.structured,null,2),S=l.structured):(u=l.raw||JSON.stringify(l,null,2),S=l.extracted||null),_)try{let a=H(_,this.name,"raw_stream_output.txt");_t(yt(a),{recursive:!0}),G(a,typeof u=="string"?u:JSON.stringify(u),"utf-8")}catch(a){d.debug(`[workflow] could not save raw output: ${a.message}`)}if(this.isZodSchema&&S){d.info(`[workflow] node '${this.name}': output validated: ${JSON.stringify(S,null,2)}`);let a=S;if(typeof this.onComplete=="function")try{a=await this.onComplete(o(),S)}catch(m){d.warn(`[workflow] onComplete hook failed: ${m.message}`)}return{success:!0,output:a,raw:u}}if(typeof this.onComplete=="function")try{return{success:!0,output:await this.onComplete(o(),{raw:u}),raw:u}}catch(a){throw new Error(`onComplete failed: ${a.message}`,{cause:a})}if(this.parser){let a=this.parser.parse(u);return d.info(`[workflow] node '${this.name}': parsed output: ${JSON.stringify(a,null,2)}`),V.step("Output parsed"),{success:!0,output:a,raw:u}}return{success:!0,output:u,raw:u}}catch(h){k=h,n<this.retries&&d.info(`[workflow] node '${this.name}' failed, retrying (${n+1}/${this.retries})\u2026`)}return{success:!1,error:k.message,raw:null}}};export{lt as Node};
|
|
@@ -51,6 +51,31 @@ export declare function listStrategies(): any;
|
|
|
51
51
|
* config is one more chance to drift; this wires the last container-side one.
|
|
52
52
|
*/
|
|
53
53
|
export declare function resolveInvocationModel({ config, options, strategyName, envModel, nodeConfigModel }?: any): any;
|
|
54
|
+
/**
|
|
55
|
+
* The per-invocation PASSTHROUGH options — the fields a strategy needs that are
|
|
56
|
+
* sourced from (in order) the explicit per-call options, the run state, then the
|
|
57
|
+
* node context. ONE implementation, imported by every invokeAgent.
|
|
58
|
+
*
|
|
59
|
+
* WHY IT IS A SHARED FUNCTION: this package's invokeAgent (the engine path) and
|
|
60
|
+
* @zibby/core's invokeAgent (what a template CODE NODE imports) each built this
|
|
61
|
+
* object by hand, and the two lists drifted — core's never carried
|
|
62
|
+
* `extraMcpServers` at all, so an agent's custom MCP servers reached prompt
|
|
63
|
+
* nodes and silently vanished for every code node. Found live 2026-08-18: a
|
|
64
|
+
* browser MCP server attached to frontend-specialist, resolved by the executor,
|
|
65
|
+
* delivered into the run container, parsed by the CLI onto
|
|
66
|
+
* state.extraMcpServers — and then dropped one line before the strategy, so the
|
|
67
|
+
* QA node reported "no browser tools". Same executor-drift class as the model
|
|
68
|
+
* chain (resolveInvocationModel, above), same cure: derive, don't duplicate.
|
|
69
|
+
* A field added here now reaches BOTH paths by construction.
|
|
70
|
+
*/
|
|
71
|
+
export declare function resolveInvocationExtras({ options, stateView, context }?: any): {
|
|
72
|
+
workspace: any;
|
|
73
|
+
schema: any;
|
|
74
|
+
images: any;
|
|
75
|
+
skills: any;
|
|
76
|
+
extraMcpServers: any;
|
|
77
|
+
plugins: any;
|
|
78
|
+
};
|
|
54
79
|
export declare function getAgentStrategy(context?: any): any;
|
|
55
80
|
/**
|
|
56
81
|
* Invoke an agent with automatic strategy selection.
|
|
@@ -1,17 +1,17 @@
|
|
|
1
|
-
var
|
|
1
|
+
var _=()=>{},M={debug:_,info:_,warn:(...e)=>console.warn("[workflow]",...e),error:(...e)=>console.error("[workflow]",...e)},k={impl:M};var S={debug:(...e)=>k.impl.debug?.(...e),info:(...e)=>k.impl.info?.(...e),warn:(...e)=>k.impl.warn?.(...e),error:(...e)=>k.impl.error?.(...e)};var T=Symbol.for("@zibby/agent-workflow.skills"),A=Symbol.for("@zibby/agent-workflow.skills.sources");globalThis[T]||(globalThis[T]=new Map);globalThis[A]||(globalThis[A]=new Map);var P=globalThis[T],D=globalThis[A];function R(e){return P.get(e)||null}var I=Symbol.for("@zibby/agent-workflow.strategies");globalThis[I]||(globalThis[I]=[]);var m=globalThis[I];function U(e){if(!e||typeof e.getName!="function"||typeof e.invoke!="function")throw new Error("strategy must implement getName() and invoke() (AgentStrategy shape)");let t=m.findIndex(n=>n.getName()===e.getName());t>=0?m[t]=e:m.push(e)}function V(){return m.map(e=>e.getName())}function j({config:e={},options:t={},strategyName:n,envModel:r,nodeConfigModel:l}={}){let i=e.models||{},g=typeof l=="string"&&l.trim()&&l.trim()!=="auto"?l.trim():null,f=t.nodeName&&i[t.nodeName]||null,y=i.default||null,E=e.agent?.[n]?.model||null,v=(typeof r=="string"?r.trim():"")||null;return g||f||y||E||t.model||v||null}function L({options:e={},stateView:t={},context:n={}}={}){return{workspace:t.workspace||e.workspace,schema:e.schema||n.schema,images:e.images||n.images||[],skills:e.skills||n.skills||[],extraMcpServers:e.extraMcpServers||t.extraMcpServers||n.extraMcpServers||[],plugins:e.plugins||n.plugins||[]}}function C(e={}){let{state:t={},preferredAgent:n=null}=e,r=n||t.agentType||process.env.AGENT_TYPE;if(!r){let i=m.map(g=>g.getName()).join(", ")||"none registered";throw new Error(`No agent specified. Set agentType in state or AGENT_TYPE env var. Available: ${i}`)}S.debug(`[workflow] agent selection: requested=${r}`);let l=m.find(i=>i.getName()===r);if(!l){let i=m.map(g=>g.getName()).join(", ")||"none registered";throw new Error(`Unknown agent '${r}'. Available: ${i}`)}if(!l.canHandle(e))throw new Error(`Agent '${r}' is not available in this environment. Check credentials/environment.`);return S.debug(`[workflow] using agent: ${l.getName()}`),l}async function B(e,t={},n={}){let r=t.state&&typeof t.state.getAll=="function"?t.state.getAll():t.state||{},l={...t,state:r},i=C(l),g=r.config||n.config||{},f=r._currentNodeConfig||{},y=typeof f.agent=="string"?f.agent.trim():"",E=!y||y===i.name?f.model:null,v=j({config:g,options:n,strategyName:i.name,envModel:process.env.MODEL,nodeConfigModel:E}),O={...n,model:v,...L({options:n,stateView:r,context:t}),config:g},u=e,$=O.skills||[];if($.length>0&&!n.skipPromptFragments){let p=n.connectedIntegrations;if(!p){let s=process.env.WORKFLOW_CONNECTED_INTEGRATIONS;if(typeof s=="string"&&s.trim()!==""){p={};for(let a of s.split(",").map(c=>c.trim()).filter(Boolean))p[a]=!0}}let N=s=>{let a=s&&s.requiresIntegration;return!a||!p?!0:(Array.isArray(a)?a:[a]).some(b=>p[b]===!0)},o=i.name==="assistant",h=s=>s?.inProcessOnly!==!0||o,w=$.map(s=>{let a=R(s);if(!N(a)||!h(a))return null;let c=a?.promptFragment;return typeof c=="function"?c():c}).filter(Boolean);w.length>0&&(u+=`
|
|
2
2
|
|
|
3
3
|
${w.join(`
|
|
4
4
|
|
|
5
|
-
`)}`)}let d=r._currentNodeConfig?.stores;if(Array.isArray(d)&&d.length>0&&typeof d[0]=="object"){let
|
|
5
|
+
`)}`)}let d=r._currentNodeConfig?.stores;if(Array.isArray(d)&&d.length>0&&typeof d[0]=="object"){let p=d.length<=8,N=d.map(o=>{let h=o?.id??o?.storeId??"",w=(o?.name??"").toString().trim()||h,s=o?.type?` \xB7 ${o.type}`:"",a=(o?.description||"").toString().replace(/\s+/g," ").trim(),c=`- ${w} \xB7 ${a||"(no description)"}${s} (id: ${h})`;if(p&&o?.schema&&typeof o.schema=="object"){let b=o.schema.properties&&typeof o.schema.properties=="object"?Object.keys(o.schema.properties):Object.keys(o.schema);b.length&&(c+=`
|
|
6
6
|
fields: ${b.join(", ")}`)}return c});u+=`
|
|
7
7
|
|
|
8
8
|
AVAILABLE STORES (pick a store by its description and pass its NAME to the store tool):
|
|
9
|
-
${
|
|
10
|
-
`)}`}let
|
|
9
|
+
${N.join(`
|
|
10
|
+
`)}`}let x=r._currentNodeConfig?.extraPromptInstructions?.trim();return x&&(u+=`
|
|
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
|
-
`),S.debug(`[workflow] prompt length: ${u.length} chars`),i.invoke(u,O)}export{
|
|
16
|
+
${x}
|
|
17
|
+
`),S.debug(`[workflow] prompt length: ${u.length} chars`),i.invoke(u,O)}export{C as getAgentStrategy,B as invokeAgent,V as listStrategies,U as registerStrategy,L as resolveInvocationExtras,j 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 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};
|
|
1
|
+
var M=()=>{},K={debug:M,info:M,warn:(...e)=>console.warn("[workflow]",...e),error:(...e)=>console.error("[workflow]",...e)},R={impl:K};var g={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 ee,existsSync as x,statSync as X,readdirSync as Y,rmSync as te}from"node:fs";import{spawn as F}from"node:child_process";import{join as T}from"node:path";import{pathToFileURL as ne}from"node:url";import{AsyncLocalStorage as re}from"node:async_hooks";import{AsyncLocalStorage as Q}from"node:async_hooks";var U=new Q;function B(){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,agent:null,signal:null})}function L(e,t){let r=U.getStore()||B(),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 U.run(n,t)}var j=new Map,D=new Map,G=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");j.set(e,t),D.set(e,"ready"),G.set(e,{...r,cachedAt:Date.now()})}function z(e,t){D.set(e,"failed"),G.set(e,{error:t?.message||String(t),failedAt:Date.now()}),j.delete(e)}function J(e){return D.get(e)==="ready"?j.get(e):null}var C=process.env.ZIBBY_SUBGRAPH_CACHE_DIR||"/tmp/zibby/subgraphs";function se(){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"}},q=new re,N=Promise.resolve();async function ae(e,t){let r=e&&typeof e=="object"&&!Array.isArray(e)?Object.entries(e).filter(([a,c])=>typeof a=="string"&&a&&typeof c=="string"):[];if(r.length===0)return t();let n=q.getStore()===!0,o=null;if(!n){let a=N;N=new Promise(c=>{o=c}),await a}let s=new Map;try{for(let[a,c]of r)s.set(a,Object.prototype.hasOwnProperty.call(process.env,a)?process.env[a]:void 0),process.env[a]=c;return g.debug(`[in-process subgraph] scoped ${r.length} child env var(s)${n?" (nested)":""}`),await q.run(!0,t)}finally{for(let[a,c]of s)c===void 0?delete process.env[a]:process.env[a]=c;o&&o()}}function oe(){let e=(process.env.SUBGRAPH_INTERNAL_URL||"").replace(/\/$/,""),t=(process.env.PROGRESS_API_URL||"").replace(/\/executions\/?$/,""),r=e||t,n=process.env.PROJECT_ID,o=process.env.PROJECT_API_TOKEN;if(!r||!n||!o)throw new f("env","SUBGRAPH_INTERNAL_URL/PROGRESS_API_URL/PROJECT_ID/PROJECT_API_TOKEN missing");return{apiBase:r,projectId:n,authToken:o}}async function ie({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(s){throw new f("network",`begin fetch failed: ${s.message}`)}let o=null;try{o=await n.json()}catch{}if(!n.ok){if(n.status===404){let s=new Error(`Sub-graph child '${r.childWorkflowType}' not found in project`);throw s.code="SUBGRAPH_NOT_FOUND",s.status=404,s}if(n.status===429){let s=o?.quotaInfo||{},a=new Error(`Sub-graph blocked by quota (${s.used??"?"}/${s.limit??"?"} on ${s.planId||"plan"})`);throw a.code="SUBGRAPH_QUOTA_EXCEEDED",a.status=429,a.quotaInfo=s,a}if(n.status===400&&o?.validationErrors){let s=new Error(`Sub-graph rejected input: ${o?.error||o?.message||"validation failed"}`);throw s.code="SUBGRAPH_INVALID_INPUT",s.status=400,s.validationErrors=o.validationErrors,s.missing=o.missing,s}throw new f("begin-status",`begin returned ${n.status}`)}return o?.data||o}async function A({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||g.warn(`[in-process subgraph] finalize returned ${n.status} for ${r.childExecutionId}`)}catch(n){g.warn(`[in-process subgraph] finalize failed: ${n.message}`)}}async function ce(e,t){let r=T(t,".ready"),n=T(t,"graph.mjs");if(x(r)&&x(n))return;ee(t,{recursive:!0});let o=T(t,".lock"),s=!1;try{let{openSync:a,closeSync:c}=await import("node:fs"),y=a(o,"wx");c(y),s=!0}catch(a){if(a.code!=="EEXIST")throw a}if(!s){let a=Date.now()+3e4;for(;Date.now()<a;){if(x(r)&&x(n))return;await new Promise(c=>setTimeout(c,100))}throw new f("bundle-extract-timeout","sibling extract did not complete within 30s")}try{await new Promise((y,b)=>{let p=F("curl",["-fsSL",e],{stdio:["ignore","pipe","inherit"]}),m=F("tar",["-xzf","-","-C",t],{stdio:["pipe","inherit","inherit"]});p.stdout.pipe(m.stdin);let l,I,w=()=>{if(l!==void 0&&I!==void 0){if(l!==0)return b(new Error(`curl exited ${l}`));if(I!==0)return b(new Error(`tar exited ${I}`));y()}};p.on("close",S=>{l=S,w()}),m.on("close",S=>{I=S,w()}),p.on("error",b),m.on("error",b)});let{writeFileSync:a,unlinkSync:c}=await import("node:fs");a(r,"");try{c(o)}catch{}}catch(a){try{let{unlinkSync:c}=await import("node:fs");c(o)}catch{}throw new f("bundle-extract-failed",a.message)}}async function ue(e){let t=T(e,"graph.mjs");if(!x(t))throw new f("entry-missing",`graph.mjs missing under ${e}`);let r;try{r=await import(ne(t).href)}catch(o){throw new f("import-failed",`${o?.code||o?.name||"unknown"}: ${o.message}`)}let n=r.default||Object.values(r).find(o=>typeof o=="function"&&o.prototype?.buildGraph);if(!n)throw new f("entry-class-missing","no buildGraph() class export found");return n}async function V(e,t={}){if(!e||typeof e!="string")throw new Error("runInProcessSubgraph: workflowName (string) is required");let r=B(),n;try{n=oe()}catch(i){throw i}g.debug(`[in-process subgraph] begin '${e}' parent=${r.executionId||"<root>"}`);let o=await ie({apiBase:n.apiBase,authToken:n.authToken,body:{parentExecutionId:r.executionId,childWorkflowType:e,input:t.input||{},...t.conversationId?{conversationId:t.conversationId}:{}}}),{childExecutionId:s,runtimeTag:a,bundlePresignedUrl:c,sourcesPresignedUrl:y,workflowVersion:b,workflowUuid:p,bundleReady:m,nodeConfigs:l}=o,I=se();if(a&&a!==I)throw await A({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,discard:!0}}),new f("runtime-mismatch",`${I} vs ${a}`);if(!m||!c)throw await A({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,discard:!0}}),new f("no-bundle","workflow bundle not built yet");let w=J(e);if(!w){let i=T(C,`${p}@${b||"0"}`);try{await ce(c,i);try{de()}catch{}}catch(u){throw u.fallback&&await A({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"failed",error:{message:u.message,code:u.reason}}}),u}try{w=await ue(i),H(e,w,{workflowUuid:p,version:b,runtimeTag:a,cacheDir:i})}catch(u){throw z(e,u),await A({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"failed",error:{message:u.message,code:u.reason||"IMPORT_FAILED"}}}),u.fallback?u:new f("import-failed",u.message)}}let S=Date.now(),O=o.env&&typeof o.env=="object"&&!Array.isArray(o.env)?o.env:null,E=l&&typeof l=="object"&&!Array.isArray(l)&&Object.keys(l).length>0,P={...t.input||{},...E?{nodeConfigs:l}:{}},h,$;try{h=await ae(O,async()=>{let u=await(typeof w=="function"&&w.prototype?.buildGraph?new w:w).buildGraph();return L({executionId:s,parentExecutionId:r.executionId,conversationId:t.conversationId!==void 0?t.conversationId:r.conversationId,dispatchMode:"inprocess"},()=>u.run(t.parentAgent,P,{signal:t.signal}))}),$=h&&typeof h=="object"&&"state"in h?h.state:h}catch(i){throw await A({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"failed",error:{message:i.message,code:i.code||"CHILD_THREW",stack:i.stack},durationMs:Date.now()-S}}),i}if(h&&typeof h=="object"&&h.stoppedExternally){await A({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"canceled",finalState:$,durationMs:Date.now()-S}});let i=new Error(`Sub-graph '${e}' canceled by parent abort`);throw i.code="SUBGRAPH_CANCELED",i.subgraphJobId=s,i}return await A({apiBase:n.apiBase,authToken:n.authToken,payload:{childExecutionId:s,status:"completed",finalState:$,durationMs:Date.now()-S}}),{finalState:$,executionId:s}}function le(e){let t=0,r=[e];for(;r.length;){let n=r.pop(),o;try{o=X(n)}catch{continue}if(o.isDirectory()){let s;try{s=Y(n)}catch{continue}for(let a of s)r.push(T(n,a))}else t+=o.size}return t}function de({cap:e=Number(process.env.ZIBBY_SUBGRAPH_CACHE_CAP_BYTES||2*1024*1024*1024)}={}){try{if(!x(C))return{evicted:0,freedBytes:0};let t=Y(C),r=[],n=0;for(let c of t){let y=T(C,c),b;try{b=X(y)}catch{continue}let p=b.isDirectory()?le(y):b.size;n+=p,r.push({name:c,full:y,size:p,mtimeMs:b.mtimeMs})}if(n<=e)return{evicted:0,freedBytes:0,totalBytes:n};r.sort((c,y)=>c.mtimeMs-y.mtimeMs);let o=Math.floor(e*.7),s=0,a=0;for(let c of r){if(n-s<=o)break;if(!x(T(c.full,".lock")))try{te(c.full,{recursive:!0,force:!0}),s+=c.size,a+=1}catch(y){g.debug(`[sub-graph cache] evict skip ${c.name}: ${y.message}`)}}return a>0&&g.info(`[sub-graph cache] evicted ${a} entr(y/ies), freed ${(s/1024/1024).toFixed(1)}MB`),{evicted:a,freedBytes:s,totalBytes:n-s}}catch(t){return g.debug(`[sub-graph cache] evict failed: ${t.message}`),{evicted:0,freedBytes:0}}}var pe=2e3,he=600*1e3,ge=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 ye(){let e=process.env.PROJECT_ID;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_ID env var.");return e}function be(){let e=process.env.PROJECT_API_TOKEN;if(!e)throw new Error("Sub-graph dispatch requires PROJECT_API_TOKEN env var.");return e}function we(){return process.env.EXECUTION_ID||null}function Z(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 Oe(e,t={}){if(!e||typeof e!="string")throw new Error("dispatchSubgraph: workflowName (string) is required");let r=B();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{g.debug(`[sub-graph] trying in-process for '${e}'`);let{finalState:i}=await V(e,{input:t.input,conversationId:t.conversationId,signal:t.signal,parentAgent:t.parentAgent}),u=Z(i,t.output);return g.info(`[sub-graph] '${e}' completed in-process`),u}catch(i){if(i instanceof f||i?.fallback)g.info(`[sub-graph] in-process fallback for '${e}': ${i.reason||"unknown"} \u2014 using HTTP`);else throw i}let o=fe(),s=ye(),a=be(),c=we(),y=`${o}/projects/${encodeURIComponent(s)}/workflows/${encodeURIComponent(e)}/trigger`,b={input:t.input||{},...c?{parentExecutionId:c}:{},...t.conversationId?{conversationId:t.conversationId}:{}};g.info(`[sub-graph] dispatching '${e}' (${t.async?"async":"sync"}) from parent ${c||"<none>"}`);let p=await fetch(y,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${a}`},body:JSON.stringify(b)});if(!p.ok){let i=null,u="";try{i=await p.json(),u=i?.error||i?.message||JSON.stringify(i)}catch{u=await p.text().catch(()=>"")}if(p.status===429){let d=i?.quotaInfo||{},_=new Error(`Sub-graph '${e}' blocked by execution quota (${d.used??"?"}/${d.limit??"?"} on plan ${d.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=d,_}if(p.status===400){let d=new Error(`Sub-graph '${e}' rejected input: ${u}`);throw d.code="SUBGRAPH_INVALID_INPUT",d.status=400,d.subgraph=e,d.validationErrors=i?.validationErrors||null,d.missing=i?.missing||null,d}let v=new Error(`Sub-graph '${e}' trigger rejected (${p.status}): ${u}`);throw v.code="SUBGRAPH_TRIGGER_FAILED",v.status=p.status,v.subgraph=e,v}let m=await p.json(),l=m?.data?.jobId||m?.jobId;if(!l)throw new Error(`Sub-graph '${e}' trigger returned no jobId: ${JSON.stringify(m).slice(0,200)}`);if(t.async)return g.info(`[sub-graph] async dispatch of '${e}' \u2192 jobId=${l} (not waiting)`),{jobId:l,status:"accepted",workflow:e};let I=Number.isFinite(t.timeoutMs)?t.timeoutMs:he,w=Number.isFinite(t.pollIntervalMs)?t.pollIntervalMs:pe,S=`${o}/executions/${encodeURIComponent(l)}`,O=Date.now()+I,E="accepted",P=0,h=null;for(;Date.now()<O;){await new Promise(d=>setTimeout(d,w)),P+=1;let i;try{i=await fetch(S,{headers:{Authorization:`Bearer ${a}`}})}catch(d){h=d?.message||String(d),g.warn(`[sub-graph] status poll for ${l} could not reach the API (${h}), will retry`);continue}if(!i.ok){if(i.status>=500){g.warn(`[sub-graph] status poll for ${l} returned ${i.status}, will retry`);continue}throw new Error(`Sub-graph status poll failed for ${l}: ${i.status}`)}let u;try{u=await i.json()}catch(d){h=d?.message||String(d),g.warn(`[sub-graph] status poll for ${l} returned an unreadable body (${h}), will retry`);continue}let v=u?.data||u?.execution||u;if(E=v?.status||E,ge.has(E)){if(E!=="completed"){let k=new Error(`Sub-graph '${e}' (${l}) ended in status '${E}'`);throw k.subgraphJobId=l,k.subgraphStatus=E,k}let d=v?.finalState||v?.state||{},_=Z(d,t.output);return g.info(`[sub-graph] '${e}' (${l}) completed after ${P} polls`),_}}let $=new Error(`Sub-graph '${e}' (${l}) timed out after ${Math.round(I/1e3)}s (last status: ${E})`+(h?`; the status API was unreachable on the last attempt (${h})`:""));throw $.subgraphJobId=l,$.subgraphStatus=E,h&&($.subgraphTransportError=h),$}export{Oe as dispatchSubgraph};
|