Function makeMockSdk

test/proxyBedrock.test.js:13–38

(handlers)

// event-stream binary fixtures (which require internal smithy modules
// that are deep transitive deps).
function makeMockSdk(handlers) {
  let lastClientArgs = null;
  let lastCommand = null;

  class MockBedrockRuntimeClient {
    constructor(args) { lastClientArgs = args; }
    async send(command) {
      lastCommand = command;
      const isStream = command instanceof MockInvokeModelWithResponseStreamCommand;
      return isStream
        ? handlers.stream(command.input)
        : handlers.invoke(command.input);
    }
  }
  class MockInvokeModelCommand { constructor(input) { this.input = input; } }
  class MockInvokeModelWithResponseStreamCommand { constructor(input) { this.input = input; } }

  return {
    sdk: {
      BedrockRuntimeClient: MockBedrockRuntimeClient,
      InvokeModelCommand: MockInvokeModelCommand,
      InvokeModelWithResponseStreamCommand: MockInvokeModelWithResponseStreamCommand,
    },
    inspect: () => ({ lastClientArgs, lastCommand }),
  };
}

async function readSseStream(stream) {
  const reader = stream.getReader();