bedrock constants.ts

0% Statements 0/44

0% Branches 0/1

0% Functions 0/1

0% Lines 0/44

Press n or j to go to the next uncovered block, b, p or k for the previous block.


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  export const LLAMA_2_SPECIAL_TOKENS = {
  BEGINNING_OF_SENTENCE: '<s>',
  END_OF_SENTENCE: '</s>',
  CONVERSATION_TURN_START: '[INST]',
  CONVERSATION_TURN_END: '[/INST]',
  SYSTEM_MESSAGE_START: '<<SYS>>\n',
  SYSTEM_MESSAGE_END: '\n<</SYS>>\n\n',
};
 
export const LLAMA_3_SPECIAL_TOKENS = {
  PROMPT_START: '<|begin_of_text|>',
  PROMPT_END: '<|end_of_text|>',
  PADDING: '<|finetune_right_pad_id|>',
  ROLE_START: '<|start_header_id|>',
  ROLE_END: '<|end_header_id|>',
  END_OF_MESSAGE: '<|eom_id|>',
  END_OF_TURN: '<|eot_id|>',
  TOOL_CALL: '<|python_tag|>',
};
 
export const MISTRAL_CONTROL_TOKENS = {
  UNKNOWN: '<unk>',
  BEGINNING_OF_SENTENCE: '<s>',
  END_OF_SENTENCE: '</s>',
  CONVERSATION_TURN_START: '[INST]',
  CONVERSATION_TURN_END: '[/INST]',
  AVAILABLE_TOOLS_START: '[AVAILABLE_TOOLS]',
  AVAILABLE_TOOLS_END: '[/AVAILABLE_TOOLS]',
  TOOL_RESULTS_START: '[TOOL_RESULTS]',
  TOOL_RESULTS_END: '[/TOOL_RESULTS]',
  TOOL_CALLS_START: '[TOOL_CALLS]',
  PADDING: '<pad>',
  PREFIX: '[PREFIX]',
  MIDDLE: '[MIDDLE]',
  SUFFIX: '[SUFFIX]',
};
 
export const BEDROCK_STABILITY_V1_MODELS = [
  'stable-diffusion-xl-v0',
  'stable-diffusion-xl-v1',
];
 
export const bedrockInvokeModels = [
  'cohere.command-light-text-v14',
  'cohere.command-text-v14',
  'ai21.j2-mid-v1',
  'ai21.j2-ultra-v1',
];