mirror of
https://github.com/JimLiu/baoyu-skills.git
synced 2026-07-12 05:51:44 +08:00
Jim Liu 宝玉
26 lines
1.0 KiB
TypeScript
26 lines
1.0 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
|
|
import { collect_generated_image_urls_from_response_parts } from "./client.ts";
|
|
|
|
test("response part fallback finds generated images when legacy generated markers are absent", () => {
|
|
const generatedUrl = "https://lh3.googleusercontent.com/gg-dl/example-generated-image";
|
|
const initialCandidate = ["rcid-1", ["image generated successfully"]];
|
|
const imageCandidate = [
|
|
"rcid-1",
|
|
["image generated successfully"],
|
|
{ nestedPayload: [{ media: generatedUrl }] },
|
|
];
|
|
const responseJson = [
|
|
["wrb.fr", null, JSON.stringify([null, [], null, null, [initialCandidate]])],
|
|
["wrb.fr", null, JSON.stringify([null, [], null, null, [imageCandidate]])],
|
|
];
|
|
|
|
assert.equal(initialCandidate[12], undefined);
|
|
assert.equal(
|
|
/http:\/\/googleusercontent\.com\/image_generation_content\/\d+/.test(String(initialCandidate[1]?.[0])),
|
|
false,
|
|
);
|
|
assert.deepEqual(collect_generated_image_urls_from_response_parts(responseJson, 0, 0), [generatedUrl]);
|
|
});
|