fix(core/Node.java): Ensured the equals method only casts to `Node<…

…?>` if the object is not null, work on #73
bsorrentino · Feb 6, 2025 · 79f8395 · 79f8395
1 parent 103becc
commit 79f8395
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/core/src/main/java/org/bsc/langgraph4j/Node.java b/core/src/main/java/org/bsc/langgraph4j/Node.java
@@ -78,9 +78,12 @@ public Node<State> withIdUpdated( Function<String,String> newId ) {
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
-        if (o == null || getClass() != o.getClass()) return false;
-        Node<?> node = (Node<?>) o;
-        return Objects.equals(id, node.id);
+        if (o == null ) return false;
+        if( o instanceof Node<?> node ) {
+            return Objects.equals(id, node.id);
+        }
+        return false;
+
     }
 
     /**