Parcourir la source

fix(tool): tool node error (#7459)

Co-authored-by: hobo.l <hobo.l@binance.com>
tags/0.7.2
luckylhb90 il y a 1 an
Parent
révision
9a715f6b68
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2
    2
      api/core/workflow/nodes/tool/tool_node.py

+ 2
- 2
api/core/workflow/nodes/tool/tool_node.py Voir le fichier

@@ -2,7 +2,7 @@ from collections.abc import Mapping, Sequence
from os import path
from typing import Any, cast

from core.app.segments import ArrayAnyVariable, parser
from core.app.segments import ArrayAnySegment, ArrayAnyVariable, parser
from core.callback_handler.workflow_tool_callback_handler import DifyWorkflowCallbackHandler
from core.file.file_obj import FileTransferMethod, FileType, FileVar
from core.tools.entities.tool_entities import ToolInvokeMessage, ToolParameter
@@ -142,7 +142,7 @@ class ToolNode(BaseNode):

def _fetch_files(self, variable_pool: VariablePool) -> list[FileVar]:
variable = variable_pool.get(['sys', SystemVariableKey.FILES.value])
assert isinstance(variable, ArrayAnyVariable)
assert isinstance(variable, ArrayAnyVariable | ArrayAnySegment)
return list(variable.value) if variable else []

def _convert_tool_messages(self, messages: list[ToolInvokeMessage]):

Chargement…
Annuler
Enregistrer