diff --git a/apps/demos/Demos/Chat/PageLevelAIAssistant/Angular/app/app.service.ts b/apps/demos/Demos/Chat/PageLevelAIAssistant/Angular/app/app.service.ts
index e24da4a89d5b..8c22123a59f7 100644
--- a/apps/demos/Demos/Chat/PageLevelAIAssistant/Angular/app/app.service.ts
+++ b/apps/demos/Demos/Chat/PageLevelAIAssistant/Angular/app/app.service.ts
@@ -124,6 +124,8 @@ function buildGridResultsPromise(
   const columnNames = getGridColumnNames(gridInstance);
   const prompt = `${buildGridSystemPrompt(columnNames)}\n\nUser request: '${text}'`;
 
+  gridInstance?.instance.beginCustomLoading('');
+
   return executeAiCommand(prompt, aiIntegration)
     .then((parsed) => {
       const actions = Array.isArray((parsed as { actions?: ExecuteGridAssistantAction[] })?.actions)
@@ -213,14 +215,12 @@ async function runCommand(
   }
 
   if (target === 'form') {
-    const { results: formResults, error: formError } = await buildFormResultsPromise(form, formAction, text);
+    const { results: formResults, error: formError } = await buildFormResultsPromise(form, formAction ?? { type: 'smart_paste' }, text);
 
     return joinSucceededOrThrow(formResults, formError);
   }
 
   if (target === 'grid') {
-    gridInstance?.instance.beginCustomLoading('');
-
     const { results: gridResults, error: gridError } = await buildGridResultsPromise(gridInstance, aiIntegration, text);
 
     return joinSucceededOrThrow(gridResults, gridError);
diff --git a/apps/demos/Demos/Chat/PageLevelAIAssistant/React/chat-router.ts b/apps/demos/Demos/Chat/PageLevelAIAssistant/React/chat-router.ts
index 9f2fc2939d96..b95dd22953fe 100644
--- a/apps/demos/Demos/Chat/PageLevelAIAssistant/React/chat-router.ts
+++ b/apps/demos/Demos/Chat/PageLevelAIAssistant/React/chat-router.ts
@@ -212,7 +212,7 @@ async function runCommand(text: string, { form, gridInstance, aiIntegration }: R
   }
 
   if (target === 'form') {
-    const { results, error } = await buildFormResultsPromise(form, formAction, text);
+    const { results, error } = await buildFormResultsPromise(form, formAction ?? { type: 'smart_paste' }, text);
     return joinSucceededOrThrow(results, error);
   }
 
diff --git a/apps/demos/Demos/Chat/PageLevelAIAssistant/ReactJs/chat-router.js b/apps/demos/Demos/Chat/PageLevelAIAssistant/ReactJs/chat-router.js
index 663870eda590..d24ca6f5ff5d 100644
--- a/apps/demos/Demos/Chat/PageLevelAIAssistant/ReactJs/chat-router.js
+++ b/apps/demos/Demos/Chat/PageLevelAIAssistant/ReactJs/chat-router.js
@@ -176,7 +176,11 @@ async function runCommand(text, { form, gridInstance, aiIntegration }) {
     );
   }
   if (target === 'form') {
-    const { results, error } = await buildFormResultsPromise(form, formAction, text);
+    const { results, error } = await buildFormResultsPromise(
+      form,
+      formAction ?? { type: 'smart_paste' },
+      text,
+    );
     return joinSucceededOrThrow(results, error);
   }
   if (target === 'grid') {
diff --git a/apps/demos/Demos/Chat/PageLevelAIAssistant/Vue/chatRouter.ts b/apps/demos/Demos/Chat/PageLevelAIAssistant/Vue/chatRouter.ts
index 4d711a1d243c..06e09175c776 100644
--- a/apps/demos/Demos/Chat/PageLevelAIAssistant/Vue/chatRouter.ts
+++ b/apps/demos/Demos/Chat/PageLevelAIAssistant/Vue/chatRouter.ts
@@ -126,6 +126,8 @@ function buildGridResultsPromise(
   const columnNames = getGridColumnNames(gridInstance);
   const prompt = `${buildGridSystemPrompt(columnNames)}\n\nUser request: '${text}'`;
 
+  gridInstance?.beginCustomLoading('');
+
   return executeAiCommand(prompt, aiIntegration)
     .then((parsed) => {
       const actions = Array.isArray((parsed as { actions?: ExecuteGridAssistantAction[] })?.actions)
@@ -221,8 +223,6 @@ async function runCommand(
   }
 
   if (target === 'grid') {
-    gridInstance?.beginCustomLoading('');
-
     const { results: gridResults, error: gridError } = await buildGridResultsPromise(gridInstance, aiIntegration, text);
 
     return joinSucceededOrThrow(gridResults, gridError);
diff --git a/apps/demos/Demos/Chat/PageLevelAIAssistant/jQuery/index.js b/apps/demos/Demos/Chat/PageLevelAIAssistant/jQuery/index.js
index a600327f0824..05c7fc01b1de 100644
--- a/apps/demos/Demos/Chat/PageLevelAIAssistant/jQuery/index.js
+++ b/apps/demos/Demos/Chat/PageLevelAIAssistant/jQuery/index.js
@@ -811,6 +811,8 @@ function buildGridResultsPromise(gridInstance, aiIntegration, text) {
   const columnNames = getGridColumnNames(gridInstance);
   const prompt = `${buildGridSystemPrompt(columnNames)}\n\nUser request: '${text}'`;
 
+  gridInstance?.beginCustomLoading('');
+
   return executeAiCommand(prompt, aiIntegration)
     .then((parsed) => {
       const actions = Array.isArray(parsed.actions) ? parsed.actions : [];
@@ -900,14 +902,16 @@ async function runCommand(text, { form, gridInstance, aiIntegration }) {
 
   if (target === 'form') {
     const { results: formResults, error: formError } =
-      await buildFormResultsPromise(form, formAction, text);
+      await buildFormResultsPromise(
+        form,
+        formAction ?? { type: 'smart_paste' },
+        text,
+      );
 
     return joinSucceededOrThrow(formResults, formError);
   }
 
   if (target === 'grid') {
-    gridInstance?.beginCustomLoading('');
-
     const { results: gridResults, error: gridError } =
       await buildGridResultsPromise(gridInstance, aiIntegration, text);