refactor: rename getCheckpointSaver() to checkpointSaver() because it…

… returns an Optional
bsorrentino · Aug 28, 2024 · fd072d2 · fd072d2
1 parent a00054b
commit fd072d2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 10 deletions.
diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/CompileConfig.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/CompileConfig.java
@@ -1,5 +1,6 @@
 package org.bsc.langgraph4j;
 
+import lombok.Getter;
 import org.bsc.langgraph4j.checkpoint.BaseCheckpointSaver;
 
 import java.util.Optional;
@@ -8,12 +9,12 @@
 public class CompileConfig {
 
     private BaseCheckpointSaver checkpointSaver;
+    @Getter
     private String[] interruptBefore = {};
+    @Getter
     private String[] interruptAfter = {};
 
-    public Optional<BaseCheckpointSaver> getCheckpointSaver() { return Optional.ofNullable(checkpointSaver); }
-    public String[] getInterruptBefore() { return interruptBefore; }
-    public String[] getInterruptAfter() { return interruptAfter; }
+    public Optional<BaseCheckpointSaver> checkpointSaver() { return Optional.ofNullable(checkpointSaver); }
 
     public static Builder builder() {
         return new Builder();

diff --git a/core-jdk8/src/main/java/org/bsc/langgraph4j/CompiledGraph.java b/core-jdk8/src/main/java/org/bsc/langgraph4j/CompiledGraph.java
@@ -61,23 +61,23 @@ protected CompiledGraph(StateGraph<State> stateGraph, CompileConfig compileConfi
     }
 
     public Collection<StateSnapshot<State>> getStateHistory( RunnableConfig config ) {
-        var saver = compileConfig.getCheckpointSaver().orElseThrow( () -> (new IllegalStateException("Missing CheckpointSaver!")) );
+        var saver = compileConfig.checkpointSaver().orElseThrow( () -> (new IllegalStateException("Missing CheckpointSaver!")) );
 
         return saver.list(config).stream()
                 .map( checkpoint -> StateSnapshot.of( checkpoint, config, stateGraph.getStateFactory() ) )
                 .collect(Collectors.toList());
     }
 
     public StateSnapshot<State> getState( RunnableConfig config ) {
-        var saver = compileConfig.getCheckpointSaver().orElseThrow( () -> (new IllegalStateException("Missing CheckpointSaver!")) );
+        var saver = compileConfig.checkpointSaver().orElseThrow( () -> (new IllegalStateException("Missing CheckpointSaver!")) );
 
         var checkpoint = saver.get(config).orElseThrow( () -> (new IllegalStateException("Missing Checkpoint!")) );
 
         return StateSnapshot.of( checkpoint, config, stateGraph.getStateFactory() );
     }
 
     public RunnableConfig updateState( RunnableConfig config, Map<String,Object> values, String asNode ) throws Exception {
-        var saver = compileConfig.getCheckpointSaver().orElseThrow( () -> (new IllegalStateException("Missing CheckpointSaver!")) );
+        var saver = compileConfig.checkpointSaver().orElseThrow( () -> (new IllegalStateException("Missing CheckpointSaver!")) );
 
         // merge values with checkpoint values
         var updatedCheckpoint = saver.get(config)
@@ -152,13 +152,13 @@ private String getEntryPoint( State state ) throws Exception {
     }
 
     private void addCheckpoint( RunnableConfig config, String nodeId, State state, String nextNodeId ) throws Exception {
-        if( compileConfig.getCheckpointSaver().isPresent() ) {
+        if( compileConfig.checkpointSaver().isPresent() ) {
             Checkpoint cp =  Checkpoint.builder()
                                 .nodeId( nodeId )
                                 .state( state )
                                 .nextNodeId( nextNodeId )
                                 .build();
-            compileConfig.getCheckpointSaver().get().put( config, cp );
+            compileConfig.checkpointSaver().get().put( config, cp );
         }
     }
 
@@ -172,7 +172,7 @@ Map<String,Object> getInitialStateFromSchema() {
 
     Map<String,Object> getInitialState(Map<String,Object> inputs, RunnableConfig config) {
 
-        return compileConfig.getCheckpointSaver()
+        return compileConfig.checkpointSaver()
                 .flatMap( saver -> saver.get( config ) )
                 .map( cp -> AgentState.updateState( cp.getState(), inputs, stateGraph.getChannels() ))
                 .orElseGet( () -> AgentState.updateState(getInitialStateFromSchema(), inputs, stateGraph.getChannels() ));
@@ -258,7 +258,7 @@ public AsyncGenerator<NodeOutput<State>> stream(Map<String,Object> inputs, Runna
 
         if( isResumeRequest ) {
 
-            BaseCheckpointSaver saver = compileConfig.getCheckpointSaver().orElseThrow(() -> (new IllegalStateException("inputs cannot be null (ie. resume request) if no checkpoint saver is configured")));
+            BaseCheckpointSaver saver = compileConfig.checkpointSaver().orElseThrow(() -> (new IllegalStateException("inputs cannot be null (ie. resume request) if no checkpoint saver is configured")));
 
             Checkpoint startCheckpoint = saver.get( config ).orElseThrow( () -> (new IllegalStateException("Resume request without a saved checkpoint!")) );