Manage an "in_progress" property on each node.

lynxkite-app/src/lynxkite_app/crdt.py

@@ -86,6 +86,7 @@ def clean_input(ws_pyd):
     for node in ws_pyd.nodes:
         node.data.display = None
         node.data.error = None
+        node.data.in_progress = False
         node.position.x = 0
         node.position.y = 0
         if node.model_extra:
@@ -175,7 +176,6 @@ delayed_executions = {}
 async def workspace_changed(name: str, changes: pycrdt.MapEvent, ws_crdt: pycrdt.Map):
     """Callback to react to changes in the workspace.
 
-
     Args:
         name: Name of the workspace.
         changes: Changes performed to the workspace.
@@ -224,18 +224,17 @@ async def execute(
     assert path.is_relative_to(config.DATA_PATH), "Provided workspace path is invalid"
     # Save user changes before executing, in case the execution fails.
     workspace.save(ws_pyd, path)
-    add_crdt_bindings(ws_pyd, ws_crdt)
+    with ws_crdt.doc.transaction():
+        for nc, np in zip(ws_crdt["nodes"], ws_pyd.nodes):
+            if "data" not in nc:
+                nc["data"] = pycrdt.Map()
+            nc["data"]["in_progress"] = True
+            # Nodes get a reference to their CRDT maps, so they can update them as the results come in.
+            np._crdt = nc
     await workspace.execute(ws_pyd)
     workspace.save(ws_pyd, path)
 
 
-def add_crdt_bindings(ws_pyd: workspace.Workspace, ws_crdt: pycrdt.Map):
-    for nc, np in zip(ws_crdt["nodes"], ws_pyd.nodes):
-        if "data" not in nc:
-            nc["data"] = pycrdt.Map()
-        np._crdt = nc
-
-
 @contextlib.asynccontextmanager
 async def lifespan(app):
     global websocket_server

lynxkite-app/web/src/apiTypes.ts

@@ -41,6 +41,7 @@ export interface WorkspaceNodeData {
   };
   display?: unknown;
   error?: string | null;
+  in_progress?: boolean;
   [k: string]: unknown;
 }
 export interface Position {

lynxkite-app/web/src/index.css

@@ -90,11 +90,14 @@ body {
   }
 
   .lynxkite-node .title {
-    /* background: oklch(75% 0.2 55); */
     font-weight: bold;
     padding: 8px;
   }
 
+  .lynxkite-node .title.in-progress {
+    background: yellow;
+  }
+
   .handle-name {
     font-size: 10px;
     color: black;

lynxkite-app/web/src/workspace/nodes/LynxKiteNode.tsx

@@ -71,7 +71,10 @@ export default function LynxKiteNode(props: LynxKiteNodeProps) {
       }}
     >
       <div className="lynxkite-node" style={props.nodeStyle}>
-        <div className="title bg-primary" onClick={titleClicked}>
+        <div
+          className={`title bg-primary ${data.in_progress ? "in-progress" : ""}`}
+          onClick={titleClicked}
+        >
           {data.title}
           {data.error && <span className="title-icon">⚠️</span>}
           {expanded || <span className="title-icon">⋯</span>}

lynxkite-core/src/lynxkite/core/workspace.py

@@ -26,6 +26,7 @@ class WorkspaceNodeData(BaseConfig):
     params: dict
     display: Optional[object] = None
     error: Optional[str] = None
+    in_progress: bool = False
     # Also contains a "meta" field when going out.
     # This is ignored when coming back from the frontend.
 
@@ -44,6 +45,7 @@ class WorkspaceNode(BaseConfig):
         with self._crdt.doc.transaction():
             self._crdt["data"]["display"] = result.display
             self._crdt["data"]["error"] = result.error
+            self._crdt["data"]["in_progress"] = False
 
     def publish_error(self, error: Exception | str):
         self.publish_result(ops.Result(error=str(error)))