schema

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

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"`
	JobStatuses  []status.JobStatus `json:"job_statuses,omitempty"`
	APIVersions  []APIVersion       `json:"api_versions,omitempty"`
}

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

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

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

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

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

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

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

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

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

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

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 struct {
	Shape []interface{} `json:"shape"`
	Type  string        `json:"type"`
}

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

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

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

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 struct {
	Message string `json:"message"`
}

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

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"`
}