Class TwoBatchAgent

packages/agent-runtime/src/core/__tests__/runtime.test.ts:1661–1717

      // This tests a scenario where we execute two separate batch tool calls
      // The second batch should not re-add the tool messages from the first batch
      class TwoBatchAgent implements Agent {
        private batchCount = 0;

        tools = {
          tool_a: vi.fn().mockResolvedValue({ result: 'a' }),
          tool_b: vi.fn().mockResolvedValue({ result: 'b' }),
        };

        async runner(context: AgentRuntimeContext, _state: AgentState) {
          if (context.phase === 'user_input' || context.phase === 'tools_batch_result') {
            this.batchCount++;
            if (this.batchCount === 1) {
              // First batch
              return {
                type: 'call_tools_batch' as const,
                payload: {
                  parentMessageId: 'msg',
                  toolsCalling: [
                    {
                      id: 'batch1_call_a',
                      type: 'default' as const,
                      apiName: 'tool_a',
                      identifier: 'tool_a',
                      arguments: '{}',
                    },
                  ],
                },
              };
            } else if (this.batchCount === 2) {
              // Second batch - should not duplicate first batch's tool messages
              return {
                type: 'call_tools_batch' as const,
                payload: {
                  parentMessageId: 'msg',
                  toolsCalling: [
                    {
                      id: 'batch2_call_a',
                      type: 'default' as const,
                      apiName: 'tool_a',
                      identifier: 'tool_a',
                      arguments: '{}',
                    },
                    {
                      id: 'batch2_call_b',
                      type: 'default' as const,
                      apiName: 'tool_b',
                      identifier: 'tool_b',
                      arguments: '{}',
                    },
                  ],
                },
              };
            }
          }
          return { type: 'finish' as const, reason: 'completed' as const };
        }
      }