禁止管理员外用户使用某些模型

This commit is contained in:
sijinhui
2024-07-01 17:53:59 +08:00
parent 7603b21cbc
commit 9bab4b736f
3 changed files with 10 additions and 3 deletions

View File

@@ -7,7 +7,7 @@ export function useAllModels() {
const accessStore = useAccessStore();
const configStore = useAppConfig();
const { data: session, status } = useSession();
console.log("-sssssssssss", session, status);
const models = useMemo(() => {
return collectModelsWithDefaultModel(
configStore.models,
@@ -22,5 +22,10 @@ export function useAllModels() {
configStore.dontUseModel,
]);
// @ts-expect-error
if (status === "authenticated" && !session?.user?.isAdmin) {
// 过滤非管理员用户可使用的模型
return models.filter((m) => !m.name.endsWith("-all"));
}
return models;
}