jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: mypy for all type check (#10921)

Browse source
This commit is contained in:
yihong 2024-12-24 18:38:51 +08:00 • committed by GitHub
commit 56e15d09a9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
584 changed files with 3975 additions and 2826 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/core/file/file_manager.py
View file

@ -1,4 +1,5 @@
import base64
from collections.abc import Mapping
from configs import dify_config
from core.helper import ssrf_proxy
@ -55,7 +56,7 @@ def to_prompt_message_content(
if f.type == FileType.IMAGE:
params["detail"] = image_detail_config or ImagePromptMessageContent.DETAIL.LOW
prompt_class_map = {
prompt_class_map: Mapping[FileType, type[MultiModalPromptMessageContent]] = {
FileType.IMAGE: ImagePromptMessageContent,
FileType.AUDIO: AudioPromptMessageContent,
FileType.VIDEO: VideoPromptMessageContent,
@ -63,7 +64,7 @@ def to_prompt_message_content(
}
try:
return prompt_class_map[f.type](**params)
return prompt_class_map[f.type].model_validate(params)
except KeyError:
raise ValueError(f"file type {f.type} is not supported")

4
api/core/file/tool_file_parser.py
View file

@ -1,4 +1,4 @@
from typing import TYPE_CHECKING, Any
from typing import TYPE_CHECKING, Any, cast
if TYPE_CHECKING:
from core.tools.tool_file_manager import ToolFileManager
@ -9,4 +9,4 @@ tool_file_manager: dict[str, Any] = {"manager": None}
class ToolFileParser:
@staticmethod
def get_tool_file_manager() -> "ToolFileManager":
return tool_file_manager["manager"]
return cast("ToolFileManager", tool_file_manager["manager"])