@providerprotocol/agents 0.0.2 → 0.0.4

package/tests/unit/execution/subagent-events.test.ts

@@ -1,235 +0,0 @@
-/**
- * Unit tests for sub-agent event types.
- *
- * @see UAP-1.0 Spec Section 8.7
- */
-
-import { describe, test, expect } from 'bun:test';
-import type {
-  SubagentEvent,
-  SubagentStartEvent,
-  SubagentInnerEvent,
-  SubagentEndEvent,
-  OnSubagentEvent,
-  AgentStreamEvent,
-} from '../../../src/execution/types.ts';
-
-describe('SubagentEvent types', () => {
-  describe('SubagentStartEvent', () => {
-    test('should have correct structure', () => {
-      const event: SubagentStartEvent = {
-        type: 'subagent_start',
-        subagentId: 'sub-123',
-        subagentType: 'explorer',
-        parentToolCallId: 'tool-456',
-        prompt: 'Find all TypeScript files',
-        timestamp: Date.now(),
-      };
-
-      expect(event.type).toBe('subagent_start');
-      expect(event.subagentId).toBe('sub-123');
-      expect(event.subagentType).toBe('explorer');
-      expect(event.parentToolCallId).toBe('tool-456');
-      expect(event.prompt).toBe('Find all TypeScript files');
-      expect(typeof event.timestamp).toBe('number');
-    });
-  });
-
-  describe('SubagentInnerEvent', () => {
-    test('should wrap inner AgentStreamEvent', () => {
-      const innerEvent: AgentStreamEvent = {
-        source: 'upp',
-        upp: {
-          type: 'text_delta',
-          index: 0,
-          delta: { text: 'Hello' },
-        },
-      };
-
-      const event: SubagentInnerEvent = {
-        type: 'subagent_event',
-        subagentId: 'sub-123',
-        subagentType: 'explorer',
-        parentToolCallId: 'tool-456',
-        innerEvent,
-      };
-
-      expect(event.type).toBe('subagent_event');
-      expect(event.innerEvent.source).toBe('upp');
-      expect(event.innerEvent.upp?.type).toBe('text_delta');
-    });
-
-    test('should wrap UAP events from sub-agent', () => {
-      const innerEvent: AgentStreamEvent = {
-        source: 'uap',
-        uap: {
-          type: 'step_start',
-          step: 1,
-          agentId: 'sub-123',
-          data: { iteration: 1 },
-        },
-      };
-
-      const event: SubagentInnerEvent = {
-        type: 'subagent_event',
-        subagentId: 'sub-123',
-        subagentType: 'planner',
-        parentToolCallId: 'tool-789',
-        innerEvent,
-      };
-
-      expect(event.innerEvent.source).toBe('uap');
-      expect(event.innerEvent.uap?.type).toBe('step_start');
-    });
-  });
-
-  describe('SubagentEndEvent', () => {
-    test('should have correct structure for successful completion', () => {
-      const event: SubagentEndEvent = {
-        type: 'subagent_end',
-        subagentId: 'sub-123',
-        subagentType: 'explorer',
-        parentToolCallId: 'tool-456',
-        success: true,
-        result: 'Found 42 TypeScript files',
-        timestamp: Date.now(),
-        toolExecutions: [
-          {
-            toolName: 'Glob',
-            arguments: { pattern: '**/*.ts' },
-            result: 'file1.ts\nfile2.ts',
-          },
-        ],
-      };
-
-      expect(event.type).toBe('subagent_end');
-      expect(event.success).toBe(true);
-      expect(event.result).toBe('Found 42 TypeScript files');
-      expect(event.error).toBeUndefined();
-      expect(event.toolExecutions).toHaveLength(1);
-      expect(event.toolExecutions?.[0]?.toolName).toBe('Glob');
-    });
-
-    test('should have correct structure for failed completion', () => {
-      const event: SubagentEndEvent = {
-        type: 'subagent_end',
-        subagentId: 'sub-123',
-        subagentType: 'explorer',
-        parentToolCallId: 'tool-456',
-        success: false,
-        error: 'Rate limit exceeded',
-        timestamp: Date.now(),
-      };
-
-      expect(event.type).toBe('subagent_end');
-      expect(event.success).toBe(false);
-      expect(event.error).toBe('Rate limit exceeded');
-      expect(event.result).toBeUndefined();
-    });
-  });
-
-  describe('SubagentEvent union type', () => {
-    test('should discriminate by type field', () => {
-      const events: SubagentEvent[] = [
-        {
-          type: 'subagent_start',
-          subagentId: 'sub-1',
-          subagentType: 'explorer',
-          parentToolCallId: 'tool-1',
-          prompt: 'test',
-          timestamp: 1000,
-        },
-        {
-          type: 'subagent_event',
-          subagentId: 'sub-1',
-          subagentType: 'explorer',
-          parentToolCallId: 'tool-1',
-          innerEvent: { source: 'upp', upp: { type: 'text_delta', index: 0, delta: { text: 'hi' } } },
-        },
-        {
-          type: 'subagent_end',
-          subagentId: 'sub-1',
-          subagentType: 'explorer',
-          parentToolCallId: 'tool-1',
-          success: true,
-          timestamp: 2000,
-        },
-      ];
-
-      // Type narrowing should work correctly
-      for (const event of events) {
-        switch (event.type) {
-          case 'subagent_start':
-            expect(event.prompt).toBeDefined();
-            break;
-          case 'subagent_event':
-            expect(event.innerEvent).toBeDefined();
-            break;
-          case 'subagent_end':
-            expect(typeof event.success).toBe('boolean');
-            break;
-        }
-      }
-    });
-  });
-
-  describe('OnSubagentEvent callback', () => {
-    test('should be callable with any SubagentEvent', () => {
-      const receivedEvents: SubagentEvent[] = [];
-
-      const callback: OnSubagentEvent = (event) => {
-        receivedEvents.push(event);
-      };
-
-      // Emit start event
-      callback({
-        type: 'subagent_start',
-        subagentId: 'sub-1',
-        subagentType: 'explorer',
-        parentToolCallId: 'tool-1',
-        prompt: 'test',
-        timestamp: Date.now(),
-      });
-
-      // Emit inner event
-      callback({
-        type: 'subagent_event',
-        subagentId: 'sub-1',
-        subagentType: 'explorer',
-        parentToolCallId: 'tool-1',
-        innerEvent: { source: 'upp', upp: { type: 'message_stop', index: 0, delta: {} } },
-      });
-
-      // Emit end event
-      callback({
-        type: 'subagent_end',
-        subagentId: 'sub-1',
-        subagentType: 'explorer',
-        parentToolCallId: 'tool-1',
-        success: true,
-        result: 'done',
-        timestamp: Date.now(),
-      });
-
-      expect(receivedEvents).toHaveLength(3);
-      expect(receivedEvents[0]?.type).toBe('subagent_start');
-      expect(receivedEvents[1]?.type).toBe('subagent_event');
-      expect(receivedEvents[2]?.type).toBe('subagent_end');
-    });
-  });
-});
-
-describe('UAPEventType includes subagent events', () => {
-  test('subagent event types are valid UAPEventType values', () => {
-    // These should compile without errors
-    const types: import('../../../src/execution/types.ts').UAPEventType[] = [
-      'subagent_start',
-      'subagent_event',
-      'subagent_end',
-    ];
-
-    expect(types).toContain('subagent_start');
-    expect(types).toContain('subagent_event');
-    expect(types).toContain('subagent_end');
-  });
-});

package/tests/unit/execution/tool-ordering.test.ts

@@ -1,310 +0,0 @@
-import { describe, test, expect } from 'bun:test';
-import type { ToolCall } from '@providerprotocol/ai';
-import {
-  orderToolCalls,
-  hasToolDependencies,
-  hasCallDependencies,
-} from '../../../src/execution/tool-ordering.ts';
-import type {
-  ToolWithDependencies,
-  OrderedToolCall,
-} from '../../../src/execution/types.ts';
-
-// Helper to create a mock tool
-function createTool(
-  name: string,
-  options?: { sequential?: boolean; dependsOn?: string[] },
-): ToolWithDependencies {
-  return {
-    name,
-    description: `Tool ${name}`,
-    parameters: { type: 'object', properties: {} },
-    run: async () => 'result',
-    sequential: options?.sequential,
-    dependsOn: options?.dependsOn,
-  };
-}
-
-// Helper to create a mock tool call
-function createToolCall(
-  toolName: string,
-  id?: string,
-  after?: string[],
-): ToolCall | OrderedToolCall {
-  const call: OrderedToolCall = {
-    toolCallId: id ?? `call-${toolName}`,
-    toolName,
-    arguments: {},
-  };
-  if (after) {
-    call.after = after;
-  }
-  return call;
-}
-
-describe('orderToolCalls()', () => {
-  test('returns empty array for no tool calls', () => {
-    const groups = orderToolCalls([], []);
-    expect(groups).toEqual([]);
-  });
-
-  test('groups independent tools together', () => {
-    const tools = [
-      createTool('tool_a'),
-      createTool('tool_b'),
-      createTool('tool_c'),
-    ];
-
-    const calls = [
-      createToolCall('tool_a'),
-      createToolCall('tool_b'),
-      createToolCall('tool_c'),
-    ];
-
-    const groups = orderToolCalls(calls, tools);
-
-    expect(groups).toHaveLength(1);
-    const firstGroup = groups[0];
-    expect(firstGroup).toBeDefined();
-    if (firstGroup) {
-      expect(firstGroup.calls).toHaveLength(3);
-      expect(firstGroup.isBarrier).toBe(false);
-    }
-  });
-
-  test('sequential tool creates barrier', () => {
-    const tools = [
-      createTool('read', { sequential: true }),
-      createTool('process'),
-    ];
-
-    const calls = [
-      createToolCall('read'),
-      createToolCall('process'),
-    ];
-
-    const groups = orderToolCalls(calls, tools);
-
-    // read should be in its own group as a barrier
-    expect(groups.length).toBeGreaterThanOrEqual(1);
-    const readGroup = groups.find((g) => g.calls.some((c) => c.toolName === 'read'));
-    expect(readGroup?.isBarrier).toBe(true);
-  });
-
-  test('dependsOn creates ordering', () => {
-    const tools = [
-      createTool('read'),
-      createTool('write', { dependsOn: ['read'] }),
-    ];
-
-    const calls = [
-      createToolCall('read'),
-      createToolCall('write'),
-    ];
-
-    const groups = orderToolCalls(calls, tools);
-
-    // Should have at least 2 groups due to dependency
-    expect(groups.length).toBeGreaterThanOrEqual(2);
-
-    // Find read and write group indices
-    const readGroupIndex = groups.findIndex(
-      (g) => g.calls.some((c) => c.toolName === 'read'),
-    );
-    const writeGroupIndex = groups.findIndex(
-      (g) => g.calls.some((c) => c.toolName === 'write'),
-    );
-
-    // read should come before write
-    expect(readGroupIndex).toBeLessThan(writeGroupIndex);
-  });
-
-  test('model-driven dependencies (after field)', () => {
-    const tools = [
-      createTool('tool_a'),
-      createTool('tool_b'),
-    ];
-
-    const calls = [
-      createToolCall('tool_a', 'call-1'),
-      createToolCall('tool_b', 'call-2', ['call-1']), // depends on call-1
-    ] as OrderedToolCall[];
-
-    const groups = orderToolCalls(calls, tools);
-
-    // Should have at least 2 groups due to dependency
-    expect(groups.length).toBeGreaterThanOrEqual(2);
-
-    // Find group indices
-    const aGroupIndex = groups.findIndex(
-      (g) => g.calls.some((c) => c.toolCallId === 'call-1'),
-    );
-    const bGroupIndex = groups.findIndex(
-      (g) => g.calls.some((c) => c.toolCallId === 'call-2'),
-    );
-
-    // call-1 should come before call-2
-    expect(aGroupIndex).toBeLessThan(bGroupIndex);
-  });
-
-  test('complex dependency chain', () => {
-    // A -> B -> C (chain dependency)
-    const tools = [
-      createTool('tool_a'),
-      createTool('tool_b', { dependsOn: ['tool_a'] }),
-      createTool('tool_c', { dependsOn: ['tool_b'] }),
-    ];
-
-    const calls = [
-      createToolCall('tool_a'),
-      createToolCall('tool_b'),
-      createToolCall('tool_c'),
-    ];
-
-    const groups = orderToolCalls(calls, tools);
-
-    // Should have 3 groups, one for each tool
-    expect(groups).toHaveLength(3);
-
-    // Verify order
-    expect(groups[0]?.calls[0]?.toolName).toBe('tool_a');
-    expect(groups[1]?.calls[0]?.toolName).toBe('tool_b');
-    expect(groups[2]?.calls[0]?.toolName).toBe('tool_c');
-  });
-
-  test('parallel tools with shared dependency', () => {
-    // D depends on both A and B (which can run in parallel)
-    const tools = [
-      createTool('tool_a'),
-      createTool('tool_b'),
-      createTool('tool_d', { dependsOn: ['tool_a', 'tool_b'] }),
-    ];
-
-    const calls = [
-      createToolCall('tool_a'),
-      createToolCall('tool_b'),
-      createToolCall('tool_d'),
-    ];
-
-    const groups = orderToolCalls(calls, tools);
-
-    // First group should have A and B (parallel)
-    // Second group should have D
-    expect(groups.length).toBeGreaterThanOrEqual(2);
-
-    const firstGroup = groups[0];
-    expect(firstGroup).toBeDefined();
-    if (firstGroup) {
-      const hasA = firstGroup.calls.some((c) => c.toolName === 'tool_a');
-      const hasB = firstGroup.calls.some((c) => c.toolName === 'tool_b');
-      expect(hasA).toBe(true);
-      expect(hasB).toBe(true);
-    }
-
-    const dGroupIndex = groups.findIndex(
-      (g) => g.calls.some((c) => c.toolName === 'tool_d'),
-    );
-    expect(dGroupIndex).toBe(groups.length - 1); // D should be last
-  });
-
-  test('handles unknown tools gracefully', () => {
-    const tools = [createTool('known')];
-    const calls = [
-      createToolCall('known'),
-      createToolCall('unknown'),
-    ];
-
-    // Should not throw
-    const groups = orderToolCalls(calls, tools);
-    expect(groups.length).toBeGreaterThan(0);
-  });
-
-  test('handles cyclic dependencies by executing remaining items', () => {
-    // A depends on B, B depends on A (cycle)
-    const tools = [
-      createTool('tool_a', { dependsOn: ['tool_b'] }),
-      createTool('tool_b', { dependsOn: ['tool_a'] }),
-    ];
-
-    const calls = [
-      createToolCall('tool_a'),
-      createToolCall('tool_b'),
-    ];
-
-    // Should not hang - should execute items despite cycle
-    const groups = orderToolCalls(calls, tools);
-    expect(groups.length).toBeGreaterThan(0);
-
-    // All calls should be included
-    const allCalls = groups.flatMap((g) => g.calls);
-    expect(allCalls).toHaveLength(2);
-  });
-});
-
-describe('hasToolDependencies()', () => {
-  test('returns false for tools without dependencies', () => {
-    const tools = [
-      createTool('tool_a'),
-      createTool('tool_b'),
-    ];
-
-    expect(hasToolDependencies(tools)).toBe(false);
-  });
-
-  test('returns true for sequential tool', () => {
-    const tools = [
-      createTool('tool_a'),
-      createTool('tool_b', { sequential: true }),
-    ];
-
-    expect(hasToolDependencies(tools)).toBe(true);
-  });
-
-  test('returns true for tool with dependsOn', () => {
-    const tools = [
-      createTool('tool_a'),
-      createTool('tool_b', { dependsOn: ['tool_a'] }),
-    ];
-
-    expect(hasToolDependencies(tools)).toBe(true);
-  });
-
-  test('returns false for empty array', () => {
-    expect(hasToolDependencies([])).toBe(false);
-  });
-});
-
-describe('hasCallDependencies()', () => {
-  test('returns false for calls without after field', () => {
-    const calls = [
-      createToolCall('tool_a'),
-      createToolCall('tool_b'),
-    ];
-
-    expect(hasCallDependencies(calls)).toBe(false);
-  });
-
-  test('returns true for call with after field', () => {
-    const calls = [
-      createToolCall('tool_a', 'call-1'),
-      createToolCall('tool_b', 'call-2', ['call-1']),
-    ] as OrderedToolCall[];
-
-    expect(hasCallDependencies(calls)).toBe(true);
-  });
-
-  test('returns false for empty array', () => {
-    expect(hasCallDependencies([])).toBe(false);
-  });
-
-  test('returns false for empty after array', () => {
-    const call: OrderedToolCall = {
-      toolCallId: 'call-1',
-      toolName: 'tool_a',
-      arguments: {},
-      after: [],
-    };
-
-    expect(hasCallDependencies([call])).toBe(false);
-  });
-});