Function insertUserTextMessage

coderd/x/chatd/chatd_test.go:6223–6250

(
	t *testing.T,
	db database.Store,
	chatID uuid.UUID,
	userID uuid.UUID,
	modelConfigID uuid.UUID,
	text string,
	contextLimit ...int64,
)

}

func insertUserTextMessage(
	t *testing.T,
	db database.Store,
	chatID uuid.UUID,
	userID uuid.UUID,
	modelConfigID uuid.UUID,
	text string,
	contextLimit ...int64,
) database.ChatMessage {
	t.Helper()
	require.LessOrEqual(t, len(contextLimit), 1)

	contextLimitValue := int64(0)
	if len(contextLimit) == 1 {
		contextLimitValue = contextLimit[0]
	}
	content, err := chatprompt.MarshalParts([]codersdk.ChatMessagePart{codersdk.ChatMessageText(text)})
	require.NoError(t, err)

	return dbgen.ChatMessage(t, db, database.ChatMessage{
		ChatID:        chatID,
		CreatedBy:     uuid.NullUUID{UUID: userID, Valid: true},
		ModelConfigID: uuid.NullUUID{UUID: modelConfigID, Valid: true},
		Role:          database.ChatMessageRoleUser,
		Content:       pqtype.NullRawMessage{RawMessage: content.RawMessage, Valid: true},
		ContextLimit:  sql.NullInt64{Int64: contextLimitValue, Valid: contextLimitValue != 0},
	})
}

// seedWorkspaceWithAgent creates a full workspace chain with a connected
// agent. This is the common setup needed by tests that exercise tool