antfly-sdk 0.0.2.dev2__py3-none-any.whl

antfly/client_generated/models/inference_document_token_box.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="InferenceDocumentTokenBox")
+
+
+@_attrs_define
+class InferenceDocumentTokenBox:
+    """
+    Attributes:
+        text (str):
+        bbox (list[int]): Bounding box normalized to the same 0-1000 layout space used by training Example: [0, 0, 120,
+            24].
+    """
+
+    text: str
+    bbox: list[int]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        text = self.text
+
+        bbox = self.bbox
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "text": text,
+                "bbox": bbox,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        text = d.pop("text")
+
+        bbox = cast(list[int], d.pop("bbox"))
+
+        inference_document_token_box = cls(
+            text=text,
+            bbox=bbox,
+        )
+
+        inference_document_token_box.additional_properties = d
+        return inference_document_token_box
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/inference_document_token_classification_features.py

@@ -0,0 +1,101 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="InferenceDocumentTokenClassificationFeatures")
+
+
+@_attrs_define
+class InferenceDocumentTokenClassificationFeatures:
+    """
+    Attributes:
+        text_length (int):
+        bbox (list[int]):
+        width (float):
+        height (float):
+        relative_position (float):
+        bbox_phase_sin (float):
+    """
+
+    text_length: int
+    bbox: list[int]
+    width: float
+    height: float
+    relative_position: float
+    bbox_phase_sin: float
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        text_length = self.text_length
+
+        bbox = self.bbox
+
+        width = self.width
+
+        height = self.height
+
+        relative_position = self.relative_position
+
+        bbox_phase_sin = self.bbox_phase_sin
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "text_length": text_length,
+                "bbox": bbox,
+                "width": width,
+                "height": height,
+                "relative_position": relative_position,
+                "bbox_phase_sin": bbox_phase_sin,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        text_length = d.pop("text_length")
+
+        bbox = cast(list[int], d.pop("bbox"))
+
+        width = d.pop("width")
+
+        height = d.pop("height")
+
+        relative_position = d.pop("relative_position")
+
+        bbox_phase_sin = d.pop("bbox_phase_sin")
+
+        inference_document_token_classification_features = cls(
+            text_length=text_length,
+            bbox=bbox,
+            width=width,
+            height=height,
+            relative_position=relative_position,
+            bbox_phase_sin=bbox_phase_sin,
+        )
+
+        inference_document_token_classification_features.additional_properties = d
+        return inference_document_token_classification_features
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/inference_document_token_classification_object.py

@@ -0,0 +1,124 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.inference_document_token_classification_object_object import (
+    InferenceDocumentTokenClassificationObjectObject,
+)
+
+if TYPE_CHECKING:
+    from ..models.inference_document_token_classification_prediction import (
+        InferenceDocumentTokenClassificationPrediction,
+    )
+
+
+T = TypeVar("T", bound="InferenceDocumentTokenClassificationObject")
+
+
+@_attrs_define
+class InferenceDocumentTokenClassificationObject:
+    """
+    Attributes:
+        object_ (InferenceDocumentTokenClassificationObjectObject):
+        index (int):
+        checkpoint_path (str):
+        prefix (str):
+        num_tokens (int):
+        predictions (list[InferenceDocumentTokenClassificationPrediction]): Token classification predictions sorted by
+            score descending.
+    """
+
+    object_: InferenceDocumentTokenClassificationObjectObject
+    index: int
+    checkpoint_path: str
+    prefix: str
+    num_tokens: int
+    predictions: list[InferenceDocumentTokenClassificationPrediction]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        object_ = self.object_.value
+
+        index = self.index
+
+        checkpoint_path = self.checkpoint_path
+
+        prefix = self.prefix
+
+        num_tokens = self.num_tokens
+
+        predictions = []
+        for predictions_item_data in self.predictions:
+            predictions_item = predictions_item_data.to_dict()
+            predictions.append(predictions_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "object": object_,
+                "index": index,
+                "checkpoint_path": checkpoint_path,
+                "prefix": prefix,
+                "num_tokens": num_tokens,
+                "predictions": predictions,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_document_token_classification_prediction import (
+            InferenceDocumentTokenClassificationPrediction,
+        )
+
+        d = dict(src_dict)
+        object_ = InferenceDocumentTokenClassificationObjectObject(d.pop("object"))
+
+        index = d.pop("index")
+
+        checkpoint_path = d.pop("checkpoint_path")
+
+        prefix = d.pop("prefix")
+
+        num_tokens = d.pop("num_tokens")
+
+        predictions = []
+        _predictions = d.pop("predictions")
+        for predictions_item_data in _predictions:
+            predictions_item = InferenceDocumentTokenClassificationPrediction.from_dict(predictions_item_data)
+
+            predictions.append(predictions_item)
+
+        inference_document_token_classification_object = cls(
+            object_=object_,
+            index=index,
+            checkpoint_path=checkpoint_path,
+            prefix=prefix,
+            num_tokens=num_tokens,
+            predictions=predictions,
+        )
+
+        inference_document_token_classification_object.additional_properties = d
+        return inference_document_token_classification_object
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/inference_document_token_classification_object_object.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class InferenceDocumentTokenClassificationObjectObject(str, Enum):
+    DOCUMENT_TOKEN_CLASSIFICATION = "document.token_classification"
+
+    def __str__(self) -> str:
+        return str(self.value)

antfly/client_generated/models/inference_document_token_classification_prediction.py

@@ -0,0 +1,145 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.inference_document_token_classification_features import InferenceDocumentTokenClassificationFeatures
+    from ..models.inference_document_token_classification_result import InferenceDocumentTokenClassificationResult
+
+
+T = TypeVar("T", bound="InferenceDocumentTokenClassificationPrediction")
+
+
+@_attrs_define
+class InferenceDocumentTokenClassificationPrediction:
+    """
+    Attributes:
+        token_index (int):
+        text (str):
+        bbox (list[int]):
+        features (InferenceDocumentTokenClassificationFeatures):
+        scores (list[InferenceDocumentTokenClassificationResult]):
+        best (InferenceDocumentTokenClassificationResult | None | Unset):
+    """
+
+    token_index: int
+    text: str
+    bbox: list[int]
+    features: InferenceDocumentTokenClassificationFeatures
+    scores: list[InferenceDocumentTokenClassificationResult]
+    best: InferenceDocumentTokenClassificationResult | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.inference_document_token_classification_result import InferenceDocumentTokenClassificationResult
+
+        token_index = self.token_index
+
+        text = self.text
+
+        bbox = self.bbox
+
+        features = self.features.to_dict()
+
+        scores = []
+        for scores_item_data in self.scores:
+            scores_item = scores_item_data.to_dict()
+            scores.append(scores_item)
+
+        best: dict[str, Any] | None | Unset
+        if isinstance(self.best, Unset):
+            best = UNSET
+        elif isinstance(self.best, InferenceDocumentTokenClassificationResult):
+            best = self.best.to_dict()
+        else:
+            best = self.best
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "token_index": token_index,
+                "text": text,
+                "bbox": bbox,
+                "features": features,
+                "scores": scores,
+            }
+        )
+        if best is not UNSET:
+            field_dict["best"] = best
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_document_token_classification_features import (
+            InferenceDocumentTokenClassificationFeatures,
+        )
+        from ..models.inference_document_token_classification_result import InferenceDocumentTokenClassificationResult
+
+        d = dict(src_dict)
+        token_index = d.pop("token_index")
+
+        text = d.pop("text")
+
+        bbox = cast(list[int], d.pop("bbox"))
+
+        features = InferenceDocumentTokenClassificationFeatures.from_dict(d.pop("features"))
+
+        scores = []
+        _scores = d.pop("scores")
+        for scores_item_data in _scores:
+            scores_item = InferenceDocumentTokenClassificationResult.from_dict(scores_item_data)
+
+            scores.append(scores_item)
+
+        def _parse_best(data: object) -> InferenceDocumentTokenClassificationResult | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                best_type_1 = InferenceDocumentTokenClassificationResult.from_dict(data)
+
+                return best_type_1
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(InferenceDocumentTokenClassificationResult | None | Unset, data)
+
+        best = _parse_best(d.pop("best", UNSET))
+
+        inference_document_token_classification_prediction = cls(
+            token_index=token_index,
+            text=text,
+            bbox=bbox,
+            features=features,
+            scores=scores,
+            best=best,
+        )
+
+        inference_document_token_classification_prediction.additional_properties = d
+        return inference_document_token_classification_prediction
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/inference_document_token_classification_request.py

@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.inference_document_token_box import InferenceDocumentTokenBox
+
+
+T = TypeVar("T", bound="InferenceDocumentTokenClassificationRequest")
+
+
+@_attrs_define
+class InferenceDocumentTokenClassificationRequest:
+    """
+    Attributes:
+        model (str): Name or path of the document token classification model directory or checkpoint Example:
+            acme/layoutdoc-token-tags.
+        labels (list[str]): Labels in the same order expected by the checkpoint output head Example: ['O', 'B-KEY',
+            'I-KEY'].
+        tokens (list[InferenceDocumentTokenBox]):
+        prefix (str | Unset): Optional tensor prefix inside the safetensors checkpoint Default: 'layoutdoc_token_head'.
+            Example: layoutdoc_token_head.
+    """
+
+    model: str
+    labels: list[str]
+    tokens: list[InferenceDocumentTokenBox]
+    prefix: str | Unset = "layoutdoc_token_head"
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        model = self.model
+
+        labels = self.labels
+
+        tokens = []
+        for tokens_item_data in self.tokens:
+            tokens_item = tokens_item_data.to_dict()
+            tokens.append(tokens_item)
+
+        prefix = self.prefix
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "model": model,
+                "labels": labels,
+                "tokens": tokens,
+            }
+        )
+        if prefix is not UNSET:
+            field_dict["prefix"] = prefix
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_document_token_box import InferenceDocumentTokenBox
+
+        d = dict(src_dict)
+        model = d.pop("model")
+
+        labels = cast(list[str], d.pop("labels"))
+
+        tokens = []
+        _tokens = d.pop("tokens")
+        for tokens_item_data in _tokens:
+            tokens_item = InferenceDocumentTokenBox.from_dict(tokens_item_data)
+
+            tokens.append(tokens_item)
+
+        prefix = d.pop("prefix", UNSET)
+
+        inference_document_token_classification_request = cls(
+            model=model,
+            labels=labels,
+            tokens=tokens,
+            prefix=prefix,
+        )
+
+        inference_document_token_classification_request.additional_properties = d
+        return inference_document_token_classification_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/inference_document_token_classification_response.py

@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.inference_document_token_classification_response_object import (
+    InferenceDocumentTokenClassificationResponseObject,
+)
+
+if TYPE_CHECKING:
+    from ..models.inference_document_token_classification_object import InferenceDocumentTokenClassificationObject
+    from ..models.inference_generate_usage import InferenceGenerateUsage
+
+
+T = TypeVar("T", bound="InferenceDocumentTokenClassificationResponse")
+
+
+@_attrs_define
+class InferenceDocumentTokenClassificationResponse:
+    """
+    Attributes:
+        object_ (InferenceDocumentTokenClassificationResponseObject): Object type, always "list"
+        data (list[InferenceDocumentTokenClassificationObject]):
+        model (str):
+        usage (InferenceGenerateUsage):
+    """
+
+    object_: InferenceDocumentTokenClassificationResponseObject
+    data: list[InferenceDocumentTokenClassificationObject]
+    model: str
+    usage: InferenceGenerateUsage
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        object_ = self.object_.value
+
+        data = []
+        for data_item_data in self.data:
+            data_item = data_item_data.to_dict()
+            data.append(data_item)
+
+        model = self.model
+
+        usage = self.usage.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "object": object_,
+                "data": data,
+                "model": model,
+                "usage": usage,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_document_token_classification_object import InferenceDocumentTokenClassificationObject
+        from ..models.inference_generate_usage import InferenceGenerateUsage
+
+        d = dict(src_dict)
+        object_ = InferenceDocumentTokenClassificationResponseObject(d.pop("object"))
+
+        data = []
+        _data = d.pop("data")
+        for data_item_data in _data:
+            data_item = InferenceDocumentTokenClassificationObject.from_dict(data_item_data)
+
+            data.append(data_item)
+
+        model = d.pop("model")
+
+        usage = InferenceGenerateUsage.from_dict(d.pop("usage"))
+
+        inference_document_token_classification_response = cls(
+            object_=object_,
+            data=data,
+            model=model,
+            usage=usage,
+        )
+
+        inference_document_token_classification_response.additional_properties = d
+        return inference_document_token_classification_response
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/inference_document_token_classification_response_object.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class InferenceDocumentTokenClassificationResponseObject(str, Enum):
+    LIST = "list"
+
+    def __str__(self) -> str:
+        return str(self.value)