diff --git a/source/blender/functions/FN_lazy_function_graph.hh b/source/blender/functions/FN_lazy_function_graph.hh
index 52d0eba1280..6c5090adb63 100644
--- a/source/blender/functions/FN_lazy_function_graph.hh
+++ b/source/blender/functions/FN_lazy_function_graph.hh
@@ -239,6 +239,9 @@ class Graph : NonCopyable, NonMovable {
   Span<const FunctionNode *> function_nodes() const;
   Span<FunctionNode *> function_nodes();
 
+  Span<GraphInputSocket *> graph_inputs();
+  Span<GraphOutputSocket *> graph_outputs();
+
   Span<const GraphInputSocket *> graph_inputs() const;
   Span<const GraphOutputSocket *> graph_outputs() const;
 
@@ -506,6 +509,16 @@ inline Span<FunctionNode *> Graph::function_nodes()
   return nodes_.as_span().drop_front(2).cast<FunctionNode *>();
 }
 
+inline Span<GraphInputSocket *> Graph::graph_inputs()
+{
+  return graph_inputs_;
+}
+
+inline Span<GraphOutputSocket *> Graph::graph_outputs()
+{
+  return graph_outputs_;
+}
+
 inline Span<const GraphInputSocket *> Graph::graph_inputs() const
 {
   return graph_inputs_;