@eventcatalog/sdk 2.21.2 → 2.22.0

package/dist/index.mjs

@@ -2494,6 +2494,7 @@ import { join as join15 } from "path";
 var cloneStep = (step) => ({
   ...step,
   ...step.message ? { message: { ...step.message } } : {},
+  ...step.agent ? { agent: { ...step.agent } } : {},
   ...step.service ? { service: { ...step.service } } : {},
   ...step.flow ? { flow: { ...step.flow } } : {},
   ...step.container ? { container: { ...step.container } } : {},
@@ -2642,6 +2643,40 @@ var FlowBuilder = class _FlowBuilder {
       step.nextSteps
     );
   }
+  /**
+   * Add an agent step.
+   *
+   * @param step - The agent step payload.
+   *
+   * @example
+   * ```ts
+   * FlowBuilder.create({
+   *   id: 'OrderFlow',
+   *   name: 'Order Flow',
+   *   version: '1.0.0',
+   *   markdown: '',
+   * })
+   *   .addAgentStep({
+   *     id: 'FraudReviewAgent',
+   *     agent: { id: 'FraudReviewAgent', version: '1.0.0' },
+   *   });
+   * ```
+   */
+  addAgentStep(step) {
+    const agent = step.agent || {
+      id: String(step.id),
+      ...step.version ? { version: step.version } : {}
+    };
+    return this.addStepWithNext(
+      {
+        id: step.id,
+        title: step.title || String(step.id),
+        ...step.summary ? { summary: step.summary } : {},
+        agent: { ...agent, ...step.version && !agent.version ? { version: step.version } : {} }
+      },
+      step.nextSteps
+    );
+  }
   /**
    * Add a data store step.
    *