// Package wikipedia implements the wikipedia-bot agent rules.
// Handles DMs and mentions by passing them to the LLM (with Wikipedia tool).
package wikipedia

import (
	"github.com/enmanuel/agents/devagents"
	"github.com/enmanuel/agents/pkg/decision"
)

func init() {
	devagents.Register("wikipedia-bot", Rules)
}

// Rules returns the decision rules for wikipedia-bot.
// Any DM or mention is routed to the LLM which has access to wikipedia_search.
func Rules() []decision.Rule {
	return []decision.Rule{
		{
			Name: "llm-all",
			Match: func(ctx decision.MessageContext) bool {
				return ctx.IsDirectMsg || ctx.IsMention
			},
			Actions: []decision.Action{{
				Kind: decision.ActionKindLLM,
				LLM:  &decision.LLMAction{},
			}},
		},
	}
}