1import assert from "node:assert/strict";
2import test from "node:test";
3 
4import type { CliArgs } from "../types.ts";
5import {
6  buildContent,
7  buildRequestBody,
8  extractImageFromResponse,
9  getAspectRatio,
10  getImageSize,
11  validateArgs,
12} from "./openrouter.ts";
13 
14const GEMINI_MODEL = "google/gemini-3.1-flash-image-preview";
15const GEMINI_25_MODEL = "google/gemini-2.5-flash-image";
16const GPT_5_IMAGE_MODEL = "openai/gpt-5-image";
17const OPENROUTER_AUTO_MODEL = "openrouter/auto";
18const FLUX_MODEL = "black-forest-labs/flux.2-pro";
19 
20function makeArgs(overrides: Partial<CliArgs> = {}): CliArgs {
21  return {
22    prompt: null,
23    promptFiles: [],
24    imagePath: null,
25    provider: null,
26    model: null,
27    aspectRatio: null,
28    size: null,
29    quality: null,
30    imageSize: null,
31    imageApiDialect: null,
32    referenceImages: [],
33    n: 1,
34    batchFile: null,
35    jobs: null,
36    json: false,
37    help: false,
38    ...overrides,
39  };
40}
41 
42test("OpenRouter request body uses image_config and string content for text-only prompts", () => {
43  const args = makeArgs({ aspectRatio: "16:9", quality: "2k" });
44  const body = buildRequestBody("hello", GEMINI_MODEL, args, []);
45 
46  assert.deepEqual(body.image_config, {
47    image_size: "2K",
48    aspect_ratio: "16:9",
49  });
50  assert.deepEqual(body.provider, {
51    require_parameters: true,
52  });
53  assert.deepEqual(body.modalities, ["image", "text"]);
54  assert.equal(body.stream, false);
55  assert.equal(body.messages[0].content, "hello");
56});
57 
58test("OpenRouter request body keeps text+image modalities for current text+image models", () => {
59  for (const model of [GEMINI_MODEL, GEMINI_25_MODEL, GPT_5_IMAGE_MODEL, OPENROUTER_AUTO_MODEL]) {
60    const body = buildRequestBody("hello", model, makeArgs({ quality: "2k" }), []);
61 
62    assert.deepEqual(body.image_config, {
63      image_size: "2K",
64    });
65    assert.deepEqual(body.provider, {
66      require_parameters: true,
67    });
68    assert.deepEqual(body.modalities, ["image", "text"]);
69    assert.equal(body.messages[0].content, "hello");
70  }
71});
72 
73test("OpenRouter request body uses image-only modalities for image-only models under CLI defaults", () => {
74  const body = buildRequestBody("hello", FLUX_MODEL, makeArgs({ quality: "2k" }), []);
75 
76  assert.deepEqual(body.image_config, {
77    image_size: "2K",
78  });
79  assert.deepEqual(body.provider, {
80    require_parameters: true,
81  });
82  assert.deepEqual(body.modalities, ["image"]);
83  assert.equal(body.stream, false);
84  assert.equal(body.messages[0].content, "hello");
85});
86 
87test("OpenRouter helper omits image_config when no size or quality is passed", () => {
88  const body = buildRequestBody("hello", FLUX_MODEL, makeArgs(), []);
89 
90  assert.equal(body.image_config, undefined);
91  assert.equal(body.provider, undefined);
92  assert.deepEqual(body.modalities, ["image"]);
93  assert.equal(body.stream, false);
94  assert.equal(body.messages[0].content, "hello");
95});
96 
97test("OpenRouter request body keeps multimodal array content when references are provided", () => {
98  const content = buildContent("hello", ["data:image/png;base64,abc"]);
99  assert.ok(Array.isArray(content));
100  assert.deepEqual(content[0], { type: "text", text: "hello" });
101  assert.deepEqual(content[1], {
102    type: "image_url",
103    image_url: { url: "data:image/png;base64,abc" },
104  });
105});
106 
107test("OpenRouter size and aspect helpers infer supported values", () => {
108  assert.equal(getImageSize(makeArgs()), null);
109  assert.equal(getImageSize(makeArgs({ quality: "normal" })), "1K");
110  assert.equal(getImageSize(makeArgs({ size: "2048x1024" })), "2K");
111  assert.equal(getAspectRatio(GEMINI_MODEL, makeArgs({ size: "1600x900" })), "16:9");
112  assert.equal(getAspectRatio(GEMINI_MODEL, makeArgs({ size: "1024x4096" })), "1:4");
113  assert.equal(getAspectRatio(GEMINI_25_MODEL, makeArgs({ size: "1600x900" })), "16:9");
114  assert.equal(getAspectRatio(FLUX_MODEL, makeArgs({ size: "1024x4096" })), null);
115});
116 
117test("OpenRouter validates explicit aspect ratios and inferred size ratios against model support", () => {
118  assert.doesNotThrow(() =>
119    validateArgs(GEMINI_MODEL, makeArgs({ aspectRatio: "1:4" })),
120  );
121  assert.doesNotThrow(() =>
122    validateArgs(GEMINI_MODEL, makeArgs({ size: "1024x4096" })),
123  );
124  assert.throws(
125    () => validateArgs(GEMINI_25_MODEL, makeArgs({ aspectRatio: "1:4" })),
126    /does not support aspect ratio 1:4/,
127  );
128  assert.throws(
129    () => validateArgs(FLUX_MODEL, makeArgs({ aspectRatio: "1:4" })),
130    /does not support aspect ratio 1:4/,
131  );
132  assert.throws(
133    () => validateArgs(GEMINI_MODEL, makeArgs({ size: "2048x1024" })),
134    /does not support size 2048x1024 \(aspect ratio 2:1\)/,
135  );
136});
137 
138test("OpenRouter response extraction supports inline image data and finish_reason errors", async () => {
139  const bytes = await extractImageFromResponse({
140    choices: [
141      {
142        message: {
143          images: [
144            {
145              image_url: {
146                url: `data:image/png;base64,${Buffer.from("hello").toString("base64")}`,
147              },
148            },
149          ],
150        },
151      },
152    ],
153  });
154  assert.equal(Buffer.from(bytes).toString("utf8"), "hello");
155 
156  await assert.rejects(
157    () =>
158      extractImageFromResponse({
159        choices: [
160          {
161            finish_reason: "error",
162            native_finish_reason: "MALFORMED_FUNCTION_CALL",
163            message: { content: null },
164          },
165        ],
166      }),
167    /finish_reason=MALFORMED_FUNCTION_CALL/,
168  );
169});
170