from metadata.sdk import configure
from metadata.sdk.entities import MLModels

configure(
    host="https://your-company.open-metadata.org/api",
    jwt_token="your-jwt-token"
)

# List first page
models = MLModels.list(limit=50)
for m in models.entities:
    print(f"{m.fullyQualifiedName.root} ({m.algorithm})")

# List all with auto-pagination
for m in MLModels.list_all():
    print(f"{m.fullyQualifiedName.root}")

# Filter by service
models = MLModels.list(
    filters={"service": "mlflow_svc"},
    fields=["owners", "tags", "domains"],
    limit=50
)

for m in models.entities:
    print(f"{m.fullyQualifiedName.root}")
    if m.owners:
        print(f"  Owners: {[o.name for o in m.owners.root]}")
    if m.tags:
        print(f"  Tags: {[tag.tagFQN.root for tag in m.tags]}")

import org.openmetadata.schema.entity.data.MlModel;
import org.openmetadata.sdk.models.ListParams;
import org.openmetadata.sdk.models.ListResponse;

// client is an initialized OpenMetadataClient.

// List first page
ListResponse<MlModel> firstPage = client.mlModels().list(new ListParams().setLimit(50));

for (MlModel item : firstPage.getData()) {
    System.out.println(item.getFullyQualifiedName());
}

// Filter by service with fields
ListResponse<MlModel> filtered = client.mlModels().list(
    new ListParams()
        .setService("mlflow_svc")
        .setFields("owners,tags,domains")
        .setLimit(50)
);

for (MlModel item : filtered.getData()) {
    System.out.println(item.getFullyQualifiedName());
}

# List all
curl "{base_url}/api/v1/mlmodels?limit=50" \
  -H "Authorization: Bearer {access_token}"

# Filter by service
curl "{base_url}/api/v1/mlmodels?service=mlflow_svc&limit=50" \
  -H "Authorization: Bearer {access_token}"

# With fields
curl "{base_url}/api/v1/mlmodels?service=mlflow_svc&fields=owners,tags,domains&limit=50" \
  -H "Authorization: Bearer {access_token}"

{
  "data": [
    {
      "id": "6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2",
      "name": "customer_segmentation",
      "fullyQualifiedName": "mlflow_svc.customer_segmentation",
      "displayName": "Customer Segmentation Model",
      "algorithm": "KMeans",
      "version": 0.1,
      "updatedAt": 1769982669247,
      "updatedBy": "admin",
      "service": {
        "id": "ca22d46e-81b9-4e48-85b5-0adc44980da9",
        "type": "mlmodelService",
        "name": "mlflow_svc",
        "fullyQualifiedName": "mlflow_svc",
        "deleted": false
      },
      "serviceType": "Mlflow",
      "href": "http://localhost:8585/api/v1/mlmodels/6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2",
      "deleted": false,
      "owners": [],
      "tags": [],
      "followers": [],
      "votes": {
        "upVotes": 0,
        "downVotes": 0
      },
      "domains": []
    }
  ],
  "paging": {
    "after": "...",
    "total": 5
  }
}