models

type Message struct {
	gorm.Model
	SessionID   string
	ID          uint  `gorm:"PrimaryKey;autoIncrement"`
	TimeCreated int64 `gorm:"autoCreateTime"`
	TimeUpdated int64 `gorm:"autoUpdateTime:nano"`
	Data        string
}

type Session struct {
	gorm.Model
	ID          string `gorm:"primaryKey"`
	Directory   string
	ToDoList    string `gorm:"type:longtext"`
	Title       string
	TimeCreated int64 `gorm:"autoCreateTime"`
	TimeUpdated int64 `gorm:"autoUpdateTime:nano"`
}

type StoredMessageData struct {
	Role        string           `json:"role"`
	Content     string           `json:"content,omitempty"`
	CodeChanges []string         `json:"code_changes,omitempty"`
	Usage       *StoredUsage     `json:"usage,omitempty"`
	ToolCalls   []StoredToolCall `json:"tool_calls,omitempty"`
}

type StoredToolCall struct {
	ID        string `json:"id"`
	Name      string `json:"name"`
	Arguments string `json:"arguments,omitempty"`
}

type StoredUsage struct {
	PromptTokens     int64 `json:"prompt_tokens,omitempty"`
	CompletionTokens int64 `json:"completion_tokens,omitempty"`
	TotalTokens      int64 `json:"total_tokens,omitempty"`
}

type ToDo struct {
	Index       int
	Description string
	Completed   bool
}