chore: fix test, combine preference to hook

Author: gagik

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

@@ -25,7 +25,13 @@ export const AssistantContext = createContext<AssistantContextType | null>(
 );
 
 type AssistantActionsContextType = {
-  interpretExplainPlan: (explainPlan: string) => void;
+  interpretExplainPlan: ({
+    namespace,
+    explainPlan,
+  }: {
+    namespace: string;
+    explainPlan: string;
+  }) => void;
 };
 export const AssistantActionsContext =
   createContext<AssistantActionsContextType>({
@@ -49,9 +55,12 @@ export const AssistantProvider: React.FunctionComponent<
   }>
 > = ({ chat, children }) => {
   const assistantActionsContext = useRef<AssistantActionsContextType>({
-    interpretExplainPlan: (explainPlan: string) => {
+    interpretExplainPlan: ({ namespace, explainPlan }) => {
       openDrawer(ASSISTANT_DRAWER_ID);
-      const { prompt, displayText } = buildExplainPlanPrompt(explainPlan);
+      const { prompt, displayText } = buildExplainPlanPrompt({
+        namespace,
+        explainPlan,
+      });
       void chat.sendMessage(
         {
           text: prompt,

packages/compass-assistant/src/prompts.ts

@@ -1,10 +1,15 @@
-export const buildExplainPlanPrompt = (explainPlan: string) => {
+export const buildExplainPlanPrompt = ({
+  namespace,
+  explainPlan,
+}: {
+  namespace: string;
+  explainPlan: string;
+}) => {
   return {
     prompt: `Given the MongoDB explain plan output below, provide a concise human readable explanation that explains the query execution plan and highlights aspects of the plan that might impact query performance. Respond with as much concision and clarity as possible.
 If a clear optimization should be made, please suggest the optimization and describe how it can be accomplished in MongoDB Compass. Do not advise users to create indexes without weighing the pros and cons. 
 Explain output: 
 ${explainPlan}`,
-    displayText:
-      'Given this MongoDB explain plan, provide a concise human readable explanation that explains the query execution plan and highlights aspects of the plan that might impact query performance.',
+    displayText: `Given this MongoDB explain plan for ${namespace}, provide a concise human readable explanation that explains the query execution plan and highlights aspects of the plan that might impact query performance.`,
   };
 };

packages/compass-explain-plan/src/components/explain-plan-modal.tsx

@@ -132,7 +132,7 @@ export const ExplainPlanModal: React.FunctionComponent<
             }
           />
         </div>
-        {isAssistantEnabled && (
+        {isAssistantEnabled && explainPlan && (
           <div className={headerButtonSectionStyles}>
             <Button
               size="small"
@@ -144,7 +144,10 @@ export const ExplainPlanModal: React.FunctionComponent<
               data-testid="interpret-for-me-button"
               onClick={() => {
                 onModalClose();
-                interpretExplainPlan(JSON.stringify(explainPlan));
+                interpretExplainPlan({
+                  namespace: explainPlan.namespace,
+                  explainPlan: JSON.stringify(explainPlan),
+                });
               }}
               disabled={status !== 'ready'}
             >