from uuid import UUID
from model_security_client.api import ModelSecurityAPIClient

client = ModelSecurityAPIClient(
    base_url="https://api.sase.paloaltonetworks.com/aims"
)

# Scan with labels attached
result = client.scan(
    security_group_uuid=UUID("your-security-group-uuid"),
    model_uri="https://huggingface.co/microsoft/DialoGPT-medium",
    labels={
        "env": "production",
        "team": "ml-platform",
        "region": "us-west",
        "compliance": "soc2"
    }
)

print(f"Scan {result.uuid} created with labels")
print(f"Labels: {result.labels}")