Ensure enums can be serialized.

lynxkite-app/tests/test_crdt.py

@@ -18,7 +18,7 @@ def empty_list_workspace():
     yield ws
 
 
-class MyEnum(Enum):
+class MyEnum(int, Enum):
     VALUE = 1
 
 

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

@@ -95,7 +95,7 @@ class ParameterGroup(BaseConfig):
     type: str = "group"
 
 
-class Position(enum.Enum):
+class Position(str, enum.Enum):
     """Defines the position of an input or output in the UI."""
 
     LEFT = "left"

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

@@ -175,8 +175,7 @@ class Workspace(BaseConfig):
                     data.meta = op
                     # If the node is connected to a CRDT, update that too.
                     if hasattr(node, "_crdt"):
-                        # Go through JSON to simplify the types. CRDT can't handle enums.
-                        node._crdt["data"]["meta"] = json.loads(op.model_dump_json())
+                        node._crdt["data"]["meta"] = op.model_dump()
                         print("set metadata to", op)
                 if node.type != op.type:
                     node.type = op.type

lynxkite-core/tests/test_ops.py

@@ -54,7 +54,7 @@ def test_op_decorator_with_params_and_types_():
 
 
 def test_op_decorator_with_complex_types():
-    class Color(enum.Enum):
+    class Color(int, enum.Enum):
         RED = 1
         GREEN = 2
         BLUE = 3

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

@@ -148,7 +148,7 @@ VIRIDIS = [
 ]
 
 
-class UMAPMetric(enum.Enum):
+class UMAPMetric(str, enum.Enum):
     l1 = "l1"
     cityblock = "cityblock"
     taxicab = "taxicab"

lynxkite-graph-analytics/src/lynxkite_graph_analytics/pytorch/pytorch_ops.py

@@ -65,7 +65,7 @@ def linear(x, *, output_dim=1024):
     return pyg_nn.Linear(-1, output_dim)
 
 
-class ActivationTypes(enum.Enum):
+class ActivationTypes(str, enum.Enum):
     ReLU = "ReLU"
     Leaky_ReLU = "Leaky ReLU"
     Tanh = "Tanh"

lynxkite-lynxscribe/src/lynxkite_lynxscribe/llm_ops.py

@@ -179,7 +179,7 @@ def branch(input, *, expression: str):
     return one_by_one.Output(output_handle=str(bool(res)).lower(), value=input)
 
 
-class RagEngine(enum.Enum):
+class RagEngine(str, enum.Enum):
     Chroma = "Chroma"
     Custom = "Custom"
 

lynxkite-lynxscribe/src/lynxkite_lynxscribe/lynxscribe_ops.py

@@ -45,18 +45,18 @@ output_on_top = ops.output_position(output="top")
 
 
 # defining the cloud provider enum
-class CloudProvider(Enum):
+class CloudProvider(str, Enum):
     GCP = "gcp"
     AWS = "aws"
     AZURE = "azure"
 
 
-class RAGVersion(Enum):
+class RAGVersion(str, Enum):
     V1 = "v1"
     V2 = "v2"
 
 
-class MessageRole(Enum):
+class MessageRole(str, Enum):
     SYSTEM = "system"
     USER = "user"