@mlhub/models-ts-sdk 0.14.0 → 0.16.0

package/src/models/ModelMetadata.ts

@@ -22,14 +22,6 @@ import {
     DeploymentStrategyReferenceFromJSON,
     DeploymentStrategyReferenceFromJSONTyped,
     DeploymentStrategyReferenceToJSON,
-    HardwareRequirements,
-    HardwareRequirementsFromJSON,
-    HardwareRequirementsFromJSONTyped,
-    HardwareRequirementsToJSON,
-    ModelIO,
-    ModelIOFromJSON,
-    ModelIOFromJSONTyped,
-    ModelIOToJSON,
     Task,
     TaskFromJSON,
     TaskFromJSONTyped,
@@ -42,24 +34,12 @@ import {
  * @interface ModelMetadata
  */
 export interface ModelMetadata {
-    /**
-     * 
-     * @type {any}
-     * @memberof ModelMetadata
-     */
-    annotations?: any | null;
     /**
      * 
      * @type {string}
      * @memberof ModelMetadata
      */
     author: string;
-    /**
-     * 
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    bias_evaluation_score?: number | null;
     /**
      * 
      * @type {Canonical}
@@ -78,78 +58,6 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     description?: string | null;
-    /**
-     * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    edge_optimized?: boolean | null;
-    /**
-     * 
-     * @type {Array<string>}
-     * @memberof ModelMetadata
-     */
-    finetuning_datasets?: Array<string> | null;
-    /**
-     * 
-     * @type {string}
-     * @memberof ModelMetadata
-     */
-    image?: string | null;
-    /**
-     * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    inference_distributed?: boolean | null;
-    /**
-     * 
-     * @type {HardwareRequirements}
-     * @memberof ModelMetadata
-     */
-    inference_hardware?: HardwareRequirements;
-    /**
-     * 
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    inference_max_compute_utilization_percentage?: number | null;
-    /**
-     * 
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    inference_max_energy_consumption_watts?: number | null;
-    /**
-     * Inference performance fields
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    inference_max_latency_ms?: number | null;
-    /**
-     * 
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    inference_max_memory_usage_mb?: number | null;
-    /**
-     * 
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    inference_min_throughput?: number | null;
-    /**
-     * 
-     * @type {string}
-     * @memberof ModelMetadata
-     */
-    inference_precision?: string | null;
-    /**
-     * 
-     * @type {Array<string>}
-     * @memberof ModelMetadata
-     */
-    inference_software_dependencies?: Array<string> | null;
     /**
      * 
      * @type {Array<string>}
@@ -162,30 +70,12 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     license?: string | null;
-    /**
-     * 
-     * @type {Array<ModelIO>}
-     * @memberof ModelMetadata
-     */
-    model_inputs?: Array<ModelIO> | null;
-    /**
-     * 
-     * @type {Array<ModelIO>}
-     * @memberof ModelMetadata
-     */
-    model_outputs?: Array<ModelIO> | null;
     /**
      * 
      * @type {string}
      * @memberof ModelMetadata
      */
     model_type?: string | null;
-    /**
-     * Architecture fields
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    multi_modal?: boolean | null;
     /**
      * 
      * @type {string}
@@ -194,55 +84,18 @@ export interface ModelMetadata {
     name: string;
     /**
      * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    pretrained?: boolean | null;
-    /**
-     * 
-     * @type {Array<string>}
-     * @memberof ModelMetadata
-     */
-    pretraining_datasets?: Array<string> | null;
-    /**
-     * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    pruned?: boolean | null;
-    /**
-     * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    quantization_aware?: boolean | null;
-    /**
-     * Regulatory and Compliance Fields
-     * A vector or strings that represent regulatory standards. Ex HIPPA
      * @type {Array<string>}
      * @memberof ModelMetadata
      */
     regulatory?: Array<string> | null;
     /**
      * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    slimmed?: boolean | null;
-    /**
-     * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    supports_quantization?: boolean | null;
-    /**
-     * Arbitrary labels
      * @type {Array<string>}
      * @memberof ModelMetadata
      */
     tags?: Array<string> | null;
     /**
-     * Inference Fields
+     * 
      * @type {Array<Task>}
      * @memberof ModelMetadata
      */
@@ -253,36 +106,6 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     tenant_id: string;
-    /**
-     * 
-     * @type {boolean}
-     * @memberof ModelMetadata
-     */
-    training_distributed?: boolean | null;
-    /**
-     * 
-     * @type {HardwareRequirements}
-     * @memberof ModelMetadata
-     */
-    training_hardware?: HardwareRequirements;
-    /**
-     * Training performance fields
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    training_max_energy_consumption_watts?: number | null;
-    /**
-     * 
-     * @type {string}
-     * @memberof ModelMetadata
-     */
-    training_precision?: string | null;
-    /**
-     * Training-related Fields
-     * @type {number}
-     * @memberof ModelMetadata
-     */
-    training_time?: number | null;
 }
 
 export function ModelMetadataFromJSON(json: any): ModelMetadata {
@@ -295,46 +118,18 @@ export function ModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boole
     }
     return {
         
-        'annotations': !exists(json, 'annotations') ? undefined : json['annotations'],
         'author': json['author'],
-        'bias_evaluation_score': !exists(json, 'bias_evaluation_score') ? undefined : json['bias_evaluation_score'],
         'canonical': !exists(json, 'canonical') ? undefined : CanonicalFromJSON(json['canonical']),
         'deployment_strategy_refs': ((json['deployment_strategy_refs'] as Array<any>).map(DeploymentStrategyReferenceFromJSON)),
         'description': !exists(json, 'description') ? undefined : json['description'],
-        'edge_optimized': !exists(json, 'edge_optimized') ? undefined : json['edge_optimized'],
-        'finetuning_datasets': !exists(json, 'finetuning_datasets') ? undefined : json['finetuning_datasets'],
-        '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'],
         '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)),
         'model_type': !exists(json, 'model_type') ? undefined : json['model_type'],
-        'multi_modal': !exists(json, 'multi_modal') ? undefined : json['multi_modal'],
         'name': 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'],
         'tags': !exists(json, 'tags') ? undefined : json['tags'],
         'task_types': !exists(json, 'task_types') ? undefined : (json['task_types'] === null ? null : (json['task_types'] as Array<any>).map(TaskFromJSON)),
         'tenant_id': json['tenant_id'],
-        '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'],
     };
 }
 
@@ -347,46 +142,18 @@ export function ModelMetadataToJSON(value?: ModelMetadata | null): any {
     }
     return {
         
-        'annotations': value.annotations,
         'author': value.author,
-        'bias_evaluation_score': value.bias_evaluation_score,
         'canonical': CanonicalToJSON(value.canonical),
         'deployment_strategy_refs': ((value.deployment_strategy_refs as Array<any>).map(DeploymentStrategyReferenceToJSON)),
         'description': value.description,
-        'edge_optimized': value.edge_optimized,
-        'finetuning_datasets': value.finetuning_datasets,
-        '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,
         '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)),
         '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,
         'tags': value.tags,
         'task_types': value.task_types === undefined ? undefined : (value.task_types === null ? null : (value.task_types as Array<any>).map(TaskToJSON)),
         'tenant_id': value.tenant_id,
-        '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,
     };
 }
 

package/src/models/index.ts

@@ -18,6 +18,7 @@ export * from './DiscoverModelsByPlatformResponse';
 export * from './DiscoverModelsResponse';
 export * from './DiscoveryCriteria';
 export * from './DiscoveryCriterion';
+export * from './ForkModelResponse';
 export * from './GetModelArtifactResponse';
 export * from './GetModelByPlatformResponse';
 export * from './GetModelIngestionResponse';