tbotopenai

type AI interface {
	GenerateText(ctx context.Context, prompt string) ([]byte, error)
	GenerateImage(ctx context.Context, prompt string) ([]byte, string, error)
}

type ChatGPTBot struct{}

type ChatGPTSettings struct {
	Timeout time.Duration `yaml:"timeout"`
}

type Config struct {
	Telegram                TelegramSettings   `yaml:"telegram"`
	ChatGPT                 ChatGPTSettings    `yaml:"chatgpt"`
	OpenAI                  OpenAISettings     `yaml:"openai"`
	DreamBooth              DreamBoothSettings `yaml:"dreambooth"`
	Roles                   RolesSettings      `yaml:"roles"`
	Permissions             PermissionSettings `yaml:"permissions"`
	Stats                   StatsSettings      `yaml:"stats"`
	Logger                  zap.Config         `yaml:"log"`
	LenMessageChan          int                `yaml:"len_message_chan"`
	LenQueueTaskChan        int                `yaml:"len_queue_task_chan"`
	QueueMessageWorkers     int                `yaml:"queue_message_workers"`
	MaxClientOpenAIJobs     int                `yaml:"max_client_openai_jobs"`
	MaxClientChatGPTJobs    int                `yaml:"max_client_chatgpt_jobs"`
	MaxClientDreamBoothJobs int                `yaml:"max_client_dreambooth_jobs"`
	MaxLogRows              int                `yaml:"max_log_rows"`
	PathBlackList           string             `yaml:"path_blacklist"`
}

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

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

type DreamBoothSettings struct {
	Tokens        []string      `yaml:"tokens"`
	RetryInterval time.Duration `yaml:"retry_interval"`
	Timeout       time.Duration `yaml:"timeout"`
}

type Messenger interface {
	Run()
	Stop()
	ReplyText(int, int64, string) error
	ReplyFile(int, int64, []byte, string) error
}

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

type OpenAISettings struct {
	Token         string        `yaml:"token"`
	RetryCount    int           `yaml:"retry_count"`
	RetryInterval time.Duration `yaml:"retry_interval"`
	Timeout       time.Duration `yaml:"timeout"`
}

type PermissionSettings struct {
	AdminCommands []string `yaml:"admin"`
	UserCommands  []string `yaml:"user"`
}

type RolesSettings struct {
	Admins []string `yaml:"admin"`
	Users  []string `yaml:"user"`
}

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

type StatsSettings struct {
	Interval time.Duration `yaml:"interval"`
	Filepath string        `yaml:"filepath"`
}

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

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

type TelegramSettings struct {
	Token   string `yaml:"token"`
	Debug   bool   `yaml:"debug"`
	Timeout int    `yaml:"timeout"`
}