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

chore: apply ruff's pyupgrade linter rules to modernize Python code with targeted version (#2419)

Browse source
This commit is contained in:
Bowen Liang 2024-02-09 15:21:33 +08:00 • committed by GitHub
commit 063191889d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
246 changed files with 912 additions and 937 deletions
Download patch file Download diff file Expand all files Collapse all files

7
api/core/tools/tool/builtin_tool.py
View file

@ -1,4 +1,3 @@
from typing import List
from core.model_runtime.entities.llm_entities import LLMResult
from core.model_runtime.entities.message_entities import PromptMessage, SystemPromptMessage, UserPromptMessage
@ -22,7 +21,7 @@ class BuiltinTool(Tool):
"""
def invoke_model(
self, user_id: str, prompt_messages: List[PromptMessage], stop: List[str]
self, user_id: str, prompt_messages: list[PromptMessage], stop: list[str]
) -> LLMResult:
"""
invoke model
@ -52,7 +51,7 @@ class BuiltinTool(Tool):
tenant_id=self.runtime.tenant_id,
)
def get_prompt_tokens(self, prompt_messages: List[PromptMessage]) -> int:
def get_prompt_tokens(self, prompt_messages: list[PromptMessage]) -> int:
"""
get prompt tokens
@ -104,7 +103,7 @@ class BuiltinTool(Tool):
new_lines.append(line)
# merge lines into messages with max tokens
messages: List[str] = []
messages: list[str] = []
for i in new_lines:
if len(messages) == 0:
messages.append(i)