feat: add toString()

bsorrentino · Sep 14, 2024 · 7445a88 · 7445a88
1 parent b210f38
commit 7445a88
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/state/StateSnapshot.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/state/StateSnapshot.java
@@ -5,6 +5,8 @@
 import org.bsc.langgraph4j.RunnableConfig;
 import org.bsc.langgraph4j.checkpoint.Checkpoint;
 
+import static java.lang.String.*;
+
 public final class StateSnapshot<State extends AgentState> extends NodeOutput<State> {
     private final RunnableConfig config;
 
@@ -37,6 +39,11 @@ private StateSnapshot(@NonNull String node, @NonNull State state, @NonNull Runna
         this.config = config;
     }
 
+    @Override
+    public String toString() {
+
+        return format("StateSnapshot{node=%s, state=%s, config=%s}", node(), state(), config());
+    }
     public static <State extends AgentState> StateSnapshot<State> of(Checkpoint checkpoint, RunnableConfig config, AgentStateFactory<State> factory) {
 
         RunnableConfig newConfig = RunnableConfig.builder(config)
@@ -46,4 +53,5 @@ public static <State extends AgentState> StateSnapshot<State> of(Checkpoint chec
         return new StateSnapshot<>( checkpoint.getNodeId(), factory.apply(checkpoint.getState()), newConfig);
     }
 
+
 }