convo

type CacheDetailsMsg struct {
	WriteID string // ID to write cache to
	ReadID  string // ID to read cache from
	Title   string // Title of the conversation
	Model   string // Model used for the conversation
}

type ChatMessageHistory interface {
	// AddAIMessage is a convenience method for adding an AI message string to
	// the store.
	AddAIMessage(ctx context.Context, convoID, message string) error
	// AddUserMessage is a convenience method for adding a human message string
	// to the store.
	AddUserMessage(ctx context.Context, convoID, message string) error
	// AddMessage adds a message to the store.
	AddMessage(ctx context.Context, convoID string, message llms.ChatMessage) error
	// SetMessages replaces existing messages in the store
	SetMessages(ctx context.Context, convoID string, messages []llms.ChatMessage) error
	// Messages retrieves all messages from the store
	Messages(ctx context.Context, convoID string) ([]llms.ChatMessage, error)
	// PersistentMessages saves messages to persistent storage
	PersistentMessages(ctx context.Context, convoID string) error
	// InvalidateMessages removes messages from persistent storage
	InvalidateMessages(ctx context.Context, convoID string) error
}

type ContentType string

type Conversation struct {
	// ID is the unique identifier for the convo
	ID string `db:"id" json:"id"`

	// Title is a human-readable name for the convo
	Title string `db:"title" json:"title"`

	// UpdatedAt tracks the last modification time of the convo
	UpdatedAt time.Time `db:"updated_at" json:"updatedAt"`

	// Model optionally specifies the AI model used in the convo
	Model *string `db:"model" json:"model"`
}

type Factory interface {
	// Type unique type of the convo store
	Type() string
	// Create relevant convo store by type
	Create(options *options.Config) (Store, error)
}

type LoadContext struct {
	// ID is the auto-increment primary key
	ID uint64 `db:"id" json:"id"`

	// Type indicates the content source type (file, URL, or text)
	Type ContentType `db:"type" json:"type"`

	// URL contains the source URL if loading from a remote resource
	URL string `db:"url" json:"url"`

	// FilePath contains the local file path if loading from a file
	FilePath string `db:"file_path" json:"filePath"`

	// Content contains the actual loaded content as a string
	Content string `db:"content" json:"content"`

	// Name is a human-readable identifier for the content
	Name string `db:"name" json:"name"`

	// ConversationID associates the content with a specific convo
	ConversationID string `db:"conversation_id" json:"conversationId"`

	// UpdatedAt tracks the last modification time of the convo
	UpdatedAt time.Time `db:"updated_at" json:"updatedAt"`
}

type LoadContextStore interface {
	// SaveContext saves a load context
	SaveContext(ctx context.Context, lc *LoadContext) error
	// GetContext retrieves a load context by ID
	GetContext(ctx context.Context, id uint64) (*LoadContext, error)
	// ListContextsByteConvoID retrieves all load contexts for a convo
	ListContextsByteConvoID(ctx context.Context, conversationID string) ([]LoadContext, error)
	// DeleteContexts removes a load context
	DeleteContexts(ctx context.Context, id uint64) error
	// CleanContexts removes all load contexts for a convo and returns the count deleted
	CleanContexts(ctx context.Context, conversationID string) (int64, error)
}

type SimpleChatHistoryStore struct {
	sync.RWMutex // protects access to messages and loaded maps
	// contains filtered or unexported fields
}

type Store interface {
	ChatMessageHistory
	LoadContextStore

	// LatestConversation returns the last message in the chat convo.
	LatestConversation(ctx context.Context) (*Conversation, error)
	// GetConversation retrieves a convo from the store
	GetConversation(ctx context.Context, convoID string) (*Conversation, error)
	// ListConversations retrieves all convo id from the store
	ListConversations(ctx context.Context) ([]Conversation, error)
	// ListConversationsOlderThan retrieves all convo id from the store that are older than the given time.
	ListConversationsOlderThan(ctx context.Context, t time.Duration) ([]Conversation, error)
	// SaveConversation saves a convo to the store
	SaveConversation(ctx context.Context, id, title, model string) error
	// DeleteConversation removes a convo from the store
	DeleteConversation(ctx context.Context, convoID string) error
	// ClearConversations removes all convo from the store.
	ClearConversations(ctx context.Context) error
	// ConversationExists checks if the given chat convo exists.
	ConversationExists(ctx context.Context, sessionID string) (bool, error)
}