from uuid import UUID
from model_security_client.api import ModelSecurityAPIClient
from model_security_client.generated.data.models.LabelsCreateRequestSchema import LabelsCreateRequestSchema
from model_security_client.generated.data.models.LabelSchema import LabelSchema
client = ModelSecurityAPIClient(
base_url="https://api.sase.paloaltonetworks.com/aims"
)
scan_uuid = UUID("550e8400-e29b-41d4-a716-446655440000")
# Add new labels or update existing ones
client.add_scan_labels(
scan_uuid=scan_uuid,
data=LabelsCreateRequestSchema(
labels=[
LabelSchema(key="owner", value="alice"),
LabelSchema(key="priority", value="high"),
LabelSchema(key="reviewed", value="true"),
]
)
)