chore(api/core): apply ruff reformatting (#7624)
This commit is contained in:
@@ -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
|
||||
|
Reference in New Issue
Block a user