mirror of
https://github.com/langgenius/dify.git
synced 2026-02-24 20:01:22 +08:00
feat: mypy for all type check (#10921)
This commit is contained in:
@ -1,6 +1,6 @@
|
||||
import logging
|
||||
from collections.abc import Callable, Generator
|
||||
from typing import Union
|
||||
from typing import Literal, Union, overload
|
||||
|
||||
from flask import Flask
|
||||
|
||||
@ -79,6 +79,12 @@ class Storage:
|
||||
logger.exception(f"Failed to save file {filename}")
|
||||
raise e
|
||||
|
||||
@overload
|
||||
def load(self, filename: str, /, *, stream: Literal[False] = False) -> bytes: ...
|
||||
|
||||
@overload
|
||||
def load(self, filename: str, /, *, stream: Literal[True]) -> Generator: ...
|
||||
|
||||
def load(self, filename: str, /, *, stream: bool = False) -> Union[bytes, Generator]:
|
||||
try:
|
||||
if stream:
|
||||
|
||||
Reference in New Issue
Block a user