chore: encode url for hf
This commit is contained in:
@@ -14,6 +14,10 @@ export const MODELS_API = '/models';
|
|||||||
|
|
||||||
export const MODEL_INSTANCE_API = '/model-instances';
|
export const MODEL_INSTANCE_API = '/model-instances';
|
||||||
|
|
||||||
|
const setProxyUrl = (url: string) => {
|
||||||
|
return `/proxy?url=${encodeURIComponent(url)}`;
|
||||||
|
};
|
||||||
|
|
||||||
// ===================== Models =====================
|
// ===================== Models =====================
|
||||||
export async function queryModelsList(
|
export async function queryModelsList(
|
||||||
params: Global.SearchParams,
|
params: Global.SearchParams,
|
||||||
@@ -128,7 +132,8 @@ export async function queryHuggingfaceModelDetail(
|
|||||||
params: { repo: string },
|
params: { repo: string },
|
||||||
options?: any
|
options?: any
|
||||||
) {
|
) {
|
||||||
return request(`${HUGGINGFACE_API}/${params.repo}`, {
|
const url = `https://huggingface.co/api/models/${params.repo}`;
|
||||||
|
return request(setProxyUrl(url), {
|
||||||
method: 'GET',
|
method: 'GET',
|
||||||
cancelToken: options?.token
|
cancelToken: options?.token
|
||||||
});
|
});
|
||||||
@@ -228,12 +233,11 @@ export async function queryHuggingfaceModels(
|
|||||||
limit: 100,
|
limit: 100,
|
||||||
additionalFields: ['sha', 'tags'],
|
additionalFields: ['sha', 'tags'],
|
||||||
fetch(_url: string, config: any) {
|
fetch(_url: string, config: any) {
|
||||||
const url = `/proxy?url=${_url}`;
|
const url = params.search.sort
|
||||||
|
? `${_url}&sort=${params.search.sort}`
|
||||||
|
: _url;
|
||||||
try {
|
try {
|
||||||
const newUrl = params.search.sort
|
return fetch(setProxyUrl(url), {
|
||||||
? `${url}&sort=${params.search.sort}`
|
|
||||||
: url;
|
|
||||||
return fetch(`${newUrl}`, {
|
|
||||||
...config,
|
...config,
|
||||||
signal: options.signal
|
signal: options.signal
|
||||||
});
|
});
|
||||||
@@ -259,7 +263,7 @@ export async function queryHuggingfaceModelFiles(
|
|||||||
recursive: true,
|
recursive: true,
|
||||||
fetch(url: string, config: any) {
|
fetch(url: string, config: any) {
|
||||||
try {
|
try {
|
||||||
return fetch(`/proxy?url=${url}`, {
|
return fetch(setProxyUrl(url), {
|
||||||
...config,
|
...config,
|
||||||
signal: options?.signal
|
signal: options?.signal
|
||||||
});
|
});
|
||||||
@@ -286,7 +290,7 @@ export async function downloadModelFile(
|
|||||||
revision: revision,
|
revision: revision,
|
||||||
path: path,
|
path: path,
|
||||||
fetch(url: string, config: any) {
|
fetch(url: string, config: any) {
|
||||||
return fetch(`/proxy?url=${url}`, {
|
return fetch(setProxyUrl(url), {
|
||||||
...config,
|
...config,
|
||||||
signal: options?.signal
|
signal: options?.signal
|
||||||
});
|
});
|
||||||
|
|||||||
Reference in New Issue
Block a user