diff --git a/src/components/Common/Playground/MessageSource.tsx b/src/components/Common/Playground/MessageSource.tsx
index d0a006a..b0d3ad9 100644
--- a/src/components/Common/Playground/MessageSource.tsx
+++ b/src/components/Common/Playground/MessageSource.tsx
@@ -13,7 +13,7 @@ type Props = {
 }
 
 export const MessageSource: React.FC<Props> = ({ source, onSourceClick }) => {
-  if (source?.mode === "rag") {
+  if (source?.mode === "rag" || source?.mode === "chat") {
     return (
       <button
         onClick={() => {