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

8
api/core/tools/tool_label_manager.py
View file

@ -84,13 +84,17 @@ class ToolLabelManager:
if not isinstance(controller, ApiToolProviderController | WorkflowToolProviderController):
raise ValueError("Unsupported tool type")
provider_ids = [controller.provider_id for controller in tool_providers]
provider_ids = [
controller.provider_id
for controller in tool_providers
if isinstance(controller, (ApiToolProviderController, WorkflowToolProviderController))
]
labels: list[ToolLabelBinding] = (
db.session.query(ToolLabelBinding).filter(ToolLabelBinding.tool_id.in_(provider_ids)).all()
)
tool_labels = {label.tool_id: [] for label in labels}
tool_labels: dict[str, list[str]] = {label.tool_id: [] for label in labels}
for label in labels:
tool_labels[label.tool_id].append(label.label_name)