Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix some bugs in HubApi (#1369)

Browse Source
This commit is contained in:
Xingjun.Wang
2025-06-19 10:36:44 +08:00
committed by GitHub
parent 86dd6c3434
commit c9d0c46412
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modelscope/hub/api.py
View File

@@ -32,6 +32,7 @@ from modelscope.hub.constants import (API_HTTP_CLIENT_MAX_RETRIES,
API_RESPONSE_FIELD_DATA, API_RESPONSE_FIELD_DATA,
API_RESPONSE_FIELD_EMAIL, API_RESPONSE_FIELD_EMAIL,
API_RESPONSE_FIELD_GIT_ACCESS_TOKEN, API_RESPONSE_FIELD_GIT_ACCESS_TOKEN,
API_RESPONSE_FIELD_MESSAGE,
API_RESPONSE_FIELD_USERNAME, API_RESPONSE_FIELD_USERNAME,
DEFAULT_CREDENTIALS_PATH, DEFAULT_CREDENTIALS_PATH,
DEFAULT_MAX_WORKERS, DEFAULT_MAX_WORKERS,
@@ -372,6 +373,7 @@ class HubApi:
set in HubApi class (self.endpoint) set in HubApi class (self.endpoint)
re_raise(`bool`): re_raise(`bool`):
raise exception when error raise exception when error
token (`str`, *optional*): access token to use for checking existence.
Returns: Returns:
True if the repository exists, False otherwise. True if the repository exists, False otherwise.
""" """
@@ -1426,6 +1428,7 @@ class HubApi:
endpoint (Optional[str]): The endpoint to use. endpoint (Optional[str]): The endpoint to use.
In the format of `https://www.modelscope.cn` or 'https://www.modelscope.ai' In the format of `https://www.modelscope.cn` or 'https://www.modelscope.ai'
exist_ok (Optional[bool]): If the repo exists, whether to return the repo url directly. exist_ok (Optional[bool]): If the repo exists, whether to return the repo url directly.
create_default_config (Optional[bool]): If True, create a default configuration file in the model repo.
**kwargs: The additional arguments. **kwargs: The additional arguments.
Returns: Returns:
@@ -1437,15 +1440,15 @@ class HubApi:
if not endpoint: if not endpoint:
endpoint = self.endpoint endpoint = self.endpoint
repo_exists: bool = self.repo_exists(repo_id, repo_type=repo_type, endpoint=endpoint) self.login(access_token=token, endpoint=endpoint)
repo_exists: bool = self.repo_exists(repo_id, repo_type=repo_type, endpoint=endpoint, token=token)
if repo_exists: if repo_exists:
if exist_ok: if exist_ok:
return f'{endpoint}/{repo_type}s/{repo_id}' return f'{endpoint}/{repo_type}s/{repo_id}'
else: else:
raise ValueError(f'Repo {repo_id} already exists!') raise ValueError(f'Repo {repo_id} already exists!')
self.login(access_token=token, endpoint=endpoint)
repo_id_list = repo_id.split('/') repo_id_list = repo_id.split('/')
if len(repo_id_list) != 2: if len(repo_id_list) != 2:
raise ValueError('Invalid repo id, should be in the format of `owner_name/repo_name`') raise ValueError('Invalid repo id, should be in the format of `owner_name/repo_name`')