prompts

type AIMessagePromptTemplate struct {
	Prompt PromptTemplate
}

type ChatPromptTemplate struct {
	// Messages is the list of the messages to be formatted.
	Messages []MessageFormatter

	// PartialVariables represents a map of variable names to values or functions that return values.
	// If the value is a function, it will be called when the prompt template is rendered.
	PartialVariables map[string]any
}

type ChatPromptValue []schema.ChatMessage

type ExampleSelector interface {
	AddExample(example map[string]string) string
	SelectExamples(inputVariables map[string]string) []map[string]string
}

type FewShotPrompt struct {
	// Examples to format into the prompt. Either this or ExamplePrompt should be provided.
	Examples []map[string]string
	// ExampleSelector to choose the examples to format into the prompt. Either this or Examples should be provided.
	ExampleSelector ExampleSelector
	// ExamplePrompt is used to format an individual example.
	ExamplePrompt PromptTemplate
	// A prompt template string to put before the examples.
	Prefix string
	// A prompt template string to put after the examples.
	Suffix string
	// A list of the names of the variables the prompt template expects.
	InputVariables map[string]any
	// Represents a map of variable names to values or functions that return values. If the value is a function, it will
	// be called when the prompt template is rendered.
	PartialVariables map[string]any
	// String separator used to join the prefix, the examples, and suffix.
	ExampleSeparator string
	// The format of the prompt template. Options are: 'f-string', 'jinja2'.
	TemplateFormat TemplateFormat
	// Whether to try validating the template.
	ValidateTemplate bool
}

type FormatPrompter interface {
	FormatPrompt(values map[string]any) (schema.PromptValue, error)
	GetInputVariables() []string
}

type Formatter interface {
	Format(values map[string]any) (string, error)
}

type GenericMessagePromptTemplate struct {
	Prompt PromptTemplate
	Role   string
}

type HumanMessagePromptTemplate struct {
	Prompt PromptTemplate
}

type MessageFormatter interface {
	FormatMessages(values map[string]any) ([]schema.ChatMessage, error)
	GetInputVariables() []string
}

type PromptTemplate struct {
	// Template is the prompt template.
	Template string

	// A list of variable names the prompt template expects.
	InputVariables []string

	// TemplateFormat is the format of the prompt template.
	TemplateFormat TemplateFormat

	// OutputParser is a function that parses the output of the prompt template.
	OutputParser schema.OutputParser[any]

	// PartialVariables represents a map of variable names to values or functions that return values.
	// If the value is a function, it will be called when the prompt template is rendered.
	PartialVariables map[string]any
}

type StringPromptValue string

type SystemMessagePromptTemplate struct {
	Prompt PromptTemplate
}

type TemplateFormat string