updates mistral modelids

This commit is contained in:
nai-degen 2024-02-26 23:55:03 -06:00
parent 428e103323
commit 627559b729
2 changed files with 21 additions and 2 deletions

View File

@ -24,10 +24,25 @@ import {
// https://docs.mistral.ai/platform/endpoints
export const KNOWN_MISTRAL_AI_MODELS = [
// Mistral 7b (open weight, legacy)
"open-mistral-7b",
"mistral-tiny-2312",
// Mixtral 8x7b (open weight, legacy)
"open-mixtral-8x7b",
"mistral-small-2312",
// Mixtral Small (newer 8x7b, closed weight)
"mistral-small-latest",
"mistral-small-2402",
// Mistral Medium
"mistral-medium-latest",
"mistral-medium-2312",
// Mistral Large
"mistral-large-latest",
"mistral-large-2402",
// Deprecated identifiers (2024-05-01)
"mistral-tiny",
"mistral-small",
"mistral-medium",
"mistral-large",
];
let modelsCache: any = null;

View File

@ -130,13 +130,17 @@ export function getGoogleAIModelFamily(_model: string): ModelFamily {
}
export function getMistralAIModelFamily(model: string): MistralAIModelFamily {
const prunedModel = model.replace(/-latest$/, "");
const prunedModel = model.replace(/-(latest|\d{4})$/, "");
switch (prunedModel) {
case "mistral-tiny":
case "mistral-small":
case "mistral-medium":
case "mistral-large":
return model as MistralAIModelFamily;
case "open-mistral-7b":
return "mistral-tiny";
case "open-mixtral-8x7b":
return "mistral-small";
default:
return "mistral-tiny";
}