cribl-control-plane 0.0.38__py3-none-any.whl → 0.4.0a6__py3-none-any.whl

cribl_control_plane/models/authtoken.py

@@ -2,12 +2,16 @@
 
 from __future__ import annotations
 from cribl_control_plane.types import BaseModel
-from typing_extensions import TypedDict
+import pydantic
+from typing_extensions import Annotated, TypedDict
 
 
 class AuthTokenTypedDict(TypedDict):
+    force_password_change: bool
     token: str
 
 
 class AuthToken(BaseModel):
+    force_password_change: Annotated[bool, pydantic.Field(alias="forcePasswordChange")]
+
     token: str

cribl_control_plane/models/{routecloneconf.py → branchinfo.py}

@@ -5,9 +5,9 @@ from cribl_control_plane.types import BaseModel
 from typing_extensions import TypedDict
 
 
-class RouteCloneConfTypedDict(TypedDict):
-    pass
+class BranchInfoTypedDict(TypedDict):
+    id: str
 
 
-class RouteCloneConf(BaseModel):
-    pass
+class BranchInfo(BaseModel):
+    id: str

cribl_control_plane/models/cacheconnection.py

@@ -3,8 +3,12 @@
 from __future__ import annotations
 from .cacheconnectionbackfillstatus import CacheConnectionBackfillStatus
 from .lakehouseconnectiontype import LakehouseConnectionType
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 import pydantic
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -31,14 +35,38 @@ class CacheConnection(BaseModel):
     ] = None
 
     backfill_status: Annotated[
-        Optional[CacheConnectionBackfillStatus], pydantic.Field(alias="backfillStatus")
+        Annotated[
+            Optional[CacheConnectionBackfillStatus],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="backfillStatus"),
     ] = None
 
     lakehouse_connection_type: Annotated[
-        Optional[LakehouseConnectionType],
+        Annotated[
+            Optional[LakehouseConnectionType], PlainValidator(validate_open_enum(False))
+        ],
         pydantic.Field(alias="lakehouseConnectionType"),
     ] = None
 
     migration_query_id: Annotated[
         Optional[str], pydantic.Field(alias="migrationQueryId")
     ] = None
+
+    @field_serializer("backfill_status")
+    def serialize_backfill_status(self, value):
+        if isinstance(value, str):
+            try:
+                return models.CacheConnectionBackfillStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("lakehouse_connection_type")
+    def serialize_lakehouse_connection_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.LakehouseConnectionType(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/cacheconnectionbackfillstatus.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class CacheConnectionBackfillStatus(str, Enum):
+class CacheConnectionBackfillStatus(str, Enum, metaclass=utils.OpenEnumMeta):
     SCHEDULED = "scheduled"
     PENDING = "pending"
     STARTED = "started"

cribl_control_plane/models/cloudprovider.py

@@ -1,9 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class CloudProvider(str, Enum):
+class CloudProvider(str, Enum, metaclass=utils.OpenEnumMeta):
     AWS = "aws"
     AZURE = "azure"

cribl_control_plane/models/configgroup.py

@@ -4,20 +4,47 @@ from __future__ import annotations
 from .commit import Commit, CommitTypedDict
 from .configgroupcloud import ConfigGroupCloud, ConfigGroupCloudTypedDict
 from .configgrouplookups import ConfigGroupLookups, ConfigGroupLookupsTypedDict
+from cribl_control_plane import models, utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class GitTypedDict(TypedDict):
+class ConfigGroupEstimatedIngestRate(int, Enum, metaclass=utils.OpenEnumMeta):
+    r"""Maximum expected volume of data ingested by the @{group}. (This setting is available only on @{group}s consisting of Cribl-managed Cribl.Cloud @{node}s.)"""
+
+    # 12 MB/sec
+    RATE12_MB_PER_SEC = 1024
+    # 24 MB/sec
+    RATE24_MB_PER_SEC = 2048
+    # 36 MB/sec
+    RATE36_MB_PER_SEC = 3072
+    # 48 MB/sec
+    RATE48_MB_PER_SEC = 4096
+    # 60 MB/sec
+    RATE60_MB_PER_SEC = 5120
+    # 84 MB/sec
+    RATE84_MB_PER_SEC = 7168
+    # 120 MB/sec
+    RATE120_MB_PER_SEC = 10240
+    # 156 MB/sec
+    RATE156_MB_PER_SEC = 13312
+    # 180 MB/sec
+    RATE180_MB_PER_SEC = 15360
+
+
+class ConfigGroupGitTypedDict(TypedDict):
     commit: NotRequired[str]
     local_changes: NotRequired[float]
     log: NotRequired[List[CommitTypedDict]]
 
 
-class Git(BaseModel):
+class ConfigGroupGit(BaseModel):
     commit: Optional[str] = None
 
     local_changes: Annotated[Optional[float], pydantic.Field(alias="localChanges")] = (
@@ -27,18 +54,19 @@ class Git(BaseModel):
     log: Optional[List[Commit]] = None
 
 
-class ConfigGroupType(str, Enum):
+class ConfigGroupType(str, Enum, metaclass=utils.OpenEnumMeta):
     LAKE_ACCESS = "lake_access"
 
 
 class ConfigGroupTypedDict(TypedDict):
-    config_version: str
     id: str
     cloud: NotRequired[ConfigGroupCloudTypedDict]
+    config_version: NotRequired[str]
     deploying_worker_count: NotRequired[float]
     description: NotRequired[str]
-    estimated_ingest_rate: NotRequired[float]
-    git: NotRequired[GitTypedDict]
+    estimated_ingest_rate: NotRequired[ConfigGroupEstimatedIngestRate]
+    r"""Maximum expected volume of data ingested by the @{group}. (This setting is available only on @{group}s consisting of Cribl-managed Cribl.Cloud @{node}s.)"""
+    git: NotRequired[ConfigGroupGitTypedDict]
     incompatible_worker_count: NotRequired[float]
     inherits: NotRequired[str]
     is_fleet: NotRequired[bool]
@@ -57,12 +85,14 @@ class ConfigGroupTypedDict(TypedDict):
 
 
 class ConfigGroup(BaseModel):
-    config_version: Annotated[str, pydantic.Field(alias="configVersion")]
-
     id: str
 
     cloud: Optional[ConfigGroupCloud] = None
 
+    config_version: Annotated[Optional[str], pydantic.Field(alias="configVersion")] = (
+        None
+    )
+
     deploying_worker_count: Annotated[
         Optional[float], pydantic.Field(alias="deployingWorkerCount")
     ] = None
@@ -70,10 +100,15 @@ class ConfigGroup(BaseModel):
     description: Optional[str] = None
 
     estimated_ingest_rate: Annotated[
-        Optional[float], pydantic.Field(alias="estimatedIngestRate")
+        Annotated[
+            Optional[ConfigGroupEstimatedIngestRate],
+            PlainValidator(validate_open_enum(True)),
+        ],
+        pydantic.Field(alias="estimatedIngestRate"),
     ] = None
+    r"""Maximum expected volume of data ingested by the @{group}. (This setting is available only on @{group}s consisting of Cribl-managed Cribl.Cloud @{node}s.)"""
 
-    git: Optional[Git] = None
+    git: Optional[ConfigGroupGit] = None
 
     incompatible_worker_count: Annotated[
         Optional[float], pydantic.Field(alias="incompatibleWorkerCount")
@@ -103,7 +138,9 @@ class ConfigGroup(BaseModel):
 
     tags: Optional[str] = None
 
-    type: Optional[ConfigGroupType] = None
+    type: Annotated[
+        Optional[ConfigGroupType], PlainValidator(validate_open_enum(False))
+    ] = None
 
     upgrade_version: Annotated[
         Optional[str], pydantic.Field(alias="upgradeVersion")
@@ -114,3 +151,21 @@ class ConfigGroup(BaseModel):
     worker_remote_access: Annotated[
         Optional[bool], pydantic.Field(alias="workerRemoteAccess")
     ] = None
+
+    @field_serializer("estimated_ingest_rate")
+    def serialize_estimated_ingest_rate(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ConfigGroupEstimatedIngestRate(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("type")
+    def serialize_type(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ConfigGroupType(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/configgroupcloud.py

@@ -2,9 +2,12 @@
 
 from __future__ import annotations
 from .cloudprovider import CloudProvider
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel, Nullable, UNSET_SENTINEL
-from pydantic import model_serializer
-from typing_extensions import TypedDict
+from cribl_control_plane.utils import validate_open_enum
+from pydantic import field_serializer, model_serializer
+from pydantic.functional_validators import PlainValidator
+from typing_extensions import Annotated, TypedDict
 
 
 class ConfigGroupCloudTypedDict(TypedDict):
@@ -13,10 +16,21 @@ class ConfigGroupCloudTypedDict(TypedDict):
 
 
 class ConfigGroupCloud(BaseModel):
-    provider: Nullable[CloudProvider]
+    provider: Annotated[
+        Nullable[CloudProvider], PlainValidator(validate_open_enum(False))
+    ]
 
     region: str
 
+    @field_serializer("provider")
+    def serialize_provider(self, value):
+        if isinstance(value, str):
+            try:
+                return models.CloudProvider(value)
+            except ValueError:
+                return value
+        return value
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = []

cribl_control_plane/models/createconfiggroupbyproductop.py

@@ -2,32 +2,50 @@
 
 from __future__ import annotations
 from .configgroup import ConfigGroup, ConfigGroupTypedDict
+from .groupcreaterequest import GroupCreateRequest, GroupCreateRequestTypedDict
 from .productscore import ProductsCore
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
-from cribl_control_plane.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from cribl_control_plane.utils import (
+    FieldMetadata,
+    PathParamMetadata,
+    RequestMetadata,
+    validate_open_enum,
+)
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class CreateConfigGroupByProductRequestTypedDict(TypedDict):
     product: ProductsCore
-    r"""Name of the Cribl product to add the Worker Group or Edge Fleet to."""
-    config_group: ConfigGroupTypedDict
-    r"""ConfigGroup object"""
+    r"""required Name of the Cribl product to add the Worker Group or Edge Fleet to."""
+    group_create_request: GroupCreateRequestTypedDict
+    r"""GroupCreateRequest object"""
 
 
 class CreateConfigGroupByProductRequest(BaseModel):
     product: Annotated[
-        ProductsCore,
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
         FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
     ]
-    r"""Name of the Cribl product to add the Worker Group or Edge Fleet to."""
+    r"""required Name of the Cribl product to add the Worker Group or Edge Fleet to."""
 
-    config_group: Annotated[
-        ConfigGroup,
+    group_create_request: Annotated[
+        GroupCreateRequest,
         FieldMetadata(request=RequestMetadata(media_type="application/json")),
     ]
-    r"""ConfigGroup object"""
+    r"""GroupCreateRequest object"""
+
+    @field_serializer("product")
+    def serialize_product(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ProductsCore(value)
+            except ValueError:
+                return value
+        return value
 
 
 class CreateConfigGroupByProductResponseTypedDict(TypedDict):

cribl_control_plane/models/createinputhectokenbyidop.py

@@ -2,9 +2,10 @@
 
 from __future__ import annotations
 from .addhectokenrequest import AddHecTokenRequest, AddHecTokenRequestTypedDict
+from .inputsplunkhec import InputSplunkHec, InputSplunkHecTypedDict
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import FieldMetadata, PathParamMetadata, RequestMetadata
-from typing import Any, Dict, List, Optional
+from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -29,17 +30,17 @@ class CreateInputHecTokenByIDRequest(BaseModel):
 
 
 class CreateInputHecTokenByIDResponseTypedDict(TypedDict):
-    r"""a list of any objects"""
+    r"""a list of InputSplunkHec objects"""
 
     count: NotRequired[int]
     r"""number of items present in the items array"""
-    items: NotRequired[List[Dict[str, Any]]]
+    items: NotRequired[List[InputSplunkHecTypedDict]]
 
 
 class CreateInputHecTokenByIDResponse(BaseModel):
-    r"""a list of any objects"""
+    r"""a list of InputSplunkHec objects"""
 
     count: Optional[int] = None
     r"""number of items present in the items array"""
 
-    items: Optional[List[Dict[str, Any]]] = None
+    items: Optional[List[InputSplunkHec]] = None

cribl_control_plane/models/createroutesappendbyidop.py

@@ -11,7 +11,7 @@ from typing_extensions import Annotated, NotRequired, TypedDict
 
 class CreateRoutesAppendByIDRequestTypedDict(TypedDict):
     id: str
-    r"""the route table to be appended to - currently default is the only supported value"""
+    r"""The <code>id</code> of the Routing table to add the Route to. The supported value is <code>default</code>."""
     request_body: List[RouteConfTypedDict]
     r"""RouteDefinitions object"""
 
@@ -20,7 +20,7 @@ class CreateRoutesAppendByIDRequest(BaseModel):
     id: Annotated[
         str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
     ]
-    r"""the route table to be appended to - currently default is the only supported value"""
+    r"""The <code>id</code> of the Routing table to add the Route to. The supported value is <code>default</code>."""
 
     request_body: Annotated[
         List[RouteConf],

cribl_control_plane/models/createversionpushop.py

@@ -2,22 +2,22 @@
 
 from __future__ import annotations
 from cribl_control_plane.types import BaseModel
-from typing import Any, Dict, List, Optional
+from typing import List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
 class CreateVersionPushResponseTypedDict(TypedDict):
-    r"""a list of any objects"""
+    r"""a list of string objects"""
 
     count: NotRequired[int]
     r"""number of items present in the items array"""
-    items: NotRequired[List[Dict[str, Any]]]
+    items: NotRequired[List[str]]
 
 
 class CreateVersionPushResponse(BaseModel):
-    r"""a list of any objects"""
+    r"""a list of string objects"""
 
     count: Optional[int] = None
     r"""number of items present in the items array"""
 
-    items: Optional[List[Dict[str, Any]]] = None
+    items: Optional[List[str]] = None

cribl_control_plane/models/createversionrevertop.py

@@ -14,7 +14,7 @@ class CreateVersionRevertRequestTypedDict(TypedDict):
     git_revert_params: GitRevertParamsTypedDict
     r"""GitRevertParams object"""
     group_id: NotRequired[str]
-    r"""Group ID"""
+    r"""The <code>id</code> of the Worker Group or Edge Fleet to revert the commit for. Required in Distributed deployments. Omit in Single-instance deployments."""
 
 
 class CreateVersionRevertRequest(BaseModel):
@@ -29,7 +29,7 @@ class CreateVersionRevertRequest(BaseModel):
         pydantic.Field(alias="groupId"),
         FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
     ] = None
-    r"""Group ID"""
+    r"""The <code>id</code> of the Worker Group or Edge Fleet to revert the commit for. Required in Distributed deployments. Omit in Single-instance deployments."""
 
 
 class CreateVersionRevertResponseTypedDict(TypedDict):

cribl_control_plane/models/createversionundoop.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import FieldMetadata, QueryParamMetadata
 import pydantic
-from typing import List, Optional
+from typing import Any, Dict, List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -27,7 +27,7 @@ class CreateVersionUndoResponseTypedDict(TypedDict):
 
     count: NotRequired[int]
     r"""number of items present in the items array"""
-    items: NotRequired[List[str]]
+    items: NotRequired[List[Dict[str, Any]]]
 
 
 class CreateVersionUndoResponse(BaseModel):
@@ -36,4 +36,4 @@ class CreateVersionUndoResponse(BaseModel):
     count: Optional[int] = None
     r"""number of items present in the items array"""
 
-    items: Optional[List[str]] = None
+    items: Optional[List[Dict[str, Any]]] = None

cribl_control_plane/models/cribllakedataset.py

@@ -2,18 +2,23 @@
 
 from __future__ import annotations
 from .cacheconnection import CacheConnection, CacheConnectionTypedDict
+from .lakedatasetmetrics import LakeDatasetMetrics, LakeDatasetMetricsTypedDict
 from .lakedatasetsearchconfig import (
     LakeDatasetSearchConfig,
     LakeDatasetSearchConfigTypedDict,
 )
+from cribl_control_plane import models, utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class CriblLakeDatasetFormat(str, Enum):
+class CriblLakeDatasetFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     JSON = "json"
     DDSS = "ddss"
     PARQUET = "parquet"
@@ -28,6 +33,7 @@ class CriblLakeDatasetTypedDict(TypedDict):
     description: NotRequired[str]
     format_: NotRequired[CriblLakeDatasetFormat]
     http_da_used: NotRequired[bool]
+    metrics: NotRequired[LakeDatasetMetricsTypedDict]
     retention_period_in_days: NotRequired[float]
     search_config: NotRequired[LakeDatasetSearchConfigTypedDict]
     storage_location_id: NotRequired[str]
@@ -54,11 +60,16 @@ class CriblLakeDataset(BaseModel):
     description: Optional[str] = None
 
     format_: Annotated[
-        Optional[CriblLakeDatasetFormat], pydantic.Field(alias="format")
+        Annotated[
+            Optional[CriblLakeDatasetFormat], PlainValidator(validate_open_enum(False))
+        ],
+        pydantic.Field(alias="format"),
     ] = None
 
     http_da_used: Annotated[Optional[bool], pydantic.Field(alias="httpDAUsed")] = None
 
+    metrics: Optional[LakeDatasetMetrics] = None
+
     retention_period_in_days: Annotated[
         Optional[float], pydantic.Field(alias="retentionPeriodInDays")
     ] = None
@@ -72,3 +83,12 @@ class CriblLakeDataset(BaseModel):
     ] = None
 
     view_name: Annotated[Optional[str], pydantic.Field(alias="viewName")] = None
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.CriblLakeDatasetFormat(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/cribllakedatasetupdate.py

@@ -0,0 +1,95 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .cacheconnection import CacheConnection, CacheConnectionTypedDict
+from .lakedatasetmetrics import LakeDatasetMetrics, LakeDatasetMetricsTypedDict
+from .lakedatasetsearchconfig import (
+    LakeDatasetSearchConfig,
+    LakeDatasetSearchConfigTypedDict,
+)
+from cribl_control_plane import models, utils
+from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
+from enum import Enum
+import pydantic
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
+from typing import List, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class CriblLakeDatasetUpdateFormat(str, Enum, metaclass=utils.OpenEnumMeta):
+    JSON = "json"
+    DDSS = "ddss"
+    PARQUET = "parquet"
+
+
+class CriblLakeDatasetUpdateTypedDict(TypedDict):
+    accelerated_fields: NotRequired[List[str]]
+    bucket_name: NotRequired[str]
+    cache_connection: NotRequired[CacheConnectionTypedDict]
+    deletion_started_at: NotRequired[float]
+    description: NotRequired[str]
+    format_: NotRequired[CriblLakeDatasetUpdateFormat]
+    http_da_used: NotRequired[bool]
+    id: NotRequired[str]
+    metrics: NotRequired[LakeDatasetMetricsTypedDict]
+    retention_period_in_days: NotRequired[float]
+    search_config: NotRequired[LakeDatasetSearchConfigTypedDict]
+    storage_location_id: NotRequired[str]
+    view_name: NotRequired[str]
+
+
+class CriblLakeDatasetUpdate(BaseModel):
+    accelerated_fields: Annotated[
+        Optional[List[str]], pydantic.Field(alias="acceleratedFields")
+    ] = None
+
+    bucket_name: Annotated[Optional[str], pydantic.Field(alias="bucketName")] = None
+
+    cache_connection: Annotated[
+        Optional[CacheConnection], pydantic.Field(alias="cacheConnection")
+    ] = None
+
+    deletion_started_at: Annotated[
+        Optional[float], pydantic.Field(alias="deletionStartedAt")
+    ] = None
+
+    description: Optional[str] = None
+
+    format_: Annotated[
+        Annotated[
+            Optional[CriblLakeDatasetUpdateFormat],
+            PlainValidator(validate_open_enum(False)),
+        ],
+        pydantic.Field(alias="format"),
+    ] = None
+
+    http_da_used: Annotated[Optional[bool], pydantic.Field(alias="httpDAUsed")] = None
+
+    id: Optional[str] = None
+
+    metrics: Optional[LakeDatasetMetrics] = None
+
+    retention_period_in_days: Annotated[
+        Optional[float], pydantic.Field(alias="retentionPeriodInDays")
+    ] = None
+
+    search_config: Annotated[
+        Optional[LakeDatasetSearchConfig], pydantic.Field(alias="searchConfig")
+    ] = None
+
+    storage_location_id: Annotated[
+        Optional[str], pydantic.Field(alias="storageLocationId")
+    ] = None
+
+    view_name: Annotated[Optional[str], pydantic.Field(alias="viewName")] = None
+
+    @field_serializer("format_")
+    def serialize_format_(self, value):
+        if isinstance(value, str):
+            try:
+                return models.CriblLakeDatasetUpdateFormat(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/datasetmetadata.py

@@ -5,14 +5,18 @@ from .datasetmetadataruninfo import (
     DatasetMetadataRunInfo,
     DatasetMetadataRunInfoTypedDict,
 )
+from cribl_control_plane import models, utils
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-class ScanMode(str, Enum):
+class ScanMode(str, Enum, metaclass=utils.OpenEnumMeta):
     DETAILED = "detailed"
     QUICK = "quick"
 
@@ -32,8 +36,20 @@ class DatasetMetadata(BaseModel):
 
     field_list: Annotated[List[str], pydantic.Field(alias="fieldList")]
 
-    scan_mode: Annotated[ScanMode, pydantic.Field(alias="scanMode")]
+    scan_mode: Annotated[
+        Annotated[ScanMode, PlainValidator(validate_open_enum(False))],
+        pydantic.Field(alias="scanMode"),
+    ]
 
     latest_run_info: Annotated[
         Optional[DatasetMetadataRunInfo], pydantic.Field(alias="latestRunInfo")
     ] = None
+
+    @field_serializer("scan_mode")
+    def serialize_scan_mode(self, value):
+        if isinstance(value, str):
+            try:
+                return models.ScanMode(value)
+            except ValueError:
+                return value
+        return value