llm

type Conversation interface {
	Chat(ctx context.Context, message string, inputs map[string]any) (string, error)
}

type ConversationMgr interface {
	NewConversation(ctx context.Context, sessionID string) (Conversation, error)
}

type ConversationMgrOption func(mgr *conversationMgr)

type ConversationPromptConfig struct {
	SystemPrompt *struct {
		Template string   `yaml:"template"`
		Inputs   []string `yaml:"input_variables"`
	} `yaml:"system_prompt"`
}

type LLMReply interface {
	MessageTextReply(
		ctx context.Context, typ linebot.EventSourceType,
		groupID, userID, msg string, mysession sessions.Session,
	) ([]linebot.SendingMessage, error)
}

type LLMReplyConfig struct {
	Model           string
	APIKey          string
	ConfigFile      string
	MongoURI        string
	MongoDB         string
	MongoCollection string
	MCPBaseUrls     []string
	AllowMsgType    []linebot.EventSourceType
	AllowUser       bool
}

type LLMReplyOption func(cfg *LLMReplyConfig)

type MemoryMongoCfg struct {
	ConnectionURL  string `yaml:"connection_url"`
	DatabaseName   string `yaml:"database_name"`
	CollectionName string `yaml:"collection_name"`
}