diff --git a/packages/memory-graph/src/components/memory-graph.tsx b/packages/memory-graph/src/components/memory-graph.tsx index b01bf92a..630a47c2 100644 --- a/packages/memory-graph/src/components/memory-graph.tsx +++ b/packages/memory-graph/src/components/memory-graph.tsx @@ -31,9 +31,13 @@ export function MemoryGraph({ totalCount, }: MemoryGraphProps) { const resolvedColors = useGraphTheme(colorOverrides) - const colors: GraphThemeColors = colorOverrides - ? { ...resolvedColors, ...colorOverrides } - : resolvedColors + const colors = useMemo( + () => + colorOverrides + ? { ...resolvedColors, ...colorOverrides } + : resolvedColors, + [resolvedColors, colorOverrides], + ) const [containerSize, setContainerSize] = useState({ width: 0, height: 0 }) const [containerBounds, setContainerBounds] = useState(null)