ruff format code

This commit is contained in:
hjlarry 2025-10-01 23:39:46 +08:00
parent e6a44a0860
commit 82f2f76dc4
7 changed files with 14 additions and 20 deletions

View File

@ -999,17 +999,14 @@ class DraftWorkflowTriggerNodeApi(Resource):
@api.doc("poll_draft_workflow_trigger_node")
@api.doc(description="Poll for trigger events and execute single node when event arrives")
@api.doc(params={
"app_id": "Application ID",
"node_id": "Node ID"
})
@api.doc(params={"app_id": "Application ID", "node_id": "Node ID"})
@api.expect(
api.model(
"DraftWorkflowTriggerNodeRequest",
{
"trigger_name": fields.String(required=True, description="Trigger name"),
"subscription_id": fields.String(required=True, description="Subscription ID"),
}
},
)
)
@api.response(200, "Trigger event received and node executed successfully")
@ -1087,7 +1084,7 @@ class DraftWorkflowTriggerRunApi(Resource):
"node_id": fields.String(required=True, description="Node ID"),
"trigger_name": fields.String(required=True, description="Trigger name"),
"subscription_id": fields.String(required=True, description="Subscription ID"),
}
},
)
)
@api.response(200, "Trigger event received and workflow executed successfully")
@ -1168,7 +1165,7 @@ class DraftWorkflowTriggerWebhookRunApi(Resource):
"DraftWorkflowTriggerWebhookRunRequest",
{
"node_id": fields.String(required=True, description="Node ID"),
}
},
)
)
@api.response(200, "Workflow executed successfully")
@ -1252,7 +1249,7 @@ class DraftWorkflowTriggerScheduleRunApi(Resource):
"DraftWorkflowTriggerScheduleRunRequest",
{
"node_id": fields.String(required=True, description="Node ID"),
}
},
)
)
@api.response(200, "Workflow executed successfully")
@ -1287,7 +1284,7 @@ class DraftWorkflowTriggerScheduleRunApi(Resource):
args=workflow_args,
invoke_from=InvokeFrom.DEBUGGER,
streaming=True,
root_node_id=node_id
root_node_id=node_id,
)
return helper.compact_generate_response(response)
except InvokeRateLimitError as ex:

View File

@ -90,7 +90,7 @@ class WorkflowAppRunner(WorkflowBasedAppRunner):
workflow_id=self._workflow.id,
tenant_id=self._workflow.tenant_id,
user_id=self.application_generate_entity.user_id,
root_node_id=self._root_node_id
root_node_id=self._root_node_id,
)
# RUN WORKFLOW

View File

@ -80,7 +80,7 @@ class WorkflowBasedAppRunner:
workflow_id: str = "",
tenant_id: str = "",
user_id: str = "",
root_node_id: Optional[str] = None
root_node_id: Optional[str] = None,
) -> Graph:
"""
Init graph

View File

@ -126,7 +126,7 @@ class TriggerManager:
"""
provider = cls.get_trigger_provider(tenant_id, provider_id)
return provider.get_events()
@classmethod
def invoke_trigger(
cls,

View File

@ -61,4 +61,4 @@ class DatasourceProviderID(GenericProviderID):
class TriggerProviderID(GenericProviderID):
def __init__(self, value: str, is_hardcoded: bool = False) -> None:
super().__init__(value, is_hardcoded)
super().__init__(value, is_hardcoded)

View File

@ -87,10 +87,9 @@ class PluginParameterService:
provider_controller = TriggerManager.get_trigger_provider(tenant_id, TriggerProviderID(provider))
subscription: TriggerProviderSubscriptionApiEntity | SubscriptionBuilder | None
if credential_id:
subscription = (
TriggerSubscriptionBuilderService.get_subscription_builder(credential_id)
or TriggerProviderService.get_subscription_by_id(tenant_id, credential_id)
)
subscription = TriggerSubscriptionBuilderService.get_subscription_builder(
credential_id
) or TriggerProviderService.get_subscription_by_id(tenant_id, credential_id)
else:
subscription = TriggerProviderService.get_subscription_by_id(tenant_id)

View File

@ -71,9 +71,7 @@ class ToolTransformService:
elif isinstance(provider, ToolProviderApiEntity):
if provider.plugin_id:
if isinstance(provider.icon, str):
provider.icon = PluginService.get_plugin_icon_url(
tenant_id=tenant_id, filename=provider.icon
)
provider.icon = PluginService.get_plugin_icon_url(tenant_id=tenant_id, filename=provider.icon)
if isinstance(provider.icon_dark, str) and provider.icon_dark:
provider.icon_dark = PluginService.get_plugin_icon_url(
tenant_id=tenant_id, filename=provider.icon_dark