Showing used model for generated message (#2003)

drizzle/0020_nebulous_patch.sql

+ALTER TABLE `messages` ADD `model` text;
\ No newline at end of file

drizzle/meta/_journal.json

@@ -141,6 +141,13 @@
       "when": 1766529533747,
       "tag": "0019_cute_carnage",
       "breakpoints": true
+    },
+    {
+      "idx": 20,
+      "version": "6",
+      "when": 1766619976318,
+      "tag": "0020_nebulous_patch",
+      "breakpoints": true
     }
   ]
 }
\ No newline at end of file

src/components/chat/ChatMessage.tsx

@@ -13,6 +13,7 @@ import {
   Copy,
   Check,
   Info,
+  Bot,
 } from "lucide-react";
 import { formatDistanceToNow, format } from "date-fns";
 import { useVersions } from "@/hooks/useVersions";
@@ -158,10 +159,7 @@ const ChatMessage = ({ message, isLastMessage }: ChatMessageProps) => {
           message.approvalState ? (
             <div
               className={`mt-2 flex items-center ${
-                message.role === "assistant" &&
-                message.content &&
-                !isStreaming &&
-                message.approvalState
+                message.role === "assistant" && message.content && !isStreaming
                   ? "justify-between"
                   : ""
               } text-xs`}
@@ -191,21 +189,29 @@ const ChatMessage = ({ message, isLastMessage }: ChatMessageProps) => {
                     </Tooltip>
                   </TooltipProvider>
                 )}
-              {message.approvalState && (
-                <div className="flex items-center space-x-1">
-                  {message.approvalState === "approved" ? (
-                    <>
-                      <CheckCircle className="h-4 w-4 text-green-500" />
-                      <span>Approved</span>
-                    </>
-                  ) : message.approvalState === "rejected" ? (
-                    <>
-                      <XCircle className="h-4 w-4 text-red-500" />
-                      <span>Rejected</span>
-                    </>
-                  ) : null}
-                </div>
-              )}
+              <div className="flex flex-wrap gap-2">
+                {message.approvalState && (
+                  <div className="flex items-center space-x-1">
+                    {message.approvalState === "approved" ? (
+                      <>
+                        <CheckCircle className="h-4 w-4 text-green-500" />
+                        <span>Approved</span>
+                      </>
+                    ) : message.approvalState === "rejected" ? (
+                      <>
+                        <XCircle className="h-4 w-4 text-red-500" />
+                        <span>Rejected</span>
+                      </>
+                    ) : null}
+                  </div>
+                )}
+                {message.role === "assistant" && message.model && (
+                  <div className="flex items-center gap-1 text-gray-500 dark:text-gray-400 w-full sm:w-auto">
+                    <Bot className="h-4 w-4 flex-shrink-0" />
+                    <span>{message.model}</span>
+                  </div>
+                )}
+              </div>
             </div>
           ) : null}
         </div>

src/db/schema.ts

@@ -89,6 +89,8 @@ export const messages = sqliteTable("messages", {
   requestId: text("request_id"),
   // Max tokens used for this message (only for assistant messages)
   maxTokensUsed: integer("max_tokens_used"),
+  // Model name used for this message (only for assistant messages)
+  model: text("model"),
   // AI SDK messages (v5 envelope) for preserving tool calls/results in agent mode
   aiMessagesJson: text("ai_messages_json", {
     mode: "json",

src/ipc/handlers/chat_stream_handlers.ts

@@ -422,6 +422,7 @@ ${componentSnippet}
           role: "assistant",
           content: "", // Start with empty content
           requestId: dyadRequestId,
+          model: settings.selectedModel.name,
           sourceCommitHash: await getCurrentCommitHash({
             path: getDyadAppPath(chat.app.path),
           }),

src/ipc/ipc_types.ts

@@ -85,6 +85,7 @@ export interface Message {
   createdAt?: Date | string;
   requestId?: string | null;
   totalTokens?: number | null;
+  model?: string | null;
 }
 
 export interface Chat {