style(agent): 调整agent结构
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user