taegis-sdk-python 1.2.23__tar.gz → 1.2.25__tar.gz

{taegis_sdk_python-1.2.23/taegis_sdk_python.egg-info → taegis_sdk_python-1.2.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: taegis-sdk-python
-Version: 1.2.23
+Version: 1.2.25
 Summary: Taegis Python SDK
 Home-page: https://github.com/secureworks/taegis-sdk-python
 Author: Secureworks
@@ -84,7 +84,7 @@ pp(result)
 
 ```python
 results = service.events.subscription.event_query(
-    query="FROM process EARLIEST=-30d",
+    query="FROM process EARLIEST=-1d",
     options=EventQueryOptions(
         max_rows=20,
         page_size=10,

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/README.md

@@ -58,7 +58,7 @@ pp(result)
 
 ```python
 results = service.events.subscription.event_query(
-    query="FROM process EARLIEST=-30d",
+    query="FROM process EARLIEST=-1d",
     options=EventQueryOptions(
         max_rows=20,
         page_size=10,

taegis_sdk_python-1.2.25/taegis_sdk_python/_version.py

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

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/authentication.py

@@ -48,10 +48,10 @@ def check_username(
         JSON response from username check
     """
     logger.debug("Checking login type for username...")
-    username_endpoint = "/auth/username"
+    userlogin_endpoint = "/auth/userLogin"
 
     response = post(
-        f"{request_url}{username_endpoint}", json={"username": username}, timeout=300
+        f"{request_url}{userlogin_endpoint}", json={"username": username}, timeout=300
     )
     logger.debug(response)
 
@@ -118,13 +118,22 @@ def get_token(environment: str, request_url: str) -> str:  # pragma: no cover
                 else:
                     username = input("Username: ")
                     response = check_username(request_url, username)
+                    external_provider = response.get("external_provider")
 
-                    if response.get("login_type") == "username-password":
+                    if (
+                        response.get("login_type") == "username-password"
+                        and external_provider != "COGNITO"
+                    ):
                         access_token = get_token_by_password_grant(
                             request_url, username
                         )
-                    elif response.get("login_type") == "sso":
-                        access_token = get_token_by_sso_device_code(request_url)
+                    elif (
+                        response.get("login_type") == "sso"
+                        or external_provider == "COGNITO"
+                    ):
+                        access_token = get_token_by_device_code_authorization(
+                            request_url, external_provider
+                        )
                     else:
                         raise InvalidAuthenticationMethod(
                             message="No known authentication method for user"
@@ -262,8 +271,9 @@ def get_token_by_password_grant(
     return access_token
 
 
-def get_token_by_sso_device_code(
+def get_token_by_device_code_authorization(
     request_url: str,
+    external_provider: str,
 ) -> Union[str, None]:  # pragma: no cover
     """Get a user token via Device Code authorization.
 
@@ -277,10 +287,14 @@ def get_token_by_sso_device_code(
     Union[str, None]
         Access Token
     """
-    logger.debug("Trying by SSO device code auth url...")
+    logger.debug("Trying by device code auth url...")
     init_endpoint = "/universal-auth/device/code/auth"
     token_endpoint = "/universal-auth/device/code/token"
 
+    if external_provider == "COGNITO":
+        init_endpoint += "?version=2"
+        token_endpoint += "?version=2"
+
     response = post(f"{request_url}{init_endpoint}", timeout=300)
 
     try:

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/access_points/mutations.py

@@ -1,4 +1,5 @@
 """AccessPoints Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/access_points/queries.py

@@ -1,4 +1,5 @@
 """AccessPoints Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/access_points/subscriptions.py

@@ -1,4 +1,5 @@
 """AccessPoints Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/access_points/types.py

@@ -1,4 +1,5 @@
 """AccessPoints Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/agent/mutations.py

@@ -1,4 +1,5 @@
 """Agent Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/agent/queries.py

@@ -1,4 +1,5 @@
 """Agent Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/agent/subscriptions.py

@@ -1,4 +1,5 @@
 """Agent Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/agent/types.py

@@ -1,4 +1,5 @@
 """Agent Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/alerts/mutations.py

@@ -1,4 +1,5 @@
 """Alerts Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/alerts/queries.py

@@ -1,4 +1,5 @@
 """Alerts Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/alerts/subscriptions.py

@@ -1,4 +1,5 @@
 """Alerts Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/alerts/types.py

@@ -1,4 +1,5 @@
 """Alerts Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated
@@ -2006,15 +2007,15 @@ class ImprobableLogonDetailInput:
     source_address: Optional[str] = field(
         default=None, metadata=config(field_name="source_address")
     )
-    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",
-        ),
+    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")
@@ -2058,15 +2059,15 @@ class ImprobableLogonDetail:
     source_address: Optional[str] = field(
         default=None, metadata=config(field_name="source_address")
     )
-    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",
-        ),
+    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")
@@ -2088,15 +2089,17 @@ class AggregateAlertsBySeverityInputInput:
     tenant_service_filters: Optional[List[str]] = field(
         default=None, metadata=config(field_name="tenant_service_filters")
     )
-    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",
-        ),
+    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")
@@ -2422,10 +2425,11 @@ class EnrichmentDetail:
     business_email_compromise: Optional[BusinessEmailCompromiseDetail] = field(
         default=None, metadata=config(field_name="business_email_compromise")
     )
-    account_compromise_detector_detail: Optional[
-        AccountCompromiseDetectorDetail
-    ] = field(
-        default=None, metadata=config(field_name="account_compromise_detector_detail")
+    account_compromise_detector_detail: Optional[AccountCompromiseDetectorDetail] = (
+        field(
+            default=None,
+            metadata=config(field_name="account_compromise_detector_detail"),
+        )
     )
     file_analysis_detail: Optional[FileAnalysisDetail] = field(
         default=None, metadata=config(field_name="file_analysis_detail")

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets/mutations.py

@@ -1,4 +1,5 @@
 """Assets Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets/queries.py

@@ -1,4 +1,5 @@
 """Assets Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets/subscriptions.py

@@ -1,4 +1,5 @@
 """Assets Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets/types.py

@@ -1,4 +1,5 @@
 """Assets Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets2/mutations.py

@@ -1,4 +1,5 @@
 """Assets2 Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 
@@ -156,7 +157,8 @@ class TaegisSDKAssets2Mutation:
     ) -> BulkOpPayloadV2:
         """Start a job to assign the endpoints matching the filter criteria to the
         investigation in the input. Use the task ID in the response to poll the
-        assignBulkAssetsToInvestigationStatus query to determine if the job succeeded.."""
+        assignBulkAssetsToInvestigationStatus query to determine if the job succeeded..
+        """
         endpoint = "assignBulkAssetsToInvestigation"
 
         result = self.service.execute_mutation(
@@ -177,7 +179,8 @@ class TaegisSDKAssets2Mutation:
     ) -> BulkOpPayloadV2:
         """Start a job to delete the provided investigation from the endpoints matching
         the filter criteria in the input. Use the task ID in the response to poll the
-        bulkDeleteInvestigationForEndpointsStatus query to determine if the job succeeded.."""
+        bulkDeleteInvestigationForEndpointsStatus query to determine if the job succeeded..
+        """
         endpoint = "bulkDeleteInvestigationForEndpoints"
 
         result = self.service.execute_mutation(

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets2/queries.py

@@ -1,4 +1,5 @@
 """Assets2 Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets2/subscriptions.py

@@ -1,4 +1,5 @@
 """Assets2 Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/assets2/types.py

@@ -1,4 +1,5 @@
 """Assets2 Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated
@@ -201,7 +202,16 @@ class AutoArchive:
     """AutoArchive."""
 
     enabled: Optional[bool] = field(default=None, metadata=config(field_name="enabled"))
-    period: Optional[int] = field(default=None, metadata=config(field_name="period"))
+    period: Optional[int] = field(
+        default=None,
+        metadata=config(
+            metadata={"deprecated": True, "deprecation_reason": "use periodDays"},
+            field_name="period",
+        ),
+    )
+    period_days: Optional[int] = field(
+        default=None, metadata=config(field_name="periodDays")
+    )
 
 
 @dataclass_json

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/audits/mutations.py

@@ -1,4 +1,5 @@
 """Audits Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/audits/queries.py

@@ -1,4 +1,5 @@
 """Audits Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/audits/subscriptions.py

@@ -1,4 +1,5 @@
 """Audits Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/audits/types.py

@@ -1,4 +1,5 @@
 """Audits Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/authz/mutations.py

@@ -1,4 +1,5 @@
 """Authz Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/authz/queries.py

@@ -1,4 +1,5 @@
 """Authz Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/authz/subscriptions.py

@@ -1,4 +1,5 @@
 """Authz Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/authz/types.py

@@ -1,4 +1,5 @@
 """Authz Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/byoti/mutations.py

@@ -1,4 +1,5 @@
 """Byoti Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/byoti/queries.py

@@ -1,4 +1,5 @@
 """Byoti Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/byoti/subscriptions.py

@@ -1,4 +1,5 @@
 """Byoti Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/byoti/types.py

@@ -1,4 +1,5 @@
 """Byoti Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/clients/mutations.py

@@ -1,4 +1,5 @@
 """Clients Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/clients/queries.py

@@ -1,4 +1,5 @@
 """Clients Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/clients/subscriptions.py

@@ -1,4 +1,5 @@
 """Clients Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/clients/types.py

@@ -1,4 +1,5 @@
 """Clients Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/collector/mutations.py

@@ -1,4 +1,5 @@
 """Collector Mutation."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/collector/queries.py

@@ -1,4 +1,5 @@
 """Collector Query."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 
@@ -640,7 +641,8 @@ class TaegisSDKCollectorQuery:
         self, cluster_id: Optional[str] = None
     ) -> LogLastSeenMetrics:
         """Get last seen metrics for all available log sources for a given cluster.
-        If no clusterId is specified, this will return all log sources metrics for all existing clusters."""
+        If no clusterId is specified, this will return all log sources metrics for all existing clusters.
+        """
         endpoint = "getLogLastSeenMetrics"
 
         log.warning(
@@ -727,3 +729,18 @@ class TaegisSDKCollectorQuery:
         if result.get(endpoint) is not None:
             return result.get(endpoint)
         raise GraphQLNoRowsInResultSetError("for query getClustersPasswordHash")
+
+    def get_data_collectors(self, input_: QlQueryInput) -> DataCollectorResult:
+        """Searches and/or aggregates on Data collectors cql schema."""
+        endpoint = "getDataCollectors"
+
+        result = self.service.execute_query(
+            endpoint=endpoint,
+            variables={
+                "input": prepare_input(input_),
+            },
+            output=build_output_string(DataCollectorResult),
+        )
+        if result.get(endpoint) is not None:
+            return DataCollectorResult.from_dict(result.get(endpoint))
+        raise GraphQLNoRowsInResultSetError("for query getDataCollectors")

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/collector/subscriptions.py

@@ -1,4 +1,5 @@
 """Collector Subscription."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code, wildcard-import, unused-wildcard-import, cyclic-import
 
 

{taegis_sdk_python-1.2.23 → taegis_sdk_python-1.2.25}/taegis_sdk_python/services/collector/types.py

@@ -1,4 +1,5 @@
 """Collector Types and Enums."""
+
 # pylint: disable=no-member, unused-argument, too-many-locals, duplicate-code
 
 # Autogenerated
@@ -710,6 +711,36 @@ class SyslogMessageCountV2Arguments:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class QlQueryInput:
+    """QlQueryInput."""
+
+    cql_query: Optional[str] = field(
+        default=None, metadata=config(field_name="cqlQuery")
+    )
+    cursor: Optional[str] = field(default=None, metadata=config(field_name="cursor"))
+    count: Optional[int] = field(default=None, metadata=config(field_name="count"))
+
+
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class QLPageCursor:
+    """QLPageCursor."""
+
+    next_cursor: Optional[str] = field(
+        default=None, metadata=config(field_name="nextCursor")
+    )
+    previous_cursor: Optional[str] = field(
+        default=None, metadata=config(field_name="previousCursor")
+    )
+    cursor: Optional[str] = field(default=None, metadata=config(field_name="cursor"))
+    count: Optional[int] = field(default=None, metadata=config(field_name="count"))
+    total_count: Optional[int] = field(
+        default=None, metadata=config(field_name="totalCount")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class ScheduledService:
@@ -973,6 +1004,19 @@ class DataSourceMetrics:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class NodeInfo:
+    """NodeInfo."""
+
+    node_health: Optional[str] = field(
+        default=None, metadata=config(field_name="nodeHealth")
+    )
+    network: Optional[Network] = field(
+        default=None, metadata=config(field_name="network")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class ClusterNode:
@@ -1096,6 +1140,41 @@ class Deployment:
     )
 
 
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class DataCollector:
+    """DataCollector."""
+
+    id: Optional[str] = field(default=None, metadata=config(field_name="id"))
+    name: Optional[str] = field(default=None, metadata=config(field_name="name"))
+    type: Optional[str] = field(default=None, metadata=config(field_name="type"))
+    collector_health: Optional[str] = field(
+        default=None, metadata=config(field_name="collectorHealth")
+    )
+    is_ha: Optional[bool] = field(default=None, metadata=config(field_name="isHa"))
+    tenant: Optional[str] = field(default=None, metadata=config(field_name="tenant"))
+    nodes: Optional[NodeInfo] = field(default=None, metadata=config(field_name="nodes"))
+    applications: Optional[List[Deployment]] = field(
+        default=None, metadata=config(field_name="applications")
+    )
+
+
+@dataclass_json
+@dataclass(order=True, eq=True, frozen=True)
+class DataCollectorResult:
+    """DataCollectorResult."""
+
+    aggregation: Optional[List[dict]] = field(
+        default=None, metadata=config(field_name="aggregation")
+    )
+    data_collectors: Optional[List[DataCollector]] = field(
+        default=None, metadata=config(field_name="dataCollectors")
+    )
+    pagination: Optional[QLPageCursor] = field(
+        default=None, metadata=config(field_name="pagination")
+    )
+
+
 @dataclass_json
 @dataclass(order=True, eq=True, frozen=True)
 class ClusterImageInput:
@@ -1235,6 +1314,9 @@ class Cluster:
     ha_cidr_block: Optional[str] = field(
         default=None, metadata=config(field_name="haCidrBlock")
     )
+    partner_tenant_name: Optional[str] = field(
+        default=None, metadata=config(field_name="partnerTenantName")
+    )
     cluster_type: Optional[Union[ClusterType, TaegisEnum]] = field(
         default=None,
         metadata=config(