[Bug]: Images from Flux tool not displayed by the agent for finetuned model

[leondape]

What happened?

When using the Flux tool with finetuned models in an agent context, the images are not being displayed in the chat interface. Instead, only a default message is shown: "Flux displayed an image. All generated images are already plainly visible, so don't repeat the descriptions in detail. Do not list download links as they are available in the UI already. The user may download the images by clicking on them, but do not mention anything about downloading to the user."

Expected behavior: The generated image should be displayed in the chat interface, just like it does when using non-finetuned models.

Both methods should be handling the image flow the same way (via isAgent == true) however there is different behavior.

Version Information

52f146d

Steps to Reproduce

1. Set up an agent with the Flux tool enabled
2. Use the agent to generate an image using a finetuned model (either /v1/flux-pro-finetuned or /v1/flux-pro-1.1-ultra-finetuned endpoint)
3. Provide a prompt for image generation
4. Wait for the generation to complete
5. Observe that while the tool executes successfully, no image is displayed in the chat interface
6. Only the default message appears instead of the actual image

What browsers are you seeing the problem on?

No response

Relevant log output

Screenshots

No response

Code of Conduct

• [x] I agree to follow this project's Code of Conduct