quest

type AIClient interface {
	GenerateResponse(ctx context.Context, prompt, userMessage string) (*BotLLMResponse, error)
}

type Bot interface {
	HandleMessage(ctx context.Context, userID int64, username, firstName, lastName, message string) ([]pkg.OutputMessage, error)
	HandleCommand(userID int64, username, firstName, lastName, message string) ([]pkg.OutputMessage, error)
	GetName() string
}

type BotConfig struct {
	Name               string              `json:"name" yaml:"name"`
	NameEnv            string              `json:"name_env" yaml:"name_env"`
	Token              string              `json:"token" yaml:"token"`
	TokenEnv           string              `json:"token_env" yaml:"token_env"`
	PromptFiles        map[string][]string `json:"prompt_files" yaml:"prompt_files"` // For every stage each character may have different prompts
	Description        string              `json:"description" yaml:"description"`
	CommandHandlers    map[string]func(chatID int64, sender MessageSender) error
	StateHandlers      map[string]StateTransition `json:"state_handlers" yaml:"state_handlers"`
	TransitionHandlers map[string]func(chatID int64, sender MessageSender) error
}

type BotLLMResponse struct {
	Answer    string   `json:"answer"`
	Goals     []string `json:"goals"`
	Command   string   `json:"command,omitempty"`
	Reasoning string   `json:"reasoning,omitempty"`
}

type BotManager interface {
	AddBots(cfg *Config) error
	ProcessMessage(ctx context.Context, input pkg.InputMessage) ([]pkg.OutputMessage, error)
	StopAll() error
	GetBot(name string) (Bot, bool)
	ListBots() []string
}

type Config struct {
	LLMAPIKey        string
	GenerationConfig GenerationConfig      `json:"generation_config" yaml:"generation_config"`
	Bots             map[string]*BotConfig `json:"bots" yaml:"bots"`
	LogLevel         string
}

type GameEngine struct {
	// contains filtered or unexported fields
}

type GenerationConfig struct {
	ModelName       string  `json:"model_name" yaml:"model_name"`
	Temperature     float32 `json:"temperature" yaml:"temperature"`
	MaxOutputTokens int32   `json:"max_output_tokens" yaml:"max_output_tokens"`
}

type GenerationOptions struct {
	Temperature     float32
	ModelName       string
	MaxOutputTokens int32
}

type MessageSender interface {
	SendMessage(chatID int64, text string) error
	SendPhoto(chatID int64, photoPath string) error
}

type MessengerManager interface {
	AddBots(cfg *Config) error
	Start(ctx context.Context) error
	Stop() error
}

type OptionFunc func(*GenerationOptions)

type StateTransition struct {
	TargetState string   `json:"target_state" yaml:"target_state"`
	Goals       []string `json:"goals" yaml:"goals"`
}