chore: fix predictive state update tests for several integrations

Author: ranst91

typescript-sdk/apps/dojo/e2e/tests/crewAITests/predictvieStateUpdatePage.spec.ts

@@ -1,7 +1,6 @@
 import {
   test,
   expect,
-  waitForAIResponse,
   retryOnAIFailure,
 } from "../../test-isolation-helper";
 import { PredictiveStateUpdatesPage } from "../../pages/crewAIPages/PredictiveStateUpdatesPage";
@@ -22,7 +21,7 @@ test.describe("Predictive Status Updates Feature", () => {
       await predictiveStateUpdates.sendMessage(
         "Give me a story for a dragon called Atlantis in document"
       );
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await predictiveStateUpdates.getPredictiveResponse();
       await predictiveStateUpdates.getUserApproval();
       await predictiveStateUpdates.confirmedChangesResponse.isVisible();
@@ -33,7 +32,7 @@ test.describe("Predictive Status Updates Feature", () => {
 
       // Send update to change the dragon name
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await predictiveStateUpdates.verifyHighlightedText();
       await predictiveStateUpdates.getUserApproval();
       await predictiveStateUpdates.confirmedChangesResponse.nth(1).isVisible();
@@ -70,7 +69,7 @@ test.describe("Predictive Status Updates Feature", () => {
 
       // Send update to change the dragon name
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await predictiveStateUpdates.verifyHighlightedText();
       await predictiveStateUpdates.getUserRejection();
       await predictiveStateUpdates.rejectedChangesResponse.isVisible();

typescript-sdk/apps/dojo/e2e/tests/pydanticAITests/predictvieStateUpdatePage.spec.ts

@@ -1,7 +1,6 @@
 import {
   test,
   expect,
-  waitForAIResponse,
   retryOnAIFailure,
 } from "../../test-isolation-helper";
 import { PredictiveStateUpdatesPage } from "../../pages/pydanticAIPages/PredictiveStateUpdatesPage";
@@ -22,7 +21,7 @@ test.describe("Predictive Status Updates Feature", () => {
       await predictiveStateUpdates.sendMessage(
         "Give me a story for a dragon called Atlantis in document"
       );
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await predictiveStateUpdates.getPredictiveResponse();
       await predictiveStateUpdates.getUserApproval();
       await predictiveStateUpdates.confirmedChangesResponse.isVisible();
@@ -33,7 +32,7 @@ test.describe("Predictive Status Updates Feature", () => {
 
       // Send update to change the dragon name
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await predictiveStateUpdates.verifyHighlightedText();
       await predictiveStateUpdates.getUserApproval();
       await predictiveStateUpdates.confirmedChangesResponse.nth(1).isVisible();
@@ -70,7 +69,7 @@ test.describe("Predictive Status Updates Feature", () => {
 
       // Send update to change the dragon name
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await predictiveStateUpdates.verifyHighlightedText();
       await predictiveStateUpdates.getUserRejection();
       await predictiveStateUpdates.rejectedChangesResponse.isVisible();

typescript-sdk/apps/dojo/e2e/tests/serverStarterAllFeaturesTests/predictvieStateUpdatePage.spec.ts

@@ -1,4 +1,4 @@
-import { test, expect, waitForAIResponse, retryOnAIFailure, } from "../../test-isolation-helper";
+import { test, expect, retryOnAIFailure, } from "../../test-isolation-helper";
 import { PredictiveStateUpdatesPage } from "../../pages/serverStarterAllFeaturesPages/PredictiveStateUpdatesPage";
 
 test.describe("Predictive Status Updates Feature", () => {
@@ -14,7 +14,7 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.sendMessage("Hi");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.getPredictiveResponse();
@@ -27,7 +27,7 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(3000);
 
       await predictiveStateUpdates.sendMessage("Change the dog name");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.verifyHighlightedText();
@@ -53,7 +53,7 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.sendMessage("Hi");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.getPredictiveResponse();
@@ -66,7 +66,7 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(3000);
 
       await predictiveStateUpdates.sendMessage("Change the dog name");
-      await waitForAIResponse(page);
+      await page.waitForTimeout(2000);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.verifyHighlightedText();