Feat: rename chat
This commit is contained in:
@@ -12,12 +12,14 @@ func (p ProviderSQL) Memorise(m *models.Memory) (*models.Memory, error) {
|
||||
query := "INSERT INTO memories (agent, topic, mind) VALUES (:agent, :topic, :mind) RETURNING *;"
|
||||
stmt, err := p.db.PrepareNamed(query)
|
||||
if err != nil {
|
||||
p.logger.Error("failed to prepare stmt", "query", query, "error", err)
|
||||
return nil, err
|
||||
}
|
||||
defer stmt.Close()
|
||||
var memory models.Memory
|
||||
err = stmt.Get(&memory, m)
|
||||
if err != nil {
|
||||
p.logger.Error("failed to insert memory", "query", query, "error", err)
|
||||
return nil, err
|
||||
}
|
||||
return &memory, nil
|
||||
@@ -28,6 +30,7 @@ func (p ProviderSQL) Recall(agent, topic string) (string, error) {
|
||||
var mind string
|
||||
err := p.db.Get(&mind, query, agent, topic)
|
||||
if err != nil {
|
||||
p.logger.Error("failed to get memory", "query", query, "error", err)
|
||||
return "", err
|
||||
}
|
||||
return mind, nil
|
||||
@@ -38,6 +41,7 @@ func (p ProviderSQL) RecallTopics(agent string) ([]string, error) {
|
||||
var topics []string
|
||||
err := p.db.Select(&topics, query, agent)
|
||||
if err != nil {
|
||||
p.logger.Error("failed to get topics", "query", query, "error", err)
|
||||
return nil, err
|
||||
}
|
||||
return topics, nil
|
||||
|
||||
@@ -19,6 +19,7 @@ type ChatHistory interface {
|
||||
GetLastChat() (*models.Chat, error)
|
||||
UpsertChat(chat *models.Chat) (*models.Chat, error)
|
||||
RemoveChat(id uint32) error
|
||||
ChatGetMaxID() (uint32, error)
|
||||
}
|
||||
|
||||
type ProviderSQL struct {
|
||||
@@ -66,6 +67,13 @@ func (p ProviderSQL) RemoveChat(id uint32) error {
|
||||
return err
|
||||
}
|
||||
|
||||
func (p ProviderSQL) ChatGetMaxID() (uint32, error) {
|
||||
query := "SELECT MAX(id) FROM chats;"
|
||||
var id uint32
|
||||
err := p.db.Get(&id, query)
|
||||
return id, err
|
||||
}
|
||||
|
||||
func NewProviderSQL(dbPath string, logger *slog.Logger) FullRepo {
|
||||
db, err := sqlx.Open("sqlite", dbPath)
|
||||
if err != nil {
|
||||
|
||||
Reference in New Issue
Block a user