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

Merge branch 'fix/chore-fix' into dev/plugin-deploy

Browse Source
This commit is contained in:
Novice Lee
2025-01-09 16:55:58 +08:00
parent 2b20ba697c d76af08784
commit ffbbd47d13
49 changed files with 157 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -28,7 +28,7 @@ class AgentNode(ToolNode):
Agent Node
"""
_node_data_cls = AgentNodeData
_node_data_cls = AgentNodeData # type: ignore
_node_type = NodeType.AGENT
def _run(self) -> Generator:
@ -137,7 +137,7 @@ class AgentNode(ToolNode):
"""
agent_parameters_dictionary = {parameter.name: parameter for parameter in agent_parameters}
result = {}
result: dict[str, Any] = {}
for parameter_name in node_data.agent_parameters:
parameter = agent_parameters_dictionary.get(parameter_name)
if not parameter:
@ -226,7 +226,7 @@ class AgentNode(ToolNode):
:return:
"""
node_data = cast(AgentNodeData, node_data)
result = {}
result: dict[str, Any] = {}
for parameter_name in node_data.agent_parameters:
input = node_data.agent_parameters[parameter_name]
if input.type == "mixed":

4
api/core/workflow/nodes/llm/node.py
View File

@ -233,9 +233,9 @@ class LLMNode(BaseNode[LLMNodeData]):
db.session.close()
invoke_result = model_instance.invoke_llm(
prompt_messages=prompt_messages,
prompt_messages=list(prompt_messages),
model_parameters=node_data_model.completion_params,
stop=stop,
stop=list(stop or []),
stream=True,
user=self.user_id,
)

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

@ -1,5 +1,5 @@
from collections.abc import Generator, Mapping, Sequence
from typing import Any, Optional, cast
from typing import Any, cast
from sqlalchemy import select
from sqlalchemy.orm import Session
@ -199,7 +199,7 @@ class ToolNode(BaseNode[ToolNodeData]):
json: list[dict] = []
agent_logs: list[AgentLogEvent] = []
agent_execution_metadata: Optional[Mapping[NodeRunMetadataKey, Any]] = {}
agent_execution_metadata: Mapping[NodeRunMetadataKey, Any] = {}
variables: dict[str, Any] = {}