Skip to content

Commit 6f29459

Browse files
author
Val Brodsky
committed
Add step by step reasoning ontology tool
1 parent 84d5fae commit 6f29459

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

libs/labelbox/src/labelbox/schema/ontology.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -492,6 +492,10 @@ def add_classification(self, classification: Classification) -> None:
492492

493493
def tool_cls_from_type(tool_type: str):
494494
if tool_type.lower() == ToolType.STEP_REASONING.value:
495+
from labelbox.schema.tool_building.step_reasoning_tool import (
496+
StepReasoningTool,
497+
)
498+
495499
return StepReasoningTool
496500
return Tool
497501

@@ -535,6 +539,12 @@ def __init__(self, *args, **kwargs) -> None:
535539
Union[List[Classification], List[PromptResponseClassification]]
536540
] = None
537541

542+
def _tool_deserializer_cls(self, tool: Dict[str, Any]) -> Tool:
543+
import pdb
544+
545+
pdb.set_trace()
546+
return Tool
547+
538548
def tools(self) -> List[Tool]:
539549
"""Get list of tools (AKA objects) in an Ontology."""
540550
if self._tools is None:

0 commit comments

Comments
 (0)