workflow

type FileInfo struct {
	Name string
	Data []byte
}

type Runner interface {
	Next(ctx context.Context) (*WorkflowStepResult, error)
}

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

type RunnerMock struct {
	// NextFunc mocks the Next method.
	NextFunc func(ctx context.Context) (*WorkflowStepResult, error)
	// contains filtered or unexported fields
}

type StartWorklfowRequest struct {
	Variables   map[string]any `json:"variables"`
	Model       string         `json:"model"`
	ImageModel  string         `json:"image_model"`
	Temperature float64        `json:"temperature"`
}

type StepContext struct {
	Table  string `json:"table"`
	Column string `json:"column"`
}

type Workflow struct {
	ID          string                    `json:"id"`
	Name        string                    `json:"name"`
	Description string                    `json:"description"`
	Variables   []schema.WorkflowVariable `json:"variables"`
	Steps       []schema.WorkflowStep     `json:"steps"`
}

type WorkflowAction string

type WorkflowAutofillPayload struct {
	Count          any      `json:"count"`
	Batch          any      `json:"batch"`
	Table          string   `json:"table"`
	Columns        []string `json:"columns"`
	ContextColumns []string `json:"context_columns"`
}

type WorkflowCreateColumnParams struct {
	Table       string `json:"table"`
	Name        string `json:"name"`
	Description string `json:"description"`
	Type        string `json:"type"`
}

type WorkflowCreateTablePayload struct {
	SchemaFile string                       `json:"schema_file"`
	OnExists   schema.WorkflowTableOnExists `json:"on_exists"`
	Request    table.TableGenRequest        `json:"request"`
	Datasets   []string                     `json:"datasets"`
}

type WorkflowDeleteColumnParams struct {
	Table  string `json:"table"`
	Column string `json:"column"`
}

type WorkflowDeleteTableParams struct {
	Table string `json:"table"`
}

type WorkflowExportTableParams struct {
	Table string `json:"table"`
	Path  string `json:"path"`
}

type WorkflowGeneratePayload struct {
	Count any    `json:"count"`
	Batch any    `json:"batch"`
	Table string `json:"table"`
}

type WorkflowImportFileParams struct {
	Table    string `json:"table"`
	Name     string `json:"name"`
	File     string `json:"file"`
	Prompt   string `json:"prompt"`
	Truncate bool   `json:"truncate"`
}

type WorkflowService interface {
	Get(ctx context.Context, wf string) (*ent.Workflow, error)
	Delete(ctx context.Context, wf string) error
	Create(ctx context.Context, wf *Workflow) (string, error)
	Update(ctx context.Context, id string, wf *Workflow) (string, error)
	Start(ctx context.Context, id string, request StartWorklfowRequest) (Runner, error)
	List(ctx context.Context) ([]*ent.Workflow, error)
}

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

type WorkflowServiceMock struct {
	// CreateFunc mocks the Create method.
	CreateFunc func(ctx context.Context, wf *Workflow) (string, error)

	// DeleteFunc mocks the Delete method.
	DeleteFunc func(ctx context.Context, wf string) error

	// GetFunc mocks the Get method.
	GetFunc func(ctx context.Context, wf string) (*ent.Workflow, error)

	// ListFunc mocks the List method.
	ListFunc func(ctx context.Context) ([]*ent.Workflow, error)

	// StartFunc mocks the Start method.
	StartFunc func(ctx context.Context, id string, request StartWorklfowRequest) (Runner, error)

	// UpdateFunc mocks the Update method.
	UpdateFunc func(ctx context.Context, id string, wf *Workflow) (string, error)
	// contains filtered or unexported fields
}

type WorkflowSimple struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Description string `json:"description"`
}

type WorkflowStepResult struct {
	Action     WorkflowAction
	Message    string
	ExportData string
	ExportPath string
	Generator  table.RowsGenerator
}