diff --git a/vite/src/ChatApp.svelte b/vite/src/ChatApp.svelte
index a959fee..42d00ba 100644
--- a/vite/src/ChatApp.svelte
+++ b/vite/src/ChatApp.svelte
@@ -452,7 +452,8 @@
     <div class="chat-container" bind:this="{chatContainer}" >
         {#each messages as message, index (index)}
         <ChatMessage message="{message}"
-        message_index={index} on:editMessage={handleEditMessage}
+        message_index={index} model_id={assistant.model}
+        on:editMessage={handleEditMessage}
         on:regenerate={handleRegenerateMessage}
         on:delete={handleDeleteMessage}
         container="{chatContainer.parentElement}" markdown={md} />
diff --git a/vite/src/ChatMessage.svelte b/vite/src/ChatMessage.svelte
index 4b38a99..2f10f2e 100644
--- a/vite/src/ChatMessage.svelte
+++ b/vite/src/ChatMessage.svelte
@@ -7,6 +7,7 @@
     export let container;
     export let markdown;
     export let message_index;
+    export let model_id;
 
     const dispatch = createEventDispatcher();
 
@@ -144,6 +145,7 @@
 	                <span class="icon-edit"></span>
 	            </button>
             {:else if message.role == 'assistant'}
+                <span class="d-inline-block py-1 px-2 me-2 border-end">{model_id}</span>
                 <button class="btn btn-primary btn-sm" title="Copier" on:click={handleCopy}>
                     <span class="icon-content_copy"></span>
                 </button>