Pick file type. Remove waiting from tests.

lynxkite-app/web/src/workspace/Workspace.tsx

@@ -243,7 +243,7 @@ function LynxKiteFlow() {
     state: { workspace: Workspace },
     nodes: Node[],
   ) {
-    const title = node.data.title;
+    const title = node.data?.title;
     let i = 1;
     node.id = `${title} ${i}`;
     const wnodes = state.workspace.nodes!;
@@ -325,6 +325,15 @@ function LynxKiteFlow() {
         y: e.clientY,
       });
       node.data!.params.file_path = file.name;
+      if (file.name.includes(".csv")) {
+        node.data!.params.file_format = "csv";
+      } else if (file.name.includes(".parquet")) {
+        node.data!.params.file_format = "parquet";
+      } else if (file.name.includes(".json")) {
+        node.data!.params.file_format = "json";
+      } else if (file.name.includes(".xls")) {
+        node.data!.params.file_format = "excel";
+      }
       addNode(node, state, nodes);
     } catch (error) {
       setMessage("File upload failed.");

lynxkite-app/web/tests/basic.spec.ts

@@ -21,9 +21,6 @@ test("Box creation & deletion per env", async () => {
   const envs = await workspace.getEnvs();
   for (const env of envs) {
     await workspace.setEnv(env);
-    // TODO: Opening the catalog immediately after setting the env can fail.
-    // Let's fix this!
-    await new Promise((resolve) => setTimeout(resolve, 500));
     const catalog = await workspace.getCatalog();
     expect(catalog).not.toHaveLength(0);
     const op = catalog[0];

lynxkite-app/web/tests/examples.spec.ts

@@ -17,8 +17,7 @@ test("Pytorch example", async ({ page }) => {
   await ws.expectErrorFree();
 });
 
-test.fail("AIMO example", async ({ page }) => {
-  // Fails because of missing OPENAI_API_KEY
+test("AIMO example", async ({ page }) => {
   const ws = await Workspace.open(page, "AIMO");
   await ws.expectErrorFree();
 });
@@ -35,12 +34,6 @@ test("Graph RAG", async ({ page }) => {
   await ws.expectErrorFree(process.env.CI ? 2000 : 500);
 });
 
-test.fail("RAG chatbot app", async ({ page }) => {
-  // Fail due to all operation being unknown
-  const ws = await Workspace.open(page, "RAG chatbot app");
-  await ws.expectErrorFree();
-});
-
 test("Airlines demo", async ({ page }) => {
   const ws = await Workspace.open(page, "Airlines demo");
   await ws.expectErrorFree(process.env.CI ? 10000 : 500);

lynxkite-app/web/tests/lynxkite.ts

@@ -59,7 +59,6 @@ export class Workspace {
       // Avoid overlapping with existing nodes
       const numNodes = allBoxes.length || 1;
       await this.page.mouse.wheel(0, numNodes * 400);
-      await new Promise((resolve) => setTimeout(resolve, 200));
     }
 
     await this.page.locator(".ws-name").click();
@@ -74,9 +73,9 @@ export class Workspace {
   async getCatalog() {
     await this.page.locator(".ws-name").click();
     await this.page.keyboard.press("/");
-    const catalog = await this.page
-      .locator(".node-search .matches .search-result")
-      .allInnerTexts();
+    const results = this.page.locator(".node-search .matches .search-result");
+    await expect(results.first()).toBeVisible();
+    const catalog = await results.allInnerTexts();
     // Dismiss the catalog menu
     await this.page.keyboard.press("Escape");
     await expect(this.page.locator(".node-search")).not.toBeVisible();
@@ -153,11 +152,6 @@ export class Workspace {
   }
 
   async expectErrorFree(executionWaitTime?) {
-    // TODO: Workaround, to account for workspace execution. Once
-    // we have a load indicator we can use that instead.
-    await new Promise((resolve) =>
-      setTimeout(resolve, executionWaitTime ? executionWaitTime : 500),
-    );
     await expect(this.getBoxes().locator(".error").first()).not.toBeVisible();
   }
 
@@ -188,7 +182,7 @@ export class Splash {
   }
 
   workspace(name: string) {
-    return this.page.getByRole("link", { name: name });
+    return this.page.getByRole("link", { name: name, exact: true });
   }
 
   getEntry(name: string) {