bot

Documentation

Index

• type Bot
  • func New(botAPI *tgbotapi.BotAPI, imagePredictor ImagePredictor) *Bot
  • func (bot *Bot) FetchAsync(ctx context.Context, config tgbotapi.UpdateConfig, ch chan *tgbotapi.Update)
  • func (b *Bot) Start(ctx context.Context)
  • func (b *Bot) StartWorkers(ctx context.Context, ch chan *tgbotapi.Update)
  • func (b *Bot) Stop()
• type ImagePredictor

Types

type Bot

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

func New

func New(botAPI *tgbotapi.BotAPI, imagePredictor ImagePredictor) *Bot

func (*Bot) FetchAsync

func (bot *Bot) FetchAsync(ctx context.Context, config tgbotapi.UpdateConfig, ch chan *tgbotapi.Update)

func (*Bot) Start

func (b *Bot) Start(ctx context.Context)

func (*Bot) StartWorkers

func (b *Bot) StartWorkers(ctx context.Context, ch chan *tgbotapi.Update)

func (*Bot) Stop

func (b *Bot) Stop()

type ImagePredictor

type ImagePredictor interface {
	PredictImage(imageBytes []byte) (*model.Result, error)
	Stop() error
}

A ImagePredictor predicts the class of an image