@memgrafter/flatagents 0.9.0 → 2.0.0

package/dist/index.mjs

@@ -949,6 +949,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) {
@@ -960,7 +1133,8 @@ var FlatMachine = class _FlatMachine {
     this.pendingLaunches = [];
     this.currentStep = 0;
     this.config = typeof options.config === "string" ? yaml3.parse(readFileSync3(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;
@@ -983,6 +1157,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");
@@ -1266,7 +1449,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
@@ -1476,152 +1659,13 @@ 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;
-  }
-};
 export {
   CheckpointManager,
   CompositeHooks,
   DefaultExecution,
   FlatAgent,
   FlatMachine,
+  HooksRegistry,
   LocalFileBackend,
   LocalFileLock,
   MCPToolProvider,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memgrafter/flatagents",
-  "version": "0.9.0",
+  "version": "2.0.0",
   "description": "TypeScript SDK for FlatAgents - Declarative LLM orchestration with YAML",
   "main": "dist/index.js",
   "module": "dist/index.mjs",

package/schemas/flatagent.d.ts

@@ -149,7 +149,7 @@
  * The profile field specifies which profile name to use as base.
  */
 
-export const SPEC_VERSION = "0.9.0";
+export const SPEC_VERSION = "2.0.0";
 
 export interface AgentWrapper {
   spec: "flatagent";
@@ -166,6 +166,16 @@ export interface AgentData {
   instruction_suffix?: string;
   output?: OutputSchema;
   mcp?: MCPConfig;
+  tools?: ToolDefinition[];
+}
+
+export interface ToolDefinition {
+  type: "function";
+  function: {
+    name: string;
+    description?: string;
+    parameters?: Record<string, any>;
+  };
 }
 
 export interface MCPConfig {

package/schemas/flatagent.schema.json

@@ -59,6 +59,12 @@
         },
         "mcp": {
           "$ref": "#/definitions/MCPConfig"
+        },
+        "tools": {
+          "type": "array",
+          "items": {
+            "$ref": "#/definitions/ToolDefinition"
+          }
         }
       },
       "required": [
@@ -266,6 +272,38 @@
         }
       },
       "additionalProperties": false
+    },
+    "ToolDefinition": {
+      "type": "object",
+      "properties": {
+        "type": {
+          "type": "string",
+          "const": "function"
+        },
+        "function": {
+          "type": "object",
+          "properties": {
+            "name": {
+              "type": "string"
+            },
+            "description": {
+              "type": "string"
+            },
+            "parameters": {
+              "type": "object"
+            }
+          },
+          "required": [
+            "name"
+          ],
+          "additionalProperties": false
+        }
+      },
+      "required": [
+        "type",
+        "function"
+      ],
+      "additionalProperties": false
     }
   }
 }

package/schemas/flatagent.slim.d.ts

@@ -1,4 +1,4 @@
-export const SPEC_VERSION = "0.9.0";
+export const SPEC_VERSION = "2.0.0";
 export interface AgentWrapper {
     spec: "flatagent";
     spec_version: string;
@@ -13,6 +13,15 @@ export interface AgentData {
     instruction_suffix?: string;
     output?: OutputSchema;
     mcp?: MCPConfig;
+    tools?: ToolDefinition[];
+}
+export interface ToolDefinition {
+    type: "function";
+    function: {
+        name: string;
+        description?: string;
+        parameters?: Record<string, any>;
+    };
 }
 export interface MCPConfig {
     servers: Record<string, MCPServerDef>;