Add FactCheckingTool

Author: Val Brodsky

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

@@ -1,4 +1,5 @@
 import labelbox.schema.tool_building.tool_type
+import labelbox.schema.tool_building.variant
 import labelbox.schema.tool_building.step_reasoning_tool
 import labelbox.schema.tool_building.fact_checking_tool
 import labelbox.schema.tool_building.tool_type_mapping

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

@@ -0,0 +1,38 @@
+from dataclasses import dataclass, field
+from typing import Any, Dict, List, Set
+
+
+@dataclass
+class Variant:
+    """
+    A variant is a single option in step-by-step reasoning or fact-checking tool.
+    """
+
+    id: int
+    name: str
+
+    def asdict(self) -> Dict[str, Any]:
+        return {"id": self.id, "name": self.name}
+
+
+@dataclass
+class VariantWithActions:
+    id: int
+    name: str
+    actions: List[str] = field(default_factory=list)
+    _available_actions: Set[str] = field(default_factory=set)
+
+    def set_actions(self, actions: Set[str]) -> None:
+        for action in actions:
+            if action in self._available_actions:
+                self.actions.append(action)
+
+    def reset_actions(self) -> None:
+        self.actions = []
+
+    def asdict(self) -> Dict[str, Any]:
+        return {
+            "id": self.id,
+            "name": self.name,
+            "actions": list(set(self.actions)),
+        }