"""Mock Nexos.ai API responses for testing."""
# Successful image generation response
MOCK_IMAGE_GENERATION_SUCCESS = {
"created": 1734800000,
"data": [
{
# pragma: allowlist secret
"b64_json": "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==",
"revised_prompt": "A serene mountain landscape at sunset with golden light",
}
],
}
# Rate limited response
MOCK_IMAGE_GENERATION_RATE_LIMITED = {
"error": {
"message": "Rate limit exceeded",
"type": "rate_limit_error",
"code": "rate_limit_exceeded",
}
}
# Models list response
MOCK_MODELS_LIST = {
"object": "list",
"data": [
{
"id": "imagen-4",
"nexos_model_id": "uuid-imagen-4",
"object": "model",
"owned_by": "google",
},
{
"id": "imagen-4-fast",
"nexos_model_id": "uuid-imagen-4-fast",
"object": "model",
"owned_by": "google",
},
{
"id": "imagen-4-ultra",
"nexos_model_id": "uuid-imagen-4-ultra",
"object": "model",
"owned_by": "google",
},
{
"id": "flux-1.1-pro",
"nexos_model_id": "uuid-flux-1.1-pro",
"object": "model",
"owned_by": "black-forest-labs",
},
],
}
