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

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

# List all versions
versions = MLModels.get_versions("6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2")
for v in versions:
    print(f"Version {v.version}: {v.description}")

# Get a specific version
model_v1 = MLModels.get_specific_version(
    "6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2",
    "0.1"
)
print(f"Original algorithm: {model_v1.algorithm}")

import static org.openmetadata.sdk.fluent.MlModels.*;

// List all versions
var history = MlModels.getVersions("6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2");

for (var version : history.getVersions()) {
    System.out.println("Version: " + version);
}

// Get a specific version
var v1 = MlModels.getSpecificVersion(
    "6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2",
    0.1
);
System.out.println("Algorithm: " + v1.getAlgorithm());

# List all versions
curl "{base_url}/api/v1/mlmodels/6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2/versions" \
  -H "Authorization: Bearer {access_token}"

# Get a specific version
curl "{base_url}/api/v1/mlmodels/6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2/versions/0.1" \
  -H "Authorization: Bearer {access_token}"

{
  "entityType": "mlmodel",
  "versions": [
    "{\"id\":\"6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2\",\"name\":\"customer_segmentation\",\"fullyQualifiedName\":\"mlflow_svc.customer_segmentation\",\"version\":0.2,\"description\":\"Updated: Customer segmentation using KMeans with 5 clusters\",\"algorithm\":\"KMeans\",\"serviceType\":\"Mlflow\"}",
    "{\"id\":\"6b04e1d8-b66d-4f78-ab21-beb5be2cf4f2\",\"name\":\"customer_segmentation\",\"fullyQualifiedName\":\"mlflow_svc.customer_segmentation\",\"version\":0.1,\"algorithm\":\"KMeans\",\"serviceType\":\"Mlflow\"}"
  ]
}