estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix incorrect assert type in the AgentNode class (#22964)

Browse Source
This commit is contained in:
Yongtao Huang
2025-07-25 16:31:21 +08:00
committed by GitHub
parent 9c5b1e7012
commit 31985d94fa
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/workflow/nodes/agent/agent_node.py
View File

@@ -50,6 +50,7 @@ from .exc import (
AgentInputTypeError, AgentInputTypeError,
AgentInvocationError, AgentInvocationError,
AgentMessageTransformError, AgentMessageTransformError,
AgentNodeError,
AgentVariableNotFoundError, AgentVariableNotFoundError,
AgentVariableTypeError, AgentVariableTypeError,
ToolFileNotFoundError, ToolFileNotFoundError,
@@ -593,7 +594,14 @@ class AgentNode(BaseNode):
variables[variable_name] = variable_value variables[variable_name] = variable_value
elif message.type == ToolInvokeMessage.MessageType.FILE: elif message.type == ToolInvokeMessage.MessageType.FILE:
assert message.meta is not None assert message.meta is not None
assert isinstance(message.meta, File) assert isinstance(message.meta, dict)
# Validate that meta contains a 'file' key
if "file" not in message.meta:
raise AgentNodeError("File message is missing 'file' key in meta")
# Validate that the file is an instance of File
if not isinstance(message.meta["file"], File):
raise AgentNodeError(f"Expected File object but got {type(message.meta['file']).__name__}")
files.append(message.meta["file"]) files.append(message.meta["file"])
elif message.type == ToolInvokeMessage.MessageType.LOG: elif message.type == ToolInvokeMessage.MessageType.LOG:
assert isinstance(message.message, ToolInvokeMessage.LogMessage) assert isinstance(message.message, ToolInvokeMessage.LogMessage)