feat(image): complete reviewed generation delivery
Memind CI / Test, build, and release guards (pull_request) Successful in 2m47s

This commit is contained in:
john
2026-07-20 20:04:44 +08:00
parent fc3d27aae0
commit 4b15610aa8
42 changed files with 1764 additions and 97 deletions
+199 -1
View File
@@ -3,6 +3,21 @@ import fs from 'node:fs';
import test from 'node:test';
import { createMindSpaceImageGenerationService } from './mindspace-image-generation.mjs';
function passingReviewService(overrides = {}) {
return {
async review() {
return {
ok: true,
pass: true,
verdict: 'pass',
relevanceScore: 92,
minimumScore: 70,
...overrides,
};
},
};
}
test('MindSpace asset schema accepts image_make as a source type', () => {
const schemaSql = fs.readFileSync(new URL('./schema.sql', import.meta.url), 'utf8');
const dbSource = fs.readFileSync(new URL('./db.mjs', import.meta.url), 'utf8');
@@ -44,18 +59,31 @@ test('MindSpace image generation stores the result before acknowledging temporar
},
async acknowledge(jobId, assetId) { order.push(['ack', jobId, assetId]); },
},
imageReviewService: {
async review() {
order.push(['review']);
return passingReviewService().review();
},
},
assetService: {
async createChatAsset(userId, input) {
order.push(['store', userId, input.sourceType, input.categoryCode]);
return { id: 'asset-1', publicUrl: 'https://example.test/image.webp' };
return {
id: 'asset-1',
publicUrl: 'https://example.test/image.webp',
workspaceRelativePath: 'public/images/2026-07-20/hero.webp',
};
},
},
});
const result = await service.generate({ userId: 'user-1', purpose: 'hero', prompt: 'x' });
assert.equal(result.ok, true);
assert.equal(result.asset.id, 'asset-1');
assert.equal(result.asset.workspaceRelativePath, 'public/images/2026-07-20/hero.webp');
assert.equal(result.asset.htmlSrc, 'images/2026-07-20/hero.webp');
assert.deepEqual(order, [
['generate', 'memind_dark_hero'],
['review'],
['store', 'user-1', 'image_make', 'public'],
['ack', 'job-1', 'asset-1'],
]);
@@ -76,3 +104,173 @@ test('MindSpace image generation converts provider failures into an explicit fal
message: '图片生成暂不可用,已保留原有流程',
});
});
test('MindSpace image generation does not return success without an HTML-ready source path', async () => {
const service = createMindSpaceImageGenerationService({
configService: { async resolveImageGenerationPurpose() { return { ok: true, presetId: 'x' }; } },
assetService: { async createChatAsset() { return { id: 'asset-1' }; } },
imageMakeClient: {
async generateImage() {
return {
jobId: 'job-1', buffer: Buffer.from('image'), mimeType: 'image/webp',
sha256: 'abc', width: 512, height: 512,
};
},
async acknowledge() {},
},
imageReviewService: passingReviewService(),
logger: { warn() {} },
});
const result = await service.generate({ userId: 'user-1', purpose: 'hero', prompt: 'x' });
assert.deepEqual(result, {
ok: false,
fallback: true,
code: 'IMAGE_ASSET_PATH_UNAVAILABLE',
message: '图片生成暂不可用,已保留原有流程',
});
});
test('MindSpace image generation applies an optional per-purpose size override', async () => {
let request = null;
const service = createMindSpaceImageGenerationService({
configService: { async resolveImageGenerationPurpose() { return { ok: true, presetId: 'memind_dark_hero' }; } },
assetService: {
async createChatAsset() {
return { id: 'asset-1', workspaceRelativePath: 'public/images/2026-07-20/hero.webp' };
},
},
imageMakeClient: {
async generateImage(input) {
request = input;
return {
jobId: 'job-1', buffer: Buffer.from('image'), mimeType: 'image/webp',
sha256: 'abc', width: input.width, height: input.height,
};
},
async acknowledge() {},
},
imageReviewService: passingReviewService(),
env: { IMAGE_MAKE_HERO_WIDTH: '768', IMAGE_MAKE_HERO_HEIGHT: '432' },
});
const result = await service.generate({ userId: 'user-1', purpose: 'hero', prompt: 'x' });
assert.equal(result.ok, true);
assert.equal(request.width, 768);
assert.equal(request.height, 432);
});
test('MindSpace image generation retries a semantic mismatch and stores only a reviewed result', async () => {
const generatedRequests = [];
let reviewCount = 0;
let storeCount = 0;
const service = createMindSpaceImageGenerationService({
configService: { async resolveImageGenerationPurpose() { return { ok: true, presetId: 'hero' }; } },
assetService: {
async createChatAsset() {
storeCount += 1;
return { id: 'asset-pass', workspaceRelativePath: 'public/images/hero-pass.webp' };
},
},
imageMakeClient: {
async generateImage(input) {
generatedRequests.push(input);
return {
jobId: `job-${generatedRequests.length}`,
buffer: Buffer.from(`image-${generatedRequests.length}`),
mimeType: 'image/webp', sha256: 'abc', width: 768, height: 432,
};
},
async acknowledge() {},
},
imageReviewService: {
async review() {
reviewCount += 1;
if (reviewCount === 1) {
return {
ok: true, pass: false, verdict: 'reject', relevanceScore: 0, minimumScore: 70,
observedSubjects: ['人物', '沙发'],
missingRequiredElements: ['竹海'],
reason: '主体不符',
};
}
return passingReviewService().review();
},
},
env: { IMAGE_MAKE_SEMANTIC_REVIEW_MAX_ATTEMPTS: '3' },
logger: { warn() {} },
});
const result = await service.generate({
userId: 'user-1', purpose: 'hero', prompt: '安吉竹海', idempotencyKey: 'imgreq-review',
});
assert.equal(result.ok, true);
assert.equal(result.jobId, 'job-2');
assert.equal(storeCount, 1);
assert.equal(generatedRequests.length, 2);
assert.equal(generatedRequests[0].idempotencyKey, 'imgreq-review');
assert.equal(generatedRequests[1].idempotencyKey, 'imgreq-review-review-2');
assert.match(generatedRequests[1].prompt, /必须修正上次错误/);
assert.match(generatedRequests[1].prompt, /缺失元素:竹海/);
});
test('MindSpace image generation blocks storage after all semantic review attempts fail', async () => {
let generateCount = 0;
let storeCount = 0;
const service = createMindSpaceImageGenerationService({
configService: { async resolveImageGenerationPurpose() { return { ok: true, presetId: 'hero' }; } },
assetService: { async createChatAsset() { storeCount += 1; } },
imageMakeClient: {
async generateImage() {
generateCount += 1;
return {
jobId: `job-${generateCount}`, buffer: Buffer.from('bad'), mimeType: 'image/webp',
sha256: 'abc', width: 768, height: 432,
};
},
},
imageReviewService: {
async review() {
return {
ok: true, pass: false, verdict: 'reject', relevanceScore: 5, minimumScore: 70,
observedSubjects: ['人物'], missingRequiredElements: ['竹海'], reason: '图文不符',
};
},
},
env: { IMAGE_MAKE_SEMANTIC_REVIEW_MAX_ATTEMPTS: '2' },
logger: { warn() {} },
});
const result = await service.generate({ userId: 'user-1', purpose: 'hero', prompt: '安吉竹海' });
assert.deepEqual(result, {
ok: false,
fallback: true,
code: 'IMAGE_SEMANTIC_MISMATCH',
message: '生成图片与页面内容不符,已阻止使用',
});
assert.equal(generateCount, 2);
assert.equal(storeCount, 0);
});
test('MindSpace image generation fails closed when semantic review is unavailable', async () => {
let storeCount = 0;
const service = createMindSpaceImageGenerationService({
configService: { async resolveImageGenerationPurpose() { return { ok: true, presetId: 'hero' }; } },
assetService: { async createChatAsset() { storeCount += 1; } },
imageMakeClient: {
async generateImage() {
return {
jobId: 'job-1', buffer: Buffer.from('image'), mimeType: 'image/webp',
sha256: 'abc', width: 768, height: 432,
};
},
},
imageReviewService: { async review() { return { ok: false, code: 'IMAGE_REVIEW_UNAVAILABLE' }; } },
logger: { warn() {} },
});
const result = await service.generate({ userId: 'user-1', purpose: 'hero', prompt: '安吉竹海' });
assert.equal(result.ok, false);
assert.equal(result.code, 'IMAGE_REVIEW_UNAVAILABLE');
assert.equal(storeCount, 0);
});