AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 07:29:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(api): simplify limit retrieval and return types (#9641)

Browse Source
This commit is contained in:
-LAN- 2024-10-22 16:34:16 +08:00 committed by GitHub
parent 5f12616cb9
commit 36f66d40e5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/controllers/console/datasets/file.py
View File

@ -30,13 +30,12 @@ class FileApi(Resource):
@account_initialization_required @account_initialization_required
@marshal_with(upload_config_fields) @marshal_with(upload_config_fields)
def get(self): def get(self):
file_size_limit = dify_config.UPLOAD_FILE_SIZE_LIMIT
batch_count_limit = dify_config.UPLOAD_FILE_BATCH_LIMIT
image_file_size_limit = dify_config.UPLOAD_IMAGE_FILE_SIZE_LIMIT
return { return {
"file_size_limit": file_size_limit, "file_size_limit": dify_config.UPLOAD_FILE_SIZE_LIMIT,
"batch_count_limit": batch_count_limit, "batch_count_limit": dify_config.UPLOAD_FILE_BATCH_LIMIT,
"image_file_size_limit": image_file_size_limit, "image_file_size_limit": dify_config.UPLOAD_IMAGE_FILE_SIZE_LIMIT,
"video_file_size_limit": dify_config.UPLOAD_VIDEO_FILE_SIZE_LIMIT,
"audio_file_size_limit": dify_config.UPLOAD_AUDIO_FILE_SIZE_LIMIT,
}, 200 }, 200
@setup_required @setup_required

4
api/core/app/task_pipeline/based_generate_task_pipeline.py
View File

@ -53,7 +53,7 @@ class BasedGenerateTaskPipeline:
self._output_moderation_handler = self._init_output_moderation() self._output_moderation_handler = self._init_output_moderation()
self._stream = stream self._stream = stream
def _handle_error(self, event: QueueErrorEvent, message: Optional[Message] = None) -> Exception: def _handle_error(self, event: QueueErrorEvent, message: Optional[Message] = None):
""" """
Handle error event. Handle error event.
:param event: event :param event: event
@ -100,7 +100,7 @@ class BasedGenerateTaskPipeline:
return message return message
def _error_to_stream_response(self, e: Exception) -> ErrorStreamResponse: def _error_to_stream_response(self, e: Exception):
""" """
Error to stream response. Error to stream response.
:param e: exception :param e: exception