Skip to content

Commit 0b9e931

Browse files
author
Val Brodsky
committed
Refactor StepReasoning to also reuse Variants
1 parent e2e2d91 commit 0b9e931

File tree

2 files changed

+19
-3
lines changed

2 files changed

+19
-3
lines changed

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

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,18 @@ def build_fact_checking_definition():
6161
return _Definition(variants=variants)
6262

6363

64+
class UnsupportedStepActions(Enum):
65+
WRITE_JUSTIFICATION = "writeJustification"
66+
67+
68+
class CanConfidentlyAssessStepActions(Enum):
69+
WRITE_JUSTIFICATION = "writeJustification"
70+
71+
72+
class NoFactualInformationStepActions(Enum):
73+
WRITE_JUSTIFICATION = "writeJustification"
74+
75+
6476
@dataclass
6577
class FactCheckingTool(_BaseStepReasoningTool):
6678
"""

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

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,8 @@ class VariantWithActions:
2222
actions: List[str] = field(default_factory=list)
2323
_available_actions: Set[str] = field(default_factory=set)
2424

25-
def set_actions(self, actions: Set[str]) -> None:
25+
def set_actions(self, actions: List[str]) -> None:
26+
self.actions = []
2627
for action in actions:
2728
if action in self._available_actions:
2829
self.actions.append(action)
@@ -31,8 +32,11 @@ def reset_actions(self) -> None:
3132
self.actions = []
3233

3334
def asdict(self) -> Dict[str, Any]:
34-
return {
35+
data = {
3536
"id": self.id,
3637
"name": self.name,
37-
"actions": list(set(self.actions)),
3838
}
39+
if len(self.actions) > 0:
40+
data["actions"] = self.actions
41+
42+
return data

0 commit comments

Comments
 (0)