Dropdown for enum parameters.

server/lynxscribe_ops.py

@@ -4,7 +4,7 @@ from .ops import register_passive_op, Parameter as P
 register_passive_op('Scrape documents', inputs={}, params=[P('url', '')])
 register_passive_op('Extract graph')
 register_passive_op('Compute embeddings')
-register_passive_op('Vector DB', params=[P('backend', 'FAISS')])
+register_passive_op('Vector DB', params=[P.options('backend', ['FAISS', 'ANN', 'HNSW'])])
 register_passive_op('Chat UI', outputs={})
 register_passive_op('Chat backend', inputs={})
 register_passive_op('WhatsApp', inputs={})

server/networkx_ops.py

@@ -37,8 +37,6 @@ for (name, func) in nx.__dict__.items():
         # No annotation, no default — we must guess the type.
         if len(k) == 1:
           params[k] = 1
-    if name == 'ladder_graph':
-      print(params)
     name = "NX › " + name.replace('_', ' ').title()
     op = ops.Op(wrapped(func), name, params=params, inputs=inputs, outputs={'output': 'yes'}, type='basic')
     ops.ALL_OPS[name] = op

server/ops.py

@@ -1,5 +1,6 @@
 '''API for implementing LynxKite operations.'''
 import dataclasses
+import enum
 import functools
 import inspect
 import networkx as nx
@@ -16,14 +17,22 @@ class Parameter:
   default: any
   type: PARAM_TYPE = None
 
+  @staticmethod
+  def options(name, options):
+    return Parameter(name, options[0], enum.Enum(f'OptionsFor{name}', options))
+
   def __post_init__(self):
     if self.type is None:
       self.type = type(self.default)
   def to_json(self):
+    if isinstance(self.type, type) and issubclass(self.type, enum.Enum):
+      t = {'enum': list(self.type.__members__.keys())}
+    else:
+      t = str(self.type)
     return {
       'name': self.name,
       'default': self.default,
-      'type': str(self.type),
+      'type': t,
     }
 
 @dataclasses.dataclass

server/pytorch_model_ops.py

@@ -32,7 +32,7 @@ def register_layer(name):
   return decorator
 
 @register_layer('LayerNorm')
-def normalization():
+def layernorm():
   return 'LayerNorm'
 
 @register_layer('Dropout')

server/workspace.py

@@ -42,7 +42,6 @@ class Workspace(BaseConfig):
 def execute(ws):
     # Nodes are responsible for interpreting/executing their child nodes.
     nodes = [n for n in ws.nodes if not n.parentNode]
-    print(nodes)
     children = {}
     for n in ws.nodes:
         if n.parentNode:

web/src/LynxKiteFlow.svelte

@@ -1,4 +1,5 @@
 <script lang="ts">
+  import { setContext } from 'svelte';
   import { writable, derived } from 'svelte/store';
   import {
     SvelteFlow,
@@ -139,9 +140,13 @@
         || e.targetHandle !== connection.targetHandle);
     });
   }
+  function getMeta(title) {
+    return $boxes.find((m) => m.data.title === title);
+  }
+  setContext('LynxKiteFlow', { getMeta });
   function nodeClick(e) {
     const node = e.detail.node;
-    const meta = $boxes.find(m => m.data.title === node.data.title);
+    const meta = getMeta(node.data.title);
     if (!meta) return;
     const sub_nodes = meta.sub_nodes;
     if (!sub_nodes) return;

web/src/NodeWithParams.svelte

@@ -1,10 +1,13 @@
 <script lang="ts">
+  import { getContext } from 'svelte';
   import { type NodeProps, useSvelteFlow } from '@xyflow/svelte';
   import LynxKiteNode from './LynxKiteNode.svelte';
   type $$Props = NodeProps;
   export let id: $$Props['id'];
   export let data: $$Props['data'];
   const { updateNodeData } = useSvelteFlow();
+  $: meta = getContext('LynxKiteFlow').getMeta(data.title);
+  $: metaParams = Object.fromEntries(meta.data.params.map((p) => [p.name, p]));
 </script>
 
 <LynxKiteNode {...$$props}>
@@ -12,10 +15,21 @@
     <div class="param">
       <label>
         {name}<br>
-        <input
+        {#if metaParams[name].type.enum}
+          <select
+            value={value}
+            on:change={(evt) => updateNodeData(id, { params: { ...data.params, [name]: evt.currentTarget.value } })}
+          >
+            {#each metaParams[name].type.enum as option}
+              <option value={option}>{option}</option>
+            {/each}
+          </select>
+        {:else}
+          <input
           value={value}
           on:input={(evt) => updateNodeData(id, { params: { ...data.params, [name]: evt.currentTarget.value } })}
-        />
+          />
+        {/if}
       </label>
     </div>
   {/each}
@@ -28,7 +42,8 @@
     font-size: 12px;
     display: block;
   }
-  .param input {
+  .param input,
+  .param select {
     width: calc(100% - 8px);
   }
 </style>