Import the required classes and initialize Athina API key.

import os
from athina_client.datasets import Dataset
from athina_client.keys import AthinaApiKey

AthinaApiKey.set_key(os.getenv('ATHINA_API_KEY'))

Now you can add rows to an existing dataset.

# Add rows to a dataset
try:
    Dataset.add_rows(
        dataset_id=DATASET_ID,
        rows=[
            {
                'query': 'What is the capital of France?',
                'context': ['France is a country in Western Europe.', 'Paris is the capital of France.'],
                'response': 'Paris',
                'expected_response': 'Paris'
            },
        ]
    )
except Exception as e:
    print(f"Failed to add rows: {e}")
Create a DatasetGet Dataset