# Prepare the documents to be indexed
documents = []
for line in jsonl_data:
    data_dict = json.loads(line)
    documents.append(
        {
            "_index": "cohere-wiki-embeddings",
            "_source": data_dict,
        }
    )