存档

kilostar/api/agent.py

@@ -192,24 +192,34 @@ async def create_worker_individual(
 async def get_worker_individual_list(
     token_data: TokenData = Depends(Accessor.get_current_user),
 ):
-    """列出当前登录用户名下的全部 Worker Agent。"""
+    """列出当前登录用户名下的全部 Worker Agent，并附加所有 plugin_owned slot。
+
+    plugin_owned slot 是插件登记的"占位 agent"，所有用户共享同一份配置，
+    在前端展示时会用徽标标记，并允许任何登录用户装配 provider/model。
+    """
     postgres_database = ray_actor_hook("postgres_database").postgres_database
     workers = await postgres_database.get_worker_individual_list.remote(
         owner_id=token_data.user_id
-    )
-    return {"workers": workers}
+    ) or []
+    all_workers = await postgres_database.get_all_worker_individual.remote() or []
+    seen_ids = {w.agent_id for w in workers}
+    plugin_slots = [
+        w for w in all_workers
+        if getattr(w, "plugin_owned", None) and w.agent_id not in seen_ids
+    ]
+    return {"workers": list(workers) + plugin_slots}
 
 
 @agent_router.get("/worker/{agent_id}")
 async def get_worker_individual(
     agent_id: str, token_data: TokenData = Depends(Accessor.get_current_user)
 ):
-    """按 ``agent_id`` 查询 Worker Agent；非本人的 Agent 返回 403。"""
+    """按 ``agent_id`` 查询 Worker Agent；非本人的 Agent 返回 403（plugin_owned slot 例外）。"""
     postgres_database = ray_actor_hook("postgres_database").postgres_database
     worker = await postgres_database.get_worker_individual.remote(agent_id=agent_id)
     if not worker:
         raise HTTPException(status_code=404, detail="Agent not found")
-    if worker.owner_id != token_data.user_id:
+    if not getattr(worker, "plugin_owned", None) and worker.owner_id != token_data.user_id:
         raise HTTPException(
             status_code=403, detail="Forbidden: You do not own this agent"
         )
@@ -227,7 +237,8 @@ async def update_worker_individual(
     worker = await postgres_database.get_worker_individual.remote(agent_id=agent_id)
     if not worker:
         raise HTTPException(status_code=404, detail="Agent not found")
-    if worker.owner_id != token_data.user_id:
+    # plugin_owned slot：任何登录用户都能装配 provider/model；普通 worker 仅 owner 可改
+    if not getattr(worker, "plugin_owned", None) and worker.owner_id != token_data.user_id:
         raise HTTPException(
             status_code=403, detail="Forbidden: You do not own this agent"
         )
@@ -243,6 +254,14 @@ async def update_worker_individual(
     except Exception:
         pass
 
+    # plugin_owned 时顺带触发对应插件的 reload，让新 provider/model 立刻生效
+    if getattr(worker, "plugin_owned", None):
+        try:
+            pm = ray_actor_hook("global_plugin_manager").global_plugin_manager
+            await pm.reload.remote(worker.plugin_owned)
+        except Exception:
+            pass
+
     return {"message": "success", "worker": updated_worker}