style(agent): 调整agent结构

This commit is contained in:
2026-05-18 05:33:11 +00:00
parent 6f6879dfab
commit a83c5fa5bd
5 changed files with 93 additions and 145 deletions
+4 -1
View File
@@ -19,6 +19,8 @@ from pydantic import BaseModel
from ulid import ULID
import asyncio
from kilostar.utils.access import Accessor, TokenData
from kilostar.utils.check_user.role_check import RoleChecker
from kilostar.core.postgres_database.model import UserAuthority
workflow_router = APIRouter(prefix="/api/v1/workflow", tags=["workflow"])
@@ -54,7 +56,8 @@ async def create_workflow(
@workflow_router.get("/list")
async def get_workflow_list(token_data: TokenData = Depends(Accessor.get_current_user)):
async def get_workflow_list(
token_data: TokenData = Depends(RoleChecker(allowed_roles=UserAuthority.USER))):
postgres_database = ray_actor_hook("postgres_database").postgres_database
workflows = await postgres_database.list_workflows.remote(
user_id=token_data.user_id