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

data/plugin/data_analytics/api.py

@@ -14,7 +14,7 @@ from fastapi.responses import StreamingResponse
 from pydantic import BaseModel, Field
 
 from kilostar.utils.access import Accessor, TokenData
-from kilostar.utils.ray_hook import ray_actor_hook
+from kilostar.utils.actor import get_actor
 
 router = APIRouter(tags=["data_analytics"])
 
@@ -40,7 +40,7 @@ class JobCreate(BaseModel):
 
 def _get_org():
     try:
-        return ray_actor_hook("org_data_analytics").org_data_analytics
+        return get_actor("org_data_analytics")
     except Exception as e:
         raise HTTPException(503, f"data_analytics 插件未就绪：{e}")
 
@@ -53,7 +53,7 @@ async def list_credentials(
     token_data: TokenData = Depends(Accessor.get_current_user),
 ):
     org = _get_org()
-    rows = await org.cred_list.remote(token_data.username)
+    rows = await org.cred_list(token_data.username)
     return {"credentials": rows}
 
 
@@ -63,7 +63,7 @@ async def create_credential(
     token_data: TokenData = Depends(Accessor.get_current_user),
 ):
     org = _get_org()
-    row = await org.cred_create.remote(
+    row = await org.cred_create(
         user_id=token_data.username,
         display_name=payload.display_name,
         access_key=payload.access_key,
@@ -80,7 +80,7 @@ async def delete_credential(
     token_data: TokenData = Depends(Accessor.get_current_user),
 ):
     org = _get_org()
-    ok = await org.cred_delete.remote(cred_id, token_data.username)
+    ok = await org.cred_delete(cred_id, token_data.username)
     if not ok:
         raise HTTPException(404, "凭证不存在或不属于当前用户")
     return {"status": "ok"}
@@ -96,7 +96,7 @@ async def create_job(
 ):
     org = _get_org()
     try:
-        return await org.job_create.remote(
+        return await org.job_create(
             user_id=token_data.username,
             cred_id=payload.cred_id,
             description=payload.description,
@@ -110,7 +110,7 @@ async def list_jobs(
     token_data: TokenData = Depends(Accessor.get_current_user),
 ):
     org = _get_org()
-    rows = await org.job_list.remote(token_data.username)
+    rows = await org.job_list(token_data.username)
     return {"jobs": rows}
 
 
@@ -120,7 +120,7 @@ async def get_job(
     token_data: TokenData = Depends(Accessor.get_current_user),
 ):
     org = _get_org()
-    row = await org.job_get.remote(job_id, token_data.username)
+    row = await org.job_get(job_id, token_data.username)
     if row is None:
         raise HTTPException(404, "任务不存在")
     return row
@@ -135,7 +135,7 @@ async def stream_job(
     import json
 
     org = _get_org()
-    row = await org.job_get.remote(job_id, token_data.username)
+    row = await org.job_get(job_id, token_data.username)
     if row is None:
         raise HTTPException(404, "任务不存在")
     org_task_id = row.get("org_task_id")
@@ -143,7 +143,7 @@ async def stream_job(
         raise HTTPException(409, "任务尚未投递到 organization")
 
     async def _generate():
-        async for event in await org.stream.remote(org_task_id):
+        async for event in await org.stream(org_task_id):
             payload = event if isinstance(event, str) else json.dumps(event, ensure_ascii=False)
             yield f"data: {payload}\n\n"