model

Documentation

Index

• type Content
• type History
  • func NewHistory() *History
  • func (h *History) Append(role Role, text string) *History
  • func (h *History) ToGeminiContents() (genai.Text, []*genai.Content)
• type Message
  • func NewMessage(messageID, unixTimestamp, chatID, userID int64, ...) *Message
• type Model
  • func New(db *database.Database) *Model
  • func (m *Model) GetReplies(ctx context.Context, msgID int64) ([]Message, error)
  • func (m *Model) StoreMessages(ctx context.Context, msgs []Message) error
• type Role

Types

type Content

type Content struct {
	Role Role   `json:"role"`
	Text string `json:"text"`
}

type History

type History struct {
	Contents []*Content `json:"contents"`
}

func NewHistory

func NewHistory() *History

func (*History) Append

func (h *History) Append(role Role, text string) *History

func (*History) ToGeminiContents

func (h *History) ToGeminiContents() (genai.Text, []*genai.Content)

Returns a tuple of the user message and all historical messages that is ready to be consumed by Gemini's genai package.

type Message

type Message struct {
	MessageID int64  `json:"message_id"`
	CreatedAt int64  `json:"created_at"`
	ChatID    int64  `json:"chat_id"`
	SenderID  int64  `json:"sender_id"`
	ReplyID   *int64 `json:"reply_id,omitempty"`
	Content   string `json:"content"`
}

func NewMessage

func NewMessage(
	messageID, unixTimestamp, chatID, userID int64,
	messageReplied *gotgbot.Message,
	content string,
) *Message

type Model

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

func New

func New(db *database.Database) *Model

func (*Model) GetReplies

func (m *Model) GetReplies(ctx context.Context, msgID int64) ([]Message, error)

Returns all replies of the given message.

func (*Model) StoreMessages

func (m *Model) StoreMessages(ctx context.Context, msgs []Message) error

Stores messages into the database, returning the first error encountered.

type Role

type Role string

const (
	RoleUser  Role = "user"
	RoleModel Role = "model"
)