package domain

// TrackingConfig specifies experiment tracking tools to enable for a task.
type TrackingConfig struct {
	MLflow      *MLflowTrackingConfig      `json:"mlflow,omitempty"`
	TensorBoard *TensorBoardTrackingConfig `json:"tensorboard,omitempty"`
	Wandb       *WandbTrackingConfig       `json:"wandb,omitempty"`
}

// MLflowTrackingConfig controls MLflow integration.
type MLflowTrackingConfig struct {
	Mode        string `json:"mode,omitempty"`
	TrackingURI string `json:"tracking_uri,omitempty"`
	Enabled     bool   `json:"enabled"`
}

// TensorBoardTrackingConfig controls TensorBoard integration.
type TensorBoardTrackingConfig struct {
	Mode    string `json:"mode,omitempty"`
	Enabled bool   `json:"enabled"`
}

// WandbTrackingConfig controls Weights & Biases integration.
type WandbTrackingConfig struct {
	Mode    string `json:"mode,omitempty"`
	APIKey  string `json:"api_key,omitempty"`
	Project string `json:"project,omitempty"`
	Entity  string `json:"entity,omitempty"`
	Enabled bool   `json:"enabled"`
}
