schema

Documentation

Index

• Constants
• type APIModelSummary
• type APIResponse
• type APITFLiveReloadingSummary
• type APIVersion
• type BatchJobResponse
• type BatchJobSubmission
• type DeleteResponse
• type DelimitedFiles
• type DeployResult
• type ErrorResponse
• type FilePathLister
• type GCPLogsResponse
• type GenericModelMetadata
• type InfoGCPResponse
• type InfoResponse
• type InputSignature
• type ItemList
• type NodeInfo
• type RefreshResponse
• type S3Lister
• type TFModelIDMetadata
• type TaskJobResponse
• type TaskJobSubmission
• type VerifyCortexResponse

Constants

const (
	// Job Submission
	BatchSizeKey          = "batch_size"
	ItemsKey              = "items"
	ItemListKey           = "item_list"
	FilePathListerKey     = "file_path_lister"
	DelimitedFilesKey     = "delimited_files"
	S3PathsKey            = "s3_paths"
	IncludesKey           = "includes"
	ExcludesKey           = "excludes"
	WorkersKey            = "workers"
	TimeoutKey            = "timeout"
	MaxReceiveCountKey    = "max_receive_count"
	ARNKey                = "arn"
	SQSDeadLetterQueueKey = "sqs_dead_letter_queue"
)

Types

type APIModelSummary

type APIModelSummary struct {
	Message       string                          `json:"message"`
	ModelMetadata map[string]GenericModelMetadata `json:"model_metadata"`
}

type APIResponse

type APIResponse struct {
	Spec             spec.API                `json:"spec"`
	Status           *status.Status          `json:"status,omitempty"`
	Metrics          *metrics.Metrics        `json:"metrics,omitempty"`
	Endpoint         string                  `json:"endpoint"`
	DashboardURL     *string                 `json:"dashboard_url,omitempty"`
	BatchJobStatuses []status.BatchJobStatus `json:"batch_job_statuses,omitempty"`
	TaskJobStatuses  []status.TaskJobStatus  `json:"task_job_statuses,omitempty"`
	APIVersions      []APIVersion            `json:"api_versions,omitempty"`
}

type APITFLiveReloadingSummary

type APITFLiveReloadingSummary struct {
	Message       string                       `json:"message"`
	ModelMetadata map[string]TFModelIDMetadata `json:"model_metadata"`
}

type APIVersion

type APIVersion struct {
	APIID       string `json:"api_id"`
	LastUpdated int64  `json:"last_updated"`
}

type BatchJobResponse

type BatchJobResponse struct {
	APISpec   spec.API              `json:"api_spec"`
	JobStatus status.BatchJobStatus `json:"job_status"`
	Endpoint  string                `json:"endpoint"`
}

type BatchJobSubmission

type BatchJobSubmission struct {
	spec.RuntimeBatchJobConfig
	ItemList       *ItemList       `json:"item_list"`
	FilePathLister *FilePathLister `json:"file_path_lister"`
	DelimitedFiles *DelimitedFiles `json:"delimited_files"`
}

type DeleteResponse

type DeleteResponse struct {
	Message string `json:"message"`
}

type DelimitedFiles

type DelimitedFiles struct {
	S3Lister
	BatchSize int `json:"batch_size"`
}

type DeployResult

type DeployResult struct {
	API     *APIResponse `json:"api"`
	Message string       `json:"message"`
	Error   string       `json:"error"`
}

type ErrorResponse

type ErrorResponse struct {
	Kind    string `json:"kind"`
	Message string `json:"message"`
}

type FilePathLister

type FilePathLister struct {
	S3Lister
	BatchSize int `json:"batch_size"`
}

type GCPLogsResponse

type GCPLogsResponse struct {
	QueryParams map[string]string `json:"query_params"`
}

type GenericModelMetadata

type GenericModelMetadata struct {
	Versions   []string `json:"versions"`
	Timestamps []int64  `json:"timestamps"`
}

type InfoGCPResponse

type InfoGCPResponse struct {
	ClusterConfig clusterconfig.InternalGCPConfig `json:"cluster_config"`
}

type InfoResponse

type InfoResponse struct {
	MaskedAWSAccessKeyID string                       `json:"masked_aws_access_key_id"`
	ClusterConfig        clusterconfig.InternalConfig `json:"cluster_config"`
	NodeInfos            []NodeInfo                   `json:"node_infos"`
	NumPendingReplicas   int                          `json:"num_pending_replicas"`
}

type InputSignature

type InputSignature struct {
	Shape []interface{} `json:"shape"`
	Type  string        `json:"type"`
}

type ItemList

type ItemList struct {
	Items     []json.RawMessage `json:"items"`
	BatchSize int               `json:"batch_size"`
}

type NodeInfo

type NodeInfo struct {
	Name                 string             `json:"name"`
	InstanceType         string             `json:"instance_type"`
	IsSpot               bool               `json:"is_spot"`
	Price                float64            `json:"price"`
	NumReplicas          int                `json:"num_replicas"`
	ComputeUserCapacity  userconfig.Compute `json:"compute_user_capacity"`  // the total resources available to the user on a node
	ComputeAvailable     userconfig.Compute `json:"compute_available"`      // unused resources on a node
	ComputeUserRequested userconfig.Compute `json:"compute_user_requested"` // total resources requested by user on a node
}

type RefreshResponse

type RefreshResponse struct {
	Message string `json:"message"`
}

type S3Lister

type S3Lister struct {
	S3Paths  []string `json:"s3_paths"` // s3://<bucket_name>/key
	Includes []string `json:"includes"`
	Excludes []string `json:"excludes"`
}

type TFModelIDMetadata

type TFModelIDMetadata struct {
	DiskPath        string                    `json:"disk_path"`
	SignatureKey    string                    `json:"signature_key"`
	InputSignatures map[string]InputSignature `json:"input_signatures"`
	Timestamp       int64                     `json:"timestamp"`
	SignatureDef    map[string]interface{}    `json:"signature_def"`
}

type TaskJobResponse

type TaskJobResponse struct {
	APISpec   spec.API             `json:"api_spec"`
	JobStatus status.TaskJobStatus `json:"job_status"`
	Endpoint  string               `json:"endpoint"`
}

type TaskJobSubmission

type TaskJobSubmission struct {
	spec.RuntimeTaskJobConfig
}

type VerifyCortexResponse

type VerifyCortexResponse struct {
	Provider types.ProviderType `json:"provider"`
}