iod/page-assist
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Update message formatting to remove model suffix in chat display

Browse Source
This commit is contained in:
n4ze3m 2025-01-19 12:52:43 +05:30
parent 6be1b2b0de
commit b39d60fc3c
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/components/Layouts/MoreOptions.tsx
View File

@ -11,6 +11,7 @@ import { Message } from "@/types/message"
import { useState } from "react" import { useState } from "react"
import { ShareModal } from "../Common/ShareModal" import { ShareModal } from "../Common/ShareModal"
import { useTranslation } from "react-i18next" import { useTranslation } from "react-i18next"
import { removeModelSuffix } from "@/db/models"
interface MoreOptionsProps { interface MoreOptionsProps {
messages: Message[] messages: Message[]
@ -25,11 +26,10 @@ const formatAsText = (messages: Message[]) => {
}) })
.join("\n\n") .join("\n\n")
} }
const formatAsMarkdown = (messages: Message[]) => { const formatAsMarkdown = (messages: Message[]) => {
return messages return messages
.map((msg) => { .map((msg) => {
let content = `**${msg.isBot ? msg.name : "You"}**:\n${msg.message}` let content = `**${msg.isBot ? removeModelSuffix(msg.name?.replaceAll(/accounts\/[^\/]+\/models\//g, "")) : "You"}**:\n${msg.message}`
if (msg.images && msg.images.length > 0) { if (msg.images && msg.images.length > 0) {
const imageMarkdown = msg.images const imageMarkdown = msg.images
@ -119,7 +119,6 @@ const generateChatImage = async (messages: Message[]) => {
} }
totalHeight += 30 totalHeight += 30
}) })
canvas.height = totalHeight canvas.height = totalHeight
@ -132,7 +131,7 @@ const generateChatImage = async (messages: Message[]) => {
for (const msg of messages) { for (const msg of messages) {
ctx.font = "bold 18px Inter, Arial" ctx.font = "bold 18px Inter, Arial"
ctx.fillStyle = msg.isBot ? "#1A202C" : "#1E4E8C" ctx.fillStyle = msg.isBot ? "#1A202C" : "#1E4E8C"
ctx.fillText(`${msg.isBot ? msg.name : "You"}:`, padding, yPosition) ctx.fillText(`${msg.isBot ? removeModelSuffix(msg.name?.replaceAll(/accounts\/[^\/]+\/models\//g, "")) : "You"}:`, padding, yPosition)
yPosition += 35 yPosition += 35
if (msg.message.includes("```")) { if (msg.message.includes("```")) {