mirror of
https://github.com/langgenius/dify.git
synced 2026-03-27 01:00:13 +08:00
56593f20b0
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: WH-2099 <wh2099@pm.me>
72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
"""
|
|
Domain entities for workflow execution.
|
|
|
|
Models describe graph runtime state and avoid infrastructure-specific details.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Mapping
|
|
from datetime import UTC, datetime
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from dify_graph.enums import WorkflowExecutionStatus, WorkflowType
|
|
|
|
|
|
class WorkflowExecution(BaseModel):
|
|
"""
|
|
Domain model for a workflow execution within the graph runtime.
|
|
"""
|
|
|
|
id_: str = Field(...)
|
|
workflow_id: str = Field(...)
|
|
workflow_version: str = Field(...)
|
|
workflow_type: WorkflowType = Field(...)
|
|
graph: Mapping[str, Any] = Field(...)
|
|
|
|
inputs: Mapping[str, Any] = Field(...)
|
|
outputs: Mapping[str, Any] | None = None
|
|
|
|
status: WorkflowExecutionStatus = WorkflowExecutionStatus.RUNNING
|
|
error_message: str = Field(default="")
|
|
total_tokens: int = Field(default=0)
|
|
total_steps: int = Field(default=0)
|
|
exceptions_count: int = Field(default=0)
|
|
|
|
started_at: datetime = Field(...)
|
|
finished_at: datetime | None = None
|
|
|
|
@property
|
|
def elapsed_time(self) -> float:
|
|
"""
|
|
Calculate elapsed time in seconds.
|
|
If workflow is not finished, use current time.
|
|
"""
|
|
end_time = self.finished_at or datetime.now(UTC).replace(tzinfo=None)
|
|
return (end_time - self.started_at).total_seconds()
|
|
|
|
@classmethod
|
|
def new(
|
|
cls,
|
|
*,
|
|
id_: str,
|
|
workflow_id: str,
|
|
workflow_type: WorkflowType,
|
|
workflow_version: str,
|
|
graph: Mapping[str, Any],
|
|
inputs: Mapping[str, Any],
|
|
started_at: datetime,
|
|
) -> WorkflowExecution:
|
|
return WorkflowExecution(
|
|
id_=id_,
|
|
workflow_id=workflow_id,
|
|
workflow_type=workflow_type,
|
|
workflow_version=workflow_version,
|
|
graph=graph,
|
|
inputs=inputs,
|
|
status=WorkflowExecutionStatus.RUNNING,
|
|
started_at=started_at,
|
|
)
|