Skip to content

Commit 7628bc5

Browse files
committed
don't include ensureOptInAndSend in useAssistantActions()
1 parent eef29cb commit 7628bc5

File tree

2 files changed

+17
-5
lines changed

2 files changed

+17
-5
lines changed

packages/compass-assistant/src/assistant-chat.tsx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
import React, { useCallback } from 'react';
1+
import React, { useCallback, useContext } from 'react';
22
import type { AssistantMessage } from './compass-assistant-provider';
3-
import { useAssistantActions } from './compass-assistant-provider';
3+
import { AssistantActionsContext } from './compass-assistant-provider';
44
import type { Chat } from './@ai-sdk/react/chat-react';
55
import { useChat } from './@ai-sdk/react/use-chat';
66
import {
@@ -111,7 +111,7 @@ export const AssistantChat: React.FunctionComponent<AssistantChatProps> = ({
111111
}) => {
112112
const track = useTelemetry();
113113
const darkMode = useDarkMode();
114-
const { ensureOptInAndSend } = useAssistantActions();
114+
const { ensureOptInAndSend } = useContext(AssistantActionsContext);
115115
const { messages, status, error, clearError } = useChat({
116116
chat,
117117
onError: (error) => {

packages/compass-assistant/src/compass-assistant-provider.tsx

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,13 +73,25 @@ export const AssistantActionsContext =
7373
ensureOptInAndSend: async () => {},
7474
});
7575

76-
export function useAssistantActions(): AssistantActionsContextType & {
76+
export function useAssistantActions(): Omit<
77+
AssistantActionsContextType,
78+
'ensureOptInAndSend'
79+
> & {
7780
isAssistantEnabled: boolean;
7881
} {
7982
const isAssistantEnabled = usePreference('enableAIAssistant');
83+
const {
84+
interpretExplainPlan,
85+
interpretConnectionError,
86+
tellMoreAboutInsight,
87+
clearChat,
88+
} = useContext(AssistantActionsContext);
8089

8190
return {
82-
...useContext(AssistantActionsContext),
91+
interpretExplainPlan,
92+
interpretConnectionError,
93+
tellMoreAboutInsight,
94+
clearChat,
8395
isAssistantEnabled,
8496
};
8597
}

0 commit comments

Comments
 (0)