template

Documentation

Index

• Variables
• func NewRAGChatTemplate(cfg *config.ServerConfig) (prompt.ChatTemplate, error)
• type ChatTemplateConfig

Variables

var ModuleChatTemplate = fx.Module("chat-template",
	fx.Provide(
		fx.Annotate(
			NewRAGChatTemplate,
			fx.As(new(prompt.ChatTemplate))),
	),
)

ModuleChatTemplate is a fx module for chat template

Functions

func NewRAGChatTemplate

func NewRAGChatTemplate(cfg *config.ServerConfig) (prompt.ChatTemplate, error)

NewRAGChatTemplate creates a new RAG chat template

Types

type ChatTemplateConfig

type ChatTemplateConfig struct {
	FormatType schema.FormatType
	Templates  []schema.MessagesTemplate
}

ChatTemplateConfig is a configuration for a chat template