Define a type

apps/app/src/features/openai/client/components/AiAssistant/AiAssistantManagementModal/AiAssistantManagementEditInstruction.tsx

@@ -19,15 +19,18 @@ type Props = {
   instruction: string;
   onChange: (value: string) => void;
   onReset: () => void;
+  backToPageMode?: AiAssistantManagementModalPageMode;
 }
 
 export const AiAssistantManagementEditInstruction = (props: Props): JSX.Element => {
-  const { instruction, onChange, onReset } = props;
+  const {
+    instruction, onChange, onReset, backToPageMode = AiAssistantManagementModalPageMode.HOME,
+  } = props;
   const { t } = useTranslation();
   const { changePageMode } = useAiAssistantManagementModal();
 
   const handleComplete = () => {
-    changePageMode(AiAssistantManagementModalPageMode.HOME);
+    changePageMode(backToPageMode);
   };
 
   return (