cribl-control-plane 0.4.0b23__py3-none-any.whl → 0.5.0b3__py3-none-any.whl

cribl_control_plane/models/functionnotifications.py

@@ -1,6 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .functionconfschemanotifications import (
+    FunctionConfSchemaNotifications,
+    FunctionConfSchemaNotificationsTypedDict,
+)
 from cribl_control_plane.types import BaseModel
 from enum import Enum
 import pydantic
@@ -12,26 +16,6 @@ class FunctionNotificationsID(str, Enum):
     NOTIFICATIONS = "notifications"
 
 
-class FunctionNotificationsSchemaTypedDict(TypedDict):
-    id: str
-    r"""Notification ID"""
-    field: str
-    r"""Notification event state field name"""
-    deduplicate: bool
-    r"""Toggle deduplication."""
-
-
-class FunctionNotificationsSchema(BaseModel):
-    id: str
-    r"""Notification ID"""
-
-    field: str
-    r"""Notification event state field name"""
-
-    deduplicate: bool
-    r"""Toggle deduplication."""
-
-
 class FunctionNotificationsTypedDict(TypedDict):
     filename: str
     group: str
@@ -46,7 +30,7 @@ class FunctionNotificationsTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[FunctionNotificationsSchemaTypedDict]
+    schema_: NotRequired[FunctionConfSchemaNotificationsTypedDict]
 
 
 class FunctionNotifications(BaseModel):
@@ -81,5 +65,5 @@ class FunctionNotifications(BaseModel):
     sync: Optional[bool] = None
 
     schema_: Annotated[
-        Optional[FunctionNotificationsSchema], pydantic.Field(alias="schema")
+        Optional[FunctionConfSchemaNotifications], pydantic.Field(alias="schema")
     ] = None

cribl_control_plane/models/functionnotify.py

@@ -1,13 +1,13 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import models, utils
+from .functionconfschemanotify import (
+    FunctionConfSchemaNotify,
+    FunctionConfSchemaNotifyTypedDict,
+)
 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 Any, Dict, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -16,133 +16,6 @@ class FunctionNotifyID(str, Enum):
     NOTIFY = "notify"
 
 
-class TriggerType(str, Enum, metaclass=utils.OpenEnumMeta):
-    r"""Type of the trigger condition. custom applies a kusto expression over the results, and results count applies a comparison over results count"""
-
-    # Where
-    CUSTOM = "custom"
-    # Count of Results
-    RESULTS_COUNT = "resultsCount"
-
-
-class CountComparator(str, Enum, metaclass=utils.OpenEnumMeta):
-    r"""Operation to be applied over the results count"""
-
-    # greater than
-    GREATER_THAN_ = ">"
-    # less than
-    LESS_THAN_ = "<"
-    # equals
-    EQUAL_EQUAL_EQUAL_ = "==="
-    # not equal to
-    NOT_EQUAL_EQUAL_ = "!=="
-    # greater than or equal to
-    GREATER_THAN_EQUAL_ = ">="
-    # less than or equal to
-    LESS_THAN_EQUAL_ = "<="
-
-
-class NotifyConfigurationTypedDict(TypedDict):
-    search_id: str
-    r"""Id of the search this function is running on."""
-    saved_query_id: str
-    r"""Id of the saved query"""
-    search_url: str
-    r"""Url of the search results"""
-    auth_token: str
-    r"""Auth token for sending notification messages"""
-    messages_endpoint: str
-    r"""System messages api endpoint"""
-    group: NotRequired[str]
-    r"""Group the notification belongs to"""
-    notification_id: NotRequired[str]
-    r"""Workspace within the deployment to send the search results to."""
-    trigger: NotRequired[str]
-    r"""Js expression that filters events, a greater than 'Trigger Count' events will trigger the notification"""
-    trigger_type: NotRequired[TriggerType]
-    r"""Type of the trigger condition. custom applies a kusto expression over the results, and results count applies a comparison over results count"""
-    trigger_comparator: NotRequired[CountComparator]
-    r"""Operation to be applied over the results count"""
-    trigger_count: NotRequired[float]
-    r"""How many results that match trigger the condition"""
-    results_limit: NotRequired[float]
-    r"""Number of results to include in the notification event"""
-    message: NotRequired[str]
-    r"""Message content template, available fields: searchId, resultSet, savedQueryId, notificationId, searchResultsUrl"""
-    tenant_id: NotRequired[str]
-    r"""Current tenant id"""
-
-
-class NotifyConfiguration(BaseModel):
-    search_id: Annotated[str, pydantic.Field(alias="searchId")]
-    r"""Id of the search this function is running on."""
-
-    saved_query_id: Annotated[str, pydantic.Field(alias="savedQueryId")]
-    r"""Id of the saved query"""
-
-    search_url: Annotated[str, pydantic.Field(alias="searchUrl")]
-    r"""Url of the search results"""
-
-    auth_token: Annotated[str, pydantic.Field(alias="authToken")]
-    r"""Auth token for sending notification messages"""
-
-    messages_endpoint: Annotated[str, pydantic.Field(alias="messagesEndpoint")]
-    r"""System messages api endpoint"""
-
-    group: Optional[str] = "default"
-    r"""Group the notification belongs to"""
-
-    notification_id: Annotated[
-        Optional[str], pydantic.Field(alias="notificationId")
-    ] = "main"
-    r"""Workspace within the deployment to send the search results to."""
-
-    trigger: Optional[str] = None
-    r"""Js expression that filters events, a greater than 'Trigger Count' events will trigger the notification"""
-
-    trigger_type: Annotated[
-        Annotated[Optional[TriggerType], PlainValidator(validate_open_enum(False))],
-        pydantic.Field(alias="triggerType"),
-    ] = None
-    r"""Type of the trigger condition. custom applies a kusto expression over the results, and results count applies a comparison over results count"""
-
-    trigger_comparator: Annotated[
-        Annotated[Optional[CountComparator], PlainValidator(validate_open_enum(False))],
-        pydantic.Field(alias="triggerComparator"),
-    ] = None
-    r"""Operation to be applied over the results count"""
-
-    trigger_count: Annotated[Optional[float], pydantic.Field(alias="triggerCount")] = 0
-    r"""How many results that match trigger the condition"""
-
-    results_limit: Annotated[Optional[float], pydantic.Field(alias="resultsLimit")] = 50
-    r"""Number of results to include in the notification event"""
-
-    message: Optional[str] = None
-    r"""Message content template, available fields: searchId, resultSet, savedQueryId, notificationId, searchResultsUrl"""
-
-    tenant_id: Annotated[Optional[str], pydantic.Field(alias="tenantId")] = None
-    r"""Current tenant id"""
-
-    @field_serializer("trigger_type")
-    def serialize_trigger_type(self, value):
-        if isinstance(value, str):
-            try:
-                return models.TriggerType(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("trigger_comparator")
-    def serialize_trigger_comparator(self, value):
-        if isinstance(value, str):
-            try:
-                return models.CountComparator(value)
-            except ValueError:
-                return value
-        return value
-
-
 class FunctionNotifyTypedDict(TypedDict):
     filename: str
     group: str
@@ -157,7 +30,7 @@ class FunctionNotifyTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[NotifyConfigurationTypedDict]
+    schema_: NotRequired[FunctionConfSchemaNotifyTypedDict]
 
 
 class FunctionNotify(BaseModel):
@@ -192,5 +65,5 @@ class FunctionNotify(BaseModel):
     sync: Optional[bool] = None
 
     schema_: Annotated[
-        Optional[NotifyConfiguration], pydantic.Field(alias="schema")
+        Optional[FunctionConfSchemaNotify], pydantic.Field(alias="schema")
     ] = None

cribl_control_plane/models/functionnumerify.py

@@ -1,14 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import models, utils
+from .functionconfschemanumerify import (
+    FunctionConfSchemaNumerify,
+    FunctionConfSchemaNumerifyTypedDict,
+)
 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 Any, Dict, List, Optional
+from typing import Any, Dict, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -16,56 +16,6 @@ class FunctionNumerifyID(str, Enum):
     NUMERIFY = "numerify"
 
 
-class FunctionNumerifyFormat(str, Enum, metaclass=utils.OpenEnumMeta):
-    # None
-    NONE = "none"
-    # Fix
-    FIX = "fix"
-    # Floor
-    FLOOR = "floor"
-    # Ceil
-    CEIL = "ceil"
-
-
-class FunctionNumerifySchemaTypedDict(TypedDict):
-    depth: NotRequired[int]
-    r"""Depth to which the Numerify Function will search within a nested event. Depth greater than 5 (the default) could decrease performance."""
-    ignore_fields: NotRequired[List[str]]
-    r"""Fields to NOT numerify. Takes precedence over 'Include expression' when set. Supports wildcards. A '!' before field name(s) means: numerify all fields EXCEPT these. For syntax details, see [Wildcard Lists](https://docs.cribl.io/stream/introduction-reference/#wildcard-lists)."""
-    filter_expr: NotRequired[str]
-    r"""Optional JavaScript expression to determine whether a field should be numerified. If left blank, all fields will be numerified. Use the 'name' and 'value' global variables to access fields' names/values. Examples: `value != null`, `name=='fieldname'`. You can access other fields' values via `__e.<fieldName>`."""
-    format_: NotRequired[FunctionNumerifyFormat]
-
-
-class FunctionNumerifySchema(BaseModel):
-    depth: Optional[int] = 5
-    r"""Depth to which the Numerify Function will search within a nested event. Depth greater than 5 (the default) could decrease performance."""
-
-    ignore_fields: Annotated[
-        Optional[List[str]], pydantic.Field(alias="ignoreFields")
-    ] = None
-    r"""Fields to NOT numerify. Takes precedence over 'Include expression' when set. Supports wildcards. A '!' before field name(s) means: numerify all fields EXCEPT these. For syntax details, see [Wildcard Lists](https://docs.cribl.io/stream/introduction-reference/#wildcard-lists)."""
-
-    filter_expr: Annotated[Optional[str], pydantic.Field(alias="filterExpr")] = None
-    r"""Optional JavaScript expression to determine whether a field should be numerified. If left blank, all fields will be numerified. Use the 'name' and 'value' global variables to access fields' names/values. Examples: `value != null`, `name=='fieldname'`. You can access other fields' values via `__e.<fieldName>`."""
-
-    format_: Annotated[
-        Annotated[
-            Optional[FunctionNumerifyFormat], PlainValidator(validate_open_enum(False))
-        ],
-        pydantic.Field(alias="format"),
-    ] = FunctionNumerifyFormat.NONE
-
-    @field_serializer("format_")
-    def serialize_format_(self, value):
-        if isinstance(value, str):
-            try:
-                return models.FunctionNumerifyFormat(value)
-            except ValueError:
-                return value
-        return value
-
-
 class FunctionNumerifyTypedDict(TypedDict):
     filename: str
     group: str
@@ -80,7 +30,7 @@ class FunctionNumerifyTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[FunctionNumerifySchemaTypedDict]
+    schema_: NotRequired[FunctionConfSchemaNumerifyTypedDict]
 
 
 class FunctionNumerify(BaseModel):
@@ -115,5 +65,5 @@ class FunctionNumerify(BaseModel):
     sync: Optional[bool] = None
 
     schema_: Annotated[
-        Optional[FunctionNumerifySchema], pydantic.Field(alias="schema")
+        Optional[FunctionConfSchemaNumerify], pydantic.Field(alias="schema")
     ] = None

cribl_control_plane/models/functionotlplogs.py

@@ -1,6 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .functionconfschemaotlplogs import (
+    FunctionConfSchemaOtlpLogs,
+    FunctionConfSchemaOtlpLogsTypedDict,
+)
 from cribl_control_plane.types import BaseModel
 from enum import Enum
 import pydantic
@@ -12,23 +16,6 @@ class FunctionOtlpLogsID(str, Enum):
     OTLP_LOGS = "otlp_logs"
 
 
-class FunctionOtlpLogsSchemaTypedDict(TypedDict):
-    drop_non_log_events: NotRequired[bool]
-    batch_otlp_logs: NotRequired[bool]
-    r"""Batch OTLP log records by shared top-level `resource` attributes"""
-
-
-class FunctionOtlpLogsSchema(BaseModel):
-    drop_non_log_events: Annotated[
-        Optional[bool], pydantic.Field(alias="dropNonLogEvents")
-    ] = False
-
-    batch_otlp_logs: Annotated[
-        Optional[bool], pydantic.Field(alias="batchOTLPLogs")
-    ] = False
-    r"""Batch OTLP log records by shared top-level `resource` attributes"""
-
-
 class FunctionOtlpLogsTypedDict(TypedDict):
     filename: str
     group: str
@@ -43,7 +30,7 @@ class FunctionOtlpLogsTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[FunctionOtlpLogsSchemaTypedDict]
+    schema_: NotRequired[FunctionConfSchemaOtlpLogsTypedDict]
 
 
 class FunctionOtlpLogs(BaseModel):
@@ -78,5 +65,5 @@ class FunctionOtlpLogs(BaseModel):
     sync: Optional[bool] = None
 
     schema_: Annotated[
-        Optional[FunctionOtlpLogsSchema], pydantic.Field(alias="schema")
+        Optional[FunctionConfSchemaOtlpLogs], pydantic.Field(alias="schema")
     ] = None

cribl_control_plane/models/functionotlpmetrics.py

@@ -1,14 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import models, utils
+from .functionconfschemaotlpmetrics import (
+    FunctionConfSchemaOtlpMetrics,
+    FunctionConfSchemaOtlpMetricsTypedDict,
+)
 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 Any, Dict, List, Optional
+from typing import Any, Dict, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -16,55 +16,6 @@ class FunctionOtlpMetricsID(str, Enum):
     OTLP_METRICS = "otlp_metrics"
 
 
-class FunctionOTLPMetricsOTLPVersion(str, Enum, metaclass=utils.OpenEnumMeta):
-    # 0.10.0
-    ZERO_DOT_10_DOT_0 = "0.10.0"
-    # 1.3.1
-    ONE_DOT_3_DOT_1 = "1.3.1"
-
-
-class FunctionOtlpMetricsSchemaTypedDict(TypedDict):
-    resource_attribute_prefixes: NotRequired[List[str]]
-    r"""The prefixes of top-level attributes to add as resource attributes. Each attribute must match the regex pattern `^[a-zA-Z0-9_\.]+$`. Use Eval to copy nested attributes to the top level for matching."""
-    drop_non_metric_events: NotRequired[bool]
-    otlp_version: NotRequired[FunctionOTLPMetricsOTLPVersion]
-    batch_otlp_metrics: NotRequired[bool]
-    r"""Batch OTLP metrics by shared top-level `resource` attributes"""
-
-
-class FunctionOtlpMetricsSchema(BaseModel):
-    resource_attribute_prefixes: Annotated[
-        Optional[List[str]], pydantic.Field(alias="resourceAttributePrefixes")
-    ] = None
-    r"""The prefixes of top-level attributes to add as resource attributes. Each attribute must match the regex pattern `^[a-zA-Z0-9_\.]+$`. Use Eval to copy nested attributes to the top level for matching."""
-
-    drop_non_metric_events: Annotated[
-        Optional[bool], pydantic.Field(alias="dropNonMetricEvents")
-    ] = False
-
-    otlp_version: Annotated[
-        Annotated[
-            Optional[FunctionOTLPMetricsOTLPVersion],
-            PlainValidator(validate_open_enum(False)),
-        ],
-        pydantic.Field(alias="otlpVersion"),
-    ] = FunctionOTLPMetricsOTLPVersion.ZERO_DOT_10_DOT_0
-
-    batch_otlp_metrics: Annotated[
-        Optional[bool], pydantic.Field(alias="batchOTLPMetrics")
-    ] = False
-    r"""Batch OTLP metrics by shared top-level `resource` attributes"""
-
-    @field_serializer("otlp_version")
-    def serialize_otlp_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.FunctionOTLPMetricsOTLPVersion(value)
-            except ValueError:
-                return value
-        return value
-
-
 class FunctionOtlpMetricsTypedDict(TypedDict):
     filename: str
     group: str
@@ -79,7 +30,7 @@ class FunctionOtlpMetricsTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[FunctionOtlpMetricsSchemaTypedDict]
+    schema_: NotRequired[FunctionConfSchemaOtlpMetricsTypedDict]
 
 
 class FunctionOtlpMetrics(BaseModel):
@@ -114,5 +65,5 @@ class FunctionOtlpMetrics(BaseModel):
     sync: Optional[bool] = None
 
     schema_: Annotated[
-        Optional[FunctionOtlpMetricsSchema], pydantic.Field(alias="schema")
+        Optional[FunctionConfSchemaOtlpMetrics], pydantic.Field(alias="schema")
     ] = None

cribl_control_plane/models/functionotlptraces.py

@@ -1,13 +1,13 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import models, utils
+from .functionconfschemaotlptraces import (
+    FunctionConfSchemaOtlpTraces,
+    FunctionConfSchemaOtlpTracesTypedDict,
+)
 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 Any, Dict, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
@@ -16,48 +16,6 @@ class FunctionOtlpTracesID(str, Enum):
     OTLP_TRACES = "otlp_traces"
 
 
-class FunctionOTLPTracesOTLPVersion(str, Enum, metaclass=utils.OpenEnumMeta):
-    # 0.10.0
-    ZERO_DOT_10_DOT_0 = "0.10.0"
-    # 1.3.1
-    ONE_DOT_3_DOT_1 = "1.3.1"
-
-
-class FunctionOtlpTracesSchemaTypedDict(TypedDict):
-    drop_non_trace_events: NotRequired[bool]
-    otlp_version: NotRequired[FunctionOTLPTracesOTLPVersion]
-    batch_otlp_traces: NotRequired[bool]
-    r"""Batch OTLP traces by shared top-level `resource` attributes"""
-
-
-class FunctionOtlpTracesSchema(BaseModel):
-    drop_non_trace_events: Annotated[
-        Optional[bool], pydantic.Field(alias="dropNonTraceEvents")
-    ] = False
-
-    otlp_version: Annotated[
-        Annotated[
-            Optional[FunctionOTLPTracesOTLPVersion],
-            PlainValidator(validate_open_enum(False)),
-        ],
-        pydantic.Field(alias="otlpVersion"),
-    ] = FunctionOTLPTracesOTLPVersion.ZERO_DOT_10_DOT_0
-
-    batch_otlp_traces: Annotated[
-        Optional[bool], pydantic.Field(alias="batchOTLPTraces")
-    ] = False
-    r"""Batch OTLP traces by shared top-level `resource` attributes"""
-
-    @field_serializer("otlp_version")
-    def serialize_otlp_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.FunctionOTLPTracesOTLPVersion(value)
-            except ValueError:
-                return value
-        return value
-
-
 class FunctionOtlpTracesTypedDict(TypedDict):
     filename: str
     group: str
@@ -72,7 +30,7 @@ class FunctionOtlpTracesTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[FunctionOtlpTracesSchemaTypedDict]
+    schema_: NotRequired[FunctionConfSchemaOtlpTracesTypedDict]
 
 
 class FunctionOtlpTraces(BaseModel):
@@ -107,5 +65,5 @@ class FunctionOtlpTraces(BaseModel):
     sync: Optional[bool] = None
 
     schema_: Annotated[
-        Optional[FunctionOtlpTracesSchema], pydantic.Field(alias="schema")
+        Optional[FunctionConfSchemaOtlpTraces], pydantic.Field(alias="schema")
     ] = None

cribl_control_plane/models/functionpack.py

@@ -1,10 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .functionconfschemapack import (
+    FunctionConfSchemaPack,
+    FunctionConfSchemaPackTypedDict,
+)
 from cribl_control_plane.types import BaseModel
 from enum import Enum
 import pydantic
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -12,21 +16,6 @@ class FunctionPackID(str, Enum):
     PACK = "pack"
 
 
-class FunctionPackSchemaTypedDict(TypedDict):
-    unpacked_fields: List[str]
-    r"""List of fields to keep, everything else will be packed"""
-    target: NotRequired[str]
-    r"""Name of the (packed) target field"""
-
-
-class FunctionPackSchema(BaseModel):
-    unpacked_fields: Annotated[List[str], pydantic.Field(alias="unpackedFields")]
-    r"""List of fields to keep, everything else will be packed"""
-
-    target: Optional[str] = "_pack"
-    r"""Name of the (packed) target field"""
-
-
 class FunctionPackTypedDict(TypedDict):
     filename: str
     group: str
@@ -41,7 +30,7 @@ class FunctionPackTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[FunctionPackSchemaTypedDict]
+    schema_: NotRequired[FunctionConfSchemaPackTypedDict]
 
 
 class FunctionPack(BaseModel):
@@ -75,6 +64,6 @@ class FunctionPack(BaseModel):
 
     sync: Optional[bool] = None
 
-    schema_: Annotated[Optional[FunctionPackSchema], pydantic.Field(alias="schema")] = (
-        None
-    )
+    schema_: Annotated[
+        Optional[FunctionConfSchemaPack], pydantic.Field(alias="schema")
+    ] = None

cribl_control_plane/models/functionpivot.py

@@ -1,10 +1,14 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .functionconfschemapivot import (
+    FunctionConfSchemaPivot,
+    FunctionConfSchemaPivotTypedDict,
+)
 from cribl_control_plane.types import BaseModel
 from enum import Enum
 import pydantic
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
@@ -12,26 +16,6 @@ class FunctionPivotID(str, Enum):
     PIVOT = "pivot"
 
 
-class SimplePivotConfigurationTypedDict(TypedDict):
-    label_field: str
-    r"""Fields to be used for the left-most column."""
-    data_fields: List[str]
-    r"""Fields with the cell values (i.e. aggregates)"""
-    qualifier_fields: List[str]
-    r"""Fields to qualify or group data fields"""
-
-
-class SimplePivotConfiguration(BaseModel):
-    label_field: Annotated[str, pydantic.Field(alias="labelField")]
-    r"""Fields to be used for the left-most column."""
-
-    data_fields: Annotated[List[str], pydantic.Field(alias="dataFields")]
-    r"""Fields with the cell values (i.e. aggregates)"""
-
-    qualifier_fields: Annotated[List[str], pydantic.Field(alias="qualifierFields")]
-    r"""Fields to qualify or group data fields"""
-
-
 class FunctionPivotTypedDict(TypedDict):
     filename: str
     group: str
@@ -46,7 +30,7 @@ class FunctionPivotTypedDict(TypedDict):
     disabled: NotRequired[bool]
     handle_signals: NotRequired[bool]
     sync: NotRequired[bool]
-    schema_: NotRequired[SimplePivotConfigurationTypedDict]
+    schema_: NotRequired[FunctionConfSchemaPivotTypedDict]
 
 
 class FunctionPivot(BaseModel):
@@ -81,5 +65,5 @@ class FunctionPivot(BaseModel):
     sync: Optional[bool] = None
 
     schema_: Annotated[
-        Optional[SimplePivotConfiguration], pydantic.Field(alias="schema")
+        Optional[FunctionConfSchemaPivot], pydantic.Field(alias="schema")
     ] = None