2.2.0 - add ZeroShotTokenClassifier

bpm-ai-core/bpm_ai_core/token_classification/zero_shot_token_classifier.py

@@ -0,0 +1,35 @@
+from abc import ABC, abstractmethod
+from typing import Tuple
+
+from pydantic import BaseModel
+
+
+class TokenClassificationResult(BaseModel):
+    tags: list[Tuple[str, str]]
+
+
+class ZeroShotTokenClassifier(ABC):
+    """
+    Zero Shot Token Classification Model
+    """
+
+    @abstractmethod
+    async def _do_classify(
+            self,
+            text: str,
+            classes: list[str],
+            confidence_threshold: float | None = None
+    ) -> TokenClassificationResult:
+        pass
+
+    async def classify(
+            self,
+            text: str,
+            classes: list[str],
+            confidence_threshold: float | None = None
+    ) -> TokenClassificationResult:
+        return await self._do_classify(
+            text=text,
+            classes=classes,
+            confidence_threshold=confidence_threshold
+        )

bpm-ai-core/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "bpm-ai-core"
-version = "2.1.0"
+version = "2.2.0"
 description = "Core AI abstractions and helpers."
 authors = ["Bennet Krause <[email protected]>"]
 repository = "https://github.com/holunda-io/bpm-ai"