prompt

type FewShotExample struct {
	Query       string
	Thought     string
	Action      string
	ActionInput map[string]any
	Observation string
	FinalAnswer string
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type ModelConfig struct {
	Name           string
	MaxContextSize int
	Temperature    float32
	TopP           float32
	TopK           int
	RepeatPenalty  float32
	StopSequences  []string
	SystemPrompt   string
}

type OutputFormat struct {
	ThoughtPrefix     string
	ActionPrefix      string
	ActionInputPrefix string
	ObservationPrefix string
	FinalAnswerPrefix string
	JSONDelimiters    bool
	StrictFormatting  bool
}

type ReActTemplate struct {
	ModelName       string
	SystemPrompt    string
	FewShotExamples []FewShotExample
	OutputFormat    OutputFormat
}

type Tool struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	Schema      map[string]any `json:"schema,omitempty"`
}