manifest

type Defaults struct {
	SDKVersion string `yaml:"sdk_version"`
	CogVersion string `yaml:"cog_version"`
}

type Expectation struct {
	Type    string         `yaml:"type"`
	Value   any            `yaml:"value"`
	Pattern string         `yaml:"pattern"`
	Mime    string         `yaml:"mime"`
	Match   map[string]any `yaml:"match"`
	Keys    []string       `yaml:"keys"`
}

type Manifest struct {
	Defaults Defaults `yaml:"defaults"`
	Models   []Model  `yaml:"models"`
}

type Model struct {
	Name                 string            `yaml:"name"`
	Repo                 string            `yaml:"repo"`
	Path                 string            `yaml:"path"`
	GPU                  bool              `yaml:"gpu"`
	Timeout              int               `yaml:"timeout"`
	RequiresEnv          []string          `yaml:"requires_env"`
	RequiresTools        []string          `yaml:"requires_tools"`
	Env                  map[string]string `yaml:"env"`
	SDKVersion           string            `yaml:"sdk_version"`
	CogYAMLOverrides     map[string]any    `yaml:"cog_yaml_overrides"`
	Setup                []string          `yaml:"setup"`
	SkipSchemaValidation bool              `yaml:"skip_schema_validation"`
	Tests                []TestCase        `yaml:"tests"`
	TrainTests           []TestCase        `yaml:"train_tests"`
}

type TestCase struct {
	Description string         `yaml:"description"`
	Inputs      map[string]any `yaml:"inputs"`
	Expect      Expectation    `yaml:"expect"`
}