Add sequence labelling output

great_ai/views/__init__.py

@@ -6,11 +6,6 @@ from .function_metadata import FunctionMetadata
 from .health_check_response import HealthCheckResponse
 from .model import Model
 from .operators import operators
-from .outputs import (
-    ClassificationOutput,
-    MultiLabelClassificationOutput,
-    RegressionOutput,
-)
 from .query import Query
 from .route_config import RouteConfig
 from .sort_by import SortBy

great_ai/views/outputs/__init__.py

@@ -1,3 +0,0 @@
-from .classification_output import ClassificationOutput
-from .multi_label_classification_output import MultiLabelClassificationOutput
-from .regression_output import RegressionOutput

great_ai/views/outputs/sequence_labeling_output.py

@@ -1 +1,15 @@
-# todo
+from typing import Any, List, Literal, Optional
+
+from ..hashable_base_model import HashableBaseModel
+
+
+class LabeledToken(HashableBaseModel):
+    token: str
+    tag: Literal["B", "I", "O", "E", "S"]
+    confidence: float
+    explanation: Optional[Any]
+
+
+class SequenceLabelingOutput(HashableBaseModel):
+    labeled_tokens: List[LabeledToken]
+    explanation: Optional[Any]