rewrite

type Action string

type Config []Rule

type Dependencies struct {
	LLMFactory    llm.Factory // NOTE: String() with cache.
	ObjectStorage object.Storage
}

type Factory component.Factory[Rewriter, config.App, Dependencies]

type Rewriter interface {
	component.Component
	config.Watcher

	// Labels applies rewrite rules to the given labels and returns the modified labels.
	// Note: this method modifies the input labels in place.
	// If a rule's action is ActionDropFeed, it returns nil to indicate the item should be dropped.
	Labels(ctx context.Context, labels model.Labels) (model.Labels, error)
}

type Rule struct {
	// If is the condition to check before applying the rule.
	// If not set, the rule will be applied.
	If []string

	// SourceLabel specifies which label's value to use as source text.
	// Default is model.LabelContent.
	SourceLabel string

	// SkipTooShortThreshold is the threshold of the source text length.
	// If the source text is shorter than this threshold, it will be skipped.
	SkipTooShortThreshold *int

	// Transform used to transform the source text.
	// If not set, transform to original source text.
	Transform *Transform

	// Match used to match the text after transform.
	// If not set, match all.
	Match string

	// Action determines what to do if matchs.
	Action Action

	// Label is the label to create or update.
	Label string
	// contains filtered or unexported fields
}

type Speaker struct {
	Name  string
	Role  string
	Voice string
}

type ToPodcast struct {
	LLM                        string
	EstimateMaximumDuration    time.Duration
	TranscriptAdditionalPrompt string
	TTSLLM                     string
	Speakers                   []Speaker
	// contains filtered or unexported fields
}

type ToText struct {
	Type ToTextType

	// LLM is the name of the LLM to use.
	// Only used when Type is ToTextTypePrompt.
	LLM string

	// Prompt is the prompt for LLM completion.
	// The source text will automatically be injected into the prompt.
	// Only used when Type is ToTextTypePrompt.
	Prompt string
	// contains filtered or unexported fields
}

type ToTextType string

type Transform struct {
	ToText    *ToText
	ToPodcast *ToPodcast
}