packaging

type JsonSchema struct {
	// Properties configuration
	Properties []Property `json:"properties"`
	// List of required properties
	Required []string `json:"required,omitempty"`
}

type K8sPackager struct {
	// Connection where a trained model artifact is stored
	ModelHolder *connection.Connection `json:"modelHolder"`
	// Model Packaging
	ModelPackaging *ModelPackaging `json:"modelPackaging"`
	// Packaging integration
	PackagingIntegration *PackagingIntegration `json:"packagingIntegration"`
	// Name of trained model artifact name
	TrainingZipName string `json:"trainingZipName"`
	// List of targets with appropriate connections
	Targets []PackagerTarget `json:"targets"`
}

type ModelPackaging struct {
	// Model packaging id
	ID string `json:"id"`
	// Deletion mark
	DeletionMark bool `json:"deletionMark,omitempty" swaggerignore:"true"`
	// CreatedAt
	CreatedAt time.Time `json:"createdAt,omitempty"`
	// UpdatedAt
	UpdatedAt time.Time `json:"updatedAt,omitempty"`
	// Model packaging specification
	Spec ModelPackagingSpec `json:"spec,omitempty"`
	// Model packaging status
	Status v1alpha1.ModelPackagingStatus `json:"status,omitempty"`
}

type ModelPackagingSpec struct {
	// Training output artifact name
	ArtifactName string `json:"artifactName"`
	// Packaging integration ID
	IntegrationName string `json:"integrationName"`
	// Image name. Packaging integration image will be used if this parameters is missed
	Image string `json:"image,omitempty"`
	// List of arguments. This parameter depends on the specific packaging integration
	Arguments map[string]interface{} `json:"arguments"`
	// List of targets. This parameter depends on the specific packaging integration
	Targets []v1alpha1.Target `json:"targets"`
	// Resources for packager container
	// The same format like k8s uses for pod resources.
	Resources *v1alpha1.ResourceRequirements `json:"resources,omitempty"`
	// Name of Connection to storage where a packager obtain a model trained artifact.
	// Permitted connection types are defined by specific PackagingIntegration
	OutputConnection string `json:"outputConnection,omitempty"`
	// Node selector for specifying a node pool
	NodeSelector map[string]string `json:"nodeSelector,omitempty"`
}

type PackagerTarget struct {
	// Target name
	Name string `json:"name"`
	// A Connection for this target
	Connection connection.Connection `json:"connection"`
}

type PackagingIntegration struct {
	// Packaging integration id
	ID string `json:"id"`
	// CreatedAt
	CreatedAt time.Time `json:"createdAt,omitempty"`
	// UpdatedAt
	UpdatedAt time.Time `json:"updatedAt,omitempty"`
	// Packaging integration specification
	Spec PackagingIntegrationSpec `json:"spec,omitempty"`
	// Packaging integration status
	Status v1alpha1.PackagingIntegrationStatus `json:"status,omitempty"`
}

type PackagingIntegrationSpec struct {
	// Path to binary which starts a packaging process
	Entrypoint string `json:"entrypoint"`
	// Default packaging Docker image
	DefaultImage string `json:"defaultImage"`
	// Enable docker privileged flag
	Privileged bool `json:"privileged,omitempty"`
	// Schema which describes targets and arguments for specific packaging integration
	Schema Schema `json:"schema"`
}

type Parameter struct {
	// Parameter name
	Name string `json:"name"`
	// Parameter value
	Value interface{} `json:"value"`
}

type Property struct {
	// Property name
	Name string `json:"name"`
	// List of property parameters
	Parameters []Parameter `json:"parameters"`
}

type Schema struct {
	// Targets schema
	Targets []v1alpha1.TargetSchema `json:"targets"`
	// Arguments schema
	Arguments JsonSchema `json:"arguments"`
}