memory

type APIKey struct {
	gorm.Model
	User   *User  `json:"user"`
	UserID uint   `json:"userId,omitempty"`
	Hash   string `json:"hash"`
	Key    string `json:"key" gorm:"-"`
}

type CLIMessageSource struct{}

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

type DiscordMessageSource struct {
	API         *discordgo.Session
	Message     *discordgo.Message
	Interaction bool
}

type ExternalID struct {
	gorm.Model
	ExternalID string `json:"external_id"`
	UserID     uint   `json:"user_id"`
	Interface  string `json:"interface"`
}

type HTTPMessageSource struct {
	Request *http.Request
}

type Message struct {
	gorm.Model
	Context               context.Context    `json:"-" gorm:"-"`
	ExternalID            string             `json:"externalId"`
	Interface             string             `json:"interface"`
	User                  *User              `json:"user,omitempty"`
	UserID                uint               `json:"userId,omitempty"`
	Source                *MessageSource     `json:"source" gorm:"-"`
	Name                  string             `json:"name"`
	Role                  string             `json:"role"`
	Content               string             `json:"content"`
	Summary               string             `json:"summary"`
	FunctionCallName      string             `json:"functionCallName"`
	FunctionCallArguments string             `json:"functionCallArguments"`
	Moderated             bool               `json:"moderated"`
	TokenCount            int                `json:"tokenCount"`
	Moderation            *MessageModeration `json:"moderation" gorm:"embedded;embeddedPrefix:moderation_"`
	ErrorCh               chan error         `json:"-" gorm:"-"`
	// contains filtered or unexported fields
}

type MessageModeration struct {
	CategoryHate            bool `json:"categoryHate"`
	CategoryHateThreatening bool `json:"categoryHateThreatening"`
	CategorySelfHarm        bool `json:"categorySelfHarm"`
	CategorySexual          bool `json:"categorySexual"`
	CategorySexualMinors    bool `json:"categorySexualMinors"`
	CategoryViolence        bool `json:"categoryViolence"`
	CategoryViolenceGraphic bool `json:"categoryViolenceGraphic"`

	CategoryScoreHate            float32 `json:"categoryScoreHate"`
	CategoryScoreHateThreatening float32 `json:"categoryScoreHateThreatening"`
	CategoryScoreSelfHarm        float32 `json:"categoryScoreSelfHarm"`
	CategoryScoreSexual          float32 `json:"categoryScoreSexual"`
	CategoryScoreSexualMinors    float32 `json:"categoryScoreSexualMinors"`
	CategoryScoreViolence        float32 `json:"categoryScoreViolence"`
	CategoryScoreViolenceGraphic float32 `json:"categoryScoreViolenceGraphic"`

	Flagged bool `json:"flagged"`
}

type MessageSource struct {
	Telegram *TelegramMessageSource `json:"telegram,omitempty"`
	HTTP     *HTTPMessageSource     `json:"http,omitempty"`
	Discord  *DiscordMessageSource  `json:"discord,omitempty"`
	Slack    *SlackMessageSource    `json:"slack,omitempty"`
	CLI      *CLIMessageSource      `json:"cli,omitempty"`
}

type SlackMessageSource struct {
	API          *slack.Client
	Message      *slackevents.MessageEvent
	SlashCommand slack.SlashCommand
}

type TelegramMessageSource struct {
	API    *tgbotapi.BotAPI
	Update tgbotapi.Update
}

type User struct {
	gorm.Model
	FirstName string `json:"first_name"`
	LastName  string `json:"last_name"`
	Username  string `json:"username"`
	Mode      string `json:"mode"`
}