From 1e59e6056ec3a09b04a7f21be718d3b4f5a3d3e1 Mon Sep 17 00:00:00 2001 From: lc6464 <64722907+lc6464@users.noreply.github.com> Date: Wed, 29 Apr 2026 13:10:52 +0800 Subject: [PATCH] style(seahorse): format files of seahorse --- pkg/seahorse/schema_test.go | 6 ++++-- pkg/seahorse/short_engine_test.go | 9 ++++++++- pkg/seahorse/store.go | 3 ++- 3 files changed, 14 insertions(+), 4 deletions(-) diff --git a/pkg/seahorse/schema_test.go b/pkg/seahorse/schema_test.go index b3014669c..943b742b2 100644 --- a/pkg/seahorse/schema_test.go +++ b/pkg/seahorse/schema_test.go @@ -106,12 +106,14 @@ func TestRunSchemaAddsMessagesReasoningContentColumn(t *testing.T) { t.Fatalf("create legacy messages table: %v", err) } - if err := runSchema(db); err != nil { + err = runSchema(db) + if err != nil { t.Fatalf("runSchema: %v", err) } var count int - err = db.QueryRow(`SELECT count(*) FROM pragma_table_info('messages') WHERE name = 'reasoning_content'`).Scan(&count) + err = db.QueryRow(`SELECT count(*) FROM pragma_table_info('messages') WHERE name = 'reasoning_content'`). + Scan(&count) if err != nil { t.Fatalf("query pragma_table_info: %v", err) } diff --git a/pkg/seahorse/short_engine_test.go b/pkg/seahorse/short_engine_test.go index b63885f79..70c08ae9b 100644 --- a/pkg/seahorse/short_engine_test.go +++ b/pkg/seahorse/short_engine_test.go @@ -630,11 +630,18 @@ func TestBootstrapRepairsMissingReasoningContent(t *testing.T) { if err != nil { t.Fatalf("AddMessage user: %v", err) } + assistantMsg, err := eng.store.AddMessage(ctx, conv.ConversationID, "assistant", "world", 3) if err != nil { t.Fatalf("AddMessage assistant: %v", err) } - if err := eng.store.AppendContextMessages(ctx, conv.ConversationID, []int64{userMsg.ID, assistantMsg.ID}); err != nil { + + err = eng.store.AppendContextMessages( + ctx, + conv.ConversationID, + []int64{userMsg.ID, assistantMsg.ID}, + ) + if err != nil { t.Fatalf("AppendContextMessages: %v", err) } diff --git a/pkg/seahorse/store.go b/pkg/seahorse/store.go index 9035b86c0..b195ca857 100644 --- a/pkg/seahorse/store.go +++ b/pkg/seahorse/store.go @@ -360,7 +360,8 @@ func (s *Store) GetMessageCount(ctx context.Context, convID int64) (int, error) func (s *Store) GetMessageByID(ctx context.Context, messageID int64) (*Message, error) { var msg Message var createdAt string - err := s.db.QueryRowContext(ctx, + err := s.db.QueryRowContext( + ctx, "SELECT message_id, conversation_id, role, content, reasoning_content, token_count, created_at FROM messages WHERE message_id = ?", messageID, ).Scan(&msg.ID, &msg.ConversationID, &msg.Role, &msg.Content, &msg.ReasoningContent, &msg.TokenCount, &createdAt)