schema

type APISummary struct {
	Message         string                     `json:"message"`
	ModelSignatures map[string]InputSignatures `json:"model_signatures"`
}

type BatchAPI struct {
	Spec        spec.API           `json:"spec"`
	JobStatuses []status.JobStatus `json:"job_statuses"`
	Endpoint    string             `json:"endpoint"`
}

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

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

type DeployResponse struct {
	Results []DeployResult `json:"results"`
}

type DeployResult struct {
	API     *spec.API
	Message string
	Error   string
}

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

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

type GetAPIResponse struct {
	RealtimeAPI     *RealtimeAPI     `json:"realtime_api"`
	BatchAPI        *BatchAPI        `json:"batch_api"`
	TrafficSplitter *TrafficSplitter `json:"traffic_splitter"`
}

type GetAPIsResponse struct {
	RealtimeAPIs     []RealtimeAPI     `json:"realtime_apis"`
	BatchAPIs        []BatchAPI        `json:"batch_apis"`
	TrafficSplitters []TrafficSplitter `json:"traffic_splitters"`
}

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

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 InputSignatures map[string]InputSignature

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

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 RealtimeAPI struct {
	Spec         spec.API        `json:"spec"`
	Status       status.Status   `json:"status"`
	Metrics      metrics.Metrics `json:"metrics"`
	Endpoint     string          `json:"endpoint"`
	DashboardURL string          `json:"dashboard_url"`
}

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 TrafficSplitter struct {
	Spec     spec.API `json:"spec"`
	Endpoint string   `json:"endpoint"`
}