Function upsert_with_retry

tests/test_multiple_datasets.py:187–203

(start_idx, retries=20)

    )

    def upsert_with_retry(start_idx, retries=20):
        for attempt in range(retries):
            try:
                upsert_in_transaction(
                    constants["COLLECTION_NAME"],
                    txn_id,
                    vectors[start_idx : start_idx + batch_size],
                )
                return
            except Exception as e:
                print(
                    f"Upsert attempt {attempt + 1} failed for batch starting at index {start_idx}: {e}"
                )
                time.sleep(random.uniform(1, 3))
        print(
            f"Failed to upsert batch starting at index {start_idx} after {retries} attempts"
        )

    transaction = create_transaction(constants["COLLECTION_NAME"])
    txn_id = transaction["transaction_id"]