youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-25 05:06:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: mypy linter

Browse Source
This commit is contained in:
Yeuoly
2025-01-08 21:11:42 +08:00
parent fb309462ad
commit b7d168ac59
27 changed files with 62 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/plugin/backwards_invocation/app.py
View File

@ -119,7 +119,7 @@ class PluginAppBackwardsInvocation(BaseBackwardsInvocation):
stream: bool,
inputs: Mapping,
files: list[dict],
):
) -> Generator[Mapping | str, None, None] | Mapping:
"""
invoke workflow app
"""
@ -146,7 +146,7 @@ class PluginAppBackwardsInvocation(BaseBackwardsInvocation):
stream: bool,
inputs: Mapping,
files: list[dict],
):
) -> Generator[Mapping | str, None, None] | Mapping:
"""
invoke completion app
"""

16
api/core/plugin/backwards_invocation/model.py
View File

@ -268,7 +268,7 @@ Here is the extra instruction you need to follow:
return summary.message.content
lines = content.split("\n")
new_lines = []
new_lines: list[str] = []
# split long line into multiple lines
for i in range(len(lines)):
line = lines[i]
@ -286,16 +286,16 @@ Here is the extra instruction you need to follow:
# merge lines into messages with max tokens
messages: list[str] = []
for i in new_lines:
for i in new_lines: # type: ignore
if len(messages) == 0:
messages.append(i)
messages.append(i) # type: ignore
else:
if len(messages[-1]) + len(i) < max_tokens * 0.5:
messages[-1] += i
if get_prompt_tokens(messages[-1] + i) > max_tokens * 0.7:
messages.append(i)
if len(messages[-1]) + len(i) < max_tokens * 0.5: # type: ignore
messages[-1] += i # type: ignore
if get_prompt_tokens(messages[-1] + i) > max_tokens * 0.7: # type: ignore
messages.append(i) # type: ignore
else:
messages[-1] += i
messages[-1] += i # type: ignore
summaries = []
for i in range(len(messages)):