fix: import

This commit is contained in:
Stream 2025-11-06 17:50:18 +08:00
parent 02b11c2fe3
commit da2421f378
No known key found for this signature in database
GPG Key ID: 033728094B100D70
2 changed files with 1 additions and 14 deletions

View File

@ -231,16 +231,3 @@ class Tool(ABC):
variable_name=variable_name, variable_value=variable_value, stream=stream
),
)
def create_variable_message(
self, variable_name: str, variable_value: Any, stream: bool = False
) -> ToolInvokeMessage:
"""
create a variable message
"""
return ToolInvokeMessage(
type=ToolInvokeMessage.MessageType.VARIABLE,
message=ToolInvokeMessage.VariableMessage(
variable_name=variable_name, variable_value=variable_value, stream=stream
),
)

View File

@ -21,7 +21,7 @@ from core.workflow.constants import (
MEMORY_BLOCK_VARIABLE_NODE_ID,
SYSTEM_VARIABLE_NODE_ID,
)
from core.workflow.enums import NodeType
from core.workflow.enums import NodeType, WorkflowExecutionStatus
from extensions.ext_storage import Storage
from factories.variable_factory import TypeMismatchError, build_segment_with_type
from libs.datetime_utils import naive_utc_now