optim: init MlpRestClient

just-ai · Jun 10, 2024 · d73e84f · d73e84f
1 parent 3d9cf3b
commit d73e84f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlp_sdk/transport/MlpClientSDK.py b/mlp_sdk/transport/MlpClientSDK.py
@@ -189,7 +189,7 @@ def __connect(self):
             previous_channel.close()
 
 
-class MlpRestClient:
+class MlpRestClient(ApiClient):
 
     def __init__(self, url: Optional[str] = None, token=None, config=CONFIG):
         self.config = config
@@ -201,7 +201,7 @@ def __init__(self, url: Optional[str] = None, token=None, config=CONFIG):
         self.log.debug("Creating mpl client with url " + self.rest_url)
 
         configuration = Configuration(host=self.rest_url)
-        self.api_client = ApiClient(configuration, "MLP-API-KEY", self.client_token)
+        super().__init__(configuration, "MLP-API-KEY", self.client_token)
 
 
 class MlpClientException(Exception):