From db96b10d4d23a2a1709468886ff73534188cf826 Mon Sep 17 00:00:00 2001
From: Francisco Javier Arceo <farceo@redhat.com>
Date: Mon, 18 Aug 2025 14:12:39 -0400
Subject: [PATCH] chore: Adding unit tests for UI

Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
---
 .../[id]/files/[fileId]/contents/page.tsx              | 10 +++++++++-
 .../components/chat-playground/markdown-renderer.tsx   |  1 +
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/llama_stack/ui/app/logs/vector-stores/[id]/files/[fileId]/contents/page.tsx b/llama_stack/ui/app/logs/vector-stores/[id]/files/[fileId]/contents/page.tsx
index 0283db9e7..3d714a480 100644
--- a/llama_stack/ui/app/logs/vector-stores/[id]/files/[fileId]/contents/page.tsx
+++ b/llama_stack/ui/app/logs/vector-stores/[id]/files/[fileId]/contents/page.tsx
@@ -52,8 +52,10 @@ export default function ContentsListPage() {
   const [file, setFile] = useState<VectorStoreFile | null>(null);
   const [contents, setContents] = useState<VectorStoreContentItem[]>([]);
   const [isLoadingStore, setIsLoadingStore] = useState(true);
+  const [isLoadingFile, setIsLoadingFile] = useState(true);
   const [isLoadingContents, setIsLoadingContents] = useState(true);
   const [errorStore, setErrorStore] = useState<Error | null>(null);
+  const [errorFile, setErrorFile] = useState<Error | null>(null);
   const [errorContents, setErrorContents] = useState<Error | null>(null);
 
   useEffect(() => {
@@ -175,7 +177,13 @@ export default function ContentsListPage() {
           <CardTitle>Content Chunks ({contents.length})</CardTitle>
         </CardHeader>
         <CardContent>
-          {isLoadingContents ? (
+          {isLoadingFile ? (
+            <Skeleton className="h-4 w-full" />
+          ) : errorFile ? (
+            <div className="text-destructive text-sm">
+              Error loading file: {errorFile.message}
+            </div>
+          ) : isLoadingContents ? (
             <div className="space-y-2">
               <Skeleton className="h-4 w-full" />
               <Skeleton className="h-4 w-3/4" />
diff --git a/llama_stack/ui/components/chat-playground/markdown-renderer.tsx b/llama_stack/ui/components/chat-playground/markdown-renderer.tsx
index bc6bf5122..b48b5e1ba 100644
--- a/llama_stack/ui/components/chat-playground/markdown-renderer.tsx
+++ b/llama_stack/ui/components/chat-playground/markdown-renderer.tsx
@@ -187,6 +187,7 @@ const COMPONENTS = {
   code: ({
     children,
     className,
+    ...rest
   }: {
     children: React.ReactNode;
     className?: string;