mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
c60c754ac9
Some checks are pending
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/amd64, build-api-amd64) (push) Waiting to run
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/arm64, build-api-arm64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/amd64, build-web-amd64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/arm64, build-web-arm64) (push) Waiting to run
Build and Push API & Web / create-manifest (api, DIFY_API_IMAGE_NAME, merge-api-images) (push) Blocked by required conditions
Build and Push API & Web / create-manifest (web, DIFY_WEB_IMAGE_NAME, merge-web-images) (push) Blocked by required conditions
Main CI Pipeline / Check Changed Files (push) Waiting to run
Main CI Pipeline / API Tests (push) Blocked by required conditions
Main CI Pipeline / Web Tests (push) Blocked by required conditions
Main CI Pipeline / Style Check (push) Waiting to run
Main CI Pipeline / VDB Tests (push) Blocked by required conditions
Main CI Pipeline / DB Migration Test (push) Blocked by required conditions
92 lines
2.3 KiB
Python
92 lines
2.3 KiB
Python
from flask_restx import Api, Namespace, fields
|
|
|
|
from libs.helper import TimestampField
|
|
|
|
upload_config_fields = {
|
|
"file_size_limit": fields.Integer,
|
|
"batch_count_limit": fields.Integer,
|
|
"image_file_size_limit": fields.Integer,
|
|
"video_file_size_limit": fields.Integer,
|
|
"audio_file_size_limit": fields.Integer,
|
|
"workflow_file_upload_limit": fields.Integer,
|
|
}
|
|
|
|
|
|
def build_upload_config_model(api_or_ns: Api | Namespace):
|
|
"""Build the upload config model for the API or Namespace.
|
|
|
|
Args:
|
|
api_or_ns: Flask-RestX Api or Namespace instance
|
|
|
|
Returns:
|
|
The registered model
|
|
"""
|
|
return api_or_ns.model("UploadConfig", upload_config_fields)
|
|
|
|
|
|
file_fields = {
|
|
"id": fields.String,
|
|
"name": fields.String,
|
|
"size": fields.Integer,
|
|
"extension": fields.String,
|
|
"mime_type": fields.String,
|
|
"created_by": fields.String,
|
|
"created_at": TimestampField,
|
|
"preview_url": fields.String,
|
|
"source_url": fields.String,
|
|
}
|
|
|
|
|
|
def build_file_model(api_or_ns: Api | Namespace):
|
|
"""Build the file model for the API or Namespace.
|
|
|
|
Args:
|
|
api_or_ns: Flask-RestX Api or Namespace instance
|
|
|
|
Returns:
|
|
The registered model
|
|
"""
|
|
return api_or_ns.model("File", file_fields)
|
|
|
|
|
|
remote_file_info_fields = {
|
|
"file_type": fields.String(attribute="file_type"),
|
|
"file_length": fields.Integer(attribute="file_length"),
|
|
}
|
|
|
|
|
|
def build_remote_file_info_model(api_or_ns: Api | Namespace):
|
|
"""Build the remote file info model for the API or Namespace.
|
|
|
|
Args:
|
|
api_or_ns: Flask-RestX Api or Namespace instance
|
|
|
|
Returns:
|
|
The registered model
|
|
"""
|
|
return api_or_ns.model("RemoteFileInfo", remote_file_info_fields)
|
|
|
|
|
|
file_fields_with_signed_url = {
|
|
"id": fields.String,
|
|
"name": fields.String,
|
|
"size": fields.Integer,
|
|
"extension": fields.String,
|
|
"url": fields.String,
|
|
"mime_type": fields.String,
|
|
"created_by": fields.String,
|
|
"created_at": TimestampField,
|
|
}
|
|
|
|
|
|
def build_file_with_signed_url_model(api_or_ns: Api | Namespace):
|
|
"""Build the file with signed URL model for the API or Namespace.
|
|
|
|
Args:
|
|
api_or_ns: Flask-RestX Api or Namespace instance
|
|
|
|
Returns:
|
|
The registered model
|
|
"""
|
|
return api_or_ns.model("FileWithSignedUrl", file_fields_with_signed_url)
|