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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: taegis-sdk-python
-Version: 1.2.15
+Version: 1.2.17
 Summary: Taegis Python SDK
 Home-page: https://github.com/secureworks/taegis-sdk-python
 Author: Secureworks
@@ -22,6 +22,7 @@ Requires-Dist: pyjwt>=2.8.0
 Requires-Dist: requests-toolbelt
 Requires-Dist: websockets
 Requires-Dist: filelock>=3.12.4
+Requires-Dist: aiohttp
 
 # Taegis SDK for Python
 

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/requirements.txt

@@ -8,4 +8,5 @@ dataclasses-json
 pyjwt>=2.8.0
 requests-toolbelt
 websockets
-filelock>=3.12.4
+filelock>=3.12.4
+aiohttp

taegis_sdk_python-1.2.17/taegis_sdk_python/_version.py

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

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/service_core.py

@@ -6,17 +6,18 @@ Taegis ServiceCore manager.
 from __future__ import annotations
 
 import logging
+import threading
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
-import threading
+
+from graphql import GraphQLError, GraphQLField, GraphQLSchema
 
 from gql import Client, gql
 from gql.transport import Transport
-from gql.transport.requests import RequestsHTTPTransport
-from gql.transport.websockets import WebsocketsTransport
-from graphql import GraphQLError, GraphQLField, GraphQLSchema
+from gql.transport.aiohttp import AIOHTTPTransport
 from taegis_sdk_python._version import __version__
 from taegis_sdk_python.errors import InvalidGraphQLEndpoint
+from taegis_sdk_python.transport.aiohttp_websockets import AIOHTTPWebsocketsTransport
 from taegis_sdk_python.utils import async_block, prepare_variables, remove_node
 
 if TYPE_CHECKING:  # pragma: no cover
@@ -129,7 +130,8 @@ class SchemaCacheMap:
                     del self._map[transport.url]
 
     @staticmethod
-    def retrieve_schema(
+    @async_block
+    async def retrieve_schema(
         transport: Transport, introspection_args: Dict[str, Any]
     ) -> GraphQLSchema:
         """Retrieve the schema from a gql Client with the provided transport and
@@ -152,7 +154,7 @@ class SchemaCacheMap:
             introspection_args=introspection_args,
         )
 
-        with client:
+        async with client:
             schema = client.schema
 
         return schema
@@ -207,9 +209,13 @@ class ServiceCore:
     @property
     def sync_client(self) -> Client:
         """GraphQL Synchronous Transport with Client."""
-        transport = RequestsHTTPTransport(
+        client_session_args = {"trust_env": self.service.trust_env}
+
+        transport = AIOHTTPTransport(
             f"{self.sync_url}{self.gateway}",
             headers=self.service.headers,
+            ssl=self.service.ssl,
+            client_session_args=client_session_args,
         )
 
         client = Client(
@@ -239,11 +245,17 @@ class ServiceCore:
         if self.service.tenant_id:
             subprotocols.append(f"x-tenant-context-{self.service.tenant_id}")
 
-        transport = WebsocketsTransport(
+        client_session_args = {"trust_env": self.service.trust_env}
+
+        transport = AIOHTTPWebsocketsTransport(
             f"{self.wss_url}{self.gateway}",
             headers=self.service.headers,
             subprotocols=subprotocols,
-            connect_args={"max_size": None},
+            ssl=self.service.ssl,
+            client_session_args=client_session_args,
+            proxy=self.service.proxy,
+            proxy_auth=self.service.proxy_auth,
+            proxy_headers=self.service.proxy_headers,
         )
 
         client = Client(
@@ -448,7 +460,8 @@ class ServiceCore:
 
         return self.subscribe(query_string, variables)
 
-    def execute(
+    @async_block
+    async def execute(
         self, query_string: str, variables: Optional[Dict[str, Any]] = None
     ) -> Any:
         """Execute a GraphQL string.
@@ -465,10 +478,19 @@ class ServiceCore:
         Any
             Query/Mutation results
         """
+        extra_args = {}
+        if self.service.proxy:
+            extra_args = {
+                "proxy": self.service.proxy,
+                "proxy_auth": self.service.proxy_auth,
+                "proxy_headers": self.service.proxy_headers,
+            }
+
         query = gql(query_string)
-        return self.sync_client.execute(
+        return await self.sync_client.execute_async(
             query,
             variable_values=prepare_variables(variables),
+            extra_args=extra_args,
         )
 
     @async_block
@@ -500,7 +522,8 @@ class ServiceCore:
 
         return results[:-1]
 
-    def _build_validated_query(
+    @async_block
+    async def _build_validated_query(
         self,
         operation_type: str,
         endpoint: str,
@@ -537,7 +560,7 @@ class ServiceCore:
         query_string = " ".join(query_string.split())
 
         # open a connection for introspection and download schema
-        with self.sync_client as session:
+        async with self.sync_client as session:
             # if multiple fields are invalid, we want to iterate until the
             # output string is valid
             for _ in range(10000):

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/__init__.py

@@ -2,8 +2,14 @@
 This needs to be a generated file.  Need to make jinja template.
 """
 
-from typing import Dict, Optional, Any
+import logging
 import threading
+from ssl import SSLContext
+from typing import Any, Dict, Optional, Union, Literal
+
+import aiohttp
+from aiohttp.client_reqrep import Fingerprint
+from aiohttp.typedefs import LooseHeaders
 
 from taegis_sdk_python._consts import TAEGIS_ENVIRONMENT_URLS
 from taegis_sdk_python._version import __version__
@@ -60,6 +66,8 @@ from taegis_sdk_python.services.vdr import VDRService
 
 __all__ = ["GraphQLService"]
 
+log = logging.getLogger(__name__)
+
 
 class GraphQLService:
     """Taegis GraphQL Service manager."""
@@ -73,6 +81,11 @@ class GraphQLService:
         gateway: Optional[str] = None,
         extra_headers: Optional[Dict[str, Any]] = None,
         schema_expiry: int = 5,
+        proxy: Optional[str] = None,
+        proxy_auth: Optional[aiohttp.BasicAuth] = None,
+        proxy_headers: Optional[LooseHeaders] = None,
+        trust_env: bool = False,
+        ssl: Optional[Union[SSLContext, Literal[False], Fingerprint]] = None,
     ):  # pylint: disable=too-many-statements
         """
         GraphQLService
@@ -114,6 +127,18 @@ class GraphQLService:
         self._schema_expiry = schema_expiry
         self._input_value_deprecation = None
 
+        if trust_env and proxy:
+            raise ValueError("trust_env and proxy cannot be used together")
+
+        if proxy_auth and not proxy:
+            raise ValueError("proxy_auth requires proxy to be set")
+
+        self._proxy = proxy
+        self._proxy_auth = proxy_auth
+        self._proxy_headers = proxy_headers
+        self._ssl = ssl
+        self._trust_env = trust_env
+
         self._access_points = None
         self._agent = None
         self._alerts = None
@@ -270,6 +295,49 @@ class GraphQLService:
             "input_value_deprecation", self._input_value_deprecation
         )
 
+    @property
+    def proxy(self):
+        """Proxy URL for API requests."""
+        return self._context_manager.get("proxy", self._proxy)
+
+    @property
+    def proxy_auth(self):
+        """Proxy Auth for API requests."""
+        value = self._context_manager.get("proxy_auth", self._proxy_auth)
+
+        if value and not self.proxy:
+            log.warning("proxy_auth is ignored when proxy is not set")
+            return None
+
+        return value
+
+    @property
+    def proxy_headers(self):
+        """Proxy Headers for API requests."""
+        value = self._context_manager.get("proxy_headers", self._proxy_headers)
+
+        if value and not self.proxy:
+            log.warning("proxy_headers is ignored when proxy is not set")
+            return None
+
+        return value
+
+    @property
+    def ssl(self):
+        """SSL Context for API requests."""
+        return self._context_manager.get("ssl", self._ssl)
+
+    @property
+    def trust_env(self):
+        """Trust Environment for Proxy Authentication."""
+        value = self._context_manager.get("trust_env", self._trust_env)
+
+        if value and self.proxy:
+            log.warning("trust_env is ignored when proxy is set")
+            return False
+
+        return value
+
     @property
     def access_points(self):
         """Access Points Service Endpoint."""

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/access_points/mutations.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.access_points.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.access_points.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.access_points import AccessPointsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/access_points/queries.py

@@ -11,16 +11,18 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.access_points.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.access_points.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.access_points import AccessPointsService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/access_points/subscriptions.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.access_points.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.access_points.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.access_points import AccessPointsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/agent/mutations.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.agent.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.agent.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.agent import AgentService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/agent/queries.py

@@ -11,16 +11,18 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.agent.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.agent.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.agent import AgentService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/agent/subscriptions.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.agent.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.agent.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.agent import AgentService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/alerts/mutations.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.alerts.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.alerts.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.alerts import AlertsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/alerts/queries.py

@@ -11,16 +11,18 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.alerts.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.alerts.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.alerts import AlertsService
 
+
 log = logging.getLogger(__name__)
 
 

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/alerts/subscriptions.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.alerts.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.alerts.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.alerts import AlertsService

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

@@ -2104,11 +2104,11 @@ class AggregateAlertsBySeverityInputInput:
     latest: Optional[TimestampInput] = field(
         default=None, metadata=config(field_name="latest")
     )
-    excluded_severities: Optional[Union[List[AlertsSeverity], TaegisEnum]] = field(
+    excluded_severities: Optional[List[Union[AlertsSeverity, TaegisEnum]]] = field(
         default=None,
         metadata=config(
             encoder=encode_enum,
-            decoder=lambda x: decode_enum(List[AlertsSeverity], x),
+            decoder=lambda x: decode_enum(AlertsSeverity, x),
             field_name="excluded_severities",
         ),
     )

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/assets/mutations.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.assets.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets import AssetsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/assets/queries.py

@@ -11,16 +11,18 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.assets.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets import AssetsService
 
+
 log = logging.getLogger(__name__)
 
 
@@ -113,9 +115,9 @@ class TaegisSDKAssetsQuery:
         self,
         offset: Optional[int] = None,
         limit: Optional[int] = None,
-        order_by: Optional[AssetsOrderByInput] = None,
-        order_direction: Optional[AssetsOrderDirectionInput] = None,
-        filter_asset_state: Optional[AssetStateFilter] = None,
+        order_by: Optional[Union[AssetsOrderByInput, TaegisEnum]] = None,
+        order_direction: Optional[Union[AssetsOrderDirectionInput, TaegisEnum]] = None,
+        filter_asset_state: Optional[Union[AssetStateFilter, TaegisEnum]] = None,
         only_most_recent: Optional[bool] = None,
     ) -> AssetsResult:
         """Get a list of assets."""
@@ -159,7 +161,9 @@ class TaegisSDKAssetsQuery:
             return AssetsResult.from_dict(result.get(endpoint))
         raise GraphQLNoRowsInResultSetError("for query allAssetsExport")
 
-    def asset_count(self, endpoint_type: Optional[AgentType] = None) -> AssetCounts:
+    def asset_count(
+        self, endpoint_type: Optional[Union[AgentType, TaegisEnum]] = None
+    ) -> AssetCounts:
         """Count of assets of a specific endpoint_type."""
         endpoint = "assetCount"
 
@@ -323,10 +327,10 @@ class TaegisSDKAssetsQuery:
         tag: Optional[str] = None,
         host_id_partial_match: Optional[bool] = None,
         only_most_recent: Optional[bool] = None,
-        order_by: Optional[AssetsOrderByInput] = None,
-        order_direction: Optional[AssetsOrderDirectionInput] = None,
+        order_by: Optional[Union[AssetsOrderByInput, TaegisEnum]] = None,
+        order_direction: Optional[Union[AssetsOrderDirectionInput, TaegisEnum]] = None,
         or_search: Optional[bool] = None,
-        filter_asset_state: Optional[AssetStateFilter] = None,
+        filter_asset_state: Optional[Union[AssetStateFilter, TaegisEnum]] = None,
     ) -> AssetsResult:
         """search assets. Soon to be deprecated."""
         endpoint = "searchAssets"

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/assets/subscriptions.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.assets.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets import AssetsService

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/assets2/mutations.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.assets2.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets2.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets2 import Assets2Service

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/assets2/queries.py

@@ -11,16 +11,18 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.assets2.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets2.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets2 import Assets2Service
 
+
 log = logging.getLogger(__name__)
 
 
@@ -31,7 +33,7 @@ class TaegisSDKAssets2Query:
         self.service = service
 
     def facets_v2(
-        self, endpoint_type: Optional[List[EndpointTypeV2]] = None
+        self, endpoint_type: Optional[List[Union[EndpointTypeV2, TaegisEnum]]] = None
     ) -> List[FacetV2]:
         """Retrieve a list of facets for a given endpoint type.."""
         endpoint = "facetsV2"
@@ -52,7 +54,7 @@ class TaegisSDKAssets2Query:
     def facet_info_v2(
         self,
         facets: List[str],
-        order_by: Optional[FacetInfoOrderByInputV2] = None,
+        order_by: Optional[Union[FacetInfoOrderByInputV2, TaegisEnum]] = None,
         filter_: Optional[AssetFilter] = None,
     ) -> List[FacetInfoV2]:
         """Get facet info based on a currently selected facet.."""
@@ -80,7 +82,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetFilter] = None,
-        order_by: Optional[AssetSearchOrderByInputV2] = None,
+        order_by: Optional[Union[AssetSearchOrderByInputV2, TaegisEnum]] = None,
     ) -> AssetsV2:
         """Retrieve assets.."""
         endpoint = "assetsV2"
@@ -106,7 +108,7 @@ class TaegisSDKAssets2Query:
         first: Optional[int] = None,
         after: Optional[str] = None,
         filter_: Optional[AssetFilter] = None,
-        order_by: Optional[AssetSearchOrderByInputV2] = None,
+        order_by: Optional[Union[AssetSearchOrderByInputV2, TaegisEnum]] = None,
     ) -> AssetsExportOutputV2:
         """Retrieve assets in a form coercible into CSV files.."""
         endpoint = "exportAssetsV2"
@@ -342,7 +344,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetHostnameFilter] = None,
-        order_by: Optional[AssetHostnameSearchOrderByInput] = None,
+        order_by: Optional[Union[AssetHostnameSearchOrderByInput, TaegisEnum]] = None,
     ) -> AssetsHostname:
         """ "Retrieve hostname details."""
         endpoint = "assetsHostname"
@@ -370,7 +372,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetIPAddressFilter] = None,
-        order_by: Optional[AssetIPAddressSearchOrderByInput] = None,
+        order_by: Optional[Union[AssetIPAddressSearchOrderByInput, TaegisEnum]] = None,
     ) -> AssetsIPAddress:
         """ "Retrieve ipAddress details."""
         endpoint = "assetsIPAddress"
@@ -398,7 +400,7 @@ class TaegisSDKAssets2Query:
         after: Optional[str] = None,
         before: Optional[str] = None,
         filter_: Optional[AssetMacAddressFilter] = None,
-        order_by: Optional[AssetMacAddressSearchOrderByInput] = None,
+        order_by: Optional[Union[AssetMacAddressSearchOrderByInput, TaegisEnum]] = None,
     ) -> AssetsMacAddress:
         """ "Retrieve Mac details."""
         endpoint = "assetsMacAddress"

{taegis_sdk_python-1.2.15 → taegis_sdk_python-1.2.17}/taegis_sdk_python/services/assets2/subscriptions.py

@@ -11,12 +11,13 @@ import logging
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
 
 from taegis_sdk_python import GraphQLNoRowsInResultSetError
+from taegis_sdk_python._consts import TaegisEnum
+from taegis_sdk_python.services.assets2.types import *
 from taegis_sdk_python.utils import (
     build_output_string,
     parse_union_result,
     prepare_input,
 )
-from taegis_sdk_python.services.assets2.types import *
 
 if TYPE_CHECKING:  # pragma: no cover
     from taegis_sdk_python.services.assets2 import Assets2Service