bugfix, fix tests

Author: rsergeenko

redisinsight/ui/src/components/side-panels/panels/ai-assistant/components/assistance-chat/AssistanceChat.spec.tsx

@@ -131,8 +131,9 @@ describe('AssistanceChat', () => {
     fireEvent.click(screen.getByTestId('ai-general-restart-session-btn'))
 
     await waitForEuiPopoverVisible()
-
-    fireEvent.click(screen.getByTestId('ai-chat-restart-confirm'))
+    await act(async () => {
+      fireEvent.click(screen.getByTestId('ai-chat-restart-confirm'))
+    })
 
     expect(store.getActions()).toEqual([
       ...afterRenderActions,

redisinsight/ui/src/components/side-panels/panels/ai-assistant/components/assistance-chat/AssistanceChat.tsx

@@ -16,7 +16,7 @@ import { AiChatMessage, AiChatType } from 'uiSrc/slices/interfaces/aiAssistant'
 import { connectedInstanceSelector } from 'uiSrc/slices/instances/instances'
 import { appRedisCommandsSelector } from 'uiSrc/slices/app/redis-commands'
 
-import { AssistanceChatInitialMessage, ChatHistory, ChatForm } from '../shared'
+import { AssistanceChatInitialMessage, ChatHistory, ChatForm, RestartChat } from '../shared'
 
 import styles from './styles.module.scss'
 
@@ -121,13 +121,17 @@ const AssistanceChat = () => {
     <div className={styles.wrapper} data-testid="ai-general-chat">
       <div className={styles.header}>
         <span />
-        <EuiButtonEmpty
-          disabled={!!inProgressMessage || !messages?.length}
-          iconType="eraser"
-          size="xs"
-          onClick={onClearSession}
-          className={styles.headerBtn}
-          data-testid="ai-general-restart-session-btn"
+        <RestartChat
+          button={(
+            <EuiButtonEmpty
+              disabled={!!inProgressMessage || !messages?.length}
+              iconType="eraser"
+              size="xs"
+              className={styles.headerBtn}
+              data-testid="ai-general-restart-session-btn"
+            />
+          )}
+          onConfirm={onClearSession}
         />
       </div>
       <div className={styles.chatHistory}>

redisinsight/ui/src/components/side-panels/panels/ai-assistant/components/shared/error-message/ErrorMessage.spec.tsx

@@ -16,7 +16,7 @@ describe('ErrorMessage', () => {
     expect(screen.getByTestId('ai-chat-error-message')).toHaveTextContent(AiChatErrors.Default)
     expect(screen.getByTestId('ai-chat-error-report-link')).toBeInTheDocument()
 
-    expect(screen.getByTestId('ai-general-restart-session-btn')).toBeInTheDocument()
+    expect(screen.getByTestId('ai-chat-error-restart-session-btn')).toBeInTheDocument()
   })
 
   it('should not render restart button with timeout error', () => {
@@ -26,6 +26,6 @@ describe('ErrorMessage', () => {
     expect(screen.getByTestId('ai-chat-error-message')).toHaveTextContent(AiChatErrors.Timeout)
     expect(screen.getByTestId('ai-chat-error-report-link')).toBeInTheDocument()
 
-    expect(screen.queryByTestId('ai-general-restart-session-btn')).not.toBeInTheDocument()
+    expect(screen.queryByTestId('ai-chat-error-restart-session-btn')).not.toBeInTheDocument()
   })
 })

redisinsight/ui/src/components/side-panels/panels/ai-assistant/components/shared/error-message/ErrorMessage.tsx

@@ -54,7 +54,7 @@ const ErrorMessage = (props: Props) => {
               color="secondary"
               iconType="eraser"
               className={styles.restartSessionBtn}
-              data-testid="ai-general-restart-session-btn"
+              data-testid="ai-chat-error-restart-session-btn"
             >
               Restart session
             </EuiButton>

redisinsight/ui/src/utils/api/chatbots.ts

@@ -2,7 +2,7 @@ import { CustomHeaders } from 'uiSrc/constants/api'
 import { isStatusSuccessful } from 'uiSrc/utils'
 import ApiStatusCode from '../../constants/apiStatusCode'
 
-const TIMEOUT_FOR_MESSAGE_REQUEST = 50_000
+const TIMEOUT_FOR_MESSAGE_REQUEST = 30_000
 
 export const getStreamedAnswer = async (
   url: string,