diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte
index e333215af9..2622c92029 100644
--- a/src/lib/components/chat/Messages/ResponseMessage.svelte
+++ b/src/lib/components/chat/Messages/ResponseMessage.svelte
@@ -665,12 +665,12 @@
{/if}
- {#if message?.files && message.files?.filter((f) => f.type === 'image').length > 0}
+ {#if message?.files && message.files?.filter( (f) => ['image', 'file'].includes(f.type) ).length > 0}
- {#each message.files as file}
+ {#each message.files.filter((f) => ['image', 'file'].includes(f.type)) as file}
{#if file.type === 'image' || (file?.content_type ?? '').startsWith('image/')}