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

data/toolset/regulatory_toolset/query_task_list.py

@@ -9,7 +9,7 @@ regulatory_node 用以回答"我之前那份报告呢""昨天那个查询结果
 
 from typing import Any, Dict, List, Optional
 
-from kilostar.utils.ray_hook import ray_actor_hook
+from kilostar.utils.actor import get_actor
 
 
 async def query_task_list(
@@ -34,8 +34,8 @@ async def query_task_list(
             "total": int
         }
     """
-    pg = ray_actor_hook("postgres_database").postgres_database
-    rows: List[Dict[str, Any]] = await pg.list_tasks_by_user.remote(
+    pg = get_actor("postgres_database")
+    rows: List[Dict[str, Any]] = await pg.list_tasks_by_user(
         user_id=user_id,
         status=status_filter,
         limit=limit,

data/toolset/regulatory_toolset/query_workflow_status.py

@@ -6,7 +6,7 @@ regulatory_node 在与用户对话时，可以借此工具回答"我那个任务
 
 from typing import Any, Dict, List
 
-from kilostar.utils.ray_hook import ray_actor_hook
+from kilostar.utils.actor import get_actor
 
 
 async def query_workflow_status(trace_id: str, limit: int = 10) -> Dict[str, Any]:
@@ -26,10 +26,10 @@ async def query_workflow_status(trace_id: str, limit: int = 10) -> Dict[str, Any
             ]
         }
     """
-    pg = ray_actor_hook("postgres_database").postgres_database
+    pg = get_actor("postgres_database")
 
-    workflow = await pg.get_workflow.remote(trace_id)
-    events = await pg.query_event_logs.remote(trace_id=trace_id, limit=limit)
+    workflow = await pg.get_workflow(trace_id)
+    events = await pg.query_event_logs(trace_id=trace_id, limit=limit)
 
     recent: List[Dict[str, Any]] = []
     for e in events or []:

data/toolset/regulatory_toolset/send_file.py

@@ -12,7 +12,7 @@ import re
 import uuid
 from pathlib import Path
 
-from kilostar.utils.ray_hook import ray_actor_hook
+from kilostar.utils.actor import get_actor
 from kilostar.utils.settings import get_artifact_dir
 
 
@@ -56,8 +56,8 @@ async def send_file(filename: str, content: str, trace_id: str = "") -> str:
         },
         ensure_ascii=False,
     )
-    actor_list = ray_actor_hook("global_workflow_manager")
-    await actor_list.global_workflow_manager.put_pending.remote(
+    gwm = get_actor("global_workflow_manager")
+    await gwm.put_pending(
         trace_id, f"__FILE__{payload}"
     )
     return f"已发送文件: {safe_name}"