LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-04-26 07:57:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(google): add image generation provider

Browse Source
This commit is contained in:
Peter Steinberger
2026-03-16 23:20:27 -07:00
parent c1ef5748eb
commit 618d35f933
7 changed files with 359 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/plugins/contracts/registry.contract.test.ts
View File

@@ -165,6 +165,7 @@ describe("plugin contract registry", () => {
});
it("keeps bundled image-generation ownership explicit", () => {
expect(findImageGenerationProviderIdsForPlugin("google")).toEqual(["google"]);
expect(findImageGenerationProviderIdsForPlugin("openai")).toEqual(["openai"]);
});
@@ -180,6 +181,13 @@ describe("plugin contract registry", () => {
});
it("tracks speech registrations on bundled provider plugins", () => {
expect(findRegistrationForPlugin("google")).toMatchObject({
providerIds: ["google", "google-gemini-cli"],
speechProviderIds: [],
mediaUnderstandingProviderIds: ["google"],
imageGenerationProviderIds: ["google"],
webSearchProviderIds: ["gemini"],
});
expect(findRegistrationForPlugin("openai")).toMatchObject({
providerIds: ["openai", "openai-codex"],
speechProviderIds: ["openai"],
@@ -245,6 +253,9 @@ describe("plugin contract registry", () => {
});
it("keeps bundled image-generation support explicit", () => {
expect(findImageGenerationProviderForPlugin("google").generateImage).toEqual(
expect.any(Function),
);
expect(findImageGenerationProviderForPlugin("openai").generateImage).toEqual(
expect.any(Function),
);

2
src/plugins/contracts/registry.ts
View File

@@ -131,7 +131,7 @@ const bundledMediaUnderstandingPlugins: RegistrablePlugin[] = [
zaiPlugin,
];
const bundledImageGenerationPlugins: RegistrablePlugin[] = [openAIPlugin];
const bundledImageGenerationPlugins: RegistrablePlugin[] = [googlePlugin, openAIPlugin];
function captureRegistrations(plugin: RegistrablePlugin) {
const captured = createCapturedPluginRegistration();