diff --git a/public/scripts/openai.js b/public/scripts/openai.js index 4fce97258..9d01a4a13 100644 --- a/public/scripts/openai.js +++ b/public/scripts/openai.js @@ -4430,20 +4430,17 @@ async function onModelChange() { if (oai_settings.chat_completion_source === chat_completion_sources.MISTRALAI) { if (oai_settings.max_context_unlocked) { $('#openai_max_context').attr('max', unlocked_max); - } else if (['codestral-latest', 'codestral-mamba-2407', 'codestral-2411-rc5', 'codestral-2412', 'codestral-2501'].includes(oai_settings.mistralai_model)) { - $('#openai_max_context').attr('max', max_256k); - } else if (['mistral-large-2407', 'mistral-large-2411', 'mistral-large-pixtral-2411', 'mistral-large-latest'].includes(oai_settings.mistralai_model)) { - $('#openai_max_context').attr('max', max_128k); - } else if (oai_settings.mistralai_model.includes('mistral-nemo')) { - $('#openai_max_context').attr('max', max_128k); - } else if (oai_settings.mistralai_model.includes('mixtral-8x22b')) { - $('#openai_max_context').attr('max', max_64k); - } else if (oai_settings.mistralai_model.includes('pixtral')) { - $('#openai_max_context').attr('max', max_128k); + } else if (["codestral-2411-rc5", "codestral-2412", "codestral-2501", "codestral-latest", "codestral-mamba-2407", "codestral-mamba-latest", "open-codestral-mamba"].includes(oai_settings.mistralai_model)) { + $('#openai_max_context').attr('max', 262144); + } else if (["ministral-3b-2410", "ministral-3b-latest", "ministral-8b-2410", "ministral-8b-latest", "mistral-large-2407", "mistral-large-2411", "mistral-large-latest", "mistral-large-pixtral-2411", "mistral-tiny-2407", "mistral-tiny-latest", "open-mistral-nemo", "open-mistral-nemo-2407", "pixtral-12b", "pixtral-12b-2409", "pixtral-12b-latest", "pixtral-large-2411", "pixtral-large-latest"].includes(oai_settings.mistralai_model)) { + $('#openai_max_context').attr('max', 131072); + } else if (["open-mixtral-8x22b", "open-mixtral-8x22b-2404"].includes(oai_settings.mistralai_model)) { + $('#openai_max_context').attr('max', 65536); } else if (oai_settings.mistralai_model.includes('ministral')) { - $('#openai_max_context').attr('max', max_32k); + } else if (["codestral-2405", "mistral-embed", "mistral-large-2402", "mistral-medium", "mistral-medium-2312", "mistral-medium-latest", "mistral-moderation-2411", "mistral-moderation-latest", "mistral-ocr-2503", "mistral-ocr-latest", "mistral-saba-2502", "mistral-saba-latest", "mistral-small", "mistral-small-2312", "mistral-small-2402", "mistral-small-2409", "mistral-small-2501", "mistral-small-2503", "mistral-small-latest", "mistral-tiny", "mistral-tiny-2312", "open-mistral-7b", "open-mixtral-8x7b"].includes(oai_settings.mistralai_model)) { + $('#openai_max_context').attr('max', 32768); } else { - $('#openai_max_context').attr('max', max_32k); + $('#openai_max_context').attr('max', 32768); } oai_settings.openai_max_context = Math.min(oai_settings.openai_max_context, Number($('#openai_max_context').attr('max'))); $('#openai_max_context').val(oai_settings.openai_max_context).trigger('input');