@mlhub/models-ts-sdk 0.1.1 → 0.2.0

package/src/models/CreateModelMetadataBody.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 CreateModelMetadataBody
+ */
+export interface CreateModelMetadataBody {
+    /**
+     * 
+     * @type {any}
+     * @memberof CreateModelMetadataBody
+     */
+    annotations?: any | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    author?: string | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    bias_evaluation_score?: number | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    edge_optimized?: boolean | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof CreateModelMetadataBody
+     */
+    finetuning_datasets?: Array<string> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    image?: string | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_distributed?: boolean | null;
+    /**
+     * 
+     * @type {HardwareRequirements}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_hardware?: HardwareRequirements;
+    /**
+     * 
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_max_compute_utilization_percentage?: number | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_max_energy_consumption_watts?: number | null;
+    /**
+     * Inference performance fields
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_max_latency_ms?: number | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_max_memory_usage_mb?: number | null;
+    /**
+     * 
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_min_throughput?: number | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_precision?: string | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof CreateModelMetadataBody
+     */
+    inference_software_dependencies?: Array<string> | null;
+    /**
+     * Arbitrary labels
+     * @type {Array<string>}
+     * @memberof CreateModelMetadataBody
+     */
+    keywords?: Array<string> | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof CreateModelMetadataBody
+     */
+    libraries?: Array<string> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    license?: string | null;
+    /**
+     * 
+     * @type {Array<ModelIO>}
+     * @memberof CreateModelMetadataBody
+     */
+    model_inputs?: Array<ModelIO> | null;
+    /**
+     * 
+     * @type {Array<ModelIO>}
+     * @memberof CreateModelMetadataBody
+     */
+    model_outputs?: Array<ModelIO> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    model_type?: string | null;
+    /**
+     * Architecture fields
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    multi_modal?: boolean | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    name?: string | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    pretrained?: boolean | null;
+    /**
+     * 
+     * @type {Array<string>}
+     * @memberof CreateModelMetadataBody
+     */
+    pretraining_datasets?: Array<string> | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    pruned?: boolean | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    quantization_aware?: boolean | null;
+    /**
+     * Regulatory and Compliance Fields
+     * A vector or strings that represent regulatory standards. Ex HIPPA
+     * @type {Array<string>}
+     * @memberof CreateModelMetadataBody
+     */
+    regulatory?: Array<string> | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    slimmed?: boolean | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    supports_quantization?: boolean | null;
+    /**
+     * Inference Fields
+     * @type {Array<Task>}
+     * @memberof CreateModelMetadataBody
+     */
+    task_types?: Array<Task> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    tenant_id?: string | null;
+    /**
+     * 
+     * @type {boolean}
+     * @memberof CreateModelMetadataBody
+     */
+    training_distributed?: boolean | null;
+    /**
+     * 
+     * @type {HardwareRequirements}
+     * @memberof CreateModelMetadataBody
+     */
+    training_hardware?: HardwareRequirements;
+    /**
+     * Training performance fields
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    training_max_energy_consumption_watts?: number | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof CreateModelMetadataBody
+     */
+    training_precision?: string | null;
+    /**
+     * Training-related Fields
+     * @type {number}
+     * @memberof CreateModelMetadataBody
+     */
+    training_time?: number | null;
+}
+
+export function CreateModelMetadataBodyFromJSON(json: any): CreateModelMetadataBody {
+    return CreateModelMetadataBodyFromJSONTyped(json, false);
+}
+
+export function CreateModelMetadataBodyFromJSONTyped(json: any, ignoreDiscriminator: boolean): CreateModelMetadataBody {
+    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'],
+        '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'],
+        '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': !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)),
+        'tenant_id': !exists(json, 'tenant_id') ? undefined : 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'],
+    };
+}
+
+export function CreateModelMetadataBodyToJSON(value?: CreateModelMetadataBody | 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,
+        '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,
+        '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,
+        '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/DiscoveryCriteria.ts

@@ -26,12 +26,6 @@ import {
  * @interface DiscoveryCriteria
  */
 export interface DiscoveryCriteria {
-    /**
-     * 
-     * @type {number}
-     * @memberof DiscoveryCriteria
-     */
-    confidence_threshold?: number | null;
     /**
      * 
      * @type {Array<DiscoveryCriterion>}
@@ -56,7 +50,6 @@ 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(DiscoveryCriterionFromJSON)),
         'prompt': !exists(json, 'prompt') ? undefined : json['prompt'],
     };
@@ -71,7 +64,6 @@ export function DiscoveryCriteriaToJSON(value?: DiscoveryCriteria | null): any {
     }
     return {
         
-        'confidence_threshold': value.confidence_threshold,
         'criteria': ((value.criteria as Array<any>).map(DiscoveryCriterionToJSON)),
         'prompt': value.prompt,
     };

package/src/models/Locator.ts

@@ -0,0 +1,56 @@
+/* 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 Locator
+ */
+export interface Locator {
+    /**
+     * 
+     * @type {string}
+     * @memberof Locator
+     */
+    url: string;
+}
+
+export function LocatorFromJSON(json: any): Locator {
+    return LocatorFromJSONTyped(json, false);
+}
+
+export function LocatorFromJSONTyped(json: any, ignoreDiscriminator: boolean): Locator {
+    if ((json === undefined) || (json === null)) {
+        return json;
+    }
+    return {
+        
+        'url': json['url'],
+    };
+}
+
+export function LocatorToJSON(value?: Locator | null): any {
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value === null) {
+        return null;
+    }
+    return {
+        
+        'url': value.url,
+    };
+}
+

package/src/models/ModelMetadata.ts

@@ -14,6 +14,10 @@
 
 import { exists, mapValues } from '../runtime';
 import {
+    Canonical,
+    CanonicalFromJSON,
+    CanonicalFromJSONTyped,
+    CanonicalToJSON,
     HardwareRequirements,
     HardwareRequirementsFromJSON,
     HardwareRequirementsFromJSONTyped,
@@ -52,6 +56,12 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     bias_evaluation_score?: number | null;
+    /**
+     * 
+     * @type {Canonical}
+     * @memberof ModelMetadata
+     */
+    canonical?: Canonical;
     /**
      * 
      * @type {boolean}
@@ -221,6 +231,12 @@ export interface ModelMetadata {
      * @memberof ModelMetadata
      */
     task_types?: Array<Task> | null;
+    /**
+     * 
+     * @type {string}
+     * @memberof ModelMetadata
+     */
+    tenant_id?: string | null;
     /**
      * 
      * @type {boolean}
@@ -266,6 +282,7 @@ export function ModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boole
         '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'],
+        'canonical': !exists(json, 'canonical') ? undefined : CanonicalFromJSON(json['canonical']),
         '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'],
@@ -294,6 +311,7 @@ export function ModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boole
         '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)),
+        'tenant_id': !exists(json, 'tenant_id') ? undefined : 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'],
@@ -314,6 +332,7 @@ export function ModelMetadataToJSON(value?: ModelMetadata | null): any {
         'annotations': value.annotations,
         'author': value.author,
         'bias_evaluation_score': value.bias_evaluation_score,
+        'canonical': CanonicalToJSON(value.canonical),
         'edge_optimized': value.edge_optimized,
         'finetuning_datasets': value.finetuning_datasets,
         'image': value.image,
@@ -342,6 +361,7 @@ export function ModelMetadataToJSON(value?: ModelMetadata | null): any {
         '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)),
+        '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,

package/src/models/index.ts

@@ -7,9 +7,11 @@ export * from './ArtifactIngestionStatus';
 export * from './ArtifactPublication';
 export * from './ArtifactPublicationStatus';
 export * from './ArtifactType';
-export * from './AssociateModelMetadata';
+export * from './AssociateModelMetadataBody';
 export * from './AssociateModelMetadataResponse';
 export * from './BadRequestResponse';
+export * from './Canonical';
+export * from './CreateModelMetadataBody';
 export * from './CreateModelMetadataResponse';
 export * from './DiscoverModelsByPlatformResponse';
 export * from './DiscoverModelsResponse';
@@ -31,6 +33,7 @@ export * from './ListModelsByPlatformResponse';
 export * from './ListModelsResponse';
 export * from './ListPlatformsResponse';
 export * from './ListTasksResponse';
+export * from './Locator';
 export * from './ModelArtifact';
 export * from './ModelIO';
 export * from './ModelMetadata';

package/dist/models/AssociateModelMetadata.d.ts

@@ -1,33 +0,0 @@
-/**
- * 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.
- */
-/**
- *
- * @export
- * @interface AssociateModelMetadata
- */
-export interface AssociateModelMetadata {
-    /**
-     *
-     * @type {string}
-     * @memberof AssociateModelMetadata
-     */
-    author: string;
-    /**
-     *
-     * @type {string}
-     * @memberof AssociateModelMetadata
-     */
-    name: string;
-}
-export declare function AssociateModelMetadataFromJSON(json: any): AssociateModelMetadata;
-export declare function AssociateModelMetadataFromJSONTyped(json: any, ignoreDiscriminator: boolean): AssociateModelMetadata;
-export declare function AssociateModelMetadataToJSON(value?: AssociateModelMetadata | null): any;