Support optional inputs in lynxkite-graph-analytics.

lynxkite-core/tests/test_simple.py

@@ -19,7 +19,7 @@ async def test_optional_inputs():
     ws = workspace.Workspace(env="test", nodes=[], edges=[])
     a = ws.add_node(one)
     b = ws.add_node(maybe_add)
-    outputs = await ws.execute()
+    await ws.execute()
     assert b.data.error == "Missing input: a"
     ws.add_edge(a, "output", b, "a")
     outputs = await ws.execute()

lynxkite-graph-analytics/src/lynxkite_graph_analytics/core.py

@@ -187,6 +187,7 @@ async def execute(ws: workspace.Workspace):
                 todo.remove(id)
                 progress = True
                 await _execute_node(node, ws, catalog, outputs)
+    return outputs
 
 
 async def await_if_needed(obj):
@@ -213,10 +214,15 @@ async def _execute_node(
     # Convert inputs types to match operation signature.
     try:
         inputs = []
+        missing = []
         for p in op.inputs:
             if p.name not in input_map:
-                node.publish_error(f"Missing input: {p.name}")
-                return
+                opt_type = ops.get_optional_type(p.type)
+                if opt_type is not None:
+                    inputs.append(None)
+                else:
+                    missing.append(p.name)
+                continue
             x = input_map[p.name]
             if p.type == nx.Graph and isinstance(x, Bundle):
                 x = x.to_nx()
@@ -230,6 +236,9 @@ async def _execute_node(
             traceback.print_exc()
         node.publish_error(e)
         return
+    if missing:
+        node.publish_error(f"Missing input: {', '.join(missing)}")
+        return
     # Execute op.
     try:
         result = op(*inputs, **params)

lynxkite-graph-analytics/tests/test_lynxkite_ops.py

@@ -160,5 +160,28 @@ async def test_multiple_inputs():
     assert ws.nodes[-1].data.display is False
 
 
+async def test_optional_inputs():
+    @ops.op("test", "one")
+    def one():
+        return 1
+
+    @ops.op("test", "maybe add")
+    def maybe_add(a: int, b: int | None = None):
+        return a + (b or 0)
+
+    assert maybe_add.__op__.inputs == [
+        ops.Input(name="a", type=int, position="left"),
+        ops.Input(name="b", type=int | None, position="left"),
+    ]
+    ws = workspace.Workspace(env="test", nodes=[], edges=[])
+    a = ws.add_node(one)
+    b = ws.add_node(maybe_add)
+    await execute(ws)
+    assert b.data.error == "Missing input: a"
+    ws.add_edge(a, "output", b, "a")
+    outputs = await execute(ws)
+    assert outputs[b.id, "output"] == 1
+
+
 if __name__ == "__main__":
     pytest.main()