chore(api/core): apply ruff reformatting (#7624)

This commit is contained in:
Bowen Liang
2024-09-10 17:00:20 +08:00
committed by GitHub
parent 178730266d
commit 2cf1187b32
724 changed files with 21180 additions and 21123 deletions

View File

@@ -15,7 +15,7 @@ class ToolLabelManager:
"""
tool_labels = [label for label in tool_labels if label in default_tool_label_name_list]
return list(set(tool_labels))
@classmethod
def update_tool_labels(cls, controller: ToolProviderController, labels: list[str]):
"""
@@ -26,20 +26,20 @@ class ToolLabelManager:
if isinstance(controller, ApiToolProviderController | WorkflowToolProviderController):
provider_id = controller.provider_id
else:
raise ValueError('Unsupported tool type')
raise ValueError("Unsupported tool type")
# delete old labels
db.session.query(ToolLabelBinding).filter(
ToolLabelBinding.tool_id == provider_id
).delete()
db.session.query(ToolLabelBinding).filter(ToolLabelBinding.tool_id == provider_id).delete()
# insert new labels
for label in labels:
db.session.add(ToolLabelBinding(
tool_id=provider_id,
tool_type=controller.provider_type.value,
label_name=label,
))
db.session.add(
ToolLabelBinding(
tool_id=provider_id,
tool_type=controller.provider_type.value,
label_name=label,
)
)
db.session.commit()
@@ -53,12 +53,16 @@ class ToolLabelManager:
elif isinstance(controller, BuiltinToolProviderController):
return controller.tool_labels
else:
raise ValueError('Unsupported tool type')
raise ValueError("Unsupported tool type")
labels: list[ToolLabelBinding] = db.session.query(ToolLabelBinding.label_name).filter(
ToolLabelBinding.tool_id == provider_id,
ToolLabelBinding.tool_type == controller.provider_type.value,
).all()
labels: list[ToolLabelBinding] = (
db.session.query(ToolLabelBinding.label_name)
.filter(
ToolLabelBinding.tool_id == provider_id,
ToolLabelBinding.tool_type == controller.provider_type.value,
)
.all()
)
return [label.label_name for label in labels]
@@ -75,22 +79,20 @@ class ToolLabelManager:
"""
if not tool_providers:
return {}
for controller in tool_providers:
if not isinstance(controller, ApiToolProviderController | WorkflowToolProviderController):
raise ValueError('Unsupported tool type')
raise ValueError("Unsupported tool type")
provider_ids = [controller.provider_id for controller in tool_providers]
labels: list[ToolLabelBinding] = db.session.query(ToolLabelBinding).filter(
ToolLabelBinding.tool_id.in_(provider_ids)
).all()
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 = {label.tool_id: [] for label in labels}
for label in labels:
tool_labels[label.tool_id].append(label.label_name)
return tool_labels
return tool_labels