diff --git a/tools/server/webui/src/components/ChatMessage.tsx b/tools/server/webui/src/components/ChatMessage.tsx index ee59de450d1ff..5af2c75e7fd0f 100644 --- a/tools/server/webui/src/components/ChatMessage.tsx +++ b/tools/server/webui/src/components/ChatMessage.tsx @@ -37,7 +37,7 @@ export default function ChatMessage({ onChangeSibling(sibling: Message['id']): void; isPending?: boolean; }) { - const { viewingChat, config } = useAppContext(); + const { viewingChat, config, serverProps } = useAppContext(); const [editingContent, setEditingContent] = useState(null); const timings = useMemo( () => @@ -189,11 +189,18 @@ export default function ChatMessage({ )} + + {/* Show model name only after AI messages */} + {msg.role === 'assistant' && serverProps?.model_path && ( +
+ {serverProps.model_path.split(/(\\|\/)/).pop()} +
+ )} )} - + {/* actions for each message */} {msg.content !== null && (
{ + async createConversation( + name: string, + modelName?: string + ): Promise { const now = Date.now(); const msgId = now; const conv: Conversation = { @@ -101,6 +110,7 @@ const StorageUtils = { lastModified: now, currNode: msgId, name, + modelName, }; await db.conversations.add(conv); // create a root node diff --git a/tools/server/webui/src/utils/types.ts b/tools/server/webui/src/utils/types.ts index ea7d641dc748b..35ba58a1b143c 100644 --- a/tools/server/webui/src/utils/types.ts +++ b/tools/server/webui/src/utils/types.ts @@ -103,6 +103,7 @@ export interface Conversation { lastModified: number; // timestamp from Date.now() currNode: Message['id']; // the current message node being viewed name: string; + modelName?: string; // optional model name extracted from serverProps } export interface ViewingChat {