opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 11:56:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix typing errors in core/model_runtime (#26462)

Browse Source 
Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Asuka Minato 2025-09-30 11:24:05 +09:00 committed by GitHub
parent 2619c7553a
commit f7b1348623
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/model_runtime/entities/message_entities.py
View File

@ -74,7 +74,7 @@ class TextPromptMessageContent(PromptMessageContent):
Model class for text prompt message content. Model class for text prompt message content.
""" """
type: Literal[PromptMessageContentType.TEXT] = PromptMessageContentType.TEXT type: Literal[PromptMessageContentType.TEXT] = PromptMessageContentType.TEXT # type: ignore
data: str data: str
@ -95,11 +95,11 @@ class MultiModalPromptMessageContent(PromptMessageContent):
class VideoPromptMessageContent(MultiModalPromptMessageContent): class VideoPromptMessageContent(MultiModalPromptMessageContent):
type: Literal[PromptMessageContentType.VIDEO] = PromptMessageContentType.VIDEO type: Literal[PromptMessageContentType.VIDEO] = PromptMessageContentType.VIDEO # type: ignore
class AudioPromptMessageContent(MultiModalPromptMessageContent): class AudioPromptMessageContent(MultiModalPromptMessageContent):
type: Literal[PromptMessageContentType.AUDIO] = PromptMessageContentType.AUDIO type: Literal[PromptMessageContentType.AUDIO] = PromptMessageContentType.AUDIO # type: ignore
class ImagePromptMessageContent(MultiModalPromptMessageContent): class ImagePromptMessageContent(MultiModalPromptMessageContent):
@ -111,12 +111,12 @@ class ImagePromptMessageContent(MultiModalPromptMessageContent):
LOW = auto() LOW = auto()
HIGH = auto() HIGH = auto()
type: Literal[PromptMessageContentType.IMAGE] = PromptMessageContentType.IMAGE type: Literal[PromptMessageContentType.IMAGE] = PromptMessageContentType.IMAGE # type: ignore
detail: DETAIL = DETAIL.LOW detail: DETAIL = DETAIL.LOW
class DocumentPromptMessageContent(MultiModalPromptMessageContent): class DocumentPromptMessageContent(MultiModalPromptMessageContent):
type: Literal[PromptMessageContentType.DOCUMENT] = PromptMessageContentType.DOCUMENT type: Literal[PromptMessageContentType.DOCUMENT] = PromptMessageContentType.DOCUMENT # type: ignore
PromptMessageContentUnionTypes = Annotated[ PromptMessageContentUnionTypes = Annotated[

2
api/core/model_runtime/model_providers/__base/tokenizers/gpt2_tokenizer.py
View File

@ -15,7 +15,7 @@ class GPT2Tokenizer:
use gpt2 tokenizer to get num tokens use gpt2 tokenizer to get num tokens
""" """
_tokenizer = GPT2Tokenizer.get_encoder() _tokenizer = GPT2Tokenizer.get_encoder()
tokens = _tokenizer.encode(text) tokens = _tokenizer.encode(text) # type: ignore
return len(tokens) return len(tokens)
@staticmethod @staticmethod

6
api/core/model_runtime/utils/encoders.py
View File

@ -196,15 +196,15 @@ def jsonable_encoder(
return encoder(obj) return encoder(obj)
try: try:
data = dict(obj) data = dict(obj) # type: ignore
except Exception as e: except Exception as e:
errors: list[Exception] = [] errors: list[Exception] = []
errors.append(e) errors.append(e)
try: try:
data = vars(obj) data = vars(obj) # type: ignore
except Exception as e: except Exception as e:
errors.append(e) errors.append(e)
raise ValueError(errors) from e raise ValueError(str(errors)) from e
return jsonable_encoder( return jsonable_encoder(
data, data,
by_alias=by_alias, by_alias=by_alias,

3
api/pyrightconfig.json
View File

@ -7,7 +7,6 @@
"core/rag", "core/rag",
"extensions", "extensions",
"core/ops", "core/ops",
"core/model_runtime",
"core/workflow/nodes", "core/workflow/nodes",
"core/app/app_config/easy_ui_based_app/dataset" "core/app/app_config/easy_ui_based_app/dataset"
], ],
@ -37,4 +36,4 @@
"reportAttributeAccessIssue": "hint", "reportAttributeAccessIssue": "hint",
"pythonVersion": "3.11", "pythonVersion": "3.11",
"pythonPlatform": "All" "pythonPlatform": "All"
} }