from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import (
    AuthProvider,
    OpenMetadataConnection,
)
from metadata.generated.schema.security.client.openMetadataJWTClientConfig import (
    OpenMetadataJWTClientConfig,
)
from metadata.ingestion.ometa.ometa_api import OpenMetadata
from metadata.generated.schema.entity.services.mlmodelService import MlModelService

metadata = OpenMetadata(
    OpenMetadataConnection(
        hostPort="https://your-company.open-metadata.org/api",
        authProvider=AuthProvider.openmetadata,
        securityConfig=OpenMetadataJWTClientConfig(jwtToken="your-jwt-token"),
    )
)

# List first page
services = metadata.list_entities(entity=MlModelService, limit=50)
for svc in services.entities:
    print(f"{svc.fullyQualifiedName.root} ({svc.serviceType.value})")

# List all with auto-pagination
for svc in metadata.list_all_entities(entity=MlModelService):
    print(f"{svc.fullyQualifiedName.root}")

# List with fields
services = metadata.list_entities(
    entity=MlModelService,
    fields=["owners", "tags", "domains"],
    limit=50
)

for svc in services.entities:
    print(f"{svc.fullyQualifiedName.root}")
    if svc.owners:
        print(f"  Owners: {[o.name for o in svc.owners.root]}")

import org.openmetadata.schema.entity.services.MlModelService;
import org.openmetadata.sdk.models.ListParams;
import org.openmetadata.sdk.models.ListResponse;

// client is an initialized OpenMetadataClient.

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

for (MlModelService service : firstPage.getData()) {
    System.out.println(service.getFullyQualifiedName());
}

// With fields
ListResponse<MlModelService> withFields = client.mlModelServices().list(
    new ListParams().setFields("owners,tags,domains").setLimit(50)
);

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

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

{
  "data": [
    {
      "id": "ca22d46e-81b9-4e48-85b5-0adc44980da9",
      "name": "mlflow_svc",
      "fullyQualifiedName": "mlflow_svc",
      "serviceType": "Mlflow",
      "version": 0.1,
      "updatedAt": 1769982621618,
      "updatedBy": "admin",
      "href": "http://localhost:8585/api/v1/services/mlmodelServices/ca22d46e-81b9-4e48-85b5-0adc44980da9",
      "connection": {
        "config": {
          "type": "Mlflow",
          "trackingUri": "http://localhost:8088",
          "registryUri": "http://localhost:8088",
          "supportsMetadataExtraction": true
        }
      },
      "owners": [],
      "tags": [],
      "deleted": false,
      "domains": []
    }
  ],
  "paging": {
    "after": "...",
    "total": 3
  }
}