From 52e37893633ce27eb59c096fde007b71cb54c34f Mon Sep 17 00:00:00 2001 From: ytqh Date: Fri, 2 May 2025 22:38:01 +0800 Subject: [PATCH] feat: fix --- api/models/model.py | 9 +++++---- api/schedule/user_profile_generate_task.py | 5 +++-- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/api/models/model.py b/api/models/model.py index 82f6ff3b68..c9907b4d7a 100644 --- a/api/models/model.py +++ b/api/models/model.py @@ -15,18 +15,19 @@ if TYPE_CHECKING: from models.workflow import Workflow import sqlalchemy as sa +from flask import request +from flask_login import UserMixin # type: ignore +from sqlalchemy import Float, func, text +from sqlalchemy.orm import Mapped, Session, mapped_column + from configs import dify_config from constants import DEFAULT_FILE_NUMBER_LIMITS from core.file import FILE_MODEL_IDENTITY, File, FileTransferMethod, FileType from core.file import helpers as file_helpers -from flask import request -from flask_login import UserMixin # type: ignore from libs.helper import generate_string from models.base import Base from models.enums import CreatedByRole from models.workflow import WorkflowRunStatus -from sqlalchemy import Float, Index, PrimaryKeyConstraint, func, text -from sqlalchemy.orm import Mapped, Session, mapped_column from .account import Account, Tenant from .engine import db diff --git a/api/schedule/user_profile_generate_task.py b/api/schedule/user_profile_generate_task.py index 3b0d92f767..ea98c06cf7 100644 --- a/api/schedule/user_profile_generate_task.py +++ b/api/schedule/user_profile_generate_task.py @@ -1,13 +1,14 @@ import time from datetime import datetime -import app import click +from sqlalchemy import asc, func, or_ + +import app from configs import dify_config from core.app.entities.app_invoke_entities import InvokeFrom from models.model import App, EndUser, Message, db from services.app_generate_service import AppGenerateService -from sqlalchemy import asc, func, or_ @app.celery.task(queue="dataset")