Modify testing

Signed-off-by: Luis Valdes <[email protected]>

Author: LuisValdesZero

typescript-sdk/apps/dojo/e2e/featurePages/ToolBaseGenUIPage.ts

@@ -73,7 +73,17 @@ export class ToolBaseGenUIPage {
   }
 
   async extractMainDisplayHaikuContent(page: Page): Promise<string> {
+    // Wait for the main haiku display to be visible
+    await page.waitForSelector('[data-testid="main-haiku-display"]', { state: 'visible' });
+
     const mainDisplayLines = page.locator('[data-testid="main-haiku-line"]');
+
+    // Wait for at least 3 haiku lines to be present
+    await page.waitForFunction(() => {
+      const elements = document.querySelectorAll('[data-testid="main-haiku-line"]');
+      return elements.length >= 3;
+    });
+
     const mainCount = await mainDisplayLines.count();
     const lines: string[] = [];
 

typescript-sdk/integrations/langgraph/examples/python/agents/agentic_chat/agent.py

@@ -14,20 +14,13 @@
 from langgraph.types import Command
 from langchain_core.tools import tool
 
+
 class AgentState(MessagesState):
     """
     State of our graph.
     """
     tools: List[Any] = []
 
-@tool
-def change_background(background: str) -> str: # pylint: disable=unused-argument
-    """
-    Change the background color of the chat. Can be anything that the CSS background attribute accepts. Regular colors, linear of radial gradients etc.
-
-    Args:
-        background: str: The background color to change to. Can be anything that the CSS background attribute accepts. Regular colors, linear of radial gradients etc.
-    """ # pylint: disable=line-too-long
 
 async def chat_node(state: AgentState, config: Optional[RunnableConfig] = None):
     """
@@ -52,7 +45,6 @@ async def chat_node(state: AgentState, config: Optional[RunnableConfig] = None):
     model_with_tools = model.bind_tools(
         [
             *state.get("tools", []),
-            change_background,
             # your_tool_here
         ],