[AI Playground] Move ChatContext to AIPlayground provider

packages/kbn-ai-playground/hooks/useChatContext.ts

@@ -6,7 +6,7 @@
  */
 
 import { useContext } from 'react';
-import { ChatContext } from '../providers/chat_provider';
+import { ChatContext } from '../providers/ai_playground_provider';
 
 export const useChatContext = () => {
   return useContext(ChatContext);

packages/kbn-ai-playground/providers/ai_playground_provider.tsx

@@ -6,17 +6,18 @@
  * Side Public License, v 1.
  */
 
-import React, { ReactNode } from 'react';
+import React, { createContext, ReactNode } from 'react';
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
 import { FormProvider, useForm } from 'react-hook-form';
 import { ChatForm } from '../types';
-import { ChatProvider } from '@kbn/ai-playground/providers/chat_provider';
 
 interface AIPlaygroundProviderProps {
   navigateToIndexPage: () => void;
   children: ReactNode;
 }
 
+export const ChatContext = createContext<{ navigateToIndexPage?: () => void }>({});
+
 const queryClient = new QueryClient({});
 
 export const AIPlaygroundProvider: React.FC<AIPlaygroundProviderProps> = ({
@@ -26,10 +27,10 @@ export const AIPlaygroundProvider: React.FC<AIPlaygroundProviderProps> = ({
   const form = useForm<ChatForm>();
 
   return (
-    <ChatProvider navigateToIndexPage={navigateToIndexPage}>
+    <ChatContext.Provider value={{ navigateToIndexPage }}>
       <QueryClientProvider client={queryClient}>
         <FormProvider {...form}>{children}</FormProvider>
       </QueryClientProvider>
-    </ChatProvider>
+    </ChatContext.Provider>
   );
 };