refactor(api): switch to dify_config with Pydantic in controllers and schedule (#6237)

This commit is contained in:
天魂
2024-07-12 16:51:43 +08:00
committed by GitHub
parent a9ee52f2d7
commit 1df71ec64d
7 changed files with 17 additions and 16 deletions

View File

@@ -2,10 +2,10 @@ import datetime
import time
import click
from flask import current_app
from werkzeug.exceptions import NotFound
import app
from configs import dify_config
from extensions.ext_database import db
from models.dataset import Embedding
@@ -13,7 +13,7 @@ from models.dataset import Embedding
@app.celery.task(queue='dataset')
def clean_embedding_cache_task():
click.echo(click.style('Start clean embedding cache.', fg='green'))
clean_days = int(current_app.config.get('CLEAN_DAY_SETTING'))
clean_days = int(dify_config.CLEAN_DAY_SETTING)
start_at = time.perf_counter()
thirty_days_ago = datetime.datetime.now() - datetime.timedelta(days=clean_days)
page = 1