fix(server): generate image (#11599)

This commit is contained in:
DarkSky
2025-04-10 16:25:02 +08:00
committed by GitHub
parent d1bd809608
commit 44d2f301de
2 changed files with 16 additions and 3 deletions

View File

@@ -476,6 +476,19 @@ const actions = [
},
type: 'image' as const,
},
{
promptName: ['debug:action:dalle3'],
messages: [
{
role: 'user' as const,
content: 'Panda',
},
],
verifier: (t: ExecutionContext<Tester>, link: string) => {
t.truthy(checkUrl(link), 'should be a valid url');
},
type: 'image' as const,
},
];
for (const { name, promptName, messages, verifier, type } of actions) {

View File

@@ -313,9 +313,9 @@ export class OpenAIProvider
prompt,
});
return result.images
.map(image => image.base64)
.filter((v): v is string => !!v);
return result.images.map(
image => `data:image/png;base64,${image.base64}`
);
} catch (e: any) {
metrics.ai.counter('generate_images_errors').add(1, { model });
throw this.handleError(e, model, options);