From 890b5d222fd714ebf0c76993dd48d7d02d7dfdc4 Mon Sep 17 00:00:00 2001 From: hjlarry Date: Fri, 12 Dec 2025 14:07:19 +0800 Subject: [PATCH] fix mypy --- api/services/clear_free_plan_expired_workflow_run_logs.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/api/services/clear_free_plan_expired_workflow_run_logs.py b/api/services/clear_free_plan_expired_workflow_run_logs.py index 17cce15311..1fe2bad2d0 100644 --- a/api/services/clear_free_plan_expired_workflow_run_logs.py +++ b/api/services/clear_free_plan_expired_workflow_run_logs.py @@ -40,6 +40,7 @@ class WorkflowRunCleanup: self.batch_size = batch_size self.billing_cache: dict[str, CloudPlan | None] = {} + self.workflow_run_repo: APIWorkflowRunRepository if workflow_run_repo: self.workflow_run_repo = workflow_run_repo else: @@ -47,9 +48,7 @@ class WorkflowRunCleanup: from repositories.factory import DifyAPIRepositoryFactory session_maker = sessionmaker(bind=db.engine, expire_on_commit=False) - self.workflow_run_repo: APIWorkflowRunRepository = ( - DifyAPIRepositoryFactory.create_api_workflow_run_repository(session_maker) - ) + self.workflow_run_repo = DifyAPIRepositoryFactory.create_api_workflow_run_repository(session_maker) def run(self) -> None: click.echo(