feat(packages): Add selective pi-ai base entrypoints (#5348)

This commit is contained in:
Fred K. Schott
2026-06-18 15:41:49 -07:00
committed by GitHub
parent 1a418ad223
commit 0d89a33373
78 changed files with 671 additions and 495 deletions

View File

@@ -1,5 +1,8 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
import { generateImages } from "../src/images.ts";
import { clearImagesApiProviders, getImagesApiProvider } from "../src/images-api-registry.ts";
import { register as registerOpenRouterImages } from "../src/providers/images/openrouter.ts";
import { registerBuiltInImagesApiProviders } from "../src/providers/register-builtins.ts";
import type { ImagesContext, ImagesModel } from "../src/types.ts";
const mockState = vi.hoisted(() => ({
@@ -62,6 +65,15 @@ describe("openrouter images", () => {
beforeEach(() => {
mockState.lastParams = undefined;
mockState.lastRequestOptions = undefined;
clearImagesApiProviders();
registerBuiltInImagesApiProviders();
});
it("registers the direct OpenRouter images transport explicitly", () => {
clearImagesApiProviders();
expect(getImagesApiProvider("openrouter-images")).toBeUndefined();
registerOpenRouterImages();
expect(getImagesApiProvider("openrouter-images")?.api).toBe("openrouter-images");
});
it("returns text plus images in final output", async () => {