Skip to content

Commit 2d9e978

Browse files
committed
chore: move agent node back to core/workflows/nodes
1 parent 72496a5 commit 2d9e978

File tree

10 files changed

+18
-18
lines changed

10 files changed

+18
-18
lines changed

api/.importlinter

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ forbidden_modules =
4343
extensions.ext_redis
4444
allow_indirect_imports = True
4545
ignore_imports =
46-
dify_graph.nodes.agent.agent_node -> extensions.ext_database
46+
core.workflow.nodes.agent.agent_node-> extensions.ext_database
4747
dify_graph.nodes.llm.node -> extensions.ext_database
4848
dify_graph.model_runtime.model_providers.__base.ai_model -> extensions.ext_redis
4949
dify_graph.model_runtime.model_providers.model_provider_factory -> extensions.ext_redis
@@ -90,18 +90,18 @@ forbidden_modules =
9090
core.trigger
9191
core.variables
9292
ignore_imports =
93-
dify_graph.nodes.agent.agent_node -> core.model_manager
94-
dify_graph.nodes.agent.agent_node -> core.provider_manager
95-
dify_graph.nodes.agent.agent_node -> core.tools.tool_manager
93+
core.workflow.nodes.agent.agent_node -> core.model_manager
94+
core.workflow.nodes.agent.agent_node -> core.provider_manager
95+
core.workflow.nodes.agent.agent_node -> core.tools.tool_manager
9696
dify_graph.nodes.llm.llm_utils -> core.model_manager
9797
dify_graph.nodes.llm.protocols -> core.model_manager
9898
dify_graph.nodes.llm.llm_utils -> dify_graph.model_runtime.model_providers.__base.large_language_model
9999
dify_graph.nodes.llm.node -> core.tools.signature
100100
dify_graph.nodes.tool.tool_node -> core.callback_handler.workflow_tool_callback_handler
101101
dify_graph.nodes.tool.tool_node -> core.tools.tool_engine
102102
dify_graph.nodes.tool.tool_node -> core.tools.tool_manager
103-
dify_graph.nodes.agent.agent_node -> core.agent.entities
104-
dify_graph.nodes.agent.agent_node -> core.agent.plugin_entities
103+
core.workflow.nodes.agent.agent_node -> core.agent.entities
104+
core.workflow.nodes.agent.agent_node -> core.agent.plugin_entities
105105
dify_graph.nodes.knowledge_retrieval.knowledge_retrieval_node -> core.app.app_config.entities
106106
dify_graph.nodes.parameter_extractor.parameter_extractor_node -> core.prompt.advanced_prompt_transform
107107
dify_graph.nodes.parameter_extractor.parameter_extractor_node -> core.prompt.simple_prompt_transform
@@ -110,12 +110,12 @@ ignore_imports =
110110
dify_graph.nodes.parameter_extractor.parameter_extractor_node -> core.model_manager
111111
dify_graph.nodes.question_classifier.question_classifier_node -> core.model_manager
112112
dify_graph.nodes.tool.tool_node -> core.tools.utils.message_transformer
113-
dify_graph.nodes.agent.agent_node -> models.model
113+
core.workflow.nodes.agent.agent_node -> models.model
114114
dify_graph.nodes.llm.node -> core.helper.code_executor
115115
dify_graph.nodes.llm.node -> core.llm_generator.output_parser.errors
116116
dify_graph.nodes.llm.node -> core.llm_generator.output_parser.structured_output
117117
dify_graph.nodes.llm.node -> core.model_manager
118-
dify_graph.nodes.agent.entities -> core.prompt.entities.advanced_prompt_entities
118+
core.workflow.nodes.agent.entities -> core.prompt.entities.advanced_prompt_entities
119119
dify_graph.nodes.llm.entities -> core.prompt.entities.advanced_prompt_entities
120120
dify_graph.nodes.llm.node -> core.prompt.entities.advanced_prompt_entities
121121
dify_graph.nodes.llm.node -> core.prompt.utils.prompt_message_util
@@ -126,15 +126,15 @@ ignore_imports =
126126
dify_graph.nodes.question_classifier.question_classifier_node -> core.prompt.utils.prompt_message_util
127127
dify_graph.nodes.knowledge_index.entities -> core.rag.retrieval.retrieval_methods
128128
dify_graph.nodes.llm.node -> models.dataset
129-
dify_graph.nodes.agent.agent_node -> core.tools.utils.message_transformer
129+
core.workflow.nodes.agent.agent_node -> core.tools.utils.message_transformer
130130
dify_graph.nodes.llm.file_saver -> core.tools.signature
131131
dify_graph.nodes.llm.file_saver -> core.tools.tool_file_manager
132132
dify_graph.nodes.tool.tool_node -> core.tools.errors
133-
dify_graph.nodes.agent.agent_node -> extensions.ext_database
133+
core.workflow.nodes.agent.agent_node -> extensions.ext_database
134134
dify_graph.nodes.llm.node -> extensions.ext_database
135-
dify_graph.nodes.agent.agent_node -> models
135+
core.workflow.nodes.agent.agent_node-> models
136136
dify_graph.nodes.llm.node -> models.model
137-
dify_graph.nodes.agent.agent_node -> services
137+
core.workflow.nodes.agent.agent_node -> services
138138
dify_graph.nodes.tool.tool_node -> services
139139
dify_graph.model_runtime.model_providers.__base.ai_model -> configs
140140
dify_graph.model_runtime.model_providers.__base.ai_model -> extensions.ext_redis

api/core/agent/cot_agent_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
from core.tools.__base.tool import Tool
1515
from core.tools.entities.tool_entities import ToolInvokeMeta
1616
from core.tools.tool_engine import ToolEngine
17+
from core.workflow.nodes.agent.exc import AgentMaxIterationError
1718
from dify_graph.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta, LLMUsage
1819
from dify_graph.model_runtime.entities.message_entities import (
1920
AssistantPromptMessage,
@@ -22,7 +23,6 @@
2223
ToolPromptMessage,
2324
UserPromptMessage,
2425
)
25-
from dify_graph.nodes.agent.exc import AgentMaxIterationError
2626
from models.model import Message
2727

2828
logger = logging.getLogger(__name__)

api/core/agent/fc_agent_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from core.prompt.agent_history_prompt_transform import AgentHistoryPromptTransform
1111
from core.tools.entities.tool_entities import ToolInvokeMeta
1212
from core.tools.tool_engine import ToolEngine
13+
from core.workflow.nodes.agent.exc import AgentMaxIterationError
1314
from dify_graph.file import file_manager
1415
from dify_graph.model_runtime.entities import (
1516
AssistantPromptMessage,
@@ -25,7 +26,6 @@
2526
UserPromptMessage,
2627
)
2728
from dify_graph.model_runtime.entities.message_entities import ImagePromptMessageContent, PromptMessageContentUnionTypes
28-
from dify_graph.nodes.agent.exc import AgentMaxIterationError
2929
from models.model import Message
3030

3131
logger = logging.getLogger(__name__)

api/core/workflow/nodes/__init__.py

Whitespace-only changes.

api/dify_graph/nodes/agent/agent_node.py renamed to api/core/workflow/nodes/agent/agent_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
)
2323
from core.tools.tool_manager import ToolManager
2424
from core.tools.utils.message_transformer import ToolFileMessageTransformer
25+
from core.workflow.nodes.agent.entities import AgentNodeData, AgentOldVersionModelFeatures, ParamsAutoGenerated
2526
from dify_graph.enums import (
2627
NodeType,
2728
SystemVariableKey,
@@ -39,7 +40,6 @@
3940
StreamChunkEvent,
4041
StreamCompletedEvent,
4142
)
42-
from dify_graph.nodes.agent.entities import AgentNodeData, AgentOldVersionModelFeatures, ParamsAutoGenerated
4343
from dify_graph.nodes.base.node import Node
4444
from dify_graph.nodes.base.variable_template_parser import VariableTemplateParser
4545
from dify_graph.runtime import VariablePool

api/dify_graph/nodes/base/node.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -385,8 +385,8 @@ def run(self) -> Generator[GraphNodeEventBase, None, None]:
385385
start_event.provider_id = getattr(self.node_data, "provider_id", "")
386386
start_event.provider_type = getattr(self.node_data, "provider_type", "")
387387

388-
from dify_graph.nodes.agent.agent_node import AgentNode
389-
from dify_graph.nodes.agent.entities import AgentNodeData
388+
from core.workflow.nodes.agent import AgentNode
389+
from core.workflow.nodes.agent.entities import AgentNodeData
390390

391391
if isinstance(self, AgentNode):
392392
start_event.agent_strategy = AgentNodeStrategyInit(

api/tests/unit_tests/core/workflow/graph_engine/test_mock_nodes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@
1111
from unittest.mock import MagicMock
1212

1313
from core.model_manager import ModelInstance
14+
from core.workflow.nodes.agent import AgentNode
1415
from dify_graph.enums import WorkflowNodeExecutionMetadataKey, WorkflowNodeExecutionStatus
1516
from dify_graph.model_runtime.entities.llm_entities import LLMUsage
1617
from dify_graph.node_events import NodeRunResult, StreamChunkEvent, StreamCompletedEvent
17-
from dify_graph.nodes.agent import AgentNode
1818
from dify_graph.nodes.code import CodeNode
1919
from dify_graph.nodes.document_extractor import DocumentExtractorNode
2020
from dify_graph.nodes.http_request import HttpRequestNode

0 commit comments

Comments
 (0)