def push_tasks(tasks: list[dict]) -> list[int]:
    """Import tasks into the project; returns the created task IDs."""
    result = ls.projects.import_tasks(id=PROJECT_ID, request=tasks)
    print(f"Imported {len(tasks)} tasks into project {PROJECT_ID}")
    return result.task_ids if hasattr(result, "task_ids") else []

from braintrust import init_project

project = init_project(name="customer-support")
experiment = project.experiments.fetch("v3-eval")
records = experiment.fetch_all()
scored = [r for r in records if r.scores.get("accuracy")]
print(f"Retrieved {len(scored)} scored eval records from Braintrust")

from databricks import sql

with sql.connect(server_hostname=HOST, http_path=PATH, access_token=TOKEN) as conn:
    cur = conn.cursor()
    cur.execute("SELECT id, transcript FROM main.support.tickets LIMIT 1000")
    rows = cur.fetchall()
print(f"Retrieved {len(rows)} support transcripts from Databricks")