loader

type BaseModelLoader interface {
	// LoadBaseModels loads base models from the given repo and environment.
	LoadBaseModels(ctx context.Context, repo string, env models.Environment) ([]models.BaseModel, error)
}

type Composite interface {
	DatasetLoader
	BaseModelLoader
	ImportedModelLoader
	GPUPoolLoader
	GPUNodeLoader
	DedicatedAIClusterLoader
	TenancyOverrideLoader
	RegionalOverrideLoader
}

type DatasetLoader interface {
	// LoadDataset loads a dataset from the given repo and environment.
	LoadDataset(ctx context.Context, repo string, env models.Environment) (*models.Dataset, error)
}

type DedicatedAIClusterLoader interface {
	// LoadDedicatedAIClusters loads dedicated AI clusters from the given kube config and environment.
	LoadDedicatedAIClusters(ctx context.Context, kubeCfg string, env models.Environment) (map[string][]models.DedicatedAICluster, error)
}

type GPUNodeLoader interface {
	// LoadGPUNodesByPool loads GPU nodes from the given kube config and environment.
	LoadGPUNodesByPool(ctx context.Context, kubeCfg string, env models.Environment) (map[string][]models.GPUNode, error)
}

type GPUPoolLoader interface {
	// LoadGPUPools loads GPU pools from the given repo and environment.
	LoadGPUPools(ctx context.Context, repo string, env models.Environment) ([]models.GPUPool, error)
}

type ImportedModelLoader interface {
	// LoadImportedModels loads imported models from the given kube config and environment.
	// Returns a tenant-keyed map (raw TenantID, or `"UNKNOWN_TENANCY"` for orphans);
	// re-keying by Tenant.Name is handled by Dataset.SetImportedModelMap.
	LoadImportedModels(ctx context.Context, kubeCfg string, env models.Environment) (map[string][]models.ImportedModel, error)
}

type RegionalOverrideLoader interface {
	LoadLimitRegionalOverrides(ctx context.Context, repo string, env models.Environment) ([]models.LimitRegionalOverride, error)
	LoadConsolePropertyRegionalOverrides(ctx context.Context, repo string, env models.Environment) ([]models.ConsolePropertyRegionalOverride, error)
	LoadPropertyRegionalOverrides(ctx context.Context, repo string, env models.Environment) ([]models.PropertyRegionalOverride, error)
}

type TenancyOverrideLoader interface {
	LoadTenancyOverrideGroup(ctx context.Context, repo string, env models.Environment) (models.TenancyOverrideGroup, error)
}