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

antfly/client_generated/models/inference_models_response_classifiers.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseClassifiers")
+
+
+@_attrs_define
+class InferenceModelsResponseClassifiers:
+    """Available zero-shot classification models"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_classifiers = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_classifiers.additional_properties = additional_properties
+        return inference_models_response_classifiers
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_data_item.py

@@ -0,0 +1,47 @@
+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="InferenceModelsResponseDataItem")
+
+
+@_attrs_define
+class InferenceModelsResponseDataItem:
+    """ """
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        inference_models_response_data_item = cls()
+
+        inference_models_response_data_item.additional_properties = d
+        return inference_models_response_data_item
+
+    @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_models_response_embedders.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseEmbedders")
+
+
+@_attrs_define
+class InferenceModelsResponseEmbedders:
+    """Available embedding models from models_dir/embedders/"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_embedders = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_embedders.additional_properties = additional_properties
+        return inference_models_response_embedders
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_extractors.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseExtractors")
+
+
+@_attrs_define
+class InferenceModelsResponseExtractors:
+    """Available extractor models (models with 'extraction' capability)"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_extractors = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_extractors.additional_properties = additional_properties
+        return inference_models_response_extractors
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_generators.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseGenerators")
+
+
+@_attrs_define
+class InferenceModelsResponseGenerators:
+    """Available generator/LLM models from models_dir/generators/"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_generators = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_generators.additional_properties = additional_properties
+        return inference_models_response_generators
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_object.py

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

antfly/client_generated/models/inference_models_response_readers.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseReaders")
+
+
+@_attrs_define
+class InferenceModelsResponseReaders:
+    """Available reader/OCR models from models_dir/readers/"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_readers = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_readers.additional_properties = additional_properties
+        return inference_models_response_readers
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_recognizers.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseRecognizers")
+
+
+@_attrs_define
+class InferenceModelsResponseRecognizers:
+    """Available recognizer models from models_dir/recognizers/"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_recognizers = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_recognizers.additional_properties = additional_properties
+        return inference_models_response_recognizers
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_rerankers.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseRerankers")
+
+
+@_attrs_define
+class InferenceModelsResponseRerankers:
+    """Available reranking models"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_rerankers = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_rerankers.additional_properties = additional_properties
+        return inference_models_response_rerankers
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_rewriters.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseRewriters")
+
+
+@_attrs_define
+class InferenceModelsResponseRewriters:
+    """Available Seq2Seq rewriter models from models_dir/rewriters/"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_rewriters = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_rewriters.additional_properties = additional_properties
+        return inference_models_response_rewriters
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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_models_response_transcribers.py

@@ -0,0 +1,60 @@
+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
+
+if TYPE_CHECKING:
+    from ..models.inference_model_info import InferenceModelInfo
+
+
+T = TypeVar("T", bound="InferenceModelsResponseTranscribers")
+
+
+@_attrs_define
+class InferenceModelsResponseTranscribers:
+    """Available transcriber/speech-to-text models from models_dir/transcribers/"""
+
+    additional_properties: dict[str, InferenceModelInfo] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = prop.to_dict()
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.inference_model_info import InferenceModelInfo
+
+        d = dict(src_dict)
+        inference_models_response_transcribers = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+            additional_property = InferenceModelInfo.from_dict(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        inference_models_response_transcribers.additional_properties = additional_properties
+        return inference_models_response_transcribers
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> InferenceModelInfo:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: InferenceModelInfo) -> 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