@mlhub/models-ts-sdk 0.0.0 → 0.0.1

package/dist/models/ModelMetadata.js

@@ -25,10 +25,10 @@ function ModelMetadataFromJSONTyped(json, ignoreDiscriminator) {
     }
     return {
         'annotation': !runtime_1.exists(json, 'annotation') ? undefined : json['annotation'],
+        'author': !runtime_1.exists(json, 'author') ? undefined : json['author'],
         'bias_evaluation_score': !runtime_1.exists(json, 'bias_evaluation_score') ? undefined : json['bias_evaluation_score'],
         'edge_optimized': !runtime_1.exists(json, 'edge_optimized') ? undefined : json['edge_optimized'],
         'finetuning_datasets': !runtime_1.exists(json, 'finetuning_datasets') ? undefined : json['finetuning_datasets'],
-        'framework': !runtime_1.exists(json, 'framework') ? undefined : json['framework'],
         'image': !runtime_1.exists(json, 'image') ? undefined : json['image'],
         'inference_distributed': !runtime_1.exists(json, 'inference_distributed') ? undefined : json['inference_distributed'],
         'inference_hardware': !runtime_1.exists(json, 'inference_hardware') ? undefined : _1.HardwareRequirementsFromJSON(json['inference_hardware']),
@@ -40,6 +40,7 @@ function ModelMetadataFromJSONTyped(json, ignoreDiscriminator) {
         'inference_precision': !runtime_1.exists(json, 'inference_precision') ? undefined : json['inference_precision'],
         'inference_software_dependencies': !runtime_1.exists(json, 'inference_software_dependencies') ? undefined : json['inference_software_dependencies'],
         'keywords': !runtime_1.exists(json, 'keywords') ? undefined : json['keywords'],
+        'libraries': !runtime_1.exists(json, 'libraries') ? undefined : json['libraries'],
         'license': !runtime_1.exists(json, 'license') ? undefined : json['license'],
         'model_inputs': !runtime_1.exists(json, 'model_inputs') ? undefined : (json['model_inputs'] === null ? null : json['model_inputs'].map(_1.ModelIOFromJSON)),
         'model_outputs': !runtime_1.exists(json, 'model_outputs') ? undefined : (json['model_outputs'] === null ? null : json['model_outputs'].map(_1.ModelIOFromJSON)),
@@ -53,13 +54,12 @@ function ModelMetadataFromJSONTyped(json, ignoreDiscriminator) {
         'regulatory': !runtime_1.exists(json, 'regulatory') ? undefined : json['regulatory'],
         'slimmed': !runtime_1.exists(json, 'slimmed') ? undefined : json['slimmed'],
         'supports_quantization': !runtime_1.exists(json, 'supports_quantization') ? undefined : json['supports_quantization'],
-        'task_types': !runtime_1.exists(json, 'task_types') ? undefined : json['task_types'],
+        'task_types': !runtime_1.exists(json, 'task_types') ? undefined : (json['task_types'] === null ? null : json['task_types'].map(_1.TaskFromJSON)),
         'training_distributed': !runtime_1.exists(json, 'training_distributed') ? undefined : json['training_distributed'],
         'training_hardware': !runtime_1.exists(json, 'training_hardware') ? undefined : _1.HardwareRequirementsFromJSON(json['training_hardware']),
         'training_max_energy_consumption_watts': !runtime_1.exists(json, 'training_max_energy_consumption_watts') ? undefined : json['training_max_energy_consumption_watts'],
         'training_precision': !runtime_1.exists(json, 'training_precision') ? undefined : json['training_precision'],
         'training_time': !runtime_1.exists(json, 'training_time') ? undefined : json['training_time'],
-        'version': !runtime_1.exists(json, 'version') ? undefined : json['version'],
     };
 }
 exports.ModelMetadataFromJSONTyped = ModelMetadataFromJSONTyped;
@@ -72,10 +72,10 @@ function ModelMetadataToJSON(value) {
     }
     return {
         'annotation': value.annotation,
+        'author': value.author,
         'bias_evaluation_score': value.bias_evaluation_score,
         'edge_optimized': value.edge_optimized,
         'finetuning_datasets': value.finetuning_datasets,
-        'framework': value.framework,
         'image': value.image,
         'inference_distributed': value.inference_distributed,
         'inference_hardware': _1.HardwareRequirementsToJSON(value.inference_hardware),
@@ -87,6 +87,7 @@ function ModelMetadataToJSON(value) {
         'inference_precision': value.inference_precision,
         'inference_software_dependencies': value.inference_software_dependencies,
         'keywords': value.keywords,
+        'libraries': value.libraries,
         'license': value.license,
         'model_inputs': value.model_inputs === undefined ? undefined : (value.model_inputs === null ? null : value.model_inputs.map(_1.ModelIOToJSON)),
         'model_outputs': value.model_outputs === undefined ? undefined : (value.model_outputs === null ? null : value.model_outputs.map(_1.ModelIOToJSON)),
@@ -100,13 +101,12 @@ function ModelMetadataToJSON(value) {
         'regulatory': value.regulatory,
         'slimmed': value.slimmed,
         'supports_quantization': value.supports_quantization,
-        'task_types': value.task_types,
+        'task_types': value.task_types === undefined ? undefined : (value.task_types === null ? null : value.task_types.map(_1.TaskToJSON)),
         'training_distributed': value.training_distributed,
         'training_hardware': _1.HardwareRequirementsToJSON(value.training_hardware),
         'training_max_energy_consumption_watts': value.training_max_energy_consumption_watts,
         'training_precision': value.training_precision,
         'training_time': value.training_time,
-        'version': value.version,
     };
 }
 exports.ModelMetadataToJSON = ModelMetadataToJSON;

package/dist/models/Task.d.ts

@@ -0,0 +1,66 @@
+/**
+ * MLHub Models API
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 0.1.0
+ *
+ *
+ * NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
+ * https://openapi-generator.tech
+ * Do not edit the class manually.
+ */
+/**
+ * An enum of all task types available on Huggingface
+ * @export
+ * @enum {string}
+ */
+export declare enum Task {
+    AnyToAny = "AnyToAny",
+    AudioClassification = "AudioClassification",
+    AudioToAudio = "AudioToAudio",
+    AudioTextToText = "AudioTextToText",
+    AutomaticSpeechRecognition = "AutomaticSpeechRecognition",
+    DepthEstimation = "DepthEstimation",
+    DocumentQuestionAnswering = "DocumentQuestionAnswering",
+    VisualDocumentRetrieval = "VisualDocumentRetrieval",
+    FeatureExtraction = "FeatureExtraction",
+    FillMask = "FillMask",
+    ImageClassification = "ImageClassification",
+    ImageFeatureExtraction = "ImageFeatureExtraction",
+    ImageSegmentation = "ImageSegmentation",
+    ImageToImage = "ImageToImage",
+    ImageTextToText = "ImageTextToText",
+    ImageToText = "ImageToText",
+    ImageToVideo = "ImageToVideo",
+    KeypointDetection = "KeypointDetection",
+    MaskGeneration = "MaskGeneration",
+    ObjectDetection = "ObjectDetection",
+    VideoClassification = "VideoClassification",
+    QuestionAnswering = "QuestionAnswering",
+    ReinforcementLearning = "ReinforcementLearning",
+    SentenceSimilarity = "SentenceSimilarity",
+    Summarization = "Summarization",
+    TableQuestionAnswering = "TableQuestionAnswering",
+    TabularClassification = "TabularClassification",
+    TabularRegression = "TabularRegression",
+    TextClassification = "TextClassification",
+    TextGeneration = "TextGeneration",
+    TextRanking = "TextRanking",
+    TextToImage = "TextToImage",
+    TextToSpeech = "TextToSpeech",
+    TextToVideo = "TextToVideo",
+    TokenClassification = "TokenClassification",
+    Translation = "Translation",
+    UnconditionalImageGeneration = "UnconditionalImageGeneration",
+    VideoTextToText = "VideoTextToText",
+    VideoToVideo = "VideoToVideo",
+    VisualQuestionAnswering = "VisualQuestionAnswering",
+    ZeroShotClassification = "ZeroShotClassification",
+    ZeroShotImageClassification = "ZeroShotImageClassification",
+    ZeroShotObjectDetection = "ZeroShotObjectDetection",
+    TextTo3d = "TextTo3d",
+    ImageTo3d = "ImageTo3d"
+}
+export declare function TaskFromJSON(json: any): Task;
+export declare function TaskFromJSONTyped(json: any, ignoreDiscriminator: boolean): Task;
+export declare function TaskToJSON(value?: Task | null): any;

package/dist/models/Task.js

@@ -0,0 +1,80 @@
+"use strict";
+/* tslint:disable */
+/* eslint-disable */
+/**
+ * MLHub Models API
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 0.1.0
+ *
+ *
+ * NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
+ * https://openapi-generator.tech
+ * Do not edit the class manually.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+/**
+ * An enum of all task types available on Huggingface
+ * @export
+ * @enum {string}
+ */
+var Task;
+(function (Task) {
+    Task["AnyToAny"] = "AnyToAny";
+    Task["AudioClassification"] = "AudioClassification";
+    Task["AudioToAudio"] = "AudioToAudio";
+    Task["AudioTextToText"] = "AudioTextToText";
+    Task["AutomaticSpeechRecognition"] = "AutomaticSpeechRecognition";
+    Task["DepthEstimation"] = "DepthEstimation";
+    Task["DocumentQuestionAnswering"] = "DocumentQuestionAnswering";
+    Task["VisualDocumentRetrieval"] = "VisualDocumentRetrieval";
+    Task["FeatureExtraction"] = "FeatureExtraction";
+    Task["FillMask"] = "FillMask";
+    Task["ImageClassification"] = "ImageClassification";
+    Task["ImageFeatureExtraction"] = "ImageFeatureExtraction";
+    Task["ImageSegmentation"] = "ImageSegmentation";
+    Task["ImageToImage"] = "ImageToImage";
+    Task["ImageTextToText"] = "ImageTextToText";
+    Task["ImageToText"] = "ImageToText";
+    Task["ImageToVideo"] = "ImageToVideo";
+    Task["KeypointDetection"] = "KeypointDetection";
+    Task["MaskGeneration"] = "MaskGeneration";
+    Task["ObjectDetection"] = "ObjectDetection";
+    Task["VideoClassification"] = "VideoClassification";
+    Task["QuestionAnswering"] = "QuestionAnswering";
+    Task["ReinforcementLearning"] = "ReinforcementLearning";
+    Task["SentenceSimilarity"] = "SentenceSimilarity";
+    Task["Summarization"] = "Summarization";
+    Task["TableQuestionAnswering"] = "TableQuestionAnswering";
+    Task["TabularClassification"] = "TabularClassification";
+    Task["TabularRegression"] = "TabularRegression";
+    Task["TextClassification"] = "TextClassification";
+    Task["TextGeneration"] = "TextGeneration";
+    Task["TextRanking"] = "TextRanking";
+    Task["TextToImage"] = "TextToImage";
+    Task["TextToSpeech"] = "TextToSpeech";
+    Task["TextToVideo"] = "TextToVideo";
+    Task["TokenClassification"] = "TokenClassification";
+    Task["Translation"] = "Translation";
+    Task["UnconditionalImageGeneration"] = "UnconditionalImageGeneration";
+    Task["VideoTextToText"] = "VideoTextToText";
+    Task["VideoToVideo"] = "VideoToVideo";
+    Task["VisualQuestionAnswering"] = "VisualQuestionAnswering";
+    Task["ZeroShotClassification"] = "ZeroShotClassification";
+    Task["ZeroShotImageClassification"] = "ZeroShotImageClassification";
+    Task["ZeroShotObjectDetection"] = "ZeroShotObjectDetection";
+    Task["TextTo3d"] = "TextTo3d";
+    Task["ImageTo3d"] = "ImageTo3d";
+})(Task = exports.Task || (exports.Task = {}));
+function TaskFromJSON(json) {
+    return TaskFromJSONTyped(json, false);
+}
+exports.TaskFromJSON = TaskFromJSON;
+function TaskFromJSONTyped(json, ignoreDiscriminator) {
+    return json;
+}
+exports.TaskFromJSONTyped = TaskFromJSONTyped;
+function TaskToJSON(value) {
+    return value;
+}
+exports.TaskToJSON = TaskToJSON;

package/dist/models/index.d.ts

@@ -5,14 +5,19 @@ export * from './ArtifactIngestionStatus';
 export * from './ArtifactPublication';
 export * from './ArtifactPublicationStatus';
 export * from './ArtifactType';
+export * from './AssociateModelMetadata';
+export * from './AssociateModelMetadataResponse';
 export * from './BadRequestResponse';
 export * from './CreateModelMetadataResponse';
 export * from './DiscoverModelsByPlatformResponse';
+export * from './DiscoverModelsResponse';
 export * from './DiscoveryCriteria';
+export * from './DiscoveryCriterion';
 export * from './GetModelArtifactResponse';
 export * from './GetModelByPlatformResponse';
 export * from './GetModelIngestionResponse';
 export * from './GetModelPublicationResponse';
+export * from './GetModelResponse';
 export * from './HardwareRequirements';
 export * from './IngestArtifactRequest';
 export * from './IngestModelArtifactResponse';
@@ -21,7 +26,10 @@ export * from './ListModelIngestionsResponse';
 export * from './ListModelPublicationsForArtifactResponse';
 export * from './ListModelPublicationsResponse';
 export * from './ListModelsByPlatformResponse';
+export * from './ListModelsResponse';
 export * from './ListPlatformsResponse';
+export * from './ListTasksResponse';
+export * from './ModelArtifact';
 export * from './ModelIO';
 export * from './ModelMetadata';
 export * from './NotFoundResponse';
@@ -31,3 +39,4 @@ export * from './PublishArtifactRequest';
 export * from './PublishModelArtifactResponse';
 export * from './ServerErrorResponse';
 export * from './SystemRequirement';
+export * from './Task';

package/dist/models/index.js

@@ -12,14 +12,19 @@ __export(require("./ArtifactIngestionStatus"));
 __export(require("./ArtifactPublication"));
 __export(require("./ArtifactPublicationStatus"));
 __export(require("./ArtifactType"));
+__export(require("./AssociateModelMetadata"));
+__export(require("./AssociateModelMetadataResponse"));
 __export(require("./BadRequestResponse"));
 __export(require("./CreateModelMetadataResponse"));
 __export(require("./DiscoverModelsByPlatformResponse"));
+__export(require("./DiscoverModelsResponse"));
 __export(require("./DiscoveryCriteria"));
+__export(require("./DiscoveryCriterion"));
 __export(require("./GetModelArtifactResponse"));
 __export(require("./GetModelByPlatformResponse"));
 __export(require("./GetModelIngestionResponse"));
 __export(require("./GetModelPublicationResponse"));
+__export(require("./GetModelResponse"));
 __export(require("./HardwareRequirements"));
 __export(require("./IngestArtifactRequest"));
 __export(require("./IngestModelArtifactResponse"));
@@ -28,7 +33,10 @@ __export(require("./ListModelIngestionsResponse"));
 __export(require("./ListModelPublicationsForArtifactResponse"));
 __export(require("./ListModelPublicationsResponse"));
 __export(require("./ListModelsByPlatformResponse"));
+__export(require("./ListModelsResponse"));
 __export(require("./ListPlatformsResponse"));
+__export(require("./ListTasksResponse"));
+__export(require("./ModelArtifact"));
 __export(require("./ModelIO"));
 __export(require("./ModelMetadata"));
 __export(require("./NotFoundResponse"));
@@ -38,3 +46,4 @@ __export(require("./PublishArtifactRequest"));
 __export(require("./PublishModelArtifactResponse"));
 __export(require("./ServerErrorResponse"));
 __export(require("./SystemRequirement"));
+__export(require("./Task"));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mlhub/models-ts-sdk",
-  "version": "0.0.0",
+  "version": "0.0.1",
   "description": "OpenAPI client for @mlhub/models-ts-sdk",
   "author": "OpenAPI-Generator",
   "main": "./dist/index.js",

package/src/apis/ArtifactsApi.ts

@@ -15,6 +15,12 @@
 
 import * as runtime from '../runtime';
 import {
+    AssociateModelMetadata,
+    AssociateModelMetadataFromJSON,
+    AssociateModelMetadataToJSON,
+    AssociateModelMetadataResponse,
+    AssociateModelMetadataResponseFromJSON,
+    AssociateModelMetadataResponseToJSON,
     BadRequestResponse,
     BadRequestResponseFromJSON,
     BadRequestResponseToJSON,
@@ -32,6 +38,11 @@ import {
     ServerErrorResponseToJSON,
 } from '../models';
 
+export interface AssociateModelMetadataWithArtifactRequest {
+    artifactId: string;
+    associateModelMetadata: AssociateModelMetadata;
+}
+
 export interface GetModelArtifactRequest {
     artifactId: string;
 }
@@ -41,6 +52,43 @@ export interface GetModelArtifactRequest {
  */
 export class ArtifactsApi extends runtime.BaseAPI {
 
+    /**
+     * Associate existing model metadata to a model artifact
+     */
+    async associateModelMetadataWithArtifactRaw(requestParameters: AssociateModelMetadataWithArtifactRequest, initOverrides?: RequestInit): Promise<runtime.ApiResponse<AssociateModelMetadataResponse>> {
+        if (requestParameters.artifactId === null || requestParameters.artifactId === undefined) {
+            throw new runtime.RequiredError('artifactId','Required parameter requestParameters.artifactId was null or undefined when calling associateModelMetadataWithArtifact.');
+        }
+
+        if (requestParameters.associateModelMetadata === null || requestParameters.associateModelMetadata === undefined) {
+            throw new runtime.RequiredError('associateModelMetadata','Required parameter requestParameters.associateModelMetadata was null or undefined when calling associateModelMetadataWithArtifact.');
+        }
+
+        const queryParameters: any = {};
+
+        const headerParameters: runtime.HTTPHeaders = {};
+
+        headerParameters['Content-Type'] = 'application/json';
+
+        const response = await this.request({
+            path: `/models-api/artifacts/{artifact_id}/metadata`.replace(`{${"artifact_id"}}`, encodeURIComponent(String(requestParameters.artifactId))),
+            method: 'POST',
+            headers: headerParameters,
+            query: queryParameters,
+            body: AssociateModelMetadataToJSON(requestParameters.associateModelMetadata),
+        }, initOverrides);
+
+        return new runtime.JSONApiResponse(response, (jsonValue) => AssociateModelMetadataResponseFromJSON(jsonValue));
+    }
+
+    /**
+     * Associate existing model metadata to a model artifact
+     */
+    async associateModelMetadataWithArtifact(requestParameters: AssociateModelMetadataWithArtifactRequest, initOverrides?: RequestInit): Promise<AssociateModelMetadataResponse> {
+        const response = await this.associateModelMetadataWithArtifactRaw(requestParameters, initOverrides);
+        return await response.value();
+    }
+
     /**
      * Fetches the model artifact by the provided id
      */

package/src/apis/ModelsApi.ts

@@ -0,0 +1,236 @@
+/* tslint:disable */
+/* eslint-disable */
+/**
+ * MLHub Models API
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 0.1.0
+ * 
+ *
+ * NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
+ * https://openapi-generator.tech
+ * Do not edit the class manually.
+ */
+
+
+import * as runtime from '../runtime';
+import {
+    BadRequestResponse,
+    BadRequestResponseFromJSON,
+    BadRequestResponseToJSON,
+    CreateModelMetadataResponse,
+    CreateModelMetadataResponseFromJSON,
+    CreateModelMetadataResponseToJSON,
+    DiscoverModelsResponse,
+    DiscoverModelsResponseFromJSON,
+    DiscoverModelsResponseToJSON,
+    DiscoveryCriteria,
+    DiscoveryCriteriaFromJSON,
+    DiscoveryCriteriaToJSON,
+    GetModelResponse,
+    GetModelResponseFromJSON,
+    GetModelResponseToJSON,
+    ListModelsResponse,
+    ListModelsResponseFromJSON,
+    ListModelsResponseToJSON,
+    ModelMetadata,
+    ModelMetadataFromJSON,
+    ModelMetadataToJSON,
+    NotFoundResponse,
+    NotFoundResponseFromJSON,
+    NotFoundResponseToJSON,
+    ServerErrorResponse,
+    ServerErrorResponseFromJSON,
+    ServerErrorResponseToJSON,
+} from '../models';
+
+export interface CreateModelMetadataRequest {
+    modelMetadata: ModelMetadata;
+}
+
+export interface DiscoverModelsRequest {
+    discoveryCriteria: DiscoveryCriteria;
+}
+
+export interface GetModelRequest {
+    author: string;
+    name: string;
+}
+
+export interface ListModelsRequest {
+    modelMetadata: ModelMetadata;
+}
+
+export interface ListModelsByAuthorRequest {
+    author: string;
+}
+
+/**
+ * 
+ */
+export class ModelsApi extends runtime.BaseAPI {
+
+    /**
+     * Create a model metadata
+     */
+    async createModelMetadataRaw(requestParameters: CreateModelMetadataRequest, initOverrides?: RequestInit): Promise<runtime.ApiResponse<CreateModelMetadataResponse>> {
+        if (requestParameters.modelMetadata === null || requestParameters.modelMetadata === undefined) {
+            throw new runtime.RequiredError('modelMetadata','Required parameter requestParameters.modelMetadata was null or undefined when calling createModelMetadata.');
+        }
+
+        const queryParameters: any = {};
+
+        const headerParameters: runtime.HTTPHeaders = {};
+
+        headerParameters['Content-Type'] = 'application/json';
+
+        const response = await this.request({
+            path: `/models-api/models`,
+            method: 'POST',
+            headers: headerParameters,
+            query: queryParameters,
+            body: ModelMetadataToJSON(requestParameters.modelMetadata),
+        }, initOverrides);
+
+        return new runtime.JSONApiResponse(response, (jsonValue) => CreateModelMetadataResponseFromJSON(jsonValue));
+    }
+
+    /**
+     * Create a model metadata
+     */
+    async createModelMetadata(requestParameters: CreateModelMetadataRequest, initOverrides?: RequestInit): Promise<CreateModelMetadataResponse> {
+        const response = await this.createModelMetadataRaw(requestParameters, initOverrides);
+        return await response.value();
+    }
+
+    /**
+     * Discover models known to MLHub
+     */
+    async discoverModelsRaw(requestParameters: DiscoverModelsRequest, initOverrides?: RequestInit): Promise<runtime.ApiResponse<DiscoverModelsResponse>> {
+        if (requestParameters.discoveryCriteria === null || requestParameters.discoveryCriteria === undefined) {
+            throw new runtime.RequiredError('discoveryCriteria','Required parameter requestParameters.discoveryCriteria was null or undefined when calling discoverModels.');
+        }
+
+        const queryParameters: any = {};
+
+        const headerParameters: runtime.HTTPHeaders = {};
+
+        headerParameters['Content-Type'] = 'application/json';
+
+        const response = await this.request({
+            path: `/models-api/models/search`,
+            method: 'POST',
+            headers: headerParameters,
+            query: queryParameters,
+            body: DiscoveryCriteriaToJSON(requestParameters.discoveryCriteria),
+        }, initOverrides);
+
+        return new runtime.JSONApiResponse(response, (jsonValue) => DiscoverModelsResponseFromJSON(jsonValue));
+    }
+
+    /**
+     * Discover models known to MLHub
+     */
+    async discoverModels(requestParameters: DiscoverModelsRequest, initOverrides?: RequestInit): Promise<DiscoverModelsResponse> {
+        const response = await this.discoverModelsRaw(requestParameters, initOverrides);
+        return await response.value();
+    }
+
+    /**
+     * Get model by the author and name
+     */
+    async getModelRaw(requestParameters: GetModelRequest, initOverrides?: RequestInit): Promise<runtime.ApiResponse<GetModelResponse>> {
+        if (requestParameters.author === null || requestParameters.author === undefined) {
+            throw new runtime.RequiredError('author','Required parameter requestParameters.author was null or undefined when calling getModel.');
+        }
+
+        if (requestParameters.name === null || requestParameters.name === undefined) {
+            throw new runtime.RequiredError('name','Required parameter requestParameters.name was null or undefined when calling getModel.');
+        }
+
+        const queryParameters: any = {};
+
+        const headerParameters: runtime.HTTPHeaders = {};
+
+        const response = await this.request({
+            path: `/models-api/models/{author}/{name}`.replace(`{${"author"}}`, encodeURIComponent(String(requestParameters.author))).replace(`{${"name"}}`, encodeURIComponent(String(requestParameters.name))),
+            method: 'GET',
+            headers: headerParameters,
+            query: queryParameters,
+        }, initOverrides);
+
+        return new runtime.JSONApiResponse(response, (jsonValue) => GetModelResponseFromJSON(jsonValue));
+    }
+
+    /**
+     * Get model by the author and name
+     */
+    async getModel(requestParameters: GetModelRequest, initOverrides?: RequestInit): Promise<GetModelResponse> {
+        const response = await this.getModelRaw(requestParameters, initOverrides);
+        return await response.value();
+    }
+
+    /**
+     * List all models
+     */
+    async listModelsRaw(requestParameters: ListModelsRequest, initOverrides?: RequestInit): Promise<runtime.ApiResponse<ListModelsResponse>> {
+        if (requestParameters.modelMetadata === null || requestParameters.modelMetadata === undefined) {
+            throw new runtime.RequiredError('modelMetadata','Required parameter requestParameters.modelMetadata was null or undefined when calling listModels.');
+        }
+
+        const queryParameters: any = {};
+
+        const headerParameters: runtime.HTTPHeaders = {};
+
+        headerParameters['Content-Type'] = 'application/json';
+
+        const response = await this.request({
+            path: `/models-api/models`,
+            method: 'GET',
+            headers: headerParameters,
+            query: queryParameters,
+            body: ModelMetadataToJSON(requestParameters.modelMetadata),
+        }, initOverrides);
+
+        return new runtime.JSONApiResponse(response, (jsonValue) => ListModelsResponseFromJSON(jsonValue));
+    }
+
+    /**
+     * List all models
+     */
+    async listModels(requestParameters: ListModelsRequest, initOverrides?: RequestInit): Promise<ListModelsResponse> {
+        const response = await this.listModelsRaw(requestParameters, initOverrides);
+        return await response.value();
+    }
+
+    /**
+     * List models by author
+     */
+    async listModelsByAuthorRaw(requestParameters: ListModelsByAuthorRequest, initOverrides?: RequestInit): Promise<runtime.ApiResponse<ListModelsResponse>> {
+        if (requestParameters.author === null || requestParameters.author === undefined) {
+            throw new runtime.RequiredError('author','Required parameter requestParameters.author was null or undefined when calling listModelsByAuthor.');
+        }
+
+        const queryParameters: any = {};
+
+        const headerParameters: runtime.HTTPHeaders = {};
+
+        const response = await this.request({
+            path: `/models-api/models/{author}`.replace(`{${"author"}}`, encodeURIComponent(String(requestParameters.author))),
+            method: 'GET',
+            headers: headerParameters,
+            query: queryParameters,
+        }, initOverrides);
+
+        return new runtime.JSONApiResponse(response, (jsonValue) => ListModelsResponseFromJSON(jsonValue));
+    }
+
+    /**
+     * List models by author
+     */
+    async listModelsByAuthor(requestParameters: ListModelsByAuthorRequest, initOverrides?: RequestInit): Promise<ListModelsResponse> {
+        const response = await this.listModelsByAuthorRaw(requestParameters, initOverrides);
+        return await response.value();
+    }
+
+}

package/src/apis/TasksApi.ts

@@ -0,0 +1,63 @@
+/* tslint:disable */
+/* eslint-disable */
+/**
+ * MLHub Models API
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 0.1.0
+ * 
+ *
+ * NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
+ * https://openapi-generator.tech
+ * Do not edit the class manually.
+ */
+
+
+import * as runtime from '../runtime';
+import {
+    BadRequestResponse,
+    BadRequestResponseFromJSON,
+    BadRequestResponseToJSON,
+    ListTasksResponse,
+    ListTasksResponseFromJSON,
+    ListTasksResponseToJSON,
+    NotFoundResponse,
+    NotFoundResponseFromJSON,
+    NotFoundResponseToJSON,
+    ServerErrorResponse,
+    ServerErrorResponseFromJSON,
+    ServerErrorResponseToJSON,
+} from '../models';
+
+/**
+ * 
+ */
+export class TasksApi extends runtime.BaseAPI {
+
+    /**
+     * List all model tasks/capabilities
+     */
+    async listTasksRaw(initOverrides?: RequestInit): Promise<runtime.ApiResponse<ListTasksResponse>> {
+        const queryParameters: any = {};
+
+        const headerParameters: runtime.HTTPHeaders = {};
+
+        const response = await this.request({
+            path: `/models-api/tasks`,
+            method: 'GET',
+            headers: headerParameters,
+            query: queryParameters,
+        }, initOverrides);
+
+        return new runtime.JSONApiResponse(response, (jsonValue) => ListTasksResponseFromJSON(jsonValue));
+    }
+
+    /**
+     * List all model tasks/capabilities
+     */
+    async listTasks(initOverrides?: RequestInit): Promise<ListTasksResponse> {
+        const response = await this.listTasksRaw(initOverrides);
+        return await response.value();
+    }
+
+}

package/src/apis/index.ts

@@ -2,6 +2,7 @@
 /* eslint-disable */
 export * from './ArtifactsApi';
 export * from './IngestionsApi';
-export * from './MetadataApi';
+export * from './ModelsApi';
 export * from './PlatformsApi';
 export * from './PublicationsApi';
+export * from './TasksApi';