llama-index

1
from typing import Dict
2

3
MISTRALAI_MODELS: Dict[str, int] = {
4
    "mistral-tiny": 32000,
5
    "mistral-small": 32000,
6
    "mistral-medium": 32000,
7
}
8

9

10
def mistralai_modelname_to_contextsize(modelname: str) -> int:
11
    if modelname not in MISTRALAI_MODELS:
12
        raise ValueError(
13
            f"Unknown model: {modelname}. Please provide a valid MistralAI model name."
14
            "Known models are: " + ", ".join(MISTRALAI_MODELS.keys())
15
        )
16

17
    return MISTRALAI_MODELS[modelname]
18