shumengya/sproutclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: modalities from openrouter

Browse Source
This commit is contained in:
Cristina Poncela Cubeiro
2026-04-29 09:51:57 +02:00
parent e9414b0500
commit 59a89e0c1c
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/ai/src/providers/openai-completions.ts
View File

@@ -575,6 +575,10 @@ function buildParams(
prompt_cache_retention: cacheRetention === "long" && compat.supportsLongCacheRetention ? "24h" : undefined,
};
if (isOpenRouterImageGenerationModel(model)) {
Reflect.set(params, "modalities", ["image"]);
}
if (compat.supportsUsageInStreaming !== false) {
(params as any).stream_options = { include_usage: true };
}

3
packages/ai/test/openai-completions-image-output.test.ts
View File

@@ -92,8 +92,9 @@ describe("openai-completions image output", () => {
data: "ZmFrZS1wbmc=",
});
const params = mockState.lastParams as { stream?: boolean; stream_options?: unknown };
const params = mockState.lastParams as { stream?: boolean; stream_options?: unknown; modalities?: string[] };
expect(params.stream).toBe(false);
expect("stream_options" in (params as object)).toBe(false);
expect(params.modalities).toEqual(["image"]);
});
});