Update FactCheckingTool to match the latest api

Author: Val Brodsky

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

@@ -5,11 +5,14 @@
 
 from labelbox.schema.tool_building.tool_type import ToolType
 from labelbox.schema.tool_building.variant import (
-    Variant,
     VariantWithActions,
 )
 
 
+class FactCheckingActions(Enum):
+    WRITE_JUSTIFICATION = "writeJustification"
+
+
 class UnsupportedStepActions(Enum):
     WRITE_JUSTIFICATION = "writeJustification"
 
@@ -29,43 +32,53 @@ class FactCheckingVariants:
     NOTE do not change the variants directly
     """
 
-    accurate_step: Variant = field(
-        default_factory=lambda: Variant(id=0, name="Accurate")
+    accurate_step: VariantWithActions = field(
+        default_factory=lambda: VariantWithActions(
+            id=0,
+            name="Accurate",
+            _available_actions={action.value for action in FactCheckingActions},
+            actions=[action.value for action in FactCheckingActions],
+        )
     )
-    inaccurate_step: Variant = field(
-        default_factory=lambda: Variant(id=1, name="Inaccurate")
+
+    inaccurate_step: VariantWithActions = field(
+        default_factory=lambda: VariantWithActions(
+            id=1,
+            name="Inaccurate",
+            _available_actions={action.value for action in FactCheckingActions},
+            actions=[action.value for action in FactCheckingActions],
+        )
     )
-    disputed_step: Variant = field(
-        default_factory=lambda: Variant(id=2, name="Disputed")
+    disputed_step: VariantWithActions = field(
+        default_factory=lambda: VariantWithActions(
+            id=2,
+            name="Disputed",
+            _available_actions={action.value for action in FactCheckingActions},
+            actions=[action.value for action in FactCheckingActions],
+        )
     )
     unsupported_step: VariantWithActions = field(
         default_factory=lambda: VariantWithActions(
             id=3,
             name="Unsupported",
-            _available_actions={
-                action.value for action in UnsupportedStepActions
-            },
-            actions=[UnsupportedStepActions.WRITE_JUSTIFICATION.value],
+            _available_actions=set(),
+            actions=[],
         )
     )
     cant_confidently_assess_step: VariantWithActions = field(
         default_factory=lambda: VariantWithActions(
             id=4,
             name="Can't confidently assess",
-            _available_actions={
-                action.value for action in CanConfidentlyAssessStepActions
-            },
-            actions=[CanConfidentlyAssessStepActions.WRITE_JUSTIFICATION.value],
+            _available_actions=set(),
+            actions=[],
         )
     )
     no_factual_information_step: VariantWithActions = field(
         default_factory=lambda: VariantWithActions(
             id=5,
             name="No factual information",
-            _available_actions={
-                action.value for action in NoFactualInformationStepActions
-            },
-            actions=[NoFactualInformationStepActions.WRITE_JUSTIFICATION.value],
+            _available_actions=set(),
+            actions=[],
         )
     )
 
@@ -90,11 +103,11 @@ def from_dict(cls, dictionary: List[Dict[str, Any]]):
 
         for variant in dictionary:
             if variant["id"] == 0:
-                accurate_step = Variant(**variant)
+                accurate_step = VariantWithActions(**variant)
             elif variant["id"] == 1:
-                inaccurate_step = Variant(**variant)
+                inaccurate_step = VariantWithActions(**variant)
             elif variant["id"] == 2:
-                disputed_step = Variant(**variant)
+                disputed_step = VariantWithActions(**variant)
             elif variant["id"] == 3:
                 unsupported_step = VariantWithActions(**variant)
             elif variant["id"] == 4:
@@ -126,6 +139,12 @@ def from_dict(cls, dictionary: List[Dict[str, Any]]):
 
 @dataclass
 class FactCheckingDefinition:
+    name: str
+    type: ToolType = field(default=ToolType.FACT_CHECKING, init=False)
+    required: bool = False
+    schema_id: Optional[str] = None
+    feature_schema_id: Optional[str] = None
+    color: Optional[str] = None
     variants: FactCheckingVariants = field(default_factory=FactCheckingVariants)
     version: int = field(default=1)
     title: Optional[str] = None
@@ -145,10 +164,11 @@ def asdict(self) -> Dict[str, Any]:
 
     @classmethod
     def from_dict(cls, dictionary: Dict[str, Any]) -> "FactCheckingDefinition":
+        name = dictionary["name"]
         variants = FactCheckingVariants.from_dict(dictionary["variants"])
         title = dictionary.get("title", None)
         value = dictionary.get("value", None)
-        return cls(variants=variants, title=title, value=value)
+        return cls(name=name, variants=variants, title=title, value=value)
 
 
 @dataclass