opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 03:36:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct the provider type

Browse Source
This commit is contained in:
Novice Lee 2025-01-14 14:20:03 +08:00
parent d6b9648332
commit 11428172eb
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -161,9 +161,10 @@ class AgentNode(ToolNode):
value = cast(list[dict[str, Any]], value) value = cast(list[dict[str, Any]], value)
tool_value = [] tool_value = []
for tool in value: for tool in value:
provider_type = ToolProviderType(tool.get("type", ToolProviderType.BUILT_IN.value))
entity = AgentToolEntity( entity = AgentToolEntity(
provider_id=tool.get("provider_name", ""), provider_id=tool.get("provider_name", ""),
provider_type=ToolProviderType.BUILT_IN, provider_type=provider_type,
tool_name=tool.get("tool_name", ""), tool_name=tool.get("tool_name", ""),
tool_parameters=tool.get("parameters", {}), tool_parameters=tool.get("parameters", {}),
plugin_unique_identifier=tool.get("plugin_unique_identifier", None), plugin_unique_identifier=tool.get("plugin_unique_identifier", None),
@ -183,6 +184,7 @@ class AgentNode(ToolNode):
{ {
**tool_runtime.entity.model_dump(mode="json"), **tool_runtime.entity.model_dump(mode="json"),
"runtime_parameters": tool_runtime.runtime.runtime_parameters, "runtime_parameters": tool_runtime.runtime.runtime_parameters,
"provider_type": provider_type.value,
} }
) )
value = tool_value value = tool_value