from adaline.main import Adaline

adaline = Adaline()
rows = adaline.datasets.rows  # DatasetRowsClient

from adaline.clients import DatasetRowsClient

from adaline_api.models.dataset_row import DatasetRow
from adaline_api.models.dataset_row_input import DatasetRowInput
from adaline_api.models.add_dataset_rows_response import AddDatasetRowsResponse
from adaline_api.models.list_dataset_rows_response import ListDatasetRowsResponse

async def list(
    *,
    dataset_id: str,
    columns: Optional[str] = None,
    sort: Optional[SortOrderInput] = None,
    created_after: Optional[int] = None,
    created_before: Optional[int] = None,
    limit: Optional[int] = None,
    cursor: Optional[str] = None,
) -> ListDatasetRowsResponse

response = await adaline.datasets.rows.list(
    dataset_id="dataset_abc123",
    columns="question,expected",
    sort="createdAt:desc",
    limit=100,
)

async def create(
    *,
    dataset_id: str,
    rows: List[DatasetRowInput],
    values_by: Optional[str] = None,  # "columnId" or "columnName"
) -> AddDatasetRowsResponse

from adaline_api.models.dataset_row_input import DatasetRowInput

response = await adaline.datasets.rows.create(
    dataset_id="dataset_abc123",
    values_by="columnName",
    rows=[
        DatasetRowInput(values={"question": "How do I reset my password?", "expected": "Link to reset"}),
        DatasetRowInput(values={"question": "How do I cancel my plan?", "expected": "Link to billing"}),
    ],
)

print(f"Inserted {len(response.rows)} rows")

async def update(
    *,
    dataset_id: str,
    row_id: str,
    row: DatasetRowInput,
    values_by: Optional[str] = None,
) -> DatasetRow

from adaline_api.models.dataset_row_input import DatasetRowInput

await adaline.datasets.rows.update(
    dataset_id="dataset_abc123",
    row_id="row_def456",
    values_by="columnName",
    row=DatasetRowInput(values={"expected": "Updated expected answer"}),
)

async def delete(*, dataset_id: str, row_id: str) -> None