Browse Source

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

tags/1.7.1
Yongtao Huang 3 months ago
parent
commit
31985d94fa
No account linked to committer's email address
1 changed files with 9 additions and 1 deletions
  1. 9
    1
      api/core/workflow/nodes/agent/agent_node.py

+ 9
- 1
api/core/workflow/nodes/agent/agent_node.py View File

AgentInputTypeError, AgentInputTypeError,
AgentInvocationError, AgentInvocationError,
AgentMessageTransformError, AgentMessageTransformError,
AgentNodeError,
AgentVariableNotFoundError, AgentVariableNotFoundError,
AgentVariableTypeError, AgentVariableTypeError,
ToolFileNotFoundError, ToolFileNotFoundError,
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)

Loading…
Cancel
Save