export const bedrockAnthropicChunks = [
  { completion: ' Hello', stop_reason: null, stop: null },
  { completion: ',', stop_reason: null, stop: null },
  { completion: ' world', stop_reason: null, stop: null },
  { completion: '.', stop_reason: 'stop_sequence', stop: '\n\nHuman:' },
];

export const bedrockAnthropicV3Chunks = [
  {
    type: 'content_block_delta',
    index: 0,
    delta: { type: 'text_delta', text: ' Hello' },
  },
  {
    type: 'content_block_delta',
    index: 0,
    delta: { type: 'text_delta', text: ',' },
  },
  {
    type: 'content_block_delta',
    index: 0,
    delta: { type: 'text_delta', text: ' world' },
  },
  {
    type: 'content_block_delta',
    index: 0,
    delta: { type: 'text_delta', text: '.' },
  },
];

export const bedrockCohereChunks = [
  { is_finished: false, text: ' Hello' },
  { is_finished: false, text: '!' },
  { is_finished: false, text: ' How' },
  { is_finished: false, text: ' can' },
  { is_finished: false, text: ' I' },
  { is_finished: false, text: ' help' },
  { is_finished: false, text: ' you' },
  { is_finished: false, text: ' today' },
  { is_finished: false, text: '?' },
];

export const bedrockLlama2Chunks = [
  {
    generation: '',
    prompt_token_count: 10,
    generation_token_count: 1,
    stop_reason: null,
  },
  {
    generation: ' Hello',
    prompt_token_count: null,
    generation_token_count: 2,
    stop_reason: null,
  },
  {
    generation: ',',
    prompt_token_count: null,
    generation_token_count: 3,
    stop_reason: null,
  },
  {
    generation: ' world',
    prompt_token_count: null,
    generation_token_count: 4,
    stop_reason: null,
  },
  {
    generation: '.',
    prompt_token_count: null,
    generation_token_count: 5,
    stop_reason: null,
  },
  {
    generation: '',
    prompt_token_count: null,
    generation_token_count: 6,
    stop_reason: 'length',
  },
];