chore: remove eslint disables, fix types on plugin props

Author: gagik

packages/compass-assistant/src/docs-provider-transport.ts

@@ -1,5 +1,4 @@
 import {
-  type ChatRequestOptions,
   type ChatTransport,
   type UIMessage,
   type UIMessageChunk,
@@ -18,8 +17,7 @@ export class DocsProviderTransport implements ChatTransport<UIMessage> {
     });
   }
 
-  // eslint-disable-next-line @typescript-eslint/require-await
-  async sendMessages({
+  sendMessages({
     messages,
     abortSignal,
   }: Parameters<ChatTransport<UIMessage>['sendMessages']>[0]) {
@@ -29,15 +27,11 @@ export class DocsProviderTransport implements ChatTransport<UIMessage> {
       abortSignal: abortSignal,
     });
 
-    return result.toUIMessageStream();
+    return Promise.resolve(result.toUIMessageStream());
   }
 
-  // eslint-disable-next-line @typescript-eslint/require-await
-  async reconnectToStream(
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    options: ChatRequestOptions
-  ): Promise<ReadableStream<UIMessageChunk> | null> {
+  reconnectToStream(): Promise<ReadableStream<UIMessageChunk> | null> {
     // For this implementation, we don't support reconnecting to streams
-    return null;
+    return Promise.resolve(null);
   }
 }

packages/compass-assistant/src/index.tsx

@@ -3,11 +3,24 @@ import { AssistantProvider } from './assistant-provider';
 import { Chat } from './@ai-sdk/react/chat-react';
 import { DocsProviderTransport } from './docs-provider-transport';
 import { atlasServiceLocator } from '@mongodb-js/atlas-service/provider';
+import type { PropsWithChildren } from 'react';
+import type { UIMessage } from 'ai';
+import React from 'react';
 
 const CompassAssistantProvider = registerCompassPlugin(
   {
     name: 'CompassAssistant',
-    component: AssistantProvider,
+    component: ({
+      chat,
+      children,
+    }: PropsWithChildren<{
+      chat?: Chat<UIMessage>;
+    }>) => {
+      if (!chat) {
+        throw new Error('Chat was not provided by the state');
+      }
+      return <AssistantProvider chat={chat}>{children}</AssistantProvider>;
+    },
     activate: (initialProps, { atlasService }) => {
       // TODO: We will temporarily default to the staging url until the docs
       // API is deployed to the production environment.