Merge pull request #132 from biggraph/darabos-execute-button

.github/workflows/test.yaml

@@ -8,6 +8,8 @@ on:
 jobs:
   test:
     runs-on: ubuntu-latest
+    env:
+      UV_SYSTEM_PYTHON: 1
     steps:
       - uses: actions/checkout@v4
 
@@ -24,6 +26,8 @@ jobs:
         run: |
           eval `ssh-agent -s`
           ssh-add - <<< '${{ secrets.LYNXSCRIBE_DEPLOY_KEY }}'
+          uv venv
+          . .venv/bin/activate
           uv pip install \
             -e lynxkite-core/[dev] \
             -e lynxkite-app/[dev] \
@@ -31,15 +35,11 @@ jobs:
             -e lynxkite-bio \
             -e lynxkite-lynxscribe/ \
             -e lynxkite-pillow-example/
-        env:
-            UV_SYSTEM_PYTHON: 1
 
       - name: Run pre-commits
         run: |
           uv pip install pre-commit
           pre-commit run --all-files
-        env:
-            UV_SYSTEM_PYTHON: 1
 
       - name: Run core tests
         run: |
@@ -65,8 +65,6 @@ jobs:
         run: |
           uv pip install mkdocs-material mkdocstrings[python]
           mkdocs build
-        env:
-            UV_SYSTEM_PYTHON: 1
 
       - uses: actions/setup-node@v4
         with:

examples/Image processing.lynxkite.json

@@ -8,31 +8,31 @@
       "targetHandle": "image"
     },
     {
-      "id": "xy-edge__Open image 1output-Flip verically 1image",
-      "source": "Open image 1",
+      "id": "xy-edge__To grayscale 1output-View image 2image",
+      "source": "To grayscale 1",
       "sourceHandle": "output",
-      "target": "Flip verically 1",
+      "target": "View image 2",
       "targetHandle": "image"
     },
     {
-      "id": "xy-edge__To grayscale 1output-View image 2image",
-      "source": "To grayscale 1",
+      "id": "xy-edge__Blur 1output-To grayscale 1image",
+      "source": "Blur 1",
       "sourceHandle": "output",
-      "target": "View image 2",
+      "target": "To grayscale 1",
       "targetHandle": "image"
     },
     {
-      "id": "xy-edge__Flip verically 1output-Blur 1image",
-      "source": "Flip verically 1",
+      "id": "Open image 1 Flip vertically 1",
+      "source": "Open image 1",
       "sourceHandle": "output",
-      "target": "Blur 1",
+      "target": "Flip vertically 1",
       "targetHandle": "image"
     },
     {
-      "id": "xy-edge__Blur 1output-To grayscale 1image",
-      "source": "Blur 1",
+      "id": "Flip vertically 1 Blur 1",
+      "source": "Flip vertically 1",
       "sourceHandle": "output",
-      "target": "To grayscale 1",
+      "target": "Blur 1",
       "targetHandle": "image"
     }
   ],
@@ -120,51 +120,6 @@
       "type": "image",
       "width": 265.0
     },
-    {
-      "data": {
-        "__execution_delay": null,
-        "collapsed": true,
-        "display": null,
-        "error": null,
-        "input_metadata": null,
-        "meta": {
-          "inputs": {
-            "image": {
-              "name": "image",
-              "position": "left",
-              "type": {
-                "type": "<module 'PIL.Image' from '/media/nvme/darabos/lynxkite-2024/.venv/lib/python3.11/site-packages/PIL/Image.py'>"
-              }
-            }
-          },
-          "name": "Flip verically",
-          "outputs": {
-            "output": {
-              "name": "output",
-              "position": "right",
-              "type": {
-                "type": "None"
-              }
-            }
-          },
-          "params": {},
-          "type": "basic"
-        },
-        "params": {},
-        "status": "done",
-        "title": "Flip verically"
-      },
-      "dragHandle": ".bg-primary",
-      "height": 200.0,
-      "id": "Flip verically 1",
-      "parentId": null,
-      "position": {
-        "x": 228.3853393986406,
-        "y": 245.68255477059915
-      },
-      "type": "basic",
-      "width": 200.0
-    },
     {
       "data": {
         "display": "data:image/jpeg;base64,UklGRoADAABXRUJQVlA4IHQDAAAwNQCdASrIAMgAPm0ukkYyMaGhrtt5UkANiWluu7AObIPJl9NfJmVCwA06FIbr8ZA3WaSJ7/dHH5er4u683qcKWcW7qn4VGE7XCnSEtZUZsarT1ox9y8J2EionbqsxfkAAIOk7JTtvXiLVAeVPQlEW3R4aJSLoxXCJTuGOqcp6fJ7+tzWWmOSq1IqFDN7536TfOvEoiw/K8AQcP06J13RTpwyFow3MFoK9BWZDv0mPqMmHjuC7Hb5MkHc6QnQihSek5PyYdUN4+yG0NpM3NK0+lOSMpu5jaeY5YrFUKke5nLaJ0Mu/3it1akkzebF4fRj8xdIkJDrC4xD4RcyC8aQ1zI4MjffvaCX8X0jbNJ5n5xzGOOU9kFhpry6Uy30xVtGdrEYwbGflOPdNp1u0tiKemkGylZmdj/9rbvWo0IYnCFLs1tAQ1fpkGVQYVo4FA68IbkKvB2KNxrQ17ySt96mzi6ODrNuVzsgisAS/q2ILoOlZL2nfdCNWb523fz7hEM319hb+aWTmo2zExKfWXzUZRGWy09rt8rQZXW5qBKTs5vsYabOjubwNNfFoX7dv32XwAAD++LH6dpCJbVSQouNoBCx0XMzcYw+6o+G//GpIy1eD9A/1cvAp/zhr1mDURDqx2xBgY79tNgsjX9AArhyCNX/BRq/st/uDXRqUv7Wx+Pl/40TsVhng3/XrKZXhDYKnD2aAAyOHG8yS8FTlr5aM6Nh//ra74YvdX+3/j/RLZDm6wRAbHmTF1ZWhK3HdcgOeU+RMXPeCEL8zqEcm3nMeTpuJhH7GTnjejYZx6OrdHAWBghyiaVrgH9f2nXLm1VZ4AgOokXq16NNjRz3FgLYzr7qd94Y2TRJW/cGAcaH5CN5+iXZy4LY2JaCo5I4//pJ/TthQ8SBbQdOIDZPayUXSmAo7cTuD/rk97N3QBGdx5oIzTeSYL8FZfiH/nJLQARX7PafEYZBW9a/ZMjDSPg9GFTPL4BL42F32uL8ZcYLrpkRGRW6jbjmj7GuGSwV5+YuKOhEAf2ZVwU55UjaxRygygrjlYc8FoCe+HyL6rJ5V3jfZvZy1Nw3CdfX4CWQ5/gUjrgONEGb7g2SL/nS9dGgl2gAHKKCzEneCdL38wwsnQYPGdrerT/4uEWM1T24cHLkgBmT+zH6YiwG7DkqEmAbjF4AAAA==",
@@ -299,6 +254,50 @@
       },
       "type": "basic",
       "width": 200.0
+    },
+    {
+      "data": {
+        "__execution_delay": null,
+        "collapsed": true,
+        "display": null,
+        "error": null,
+        "input_metadata": null,
+        "meta": {
+          "inputs": {
+            "image": {
+              "name": "image",
+              "position": "left",
+              "type": {
+                "type": "<module 'PIL.Image' from '/media/nvme/darabos/lynxkite-2024/.venv/lib/python3.11/site-packages/PIL/Image.py'>"
+              }
+            }
+          },
+          "name": "Flip vertically",
+          "outputs": {
+            "output": {
+              "name": "output",
+              "position": "right",
+              "type": {
+                "type": "None"
+              }
+            }
+          },
+          "params": {},
+          "type": "basic"
+        },
+        "params": {},
+        "status": "done",
+        "title": "Flip vertically"
+      },
+      "dragHandle": ".bg-primary",
+      "height": 200.0,
+      "id": "Flip vertically 1",
+      "position": {
+        "x": 148.51544517498044,
+        "y": 288.98657171134255
+      },
+      "type": "basic",
+      "width": 200.0
     }
   ]
 }

examples/Model use.lynxkite.json

@@ -421,7 +421,7 @@
           "type": "basic"
         },
         "params": {
-          "model_workspace": "Model definition.lynxkite.json",
+          "model_workspace": "Model definition",
           "save_as": "model"
         },
         "status": "done",

examples/PyTorch demo.lynxkite.json

@@ -1,623 +0,0 @@
-{
-  "env": "PyTorch model",
-  "nodes": [
-    {
-      "id": "Input: features 1",
-      "type": "basic",
-      "data": {
-        "title": "Input: features",
-        "params": {},
-        "display": null,
-        "error": null,
-        "meta": {
-          "name": "Input: features",
-          "type": "basic",
-          "outputs": {
-            "x": {
-              "position": "top",
-              "name": "x",
-              "type": {
-                "type": "tensor"
-              }
-            }
-          },
-          "params": {},
-          "inputs": {}
-        },
-        "collapsed": true,
-        "__execution_delay": null
-      },
-      "position": {
-        "x": -108.60604658638658,
-        "y": 63.96065124378427
-      },
-      "width": 200.0,
-      "height": 200.0,
-      "parentId": null
-    },
-    {
-      "id": "Input: graph edges 1",
-      "type": "basic",
-      "data": {
-        "title": "Input: graph edges",
-        "params": {},
-        "display": null,
-        "error": null,
-        "__execution_delay": null,
-        "collapsed": true,
-        "meta": {
-          "name": "Input: graph edges",
-          "inputs": {},
-          "params": {},
-          "type": "basic",
-          "outputs": {
-            "edges": {
-              "name": "edges",
-              "type": {
-                "type": "tensor"
-              },
-              "position": "top"
-            }
-          }
-        }
-      },
-      "position": {
-        "x": 180.7373888617958,
-        "y": 58.54904654355781
-      },
-      "width": 200.0,
-      "parentId": null,
-      "height": 200.0
-    },
-    {
-      "id": "Linear 1",
-      "type": "basic",
-      "data": {
-        "title": "Linear",
-        "params": {
-          "output_dim": "same"
-        },
-        "display": null,
-        "error": null,
-        "meta": {
-          "inputs": {
-            "x": {
-              "type": {
-                "type": "tensor"
-              },
-              "position": "bottom",
-              "name": "x"
-            }
-          },
-          "type": "basic",
-          "name": "Linear",
-          "outputs": {
-            "x": {
-              "type": {
-                "type": "tensor"
-              },
-              "position": "top",
-              "name": "x"
-            }
-          },
-          "params": {
-            "output_dim": {
-              "name": "output_dim",
-              "type": {
-                "type": "<class 'str'>"
-              },
-              "default": "same"
-            }
-          }
-        }
-      },
-      "position": {
-        "x": 78.37881963123723,
-        "y": -528.3012263817914
-      },
-      "parentId": null,
-      "width": 204.0,
-      "height": 140.0
-    },
-    {
-      "id": "Activation 1",
-      "type": "basic",
-      "data": {
-        "title": "Activation",
-        "params": {
-          "type": "ReLU"
-        },
-        "display": null,
-        "error": null,
-        "meta": {
-          "outputs": {
-            "x": {
-              "position": "top",
-              "type": {
-                "type": "tensor"
-              },
-              "name": "x"
-            }
-          },
-          "name": "Activation",
-          "type": "basic",
-          "inputs": {
-            "x": {
-              "position": "bottom",
-              "name": "x",
-              "type": {
-                "type": "tensor"
-              }
-            }
-          },
-          "params": {
-            "type": {
-              "default": "OptionsFor_type.ReLU",
-              "name": "type",
-              "type": {
-                "enum": [
-                  "ReLU",
-                  "LeakyReLU",
-                  "Tanh",
-                  "Mish"
-                ]
-              }
-            }
-          }
-        }
-      },
-      "position": {
-        "x": 98.44658319023353,
-        "y": -741.1411130550297
-      },
-      "height": 154.0,
-      "parentId": null,
-      "width": 173.0
-    },
-    {
-      "id": "Dropout 1",
-      "type": "basic",
-      "data": {
-        "title": "Dropout",
-        "params": {
-          "p": 0.5
-        },
-        "display": null,
-        "error": null,
-        "meta": {
-          "type": "basic",
-          "inputs": {
-            "x": {
-              "name": "x",
-              "type": {
-                "type": "tensor"
-              },
-              "position": "bottom"
-            }
-          },
-          "name": "Dropout",
-          "params": {
-            "p": {
-              "default": 0.5,
-              "type": {
-                "type": "<class 'float'>"
-              },
-              "name": "p"
-            }
-          },
-          "outputs": {
-            "x": {
-              "position": "top",
-              "name": "x",
-              "type": {
-                "type": "tensor"
-              }
-            }
-          }
-        }
-      },
-      "position": {
-        "x": 73.61437458187964,
-        "y": -929.9824215609918
-      },
-      "width": 207.0,
-      "parentId": null,
-      "height": 159.0
-    },
-    {
-      "id": "Graph conv 1",
-      "type": "basic",
-      "data": {
-        "title": "Graph conv",
-        "params": {
-          "type": "SAGEConv"
-        },
-        "display": null,
-        "error": null,
-        "meta": {
-          "outputs": {
-            "x": {
-              "type": {
-                "type": "tensor"
-              },
-              "name": "x",
-              "position": "top"
-            }
-          },
-          "type": "basic",
-          "name": "Graph conv",
-          "params": {
-            "type": {
-              "name": "type",
-              "default": "OptionsFor_type.GCNConv",
-              "type": {
-                "enum": [
-                  "GCNConv",
-                  "GATConv",
-                  "GATv2Conv",
-                  "SAGEConv"
-                ]
-              }
-            }
-          },
-          "inputs": {
-            "x": {
-              "type": {
-                "type": "tensor"
-              },
-              "name": "x",
-              "position": "bottom"
-            },
-            "edges": {
-              "name": "edges",
-              "type": {
-                "type": "tensor"
-              },
-              "position": "bottom"
-            }
-          }
-        }
-      },
-      "position": {
-        "x": 64.08886242755246,
-        "y": -269.43023573181557
-      },
-      "height": 200.0,
-      "width": 200.0,
-      "parentId": null
-    },
-    {
-      "id": "Supervised loss 1",
-      "type": "basic",
-      "data": {
-        "title": "Supervised loss",
-        "params": {},
-        "display": null,
-        "error": null,
-        "collapsed": true,
-        "__execution_delay": null,
-        "meta": {
-          "type": "basic",
-          "params": {},
-          "name": "Supervised loss",
-          "inputs": {
-            "y": {
-              "name": "y",
-              "type": {
-                "type": "tensor"
-              },
-              "position": "bottom"
-            },
-            "x": {
-              "name": "x",
-              "type": {
-                "type": "tensor"
-              },
-              "position": "bottom"
-            }
-          },
-          "outputs": {
-            "loss": {
-              "position": "top",
-              "type": {
-                "type": "tensor"
-              },
-              "name": "loss"
-            }
-          }
-        }
-      },
-      "position": {
-        "x": 110.53693593362718,
-        "y": -1123.9976567905628
-      },
-      "parentId": null,
-      "height": 80.0,
-      "width": 204.0
-    },
-    {
-      "id": "Input: label 1",
-      "type": "basic",
-      "data": {
-        "title": "Input: label",
-        "params": {},
-        "display": null,
-        "error": null,
-        "collapsed": true,
-        "__execution_delay": null,
-        "meta": {
-          "inputs": {},
-          "outputs": {
-            "y": {
-              "type": {
-                "type": "tensor"
-              },
-              "position": "top",
-              "name": "y"
-            }
-          },
-          "name": "Input: label",
-          "type": "basic",
-          "params": {}
-        }
-      },
-      "position": {
-        "x": 666.110498676668,
-        "y": -898.6721561114967
-      },
-      "width": 200.0,
-      "height": 73.0,
-      "parentId": null
-    },
-    {
-      "id": "Optimizer 1",
-      "type": "basic",
-      "data": {
-        "title": "Optimizer",
-        "params": {
-          "lr": 0.001,
-          "type": "AdamW"
-        },
-        "display": null,
-        "error": null,
-        "meta": {
-          "name": "Optimizer",
-          "outputs": {},
-          "params": {
-            "lr": {
-              "default": 0.001,
-              "name": "lr",
-              "type": {
-                "type": "<class 'float'>"
-              }
-            },
-            "type": {
-              "type": {
-                "enum": [
-                  "AdamW",
-                  "Adafactor",
-                  "Adagrad",
-                  "SGD",
-                  "Lion",
-                  "Paged AdamW",
-                  "Galore AdamW"
-                ]
-              },
-              "name": "type",
-              "default": "OptionsFor_type.AdamW"
-            }
-          },
-          "inputs": {
-            "loss": {
-              "type": {
-                "type": "tensor"
-              },
-              "name": "loss",
-              "position": "bottom"
-            }
-          },
-          "type": "basic"
-        }
-      },
-      "position": {
-        "x": 115.25730958703494,
-        "y": -1431.5320892753364
-      },
-      "width": 200.0,
-      "height": 233.0,
-      "parentId": null
-    },
-    {
-      "id": "Repeat 3",
-      "type": "basic",
-      "data": {
-        "title": "Repeat",
-        "params": {
-          "times": 1.0
-        },
-        "display": null,
-        "error": null,
-        "meta": {
-          "type": "basic",
-          "position": {
-            "y": 340.0,
-            "x": 371.0
-          },
-          "outputs": {
-            "output": {
-              "name": "output",
-              "type": {
-                "type": "tensor"
-              },
-              "position": "bottom"
-            }
-          },
-          "name": "Repeat",
-          "params": {
-            "times": {
-              "name": "times",
-              "default": 1.0,
-              "type": {
-                "type": "<class 'int'>"
-              }
-            }
-          },
-          "inputs": {
-            "input": {
-              "type": {
-                "type": "tensor"
-              },
-              "position": "top",
-              "name": "input"
-            }
-          }
-        }
-      },
-      "position": {
-        "x": -245.1288628776232,
-        "y": -276.90661040974317
-      },
-      "width": 200.0,
-      "height": 200.0
-    },
-    {
-      "id": "Repeat 1",
-      "type": "basic",
-      "data": {
-        "title": "Repeat",
-        "params": {
-          "times": 1.0
-        },
-        "display": null,
-        "error": null,
-        "meta": {
-          "outputs": {
-            "output": {
-              "position": "bottom",
-              "type": {
-                "type": "tensor"
-              },
-              "name": "output"
-            }
-          },
-          "params": {
-            "times": {
-              "default": 1.0,
-              "type": {
-                "type": "<class 'int'>"
-              },
-              "name": "times"
-            }
-          },
-          "position": {
-            "x": 387.0,
-            "y": 337.0
-          },
-          "type": "basic",
-          "name": "Repeat",
-          "inputs": {
-            "input": {
-              "name": "input",
-              "position": "top",
-              "type": {
-                "type": "tensor"
-              }
-            }
-          }
-        }
-      },
-      "position": {
-        "x": -258.0088683218416,
-        "y": -737.3822225246788
-      },
-      "width": 200.0,
-      "height": 200.0
-    }
-  ],
-  "edges": [
-    {
-      "id": "xy-edge__Linear 1x-Activation 1x",
-      "source": "Linear 1",
-      "target": "Activation 1",
-      "sourceHandle": "x",
-      "targetHandle": "x"
-    },
-    {
-      "id": "xy-edge__Activation 1x-Dropout 1x",
-      "source": "Activation 1",
-      "target": "Dropout 1",
-      "sourceHandle": "x",
-      "targetHandle": "x"
-    },
-    {
-      "id": "xy-edge__Input: features 1x-Graph conv 1x",
-      "source": "Input: features 1",
-      "target": "Graph conv 1",
-      "sourceHandle": "x",
-      "targetHandle": "x"
-    },
-    {
-      "id": "xy-edge__Input: graph edges 1edges-Graph conv 1edges",
-      "source": "Input: graph edges 1",
-      "target": "Graph conv 1",
-      "sourceHandle": "edges",
-      "targetHandle": "edges"
-    },
-    {
-      "id": "xy-edge__Graph conv 1x-Linear 1x",
-      "source": "Graph conv 1",
-      "target": "Linear 1",
-      "sourceHandle": "x",
-      "targetHandle": "x"
-    },
-    {
-      "id": "xy-edge__Input: label 1y-Supervised loss 1y",
-      "source": "Input: label 1",
-      "target": "Supervised loss 1",
-      "sourceHandle": "y",
-      "targetHandle": "y"
-    },
-    {
-      "id": "xy-edge__Dropout 1x-Supervised loss 1x",
-      "source": "Dropout 1",
-      "target": "Supervised loss 1",
-      "sourceHandle": "x",
-      "targetHandle": "x"
-    },
-    {
-      "id": "xy-edge__Supervised loss 1loss-Optimizer 1loss",
-      "source": "Supervised loss 1",
-      "target": "Optimizer 1",
-      "sourceHandle": "loss",
-      "targetHandle": "loss"
-    },
-    {
-      "id": "Graph conv 1 Repeat 3",
-      "source": "Graph conv 1",
-      "target": "Repeat 3",
-      "sourceHandle": "x",
-      "targetHandle": "input"
-    },
-    {
-      "id": "Repeat 3 Graph conv 1",
-      "source": "Repeat 3",
-      "target": "Graph conv 1",
-      "sourceHandle": "output",
-      "targetHandle": "x"
-    },
-    {
-      "id": "Dropout 1 Repeat 1",
-      "source": "Dropout 1",
-      "target": "Repeat 1",
-      "sourceHandle": "x",
-      "targetHandle": "input"
-    },
-    {
-      "id": "Repeat 1 Linear 1",
-      "source": "Repeat 1",
-      "target": "Linear 1",
-      "sourceHandle": "output",
-      "targetHandle": "x"
-    }
-  ]
-}

examples/requirements.txt

@@ -1 +1,2 @@
-Faker
+# Example of a requirements.txt file. LynxKite will automatically install anything you put here.
+faker

examples/word2vec.py

@@ -1,5 +1,4 @@
 from lynxkite.core.ops import op
-import staticvectors
 import pandas as pd
 
 ENV = "LynxKite Graph Analytics"
@@ -7,6 +6,8 @@ ENV = "LynxKite Graph Analytics"
 
 @op(ENV, "Word2vec for the top 1000 words", slow=True)
 def word2vec_1000():
+    import staticvectors
+
     model = staticvectors.StaticVectors("neuml/word2vec-quantized")
     df = pd.read_csv(
         "https://gist.githubusercontent.com/deekayen/4148741/raw/98d35708fa344717d8eee15d11987de6c8e26d7d/1-1000.txt",

lynxkite-app/src/lynxkite_app/main.py

@@ -151,6 +151,14 @@ async def upload(req: fastapi.Request):
     return {"status": "ok"}
 
 
+@app.post("/api/execute_workspace")
+async def execute_workspace(name: str):
+    """Trigger and await the execution of a workspace."""
+    room = await crdt.ws_websocket_server.get_room(name)
+    ws_pyd = workspace.Workspace.model_validate(room.ws.to_py())
+    await crdt.execute(name, room.ws, ws_pyd)
+
+
 class SPAStaticFiles(StaticFiles):
     """Route everything to index.html. https://stackoverflow.com/a/73552966/3318517"""
 

lynxkite-app/web/src/Code.tsx

@@ -71,13 +71,13 @@ export default function Code() {
         </a>
         <div className="ws-name">{path}</div>
         <div className="tools text-secondary">
-          <a href="">
+          <button className="btn btn-link">
             <Atom />
-          </a>
-          <a href="">
+          </button>
+          <button className="btn btn-link">
             <Backspace />
-          </a>
-          <a href={`/dir/${parentDir}`}>
+          </button>
+          <a href={`/dir/${parentDir}`} className="btn btn-link">
             <Close />
           </a>
         </div>

lynxkite-app/web/src/index.css

@@ -54,7 +54,7 @@ body {
       display: flex;
       align-items: center;
 
-      a {
+      .btn {
         color: oklch(75% 0.13 230);
         font-size: 1.5em;
         padding: 0 10px;

lynxkite-app/web/src/workspace/Workspace.tsx

@@ -26,6 +26,8 @@ import Atom from "~icons/tabler/atom.jsx";
 // @ts-ignore
 import Backspace from "~icons/tabler/backspace.jsx";
 // @ts-ignore
+import Restart from "~icons/tabler/rotate-clockwise.jsx";
+// @ts-ignore
 import Close from "~icons/tabler/x.jsx";
 import type { Workspace, WorkspaceNode } from "../apiTypes.ts";
 import favicon from "../assets/favicon.ico";
@@ -181,12 +183,16 @@ function LynxKiteFlow() {
   useEffect(() => {
     const handleKeyDown = (event: KeyboardEvent) => {
       // Show the node search dialog on "/".
-      if (event.key === "/" && !nodeSearchSettings && !isTypingInFormElement()) {
+      if (nodeSearchSettings || isTypingInFormElement()) return;
+      if (event.key === "/") {
         event.preventDefault();
         setNodeSearchSettings({
           pos: { x: 100, y: 100 },
           boxes: catalog.data![state.workspace.env!],
         });
+      } else if (event.key === "r") {
+        event.preventDefault();
+        executeWorkspace();
       }
     };
     // TODO: Switch to keydown once https://github.com/xyflow/xyflow/pull/5055 is merged.
@@ -318,6 +324,12 @@ function LynxKiteFlow() {
       setMessage("File upload failed.");
     }
   }
+  async function executeWorkspace() {
+    const response = await axios.post(`/api/execute_workspace?name=${path}`);
+    if (response.status !== 200) {
+      setMessage("Workspace execution failed.");
+    }
+  }
   return (
     <div className="workspace">
       <div className="top-bar bg-neutral">
@@ -334,13 +346,16 @@ function LynxKiteFlow() {
           }}
         />
         <div className="tools text-secondary">
-          <a href="">
+          <button className="btn btn-link">
             <Atom />
-          </a>
-          <a href="">
+          </button>
+          <button className="btn btn-link">
             <Backspace />
-          </a>
-          <a href={`/dir/${parentDir}`}>
+          </button>
+          <button className="btn btn-link" onClick={executeWorkspace}>
+            <Restart />
+          </button>
+          <a className="btn btn-link" href={`/dir/${parentDir}`}>
             <Close />
           </a>
         </div>

lynxkite-app/web/tests/examples.spec.ts

@@ -2,44 +2,21 @@
 import { expect, test } from "@playwright/test";
 import { Workspace } from "./lynxkite";
 
-test("LynxKite Graph Analytics example", async ({ page }) => {
-  const ws = await Workspace.open(page, "NetworkX demo");
-  await ws.expectErrorFree(process.env.CI ? 2000 : 1000);
-});
-
-test("Bio example", async ({ page }) => {
-  const ws = await Workspace.open(page, "Bio demo");
-  await ws.expectErrorFree();
-});
-
-test("Pytorch example", async ({ page }) => {
-  const ws = await Workspace.open(page, "PyTorch demo");
-  await ws.expectErrorFree();
-});
-
-test("AIMO example", async ({ page }) => {
-  const ws = await Workspace.open(page, "AIMO");
-  await ws.expectErrorFree();
-});
-
-test("LynxScribe example", async ({ page }) => {
-  // Fails because of missing OPENAI_API_KEY
-  const ws = await Workspace.open(page, "LynxScribe demo");
-  await ws.expectErrorFree();
-});
-
-test("Graph RAG", async ({ page }) => {
-  // Fails due to some issue with ChromaDB
-  const ws = await Workspace.open(page, "Graph RAG");
-  await ws.expectErrorFree(process.env.CI ? 2000 : 500);
-});
-
-test("Airlines demo", async ({ page }) => {
-  const ws = await Workspace.open(page, "Airlines demo");
-  await ws.expectErrorFree(process.env.CI ? 10000 : 500);
-});
-
-test("Pillow example", async ({ page }) => {
-  const ws = await Workspace.open(page, "Image processing");
-  await ws.expectErrorFree();
-});
+const WORKSPACES = [
+  // "AIMO",
+  "Airlines demo",
+  "Bio Cypher demo",
+  // "Graph RAG",
+  "Image processing",
+  // "LynxScribe demo",
+  "NetworkX demo",
+  "Model use",
+];
+
+for (const name of WORKSPACES) {
+  test(name, async ({ page }) => {
+    const ws = await Workspace.open(page, name);
+    await ws.execute();
+    await ws.expectErrorFree();
+  });
+}

lynxkite-app/web/tests/lynxkite.ts

@@ -144,8 +144,14 @@ export class Workspace {
     await this.page.mouse.up();
   }
 
+  async execute() {
+    const request = this.page.waitForResponse(/api[/]execute_workspace/);
+    await this.page.keyboard.press("r");
+    await request;
+  }
+
   async expectErrorFree(executionWaitTime?) {
-    await expect(this.getBoxes().locator(".error").first()).not.toBeVisible();
+    await expect(this.getBoxes().locator("text=⚠️").first()).not.toBeVisible();
   }
 
   async close() {

lynxkite-core/src/lynxkite/core/ops.py

@@ -344,29 +344,27 @@ def load_user_scripts(workspace: str):
     assert path.is_relative_to(cwd), "Provided workspace path is invalid"
     for p in path.parents:
         print("checking user scripts in", p)
-        for f in p.glob("*.py"):
-            try:
-                run_user_script(f)
-            except Exception:
-                traceback.print_exc()
         req = p / "requirements.txt"
         if req.exists():
             try:
                 install_requirements(req)
             except Exception:
                 traceback.print_exc()
+        for f in p.glob("*.py"):
+            try:
+                run_user_script(f)
+            except Exception:
+                traceback.print_exc()
         if p == cwd:
             break
 
 
 def install_requirements(req: pathlib.Path):
-    cmd = ["uv", "pip", "install", "-r", str(req)]
-    print(f"Running {' '.join(cmd)}")
+    cmd = ["uv", "pip", "install", "-q", "-r", str(req)]
     subprocess.check_call(cmd)
 
 
 def run_user_script(script_path: pathlib.Path):
-    print(f"Running {script_path}...")
     spec = importlib.util.spec_from_file_location(script_path.stem, str(script_path))
     module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(module)

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

@@ -152,11 +152,11 @@ def types_from_doc(doc: str) -> dict[str, type]:
 
 def wrapped(name: str, func):
     @functools.wraps(func)
-    def wrapper(*args, **kwargs):
+    async def wrapper(*args, **kwargs):
         for k, v in kwargs.items():
             if v == "None":
                 kwargs[k] = None
-        res = func(*args, **kwargs)
+        res = await ops.slow(func)(*args, **kwargs)
         # Figure out what the returned value is.
         if isinstance(res, nx.Graph):
             return res