From 8d5600d2f65dade118c1ab3b115e34a8ab884298 Mon Sep 17 00:00:00 2001 From: 0007 <0007@qq.com> Date: Wed, 27 Aug 2025 19:58:54 +0800 Subject: [PATCH] Add File --- .../agentsflex/core/chain/node/EndNode.java | 101 ++++++++++++++++++ 1 file changed, 101 insertions(+) create mode 100644 agents-flex-core/src/main/java/com/agentsflex/core/chain/node/EndNode.java diff --git a/agents-flex-core/src/main/java/com/agentsflex/core/chain/node/EndNode.java b/agents-flex-core/src/main/java/com/agentsflex/core/chain/node/EndNode.java new file mode 100644 index 0000000..d16a824 --- /dev/null +++ b/agents-flex-core/src/main/java/com/agentsflex/core/chain/node/EndNode.java @@ -0,0 +1,101 @@ +/* + * Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com). + *

+ * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

+ * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.agentsflex.core.chain.node; + +import com.agentsflex.core.chain.Chain; +import com.agentsflex.core.chain.Parameter; +import com.agentsflex.core.chain.RefType; +import com.agentsflex.core.util.StringUtil; + +import java.util.Collections; +import java.util.HashMap; +import java.util.Map; + +public class EndNode extends BaseNode { + private boolean normal = true; + private String message; + + public String getMessage() { + return message; + } + + public void setMessage(String message) { + this.message = message; + } + + public boolean isNormal() { + return normal; + } + + public void setNormal(boolean normal) { + this.normal = normal; + } + + public EndNode() { + this.name = "end"; + } + + @Override + public Map execute(Chain chain) { + if (StringUtil.hasText(message)) { + if (normal) { + chain.stopNormal(message); + } else { + chain.stopError(message); + } + } + + if (this.outputDefs != null) { + Map output = new HashMap<>(); + for (Parameter outputDef : this.outputDefs) { + if (outputDef.getRefType() == RefType.REF) { + output.put(outputDef.getName(), chain.get(outputDef.getRef())); + } else if (outputDef.getRefType() == RefType.INPUT) { + output.put(outputDef.getName(), outputDef.getRef()); + } else if (outputDef.getRefType() == RefType.FIXED) { + output.put(outputDef.getName(), StringUtil.getFirstWithText(outputDef.getValue(), outputDef.getDefaultValue())); + } + // default is ref type + else if (StringUtil.hasText(outputDef.getRef())) { + output.put(outputDef.getName(), chain.get(outputDef.getRef())); + } + } + return output; + } + + + return Collections.emptyMap(); + } + + @Override + public String toString() { + return "EndNode{" + + "normal=" + normal + + ", message='" + message + '\'' + + ", description='" + description + '\'' + + ", parameters=" + parameters + + ", outputDefs=" + outputDefs + + ", id='" + id + '\'' + + ", name='" + name + '\'' + + ", async=" + async + + ", inwardEdges=" + inwardEdges + + ", outwardEdges=" + outwardEdges + + ", condition=" + condition + + ", memory=" + memory + + ", nodeStatus=" + nodeStatus + + '}'; + } +}