mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-04-22 05:39:42 +08:00
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
import mimetypes
|
|
import os
|
|
import re
|
|
import urllib.parse
|
|
from collections.abc import Mapping
|
|
from typing import Any
|
|
from uuid import uuid4
|
|
|
|
import httpx
|
|
from pydantic import BaseModel
|
|
|
|
from configs import dify_config
|
|
|
|
|
|
class FileInfo(BaseModel):
|
|
filename: str
|
|
extension: str
|
|
mimetype: str
|
|
size: int
|
|
|
|
|
|
def guess_file_info_from_response(response: httpx.Response):
|
|
url = str(response.url)
|
|
# Try to extract filename from URL
|
|
parsed_url = urllib.parse.urlparse(url)
|
|
url_path = parsed_url.path
|
|
filename = os.path.basename(url_path)
|
|
|
|
# If filename couldn't be extracted, use Content-Disposition header
|
|
if not filename:
|
|
content_disposition = response.headers.get("Content-Disposition")
|
|
if content_disposition:
|
|
filename_match = re.search(r'filename="?(.+)"?', content_disposition)
|
|
if filename_match:
|
|
filename = filename_match.group(1)
|
|
|
|
# If still no filename, generate a unique one
|
|
if not filename:
|
|
unique_name = str(uuid4())
|
|
filename = f"{unique_name}"
|
|
|
|
# Guess MIME type from filename first, then URL
|
|
mimetype, _ = mimetypes.guess_type(filename)
|
|
if mimetype is None:
|
|
mimetype, _ = mimetypes.guess_type(url)
|
|
if mimetype is None:
|
|
# If guessing fails, use Content-Type from response headers
|
|
mimetype = response.headers.get("Content-Type", "application/octet-stream")
|
|
|
|
extension = os.path.splitext(filename)[1]
|
|
|
|
# Ensure filename has an extension
|
|
if not extension:
|
|
extension = mimetypes.guess_extension(mimetype) or ".bin"
|
|
filename = f"{filename}{extension}"
|
|
|
|
return FileInfo(
|
|
filename=filename,
|
|
extension=extension,
|
|
mimetype=mimetype,
|
|
size=int(response.headers.get("Content-Length", -1)),
|
|
)
|
|
|
|
|
|
def get_parameters_from_feature_dict(*, features_dict: Mapping[str, Any], user_input_form: list[dict[str, Any]]):
|
|
return {
|
|
"opening_statement": features_dict.get("opening_statement"),
|
|
"suggested_questions": features_dict.get("suggested_questions", []),
|
|
"suggested_questions_after_answer": features_dict.get("suggested_questions_after_answer", {"enabled": False}),
|
|
"speech_to_text": features_dict.get("speech_to_text", {"enabled": False}),
|
|
"text_to_speech": features_dict.get("text_to_speech", {"enabled": False}),
|
|
"retriever_resource": features_dict.get("retriever_resource", {"enabled": False}),
|
|
"annotation_reply": features_dict.get("annotation_reply", {"enabled": False}),
|
|
"more_like_this": features_dict.get("more_like_this", {"enabled": False}),
|
|
"user_input_form": user_input_form,
|
|
"sensitive_word_avoidance": features_dict.get(
|
|
"sensitive_word_avoidance", {"enabled": False, "type": "", "configs": []}
|
|
),
|
|
"file_upload": features_dict.get(
|
|
"file_upload",
|
|
{
|
|
"image": {
|
|
"enabled": False,
|
|
"number_limits": 3,
|
|
"detail": "high",
|
|
"transfer_methods": ["remote_url", "local_file"],
|
|
}
|
|
},
|
|
),
|
|
"system_parameters": {
|
|
"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,
|
|
"file_size_limit": dify_config.UPLOAD_FILE_SIZE_LIMIT,
|
|
"workflow_file_upload_limit": dify_config.WORKFLOW_FILE_UPLOAD_LIMIT,
|
|
},
|
|
}
|