diff --git a/python/shroomdk/models/compass/core/rpc_response.py b/python/shroomdk/models/compass/core/rpc_response.py index 7b7030e..5a2dd54 100644 --- a/python/shroomdk/models/compass/core/rpc_response.py +++ b/python/shroomdk/models/compass/core/rpc_response.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, Optional +from typing import Any, Dict, Optional, Union from pydantic import BaseModel @@ -8,5 +8,5 @@ from .rpc_error import RpcError class RpcResponse(BaseModel): jsonrpc: str id: int - result: Optional[Dict[str, Any]] | None + result: Union[Optional[Dict[str, Any]], None] error: Optional[RpcError] diff --git a/python/shroomdk/models/compass/create_query_run.py b/python/shroomdk/models/compass/create_query_run.py index 5bc71d0..c11907f 100644 --- a/python/shroomdk/models/compass/create_query_run.py +++ b/python/shroomdk/models/compass/create_query_run.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List +from typing import Any, Dict, List, Union from pydantic import BaseModel @@ -33,4 +33,4 @@ class CreateQueryRunRpcResult(BaseModel): class CreateQueryRunRpcResponse(RpcResponse): - result: CreateQueryRunRpcResult | None + result: Union[CreateQueryRunRpcResult, None] diff --git a/python/shroomdk/models/compass/get_query_run.py b/python/shroomdk/models/compass/get_query_run.py index 530536c..7165da9 100644 --- a/python/shroomdk/models/compass/get_query_run.py +++ b/python/shroomdk/models/compass/get_query_run.py @@ -1,4 +1,4 @@ -from typing import Dict, List, Optional +from typing import Dict, List, Optional, Union from pydantic import BaseModel @@ -25,4 +25,4 @@ class GetQueryRunRpcResult(BaseModel): class GetQueryRunRpcResponse(RpcResponse): - result: GetQueryRunRpcResult | None + result: Union[GetQueryRunRpcResult, None] diff --git a/python/shroomdk/models/compass/get_query_run_results.py b/python/shroomdk/models/compass/get_query_run_results.py index cbbcb78..25c11e8 100644 --- a/python/shroomdk/models/compass/get_query_run_results.py +++ b/python/shroomdk/models/compass/get_query_run_results.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List, Optional +from typing import Any, Dict, List, Optional, Union from pydantic import BaseModel @@ -35,8 +35,8 @@ class SortBy(BaseModel): class GetQueryRunResultsRpcParams(BaseModel): queryRunId: str format: ResultFormat - filters: Optional[List[Filter | None]] = [] - sortBy: Optional[List[SortBy] | None] = [] + filters: Optional[Union[List[Filter], None]] = [] + sortBy: Optional[Union[List[SortBy], None]] = [] page: Page @@ -54,8 +54,8 @@ class GetQueryRunResultsRpcResult(BaseModel): sql: str format: ResultFormat originalQueryRun: QueryRun - redirectedToQueryRun: QueryRun | None + redirectedToQueryRun: Union[QueryRun, None] class GetQueryRunResultsRpcResponse(RpcResponse): - result: GetQueryRunResultsRpcResult | None + result: Union[GetQueryRunResultsRpcResult, None] diff --git a/python/shroomdk/tests/utils/mock_data/create_query_run.py b/python/shroomdk/tests/utils/mock_data/create_query_run.py index c81c3c4..e56456b 100644 --- a/python/shroomdk/tests/utils/mock_data/create_query_run.py +++ b/python/shroomdk/tests/utils/mock_data/create_query_run.py @@ -1,10 +1,12 @@ +from typing import Union + from shroomdk.models.compass.core.rpc_error import RpcError from shroomdk.models.query_status import QueryStatus def create_query_run_response( status: str = QueryStatus.Ready, - error: RpcError | None = None, + error: Union[RpcError, None] = None, result_null: bool = False, ): base = { diff --git a/python/shroomdk/tests/utils/mock_data/get_query_results.py b/python/shroomdk/tests/utils/mock_data/get_query_results.py index 6365488..7c066ac 100644 --- a/python/shroomdk/tests/utils/mock_data/get_query_results.py +++ b/python/shroomdk/tests/utils/mock_data/get_query_results.py @@ -1,9 +1,11 @@ +from typing import Union + from shroomdk.models.compass.core.rpc_error import RpcError from shroomdk.models.query_status import QueryStatus def get_query_results_response( - status: str = QueryStatus.Success, error: RpcError | None = None + status: str = QueryStatus.Success, error: Union[RpcError, None] = None ): base = { "jsonrpc": "2.0", diff --git a/python/shroomdk/tests/utils/mock_data/get_query_run.py b/python/shroomdk/tests/utils/mock_data/get_query_run.py index a251a6d..25ec4ae 100644 --- a/python/shroomdk/tests/utils/mock_data/get_query_run.py +++ b/python/shroomdk/tests/utils/mock_data/get_query_run.py @@ -1,9 +1,11 @@ +from typing import Union + from shroomdk.models.compass.core.rpc_error import RpcError from shroomdk.models.query_status import QueryStatus def get_query_run_response( - status: str = QueryStatus.Ready, error: RpcError | None = None + status: str = QueryStatus.Ready, error: Union[RpcError, None] = None ): base = { "jsonrpc": "2.0",