predict

type HealthcheckResponse struct {
	Status string `json:"status"`
}

type HelpResponse struct {
	Arguments map[string]*config.RunArgument `json:"arguments"`
}

type Input struct {
	String *string
	File   *string
	Array  *[]any
	Json   *json.RawMessage
	Float  *float32
	Int    *int32
}

type Inputs map[string]Input

type Predictor struct {
	// contains filtered or unexported fields
}

type PredictorOptions struct {
	RunOptions    command.RunOptions
	IsTrain       bool
	Docker        command.Command
	WeightManager *weights.Manager
}

type Request struct {
	// TODO: could this be Inputs?
	Input   map[string]any `json:"input"`
	Context RequestContext `json:"context"`
}

type RequestContext struct {
	ReplicateAPIToken string `json:"replicate_api_token,omitempty"`
}

type Response struct {
	Status  status         `json:"status"`
	Output  *any           `json:"output"`
	Error   string         `json:"error"`
	Metrics map[string]any `json:"metrics,omitempty"`
}

type ValidationErrorResponse struct {
	Detail []struct {
		Location []string `json:"loc"`
		Message  string   `json:"msg"`
		Type     string   `json:"type"`
	} `json:"detail"`
}