from metadata.generated.schema.type.entityLineage import EntitiesEdge
from metadata.generated.schema.type.entityReference import EntityReference
from metadata.ingestion.ometa.ometa_api import OpenMetadata
from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import (
    OpenMetadataConnection,
)
from metadata.generated.schema.security.client.openMetadataJWTClientConfig import (
    OpenMetadataJWTClientConfig,
)

server_config = OpenMetadataConnection(
    hostPort="https://your-company.getcollate.io/api",
    authProvider="openmetadata",
    securityConfig=OpenMetadataJWTClientConfig(
        jwtToken="<YOUR-JWT-TOKEN>"
    ),
)
metadata = OpenMetadata(server_config)

# Delete a lineage edge
edge = EntitiesEdge(
    fromEntity=EntityReference(id=source_table.id, type="table"),
    toEntity=EntityReference(id=target_dashboard.id, type="dashboard"),
)

metadata.delete_lineage_edge(edge=edge)

import org.openmetadata.sdk.api.Lineage;

Lineage.disconnect()
    .from("table", sourceTableId)
    .to("dashboard", dashboardId)
    .confirm();

# Delete lineage between a table and a dashboard
curl -X DELETE "{base_url}/api/v1/lineage/table/{source_table_id}/dashboard/{target_dashboard_id}" \
  -H "Authorization: Bearer {access_token}"

# Delete lineage between two tables
curl -X DELETE "{base_url}/api/v1/lineage/table/{source_table_id}/table/{target_table_id}" \
  -H "Authorization: Bearer {access_token}"

200 OK