diff --git a/internal/message/message.go b/internal/message/message.go index 37f69a89b9aef60698a90b6a94073106bf969ff2..637496268a5c802a623be9d72eef6434f94565e6 100644 --- a/internal/message/message.go +++ b/internal/message/message.go @@ -82,7 +82,7 @@ func (s *service) Create(ctx context.Context, sessionID string, params CreateMes if err != nil { return Message{}, err } - message, err := s.fromDBItem(dbMessage) + message, err := fromDBItem(dbMessage) if err != nil { return Message{}, err } @@ -134,17 +134,16 @@ func (s *service) Get(ctx context.Context, id string) (Message, error) { if err != nil { return Message{}, err } - return s.fromDBItem(dbMessage) + return fromDBItem(dbMessage) } -func (s *service) List(ctx context.Context, sessionID string) ([]Message, error) { - dbMessages, err := s.q.ListMessagesBySession(ctx, sessionID) +func convertDBMessagesToMessages(dbMessages []db.Message, err error) ([]Message, error) { if err != nil { return nil, err } messages := make([]Message, len(dbMessages)) for i, dbMessage := range dbMessages { - messages[i], err = s.fromDBItem(dbMessage) + messages[i], err = fromDBItem(dbMessage) if err != nil { return nil, err } @@ -152,12 +151,15 @@ func (s *service) List(ctx context.Context, sessionID string) ([]Message, error) return messages, nil } +func (s *service) List(ctx context.Context, sessionID string) ([]Message, error) { + return convertDBMessagesToMessages(s.q.ListMessagesBySession(ctx, sessionID)) +} + func (s *service) FullList(ctx context.Context) ([]Message, error) { - dbMessages, err := s.q.ListAllMessages(ctx) - return nil, nil + return convertDBMessagesToMessages(s.q.ListAllMessages(ctx)) } -func (s *service) fromDBItem(item db.Message) (Message, error) { +func fromDBItem(item db.Message) (Message, error) { parts, err := unmarshallParts([]byte(item.Parts)) if err != nil { return Message{}, err diff --git a/internal/tui/components/chat/editor/editor.go b/internal/tui/components/chat/editor/editor.go index a7c6b0347352132370280a064e2f7f552c559abc..1e8c1bc9748bc3392c4ba40b3832db7981dbc3e9 100644 --- a/internal/tui/components/chat/editor/editor.go +++ b/internal/tui/components/chat/editor/editor.go @@ -521,7 +521,7 @@ func (m *editorCmp) getUserMessagesAsText() ([]string, error) { ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() - allMessages, err := m.app.Messages.List(ctx, m.session.ID) + allMessages, err := m.app.Messages.FullList(ctx) if err != nil { return nil, err }