from metadata.sdk import configure, client
from metadata.sdk.entities import Pipelines

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

rest = client().ometa.client

table_id = "a1b2c3d4-e5f6-7890-abcd-ef1234567890"

# Add pipeline observability
pipeline = Pipelines.retrieve_by_name("airflow_prod.customer_etl")
rest.put(f"/tables/{table_id}/pipelineObservability", data=[{
    "pipeline": {
        "id": str(pipeline.id.root),
        "type": "pipeline",
        "fullyQualifiedName": "airflow_prod.customer_etl"
    },
    "serviceType": "Airflow",
    "scheduleInterval": "0 0 * * *",
    "lastRunTime": 1706745600000,
    "lastRunStatus": "Successful"
}])

# Get pipeline observability
observability = rest.get(f"/tables/{table_id}/pipelineObservability")
for item in observability:
    print(f"{item['pipeline']['fullyQualifiedName']}: {item['lastRunStatus']}")

import java.util.List;
import org.openmetadata.schema.api.services.CreatePipelineService.PipelineServiceType;
import org.openmetadata.schema.entity.data.Pipeline;
import org.openmetadata.schema.type.PipelineObservability;

String tableId = "a1b2c3d4-e5f6-7890-abcd-ef1234567890";

// Get pipeline observability
List<PipelineObservability> observability = client.tables().getPipelineObservability(tableId);

// Add pipeline observability
Pipeline pipeline = client.pipelines().getByName("airflow_prod.customer_etl");

PipelineObservability item = new PipelineObservability()
    .withPipeline(pipeline.getEntityReference())
    .withServiceType(PipelineServiceType.Airflow)
    .withScheduleInterval("0 0 * * *")
    .withLastRunTime(1706745600000L)
    .withLastRunStatus(PipelineObservability.LastRunStatus.SUCCESSFUL);

client.tables().addPipelineObservability(tableId, List.of(item));

# Get pipeline observability
curl "{base_url}/api/v1/tables/455e3d9d-dbbf-455e-b3be-7191daa825f3/pipelineObservability" \
  -H "Authorization: Bearer {access_token}"

# Add pipeline observability
curl -X PUT "{base_url}/api/v1/tables/455e3d9d-dbbf-455e-b3be-7191daa825f3/pipelineObservability" \
  -H "Authorization: Bearer {access_token}" \
  -H "Content-Type: application/json" \
  -d '[{
    "pipeline": {
      "id": "9a1d3c52-6f0b-4c1e-8a57-2b6f4e0d7c11",
      "type": "pipeline",
      "fullyQualifiedName": "airflow_prod.agent_metrics_etl"
    },
    "serviceType": "Airflow",
    "scheduleInterval": "0 0 * * *",
    "lastRunTime": 1769982651320,
    "lastRunStatus": "Successful"
  }]'

[
  {
    "pipeline": {
      "id": "9a1d3c52-6f0b-4c1e-8a57-2b6f4e0d7c11",
      "type": "pipeline",
      "fullyQualifiedName": "airflow_prod.customer_etl"
    },
    "serviceType": "Airflow",
    "scheduleInterval": "0 0 * * *",
    "lastRunTime": 1706745600000,
    "lastRunStatus": "Successful"
  }
]