chore: handle migrations

This commit is contained in:
Novice
2025-12-17 15:59:09 +08:00
parent 7c03bcba2b
commit cb99b8f04d
3 changed files with 5 additions and 5 deletions

View File

@@ -577,8 +577,8 @@ class Node(Generic[NodeDataT]):
@_dispatch.register
def _(self, event: ToolResultChunkEvent) -> NodeRunStreamChunkEvent:
from core.workflow.entities import ToolResult
from core.workflow.graph_events import ChunkType, ToolResultStatus
from core.workflow.entities import ToolResult, ToolResultStatus
from core.workflow.graph_events import ChunkType
tool_result = event.tool_result
status: ToolResultStatus = (

View File

@@ -12,7 +12,7 @@ from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision = '85c8b4a64f53'
down_revision = 'd57accd375ae'
down_revision = '03ea244985ce'
branch_labels = None
depends_on = None

View File

@@ -2,9 +2,9 @@
from unittest.mock import MagicMock
from core.workflow.entities import ToolResultStatus
from core.workflow.entities.tool_entities import ToolResultStatus
from core.workflow.enums import NodeType
from core.workflow.graph import Graph
from core.workflow.graph.graph import Graph
from core.workflow.graph_engine.response_coordinator.coordinator import ResponseStreamCoordinator
from core.workflow.graph_events import (
ChunkType,