[chat]Refactor global search chat integration with sendMessageWithWindow (#10932)

* Refactor global search chat integration with sendMessageWithWindow

Signed-off-by: Lin Wang <[email protected]>

* Changeset file for PR #10932 created/updated

* Fix failed unit tests

Signed-off-by: Lin Wang <[email protected]>

---------

Signed-off-by: Lin Wang <[email protected]>
Co-authored-by: opensearch-changeset-bot[bot] <154024398+opensearch-changeset-bot[bot]@users.noreply.github.com>

Author: wanglam

changelogs/fragments/10932.yml

@@ -0,0 +1,2 @@
+fix:
+- Global search chatbot integration in firefox ([#10932](https://github.com/opensearch-project/OpenSearch-Dashboards/pull/10932))

src/plugins/chat/public/plugin.test.ts

@@ -3,7 +3,6 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import React from 'react';
 import { ChatPlugin } from './plugin';
 import { ChatService } from './services/chat_service';
 import { toMountPoint } from '../../opensearch_dashboards_react/public';
@@ -255,17 +254,11 @@ describe('ChatPlugin', () => {
       );
     });
 
-    it('should call startNewConversation when global search action is triggered', async () => {
-      const mockStartNewConversation = jest.fn().mockResolvedValue(undefined);
-
-      // Mock React.createRef to return a ref with our mock function
-      const mockRef = {
-        current: {
-          startNewConversation: mockStartNewConversation,
-        },
-      };
-      jest.spyOn(React, 'createRef').mockReturnValue(mockRef as any);
-
+    it('should call sendMessageWithWindow when global search action is triggered', async () => {
+      const sendMessageWithWindowMock = jest.fn();
+      jest
+        .spyOn(ChatService.prototype, 'sendMessageWithWindow')
+        .mockImplementationOnce(sendMessageWithWindowMock);
       plugin.start(mockCoreStart, mockDeps);
 
       const registerCall = (mockCoreStart.chrome.globalSearch.registerSearchCommand as jest.Mock)
@@ -275,7 +268,9 @@ describe('ChatPlugin', () => {
       // Trigger the action
       await commandConfig.action({ content: 'test query' });
 
-      expect(mockStartNewConversation).toHaveBeenCalledWith({ content: 'test query' });
+      expect(sendMessageWithWindowMock).toHaveBeenCalledWith('test query', [], {
+        clearConversation: true,
+      });
     });
   });
 

src/plugins/chat/public/plugin.ts

@@ -12,11 +12,7 @@ import { Subscription } from 'rxjs';
 import { CoreStart, Plugin, PluginInitializerContext } from '../../../core/public';
 import { ChatPluginSetup, ChatPluginStart, AppPluginStartDependencies } from './types';
 import { ChatService, ChatWindowState } from './services/chat_service';
-import {
-  ChatHeaderButton,
-  ChatHeaderButtonInstance,
-  ChatLayoutMode,
-} from './components/chat_header_button';
+import { ChatHeaderButton, ChatLayoutMode } from './components/chat_header_button';
 import { toMountPoint } from '../../opensearch_dashboards_react/public';
 import { SuggestedActionsService } from './services/suggested_action';
 
@@ -94,8 +90,6 @@ export class ChatPlugin implements Plugin<ChatPluginSetup, ChatPluginStart> {
       };
     }
 
-    const chatHeaderButtonRef = React.createRef<ChatHeaderButtonInstance>();
-
     // Initialize chat service (it will use the server proxy)
     this.chatService = new ChatService();
 
@@ -115,7 +109,6 @@ export class ChatPlugin implements Plugin<ChatPluginSetup, ChatPluginStart> {
             chatService,
             contextProvider: deps.contextProvider,
             charts: deps.charts,
-            ref: chatHeaderButtonRef,
             suggestedActionsService: this.suggestedActionsService!,
           })
         );
@@ -149,7 +142,7 @@ export class ChatPlugin implements Plugin<ChatPluginSetup, ChatPluginStart> {
         ),
       ],
       action: async ({ content }: { content: string }) => {
-        await chatHeaderButtonRef.current?.startNewConversation({ content });
+        await chatService.sendMessageWithWindow(content, [], { clearConversation: true });
       },
     });