taegis-sdk-python 1.2.14__tar.gz → 1.2.15__tar.gz

{taegis_sdk_python-1.2.14/taegis_sdk_python.egg-info → taegis_sdk_python-1.2.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: taegis-sdk-python
-Version: 1.2.14
+Version: 1.2.15
 Summary: Taegis Python SDK
 Home-page: https://github.com/secureworks/taegis-sdk-python
 Author: Secureworks

{taegis_sdk_python-1.2.14 → taegis_sdk_python-1.2.15}/taegis_sdk_python/_consts.py

@@ -1,5 +1,7 @@
 """Taegis Constants."""
 
+from enum import Enum
+
 TAEGIS_ENVIRONMENT_URLS = {
     "charlie": "https://api.ctpx.secureworks.com",
     "production": "https://api.ctpx.secureworks.com",
@@ -24,3 +26,9 @@ TAEGIS_TENANTS_URLS = {
     "foxtrot": "https://api-tenants.foxtrot.taegis.secureworks.com",
     "US3": "https://api-tenants.foxtrot.taegis.secureworks.com",
 }
+
+
+class TaegisEnum(str, Enum):
+    """Handler for Unknown Enums."""
+
+    UNKNOWN = "TAEGIS_SDK_PYTHON_UNKNOWN"

taegis_sdk_python-1.2.15/taegis_sdk_python/_version.py

@@ -0,0 +1 @@
+__version__ = "v1.2.15"

{taegis_sdk_python-1.2.14 → taegis_sdk_python-1.2.15}/taegis_sdk_python/services/access_points/types.py

@@ -4,11 +4,11 @@
 # Autogenerated
 # DO NOT MODIFY
 
-from typing import Optional, List, Dict, Union, Any, Tuple
+from dataclasses import dataclass, field
 
+from typing import Any, Dict, List, Optional, Tuple, Union
 
-from dataclasses import dataclass, field
-from dataclasses_json import dataclass_json, config
+from dataclasses_json import config, dataclass_json
 
 
 @dataclass_json

{taegis_sdk_python-1.2.14 → taegis_sdk_python-1.2.15}/taegis_sdk_python/services/agent/types.py

@@ -4,14 +4,17 @@
 # Autogenerated
 # DO NOT MODIFY
 
-from typing import Optional, List, Dict, Union, Any, Tuple
-
+from dataclasses import dataclass, field
 
 from enum import Enum
 
+from typing import Any, Dict, List, Optional, Tuple, Union
 
-from dataclasses import dataclass, field
-from dataclasses_json import dataclass_json, config
+from dataclasses_json import config, dataclass_json
+
+
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.utils import encode_enum, decode_enum
 
 
 class AgentArch(str, Enum):
@@ -110,14 +113,29 @@ class PackageDownloadInput:
     language: Optional[str] = field(
         default=None, metadata=config(field_name="language")
     )
-    platform: Optional[AgentPlatform] = field(
-        default=None, metadata=config(field_name="platform")
+    platform: Optional[Union[AgentPlatform, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentPlatform, x),
+            field_name="platform",
+        ),
     )
-    architecture: Optional[AgentArch] = field(
-        default=None, metadata=config(field_name="architecture")
+    architecture: Optional[Union[AgentArch, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentArch, x),
+            field_name="architecture",
+        ),
     )
-    pkg_type: Optional[AgentPkgType] = field(
-        default=None, metadata=config(field_name="pkgType")
+    pkg_type: Optional[Union[AgentPkgType, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentPkgType, x),
+            field_name="pkgType",
+        ),
     )
 
 
@@ -139,14 +157,29 @@ class Package:
     supported_languages: Optional[List[str]] = field(
         default=None, metadata=config(field_name="supportedLanguages")
     )
-    platform: Optional[AgentPlatform] = field(
-        default=None, metadata=config(field_name="platform")
-    )
-    pkg_type: Optional[AgentPkgType] = field(
-        default=None, metadata=config(field_name="pkgType")
-    )
-    architecture: Optional[AgentArch] = field(
-        default=None, metadata=config(field_name="architecture")
+    platform: Optional[Union[AgentPlatform, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentPlatform, x),
+            field_name="platform",
+        ),
+    )
+    pkg_type: Optional[Union[AgentPkgType, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentPkgType, x),
+            field_name="pkgType",
+        ),
+    )
+    architecture: Optional[Union[AgentArch, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentArch, x),
+            field_name="architecture",
+        ),
     )
     meta_data: Optional[List[MetaData]] = field(
         default=None, metadata=config(field_name="metaData")
@@ -165,15 +198,35 @@ class PackageSearchInput:
     language: Optional[str] = field(
         default=None, metadata=config(field_name="language")
     )
-    platform: Optional[AgentPlatform] = field(
-        default=None, metadata=config(field_name="platform")
-    )
-    architecture: Optional[AgentArch] = field(
-        default=None, metadata=config(field_name="architecture")
-    )
-    pkg_type: Optional[AgentPkgType] = field(
-        default=None, metadata=config(field_name="pkgType")
-    )
-    channel: Optional[AgentChannel] = field(
-        default=None, metadata=config(field_name="channel")
+    platform: Optional[Union[AgentPlatform, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentPlatform, x),
+            field_name="platform",
+        ),
+    )
+    architecture: Optional[Union[AgentArch, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentArch, x),
+            field_name="architecture",
+        ),
+    )
+    pkg_type: Optional[Union[AgentPkgType, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentPkgType, x),
+            field_name="pkgType",
+        ),
+    )
+    channel: Optional[Union[AgentChannel, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AgentChannel, x),
+            field_name="channel",
+        ),
     )

{taegis_sdk_python-1.2.14 → taegis_sdk_python-1.2.15}/taegis_sdk_python/services/alerts/types.py

@@ -4,14 +4,17 @@
 # Autogenerated
 # DO NOT MODIFY
 
-from typing import Optional, List, Dict, Union, Any, Tuple
-
+from dataclasses import dataclass, field
 
 from enum import Enum
 
+from typing import Any, Dict, List, Optional, Tuple, Union
 
-from dataclasses import dataclass, field
-from dataclasses_json import dataclass_json, config
+from dataclasses_json import config, dataclass_json
+
+
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.utils import encode_enum, decode_enum
 
 
 class AggregateAlertsBySeverityInput_GroupBy(str, Enum):
@@ -1319,8 +1322,13 @@ class BulkResolutionResponse:
     is_complete: Optional[bool] = field(
         default=None, metadata=config(field_name="is_complete")
     )
-    resolution_status: Optional[ResponseStatus] = field(
-        default=None, metadata=config(field_name="resolution_status")
+    resolution_status: Optional[Union[ResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResponseStatus, x),
+            field_name="resolution_status",
+        ),
     )
 
 
@@ -1331,8 +1339,13 @@ class CreateAlertsResponse:
 
     i_ds: Optional[List[str]] = field(default=None, metadata=config(field_name="iDs"))
     reason: Optional[str] = field(default=None, metadata=config(field_name="reason"))
-    status: Optional[RPCResponseStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[RPCResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(RPCResponseStatus, x),
+            field_name="status",
+        ),
     )
 
 
@@ -1342,8 +1355,13 @@ class DeleteAlertsResponse:
     """DeleteAlertsResponse."""
 
     reason: Optional[str] = field(default=None, metadata=config(field_name="reason"))
-    status: Optional[RPCResponseStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[RPCResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(RPCResponseStatus, x),
+            field_name="status",
+        ),
     )
 
 
@@ -1352,8 +1370,13 @@ class DeleteAlertsResponse:
 class EvictResponse:
     """EvictResponse."""
 
-    status: Optional[ResponseStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[ResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResponseStatus, x),
+            field_name="status",
+        ),
     )
 
 
@@ -1377,8 +1400,13 @@ class UpdateResolutionResponse:
     """UpdateResolutionResponse."""
 
     reason: Optional[str] = field(default=None, metadata=config(field_name="reason"))
-    resolution_status: Optional[ResponseStatus] = field(
-        default=None, metadata=config(field_name="resolution_status")
+    resolution_status: Optional[Union[ResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResponseStatus, x),
+            field_name="resolution_status",
+        ),
     )
 
 
@@ -1387,8 +1415,13 @@ class UpdateResolutionResponse:
 class UpdateThreatScoreResponse:
     """UpdateThreatScoreResponse."""
 
-    status: Optional[ResponseStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[ResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResponseStatus, x),
+            field_name="status",
+        ),
     )
 
 
@@ -1582,8 +1615,13 @@ class ResolutionMetadataInput:
     timestamp: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="timestamp")
     )
-    status: Optional[ResolutionStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[ResolutionStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResolutionStatus, x),
+            field_name="status",
+        ),
     )
 
 
@@ -1656,8 +1694,13 @@ class ResolutionMetadata:
     timestamp: Optional[Timestamp] = field(
         default=None, metadata=config(field_name="timestamp")
     )
-    status: Optional[ResolutionStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[ResolutionStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResolutionStatus, x),
+            field_name="status",
+        ),
     )
 
 
@@ -1676,8 +1719,13 @@ class TuningUpdate:
     suppressed_value: Optional[bool] = field(
         default=None, metadata=config(field_name="suppressed_value")
     )
-    origin_value: Optional[Origin] = field(
-        default=None, metadata=config(field_name="origin_value")
+    origin_value: Optional[Union[Origin, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(Origin, x),
+            field_name="origin_value",
+        ),
     )
     changed_at: Optional[Timestamp] = field(
         default=None, metadata=config(field_name="changed_at")
@@ -1852,11 +1900,21 @@ class BulkResolutionRequestInput:
     reason: Optional[str] = field(default=None, metadata=config(field_name="reason"))
     user_id: Optional[str] = field(default=None, metadata=config(field_name="user_id"))
     tenant: Optional[str] = field(default=None, metadata=config(field_name="tenant"))
-    resolution_status: Optional[ResolutionStatus] = field(
-        default=None, metadata=config(field_name="resolution_status")
+    resolution_status: Optional[Union[ResolutionStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResolutionStatus, x),
+            field_name="resolution_status",
+        ),
     )
-    caller: Optional[CallerInformation] = field(
-        default=None, metadata=config(field_name="caller")
+    caller: Optional[Union[CallerInformation, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(CallerInformation, x),
+            field_name="caller",
+        ),
     )
     requested_at: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="requested_at")
@@ -1886,11 +1944,21 @@ class UpdateInvestigationRequestInput:
     )
     tenant: Optional[str] = field(default=None, metadata=config(field_name="tenant"))
     user_id: Optional[str] = field(default=None, metadata=config(field_name="user_id"))
-    operation: Optional[InvestigationOperation] = field(
-        default=None, metadata=config(field_name="operation")
+    operation: Optional[Union[InvestigationOperation, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(InvestigationOperation, x),
+            field_name="operation",
+        ),
     )
-    caller: Optional[CallerInformation] = field(
-        default=None, metadata=config(field_name="caller")
+    caller: Optional[Union[CallerInformation, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(CallerInformation, x),
+            field_name="caller",
+        ),
     )
     requested_at: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="requested_at")
@@ -1908,11 +1976,21 @@ class UpdateResolutionRequestInput:
     reason: Optional[str] = field(default=None, metadata=config(field_name="reason"))
     user_id: Optional[str] = field(default=None, metadata=config(field_name="user_id"))
     tenant: Optional[str] = field(default=None, metadata=config(field_name="tenant"))
-    resolution_status: Optional[ResolutionStatus] = field(
-        default=None, metadata=config(field_name="resolution_status")
+    resolution_status: Optional[Union[ResolutionStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResolutionStatus, x),
+            field_name="resolution_status",
+        ),
     )
-    caller: Optional[CallerInformation] = field(
-        default=None, metadata=config(field_name="caller")
+    caller: Optional[Union[CallerInformation, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(CallerInformation, x),
+            field_name="caller",
+        ),
     )
     requested_at: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="requested_at")
@@ -1928,8 +2006,15 @@ class ImprobableLogonDetailInput:
     source_address: Optional[str] = field(
         default=None, metadata=config(field_name="source_address")
     )
-    feature_name: Optional[ImprobableLogonDetail_FeatureName] = field(
-        default=None, metadata=config(field_name="feature_name")
+    feature_name: Optional[
+        Union[ImprobableLogonDetail_FeatureName, TaegisEnum]
+    ] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ImprobableLogonDetail_FeatureName, x),
+            field_name="feature_name",
+        ),
     )
     logon_anomaly: Optional[LogonAnomalyInput] = field(
         default=None, metadata=config(field_name="logon_anomaly")
@@ -1973,8 +2058,15 @@ class ImprobableLogonDetail:
     source_address: Optional[str] = field(
         default=None, metadata=config(field_name="source_address")
     )
-    feature_name: Optional[ImprobableLogonDetail_FeatureName] = field(
-        default=None, metadata=config(field_name="feature_name")
+    feature_name: Optional[
+        Union[ImprobableLogonDetail_FeatureName, TaegisEnum]
+    ] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ImprobableLogonDetail_FeatureName, x),
+            field_name="feature_name",
+        ),
     )
     logon_anomaly: Optional[LogonAnomaly] = field(
         default=None, metadata=config(field_name="logon_anomaly")
@@ -1996,8 +2088,15 @@ class AggregateAlertsBySeverityInputInput:
     tenant_service_filters: Optional[List[str]] = field(
         default=None, metadata=config(field_name="tenant_service_filters")
     )
-    group_by: Optional[AggregateAlertsBySeverityInput_GroupBy] = field(
-        default=None, metadata=config(field_name="group_by")
+    group_by: Optional[
+        Union[AggregateAlertsBySeverityInput_GroupBy, TaegisEnum]
+    ] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AggregateAlertsBySeverityInput_GroupBy, x),
+            field_name="group_by",
+        ),
     )
     earliest: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="earliest")
@@ -2005,8 +2104,13 @@ class AggregateAlertsBySeverityInputInput:
     latest: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="latest")
     )
-    excluded_severities: Optional[List[AlertsSeverity]] = field(
-        default=None, metadata=config(field_name="excluded_severities")
+    excluded_severities: Optional[Union[List[AlertsSeverity], TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(List[AlertsSeverity], x),
+            field_name="excluded_severities",
+        ),
     )
 
 
@@ -2059,7 +2163,14 @@ class AlertsMetadata:
     first_resolved_at: Optional[Timestamp] = field(
         default=None, metadata=config(field_name="first_resolved_at")
     )
-    origin: Optional[Origin] = field(default=None, metadata=config(field_name="origin"))
+    origin: Optional[Union[Origin, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(Origin, x),
+            field_name="origin",
+        ),
+    )
 
 
 @dataclass_json
@@ -2158,8 +2269,13 @@ class BulkInvestigationsResponse:
     assets: Optional[List[str]] = field(
         default=None, metadata=config(field_name="assets")
     )
-    status: Optional[ResponseStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[ResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResponseStatus, x),
+            field_name="status",
+        ),
     )
     access_vector_info: Optional[List[AlertsInvestigationInfo]] = field(
         default=None,
@@ -2186,8 +2302,13 @@ class UpdateInvestigationResponse:
     assets: Optional[List[str]] = field(
         default=None, metadata=config(field_name="assets")
     )
-    status: Optional[ResponseStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[ResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResponseStatus, x),
+            field_name="status",
+        ),
     )
     access_vector_info: Optional[List[AlertsInvestigationInfo]] = field(
         default=None,
@@ -2352,8 +2473,13 @@ class Alert2:
     metadata: Optional[AlertsMetadata] = field(
         default=None, metadata=config(field_name="metadata")
     )
-    visibility: Optional[Visibility] = field(
-        default=None, metadata=config(field_name="visibility")
+    visibility: Optional[Union[Visibility, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(Visibility, x),
+            field_name="visibility",
+        ),
     )
     tenant: Optional[TenantV4] = field(
         default=None, metadata=config(field_name="tenant")
@@ -2364,8 +2490,13 @@ class Alert2:
     alerting_rules: Optional[List[AlertRuleReference]] = field(
         default=None, metadata=config(field_name="alerting_rules")
     )
-    status: Optional[ResolutionStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[ResolutionStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(ResolutionStatus, x),
+            field_name="status",
+        ),
     )
     resolution_history: Optional[List[ResolutionMetadata]] = field(
         default=None, metadata=config(field_name="resolution_history")
@@ -2453,8 +2584,13 @@ class AlertsResponse:
         default=None, metadata=config(field_name="search_id")
     )
     query_id: Optional[str] = field(default=None, metadata=config(field_name="queryId"))
-    status: Optional[RPCResponseStatus] = field(
-        default=None, metadata=config(field_name="status")
+    status: Optional[Union[RPCResponseStatus, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(RPCResponseStatus, x),
+            field_name="status",
+        ),
     )
     alerts: Optional[AlertsList] = field(
         default=None, metadata=config(field_name="alerts")

{taegis_sdk_python-1.2.14 → taegis_sdk_python-1.2.15}/taegis_sdk_python/services/assets/types.py

@@ -4,14 +4,17 @@
 # Autogenerated
 # DO NOT MODIFY
 
-from typing import Optional, List, Dict, Union, Any, Tuple
-
+from dataclasses import dataclass, field
 
 from enum import Enum
 
+from typing import Any, Dict, List, Optional, Tuple, Union
 
-from dataclasses import dataclass, field
-from dataclasses_json import dataclass_json, config
+from dataclasses_json import config, dataclass_json
+
+
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.utils import encode_enum, decode_enum
 
 
 class AgentType(str, Enum):
@@ -419,8 +422,13 @@ class SearchAssetsInput:
     investigation_id: Optional[str] = field(
         default=None, metadata=config(field_name="investigation_id")
     )
-    filter_asset_state: Optional[AssetStateFilter] = field(
-        default=None, metadata=config(field_name="filter_asset_state")
+    filter_asset_state: Optional[Union[AssetStateFilter, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AssetStateFilter, x),
+            field_name="filter_asset_state",
+        ),
     )
 
 
@@ -431,11 +439,21 @@ class SearchAssetsPaginationInput:
 
     offset: Optional[int] = field(default=None, metadata=config(field_name="offset"))
     limit: Optional[int] = field(default=None, metadata=config(field_name="limit"))
-    order_by: Optional[AssetsOrderByInput] = field(
-        default=None, metadata=config(field_name="order_by")
-    )
-    order_direction: Optional[AssetsOrderDirectionInput] = field(
-        default=None, metadata=config(field_name="order_direction")
+    order_by: Optional[Union[AssetsOrderByInput, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AssetsOrderByInput, x),
+            field_name="order_by",
+        ),
+    )
+    order_direction: Optional[Union[AssetsOrderDirectionInput, TaegisEnum]] = field(
+        default=None,
+        metadata=config(
+            encoder=encode_enum,
+            decoder=lambda x: decode_enum(AssetsOrderDirectionInput, x),
+            field_name="order_direction",
+        ),
     )