fix: search file name by repo_id

This commit is contained in:
jialin
2024-06-15 20:41:26 +08:00
parent b416a14554
commit 2352c316c0
6 changed files with 98 additions and 14 deletions
+7 -7
View File
@@ -1,3 +1,4 @@
import { listFiles } from '@huggingface/hub';
import { request } from '@umijs/max';
import {
FormData,
@@ -115,11 +116,10 @@ export async function callHuggingfaceQuickSearch(params: any) {
});
}
export async function queryHuggingfaceModelFiles(params: any) {
return request(
`https://huggingface.co/openbmb/MiniCPM-Llama3-V-2_5-gguf/tree/main`,
{
method: 'GET'
}
);
export async function queryHuggingfaceModelFiles(params: { repo: string }) {
const result = [];
for await (const fileInfo of listFiles(params)) {
result.push(fileInfo);
}
return result;
}