feat: update backend to support MDS

CHANGELOG.md

@@ -14,3 +14,4 @@ Inspired from [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 - Fetch root agent id before executing the agent ([#165](https://github.com/opensearch-project/dashboards-assistant/pull/165))
 - Integrate chatbot with sidecar service ([#164](https://github.com/opensearch-project/dashboards-assistant/pull/164))
 - Add data source service ([#191](https://github.com/opensearch-project/dashboards-assistant/pull/191))
+- Update router and controller to support MDS ([#190](https://github.com/opensearch-project/dashboards-assistant/pull/190))

opensearch_dashboards.json

@@ -16,6 +16,6 @@
     "dataSourceManagement"
   ],
   "configPath": [
-    "assistant"
+    "assistant" 
   ]
 }

public/chat_flyout.tsx

@@ -5,13 +5,16 @@
 
 import { EuiResizableContainer } from '@elastic/eui';
 import cs from 'classnames';
-import React, { useRef } from 'react';
+import React, { useMemo, useRef } from 'react';
+import { useObservable } from 'react-use';
+import { BehaviorSubject } from 'rxjs';
 import { useChatContext } from './contexts/chat_context';
 import { ChatPage } from './tabs/chat/chat_page';
 import { ChatWindowHeader } from './tabs/chat_window_header';
 import { ChatHistoryPage } from './tabs/history/chat_history_page';
 import { AgentFrameworkTracesFlyoutBody } from './components/agent_framework_traces_flyout_body';
 import { TAB_ID } from './utils/constants';
+import { useCore } from './contexts';
 
 interface ChatFlyoutProps {
   flyoutVisible: boolean;
@@ -22,7 +25,17 @@ interface ChatFlyoutProps {
 export const ChatFlyout = (props: ChatFlyoutProps) => {
   const chatContext = useChatContext();
   const chatHistoryPageLoadedRef = useRef(false);
+  const core = useCore();
+  const defaultDataSourceSelection$ = new BehaviorSubject(new Map());
 
+  // TODO: use DataSourceService to replace
+  const selectedDS = useObservable(
+    core.services.setupDeps?.dataSourceManagement?.dataSourceSelection?.getSelection$() ??
+      defaultDataSourceSelection$
+  );
+  const currentDS = useMemo(() => {
+    return selectedDS?.values().next().value;
+  }, [selectedDS]);
   let chatPageVisible = false;
   let chatHistoryPageVisible = false;
   let chatTraceVisible = false;
@@ -87,6 +100,10 @@ export const ChatFlyout = (props: ChatFlyoutProps) => {
       <>
         <div className={cs('llm-chat-flyout-header')}>
           <ChatWindowHeader />
+          <div>
+            Data Source:
+            {currentDS?.[0]?.label}
+          </div>
         </div>
 
         {props.overrideComponent}

public/components/__tests__/chat_window_header_title.test.tsx

@@ -17,6 +17,7 @@ import * as coreContextExports from '../../contexts/core_context';
 import { IMessage } from '../../../common/types/chat_saved_object_attributes';
 
 import { ChatWindowHeaderTitle } from '../chat_window_header_title';
+import { DataSourceServiceMock } from '../../services/data_source_service.mock';
 
 const setup = ({
   messages = [],
@@ -37,6 +38,7 @@ const setup = ({
         }),
         reload: jest.fn(),
       },
+      dataSource: new DataSourceServiceMock(),
     },
   };
   useCoreMock.services.http.put.mockImplementation(() => Promise.resolve());

public/components/__tests__/edit_conversation_name_modal.test.tsx

@@ -15,10 +15,11 @@ import {
   EditConversationNameModalProps,
 } from '../edit_conversation_name_modal';
 import { HttpHandler } from '../../../../../src/core/public';
+import { DataSourceServiceMock } from '../../services/data_source_service.mock';
 
 const setup = ({ onClose, defaultTitle, conversationId }: EditConversationNameModalProps) => {
   const useCoreMock = {
-    services: coreMock.createStart(),
+    services: { ...coreMock.createStart(), dataSource: new DataSourceServiceMock() },
   };
   jest.spyOn(coreContextExports, 'useCore').mockReturnValue(useCoreMock);
 
@@ -156,7 +157,10 @@ describe('<EditConversationNameModal />', () => {
     expect(useCoreMock.services.http.put).not.toHaveBeenCalled();
 
     fireEvent.click(renderResult.getByTestId('confirmModalConfirmButton'));
-    expect(useCoreMock.services.http.put).toHaveBeenCalled();
+
+    await waitFor(() => {
+      expect(useCoreMock.services.http.put).toHaveBeenCalled();
+    });
 
     fireEvent.click(renderResult.getByTestId('confirmModalCancelButton'));