Merge pull request #74 from JefferyHcool/feature/regenerate

fix(provider): 重新启用通过 ID 获取供应商信息的接口并增强安全性
This commit is contained in:
Jianwu Huang
2025-05-04 17:48:54 +08:00
committed by GitHub
2 changed files with 11 additions and 7 deletions

View File

@@ -52,13 +52,13 @@ def get_all_providers():
except Exception as e:
return R.error(msg=e)
# @router.get("/get_provider_by_id/{id}")
# def get_provider_by_id(id: str):
# try:
# res = ProviderService.get_provider_by_id(id)
# return R.success(data=res)
# except Exception as e:
# return R.error(msg=e)
@router.get("/get_provider_by_id/{id}")
def get_provider_by_id(id: str):
try:
res = ProviderService.get_provider_by_id_safe(id)
return R.success(data=res)
except Exception as e:
return R.error(msg=e)
#
# @router.get("/get_provider_by_name/{name}")
# def get_provider_by_name(name: str):

View File

@@ -75,6 +75,10 @@ class ProviderService:
row = get_provider_by_id(id)
return ProviderService.serialize_provider(row)
@staticmethod
def get_provider_by_id_safe(id: str): # 已改为 str 类型
row = get_provider_by_id(id)
return ProviderService.serialize_provider_safe(row)
# all_models.extend(provider['models'])
@staticmethod