upgraded model strings

This commit is contained in:
hagen-danswer 2024-10-22 14:02:46 -07:00 committed by pablodanswer
parent eae1dad0fa
commit e46942e330
3 changed files with 12 additions and 11 deletions

View File

@ -100,8 +100,8 @@ def fetch_available_well_known_llms() -> list[WellKnownLLMProviderDescriptor]:
api_version_required=False,
custom_config_keys=[],
llm_names=fetch_models_for_provider(ANTHROPIC_PROVIDER_NAME),
default_model="claude-3-5-sonnet-20240620",
default_fast_model="claude-3-5-sonnet-20240620",
default_model="claude-3-5-sonnet-20241022",
default_fast_model="claude-3-5-sonnet-20241022",
),
WellKnownLLMProviderDescriptor(
name=AZURE_PROVIDER_NAME,
@ -135,8 +135,8 @@ def fetch_available_well_known_llms() -> list[WellKnownLLMProviderDescriptor]:
),
],
llm_names=fetch_models_for_provider(BEDROCK_PROVIDER_NAME),
default_model="anthropic.claude-3-5-sonnet-20240620-v1:0",
default_fast_model="anthropic.claude-3-5-sonnet-20240620-v1:0",
default_model="anthropic.claude-3-5-sonnet-claude-3-5-sonnet-20241022-v1:0",
default_fast_model="anthropic.claude-3-5-sonnet-claude-3-5-sonnet-20241022-v1:0",
),
]

View File

@ -277,7 +277,7 @@ const MODEL_DISPLAY_NAMES: { [key: string]: string } = {
"claude-2.1": "Claude 2.1",
"claude-2.0": "Claude 2.0",
"claude-instant-1.2": "Claude Instant 1.2",
"claude-3-5-sonnet-20240620": "Claude 3.5 Sonnet",
"claude-3-5-sonnet-20241022": "Claude 3.5 Sonnet",
// Bedrock models
"meta.llama3-1-70b-instruct-v1:0": "Llama 3.1 70B",
@ -300,7 +300,8 @@ const MODEL_DISPLAY_NAMES: { [key: string]: string } = {
"anthropic.claude-v1": "Claude v1",
"anthropic.claude-3-opus-20240229-v1:0": "Claude 3 Opus",
"anthropic.claude-3-haiku-20240307-v1:0": "Claude 3 Haiku",
"anthropic.claude-3-5-sonnet-20240620-v1:0": "Claude 3.5 Sonnet",
"anthropic.claude-3-5-sonnet-claude-3-5-sonnet-20241022-v1:0":
"Claude 3.5 Sonnet",
"anthropic.claude-3-sonnet-20240229-v1:0": "Claude 3 Sonnet",
"mistral.mistral-large-2402-v1:0": "Mistral Large",
"mistral.mixtral-8x7b-instruct-v0:1": "Mixtral 8x7B Instruct",
@ -323,7 +324,7 @@ export const defaultModelsByProvider: { [name: string]: string[] } = {
"meta.llama3-1-8b-instruct-v1:0",
"anthropic.claude-3-opus-20240229-v1:0",
"mistral.mistral-large-2402-v1:0",
"anthropic.claude-3-5-sonnet-20240620-v1:0",
"anthropic.claude-3-5-sonnet-claude-3-5-sonnet-20241022-v1:0",
],
anthropic: ["claude-3-opus-20240229", "claude-3-5-sonnet-20240620"],
anthropic: ["claude-3-opus-20240229", "claude-3-5-sonnet-20241022"],
};

View File

@ -69,7 +69,7 @@ const MODEL_NAMES_SUPPORTING_IMAGE_INPUT = [
"gpt-4-turbo",
"gpt-4-1106-vision-preview",
// standard claude names
"claude-3-5-sonnet-20240620",
"claude-3-5-sonnet-20241022",
"claude-3-opus-20240229",
"claude-3-sonnet-20240229",
"claude-3-haiku-20240307",
@ -77,12 +77,12 @@ const MODEL_NAMES_SUPPORTING_IMAGE_INPUT = [
"claude-3-opus-20240229-v1:0",
"claude-3-sonnet-20240229-v1:0",
"claude-3-haiku-20240307-v1:0",
"claude-3-5-sonnet-20240620-v1:0",
"claude-3-5-sonnet-claude-3-5-sonnet-20241022-v1:0",
// claude names with full AWS Bedrock names
"anthropic.claude-3-opus-20240229-v1:0",
"anthropic.claude-3-sonnet-20240229-v1:0",
"anthropic.claude-3-haiku-20240307-v1:0",
"anthropic.claude-3-5-sonnet-20240620-v1:0",
"anthropic.claude-3-5-sonnet-claude-3-5-sonnet-20241022-v1:0",
];
export function checkLLMSupportsImageInput(model: string) {