estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Feat/2070 glm 4 and glm 3 turbo (#2114)

Browse Source
This commit is contained in:
crazywoola
2024-01-21 16:58:06 +08:00
committed by GitHub
parent e19ad023d2
commit 8438d820ad
6 changed files with 92 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/file/message_file_parser.py
View File

@@ -2,7 +2,7 @@ from typing import Dict, List, Optional, Union
import requests
from core.file.file_obj import FileObj, FileTransferMethod, FileType
from core.file.upload_file_parser import SUPPORT_EXTENSIONS
from services.file_service import IMAGE_EXTENSIONS
from extensions.ext_database import db
from models.account import Account
from models.model import AppModelConfig, EndUser, MessageFile, UploadFile
@@ -83,7 +83,7 @@ class MessageFileParser:
UploadFile.tenant_id == self.tenant_id,
UploadFile.created_by == user.id,
UploadFile.created_by_role == ('account' if isinstance(user, Account) else 'end_user'),
UploadFile.extension.in_(SUPPORT_EXTENSIONS)
UploadFile.extension.in_(IMAGE_EXTENSIONS)
).first())
# check upload file is belong to tenant and user