feat: Loop uplift dataframe input and output (#8177)
* tests cases * update to loop * Update component.py * 📝 (LoopTemplate.json): update value of a configuration key from "OPENAI_API_KEY" to "ANTHROPIC_API_KEY" in order to reflect the correct API key being used * update json test loop * add dataframe support for the loop component * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes (attempt 2/3) * fix: starter project * update loop component and tests * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes (attempt 2/3) * update logic * Update loop_basic.py * Update Research Translation Loop.json * fix lint * format fix * [autofix.ci] apply automated fixes * reverting changes in component and vertex base * [autofix.ci] apply automated fixes * fix lint errors * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes (attempt 2/3) * update in the loop templates and component * [autofix.ci] apply automated fixes * Update Research Translation Loop.json * update tests * update the code and deprecate the old loop * [autofix.ci] apply automated fixes * Update loop_basic.py * WIP FIX Loop Tests * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes * ✨ (loop-component.spec.ts): Update test cases to use more descriptive names for components and actions for better clarity and understanding. * ✨ (loop-component.spec.ts): refactor loop component tests to improve readability and maintainability by updating test selectors and removing redundant test steps * update * Update loop-component.spec.ts * Update Research Translation Loop.json * Update Research Translation Loop.json * Update Research Translation Loop.json * Update Research Translation Loop.json * loop test fix --------- Co-authored-by: cristhianzl <cristhian.lousa@gmail.com> Co-authored-by: Rodrigo <rodrigosilvanader@gmail.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: italojohnny <italojohnnydosanjos@gmail.com> Co-authored-by: Mike Fortman <michael.fortman@datastax.com>
This commit is contained in:
Edwin Jose
GitHub
parent
ff37170693
commit
70b8f29099
4 changed files with 339 additions and 516 deletions
|
|
@ -43,53 +43,50 @@ test(
|
|||
|
||||
await page
|
||||
.getByTestId("logicLoop")
|
||||
.first()
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'), {
|
||||
targetPosition: { x: 280, y: 100 },
|
||||
});
|
||||
|
||||
// Add Update Data component
|
||||
await page.getByTestId("sidebar-search-input").click();
|
||||
await page.getByTestId("sidebar-search-input").fill("update data");
|
||||
await page.waitForSelector('[data-testid="processingUpdate Data"]', {
|
||||
await page.getByTestId("sidebar-search-input").fill("data operations");
|
||||
await page.waitForSelector('[data-testid="processingData Operations"]', {
|
||||
timeout: 1000,
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("processingUpdate Data")
|
||||
.getByTestId("processingData Operations")
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'), {
|
||||
targetPosition: { x: 500, y: 100 },
|
||||
});
|
||||
|
||||
// Add Parse Data component
|
||||
await page.getByTestId("sidebar-search-input").click();
|
||||
await page.getByTestId("sidebar-search-input").fill("data to message");
|
||||
await page.waitForSelector('[data-testid="processingData to Message"]', {
|
||||
await page.getByTestId("sidebar-search-input").fill("Parser");
|
||||
await page.waitForSelector('[data-testid="processingParser"]', {
|
||||
timeout: 1000,
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("processingData to Message")
|
||||
.getByTestId("processingParser")
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'), {
|
||||
targetPosition: { x: 720, y: 100 },
|
||||
});
|
||||
|
||||
//This one is for testing the wrong loop message
|
||||
|
||||
await page.getByTestId("sidebar-search-input").fill("File");
|
||||
await page.waitForSelector('[data-testid="dataFile"]', {
|
||||
timeout: 1000,
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("processingData to Message")
|
||||
.getByTestId("dataFile")
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'), {
|
||||
targetPosition: { x: 720, y: 400 },
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("handle-parsedata-shownode-data list-right")
|
||||
.nth(1)
|
||||
.click();
|
||||
|
||||
const loopItemInput = await page
|
||||
.getByTestId("handle-loopcomponent-shownode-item-left")
|
||||
.first()
|
||||
.click();
|
||||
|
||||
// Add Chat Output component
|
||||
await page.getByTestId("sidebar-search-input").click();
|
||||
await page.getByTestId("sidebar-search-input").fill("chat output");
|
||||
|
|
@ -114,7 +111,7 @@ test(
|
|||
.first()
|
||||
.click();
|
||||
await page
|
||||
.getByTestId("handle-updatedata-shownode-data-left")
|
||||
.getByTestId("handle-dataoperations-shownode-data-left")
|
||||
.first()
|
||||
.click();
|
||||
|
||||
|
|
@ -124,7 +121,7 @@ test(
|
|||
.first()
|
||||
.click();
|
||||
await page
|
||||
.getByTestId("handle-loopcomponent-shownode-data-left")
|
||||
.getByTestId("handle-loopcomponent-shownode-data or dataframe-left")
|
||||
.first()
|
||||
.click();
|
||||
|
||||
|
|
@ -134,13 +131,13 @@ test(
|
|||
.first()
|
||||
.click();
|
||||
await page
|
||||
.getByTestId("handle-parsedata-shownode-data-left")
|
||||
.getByTestId("handle-parsercomponent-shownode-data or dataframe-left")
|
||||
.first()
|
||||
.click();
|
||||
|
||||
// Parse Data -> Chat Output
|
||||
await page
|
||||
.getByTestId("handle-parsedata-shownode-message-right")
|
||||
.getByTestId("handle-parsercomponent-shownode-parsed text-right")
|
||||
.first()
|
||||
.click();
|
||||
|
||||
|
|
@ -149,7 +146,14 @@ test(
|
|||
.first()
|
||||
.click();
|
||||
|
||||
await zoomOut(page, 4);
|
||||
//Loop to File
|
||||
|
||||
await page
|
||||
.getByTestId("handle-loopcomponent-shownode-item-left")
|
||||
.first()
|
||||
.click();
|
||||
await page.getByTestId("handle-file-shownode-data-right").first().click();
|
||||
await zoomOut(page, 3);
|
||||
|
||||
await page.getByTestId("div-generic-node").nth(5).click();
|
||||
|
||||
|
|
@ -180,14 +184,16 @@ test(
|
|||
.fill("https://en.wikipedia.org/wiki/Human_intelligence");
|
||||
|
||||
await page.getByTestId("div-generic-node").nth(2).click();
|
||||
await page.getByTestId("int_int_number_of_fields").fill("1");
|
||||
await page.getByTestId("div-generic-node").nth(2).click();
|
||||
|
||||
await page.getByTestId("button_open_list_selection").click();
|
||||
|
||||
await page.getByTestId("list_item_append_or_update").click();
|
||||
|
||||
await page.getByTestId("keypair0").fill("text");
|
||||
await page.getByTestId("keypair100").fill("modified_value");
|
||||
|
||||
// Build and run, expect the wrong loop message
|
||||
await page.getByTestId("button_run_chat output").click();
|
||||
await page.getByTestId("button_run_file").click();
|
||||
await page.waitForSelector("text=The flow has an incomplete loop.", {
|
||||
timeout: 30000,
|
||||
});
|
||||
|
|
@ -197,7 +203,7 @@ test(
|
|||
|
||||
// Delete the second parse data used to test
|
||||
|
||||
await page.getByTestId("div-generic-node").nth(4).click();
|
||||
await page.getByTestId("title-File").last().click();
|
||||
|
||||
await page.getByTestId("more-options-modal").click();
|
||||
|
||||
|
|
@ -206,7 +212,7 @@ test(
|
|||
// Update Data -> Loop Item (left side)
|
||||
|
||||
await page
|
||||
.getByTestId("handle-updatedata-shownode-data-right")
|
||||
.getByTestId("handle-dataoperations-shownode-data-right")
|
||||
.first()
|
||||
.click();
|
||||
await page
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue