chore: 迁移服务器前同步本地修改

kilostar/api/provider.py

@@ -18,7 +18,7 @@ from typing import Any, Dict, Literal, Optional
 from kilostar.utils.access import TokenData, Accessor, RoleChecker
 from kilostar.core.postgres_database.model import UserAuthority
 from kilostar.core.global_state_machine.model_provider.base_provider import Provider
-from kilostar.utils.ray_hook import ray_actor_hook
+from kilostar.utils.actor import get_actor
 
 provider_router = APIRouter(prefix="/api/v1/provider", tags=["provider"])
 
@@ -39,8 +39,8 @@ async def create_provider(
     token_data: TokenData = Depends(RoleChecker(allowed_roles=UserAuthority.USER)),
 ) -> None:
     """注册一个 Provider；owner 为当前登录用户的 ``user_id``。"""
-    global_state_machine = ray_actor_hook("global_state_machine").global_state_machine
-    await global_state_machine.add_provider_wrap.remote(
+    global_state_machine = get_actor("global_state_machine")
+    await global_state_machine.add_provider_wrap(
         provider_type=provider_register.provider_type,
         provider_title=provider_register.provider_title,
         provider_url=provider_register.provider_url,
@@ -61,10 +61,10 @@ async def get_provider_list(
     _: TokenData = Depends(Accessor.get_current_user),
 ) -> Dict[str, Any]:
     """返回当前所有已注册的 Provider，前端用以展示模型清单。apikey 脱敏。"""
-    global_state_machine = ray_actor_hook("global_state_machine").global_state_machine
+    global_state_machine = get_actor("global_state_machine")
     provider_list: Dict[
         str, Provider
-    ] = await global_state_machine.get_provider_list.remote()
+    ] = await global_state_machine.get_provider_list()
     masked = {}
     for title, p in provider_list.items():
         d = p.model_dump() if hasattr(p, "model_dump") else dict(p)
@@ -138,6 +138,6 @@ async def delete_provider(
     ),
 ) -> dict:
     """删除指定 ``provider_title`` 的 Provider；仅超管可调用。"""
-    global_state_machine = ray_actor_hook("global_state_machine").global_state_machine
-    await global_state_machine.delete_provider.remote(provider_title=provider_title)
+    global_state_machine = get_actor("global_state_machine")
+    await global_state_machine.delete_provider(provider_title=provider_title)
     return {"message": "success"}