Register operations into categories, and a rough UI for displaying them.

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

@@ -3,6 +3,7 @@ import { useEffect, useMemo, useRef, useState } from "react";
 
 export type OpsOp = {
   name: string;
+  categories: string[];
   type: string;
   position: { x: number; y: number };
   params: { name: string; default: any }[];
@@ -81,6 +82,7 @@ export default function NodeSearch(props: {
             onClick={addSelected}
             className={`search-result ${index === selectedIndex ? "selected" : ""}`}
           >
+            {box.item.categories.map((category) => `${category}\u00A0›\u00A0`)}
             {box.item.name}
           </div>
         ))}

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

@@ -184,6 +184,7 @@ def _param_to_type(name, value, type):
 
 class Op(BaseConfig):
     func: typing.Callable = pydantic.Field(exclude=True)
+    categories: list[str]
     name: str
     params: list[Parameter | ParameterGroup]
     inputs: list[Input]
@@ -234,11 +235,15 @@ class Op(BaseConfig):
                 res[p.name] = _param_to_type(p.name, params[p.name], p.type)
         return res
 
+    @property
+    def id(self) -> str:
+        """The name and categories of the operation."""
+        return " > ".join(self.categories + [self.name])
+
 
 def op(
     env: str,
-    name: str,
-    *,
+    *names: str,
     view="basic",
     outputs=None,
     params=None,
@@ -247,6 +252,7 @@ def op(
     cache=None,
 ):
     """Decorator for defining an operation."""
+    [*categories, name] = names
 
     def decorator(func):
         doc = parse_doc(func)
@@ -279,6 +285,7 @@ def op(
             func=func,
             doc=doc,
             name=name,
+            categories=categories,
             params=_params,
             inputs=inputs,
             outputs=_outputs,
@@ -286,7 +293,7 @@ def op(
             color=color or "orange",
         )
         CATALOGS.setdefault(env, {})
-        CATALOGS[env][name] = op
+        CATALOGS[env][op.id] = op
         func.__op__ = op
         return func
 
@@ -364,11 +371,13 @@ def no_op(*args, **kwargs):
     return None
 
 
-def register_passive_op(env: str, name: str, inputs=[], outputs=["output"], params=[], **kwargs):
+def register_passive_op(env: str, *names: str, inputs=[], outputs=["output"], params=[], **kwargs):
     """A passive operation has no associated code."""
+    [*categories, name] = names
     op = Op(
         func=no_op,
         name=name,
+        categories=categories,
         params=params,
         inputs=[Input(name=i, type=None) if isinstance(i, str) else i for i in inputs],
         outputs=[Output(name=o, type=None) if isinstance(o, str) else o for o in outputs],
@@ -382,6 +391,7 @@ def register_passive_op(env: str, name: str, inputs=[], outputs=["output"], para
 COMMENT_OP = Op(
     func=no_op,
     name="Comment",
+    categories=[],
     params=[Parameter.basic("text", "", LongStr)],
     inputs=[],
     outputs=[],

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

@@ -242,6 +242,41 @@ _REPLACEMENTS = [
     ("Watts Strogatz", "Watts–Strogatz"),
     ("Weisfeiler Lehman", "Weisfeiler–Lehman"),
 ]
+_CATEGORY_REPLACEMENTS = [
+    ("Networkx", "NetworkX"),
+    ("D separation", "D-separation"),
+    ("Dag", "DAG"),
+    ("Pagerank alg", "PageRank alg"),
+    ("Richclub", "Rich-club"),
+    ("Smallworld", "Small-world"),
+    ("Smetric", "S-metric"),
+    ("Structuralholes", "Structural holes"),
+    ("Edgedfs", "Edge DFS"),
+    ("Edgebfs", "Edge BFS"),
+    ("Edge_kcomponents", "Edge k-components"),
+    ("Mincost", "Min cost"),
+    ("Networksimplex", "Network simplex"),
+    ("Vf2pp", "VF2++"),
+    ("Mst", "MST"),
+    ("Attrmatrix", "Attr matrix"),
+    ("Graphmatrix", "Graph matrix"),
+    ("Laplacianmatrix", "Laplacian matrix"),
+    ("Algebraicconnectivity", "Algebraic connectivity"),
+    ("Modularitymatrix", "Modularity matrix"),
+    ("Bethehessianmatrix", "Bethe–Hessian matrix"),
+]
+
+
+def _categories(func) -> list[str]:
+    """Extract categories from the function's docstring."""
+    path = func.__module__.split(".")
+    cats = []
+    for p in path:
+        p = p.replace("_", " ").capitalize()
+        for a, b in _CATEGORY_REPLACEMENTS:
+            p = p.replace(a, b)
+        cats.append(p)
+    return cats
 
 
 def register_networkx(env: str):
@@ -254,12 +289,13 @@ def register_networkx(env: str):
             except UnsupportedParameterType:
                 continue
             inputs = [ops.Input(name=k, type=nx.Graph) for k in func.graphs]
-            nicename = "NX › " + name.replace("_", " ").title()
+            nicename = name.replace("_", " ").title()
             for a, b in _REPLACEMENTS:
                 nicename = nicename.replace(a, b)
             op = ops.Op(
                 func=wrapped(name, func),
                 name=nicename,
+                categories=_categories(func),
                 params=params,
                 inputs=inputs,
                 outputs=[ops.Output(name="output", type=nx.Graph)],