mirror of
https://github.com/langgenius/dify.git
synced 2026-06-13 04:01:12 +08:00
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
from typing import override
|
|
|
|
from services.workflow.entities import WorkflowScheduleCFSPlanEntity
|
|
from services.workflow.scheduler import CFSPlanScheduler, SchedulerCommand
|
|
from tasks.workflow_cfs_scheduler.entities import AsyncWorkflowQueue
|
|
|
|
|
|
class AsyncWorkflowCFSPlanEntity(WorkflowScheduleCFSPlanEntity):
|
|
"""
|
|
Trigger workflow CFS plan entity.
|
|
"""
|
|
|
|
queue: AsyncWorkflowQueue
|
|
|
|
|
|
class AsyncWorkflowCFSPlanScheduler(CFSPlanScheduler[AsyncWorkflowCFSPlanEntity]):
|
|
"""
|
|
Trigger workflow CFS plan scheduler.
|
|
"""
|
|
|
|
@override
|
|
def can_schedule(self) -> SchedulerCommand:
|
|
"""
|
|
Check if the workflow can be scheduled.
|
|
"""
|
|
if self.plan.queue in [AsyncWorkflowQueue.PROFESSIONAL_QUEUE, AsyncWorkflowQueue.TEAM_QUEUE]:
|
|
"""
|
|
permitted all paid users to schedule the workflow any time
|
|
"""
|
|
return SchedulerCommand.NONE
|
|
|
|
# FIXME: avoid the sandbox user's workflow at a running state for ever
|
|
return SchedulerCommand.RESOURCE_LIMIT_REACHED
|