chore: fix predictive state update tests for langgraph

Author: ranst91

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

@@ -1,7 +1,6 @@
 import {
   test,
   expect,
-  waitForAIResponse,
   retryOnAIFailure,
 } from "../../test-isolation-helper";
 import { PredictiveStateUpdatesPage } from "../../pages/langGraphFastAPIPages/PredictiveStateUpdatesPage";
@@ -23,7 +22,6 @@ 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();
@@ -37,7 +35,6 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(3000);
 
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.verifyHighlightedText();
@@ -66,7 +63,6 @@ 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();
@@ -80,7 +76,6 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(3000);
 
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.verifyHighlightedText();

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

@@ -1,7 +1,6 @@
 import {
   test,
   expect,
-  waitForAIResponse,
   retryOnAIFailure,
 } from "../../test-isolation-helper";
 import { PredictiveStateUpdatesPage } from "../../pages/langGraphPages/PredictiveStateUpdatesPage";
@@ -23,7 +22,6 @@ 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();
@@ -37,7 +35,6 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(3000);
 
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.verifyHighlightedText();
@@ -66,7 +63,6 @@ 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();
@@ -80,7 +76,6 @@ test.describe("Predictive Status Updates Feature", () => {
       await page.waitForTimeout(3000);
 
       await predictiveStateUpdates.sendMessage("Change dragon name to Lola");
-      await waitForAIResponse(page);
       await page.waitForTimeout(2000);
 
       await predictiveStateUpdates.verifyHighlightedText();