common

type BasicRunConfig struct {
	InputFile   string
	InputBucket string
	MountPoint  string
	Ctx         context.Context
	// contains filtered or unexported fields
}

type BasicStepConfig struct {
	BasicRunConfig *BasicRunConfig
	StepKey        string
	StepLogic      func() (string, error)
}

type CommandStepConfig struct {
	BasicStepConfig
	CommandPath        string
	ArgsStringTemplate string
	CommandLogic       func(string) (string, error)
}

type GenAICounter struct {
	InputCounter  metric.Int64Counter
	OutputCounter metric.Int64Counter
	RetryCounter  metric.Int64Counter
}

type GenaiRunConfig struct {
	BasicRunConfig
	CloudConfig        *cloud.Config
	AgentModels        map[string]*cloud.QuotaAwareGenerativeAIModel
	TemplateService    *cloud.TemplateService
	Meter              metric.Meter
	GenAIClient        *genai.Client
	GenAIContentCaches map[string]*genai.CachedContent
	BigQueryClient     *bigquery.Client
	GenAIEmbedding     *genai.Models
}

type GenaiStepConfig struct {
	BasicStepConfig
	GenaiRunConfig *GenaiRunConfig
	Counters       *GenAICounter
}

type RunConfig struct {
	InputFile       string
	InputBucket     string
	StepMetadataKey string
	MountPoint      string
	Ctx             context.Context
}

type StepStatus struct {
	Output string `json:"output"`
	Status string `json:"status"`
}