hts/packages/isdk/tests/snapshots/aws-bedrock.ts

81 lines
1.8 KiB
TypeScript

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',
},
];