from model_security_client.api import ModelSecurityAPIClient

# Initialize the client with download configuration
client = ModelSecurityAPIClient(
    base_url="https://api.sase.paloaltonetworks.com/aims",
    download_timeout_secs=1800,            # Object storage download timeout
    download_dir="/custom/download/path",  # Object storage download location
    cleanup_download_dir=True              # Cleanup downloads after scan
)

# Perform scan with polling configuration
result = client.scan(
    security_group_uuid="12345678-1234-1234-1234-123456789012",
    model_uri="<model_uri>",
    poll_interval_secs=10,
    poll_timeout_secs=900,
)