Make variants internal

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/tool_building/variant.py

@@ -16,9 +16,7 @@ def asdict(self) -> Dict[str, Any]:
 
 
 @dataclass
-class VariantWithActions:
-    id: int
-    name: str
+class VariantWithActions(Variant):
     actions: List[str] = field(default_factory=list)
     _available_actions: Set[str] = field(default_factory=set)
 
@@ -32,11 +30,7 @@ def reset_actions(self) -> None:
         self.actions = []
 
     def asdict(self) -> Dict[str, Any]:
-        data = {
-            "id": self.id,
-            "name": self.name,
-        }
-        if len(self.actions) > 0:
-            data["actions"] = self.actions
+        data = super().asdict()
+        data["actions"] = self.actions
 
         return data

libs/labelbox/tests/unit/test_unit_fact_checking_tool.py

@@ -41,3 +41,47 @@ def test_fact_checking_as_dict_default():
     }
 
     assert tool_dict == expected_dict
+
+
+def test_step_reasoning_as_dict_with_actions():
+    tool = FactCheckingTool(name="Fact Checking Tool")
+    tool.set_unsupported_step_actions([])
+    tool.set_cant_confidently_assess_step_actions([])
+    tool.set_no_factual_information_step_actions([])
+
+    # Get the dictionary representation
+    tool_dict = tool.asdict()
+
+    # Expected dictionary structure
+    expected_dict = {
+        "tool": "fact-checking",
+        "name": "Fact Checking Tool",
+        "required": False,
+        "schemaNodeId": None,
+        "featureSchemaId": None,
+        "definition": {
+            "variants": [
+                {"id": 0, "name": "Accurate"},
+                {"id": 1, "name": "Inaccurate"},
+                {"id": 2, "name": "Disputed"},
+                {
+                    "id": 3,
+                    "name": "Unsupported",
+                    "actions": [],
+                },
+                {
+                    "id": 4,
+                    "name": "Can't confidently assess",
+                    "actions": [],
+                },
+                {
+                    "id": 5,
+                    "name": "No factual information",
+                    "actions": [],
+                },
+            ],
+            "version": 1,
+        },
+    }
+
+    assert tool_dict == expected_dict