aeri-python 4.0.0__py3-none-any.whl

aeri/api/annotation_queues/types/__init__.py

@@ -0,0 +1,84 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .annotation_queue import AnnotationQueue
+    from .annotation_queue_assignment_request import AnnotationQueueAssignmentRequest
+    from .annotation_queue_item import AnnotationQueueItem
+    from .annotation_queue_object_type import AnnotationQueueObjectType
+    from .annotation_queue_status import AnnotationQueueStatus
+    from .create_annotation_queue_assignment_response import (
+        CreateAnnotationQueueAssignmentResponse,
+    )
+    from .create_annotation_queue_item_request import CreateAnnotationQueueItemRequest
+    from .create_annotation_queue_request import CreateAnnotationQueueRequest
+    from .delete_annotation_queue_assignment_response import (
+        DeleteAnnotationQueueAssignmentResponse,
+    )
+    from .delete_annotation_queue_item_response import DeleteAnnotationQueueItemResponse
+    from .paginated_annotation_queue_items import PaginatedAnnotationQueueItems
+    from .paginated_annotation_queues import PaginatedAnnotationQueues
+    from .update_annotation_queue_item_request import UpdateAnnotationQueueItemRequest
+_dynamic_imports: typing.Dict[str, str] = {
+    "AnnotationQueue": ".annotation_queue",
+    "AnnotationQueueAssignmentRequest": ".annotation_queue_assignment_request",
+    "AnnotationQueueItem": ".annotation_queue_item",
+    "AnnotationQueueObjectType": ".annotation_queue_object_type",
+    "AnnotationQueueStatus": ".annotation_queue_status",
+    "CreateAnnotationQueueAssignmentResponse": ".create_annotation_queue_assignment_response",
+    "CreateAnnotationQueueItemRequest": ".create_annotation_queue_item_request",
+    "CreateAnnotationQueueRequest": ".create_annotation_queue_request",
+    "DeleteAnnotationQueueAssignmentResponse": ".delete_annotation_queue_assignment_response",
+    "DeleteAnnotationQueueItemResponse": ".delete_annotation_queue_item_response",
+    "PaginatedAnnotationQueueItems": ".paginated_annotation_queue_items",
+    "PaginatedAnnotationQueues": ".paginated_annotation_queues",
+    "UpdateAnnotationQueueItemRequest": ".update_annotation_queue_item_request",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"No {attr_name} found in _dynamic_imports for module name -> {__name__}"
+        )
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import {attr_name} from {module_name}: {e}"
+        ) from e
+    except AttributeError as e:
+        raise AttributeError(
+            f"Failed to get {attr_name} from {module_name}: {e}"
+        ) from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "AnnotationQueue",
+    "AnnotationQueueAssignmentRequest",
+    "AnnotationQueueItem",
+    "AnnotationQueueObjectType",
+    "AnnotationQueueStatus",
+    "CreateAnnotationQueueAssignmentResponse",
+    "CreateAnnotationQueueItemRequest",
+    "CreateAnnotationQueueRequest",
+    "DeleteAnnotationQueueAssignmentResponse",
+    "DeleteAnnotationQueueItemResponse",
+    "PaginatedAnnotationQueueItems",
+    "PaginatedAnnotationQueues",
+    "UpdateAnnotationQueueItemRequest",
+]

aeri/api/annotation_queues/types/annotation_queue.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class AnnotationQueue(UniversalBaseModel):
+    id: str
+    name: str
+    description: typing.Optional[str] = None
+    score_config_ids: typing_extensions.Annotated[
+        typing.List[str], FieldMetadata(alias="scoreConfigIds")
+    ]
+    created_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="createdAt")
+    ]
+    updated_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="updatedAt")
+    ]
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/annotation_queue_assignment_request.py

@@ -0,0 +1,16 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class AnnotationQueueAssignmentRequest(UniversalBaseModel):
+    user_id: typing_extensions.Annotated[str, FieldMetadata(alias="userId")]
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/annotation_queue_item.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+from .annotation_queue_object_type import AnnotationQueueObjectType
+from .annotation_queue_status import AnnotationQueueStatus
+
+
+class AnnotationQueueItem(UniversalBaseModel):
+    id: str
+    queue_id: typing_extensions.Annotated[str, FieldMetadata(alias="queueId")]
+    object_id: typing_extensions.Annotated[str, FieldMetadata(alias="objectId")]
+    object_type: typing_extensions.Annotated[
+        AnnotationQueueObjectType, FieldMetadata(alias="objectType")
+    ]
+    status: AnnotationQueueStatus
+    completed_at: typing_extensions.Annotated[
+        typing.Optional[dt.datetime], FieldMetadata(alias="completedAt")
+    ] = None
+    created_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="createdAt")
+    ]
+    updated_at: typing_extensions.Annotated[
+        dt.datetime, FieldMetadata(alias="updatedAt")
+    ]
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/annotation_queue_object_type.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class AnnotationQueueObjectType(enum.StrEnum):
+    TRACE = "TRACE"
+    OBSERVATION = "OBSERVATION"
+    SESSION = "SESSION"
+
+    def visit(
+        self,
+        trace: typing.Callable[[], T_Result],
+        observation: typing.Callable[[], T_Result],
+        session: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is AnnotationQueueObjectType.TRACE:
+            return trace()
+        if self is AnnotationQueueObjectType.OBSERVATION:
+            return observation()
+        if self is AnnotationQueueObjectType.SESSION:
+            return session()

aeri/api/annotation_queues/types/annotation_queue_status.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class AnnotationQueueStatus(enum.StrEnum):
+    PENDING = "PENDING"
+    COMPLETED = "COMPLETED"
+
+    def visit(
+        self,
+        pending: typing.Callable[[], T_Result],
+        completed: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is AnnotationQueueStatus.PENDING:
+            return pending()
+        if self is AnnotationQueueStatus.COMPLETED:
+            return completed()

aeri/api/annotation_queues/types/create_annotation_queue_assignment_response.py

@@ -0,0 +1,18 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class CreateAnnotationQueueAssignmentResponse(UniversalBaseModel):
+    user_id: typing_extensions.Annotated[str, FieldMetadata(alias="userId")]
+    queue_id: typing_extensions.Annotated[str, FieldMetadata(alias="queueId")]
+    project_id: typing_extensions.Annotated[str, FieldMetadata(alias="projectId")]
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/create_annotation_queue_item_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+from .annotation_queue_object_type import AnnotationQueueObjectType
+from .annotation_queue_status import AnnotationQueueStatus
+
+
+class CreateAnnotationQueueItemRequest(UniversalBaseModel):
+    object_id: typing_extensions.Annotated[str, FieldMetadata(alias="objectId")]
+    object_type: typing_extensions.Annotated[
+        AnnotationQueueObjectType, FieldMetadata(alias="objectType")
+    ]
+    status: typing.Optional[AnnotationQueueStatus] = pydantic.Field(default=None)
+    """
+    Defaults to PENDING for new queue items
+    """
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/create_annotation_queue_request.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...core.serialization import FieldMetadata
+
+
+class CreateAnnotationQueueRequest(UniversalBaseModel):
+    name: str
+    description: typing.Optional[str] = None
+    score_config_ids: typing_extensions.Annotated[
+        typing.List[str], FieldMetadata(alias="scoreConfigIds")
+    ]
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/delete_annotation_queue_assignment_response.py

@@ -0,0 +1,14 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import UniversalBaseModel
+
+
+class DeleteAnnotationQueueAssignmentResponse(UniversalBaseModel):
+    success: bool
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/delete_annotation_queue_item_response.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import UniversalBaseModel
+
+
+class DeleteAnnotationQueueItemResponse(UniversalBaseModel):
+    success: bool
+    message: str
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/paginated_annotation_queue_items.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...utils.pagination.types.meta_response import MetaResponse
+from .annotation_queue_item import AnnotationQueueItem
+
+
+class PaginatedAnnotationQueueItems(UniversalBaseModel):
+    data: typing.List[AnnotationQueueItem]
+    meta: MetaResponse
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/paginated_annotation_queues.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import UniversalBaseModel
+from ...utils.pagination.types.meta_response import MetaResponse
+from .annotation_queue import AnnotationQueue
+
+
+class PaginatedAnnotationQueues(UniversalBaseModel):
+    data: typing.List[AnnotationQueue]
+    meta: MetaResponse
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/annotation_queues/types/update_annotation_queue_item_request.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ...core.pydantic_utilities import UniversalBaseModel
+from .annotation_queue_status import AnnotationQueueStatus
+
+
+class UpdateAnnotationQueueItemRequest(UniversalBaseModel):
+    status: typing.Optional[AnnotationQueueStatus] = None
+
+    model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+        extra="allow", frozen=True
+    )

aeri/api/blob_storage_integrations/__init__.py

@@ -0,0 +1,73 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .types import (
+        BlobStorageExportFrequency,
+        BlobStorageExportMode,
+        BlobStorageIntegrationDeletionResponse,
+        BlobStorageIntegrationFileType,
+        BlobStorageIntegrationResponse,
+        BlobStorageIntegrationStatusResponse,
+        BlobStorageIntegrationType,
+        BlobStorageIntegrationsResponse,
+        BlobStorageSyncStatus,
+        CreateBlobStorageIntegrationRequest,
+    )
+_dynamic_imports: typing.Dict[str, str] = {
+    "BlobStorageExportFrequency": ".types",
+    "BlobStorageExportMode": ".types",
+    "BlobStorageIntegrationDeletionResponse": ".types",
+    "BlobStorageIntegrationFileType": ".types",
+    "BlobStorageIntegrationResponse": ".types",
+    "BlobStorageIntegrationStatusResponse": ".types",
+    "BlobStorageIntegrationType": ".types",
+    "BlobStorageIntegrationsResponse": ".types",
+    "BlobStorageSyncStatus": ".types",
+    "CreateBlobStorageIntegrationRequest": ".types",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"No {attr_name} found in _dynamic_imports for module name -> {__name__}"
+        )
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import {attr_name} from {module_name}: {e}"
+        ) from e
+    except AttributeError as e:
+        raise AttributeError(
+            f"Failed to get {attr_name} from {module_name}: {e}"
+        ) from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "BlobStorageExportFrequency",
+    "BlobStorageExportMode",
+    "BlobStorageIntegrationDeletionResponse",
+    "BlobStorageIntegrationFileType",
+    "BlobStorageIntegrationResponse",
+    "BlobStorageIntegrationStatusResponse",
+    "BlobStorageIntegrationType",
+    "BlobStorageIntegrationsResponse",
+    "BlobStorageSyncStatus",
+    "CreateBlobStorageIntegrationRequest",
+]