revert: legacy modelApi, processApi, datasetApi

just-ai · Jun 11, 2024 · 023b0ee · 023b0ee
1 parent c23b3fe
commit 023b0ee
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mlp_sdk/transport/MlpClientSDK.py b/mlp_sdk/transport/MlpClientSDK.py
@@ -7,6 +7,7 @@
 import yaml
 
 from mlp_api import ApiClient, Configuration
+from mlp_api.api import dataset_endpoint_api, model_endpoint_api, process_endpoint_api
 from mlp_sdk.grpc import mlp_grpc_pb2, mlp_grpc_pb2_grpc
 from mlp_sdk.log.setup_logging import get_logger
 from mlp_sdk.transport.MlpServiceSDK import MlpResponseHeaders
@@ -203,6 +204,10 @@ def __init__(self, url: Optional[str] = None, token: Optional[str] = None, confi
         configuration = Configuration(host=self.rest_url)
         super().__init__(configuration, "MLP-API-KEY", self.client_token)
 
+        self.modelApi = model_endpoint_api.ModelEndpointApi(self)
+        self.processApi = process_endpoint_api.ProcessEndpointApi(self)
+        self.datasetApi = dataset_endpoint_api.DatasetEndpointApi(self)
+
 
 class MlpClientException(Exception):
     def __init__(self, error_code: str, error_message: str, args: Dict[str, str]):