content: /*
   *  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.test.chain;
  
  import com.agentsflex.core.chain.*;
  import com.agentsflex.core.chain.listener.ChainEventListener;
  import com.agentsflex.core.chain.listener.ChainSuspendListener;
  import com.agentsflex.core.chain.node.LlmNode;
  import com.agentsflex.core.chain.node.StartNode;
  import com.agentsflex.core.util.Maps;
  
  import java.util.Collections;
  import java.util.List;
  import java.util.UUID;
  
  public class ChainTest {
  
      public static void main(String[] args) {
  
          StartNode startNode = new StartNode();
          startNode.setId(UUID.randomUUID().toString());
  
          LlmNode endNode = new LlmNode();
          endNode.setId(UUID.randomUUID().toString());
  
          Parameter parameter = new Parameter();
          parameter.setName("test");
          parameter.setDataType(DataType.String);
          parameter.setRequired(true);
  
          endNode.addInputParameter(parameter);
  
          Chain chain = new Chain();
          chain.addNode(startNode);
          chain.addNode(endNode);
  
          ChainEdge edge = new ChainEdge();
          edge.setId(UUID.randomUUID().toString());
          edge.setSource(startNode.getId());
          edge.setTarget(endNode.getId());
          chain.addEdge(edge);
  
  
          chain.addEventListener(new ChainEventListener() {
              @Override
              public void onEvent(ChainEvent event, Chain chain) {
                  System.out.println("onEvent: " + event.getClass());
              }
          });
  
  
          chain.addSuspendListener(new ChainSuspendListener() {
  
              @Override
              public void onSuspend(Chain chain) {
                  System.out.println("------------onSuspend-----------");
                  List<Parameter> suspendForParameters = chain.getSuspendForParameters();
                  System.out.println("suspendForParameters: " + suspendForParameters);
  
                  chain.resume(Maps.of("test","123"));
              }
          });
  
  
          chain.execute(Collections.emptyMap());
  
      }
  }
  