should be SANDBOX_EXPIRED_RECORDS_CLEAN_GRACEFUL_PERIOD

This commit is contained in:
hjlarry 2025-12-17 14:54:48 +08:00
parent a80194fe06
commit fbeb35fbb5
3 changed files with 5 additions and 5 deletions

View File

@ -1152,12 +1152,12 @@ class CeleryScheduleTasksConfig(BaseSettings):
) )
class SandboxRecordsCleanConfig(BaseSettings): class SandboxExpiredRecordsCleanConfig(BaseSettings):
SANDBOX_RECORDS_CLEAN_GRACEFUL_PERIOD: NonNegativeInt = Field( SANDBOX_EXPIRED_RECORDS_CLEAN_GRACEFUL_PERIOD: NonNegativeInt = Field(
description="Graceful period in days for sandbox records clean after subscription expiration", description="Graceful period in days for sandbox records clean after subscription expiration",
default=21, default=21,
) )
SANDBOX_RECORDS_CLEAN_BATCH_SIZE: PositiveInt = Field( SANDBOX_EXPIRED_RECORDS_CLEAN_BATCH_SIZE: PositiveInt = Field(
description="Maximum number of records to process in each batch", description="Maximum number of records to process in each batch",
default=1000, default=1000,
) )

View File

@ -45,7 +45,7 @@ class WorkflowRunCleanup:
self.batch_size = batch_size self.batch_size = batch_size
self.billing_cache: dict[str, TenantPlanInfo | None] = {} self.billing_cache: dict[str, TenantPlanInfo | None] = {}
self.dry_run = dry_run self.dry_run = dry_run
self.free_plan_grace_period_days = dify_config.SANDBOX_RECORDS_CLEAN_GRACEFUL_PERIOD self.free_plan_grace_period_days = dify_config.SANDBOX_EXPIRED_RECORDS_CLEAN_GRACEFUL_PERIOD
self.workflow_run_repo: APIWorkflowRunRepository self.workflow_run_repo: APIWorkflowRunRepository
if workflow_run_repo: if workflow_run_repo:
self.workflow_run_repo = workflow_run_repo self.workflow_run_repo = workflow_run_repo

View File

@ -76,7 +76,7 @@ def create_cleanup(
) -> WorkflowRunCleanup: ) -> WorkflowRunCleanup:
monkeypatch.setattr( monkeypatch.setattr(
cleanup_module.dify_config, cleanup_module.dify_config,
"SANDBOX_RECORDS_CLEAN_GRACEFUL_PERIOD", "SANDBOX_EXPIRED_RECORDS_CLEAN_GRACEFUL_PERIOD",
grace_period_days, grace_period_days,
) )
return WorkflowRunCleanup(workflow_run_repo=repo, **kwargs) return WorkflowRunCleanup(workflow_run_repo=repo, **kwargs)