Merge pull request #3619 from H0llyW00dzZ/fix_llm_issue

This commit is contained in:
Fred Liang 2023-12-25 03:32:02 +08:00 committed by GitHub
commit b9ca6694f7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -10,24 +10,23 @@ export function collectModelTable(
available: boolean; available: boolean;
name: string; name: string;
displayName: string; displayName: string;
provider: LLMModel["provider"]; provider?: LLMModel["provider"]; // Marked as optional
} }
> = {}; > = {};
// default models // default models
models.forEach( models.forEach((m) => {
(m) => modelTable[m.name] = {
(modelTable[m.name] = { ...m,
...m, displayName: m.name, // 'provider' is copied over if it exists
displayName: m.name, };
}), });
);
// server custom models // server custom models
customModels customModels
.split(",") .split(",")
.filter((v) => !!v && v.length > 0) .filter((v) => !!v && v.length > 0)
.map((m) => { .forEach((m) => {
const available = !m.startsWith("-"); const available = !m.startsWith("-");
const nameConfig = const nameConfig =
m.startsWith("+") || m.startsWith("-") ? m.slice(1) : m; m.startsWith("+") || m.startsWith("-") ? m.slice(1) : m;
@ -35,15 +34,15 @@ export function collectModelTable(
// enable or disable all models // enable or disable all models
if (name === "all") { if (name === "all") {
Object.values(modelTable).forEach((m) => (m.available = available)); Object.values(modelTable).forEach((model) => (model.available = available));
} else {
modelTable[name] = {
name,
displayName: displayName || name,
available,
provider: modelTable[name]?.provider, // Use optional chaining
};
} }
modelTable[name] = {
name,
displayName: displayName || name,
available,
provider: modelTable[name].provider,
};
}); });
return modelTable; return modelTable;
} }