Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Medium

Please provide a clear description of problem this feature solves

Create a TAO REST API client wrapper to communicate with TAO API toolkit server below list of endpoints.

• List datasets for user
• Create a new dataset
• Delete a dataset
• Retrieve a dataset
• Update dataset metadata
• Retrieve default action specs
• Retrieve dataset action specs
• Update dataset action specs
• Run dataset actions
• Cancel dataset actions
• List models for user
• Create a new model
• Delete a model
• Retrieve a model
• Update model metadata
• Retrieve default action specs
• Retrieve model action specs
• Update current model specs
• Run model actions
• Early stop / Cancel model actions
• Pause training
• Resume training

Describe your ideal solution

Here is an example function to wrap the endpoint.

@validate_kind
def list_resources(self, kind: _KIND, **kwargs) -> typing.Dict:
    """
    List available resources by kind.

    Parameters
    ----------
    kind : _KIND
        Endpoint type that is specific to a model or a dataset.
    **kwargs :
        Additional arguments.
    Returns
    -------
    json_resp : typing.Dict
        JSON response.
    """

    endpoint = f"{self.user_uri}/{kind}"
    logger.debug("Constructed endpoint with provided input: {}".format(endpoint))

    resp = self.session.get(endpoint, **kwargs)
    if not resp.status_code == 200:
        raise Exception("Unable to list the resources: {}".format(resp.content))

    json_resp = resp.json()
    logger.debug("Response: {}".format(json_resp))

    return json_resp

Describe any alternatives you have considered

No response

Additional context

No response

Code of Conduct

• I agree to follow this project's Code of Conduct
• I have searched the open feature requests and have found no duplicates for this feature request