zhaoxi
/
Pretor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wip: 修复了post /pretor/api/provider的返回值Bug

This commit is contained in:
朝夕 2026-04-25 10:32:46 +08:00
parent e6fcb08ce3
commit bfdb0db933
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pretor/api/provider.py
View File

@ -42,9 +42,10 @@ async def create_provider(provider_register: ProviderRegister,
@provider_router.get("/list") @provider_router.get("/list")
async def get_provider_list(_: TokenData = Depends(Accessor.get_current_user)) -> Dict[str, Provider]: async def get_provider_list(_: TokenData = Depends(Accessor.get_current_user)) -> Dict[str, Dict[str, Provider]]:
global_state_machine = ray_actor_hook("global_state_machine").global_state_machine global_state_machine = ray_actor_hook("global_state_machine").global_state_machine
provider_list: Dict[str, Provider] = await global_state_machine.provider_manager.remote("get_provider_list") provider_list: Dict[str, Provider] = await global_state_machine.provider_manager.remote("get_provider_list")
providers = list(provider_list.values()) if provider_list else []
return {"provider_list": provider_list} return {"provider_list": provider_list}
@provider_router.delete("/{provider_title}") @provider_router.delete("/{provider_title}")

3
pretor/core/global_state_machine/provider_manager.py
View File

@ -53,11 +53,8 @@ class ProviderManager:
logger.warning(f"Provider type {provider_type} is not supported.") logger.warning(f"Provider type {provider_type} is not supported.")
return None return None
provider: Provider = await provider_class.create_model(provider_args) provider: Provider = await provider_class.create_model(provider_args)
provider.provider_owner = provider_owner provider.provider_owner = provider_owner
self.provider_register[provider_title] = provider self.provider_register[provider_title] = provider
await postgres_database.provider_database.remote("add_provider", provider_title=provider.provider_title, await postgres_database.provider_database.remote("add_provider", provider_title=provider.provider_title,
provider_url=provider.provider_url, provider_url=provider.provider_url,
provider_apikey=provider.provider_apikey, provider_apikey=provider.provider_apikey,