youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-02 16:38:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: tool models

Browse Source
This commit is contained in:
Yeuoly
2024-08-30 15:55:10 +08:00
parent 1fa3b9cfd8
commit cf4e9f317e
6 changed files with 60 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/workflow/nodes/tool/tool_node.py
View File

@ -1,4 +1,4 @@
from collections.abc import Generator, Mapping, Sequence
from collections.abc import Generator, Sequence
from os import path
from typing import Any, cast
@ -100,7 +100,7 @@ class ToolNode(BaseNode):
variable_pool: VariablePool,
node_data: ToolNodeData,
for_log: bool = False,
) -> Mapping[str, Any]:
) -> dict[str, Any]:
"""
Generate parameters based on the given tool parameters, variable pool, and node data.
@ -110,7 +110,7 @@ class ToolNode(BaseNode):
node_data (ToolNodeData): The data associated with the tool node.
Returns:
Mapping[str, Any]: A dictionary containing the generated parameters.
dict[str, Any]: A dictionary containing the generated parameters.
"""
tool_parameters_dictionary = {parameter.name: parameter for parameter in tool_parameters}