@mlhub/models-ts-sdk 0.0.0 → 0.0.2

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';

package/src/models/AssociateModelMetadata.ts

@@ -0,0 +1,64 @@
+/* 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 AssociateModelMetadata
+ */
+export interface AssociateModelMetadata {
+    /**
+     * 
+     * @type {string}
+     * @memberof AssociateModelMetadata
+     */
+    author: string;
+    /**
+     * 
+     * @type {string}
+     * @memberof AssociateModelMetadata
+     */
+    name: string;
+}
+
+export function AssociateModelMetadataFromJSON(json: any): AssociateModelMetadata {
+    return AssociateModelMetadataFromJSONTyped(json, false);
+}
+
+export function AssociateModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boolean): AssociateModelMetadata {
+    if ((json === undefined) || (json === null)) {
+        return json;
+    }
+    return {
+        
+        'author': json['author'],
+        'name': json['name'],
+    };
+}
+
+export function AssociateModelMetadataToJSON(value?: AssociateModelMetadata | null): any {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value === null) {
+        return null;
+    }
+    return {
+        
+        'author': value.author,
+        'name': value.name,
+    };
+}
+

package/src/models/AssociateModelMetadataResponse.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 AssociateModelMetadataResponse
+ */
+export interface AssociateModelMetadataResponse {
+    /**
+     * 
+     * @type {string}
+     * @memberof AssociateModelMetadataResponse
+     */
+    message: string;
+    /**
+     * 
+     * @type {object}
+     * @memberof AssociateModelMetadataResponse
+     */
+    metadata: object;
+    /**
+     * 
+     * @type {object}
+     * @memberof AssociateModelMetadataResponse
+     */
+    result: object;
+    /**
+     * 
+     * @type {number}
+     * @memberof AssociateModelMetadataResponse
+     */
+    status: number;
+    /**
+     * 
+     * @type {string}
+     * @memberof AssociateModelMetadataResponse
+     */
+    version: string;
+}
+
+export function AssociateModelMetadataResponseFromJSON(json: any): AssociateModelMetadataResponse {
+    return AssociateModelMetadataResponseFromJSONTyped(json, false);
+}
+
+export function AssociateModelMetadataResponseFromJSONTyped(json: any, ignoreDiscriminator: boolean): AssociateModelMetadataResponse {
+    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 AssociateModelMetadataResponseToJSON(value?: AssociateModelMetadataResponse | 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/DiscoverModelsResponse.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 DiscoverModelsResponse
+ */
+export interface DiscoverModelsResponse {
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoverModelsResponse
+     */
+    message: string;
+    /**
+     * 
+     * @type {object}
+     * @memberof DiscoverModelsResponse
+     */
+    metadata: object;
+    /**
+     * 
+     * @type {object}
+     * @memberof DiscoverModelsResponse
+     */
+    result: object;
+    /**
+     * 
+     * @type {number}
+     * @memberof DiscoverModelsResponse
+     */
+    status: number;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoverModelsResponse
+     */
+    version: string;
+}
+
+export function DiscoverModelsResponseFromJSON(json: any): DiscoverModelsResponse {
+    return DiscoverModelsResponseFromJSONTyped(json, false);
+}
+
+export function DiscoverModelsResponseFromJSONTyped(json: any, ignoreDiscriminator: boolean): DiscoverModelsResponse {
+    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 DiscoverModelsResponseToJSON(value?: DiscoverModelsResponse | 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/DiscoveryCriteria.ts

@@ -14,10 +14,10 @@
 
 import { exists, mapValues } from '../runtime';
 import {
-    ModelMetadata,
-    ModelMetadataFromJSON,
-    ModelMetadataFromJSONTyped,
-    ModelMetadataToJSON,
+    DiscoveryCriterion,
+    DiscoveryCriterionFromJSON,
+    DiscoveryCriterionFromJSONTyped,
+    DiscoveryCriterionToJSON,
 } from './';
 
 /**
@@ -28,16 +28,22 @@ import {
 export interface DiscoveryCriteria {
     /**
      * 
-     * @type {Array<string>}
+     * @type {number}
      * @memberof DiscoveryCriteria
      */
-    confidence_threshold?: Array<string> | null;
+    confidence_threshold?: number | null;
     /**
      * 
-     * @type {Array<ModelMetadata>}
+     * @type {Array<DiscoveryCriterion>}
      * @memberof DiscoveryCriteria
      */
-    criteria: Array<ModelMetadata>;
+    criteria: Array<DiscoveryCriterion>;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriteria
+     */
+    prompt?: string | null;
 }
 
 export function DiscoveryCriteriaFromJSON(json: any): DiscoveryCriteria {
@@ -51,7 +57,8 @@ export function DiscoveryCriteriaFromJSONTyped(json: any, ignoreDiscriminator: b
     return {
         
         'confidence_threshold': !exists(json, 'confidence_threshold') ? undefined : json['confidence_threshold'],
-        'criteria': ((json['criteria'] as Array<any>).map(ModelMetadataFromJSON)),
+        'criteria': ((json['criteria'] as Array<any>).map(DiscoveryCriterionFromJSON)),
+        'prompt': !exists(json, 'prompt') ? undefined : json['prompt'],
     };
 }
 
@@ -65,7 +72,8 @@ export function DiscoveryCriteriaToJSON(value?: DiscoveryCriteria | null): any {
     return {
         
         'confidence_threshold': value.confidence_threshold,
-        'criteria': ((value.criteria as Array<any>).map(ModelMetadataToJSON)),
+        'criteria': ((value.criteria as Array<any>).map(DiscoveryCriterionToJSON)),
+        'prompt': value.prompt,
     };
 }
 

package/src/models/DiscoveryCriterion.ts

@@ -0,0 +1,360 @@
+/* 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 {
+    HardwareRequirements,
+    HardwareRequirementsFromJSON,
+    HardwareRequirementsFromJSONTyped,
+    HardwareRequirementsToJSON,
+    ModelIO,
+    ModelIOFromJSON,
+    ModelIOFromJSONTyped,
+    ModelIOToJSON,
+    Task,
+    TaskFromJSON,
+    TaskFromJSONTyped,
+    TaskToJSON,
+} from './';
+
+/**
+ * 
+ * @export
+ * @interface DiscoveryCriterion
+ */
+export interface DiscoveryCriterion {
+    /**
+     * 
+     * @type {any}
+     * @memberof DiscoveryCriterion
+     */
+    annotations?: any | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    author?: string | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    bias_evaluation_score?: number | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    edge_optimized?: boolean | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof DiscoveryCriterion
+     */
+    finetuning_datasets?: Array<string> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    framework?: string | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    image?: string | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    inference_distributed?: boolean | null;
+    /**
+     * 
+     * @type {HardwareRequirements}
+     * @memberof DiscoveryCriterion
+     */
+    inference_hardware?: HardwareRequirements | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    inference_max_compute_utilization_percentage?: number | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    inference_max_energy_consumption_watts?: number | null;
+    /**
+     * Inference performance fields
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    inference_max_latency_ms?: number | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    inference_max_memory_usage_mb?: number | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    inference_min_throughput?: number | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    inference_precision?: string | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof DiscoveryCriterion
+     */
+    inference_software_dependencies?: Array<string> | null;
+    /**
+     * Arbitrary labels
+     * @type {Array<string>}
+     * @memberof DiscoveryCriterion
+     */
+    keywords?: Array<string> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    license?: string | null;
+    /**
+     * 
+     * @type {Array<ModelIO>}
+     * @memberof DiscoveryCriterion
+     */
+    model_inputs?: Array<ModelIO> | null;
+    /**
+     * 
+     * @type {Array<ModelIO>}
+     * @memberof DiscoveryCriterion
+     */
+    model_outputs?: Array<ModelIO> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    model_type?: string | null;
+    /**
+     * Architecture fields
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    multi_modal?: boolean | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    name?: string | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    pretrained?: boolean | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof DiscoveryCriterion
+     */
+    pretraining_datasets?: Array<string> | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    pruned?: boolean | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    quantization_aware?: boolean | null;
+    /**
+     * Regulatory and Compliance Fields
+     * A vector or strings that represent regulatory standards. Ex HIPPA
+     * @type {Array<string>}
+     * @memberof DiscoveryCriterion
+     */
+    regulatory?: Array<string> | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    slimmed?: boolean | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    supports_quantization?: boolean | null;
+    /**
+     * Inference Fields
+     * @type {Array<Task>}
+     * @memberof DiscoveryCriterion
+     */
+    task_types?: Array<Task> | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof DiscoveryCriterion
+     */
+    training_distributed?: boolean | null;
+    /**
+     * 
+     * @type {HardwareRequirements}
+     * @memberof DiscoveryCriterion
+     */
+    training_hardware?: HardwareRequirements | null;
+    /**
+     * Training performance fields
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    training_max_energy_consumption_watts?: number | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    training_precision?: string | null;
+    /**
+     * Training-related Fields
+     * @type {number}
+     * @memberof DiscoveryCriterion
+     */
+    training_time?: number | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof DiscoveryCriterion
+     */
+    version?: string | null;
+}
+
+export function DiscoveryCriterionFromJSON(json: any): DiscoveryCriterion {
+    return DiscoveryCriterionFromJSONTyped(json, false);
+}
+
+export function DiscoveryCriterionFromJSONTyped(json: any, ignoreDiscriminator: boolean): DiscoveryCriterion {
+    if ((json === undefined) || (json === null)) {
+        return json;
+    }
+    return {
+        
+        'annotations': !exists(json, 'annotations') ? undefined : json['annotations'],
+        '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']),
+        'inference_max_compute_utilization_percentage': !exists(json, 'inference_max_compute_utilization_percentage') ? undefined : json['inference_max_compute_utilization_percentage'],
+        'inference_max_energy_consumption_watts': !exists(json, 'inference_max_energy_consumption_watts') ? undefined : json['inference_max_energy_consumption_watts'],
+        'inference_max_latency_ms': !exists(json, 'inference_max_latency_ms') ? undefined : json['inference_max_latency_ms'],
+        'inference_max_memory_usage_mb': !exists(json, 'inference_max_memory_usage_mb') ? undefined : json['inference_max_memory_usage_mb'],
+        'inference_min_throughput': !exists(json, 'inference_min_throughput') ? undefined : json['inference_min_throughput'],
+        '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'],
+        '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)),
+        'model_type': !exists(json, 'model_type') ? undefined : json['model_type'],
+        'multi_modal': !exists(json, 'multi_modal') ? undefined : json['multi_modal'],
+        'name': !exists(json, 'name') ? undefined : json['name'],
+        'pretrained': !exists(json, 'pretrained') ? undefined : json['pretrained'],
+        'pretraining_datasets': !exists(json, 'pretraining_datasets') ? undefined : json['pretraining_datasets'],
+        'pruned': !exists(json, 'pruned') ? undefined : json['pruned'],
+        'quantization_aware': !exists(json, 'quantization_aware') ? undefined : json['quantization_aware'],
+        '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'] === 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'],
+    };
+}
+
+export function DiscoveryCriterionToJSON(value?: DiscoveryCriterion | null): any {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value === null) {
+        return null;
+    }
+    return {
+        
+        'annotations': value.annotations,
+        '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),
+        'inference_max_compute_utilization_percentage': value.inference_max_compute_utilization_percentage,
+        'inference_max_energy_consumption_watts': value.inference_max_energy_consumption_watts,
+        'inference_max_latency_ms': value.inference_max_latency_ms,
+        'inference_max_memory_usage_mb': value.inference_max_memory_usage_mb,
+        'inference_min_throughput': value.inference_min_throughput,
+        'inference_precision': value.inference_precision,
+        'inference_software_dependencies': value.inference_software_dependencies,
+        'keywords': value.keywords,
+        '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)),
+        'model_type': value.model_type,
+        'multi_modal': value.multi_modal,
+        'name': value.name,
+        'pretrained': value.pretrained,
+        'pretraining_datasets': value.pretraining_datasets,
+        'pruned': value.pruned,
+        'quantization_aware': value.quantization_aware,
+        'regulatory': value.regulatory,
+        'slimmed': value.slimmed,
+        'supports_quantization': value.supports_quantization,
+        '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,
+    };
+}
+