@earendil-works/pi-ai
Version:
Unified LLM API with automatic model discovery and provider configuration
22 lines • 764 B
JavaScript
const imagesApiProviderRegistry = new Map();
function wrapGenerateImages(api, generateImages) {
return (model, context, options) => {
if (model.api !== api) {
throw new Error(`Mismatched api: ${model.api} expected ${api}`);
}
return generateImages(model, context, options);
};
}
export function registerImagesApiProvider(provider, sourceId) {
imagesApiProviderRegistry.set(provider.api, {
provider: {
api: provider.api,
generateImages: wrapGenerateImages(provider.api, provider.generateImages),
},
sourceId,
});
}
export function getImagesApiProvider(api) {
return imagesApiProviderRegistry.get(api)?.provider;
}
//# sourceMappingURL=images-api-registry.js.map