Add File

2025-08-27 19:58:54 +08:00
parent 33d58de976
commit 8d5600d2f6
1 changed files with 101 additions and 0 deletions
--- 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
@@ -0,0 +1,101 @@
+/*
+ *  Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com).
+ *  <p>
+ *  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
+ *  <p>
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *  <p>
+ *  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<String, Object> execute(Chain chain) {
+        if (StringUtil.hasText(message)) {
+            if (normal) {
+                chain.stopNormal(message);
+            } else {
+                chain.stopError(message);
+            }
+        }
+
+        if (this.outputDefs != null) {
+            Map<String, Object> 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 +
+            '}';
+    }
+}