chore: list models query

This commit is contained in:
jialin
2024-08-09 10:26:46 +08:00
parent ab8483fc42
commit 8490e32ef1
14 changed files with 669 additions and 114 deletions
+23 -2
View File
@@ -116,6 +116,14 @@ export async function callHuggingfaceQuickSearch(params: any) {
});
}
const HUGGINGFACE_API = 'https://huggingface.co/api/models';
export async function queryHuggingfaceModelDetail(params: { repo: string }) {
return request(`${HUGGINGFACE_API}/${params.repo}`, {
method: 'GET'
});
}
export async function queryHuggingfaceModels(
params: {
search: {
@@ -130,7 +138,18 @@ export async function queryHuggingfaceModels(
...params,
...options,
limit: 50,
additionalFields: ['cardData']
additionalFields: ['sha'],
fetch(url: string, config: any) {
try {
return fetch(`${url}`, {
...config,
signal: options.signal
});
} catch (error) {
// ignore
return [];
}
}
})) {
result.push(model);
}
@@ -139,7 +158,9 @@ export async function queryHuggingfaceModels(
export async function queryHuggingfaceModelFiles(params: { repo: string }) {
const result = [];
for await (const fileInfo of listFiles(params)) {
for await (const fileInfo of listFiles({
...params
})) {
result.push(fileInfo);
}
return result;