|
|
|
@@ -27,6 +27,7 @@ from core.app.task_pipeline.easy_ui_based_generate_task_pipeline import EasyUIBa |
|
|
|
from core.prompt.utils.prompt_template_parser import PromptTemplateParser |
|
|
|
from extensions.ext_database import db |
|
|
|
from models import Account |
|
|
|
from models.enums import CreatedByRole |
|
|
|
from models.model import App, AppMode, AppModelConfig, Conversation, EndUser, Message, MessageFile |
|
|
|
from services.errors.app_model_config import AppModelConfigBrokenError |
|
|
|
from services.errors.conversation import ConversationCompletedError, ConversationNotExistsError |
|
|
|
@@ -240,7 +241,7 @@ class MessageBasedAppGenerator(BaseAppGenerator): |
|
|
|
belongs_to="user", |
|
|
|
url=file.remote_url, |
|
|
|
upload_file_id=file.related_id, |
|
|
|
created_by_role=("account" if account_id else "end_user"), |
|
|
|
created_by_role=(CreatedByRole.ACCOUNT if account_id else CreatedByRole.END_USER), |
|
|
|
created_by=account_id or end_user_id or "", |
|
|
|
) |
|
|
|
db.session.add(message_file) |