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):
SANDBOX_RECORDS_CLEAN_GRACEFUL_PERIOD: NonNegativeInt = Field(
class SandboxExpiredRecordsCleanConfig(BaseSettings):
SANDBOX_EXPIRED_RECORDS_CLEAN_GRACEFUL_PERIOD: NonNegativeInt = Field(
description="Graceful period in days for sandbox records clean after subscription expiration",
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",
default=1000,
)

View File

@ -45,7 +45,7 @@ class WorkflowRunCleanup:
self.batch_size = batch_size
self.billing_cache: dict[str, TenantPlanInfo | None] = {}
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
if workflow_run_repo:
self.workflow_run_repo = workflow_run_repo

View File

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