Elgato_dark/void
1
0
Fork 0
mirror of https://github.com/voideditor/void synced 2026-05-22 08:58:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor model mapping logic in RefreshModelService to use flatMap for improved handling of provider-specific model IDs, including conditional filtering for Mistral models.

Browse source
This commit is contained in:
Jérôme Commaret 2026-04-24 03:43:12 +02:00
parent 03305a17e3
commit 4a4cffcfe3
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/vs/workbench/contrib/void/common/refreshModelService.ts
View file

@ -176,14 +176,17 @@ export class RefreshModelService extends Disposable implements IRefreshModelServ
// set the models to the detected models
this.voidSettingsService.setAutodetectedModels(
providerName,
models.map(model => {
if (providerName === 'ollama') return (model as OllamaModelResponse).name;
else if (providerName === 'vLLM') return (model as OpenaiCompatibleModelResponse).id;
else if (providerName === 'lmStudio') return (model as OpenaiCompatibleModelResponse).id;
else if (providerName === 'apple') return (model as OpenaiCompatibleModelResponse).id;
else if (providerName === 'mistral') return (model as OpenaiCompatibleModelResponse).id;
else throw new Error('refreshMode fn: unknown provider', providerName);
}),
models.flatMap(model => {
if (providerName === 'ollama') return [(model as OllamaModelResponse).name];
else if (providerName === 'vLLM') return [(model as OpenaiCompatibleModelResponse).id];
else if (providerName === 'lmStudio') return [(model as OpenaiCompatibleModelResponse).id];
else if (providerName === 'apple') return [(model as OpenaiCompatibleModelResponse).id];
else if (providerName === 'mistral') {
const id = (model as OpenaiCompatibleModelResponse).id;
return id.endsWith('-latest') ? [id] : [];
}
else throw new Error('refreshMode fn: unknown provider', providerName);
}),
{ enableProviderOnSuccess: options.enableProviderOnSuccess, hideRefresh: options.doNotFire }
)