chore(vendor): import pi 1.3.0
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
This commit is contained in:
1 parent
719370c8f7
commit
6081e3e325
1743 files changed
+205459
-131704
No files matched your search
+11
-11
@@ -1,18 +1,18 @@
|
||||
import type { AssistantImages, ImagesApi, ImagesContext, ImagesFunction, ImagesModel, ImagesOptions } from "./types.ts";
|
||||
import type { AssistantImages, ImageApi, ImageModel, ImagesContext, ImagesFunction, ImagesOptions } from "./types.ts";
|
||||
|
||||
export type ImagesApiFunction = (
|
||||
model: ImagesModel<ImagesApi>,
|
||||
model: ImageModel<ImageApi>,
|
||||
context: ImagesContext,
|
||||
options?: ImagesOptions,
|
||||
) => Promise<AssistantImages>;
|
||||
|
||||
export interface ImagesApiProvider<TApi extends ImagesApi = ImagesApi, TOptions extends ImagesOptions = ImagesOptions> {
|
||||
export interface ImagesApiProvider<TApi extends ImageApi = ImageApi, TOptions extends ImagesOptions = ImagesOptions> {
|
||||
api: TApi;
|
||||
generateImages: ImagesFunction<TApi, TOptions>;
|
||||
generateImages: ImagesFunction<TOptions>;
|
||||
}
|
||||
|
||||
interface ImagesApiProviderInternal {
|
||||
api: ImagesApi;
|
||||
api: ImageApi;
|
||||
generateImages: ImagesApiFunction;
|
||||
}
|
||||
|
||||
@@ -23,19 +23,19 @@ type RegisteredImagesApiProvider = {
|
||||
|
||||
const imagesApiProviderRegistry = new Map<string, RegisteredImagesApiProvider>();
|
||||
|
||||
function wrapGenerateImages<TApi extends ImagesApi, TOptions extends ImagesOptions>(
|
||||
api: TApi,
|
||||
generateImages: ImagesFunction<TApi, TOptions>,
|
||||
function wrapGenerateImages<TOptions extends ImagesOptions>(
|
||||
api: ImageApi,
|
||||
generateImages: ImagesFunction<TOptions>,
|
||||
): ImagesApiFunction {
|
||||
return (model, context, options) => {
|
||||
if (model.api !== api) {
|
||||
throw new Error(`Mismatched api: ${model.api} expected ${api}`);
|
||||
}
|
||||
return generateImages(model as ImagesModel<TApi>, context, options as TOptions);
|
||||
return generateImages(model, context, options as TOptions);
|
||||
};
|
||||
}
|
||||
|
||||
export function registerImagesApiProvider<TApi extends ImagesApi, TOptions extends ImagesOptions>(
|
||||
export function registerImagesApiProvider<TApi extends ImageApi, TOptions extends ImagesOptions>(
|
||||
provider: ImagesApiProvider<TApi, TOptions>,
|
||||
sourceId?: string,
|
||||
): void {
|
||||
@@ -48,6 +48,6 @@ export function registerImagesApiProvider<TApi extends ImagesApi, TOptions exten
|
||||
});
|
||||
}
|
||||
|
||||
export function getImagesApiProvider(api: ImagesApi): ImagesApiProviderInternal | undefined {
|
||||
export function getImagesApiProvider(api: ImageApi): ImagesApiProviderInternal | undefined {
|
||||
return imagesApiProviderRegistry.get(api)?.provider;
|
||||
}
|
||||
Reference in new issue
Block a user