imagenai
Version:
Core client library for Imagenai image generation API
80 lines (79 loc) • 3.47 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ImagenaiClient = void 0;
exports.createImagenaiClient = createImagenaiClient;
class ImagenaiClient {
constructor(config) {
this.config = Object.assign({ timeout: 30000, retries: 3 }, config);
this.baseUrl = config.baseUrl || 'https://your-domain.com';
}
/**
* Generate a single image
*/
generateImage(options) {
return __awaiter(this, void 0, void 0, function* () {
const response = yield this.makeRequest('/api/imagenai', {
method: 'POST',
body: JSON.stringify(Object.assign(Object.assign({}, options), { apiKey: this.config.apiKey }))
});
if (!response.ok) {
const error = yield response.json();
throw new Error(error.error || 'Failed to generate image');
}
return yield response.json();
});
}
/**
* Make HTTP request with retry logic
*/
makeRequest(url, options) {
return __awaiter(this, void 0, void 0, function* () {
let lastError;
for (let attempt = 1; attempt <= this.config.retries; attempt++) {
try {
const controller = new AbortController();
const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
const response = yield fetch(`${this.baseUrl}${url}`, Object.assign(Object.assign({}, options), { signal: controller.signal, headers: Object.assign({ 'Content-Type': 'application/json' }, options.headers) }));
clearTimeout(timeoutId);
return response;
}
catch (error) {
lastError = error instanceof Error ? error : new Error('Request failed');
if (attempt === this.config.retries) {
break;
}
// Wait before retry (exponential backoff, reduced for tests)
const baseDelay = process.env.NODE_ENV === 'test' ? 10 : 1000;
yield new Promise(resolve => setTimeout(resolve, Math.pow(2, attempt) * baseDelay));
}
}
throw lastError;
});
}
/**
* Handle and format errors
*/
handleError(error) {
if (error instanceof Error) {
return error;
}
if (typeof error === 'string') {
return new Error(error);
}
return new Error('An unknown error occurred');
}
}
exports.ImagenaiClient = ImagenaiClient;
// Export a factory function for easier usage
function createImagenaiClient(config) {
return new ImagenaiClient(config);
}