@@ -120,18 +120,26 @@ export const enum CHAT_PROVIDERS {
120120export const SUPPORTED_CHAT_PROVIDERS = [ CHAT_PROVIDERS . HUGGING_FACE , CHAT_PROVIDERS . ANTHROPIC , CHAT_PROVIDERS . BEDROCK ] ;
121121
122122export const enum BEDROCK_MODEL_FAMILIES {
123+ AI21 = 'ai21' ,
123124 AMAZON = 'amazon' ,
124125 ANTHROPIC = 'anthropic' ,
125- AI21 = 'ai21'
126+ COHERE = 'cohere' ,
127+ META = 'meta'
126128}
127129
128130export const SUPPORTED_BEDROCK_CHAT_MODELS = [
129131 `${ BEDROCK_MODEL_FAMILIES . AI21 } .j2-ultra` ,
130132 `${ BEDROCK_MODEL_FAMILIES . AI21 } .j2-mid` ,
131133 `${ BEDROCK_MODEL_FAMILIES . AMAZON } .titan-text-express-v1` ,
134+ `${ BEDROCK_MODEL_FAMILIES . AMAZON } .titan-text-lite-v1` ,
132135 `${ BEDROCK_MODEL_FAMILIES . ANTHROPIC } .claude-v1` ,
133136 `${ BEDROCK_MODEL_FAMILIES . ANTHROPIC } .claude-v2` ,
134- `${ BEDROCK_MODEL_FAMILIES . ANTHROPIC } .claude-instant-v1`
137+ `${ BEDROCK_MODEL_FAMILIES . ANTHROPIC } .claude-v2:1` ,
138+ `${ BEDROCK_MODEL_FAMILIES . ANTHROPIC } .claude-instant-v1` ,
139+ `${ BEDROCK_MODEL_FAMILIES . COHERE } .command-text-v14` ,
140+ `${ BEDROCK_MODEL_FAMILIES . COHERE } .command-light-text-v14` ,
141+ `${ BEDROCK_MODEL_FAMILIES . META } .llama2-13b-chat-v1` ,
142+ `${ BEDROCK_MODEL_FAMILIES . META } .llama2-70b-chat-v1`
135143] ;
136144
137145export const DEFAULT_HUGGINGFACE_TEMPERATURE = 1 ;
@@ -162,6 +170,9 @@ Current conversation:
162170Human: {input}
163171AI:` ;
164172
173+ export const DEFAULT_META_CHAT_PROMPT = `[INST] ${ DEFAULT_CHAT_PROMPT } [/INST]` ;
174+ export const DEFAULT_META_RAG_PROMPT = `[INST] ${ DEFAULT_HUGGINGFACE_RAG_PROMPT } [/INST]`
175+
165176export const SUPPORTED_ANTHROPIC_CHAT_MODELS = [ 'claude-instant-1' , 'claude-1' , 'claude-2' ] ;
166177export const DEFAULT_ANTHROPIC_TEMPERATURE = 1 ;
167178export const MIN_ANTHROPIC_TEMPERATURE = 0 ;
@@ -199,6 +210,9 @@ Assistant:`;
199210export const DEFAULT_BEDROCK_AMAZON_PROMPT = DEFAULT_CHAT_PROMPT ;
200211export const DEFAULT_BEDROCK_ANTHROPIC_PROMPT = DEFAULT_ANTHROPIC_CHAT_PROMPT ;
201212export const DEFAULT_BEDROCK_AI21_PROMPT = DEFAULT_CHAT_PROMPT ;
213+ export const DEFAULT_BEDROCK_COHERE_PROMPT = DEFAULT_CHAT_PROMPT ;
214+ export const DEFAULT_BEDROCK_META_PROMPT = DEFAULT_META_CHAT_PROMPT ;
215+
202216export const DEFAULT_BEDROCK_AMAZON_RAG_PROMPT = `Instructions: You are an AI Assistant created to help answer the User's question. You are only to answer the User's question using the provided references. You are not allowed to make things up or use your own knowledge. Only use what is provided between the <references> XML tags.
203217
204218Here are the only references you can use:
@@ -213,6 +227,28 @@ User: {input}
213227Bot:` ;
214228export const DEFAULT_BEDROCK_ANTHROPIC_RAG_PROMPT = DEFAULT_ANTHROPIC_RAG_PROMPT ;
215229export const DEFAULT_BEDROCK_AI21_RAG_PROMPT = DEFAULT_HUGGINGFACE_RAG_PROMPT ;
230+ export const DEFAULT_BEDROCK_COHERE_RAG_PROMPT = DEFAULT_HUGGINGFACE_RAG_PROMPT ;
231+ export const DEFAULT_BEDROCK_META_RAG_PROMPT = DEFAULT_META_RAG_PROMPT ;
232+
233+ export const DEFAULT_BEDROCK_ANTHROPIC_TEMPERATURE = DEFAULT_ANTHROPIC_TEMPERATURE ;
234+ export const MIN_BEDROCK_ANTHROPIC_TEMPERATURE = MIN_ANTHROPIC_TEMPERATURE ;
235+ export const MAX_BEDROCK_ANTHROPIC_TEMPERATURE = MAX_ANTHROPIC_TEMPERATURE ;
236+
237+ export const DEFAULT_BEDROCK_AMAZON_TEMPERATURE = 0.5 ;
238+ export const MIN_BEDROCK_AMAZON_TEMPERATURE = 0 ;
239+ export const MAX_BEDROCK_AMAZON_TEMPERATURE = 1 ;
240+
241+ export const DEFAULT_BEDROCK_AI21_TEMPERATURE = 1 ;
242+ export const MIN_BEDROCK_AI21_TEMPERATURE = 0 ;
243+ export const MAX_BEDROCK_AI21_TEMPERATURE = 1 ;
244+
245+ export const DEFAULT_BEDROCK_COHERE_TEMPERATURE = 0.75 ;
246+ export const MIN_BEDROCK_COHERE_TEMPERATURE = 0 ;
247+ export const MAX_BEDROCK_COHERE_TEMPERATURE = 1 ;
248+
249+ export const DEFAULT_BEDROCK_META_TEMPERATURE = 0.5 ;
250+ export const MIN_BEDROCK_META_TEMPERATURE = 0 ;
251+ export const MAX_BEDROCK_META_TEMPERATURE = 1 ;
216252
217253export const KENDRA_EDITIONS = [ 'DEVELOPER_EDITION' , 'ENTERPRISE_EDITION' ] ;
218254export const DEFAULT_KENDRA_EDITION = 'DEVELOPER_EDITION' ;
@@ -247,7 +283,7 @@ export const DEFAULT_KENDRA_STORAGE_CAPACITY_UNITS = 0;
247283export const MAX_KENDRA_QUERY_CAPACITY_UNITS = 1 ;
248284export const MAX_KENDRA_STORAGE_CAPACITY_UNITS = 5 ;
249285export const DEFAULT_KENDRA_NUMBER_OF_DOCS = 2 ;
250- export const MAX_KENDRA_NUMBER_OF_DOCS = 5 ;
286+ export const MAX_KENDRA_NUMBER_OF_DOCS = 100 ;
251287export const MIN_KENDRA_NUMBER_OF_DOCS = 1 ;
252288export const MODEL_PARAM_TYPES = [ 'string' , 'integer' , 'float' , 'boolean' , 'list' , 'dictionary' ] ;
253289
@@ -292,26 +328,40 @@ export const additionalDeploymentPlatformConfigValues = {
292328 SupportedModels : SUPPORTED_BEDROCK_CHAT_MODELS ,
293329 AllowsStreaming : true ,
294330 ModelFamilyParams : {
331+ [ BEDROCK_MODEL_FAMILIES . AI21 ] : {
332+ ChatPromptTemplate : DEFAULT_BEDROCK_AI21_PROMPT ,
333+ RAGPromptTemplate : DEFAULT_BEDROCK_AI21_RAG_PROMPT ,
334+ DefaultTemperature : DEFAULT_BEDROCK_AI21_TEMPERATURE ,
335+ MinTemperature : MIN_BEDROCK_AI21_TEMPERATURE ,
336+ MaxTemperature : MAX_BEDROCK_AI21_TEMPERATURE
337+ } ,
295338 [ BEDROCK_MODEL_FAMILIES . AMAZON ] : {
296339 ChatPromptTemplate : DEFAULT_BEDROCK_AMAZON_PROMPT ,
297340 RAGPromptTemplate : DEFAULT_BEDROCK_AMAZON_RAG_PROMPT ,
298- DefaultTemperature : DEFAULT_ANTHROPIC_TEMPERATURE ,
299- MinTemperature : MIN_ANTHROPIC_TEMPERATURE ,
300- MaxTemperature : MAX_ANTHROPIC_TEMPERATURE
341+ DefaultTemperature : DEFAULT_BEDROCK_AMAZON_TEMPERATURE ,
342+ MinTemperature : MIN_BEDROCK_AMAZON_TEMPERATURE ,
343+ MaxTemperature : MAX_BEDROCK_AMAZON_TEMPERATURE
301344 } ,
302345 [ BEDROCK_MODEL_FAMILIES . ANTHROPIC ] : {
303346 ChatPromptTemplate : DEFAULT_BEDROCK_ANTHROPIC_PROMPT ,
304347 RAGPromptTemplate : DEFAULT_BEDROCK_ANTHROPIC_RAG_PROMPT ,
305- DefaultTemperature : DEFAULT_ANTHROPIC_TEMPERATURE ,
306- MinTemperature : MIN_ANTHROPIC_TEMPERATURE ,
307- MaxTemperature : MAX_ANTHROPIC_TEMPERATURE
348+ DefaultTemperature : DEFAULT_BEDROCK_ANTHROPIC_TEMPERATURE ,
349+ MinTemperature : MIN_BEDROCK_ANTHROPIC_TEMPERATURE ,
350+ MaxTemperature : MAX_BEDROCK_ANTHROPIC_TEMPERATURE
308351 } ,
309- [ BEDROCK_MODEL_FAMILIES . AI21 ] : {
310- ChatPromptTemplate : DEFAULT_BEDROCK_AI21_PROMPT ,
311- RAGPromptTemplate : DEFAULT_BEDROCK_AI21_RAG_PROMPT ,
312- DefaultTemperature : DEFAULT_ANTHROPIC_TEMPERATURE ,
313- MinTemperature : MIN_ANTHROPIC_TEMPERATURE ,
314- MaxTemperature : MAX_ANTHROPIC_TEMPERATURE
352+ [ BEDROCK_MODEL_FAMILIES . COHERE ] : {
353+ ChatPromptTemplate : DEFAULT_BEDROCK_COHERE_PROMPT ,
354+ RAGPromptTemplate : DEFAULT_BEDROCK_COHERE_RAG_PROMPT ,
355+ DefaultTemperature : DEFAULT_BEDROCK_COHERE_TEMPERATURE ,
356+ MinTemperature : MIN_BEDROCK_COHERE_TEMPERATURE ,
357+ MaxTemperature : MAX_BEDROCK_COHERE_TEMPERATURE
358+ } ,
359+ [ BEDROCK_MODEL_FAMILIES . META ] : {
360+ ChatPromptTemplate : DEFAULT_BEDROCK_META_PROMPT ,
361+ RAGPromptTemplate : DEFAULT_BEDROCK_META_RAG_PROMPT ,
362+ DefaultTemperature : DEFAULT_BEDROCK_META_TEMPERATURE ,
363+ MinTemperature : MIN_BEDROCK_META_TEMPERATURE ,
364+ MaxTemperature : MAX_BEDROCK_META_TEMPERATURE
315365 }
316366 }
317367 }
0 commit comments