store

type InputState string

type Message struct {
	Id        int64          `db:"id"`
	ChatId    int64          `db:"chat_id"`
	Role      string         `db:"role"`
	Message   string         `db:"message"`
	Version   MessageVersion `db:"version"`
	CreatedAt ytime.Time     `db:"created_at"`
}

type MessageVersion int

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

type Store interface {
	GetUser(ctx context.Context, chatId int64) (*User, error)
	PutUser(ctx context.Context, user *User) (*User, error)
	ApproveUser(ctx context.Context, chatId int64) error
	EnsureInviteCode(ctx context.Context, user *User) error
	EnsureDiglogID(ctx context.Context, user *User) error
	ResetDiglogID(ctx context.Context, user *User) error
	CheckInviteCode(ctx context.Context, user *User, inviteCode string) error
	SetSystemPrompt(ctx context.Context, chatId int64, prompt string) error
	SetInputState(ctx context.Context, chatId int64, state InputState) error

	GetDialogMessages(ctx context.Context, chatId int64) ([]*Message, error)
	PutMessages(ctx context.Context, message []*Message) error
	ClearMessages(ctx context.Context, chatId int64) error

	PutUsage(ctx context.Context, usage *Usage) error
}

type Usage struct {
	Id               int        `db:"id"`
	ChatId           int64      `db:"chat_id"`
	UpdateId         int        `db:"update_id"`
	Model            string     `db:"model"`
	CompletionTokens int        `db:"completion_tokens"`
	PromptTokens     int        `db:"prompt_tokens"`
	TotalTokens      int        `db:"total_tokens"`
	CreatedAt        ytime.Time `db:"created_at"`
}

type User struct {
	ChatId       int64      `db:"chat_id"`
	Approved     bool       `db:"approved"`
	Username     string     `db:"username"`
	FullName     string     `db:"full_name"`
	Salt         string     `db:"salt"`
	Model        string     `db:"model"`
	InviteCode   string     `db:"invite_code"`
	SystemPrompt string     `db:"system_prompt"`
	InputState   InputState `db:"input_state"`
	DialogID     string     `db:"dialog_id"`

	CreatedAt ytime.Time `db:"created_at"`
	UpdatedAt ytime.Time `db:"updated_at"`
}