style(seahorse): format files of seahorse
This commit is contained in:
parent
85f9d21151
commit
1e59e6056e
3 changed files with 14 additions and 4 deletions
|
|
@ -106,12 +106,14 @@ func TestRunSchemaAddsMessagesReasoningContentColumn(t *testing.T) {
|
||||||
t.Fatalf("create legacy messages table: %v", err)
|
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)
|
t.Fatalf("runSchema: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
var count int
|
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 {
|
if err != nil {
|
||||||
t.Fatalf("query pragma_table_info: %v", err)
|
t.Fatalf("query pragma_table_info: %v", err)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -630,11 +630,18 @@ func TestBootstrapRepairsMissingReasoningContent(t *testing.T) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("AddMessage user: %v", err)
|
t.Fatalf("AddMessage user: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
assistantMsg, err := eng.store.AddMessage(ctx, conv.ConversationID, "assistant", "world", 3)
|
assistantMsg, err := eng.store.AddMessage(ctx, conv.ConversationID, "assistant", "world", 3)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("AddMessage assistant: %v", err)
|
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)
|
t.Fatalf("AppendContextMessages: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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) {
|
func (s *Store) GetMessageByID(ctx context.Context, messageID int64) (*Message, error) {
|
||||||
var msg Message
|
var msg Message
|
||||||
var createdAt string
|
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 = ?",
|
"SELECT message_id, conversation_id, role, content, reasoning_content, token_count, created_at FROM messages WHERE message_id = ?",
|
||||||
messageID,
|
messageID,
|
||||||
).Scan(&msg.ID, &msg.ConversationID, &msg.Role, &msg.Content, &msg.ReasoningContent, &msg.TokenCount, &createdAt)
|
).Scan(&msg.ID, &msg.ConversationID, &msg.Role, &msg.Content, &msg.ReasoningContent, &msg.TokenCount, &createdAt)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue