import os
from brainbase_labs import BrainbaseLabs
client = BrainbaseLabs(
api_key=os.environ.get("BRAINBASE_LABS_API_KEY"),
)
test = client.workers.tests.create(
worker_id="workerId",
checkpoints=[{
"description": "description",
"name": "name",
"runs": 0,
"tolerance": 0,
}],
description="description",
flow_id="flowId",
name="name",
system_prompt="systemPrompt",
test_mode="testMode",
)
print(test.id)