From 9d392e21fb19e3e8120c1751153b4001e0206465 Mon Sep 17 00:00:00 2001 From: 0007 <0007@qq.com> Date: Wed, 27 Aug 2025 19:58:05 +0800 Subject: [PATCH] Add File --- .../agentsflex/chain/node/GroovyExecNode.java | 64 +++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100644 agents-flex-chain/agents-flex-chain-groovy/src/main/java/com/agentsflex/chain/node/GroovyExecNode.java diff --git a/agents-flex-chain/agents-flex-chain-groovy/src/main/java/com/agentsflex/chain/node/GroovyExecNode.java b/agents-flex-chain/agents-flex-chain-groovy/src/main/java/com/agentsflex/chain/node/GroovyExecNode.java new file mode 100644 index 0000000..200e140 --- /dev/null +++ b/agents-flex-chain/agents-flex-chain-groovy/src/main/java/com/agentsflex/chain/node/GroovyExecNode.java @@ -0,0 +1,64 @@ +/* + * 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.chain.node; + +import com.agentsflex.core.chain.Chain; +import com.agentsflex.core.chain.node.CodeNode; +import groovy.lang.Binding; +import groovy.lang.GroovyShell; + +import java.util.HashMap; +import java.util.Map; + +public class GroovyExecNode extends CodeNode { + + @Override + protected Map executeCode(String code, Chain chain) { + Binding binding = new Binding(); + + Map parameterValues = chain.getParameterValues(this); + if (parameterValues != null) { + parameterValues.forEach(binding::setVariable); + } + + Map result = new HashMap<>(); + binding.setVariable("_result", result); + binding.setVariable("_chain", chain); + binding.setVariable("_context", chain.getNodeContext(this.id)); + + GroovyShell shell = new GroovyShell(binding); + shell.evaluate(code); + return result; + } + + @Override + public String toString() { + return "GroovyExecNode{" + + "inwardEdges=" + inwardEdges + + ", code='" + code + '\'' + + ", description='" + description + '\'' + + ", parameters=" + parameters + + ", outputDefs=" + outputDefs + + ", id='" + id + '\'' + + ", name='" + name + '\'' + + ", async=" + async + + ", outwardEdges=" + outwardEdges + + ", condition=" + condition + + ", memory=" + memory + + ", nodeStatus=" + nodeStatus + + '}'; + } +}