Fix some bugs.

server/ops.py

@@ -22,7 +22,9 @@ class Parameter:
     return Parameter(name, options[0], enum.Enum(f'OptionsFor{name}', options))
 
   def __post_init__(self):
-    if self.type is None:
+    if self.default is inspect._empty:
+      self.default = None
+    if self.type is None or self.type is inspect._empty:
       self.type = type(self.default)
   def to_json(self):
     if isinstance(self.type, type) and issubclass(self.type, enum.Enum):
@@ -47,11 +49,11 @@ class Op:
 
   def __call__(self, *inputs, **params):
     # Convert parameters.
-    for p in params.values():
+    for p in params:
       if p in self.params:
-        if p.type == int:
+        if self.params[p].type == int:
           params[p] = int(params[p])
-        elif p.type == float:
+        elif self.params[p].type == float:
           params[p] = float(params[p])
     # Convert inputs.
     inputs = list(inputs)
@@ -150,12 +152,7 @@ def op(name, *, view='basic', sub_nodes=None):
     params = {}
     for n, param in sig.parameters.items():
       if param.kind == param.KEYWORD_ONLY:
-        p = Parameter(n, param.default, param.annotation)
-        if p.default is inspect._empty:
-          p.default = None
-        if p.type is inspect._empty:
-          p.type = type(p.default)
-        params[n] = p
+        params[n] = Parameter(n, param.default, param.annotation)
     outputs = {'output': 'yes'} if view == 'basic' else {} # Maybe more fancy later.
     op = Op(func, name, params=params, inputs=inputs, outputs=outputs, type=view)
     if sub_nodes is not None:

server/pytorch_model_ops.py

@@ -23,7 +23,7 @@ def register_layer(name):
       for name, param in sig.parameters.items()
       if param.kind != param.KEYWORD_ONLY}
     params = {
-      name: param.default if param.default is not inspect._empty else None
+      name: ops.Parameter(name, param.default, param.annotation)
       for name, param in sig.parameters.items()
       if param.kind == param.KEYWORD_ONLY}
     outputs = {'x': 'tensor'}

server/workspace.py

@@ -81,7 +81,9 @@ def execute(ws):
 
 def save(ws: Workspace, path: str):
     j = ws.model_dump_json(indent=2)
-    with tempfile.NamedTemporaryFile('w', delete_on_close=False) as f:
+    dirname, basename = os.path.split(path)
+    # Create temp file in the same directory to make sure it's on the same filesystem.
+    with tempfile.NamedTemporaryFile('w', prefix=basename, dir=dirname, delete_on_close=False) as f:
         f.write(j)
         f.close()
         os.replace(f.name, path)

web/src/LynxKiteFlow.svelte

@@ -65,6 +65,7 @@
     const node = {...e.detail};
     nodes.update((n) => {
       node.position = screenToFlowPosition({x: nodeSearchSettings.pos.x, y: nodeSearchSettings.pos.y});
+      node.data = { ...node.data };
       const title = node.data.title;
       node.data.params = Object.fromEntries(
         node.data.params.map((p) => [p.name, p.default]));