@directive-run/ai 0.3.0 → 0.4.1

package/dist/testing.d.cts

@@ -1,5 +1,5 @@
-import { o as ApprovalRequest, u as BreakpointModifications, I as InputGuardrailData, O as OutputGuardrailData, M as Message, T as ToolCallGuardrailData, a_ as ToolCall, ad as GuardrailResult, A as AgentLike, b as AgentRunner, R as RunOptions, v as BreakpointRequest, C as Checkpoint, V as DagExecutionContext, X as DagNodeStatus, aN as RerouteEvent, aS as Scratchpad, a1 as DebugEventType, L as CheckpointStore, aj as InMemoryCheckpointStore, W as DagNode, Z as DagPattern, D as DebugEvent, G as GuardrailFn, aa as GuardrailContext } from './types-Co4BzMiH.cjs';
-import { w as MultiAgentOrchestrator, x as MultiAgentOrchestratorOptions, c as AgentOrchestrator, O as OrchestratorOptions, H as HealthMonitor, C as MultiplexedStreamChunk, D as DebugTimeline, X as ReflectionEvaluator } from './multi-agent-orchestrator-CxL8ycw_.cjs';
+import { w as MultiAgentOrchestrator, x as MultiAgentOrchestratorOptions, c as AgentOrchestrator, O as OrchestratorOptions, H as HealthMonitor, C as MultiplexedStreamChunk, D as DebugTimeline, ah as TaskRegistration, X as ReflectionEvaluator } from './multi-agent-orchestrator-CFOzWVNd.cjs';
+import { o as ApprovalRequest, u as BreakpointModifications, I as InputGuardrailData, O as OutputGuardrailData, M as Message, T as ToolCallGuardrailData, b2 as ToolCall, ad as GuardrailResult, A as AgentLike, b as AgentRunner, R as RunOptions, v as BreakpointRequest, C as Checkpoint, V as DagExecutionContext, X as DagNodeStatus, aN as RerouteEvent, aS as Scratchpad, a1 as DebugEventType, L as CheckpointStore, aj as InMemoryCheckpointStore, W as DagNode, Z as DagPattern, D as DebugEvent, G as GuardrailFn, aa as GuardrailContext } from './types-D5veI9su.cjs';
 import '@directive-run/core';
 import '@directive-run/core/plugins';
 
@@ -347,6 +347,12 @@ interface TestMultiAgentOrchestratorOptions extends Omit<MultiAgentOrchestratorO
     mockResponses?: Record<string, MockAgentConfig>;
     /** Default mock response for unmatched agents */
     defaultMockResponse?: MockAgentConfig;
+    /** Mock tasks keyed by task ID — auto-generates TaskRegistration wrappers */
+    mockTasks?: Record<string, {
+        output: unknown;
+        delay?: number;
+        shouldError?: boolean;
+    }>;
 }
 /** Test multi-agent orchestrator with additional testing utilities */
 interface TestMultiAgentOrchestrator extends MultiAgentOrchestrator {
@@ -397,6 +403,7 @@ declare function createTestMultiAgentOrchestrator(options: TestMultiAgentOrchest
  */
 declare function assertMultiAgentState(orchestrator: MultiAgentOrchestrator, expected: {
     agentStatus?: Record<string, "idle" | "running" | "completed" | "error">;
+    taskStatus?: Record<string, string>;
     totalTokens?: {
         agentId?: string;
         min?: number;
@@ -408,6 +415,24 @@ declare function assertMultiAgentState(orchestrator: MultiAgentOrchestrator, exp
     };
     pendingHandoffs?: number;
 }): void;
+/**
+ * Create a mock TaskRegistration for testing.
+ *
+ * @example
+ * ```typescript
+ * const task = createMockTask({ result: "processed" }, { delay: 100 });
+ * const orchestrator = createTestMultiAgentOrchestrator({
+ *   agents: { writer: { agent: { name: "writer" } } },
+ *   tasks: { process: task },
+ * });
+ * ```
+ */
+declare function createMockTask(output: unknown, options?: {
+    delay?: number;
+    shouldError?: boolean;
+    label?: string;
+    description?: string;
+}): TaskRegistration;
 
 /**
  * Create a test DAG pattern from a simplified node spec.
@@ -415,13 +440,13 @@ declare function assertMultiAgentState(orchestrator: MultiAgentOrchestrator, exp
  * @example
  * ```typescript
  * const pattern = createTestDag({
- *   A: { agent: "researcher" },
- *   B: { agent: "writer", deps: ["A"] },
- *   C: { agent: "reviewer", deps: ["B"] },
+ *   A: { handler: "researcher" },
+ *   B: { handler: "writer", deps: ["A"] },
+ *   C: { handler: "reviewer", deps: ["B"] },
  * });
  * ```
  */
-declare function createTestDag<T = unknown>(nodes: Record<string, Pick<DagNode, "agent" | "deps" | "when" | "transform" | "timeout" | "priority">>, merge?: (context: DagExecutionContext) => T | Promise<T>, options?: {
+declare function createTestDag<T = unknown>(nodes: Record<string, Pick<DagNode, "handler" | "deps" | "when" | "transform" | "timeout" | "priority">>, merge?: (context: DagExecutionContext) => T | Promise<T>, options?: {
     timeout?: number;
     maxConcurrent?: number;
     onNodeError?: "fail" | "skip-downstream" | "continue";
@@ -605,6 +630,12 @@ interface BreakpointCapable {
     resumeBreakpoint(id: string, modifications?: BreakpointModifications): void;
     cancelBreakpoint(id: string, reason?: string): void;
 }
+/**
+ * Create a breakpoint simulator that auto-resolves breakpoints for testing.
+ *
+ * @param options - Simulator options including `autoResumeDelay` and `modifications`.
+ * @returns An object with `handler`, `hits`, and `attachTo` for test assertions.
+ */
 declare function createBreakpointSimulator(options?: BreakpointSimulatorOptions): {
     handler: (request: BreakpointRequest) => void;
     hits: BreakpointRequest[];
@@ -702,4 +733,4 @@ declare function assertScratchpadState(scratchpad: Scratchpad, expected: Record<
  */
 declare function assertDerivedValues(orchestrator: MultiAgentOrchestrator, expected: Record<string, unknown>): void;
 
-export { type ApprovalSimulator, type ApprovalSimulatorOptions, type BreakpointCapable, type BreakpointSimulatorOptions, type ConstraintSnapshot, type GuardrailTestInput, type GuardrailTestResult, type MockAgentConfig, type MockAgentRunner, type MockAgentRunnerOptions, type RecordedCall, type TestMultiAgentOrchestrator, type TestMultiAgentOrchestratorOptions, type TestOrchestrator, type TestOrchestratorOptions, assertAgentHealth, assertBreakpointHit, assertCheckpoint, assertDagExecution, assertDerivedValues, assertMultiAgentState, assertMultiplexedStream, assertOrchestratorState, assertRerouted, assertScratchpadState, assertTimelineEvents, collectMultiplexedStream, createApprovalSimulator, createBreakpointSimulator, createConstraintRecorder, createFailingRunner, createMockAgentRunner, createMockSchema, createTestCheckpointStore, createTestDag, createTestMultiAgentOrchestrator, createTestOrchestrator, createTestReflectionEvaluator, createTestTimeline, createTimeController, testGuardrail, testGuardrailBatch };
+export { type ApprovalSimulator, type ApprovalSimulatorOptions, type BreakpointCapable, type BreakpointSimulatorOptions, type ConstraintSnapshot, type GuardrailTestInput, type GuardrailTestResult, type MockAgentConfig, type MockAgentRunner, type MockAgentRunnerOptions, type RecordedCall, type TestMultiAgentOrchestrator, type TestMultiAgentOrchestratorOptions, type TestOrchestrator, type TestOrchestratorOptions, assertAgentHealth, assertBreakpointHit, assertCheckpoint, assertDagExecution, assertDerivedValues, assertMultiAgentState, assertMultiplexedStream, assertOrchestratorState, assertRerouted, assertScratchpadState, assertTimelineEvents, collectMultiplexedStream, createApprovalSimulator, createBreakpointSimulator, createConstraintRecorder, createFailingRunner, createMockAgentRunner, createMockSchema, createMockTask, createTestCheckpointStore, createTestDag, createTestMultiAgentOrchestrator, createTestOrchestrator, createTestReflectionEvaluator, createTestTimeline, createTimeController, testGuardrail, testGuardrailBatch };

package/dist/testing.d.ts

@@ -1,5 +1,5 @@
-import { o as ApprovalRequest, u as BreakpointModifications, I as InputGuardrailData, O as OutputGuardrailData, M as Message, T as ToolCallGuardrailData, a_ as ToolCall, ad as GuardrailResult, A as AgentLike, b as AgentRunner, R as RunOptions, v as BreakpointRequest, C as Checkpoint, V as DagExecutionContext, X as DagNodeStatus, aN as RerouteEvent, aS as Scratchpad, a1 as DebugEventType, L as CheckpointStore, aj as InMemoryCheckpointStore, W as DagNode, Z as DagPattern, D as DebugEvent, G as GuardrailFn, aa as GuardrailContext } from './types-Co4BzMiH.js';
-import { w as MultiAgentOrchestrator, x as MultiAgentOrchestratorOptions, c as AgentOrchestrator, O as OrchestratorOptions, H as HealthMonitor, C as MultiplexedStreamChunk, D as DebugTimeline, X as ReflectionEvaluator } from './multi-agent-orchestrator-uMp8bLfV.js';
+import { w as MultiAgentOrchestrator, x as MultiAgentOrchestratorOptions, c as AgentOrchestrator, O as OrchestratorOptions, H as HealthMonitor, C as MultiplexedStreamChunk, D as DebugTimeline, ah as TaskRegistration, X as ReflectionEvaluator } from './multi-agent-orchestrator-CH-4Fqzg.js';
+import { o as ApprovalRequest, u as BreakpointModifications, I as InputGuardrailData, O as OutputGuardrailData, M as Message, T as ToolCallGuardrailData, b2 as ToolCall, ad as GuardrailResult, A as AgentLike, b as AgentRunner, R as RunOptions, v as BreakpointRequest, C as Checkpoint, V as DagExecutionContext, X as DagNodeStatus, aN as RerouteEvent, aS as Scratchpad, a1 as DebugEventType, L as CheckpointStore, aj as InMemoryCheckpointStore, W as DagNode, Z as DagPattern, D as DebugEvent, G as GuardrailFn, aa as GuardrailContext } from './types-D5veI9su.js';
 import '@directive-run/core';
 import '@directive-run/core/plugins';
 
@@ -347,6 +347,12 @@ interface TestMultiAgentOrchestratorOptions extends Omit<MultiAgentOrchestratorO
     mockResponses?: Record<string, MockAgentConfig>;
     /** Default mock response for unmatched agents */
     defaultMockResponse?: MockAgentConfig;
+    /** Mock tasks keyed by task ID — auto-generates TaskRegistration wrappers */
+    mockTasks?: Record<string, {
+        output: unknown;
+        delay?: number;
+        shouldError?: boolean;
+    }>;
 }
 /** Test multi-agent orchestrator with additional testing utilities */
 interface TestMultiAgentOrchestrator extends MultiAgentOrchestrator {
@@ -397,6 +403,7 @@ declare function createTestMultiAgentOrchestrator(options: TestMultiAgentOrchest
  */
 declare function assertMultiAgentState(orchestrator: MultiAgentOrchestrator, expected: {
     agentStatus?: Record<string, "idle" | "running" | "completed" | "error">;
+    taskStatus?: Record<string, string>;
     totalTokens?: {
         agentId?: string;
         min?: number;
@@ -408,6 +415,24 @@ declare function assertMultiAgentState(orchestrator: MultiAgentOrchestrator, exp
     };
     pendingHandoffs?: number;
 }): void;
+/**
+ * Create a mock TaskRegistration for testing.
+ *
+ * @example
+ * ```typescript
+ * const task = createMockTask({ result: "processed" }, { delay: 100 });
+ * const orchestrator = createTestMultiAgentOrchestrator({
+ *   agents: { writer: { agent: { name: "writer" } } },
+ *   tasks: { process: task },
+ * });
+ * ```
+ */
+declare function createMockTask(output: unknown, options?: {
+    delay?: number;
+    shouldError?: boolean;
+    label?: string;
+    description?: string;
+}): TaskRegistration;
 
 /**
  * Create a test DAG pattern from a simplified node spec.
@@ -415,13 +440,13 @@ declare function assertMultiAgentState(orchestrator: MultiAgentOrchestrator, exp
  * @example
  * ```typescript
  * const pattern = createTestDag({
- *   A: { agent: "researcher" },
- *   B: { agent: "writer", deps: ["A"] },
- *   C: { agent: "reviewer", deps: ["B"] },
+ *   A: { handler: "researcher" },
+ *   B: { handler: "writer", deps: ["A"] },
+ *   C: { handler: "reviewer", deps: ["B"] },
  * });
  * ```
  */
-declare function createTestDag<T = unknown>(nodes: Record<string, Pick<DagNode, "agent" | "deps" | "when" | "transform" | "timeout" | "priority">>, merge?: (context: DagExecutionContext) => T | Promise<T>, options?: {
+declare function createTestDag<T = unknown>(nodes: Record<string, Pick<DagNode, "handler" | "deps" | "when" | "transform" | "timeout" | "priority">>, merge?: (context: DagExecutionContext) => T | Promise<T>, options?: {
     timeout?: number;
     maxConcurrent?: number;
     onNodeError?: "fail" | "skip-downstream" | "continue";
@@ -605,6 +630,12 @@ interface BreakpointCapable {
     resumeBreakpoint(id: string, modifications?: BreakpointModifications): void;
     cancelBreakpoint(id: string, reason?: string): void;
 }
+/**
+ * Create a breakpoint simulator that auto-resolves breakpoints for testing.
+ *
+ * @param options - Simulator options including `autoResumeDelay` and `modifications`.
+ * @returns An object with `handler`, `hits`, and `attachTo` for test assertions.
+ */
 declare function createBreakpointSimulator(options?: BreakpointSimulatorOptions): {
     handler: (request: BreakpointRequest) => void;
     hits: BreakpointRequest[];
@@ -702,4 +733,4 @@ declare function assertScratchpadState(scratchpad: Scratchpad, expected: Record<
  */
 declare function assertDerivedValues(orchestrator: MultiAgentOrchestrator, expected: Record<string, unknown>): void;
 
-export { type ApprovalSimulator, type ApprovalSimulatorOptions, type BreakpointCapable, type BreakpointSimulatorOptions, type ConstraintSnapshot, type GuardrailTestInput, type GuardrailTestResult, type MockAgentConfig, type MockAgentRunner, type MockAgentRunnerOptions, type RecordedCall, type TestMultiAgentOrchestrator, type TestMultiAgentOrchestratorOptions, type TestOrchestrator, type TestOrchestratorOptions, assertAgentHealth, assertBreakpointHit, assertCheckpoint, assertDagExecution, assertDerivedValues, assertMultiAgentState, assertMultiplexedStream, assertOrchestratorState, assertRerouted, assertScratchpadState, assertTimelineEvents, collectMultiplexedStream, createApprovalSimulator, createBreakpointSimulator, createConstraintRecorder, createFailingRunner, createMockAgentRunner, createMockSchema, createTestCheckpointStore, createTestDag, createTestMultiAgentOrchestrator, createTestOrchestrator, createTestReflectionEvaluator, createTestTimeline, createTimeController, testGuardrail, testGuardrailBatch };
+export { type ApprovalSimulator, type ApprovalSimulatorOptions, type BreakpointCapable, type BreakpointSimulatorOptions, type ConstraintSnapshot, type GuardrailTestInput, type GuardrailTestResult, type MockAgentConfig, type MockAgentRunner, type MockAgentRunnerOptions, type RecordedCall, type TestMultiAgentOrchestrator, type TestMultiAgentOrchestratorOptions, type TestOrchestrator, type TestOrchestratorOptions, assertAgentHealth, assertBreakpointHit, assertCheckpoint, assertDagExecution, assertDerivedValues, assertMultiAgentState, assertMultiplexedStream, assertOrchestratorState, assertRerouted, assertScratchpadState, assertTimelineEvents, collectMultiplexedStream, createApprovalSimulator, createBreakpointSimulator, createConstraintRecorder, createFailingRunner, createMockAgentRunner, createMockSchema, createMockTask, createTestCheckpointStore, createTestDag, createTestMultiAgentOrchestrator, createTestOrchestrator, createTestReflectionEvaluator, createTestTimeline, createTimeController, testGuardrail, testGuardrailBatch };