Add HF token to env when spawing via ui

This commit is contained in:
Jaret Burkett
2025-02-23 14:52:12 -07:00
parent 56d8d6bd81
commit 3c7daf49f3
2 changed files with 35 additions and 8 deletions

View File

@@ -29,7 +29,6 @@ export const getDatasetsRoot = async () => {
return datasetsPath as string;
};
export const getTrainingFolder = async () => {
const key = 'TRAINING_FOLDER';
let trainingRoot = myCache.get(key) as string;
@@ -48,3 +47,22 @@ export const getTrainingFolder = async () => {
myCache.set(key, trainingRoot);
return trainingRoot as string;
};
export const getHFToken = async () => {
const key = 'HF_TOKEN';
let token = myCache.get(key) as string;
if (token) {
return token;
}
let row = await prisma.settings.findFirst({
where: {
key: key,
},
});
token = '';
if (row?.value && row.value !== '') {
token = row.value;
}
myCache.set(key, token);
return token;
};