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

antfly/client_generated/models/inference_document_token_classification_result.py

@@ -0,0 +1,69 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="InferenceDocumentTokenClassificationResult")
+
+
+@_attrs_define
+class InferenceDocumentTokenClassificationResult:
+    """
+    Attributes:
+        label (str):
+        score (float):
+    """
+
+    label: str
+    score: float
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        label = self.label
+
+        score = self.score
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "label": label,
+                "score": score,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        label = d.pop("label")
+
+        score = d.pop("score")
+
+        inference_document_token_classification_result = cls(
+            label=label,
+            score=score,
+        )
+
+        inference_document_token_classification_result.additional_properties = d
+        return inference_document_token_classification_result
+
+    @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_embed_request.py

@@ -0,0 +1,226 @@
+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 ..models.inference_embed_request_encoding_format import InferenceEmbedRequestEncodingFormat
+from ..models.inference_embed_request_input_type import InferenceEmbedRequestInputType
+from ..models.inference_embed_request_task_type import InferenceEmbedRequestTaskType
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.image_url_content_part import ImageURLContentPart
+    from ..models.media_content_part import MediaContentPart
+    from ..models.text_content_part import TextContentPart
+
+
+T = TypeVar("T", bound="InferenceEmbedRequest")
+
+
+@_attrs_define
+class InferenceEmbedRequest:
+    """OpenAI-compatible embedding request with inference multimodal content-part extension
+
+    Attributes:
+        model (str): Model name to use for embedding generation
+        input_ (list[ImageURLContentPart | MediaContentPart | TextContentPart] | list[str] | str): Input content to
+            embed.
+            Supports:
+            - a single string
+            - an array of strings
+            - an array of OpenAI-style content parts for multimodal embedding
+        encoding_format (InferenceEmbedRequestEncodingFormat | Unset): Encoding format for the embeddings (only "float"
+            supported) Default: InferenceEmbedRequestEncodingFormat.FLOAT.
+        dimensions (int | Unset): Optional truncation size for dense embeddings. Must be a positive integer no larger
+            than the model embedding size. Not supported for sparse models.
+        task_type (InferenceEmbedRequestTaskType | Unset): Optional embedding task type using Google embedding task-type
+            names. For Jina v5 text embeddings, query-side tasks use the query prefix and RETRIEVAL_DOCUMENT uses the
+            document prefix.
+        input_type (InferenceEmbedRequestInputType | Unset): Deprecated compatibility alias for task_type.
+            search_query/query map to RETRIEVAL_QUERY; search_document/document map to RETRIEVAL_DOCUMENT; classification
+            and clustering map to their Google task_type equivalents.
+    """
+
+    model: str
+    input_: list[ImageURLContentPart | MediaContentPart | TextContentPart] | list[str] | str
+    encoding_format: InferenceEmbedRequestEncodingFormat | Unset = InferenceEmbedRequestEncodingFormat.FLOAT
+    dimensions: int | Unset = UNSET
+    task_type: InferenceEmbedRequestTaskType | Unset = UNSET
+    input_type: InferenceEmbedRequestInputType | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.image_url_content_part import ImageURLContentPart
+        from ..models.text_content_part import TextContentPart
+
+        model = self.model
+
+        input_: list[dict[str, Any]] | list[str] | str
+        if isinstance(self.input_, list):
+            input_ = self.input_
+
+        elif isinstance(self.input_, list):
+            input_ = []
+            for input_type_2_item_data in self.input_:
+                input_type_2_item: dict[str, Any]
+                if isinstance(input_type_2_item_data, TextContentPart):
+                    input_type_2_item = input_type_2_item_data.to_dict()
+                elif isinstance(input_type_2_item_data, ImageURLContentPart):
+                    input_type_2_item = input_type_2_item_data.to_dict()
+                else:
+                    input_type_2_item = input_type_2_item_data.to_dict()
+
+                input_.append(input_type_2_item)
+
+        else:
+            input_ = self.input_
+
+        encoding_format: str | Unset = UNSET
+        if not isinstance(self.encoding_format, Unset):
+            encoding_format = self.encoding_format.value
+
+        dimensions = self.dimensions
+
+        task_type: str | Unset = UNSET
+        if not isinstance(self.task_type, Unset):
+            task_type = self.task_type.value
+
+        input_type: str | Unset = UNSET
+        if not isinstance(self.input_type, Unset):
+            input_type = self.input_type.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "model": model,
+                "input": input_,
+            }
+        )
+        if encoding_format is not UNSET:
+            field_dict["encoding_format"] = encoding_format
+        if dimensions is not UNSET:
+            field_dict["dimensions"] = dimensions
+        if task_type is not UNSET:
+            field_dict["task_type"] = task_type
+        if input_type is not UNSET:
+            field_dict["input_type"] = input_type
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.image_url_content_part import ImageURLContentPart
+        from ..models.media_content_part import MediaContentPart
+        from ..models.text_content_part import TextContentPart
+
+        d = dict(src_dict)
+        model = d.pop("model")
+
+        def _parse_input_(
+            data: object,
+        ) -> list[ImageURLContentPart | MediaContentPart | TextContentPart] | list[str] | str:
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                input_type_1 = cast(list[str], data)
+
+                return input_type_1
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                input_type_2 = []
+                _input_type_2 = data
+                for input_type_2_item_data in _input_type_2:
+
+                    def _parse_input_type_2_item(
+                        data: object,
+                    ) -> ImageURLContentPart | MediaContentPart | TextContentPart:
+                        try:
+                            if not isinstance(data, dict):
+                                raise TypeError()
+                            componentsschemas_content_part_type_0 = TextContentPart.from_dict(data)
+
+                            return componentsschemas_content_part_type_0
+                        except (TypeError, ValueError, AttributeError, KeyError):
+                            pass
+                        try:
+                            if not isinstance(data, dict):
+                                raise TypeError()
+                            componentsschemas_content_part_type_1 = ImageURLContentPart.from_dict(data)
+
+                            return componentsschemas_content_part_type_1
+                        except (TypeError, ValueError, AttributeError, KeyError):
+                            pass
+                        if not isinstance(data, dict):
+                            raise TypeError()
+                        componentsschemas_content_part_type_2 = MediaContentPart.from_dict(data)
+
+                        return componentsschemas_content_part_type_2
+
+                    input_type_2_item = _parse_input_type_2_item(input_type_2_item_data)
+
+                    input_type_2.append(input_type_2_item)
+
+                return input_type_2
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(list[ImageURLContentPart | MediaContentPart | TextContentPart] | list[str] | str, data)
+
+        input_ = _parse_input_(d.pop("input"))
+
+        _encoding_format = d.pop("encoding_format", UNSET)
+        encoding_format: InferenceEmbedRequestEncodingFormat | Unset
+        if isinstance(_encoding_format, Unset):
+            encoding_format = UNSET
+        else:
+            encoding_format = InferenceEmbedRequestEncodingFormat(_encoding_format)
+
+        dimensions = d.pop("dimensions", UNSET)
+
+        _task_type = d.pop("task_type", UNSET)
+        task_type: InferenceEmbedRequestTaskType | Unset
+        if isinstance(_task_type, Unset):
+            task_type = UNSET
+        else:
+            task_type = InferenceEmbedRequestTaskType(_task_type)
+
+        _input_type = d.pop("input_type", UNSET)
+        input_type: InferenceEmbedRequestInputType | Unset
+        if isinstance(_input_type, Unset):
+            input_type = UNSET
+        else:
+            input_type = InferenceEmbedRequestInputType(_input_type)
+
+        inference_embed_request = cls(
+            model=model,
+            input_=input_,
+            encoding_format=encoding_format,
+            dimensions=dimensions,
+            task_type=task_type,
+            input_type=input_type,
+        )
+
+        inference_embed_request.additional_properties = d
+        return inference_embed_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_embed_request_encoding_format.py

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

antfly/client_generated/models/inference_embed_request_input_type.py

@@ -0,0 +1,13 @@
+from enum import Enum
+
+
+class InferenceEmbedRequestInputType(str, Enum):
+    CLASSIFICATION = "classification"
+    CLUSTERING = "clustering"
+    DOCUMENT = "document"
+    QUERY = "query"
+    SEARCH_DOCUMENT = "search_document"
+    SEARCH_QUERY = "search_query"
+
+    def __str__(self) -> str:
+        return str(self.value)

antfly/client_generated/models/inference_embed_request_task_type.py

@@ -0,0 +1,15 @@
+from enum import Enum
+
+
+class InferenceEmbedRequestTaskType(str, Enum):
+    CLASSIFICATION = "CLASSIFICATION"
+    CLUSTERING = "CLUSTERING"
+    CODE_RETRIEVAL_QUERY = "CODE_RETRIEVAL_QUERY"
+    FACT_VERIFICATION = "FACT_VERIFICATION"
+    QUESTION_ANSWERING = "QUESTION_ANSWERING"
+    RETRIEVAL_DOCUMENT = "RETRIEVAL_DOCUMENT"
+    RETRIEVAL_QUERY = "RETRIEVAL_QUERY"
+    SEMANTIC_SIMILARITY = "SEMANTIC_SIMILARITY"
+
+    def __str__(self) -> str:
+        return str(self.value)

antfly/client_generated/models/inference_embed_response.py

@@ -0,0 +1,104 @@
+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_embed_response_object import InferenceEmbedResponseObject
+
+if TYPE_CHECKING:
+    from ..models.inference_embedding_object import InferenceEmbeddingObject
+    from ..models.inference_embedding_usage import InferenceEmbeddingUsage
+
+
+T = TypeVar("T", bound="InferenceEmbedResponse")
+
+
+@_attrs_define
+class InferenceEmbedResponse:
+    """OpenAI-compatible embedding response with a polymorphic `embedding` field for dense or sparse vectors
+
+    Attributes:
+        object_ (InferenceEmbedResponseObject): Object type, always "list"
+        data (list[InferenceEmbeddingObject]): List of embedding objects
+        model (str): Model used for embedding generation
+        usage (InferenceEmbeddingUsage): Token usage information
+    """
+
+    object_: InferenceEmbedResponseObject
+    data: list[InferenceEmbeddingObject]
+    model: str
+    usage: InferenceEmbeddingUsage
+    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_embedding_object import InferenceEmbeddingObject
+        from ..models.inference_embedding_usage import InferenceEmbeddingUsage
+
+        d = dict(src_dict)
+        object_ = InferenceEmbedResponseObject(d.pop("object"))
+
+        data = []
+        _data = d.pop("data")
+        for data_item_data in _data:
+            data_item = InferenceEmbeddingObject.from_dict(data_item_data)
+
+            data.append(data_item)
+
+        model = d.pop("model")
+
+        usage = InferenceEmbeddingUsage.from_dict(d.pop("usage"))
+
+        inference_embed_response = cls(
+            object_=object_,
+            data=data,
+            model=model,
+            usage=usage,
+        )
+
+        inference_embed_response.additional_properties = d
+        return inference_embed_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_embed_response_object.py

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

antfly/client_generated/models/inference_embedding_object.py

@@ -0,0 +1,113 @@
+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 ..models.inference_embedding_object_object import InferenceEmbeddingObjectObject
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.inference_sparse_vector import InferenceSparseVector
+
+
+T = TypeVar("T", bound="InferenceEmbeddingObject")
+
+
+@_attrs_define
+class InferenceEmbeddingObject:
+    """A single embedding result
+
+    Attributes:
+        object_ (InferenceEmbeddingObjectObject): Object type, always "embedding"
+        index (int): Index of the input this embedding corresponds to
+        embedding (InferenceSparseVector | list[float] | Unset): Dense float vector for dense models, or a sparse vector
+            object for sparse-capable models
+    """
+
+    object_: InferenceEmbeddingObjectObject
+    index: int
+    embedding: InferenceSparseVector | list[float] | Unset = UNSET
+    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
+
+        embedding: dict[str, Any] | list[float] | Unset
+        if isinstance(self.embedding, Unset):
+            embedding = UNSET
+        elif isinstance(self.embedding, list):
+            embedding = self.embedding
+
+        else:
+            embedding = self.embedding.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "object": object_,
+                "index": index,
+            }
+        )
+        if embedding is not UNSET:
+            field_dict["embedding"] = embedding
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_sparse_vector import InferenceSparseVector
+
+        d = dict(src_dict)
+        object_ = InferenceEmbeddingObjectObject(d.pop("object"))
+
+        index = d.pop("index")
+
+        def _parse_embedding(data: object) -> InferenceSparseVector | list[float] | Unset:
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                embedding_type_0 = cast(list[float], data)
+
+                return embedding_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            if not isinstance(data, dict):
+                raise TypeError()
+            embedding_type_1 = InferenceSparseVector.from_dict(data)
+
+            return embedding_type_1
+
+        embedding = _parse_embedding(d.pop("embedding", UNSET))
+
+        inference_embedding_object = cls(
+            object_=object_,
+            index=index,
+            embedding=embedding,
+        )
+
+        inference_embedding_object.additional_properties = d
+        return inference_embedding_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_embedding_object_object.py

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

antfly/client_generated/models/inference_embedding_usage.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="InferenceEmbeddingUsage")
+
+
+@_attrs_define
+class InferenceEmbeddingUsage:
+    """Token usage information
+
+    Attributes:
+        prompt_tokens (int): Number of tokens in the input
+        total_tokens (int): Total tokens used
+    """
+
+    prompt_tokens: int
+    total_tokens: int
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        prompt_tokens = self.prompt_tokens
+
+        total_tokens = self.total_tokens
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "prompt_tokens": prompt_tokens,
+                "total_tokens": total_tokens,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        prompt_tokens = d.pop("prompt_tokens")
+
+        total_tokens = d.pop("total_tokens")
+
+        inference_embedding_usage = cls(
+            prompt_tokens=prompt_tokens,
+            total_tokens=total_tokens,
+        )
+
+        inference_embedding_usage.additional_properties = d
+        return inference_embedding_usage
+
+    @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