Skip to content

Commit 68ae2df

Browse files
committed
refactor(DiagramGenerator.java): update edge processing streams to use 'elements'
work on #73
1 parent de25bb2 commit 68ae2df

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

core/src/main/java/org/bsc/langgraph4j/DiagramGenerator.java

+3-3
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ protected final <State extends AgentState> Context generate( StateGraph<State> s
170170

171171
final int[] conditionalEdgeCount = { 0 };
172172

173-
stateGraph.edges.stream()
173+
stateGraph.edges.elements.stream()
174174
.filter( e -> !Objects.equals(e.sourceId(), START) )
175175
.filter( e -> !e.isParallel() )
176176
.forEach( e -> {
@@ -181,7 +181,7 @@ protected final <State extends AgentState> Context generate( StateGraph<State> s
181181
}
182182
});
183183

184-
var edgeStart = stateGraph.edges.stream()
184+
var edgeStart = stateGraph.edges.elements.stream()
185185
.filter( e -> Objects.equals( e.sourceId(), START) )
186186
.findFirst()
187187
.orElseThrow();
@@ -202,7 +202,7 @@ else if( edgeStart.target().value() != null ) {
202202

203203
conditionalEdgeCount[0] = 0; // reset
204204

205-
stateGraph.edges.stream()
205+
stateGraph.edges.elements.stream()
206206
.filter( e -> !Objects.equals(e.sourceId(), START) )
207207
.forEach( v -> {
208208

0 commit comments

Comments
 (0)