chore: code format model-selector use enum (#22787)

Co-authored-by: 刘江波 <jiangbo721@163.com>
7 months ago · 5a02e599e1
parent d1572f47a0
commit 5a02e599e1
1 changed files with 1 additions and 1 deletions
--- a/api/core/workflow/nodes/agent/agent_node.py
+++ b/api/core/workflow/nodes/agent/agent_node.py
@ -309,7 +309,7 @@ class AgentNode(BaseNode):
                            }
                        )
                    value = tool_value
-                if parameter.type == "model-selector":
+                if parameter.type == AgentStrategyParameter.AgentStrategyParameterType.MODEL_SELECTOR:
                    value = cast(dict[str, Any], value)
                    model_instance, model_schema = self._fetch_model(value)
                    # memory config