From d6865e2aecc06ac5f664178c2c6a0896a1adcfeb Mon Sep 17 00:00:00 2001
From: Andras Schmelczer <andras@schmelczer.dev>
Date: Wed, 13 Jul 2022 08:47:23 +0200
Subject: [PATCH] Add sequence labelling output

---
 great_ai/__init__.py                             | 12 ++++++------
 great_ai/views/__init__.py                       |  5 -----
 great_ai/views/outputs/__init__.py               |  3 ---
 .../views/outputs/sequence_labeling_output.py    | 16 +++++++++++++++-
 4 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/great_ai/__init__.py b/great_ai/__init__.py
index c2ecdbb..a80ed2d 100644
--- a/great_ai/__init__.py
+++ b/great_ai/__init__.py
@@ -1,5 +1,5 @@
 """GreatAI."""
-__version__ = "0.1.4"
+__version__ = "0.1.5"
 
 
 from .context import configure
@@ -22,10 +22,10 @@ from .remote.call_remote_great_ai_async import call_remote_great_ai_async
 from .tracing.add_ground_truth import add_ground_truth
 from .tracing.delete_ground_truth import delete_ground_truth
 from .tracing.query_ground_truth import query_ground_truth
-from .views import (
-    ClassificationOutput,
+from .views import RouteConfig, Trace
+from .views.outputs.classification_output import ClassificationOutput
+from .views.outputs.multi_label_classification_output import (
     MultiLabelClassificationOutput,
-    RegressionOutput,
-    RouteConfig,
-    Trace,
 )
+from .views.outputs.regression_output import RegressionOutput
+from .views.outputs.sequence_labeling_output import SequenceLabelingOutput
diff --git a/great_ai/views/__init__.py b/great_ai/views/__init__.py
index 3c66933..cf82c4f 100644
--- a/great_ai/views/__init__.py
+++ b/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
diff --git a/great_ai/views/outputs/__init__.py b/great_ai/views/outputs/__init__.py
index eb9f1e0..e69de29 100644
--- a/great_ai/views/outputs/__init__.py
+++ b/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
diff --git a/great_ai/views/outputs/sequence_labeling_output.py b/great_ai/views/outputs/sequence_labeling_output.py
index 044a482..131d8f2 100644
--- a/great_ai/views/outputs/sequence_labeling_output.py
+++ b/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]