added test cases for condition

ResearchAssistant/App/frontend/src/components/ChatMessageContainer/ChatMessageContainer.test.tsx

@@ -1,6 +1,6 @@
 import React from 'react'
 import { render, screen, fireEvent } from '@testing-library/react'
-import { ChatMessageContainer } from './ChatMessageContainer'
+import { ChatMessageContainer, parseCitationFromMessage } from './ChatMessageContainer'
 import { type ChatMessage } from '../../api/models'
 import { Answer } from '../Answer'
 jest.mock('remark-supersub', () => () => {})
@@ -151,4 +151,35 @@ describe('ChatMessageContainer', () => {
     fireEvent.click(buttonEle)
     expect(mockOnCitationClicked).not.toHaveBeenCalled()
   })
+
+  it('returns citations when message role is "tool" and content is valid JSON', () => {
+    const message: ChatMessage = {
+      role: 'tool',
+      content: JSON.stringify({ citations: [{ filepath: 'path/to/file', chunk_id: '1' }] }),
+      id: '1',
+      date: ''
+    }
+    const citations = parseCitationFromMessage(message)
+    expect(citations).toEqual([{ filepath: 'path/to/file', chunk_id: '1' }])
+  })
+  it('returns an empty array when message role is "tool" and content is invalid JSON', () => {
+    const message: ChatMessage = {
+      role: 'tool',
+      content: 'invalid JSON',
+      id: '1',
+      date: ''
+    }
+    const citations = parseCitationFromMessage(message)
+    expect(citations).toEqual([])
+  })
+  it('returns an empty array when message role is not "tool"', () => {
+    const message: ChatMessage = {
+      role: 'user',
+      content: JSON.stringify({ citations: [{ filepath: 'path/to/file', chunk_id: '1' }] }),
+      id: '1',
+      date: ''
+    }
+    const citations = parseCitationFromMessage(message)
+    expect(citations).toEqual([])
+  })
 })

ResearchAssistant/App/frontend/src/components/ChatMessageContainer/ChatMessageContainer.tsx

@@ -11,7 +11,7 @@ type ChatMessageContainerProps = {
   showLoadingMessage: boolean;
 };
 
-const parseCitationFromMessage = (message: ChatMessage) => {
+export const parseCitationFromMessage = (message: ChatMessage) => {
   if (message?.role && message?.role === "tool") {
     try {
       const toolMessage = JSON.parse(message.content) as ToolMessageContent;