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