diff --git a/web/app/components/app/chat/mermaid/index.tsx b/web/app/components/app/chat/mermaid/index.tsx index af0b68e6580319..52c0a2020974ff 100644 --- a/web/app/components/app/chat/mermaid/index.tsx +++ b/web/app/components/app/chat/mermaid/index.tsx @@ -32,6 +32,14 @@ const Flowchart = React.forwardRef((props: { const chartId = useRef(`flowchart_${CryptoJS.MD5(props.PrimitiveCode).toString()}`) const [isRender, setIsRender] = useState(true) + const clearFlowchartCache = () => { + for (let i = localStorage.length - 1; i >= 0; --i) { + const key = localStorage.key(i) + if (key && key.startsWith('flowchart_')) + localStorage.removeItem(key) + } + } + const renderFlowchart = async (PrimitiveCode: string) => { try { const cachedSvg: any = localStorage.getItem(chartId.current) @@ -44,15 +52,15 @@ const Flowchart = React.forwardRef((props: { const svgGraph = await mermaidAPI.render(chartId.current, PrimitiveCode) // eslint-disable-next-line @typescript-eslint/no-use-before-define const base64Svg: any = await svgToBase64(svgGraph.svg) - localStorage.setItem(chartId.current, base64Svg) setSvgCode(base64Svg) + if (chartId.current && base64Svg) + localStorage.setItem(chartId.current, base64Svg) } } catch (error) { - localStorage.clear() - // eslint-disable-next-line @typescript-eslint/ban-ts-comment - // @ts-expect-error - console.error(error.toString()) + clearFlowchartCache() + // eslint-disable-next-line @typescript-eslint/no-use-before-define + handleReRender() } } @@ -70,7 +78,9 @@ const Flowchart = React.forwardRef((props: { const handleReRender = () => { setIsRender(false) setSvgCode(null) - localStorage.removeItem(chartId.current) + if (chartId.current) + localStorage.removeItem(chartId.current) + setTimeout(() => { setIsRender(true) renderFlowchart(props.PrimitiveCode) @@ -90,7 +100,10 @@ const Flowchart = React.forwardRef((props: { // @ts-expect-error
{ - isRender &&
{svgCode && (Mermaid chart)}
+ isRender + &&
+ {svgCode && Mermaid chart} +
}