Overview

    from adaptive_sdk import Adaptive, AsyncAdaptive
    import json
    import asyncio
    from tqdm.asyncio import tqdm_asyncio
    from tqdm import tqdm

    async def upload_data_async(
        adaptive: AsyncAdaptive,
        data: list[dict], 
        labels: dict[str,str],
        max_concurrency: int = 30
    ):
        async def coroutine(semaphore, coroutine):
            async with semaphore:
                return await coroutine

        semaphore = asyncio.Semaphore(max_concurrency)
        coroutines = []
        for i in data:
            coroutines.append(
                coroutine(
                    semaphore,
                    adaptive.interactions.create(
                        messages=i["messages"],
                        completion=i["completion"],
                        labels=labels
                    ),
                )
            )

        print("\nLoading data...")
        results = await tqdm_asyncio.gather(*coroutines)

    if __name__=="__main__":
        url = "REPLACE_ADAPTIVE_URL"
        api_key = "REPLACE_ADAPTIVE_API_KEY"
        use_case_key = "REPLACE_EXISTING_USE_CASE"

        # `data` is assumed to have this format
        data = [
            {
                "messages": [
                    {"role": "system", "content": "You are a good bot."},
                    {"role": "user", "content": "Hey there!"}
                ],
                "completion": "Hey back at you!"
            }
        ]
        labels = {"dataset": "my_custom_dataset"}
        
        adaptive = AsyncAdaptive(base_url=url, api_key=api_key)
        adaptive.set_default_use_case(use_case_key)
        asyncio.run(upload_data_async(adaptive, data, labels))