cribl-control-plane 0.2.1rc7__py3-none-any.whl → 0.2.1rc9__py3-none-any.whl

cribl_control_plane/models/cribllakedataset.py

@@ -2,16 +2,16 @@
 
 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 import 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
@@ -32,6 +32,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]
@@ -66,6 +67,8 @@ class CriblLakeDataset(BaseModel):
 
     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
@@ -79,12 +82,3 @@ 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

@@ -2,16 +2,16 @@
 
 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 import 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
@@ -32,6 +32,7 @@ class CriblLakeDatasetUpdateTypedDict(TypedDict):
     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]
@@ -67,6 +68,8 @@ class CriblLakeDatasetUpdate(BaseModel):
 
     id: Optional[str] = None
 
+    metrics: Optional[LakeDatasetMetrics] = None
+
     retention_period_in_days: Annotated[
         Optional[float], pydantic.Field(alias="retentionPeriodInDays")
     ] = None
@@ -80,12 +83,3 @@ class CriblLakeDatasetUpdate(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.CriblLakeDatasetUpdateFormat(value)
-            except ValueError:
-                return value
-        return value

cribl_control_plane/models/datasetmetadata.py

@@ -5,12 +5,11 @@ from .datasetmetadataruninfo import (
     DatasetMetadataRunInfo,
     DatasetMetadataRunInfoTypedDict,
 )
-from cribl_control_plane import models, utils
+from cribl_control_plane import 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
@@ -44,12 +43,3 @@ class DatasetMetadata(BaseModel):
     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

cribl_control_plane/models/deleteadminproductsmappingsbyproductandidop.py

@@ -0,0 +1,51 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .mappingruleset import MappingRuleset, MappingRulesetTypedDict
+from .productscore import ProductsCore
+from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import (
+    FieldMetadata,
+    PathParamMetadata,
+    validate_open_enum,
+)
+from pydantic.functional_validators import PlainValidator
+from typing import List, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class DeleteAdminProductsMappingsByProductAndIDRequestTypedDict(TypedDict):
+    product: ProductsCore
+    r"""Name of the Cribl product to delete the Mapping Ruleset for"""
+    id: str
+    r"""The <code>id</code> of the Mapping Ruleset to delete."""
+
+
+class DeleteAdminProductsMappingsByProductAndIDRequest(BaseModel):
+    product: Annotated[
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Name of the Cribl product to delete the Mapping Ruleset for"""
+
+    id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+    r"""The <code>id</code> of the Mapping Ruleset to delete."""
+
+
+class DeleteAdminProductsMappingsByProductAndIDResponseTypedDict(TypedDict):
+    r"""A list containing the deleted Mapping Ruleset objects"""
+
+    count: NotRequired[int]
+    r"""number of items present in the items array"""
+    items: NotRequired[List[MappingRulesetTypedDict]]
+
+
+class DeleteAdminProductsMappingsByProductAndIDResponse(BaseModel):
+    r"""A list containing the deleted Mapping Ruleset objects"""
+
+    count: Optional[int] = None
+    r"""number of items present in the items array"""
+
+    items: Optional[List[MappingRuleset]] = None

cribl_control_plane/models/deleteconfiggroupbyproductandidop.py

@@ -3,14 +3,12 @@
 from __future__ import annotations
 from .configgroup import ConfigGroup, ConfigGroupTypedDict
 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,
     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
@@ -35,15 +33,6 @@ class DeleteConfigGroupByProductAndIDRequest(BaseModel):
     ]
     r"""The <code>id</code> of the Worker Group or Edge Fleet to delete."""
 
-    @field_serializer("product")
-    def serialize_product(self, value):
-        if isinstance(value, str):
-            try:
-                return models.ProductsCore(value)
-            except ValueError:
-                return value
-        return value
-
 
 class DeleteConfigGroupByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of ConfigGroup objects"""

cribl_control_plane/models/evalfunction.py

@@ -0,0 +1,77 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .evalschema import EvalSchema, EvalSchemaTypedDict
+from cribl_control_plane import 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.functional_validators import PlainValidator
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class ID(str, Enum, metaclass=utils.OpenEnumMeta):
+    EVAL = "eval"
+
+
+class EvalFunctionFunctionSpecificConfigsTypedDict(TypedDict):
+    pass
+
+
+class EvalFunctionFunctionSpecificConfigs(BaseModel):
+    pass
+
+
+class Name(str, Enum, metaclass=utils.OpenEnumMeta):
+    EVAL = "Eval"
+
+
+class Group(str, Enum, metaclass=utils.OpenEnumMeta):
+    STANDARD = "Standard"
+
+
+class EvalFunctionTypedDict(TypedDict):
+    id: ID
+    conf: EvalFunctionFunctionSpecificConfigsTypedDict
+    filter_: NotRequired[str]
+    r"""Filter that selects data to be fed through this Function"""
+    description: NotRequired[str]
+    r"""Simple description of this step"""
+    disabled: NotRequired[bool]
+    r"""If true, data will not be pushed through this function"""
+    final: NotRequired[bool]
+    r"""If enabled, stops the results of this Function from being passed to the downstream Functions"""
+    group_id: NotRequired[str]
+    r"""Group ID"""
+    name: NotRequired[Name]
+    group: NotRequired[Group]
+    schema_: NotRequired[EvalSchemaTypedDict]
+
+
+class EvalFunction(BaseModel):
+    id: Annotated[ID, PlainValidator(validate_open_enum(False))]
+
+    conf: EvalFunctionFunctionSpecificConfigs
+
+    filter_: Annotated[Optional[str], pydantic.Field(alias="filter")] = "true"
+    r"""Filter that selects data to be fed through this Function"""
+
+    description: Optional[str] = None
+    r"""Simple description of this step"""
+
+    disabled: Optional[bool] = None
+    r"""If true, data will not be pushed through this function"""
+
+    final: Optional[bool] = None
+    r"""If enabled, stops the results of this Function from being passed to the downstream Functions"""
+
+    group_id: Annotated[Optional[str], pydantic.Field(alias="groupId")] = None
+    r"""Group ID"""
+
+    name: Annotated[Optional[Name], PlainValidator(validate_open_enum(False))] = None
+
+    group: Annotated[Optional[Group], PlainValidator(validate_open_enum(False))] = None
+
+    schema_: Annotated[Optional[EvalSchema], pydantic.Field(alias="schema")] = None

cribl_control_plane/models/evalschema.py

@@ -0,0 +1,44 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from cribl_control_plane.types import BaseModel
+from typing import List, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+class AddTypedDict(TypedDict):
+    value: str
+    r"""JavaScript expression to compute the value (can be constant)"""
+    name: NotRequired[str]
+    disabled: NotRequired[bool]
+    r"""Set to No to disable the evaluation of an individual expression"""
+
+
+class Add(BaseModel):
+    value: str
+    r"""JavaScript expression to compute the value (can be constant)"""
+
+    name: Optional[str] = None
+
+    disabled: Optional[bool] = False
+    r"""Set to No to disable the evaluation of an individual expression"""
+
+
+class EvalSchemaTypedDict(TypedDict):
+    add: NotRequired[List[AddTypedDict]]
+    r"""Set of key-value pairs to evaluate and add/set"""
+    keep: NotRequired[List[str]]
+    r"""List of fields to keep. Supports * wildcards. Takes precedence over 'Remove fields'."""
+    remove: NotRequired[List[str]]
+    r"""List of fields to remove. Supports * wildcards. Fields that match 'Keep fields' will not be removed. Enclose field names containing special characters in single or double quotes."""
+
+
+class EvalSchema(BaseModel):
+    add: Optional[List[Add]] = None
+    r"""Set of key-value pairs to evaluate and add/set"""
+
+    keep: Optional[List[str]] = None
+    r"""List of fields to keep. Supports * wildcards. Takes precedence over 'Remove fields'."""
+
+    remove: Optional[List[str]] = None
+    r"""List of fields to remove. Supports * wildcards. Fields that match 'Keep fields' will not be removed. Enclose field names containing special characters in single or double quotes."""

cribl_control_plane/models/getadminproductsmappingsbyproductandidop.py

@@ -0,0 +1,51 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .mappingruleset import MappingRuleset, MappingRulesetTypedDict
+from .productscore import ProductsCore
+from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import (
+    FieldMetadata,
+    PathParamMetadata,
+    validate_open_enum,
+)
+from pydantic.functional_validators import PlainValidator
+from typing import List, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class GetAdminProductsMappingsByProductAndIDRequestTypedDict(TypedDict):
+    product: ProductsCore
+    r"""Name of the Cribl product to get the Mappings for"""
+    id: str
+    r"""The <code>id</code> of the Worker Group or Edge Fleet Mapping Ruleset to get"""
+
+
+class GetAdminProductsMappingsByProductAndIDRequest(BaseModel):
+    product: Annotated[
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Name of the Cribl product to get the Mappings for"""
+
+    id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+    r"""The <code>id</code> of the Worker Group or Edge Fleet Mapping Ruleset to get"""
+
+
+class GetAdminProductsMappingsByProductAndIDResponseTypedDict(TypedDict):
+    r"""a list of MappingRuleset objects"""
+
+    count: NotRequired[int]
+    r"""number of items present in the items array"""
+    items: NotRequired[List[MappingRulesetTypedDict]]
+
+
+class GetAdminProductsMappingsByProductAndIDResponse(BaseModel):
+    r"""a list of MappingRuleset objects"""
+
+    count: Optional[int] = None
+    r"""number of items present in the items array"""
+
+    items: Optional[List[MappingRuleset]] = None

cribl_control_plane/models/getadminproductsmappingsbyproductop.py

@@ -0,0 +1,44 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .mappingruleset import MappingRuleset, MappingRulesetTypedDict
+from .productscore import ProductsCore
+from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import (
+    FieldMetadata,
+    PathParamMetadata,
+    validate_open_enum,
+)
+from pydantic.functional_validators import PlainValidator
+from typing import List, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class GetAdminProductsMappingsByProductRequestTypedDict(TypedDict):
+    product: ProductsCore
+    r"""Name of the Cribl product to list the Mapping Rulesets for"""
+
+
+class GetAdminProductsMappingsByProductRequest(BaseModel):
+    product: Annotated[
+        Annotated[ProductsCore, PlainValidator(validate_open_enum(False))],
+        FieldMetadata(path=PathParamMetadata(style="simple", explode=False)),
+    ]
+    r"""Name of the Cribl product to list the Mapping Rulesets for"""
+
+
+class GetAdminProductsMappingsByProductResponseTypedDict(TypedDict):
+    r"""a list of MappingRuleset objects"""
+
+    count: NotRequired[int]
+    r"""number of items present in the items array"""
+    items: NotRequired[List[MappingRulesetTypedDict]]
+
+
+class GetAdminProductsMappingsByProductResponse(BaseModel):
+    r"""a list of MappingRuleset objects"""
+
+    count: Optional[int] = None
+    r"""number of items present in the items array"""
+
+    items: Optional[List[MappingRuleset]] = None

cribl_control_plane/models/getconfiggroupaclbyproductandidop.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 from .productscore import ProductsCore
 from .rbacresource import RbacResource
 from .useraccesscontrollist import UserAccessControlList, UserAccessControlListTypedDict
-from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
@@ -12,7 +11,6 @@ from cribl_control_plane.utils import (
     QueryParamMetadata,
     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
@@ -45,24 +43,6 @@ class GetConfigGroupACLByProductAndIDRequest(BaseModel):
     ] = None
     r"""Filter for limiting the response to ACL entries for the specified RBAC resource type."""
 
-    @field_serializer("product")
-    def serialize_product(self, value):
-        if isinstance(value, str):
-            try:
-                return models.ProductsCore(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("type")
-    def serialize_type(self, value):
-        if isinstance(value, str):
-            try:
-                return models.RbacResource(value)
-            except ValueError:
-                return value
-        return value
-
 
 class GetConfigGroupACLByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of UserAccessControlList objects"""

cribl_control_plane/models/getconfiggroupaclteamsbyproductandidop.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 from .productscore import ProductsCore
 from .rbacresource import RbacResource
 from .teamaccesscontrollist import TeamAccessControlList, TeamAccessControlListTypedDict
-from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
@@ -12,7 +11,6 @@ from cribl_control_plane.utils import (
     QueryParamMetadata,
     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
@@ -45,24 +43,6 @@ class GetConfigGroupACLTeamsByProductAndIDRequest(BaseModel):
     ] = None
     r"""Filter for limiting the response to ACL entries for the specified RBAC resource type."""
 
-    @field_serializer("product")
-    def serialize_product(self, value):
-        if isinstance(value, str):
-            try:
-                return models.ProductsCore(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("type")
-    def serialize_type(self, value):
-        if isinstance(value, str):
-            try:
-                return models.RbacResource(value)
-            except ValueError:
-                return value
-        return value
-
 
 class GetConfigGroupACLTeamsByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of TeamAccessControlList objects"""

cribl_control_plane/models/getconfiggroupbyproductandidop.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 from .configgroup import ConfigGroup, ConfigGroupTypedDict
 from .productscore import ProductsCore
-from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
@@ -11,7 +10,6 @@ from cribl_control_plane.utils import (
     QueryParamMetadata,
     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
@@ -44,15 +42,6 @@ class GetConfigGroupByProductAndIDRequest(BaseModel):
     ] = None
     r"""Comma-separated list of additional properties to include in the response. Available values are <code>git.commit</code>, <code>git.localChanges</code>, and <code>git.log</code>."""
 
-    @field_serializer("product")
-    def serialize_product(self, value):
-        if isinstance(value, str):
-            try:
-                return models.ProductsCore(value)
-            except ValueError:
-                return value
-        return value
-
 
 class GetConfigGroupByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of ConfigGroup objects"""

cribl_control_plane/models/getconfiggroupconfigversionbyproductandidop.py

@@ -2,14 +2,12 @@
 
 from __future__ import annotations
 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,
     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
@@ -34,15 +32,6 @@ class GetConfigGroupConfigVersionByProductAndIDRequest(BaseModel):
     ]
     r"""The <code>id</code> of the Worker Group or Edge Fleet to get the configuration version for."""
 
-    @field_serializer("product")
-    def serialize_product(self, value):
-        if isinstance(value, str):
-            try:
-                return models.ProductsCore(value)
-            except ValueError:
-                return value
-        return value
-
 
 class GetConfigGroupConfigVersionByProductAndIDResponseTypedDict(TypedDict):
     r"""a list of string objects"""

cribl_control_plane/models/getsummaryop.py

@@ -3,14 +3,12 @@
 from __future__ import annotations
 from .distributedsummary import DistributedSummary, DistributedSummaryTypedDict
 from .workertypes import WorkerTypes
-from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import (
     FieldMetadata,
     QueryParamMetadata,
     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
@@ -28,15 +26,6 @@ class GetSummaryRequest(BaseModel):
     ] = None
     r"""Filter for limiting the response by Cribl product: Cribl Stream (<code>worker</code>) or Cribl Edge (<code>managed-edge</code>)."""
 
-    @field_serializer("mode")
-    def serialize_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.WorkerTypes(value)
-            except ValueError:
-                return value
-        return value
-
 
 class GetSummaryResponseTypedDict(TypedDict):
     r"""a list of DistributedSummary objects"""

cribl_control_plane/models/groupcreaterequest.py

@@ -4,12 +4,11 @@ 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 import 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
@@ -151,21 +150,3 @@ class GroupCreateRequest(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.GroupCreateRequestEstimatedIngestRate(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("type")
-    def serialize_type(self, value):
-        if isinstance(value, str):
-            try:
-                return models.GroupCreateRequestType(value)
-            except ValueError:
-                return value
-        return value

cribl_control_plane/models/hbcriblinfo.py

@@ -3,12 +3,11 @@
 from __future__ import annotations
 from .hbleaderinfo import HBLeaderInfo, HBLeaderInfoTypedDict
 from .lookupversions import LookupVersions, LookupVersionsTypedDict
-from cribl_control_plane import models, utils
+from cribl_control_plane import 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
@@ -105,12 +104,3 @@ class HBCriblInfo(BaseModel):
     )
 
     version: Optional[str] = None
-
-    @field_serializer("dist_mode")
-    def serialize_dist_mode(self, value):
-        if isinstance(value, str):
-            try:
-                return models.DistMode(value)
-            except ValueError:
-                return value
-        return value