def test_embed_documents_for_private_images(self):
        """Both image_download_headers and media_download_headers work in embed for private images."""
        content = ["https://d2k91vq0avo7lq.cloudfront.net/ai_hippo_realistic_small.png",
                   "https://d2k91vq0avo7lq.cloudfront.net/ai_hippo_realistic_small"]
        kwargs_list = [
            {"media_download_headers": {"marqo_media_header": "media_header_test_key"}},
            {"image_download_headers": {"marqo_media_header": "media_header_test_key"}}
        ]
        for index_name in [self.structured_image_index_name, self.unstructured_image_index_name]:
            for kwargs in kwargs_list:
                with self.subTest(f"{index_name} - {kwargs}"):
                    res = self.client.index(index_name).embed(
                        content,
                        **kwargs
                    )
                    self.assertEqual(2, len(res["embeddings"]))
Both image_download_headers and media_download_headers work in embed for private images.
tests/api_tests/v1/tests/api_tests/test_embed.py