@mlhub/models-ts-sdk 0.0.0 → 0.0.1

package/src/models/GetModelArtifactResponse.ts

@@ -14,10 +14,10 @@
 
 import { exists, mapValues } from '../runtime';
 import {
-    Artifact,
-    ArtifactFromJSON,
-    ArtifactFromJSONTyped,
-    ArtifactToJSON,
+    ModelArtifact,
+    ModelArtifactFromJSON,
+    ModelArtifactFromJSONTyped,
+    ModelArtifactToJSON,
 } from './';
 
 /**
@@ -40,10 +40,10 @@ export interface GetModelArtifactResponse {
     metadata: object;
     /**
      * 
-     * @type {Artifact}
+     * @type {ModelArtifact}
      * @memberof GetModelArtifactResponse
      */
-    result: Artifact;
+    result: ModelArtifact;
     /**
      * 
      * @type {number}
@@ -70,7 +70,7 @@ export function GetModelArtifactResponseFromJSONTyped(json: any, ignoreDiscrimin
         
         'message': json['message'],
         'metadata': json['metadata'],
-        'result': ArtifactFromJSON(json['result']),
+        'result': ModelArtifactFromJSON(json['result']),
         'status': json['status'],
         'version': json['version'],
     };
@@ -87,7 +87,7 @@ export function GetModelArtifactResponseToJSON(value?: GetModelArtifactResponse
         
         'message': value.message,
         'metadata': value.metadata,
-        'result': ArtifactToJSON(value.result),
+        'result': ModelArtifactToJSON(value.result),
         'status': value.status,
         'version': value.version,
     };

package/src/models/GetModelResponse.ts

@@ -0,0 +1,88 @@
+/* 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 { exists, mapValues } from '../runtime';
+/**
+ * 
+ * @export
+ * @interface GetModelResponse
+ */
+export interface GetModelResponse {
+    /**
+     * 
+     * @type {string}
+     * @memberof GetModelResponse
+     */
+    message: string;
+    /**
+     * 
+     * @type {object}
+     * @memberof GetModelResponse
+     */
+    metadata: object;
+    /**
+     * 
+     * @type {object}
+     * @memberof GetModelResponse
+     */
+    result: object;
+    /**
+     * 
+     * @type {number}
+     * @memberof GetModelResponse
+     */
+    status: number;
+    /**
+     * 
+     * @type {string}
+     * @memberof GetModelResponse
+     */
+    version: string;
+}
+
+export function GetModelResponseFromJSON(json: any): GetModelResponse {
+    return GetModelResponseFromJSONTyped(json, false);
+}
+
+export function GetModelResponseFromJSONTyped(json: any, ignoreDiscriminator: boolean): GetModelResponse {
+    if ((json === undefined) || (json === null)) {
+        return json;
+    }
+    return {
+        
+        'message': json['message'],
+        'metadata': json['metadata'],
+        'result': json['result'],
+        'status': json['status'],
+        'version': json['version'],
+    };
+}
+
+export function GetModelResponseToJSON(value?: GetModelResponse | null): any {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value === null) {
+        return null;
+    }
+    return {
+        
+        'message': value.message,
+        'metadata': value.metadata,
+        'result': value.result,
+        'status': value.status,
+        'version': value.version,
+    };
+}
+

package/src/models/ListModelsResponse.ts

@@ -0,0 +1,88 @@
+/* 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 { exists, mapValues } from '../runtime';
+/**
+ * 
+ * @export
+ * @interface ListModelsResponse
+ */
+export interface ListModelsResponse {
+    /**
+     * 
+     * @type {string}
+     * @memberof ListModelsResponse
+     */
+    message: string;
+    /**
+     * 
+     * @type {object}
+     * @memberof ListModelsResponse
+     */
+    metadata: object;
+    /**
+     * 
+     * @type {object}
+     * @memberof ListModelsResponse
+     */
+    result: object;
+    /**
+     * 
+     * @type {number}
+     * @memberof ListModelsResponse
+     */
+    status: number;
+    /**
+     * 
+     * @type {string}
+     * @memberof ListModelsResponse
+     */
+    version: string;
+}
+
+export function ListModelsResponseFromJSON(json: any): ListModelsResponse {
+    return ListModelsResponseFromJSONTyped(json, false);
+}
+
+export function ListModelsResponseFromJSONTyped(json: any, ignoreDiscriminator: boolean): ListModelsResponse {
+    if ((json === undefined) || (json === null)) {
+        return json;
+    }
+    return {
+        
+        'message': json['message'],
+        'metadata': json['metadata'],
+        'result': json['result'],
+        'status': json['status'],
+        'version': json['version'],
+    };
+}
+
+export function ListModelsResponseToJSON(value?: ListModelsResponse | null): any {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value === null) {
+        return null;
+    }
+    return {
+        
+        'message': value.message,
+        'metadata': value.metadata,
+        'result': value.result,
+        'status': value.status,
+        'version': value.version,
+    };
+}
+

package/src/models/ListTasksResponse.ts

@@ -0,0 +1,95 @@
+/* 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 { exists, mapValues } from '../runtime';
+import {
+    Task,
+    TaskFromJSON,
+    TaskFromJSONTyped,
+    TaskToJSON,
+} from './';
+
+/**
+ * 
+ * @export
+ * @interface ListTasksResponse
+ */
+export interface ListTasksResponse {
+    /**
+     * 
+     * @type {string}
+     * @memberof ListTasksResponse
+     */
+    message: string;
+    /**
+     * 
+     * @type {object}
+     * @memberof ListTasksResponse
+     */
+    metadata: object;
+    /**
+     * 
+     * @type {Array<Task>}
+     * @memberof ListTasksResponse
+     */
+    result: Array<Task>;
+    /**
+     * 
+     * @type {number}
+     * @memberof ListTasksResponse
+     */
+    status: number;
+    /**
+     * 
+     * @type {string}
+     * @memberof ListTasksResponse
+     */
+    version: string;
+}
+
+export function ListTasksResponseFromJSON(json: any): ListTasksResponse {
+    return ListTasksResponseFromJSONTyped(json, false);
+}
+
+export function ListTasksResponseFromJSONTyped(json: any, ignoreDiscriminator: boolean): ListTasksResponse {
+    if ((json === undefined) || (json === null)) {
+        return json;
+    }
+    return {
+        
+        'message': json['message'],
+        'metadata': json['metadata'],
+        'result': ((json['result'] as Array<any>).map(TaskFromJSON)),
+        'status': json['status'],
+        'version': json['version'],
+    };
+}
+
+export function ListTasksResponseToJSON(value?: ListTasksResponse | null): any {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value === null) {
+        return null;
+    }
+    return {
+        
+        'message': value.message,
+        'metadata': value.metadata,
+        'result': ((value.result as Array<any>).map(TaskToJSON)),
+        'status': value.status,
+        'version': value.version,
+    };
+}
+

package/src/models/ModelArtifact.ts

@@ -0,0 +1,99 @@
+/* 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 { exists, mapValues } from '../runtime';
+import {
+    ArtifactType,
+    ArtifactTypeFromJSON,
+    ArtifactTypeFromJSONTyped,
+    ArtifactTypeToJSON,
+    ModelMetadata,
+    ModelMetadataFromJSON,
+    ModelMetadataFromJSONTyped,
+    ModelMetadataToJSON,
+} from './';
+
+/**
+ * 
+ * @export
+ * @interface ModelArtifact
+ */
+export interface ModelArtifact {
+    /**
+     * 
+     * @type {ArtifactType}
+     * @memberof ModelArtifact
+     */
+    artifact_type: ArtifactType;
+    /**
+     * 
+     * @type {string}
+     * @memberof ModelArtifact
+     */
+    created_at: string;
+    /**
+     * 
+     * @type {string}
+     * @memberof ModelArtifact
+     */
+    id: string;
+    /**
+     * 
+     * @type {string}
+     * @memberof ModelArtifact
+     */
+    last_modified: string;
+    /**
+     * 
+     * @type {ModelMetadata}
+     * @memberof ModelArtifact
+     */
+    metadata?: ModelMetadata | null;
+}
+
+export function ModelArtifactFromJSON(json: any): ModelArtifact {
+    return ModelArtifactFromJSONTyped(json, false);
+}
+
+export function ModelArtifactFromJSONTyped(json: any, ignoreDiscriminator: boolean): ModelArtifact {
+    if ((json === undefined) || (json === null)) {
+        return json;
+    }
+    return {
+        
+        'artifact_type': ArtifactTypeFromJSON(json['artifact_type']),
+        'created_at': json['created_at'],
+        'id': json['id'],
+        'last_modified': json['last_modified'],
+        'metadata': !exists(json, 'metadata') ? undefined : ModelMetadataFromJSON(json['metadata']),
+    };
+}
+
+export function ModelArtifactToJSON(value?: ModelArtifact | null): any {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value === null) {
+        return null;
+    }
+    return {
+        
+        'artifact_type': ArtifactTypeToJSON(value.artifact_type),
+        'created_at': value.created_at,
+        'id': value.id,
+        'last_modified': value.last_modified,
+        'metadata': ModelMetadataToJSON(value.metadata),
+    };
+}
+

package/src/models/ModelMetadata.ts

@@ -22,6 +22,10 @@ import {
     ModelIOFromJSON,
     ModelIOFromJSONTyped,
     ModelIOToJSON,
+    Task,
+    TaskFromJSON,
+    TaskFromJSONTyped,
+    TaskToJSON,
 } from './';
 
 /**
@@ -36,6 +40,12 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     annotation?: any | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof ModelMetadata
+     */
+    author?: string | null;
     /**
      * 
      * @type {number}
@@ -54,12 +64,6 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     finetuning_datasets?: Array<string> | null;
-    /**
-     * 
-     * @type {string}
-     * @memberof ModelMetadata
-     */
-    framework?: string | null;
     /**
      * 
      * @type {string}
@@ -126,6 +130,12 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     keywords?: Array<string> | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof ModelMetadata
+     */
+    libraries?: Array<string> | null;
     /**
      * 
      * @type {string}
@@ -207,10 +217,10 @@ export interface ModelMetadata {
     supports_quantization?: boolean | null;
     /**
      * Inference Fields
-     * @type {Array<string>}
+     * @type {Array<Task>}
      * @memberof ModelMetadata
      */
-    task_types?: Array<string> | null;
+    task_types?: Array<Task> | null;
     /**
      * 
      * @type {boolean}
@@ -241,12 +251,6 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     training_time?: number | null;
-    /**
-     * 
-     * @type {string}
-     * @memberof ModelMetadata
-     */
-    version?: string | null;
 }
 
 export function ModelMetadataFromJSON(json: any): ModelMetadata {
@@ -260,10 +264,10 @@ export function ModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boole
     return {
         
         'annotation': !exists(json, 'annotation') ? undefined : json['annotation'],
+        'author': !exists(json, 'author') ? undefined : json['author'],
         'bias_evaluation_score': !exists(json, 'bias_evaluation_score') ? undefined : json['bias_evaluation_score'],
         'edge_optimized': !exists(json, 'edge_optimized') ? undefined : json['edge_optimized'],
         'finetuning_datasets': !exists(json, 'finetuning_datasets') ? undefined : json['finetuning_datasets'],
-        'framework': !exists(json, 'framework') ? undefined : json['framework'],
         'image': !exists(json, 'image') ? undefined : json['image'],
         'inference_distributed': !exists(json, 'inference_distributed') ? undefined : json['inference_distributed'],
         'inference_hardware': !exists(json, 'inference_hardware') ? undefined : HardwareRequirementsFromJSON(json['inference_hardware']),
@@ -275,6 +279,7 @@ export function ModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boole
         'inference_precision': !exists(json, 'inference_precision') ? undefined : json['inference_precision'],
         'inference_software_dependencies': !exists(json, 'inference_software_dependencies') ? undefined : json['inference_software_dependencies'],
         'keywords': !exists(json, 'keywords') ? undefined : json['keywords'],
+        'libraries': !exists(json, 'libraries') ? undefined : json['libraries'],
         'license': !exists(json, 'license') ? undefined : json['license'],
         'model_inputs': !exists(json, 'model_inputs') ? undefined : (json['model_inputs'] === null ? null : (json['model_inputs'] as Array<any>).map(ModelIOFromJSON)),
         'model_outputs': !exists(json, 'model_outputs') ? undefined : (json['model_outputs'] === null ? null : (json['model_outputs'] as Array<any>).map(ModelIOFromJSON)),
@@ -288,13 +293,12 @@ export function ModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boole
         'regulatory': !exists(json, 'regulatory') ? undefined : json['regulatory'],
         'slimmed': !exists(json, 'slimmed') ? undefined : json['slimmed'],
         'supports_quantization': !exists(json, 'supports_quantization') ? undefined : json['supports_quantization'],
-        'task_types': !exists(json, 'task_types') ? undefined : json['task_types'],
+        'task_types': !exists(json, 'task_types') ? undefined : (json['task_types'] === null ? null : (json['task_types'] as Array<any>).map(TaskFromJSON)),
         'training_distributed': !exists(json, 'training_distributed') ? undefined : json['training_distributed'],
         'training_hardware': !exists(json, 'training_hardware') ? undefined : HardwareRequirementsFromJSON(json['training_hardware']),
         'training_max_energy_consumption_watts': !exists(json, 'training_max_energy_consumption_watts') ? undefined : json['training_max_energy_consumption_watts'],
         'training_precision': !exists(json, 'training_precision') ? undefined : json['training_precision'],
         'training_time': !exists(json, 'training_time') ? undefined : json['training_time'],
-        'version': !exists(json, 'version') ? undefined : json['version'],
     };
 }
 
@@ -308,10 +312,10 @@ export function ModelMetadataToJSON(value?: ModelMetadata | null): any {
     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': HardwareRequirementsToJSON(value.inference_hardware),
@@ -323,6 +327,7 @@ export function ModelMetadataToJSON(value?: ModelMetadata | null): any {
         '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 as Array<any>).map(ModelIOToJSON)),
         'model_outputs': value.model_outputs === undefined ? undefined : (value.model_outputs === null ? null : (value.model_outputs as Array<any>).map(ModelIOToJSON)),
@@ -336,13 +341,12 @@ export function ModelMetadataToJSON(value?: ModelMetadata | null): any {
         '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 as Array<any>).map(TaskToJSON)),
         'training_distributed': value.training_distributed,
         'training_hardware': 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,
     };
 }
 

package/src/models/Task.ts

@@ -0,0 +1,79 @@
+/* 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.
+ */
+
+/**
+ * An enum of all task types available on Huggingface
+ * @export
+ * @enum {string}
+ */
+export 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 function TaskFromJSON(json: any): Task {
+    return TaskFromJSONTyped(json, false);
+}
+
+export function TaskFromJSONTyped(json: any, ignoreDiscriminator: boolean): Task {
+    return json as Task;
+}
+
+export function TaskToJSON(value?: Task | null): any {
+    return value as any;
+}
+

package/src/models/index.ts

@@ -7,14 +7,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';
@@ -23,7 +28,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';
@@ -33,3 +41,4 @@ export * from './PublishArtifactRequest';
 export * from './PublishModelArtifactResponse';
 export * from './ServerErrorResponse';
 export * from './SystemRequirement';
+export * from './Task';