diff --git a/packages/backend/server/src/__tests__/copilot-provider.spec.ts b/packages/backend/server/src/__tests__/copilot-provider.spec.ts index e97dd613b..a47708436 100644 --- a/packages/backend/server/src/__tests__/copilot-provider.spec.ts +++ b/packages/backend/server/src/__tests__/copilot-provider.spec.ts @@ -476,6 +476,19 @@ const actions = [ }, type: 'image' as const, }, + { + promptName: ['debug:action:dalle3'], + messages: [ + { + role: 'user' as const, + content: 'Panda', + }, + ], + verifier: (t: ExecutionContext, link: string) => { + t.truthy(checkUrl(link), 'should be a valid url'); + }, + type: 'image' as const, + }, ]; for (const { name, promptName, messages, verifier, type } of actions) { diff --git a/packages/backend/server/src/plugins/copilot/providers/openai.ts b/packages/backend/server/src/plugins/copilot/providers/openai.ts index ad7147d24..4719caf91 100644 --- a/packages/backend/server/src/plugins/copilot/providers/openai.ts +++ b/packages/backend/server/src/plugins/copilot/providers/openai.ts @@ -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);