@memgrafter/flatagents 0.9.0 → 2.0.0

package/dist/index.js

@@ -35,6 +35,7 @@ __export(index_exports, {
   DefaultExecution: () => DefaultExecution,
   FlatAgent: () => FlatAgent,
   FlatMachine: () => FlatMachine,
+  HooksRegistry: () => HooksRegistry,
   LocalFileBackend: () => LocalFileBackend,
   LocalFileLock: () => LocalFileLock,
   MCPToolProvider: () => MCPToolProvider,
@@ -1006,6 +1007,179 @@ var LocalFileLock = class {
   }
 };
 
+// src/hooks.ts
+var WebhookHooks = class {
+  constructor(url) {
+    this.url = url;
+  }
+  async send(event, data) {
+    try {
+      const body = JSON.stringify({ event, ...data, timestamp: (/* @__PURE__ */ new Date()).toISOString() }, (key, value) => {
+        if (typeof value === "object" && value !== null) {
+          const seen = /* @__PURE__ */ new WeakSet();
+          return JSON.parse(JSON.stringify(value, (k, v) => {
+            if (typeof v === "object" && v !== null) {
+              if (seen.has(v)) return "[Circular]";
+              seen.add(v);
+            }
+            return v;
+          }));
+        }
+        return value;
+      });
+      await fetch(this.url, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body
+      });
+    } catch {
+    }
+  }
+  async onMachineStart(context) {
+    await this.send("machine_start", { context });
+    return context;
+  }
+  async onMachineEnd(context, output) {
+    await this.send("machine_end", { context, output });
+    return output;
+  }
+  async onStateEnter(state, context) {
+    await this.send("state_enter", { state, context });
+    return context;
+  }
+  async onStateExit(state, context, output) {
+    await this.send("state_exit", { state, context, output });
+    return output;
+  }
+  async onAction(action, context) {
+    await this.send("action", { action, context });
+    return context;
+  }
+};
+var CompositeHooks = class {
+  constructor(hooks) {
+    this.hooks = hooks;
+  }
+  async onMachineStart(context) {
+    let result = context;
+    for (const hook of this.hooks) {
+      if (hook.onMachineStart) {
+        try {
+          result = await hook.onMachineStart(result);
+        } catch {
+        }
+      }
+    }
+    return result;
+  }
+  async onMachineEnd(context, output) {
+    let result = output;
+    for (const hook of this.hooks) {
+      if (hook.onMachineEnd) {
+        try {
+          result = await hook.onMachineEnd(context, result);
+        } catch {
+        }
+      }
+    }
+    return result;
+  }
+  async onStateEnter(state, context) {
+    let result = context;
+    for (const hook of this.hooks) {
+      if (hook.onStateEnter) {
+        try {
+          result = await hook.onStateEnter(state, result);
+        } catch {
+        }
+      }
+    }
+    return result;
+  }
+  async onStateExit(state, context, output) {
+    let result = output;
+    for (const hook of this.hooks) {
+      if (hook.onStateExit) {
+        try {
+          result = await hook.onStateExit(state, context, result);
+        } catch {
+        }
+      }
+    }
+    return result;
+  }
+  async onTransition(from, to, context) {
+    let result = to;
+    for (const hook of this.hooks) {
+      if (hook.onTransition) {
+        try {
+          result = await hook.onTransition(from, result, context);
+        } catch {
+        }
+      }
+    }
+    return result;
+  }
+  async onError(state, error, context) {
+    let result = null;
+    for (const hook of this.hooks) {
+      if (hook.onError) {
+        try {
+          const hookResult = await hook.onError(state, error, context);
+          if (hookResult !== null) result = hookResult;
+        } catch {
+        }
+      }
+    }
+    return result;
+  }
+  async onAction(action, context) {
+    let result = context;
+    for (const hook of this.hooks) {
+      if (hook.onAction) {
+        try {
+          result = await hook.onAction(action, result);
+        } catch {
+        }
+      }
+    }
+    return result;
+  }
+};
+var HooksRegistry = class {
+  constructor() {
+    this.factories = /* @__PURE__ */ new Map();
+  }
+  register(name, factory) {
+    this.factories.set(name, factory);
+  }
+  has(name) {
+    return this.factories.has(name);
+  }
+  resolve(ref) {
+    if (Array.isArray(ref)) {
+      const hooks = ref.map((entry) => this.resolveSingle(entry));
+      return new CompositeHooks(hooks);
+    }
+    return this.resolveSingle(ref);
+  }
+  resolveSingle(ref) {
+    const name = typeof ref === "string" ? ref : ref.name;
+    const args = typeof ref === "string" ? void 0 : ref.args;
+    const factory = this.factories.get(name);
+    if (!factory) {
+      throw new Error(
+        `No hooks registered for name '${name}'. Registered: [${[...this.factories.keys()].join(", ")}]`
+      );
+    }
+    try {
+      return new factory(args);
+    } catch {
+      return factory(args);
+    }
+  }
+};
+
 // src/flatmachine.ts
 var FlatMachine = class _FlatMachine {
   constructor(options) {
@@ -1017,7 +1191,8 @@ var FlatMachine = class _FlatMachine {
     this.pendingLaunches = [];
     this.currentStep = 0;
     this.config = typeof options.config === "string" ? yaml3.parse((0, import_fs5.readFileSync)(options.config, "utf-8")) : options.config;
-    this.hooks = options.hooks;
+    this._hooksRegistry = options.hooksRegistry ?? new HooksRegistry();
+    this.hooks = this.resolveHooks(options.hooks);
     this.configDir = options.configDir ?? process.cwd();
     this.profilesFile = this.resolveProfilesFile(options.profilesFile);
     this.executionId = options.executionId ?? this.executionId;
@@ -1040,6 +1215,15 @@ var FlatMachine = class _FlatMachine {
       this.checkpointEvents = new Set(events);
     }
   }
+  get hooksRegistry() {
+    return this._hooksRegistry;
+  }
+  resolveHooks(explicit) {
+    if (explicit) return explicit;
+    const hooksConfig = this.config.data.hooks;
+    if (!hooksConfig) return void 0;
+    return this._hooksRegistry.resolve(hooksConfig);
+  }
   async execute(input, resumeSnapshot) {
     if (this.config.data.expression_engine === "cel") {
       throw new Error("expression_engine 'cel' is not supported in the JS SDK yet");
@@ -1323,7 +1507,7 @@ var FlatMachine = class _FlatMachine {
       config: resolved.config,
       configDir: resolved.configDir,
       resultBackend: this.resultBackend,
-      hooks: this.hooks,
+      hooksRegistry: this._hooksRegistry,
       executionId: overrides?.executionId,
       parentExecutionId: overrides?.parentExecutionId,
       profilesFile: this.profilesFile
@@ -1533,146 +1717,6 @@ var FlatMachine = class _FlatMachine {
     return { [firstKey]: results[firstKey] };
   }
 };
-
-// src/hooks.ts
-var WebhookHooks = class {
-  constructor(url) {
-    this.url = url;
-  }
-  async send(event, data) {
-    try {
-      const body = JSON.stringify({ event, ...data, timestamp: (/* @__PURE__ */ new Date()).toISOString() }, (key, value) => {
-        if (typeof value === "object" && value !== null) {
-          const seen = /* @__PURE__ */ new WeakSet();
-          return JSON.parse(JSON.stringify(value, (k, v) => {
-            if (typeof v === "object" && v !== null) {
-              if (seen.has(v)) return "[Circular]";
-              seen.add(v);
-            }
-            return v;
-          }));
-        }
-        return value;
-      });
-      await fetch(this.url, {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body
-      });
-    } catch {
-    }
-  }
-  async onMachineStart(context) {
-    await this.send("machine_start", { context });
-    return context;
-  }
-  async onMachineEnd(context, output) {
-    await this.send("machine_end", { context, output });
-    return output;
-  }
-  async onStateEnter(state, context) {
-    await this.send("state_enter", { state, context });
-    return context;
-  }
-  async onStateExit(state, context, output) {
-    await this.send("state_exit", { state, context, output });
-    return output;
-  }
-  async onAction(action, context) {
-    await this.send("action", { action, context });
-    return context;
-  }
-};
-var CompositeHooks = class {
-  constructor(hooks) {
-    this.hooks = hooks;
-  }
-  async onMachineStart(context) {
-    let result = context;
-    for (const hook of this.hooks) {
-      if (hook.onMachineStart) {
-        try {
-          result = await hook.onMachineStart(result);
-        } catch {
-        }
-      }
-    }
-    return result;
-  }
-  async onMachineEnd(context, output) {
-    let result = output;
-    for (const hook of this.hooks) {
-      if (hook.onMachineEnd) {
-        try {
-          result = await hook.onMachineEnd(context, result);
-        } catch {
-        }
-      }
-    }
-    return result;
-  }
-  async onStateEnter(state, context) {
-    let result = context;
-    for (const hook of this.hooks) {
-      if (hook.onStateEnter) {
-        try {
-          result = await hook.onStateEnter(state, result);
-        } catch {
-        }
-      }
-    }
-    return result;
-  }
-  async onStateExit(state, context, output) {
-    let result = output;
-    for (const hook of this.hooks) {
-      if (hook.onStateExit) {
-        try {
-          result = await hook.onStateExit(state, context, result);
-        } catch {
-        }
-      }
-    }
-    return result;
-  }
-  async onTransition(from, to, context) {
-    let result = to;
-    for (const hook of this.hooks) {
-      if (hook.onTransition) {
-        try {
-          result = await hook.onTransition(from, result, context);
-        } catch {
-        }
-      }
-    }
-    return result;
-  }
-  async onError(state, error, context) {
-    let result = null;
-    for (const hook of this.hooks) {
-      if (hook.onError) {
-        try {
-          const hookResult = await hook.onError(state, error, context);
-          if (hookResult !== null) result = hookResult;
-        } catch {
-        }
-      }
-    }
-    return result;
-  }
-  async onAction(action, context) {
-    let result = context;
-    for (const hook of this.hooks) {
-      if (hook.onAction) {
-        try {
-          result = await hook.onAction(action, result);
-        } catch {
-        }
-      }
-    }
-    return result;
-  }
-};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   CheckpointManager,
@@ -1680,6 +1724,7 @@ var CompositeHooks = class {
   DefaultExecution,
   FlatAgent,
   FlatMachine,
+  HooksRegistry,
   LocalFileBackend,
   LocalFileLock,
   MCPToolProvider,