Clustering

from docent import Docent

client = Docent()

state = client.get_clustering_state("my-collection-id", "rubric-123")
print(f"Job ID: {state.get('job_id')}")
print(f"Centroids: {len(state.get('centroids', []))}")

centroids = client.get_cluster_centroids("my-collection-id", "rubric-123")
for centroid in centroids:
    print(centroid)

assignments = client.get_cluster_assignments("my-collection-id", "rubric-123")
for centroid_id, result_ids in assignments.items():
    print(f"Cluster {centroid_id}: {len(result_ids)} results")