persistence

package

v0.2.1 Latest Latest Go to latest Published: Nov 25, 2024 License: MIT Imports: 7 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Conversation struct {
	Id       int64
	Messages []Message
}

type DatabaseClient struct {
	Conn *sql.DB
}

func StartClient(projectPath string) (DatabaseClient, error)

func (c *DatabaseClient) FetchRecentConversation() (
	Conversation,
	error,
)

Simply fetches the conversation that was created most recently.

func (c *DatabaseClient) InitializeConversation() (
	Conversation,
	error,
)

Creates an empty conversation in the database.

func (c *DatabaseClient) InsertMessageIntoConversation(
	conversationId int64,
	role string,
	content string,
) (Message, error)

func (c *DatabaseClient) PruneOldConversations(maxHistory int) error

func (c *DatabaseClient) WriteToMessage(messageId int64, text string) error

Extends the existing content of a message with the provided text (used for recording streaming responses from an LLM chat).

type Message struct {
	Id      int64
	Role    string
	Content string
}