from metadata.sdk import configure
from metadata.sdk.entities import TestSuites

configure(
    host="https://your-company.getcollate.io/api",
    jwt_token="your-jwt-token"
)

# Retrieve, modify, and update
ts = TestSuites.retrieve("e86a9a11-852f-4bac-b5a7-993b2bbbb572")
ts.description = "Updated: Critical data quality checks for production tables"
updated = TestSuites.update(ts)

print(f"Updated to version {updated.version.root}")

// client is an initialized OpenMetadataClient.
var testSuiteId = "e86a9a11-852f-4bac-b5a7-993b2bbbb572";
var testSuite = client.testSuites().get(testSuiteId);
testSuite.setDescription("Updated: Critical data quality checks for production tables");

var updated = client.testSuites().update(testSuiteId, testSuite);

# Update by ID
curl -X PATCH "{base_url}/api/v1/dataQuality/testSuites/e86a9a11-852f-4bac-b5a7-993b2bbbb572" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json-patch+json" \
  -d '[
    {"op": "replace", "path": "/description", "value": "Updated: Critical data quality checks for production tables"}
  ]'

# Update by name
curl -X PATCH "{base_url}/api/v1/dataQuality/testSuites/name/sample_data.ecommerce_db.shopify.dim_address.testSuite" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json-patch+json" \
  -d '[
    {"op": "replace", "path": "/description", "value": "Updated description"}
  ]'

{
  "id": "e86a9a11-852f-4bac-b5a7-993b2bbbb572",
  "name": "sample_data.ecommerce_db.shopify.dim_address.testSuite",
  "displayName": "Data Contract - dim_address_comprehensive_contract",
  "fullyQualifiedName": "sample_data.ecommerce_db.shopify.dim_address.testSuite",
  "description": "Updated: Critical data quality checks for production tables",
  "version": 0.2,
  "updatedAt": 1769982800000,
  "updatedBy": "admin",
  "deleted": false,
  "owners": [],
  "basic": true
}