refactor(agent): standardize memory handling and update chat history logic (#8715)
* update chat history * update to agents * Update Simple Agent.json * update to templates * ruff errors * Update agent.py * Update test_agent_component.py * [autofix.ci] apply automated fixes * update templates * test fix --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Mike Fortman <michael.fortman@datastax.com>
This commit is contained in:
parent
0d74474e12
commit
38d5885fa3
25 changed files with 8672 additions and 17583 deletions
|
|
@ -1,78 +0,0 @@
|
|||
import { expect, test } from "@playwright/test";
|
||||
import * as dotenv from "dotenv";
|
||||
import path from "path";
|
||||
import { awaitBootstrapTest } from "../../utils/await-bootstrap-test";
|
||||
import { withEventDeliveryModes } from "../../utils/withEventDeliveryModes";
|
||||
|
||||
withEventDeliveryModes(
|
||||
"Financial Agent",
|
||||
{ tag: ["@release", "@starter-projects"] },
|
||||
async ({ page }) => {
|
||||
test.skip(
|
||||
!process?.env?.TAVILY_API_KEY,
|
||||
"TAVILY_API_KEY required to run this test",
|
||||
);
|
||||
|
||||
test.skip(
|
||||
!process?.env?.SAMBANOVA_API_KEY,
|
||||
"SAMBANOVA_API_KEY required to run this test",
|
||||
);
|
||||
|
||||
await page.goto("/");
|
||||
await awaitBootstrapTest(page);
|
||||
|
||||
await page.getByTestId("side_nav_options_all-templates").click();
|
||||
await page.getByRole("heading", { name: "Financial Agent" }).click();
|
||||
|
||||
await page
|
||||
.getByTestId("popover-anchor-input-api_key")
|
||||
.nth(0)
|
||||
.fill(process.env.TAVILY_API_KEY ?? "");
|
||||
|
||||
for (let i = 0; i < 2; i++) {
|
||||
await page.getByTestId("dropdown_str_agent_llm").nth(i).click();
|
||||
await page.waitForTimeout(500);
|
||||
await page.getByRole("option", { name: "SambaNova" }).click();
|
||||
}
|
||||
|
||||
for (let i = 0; i < 3; i++) {
|
||||
await page
|
||||
.getByTestId("value-dropdown-dropdown_str_model_name")
|
||||
.nth(i)
|
||||
.click();
|
||||
await page.waitForTimeout(500);
|
||||
|
||||
await page.getByRole("option").first().click();
|
||||
}
|
||||
|
||||
for (let i = 1; i <= 3; i++) {
|
||||
await page
|
||||
.getByTestId("popover-anchor-input-api_key")
|
||||
.nth(i)
|
||||
.fill(process.env.SAMBANOVA_API_KEY ?? "");
|
||||
}
|
||||
|
||||
await page.getByTestId("playground-btn-flow-io").click();
|
||||
|
||||
await page
|
||||
.getByTestId("input-chat-playground")
|
||||
.last()
|
||||
.fill("Why did Nvidia stock drop in January?");
|
||||
|
||||
await page.getByTestId("button-send").last().click();
|
||||
|
||||
const stopButton = page.getByRole("button", { name: "Stop" });
|
||||
await stopButton.waitFor({ state: "visible", timeout: 30000 });
|
||||
|
||||
if (await stopButton.isVisible()) {
|
||||
await expect(stopButton).toBeHidden({ timeout: 120000 });
|
||||
}
|
||||
|
||||
const output = await page
|
||||
.getByTestId("div-chat-message")
|
||||
.last()
|
||||
.innerText();
|
||||
expect(output.toLowerCase()).toContain("nvidia");
|
||||
expect(output.length).toBeGreaterThan(100);
|
||||
},
|
||||
);
|
||||
|
|
@ -41,17 +41,6 @@ withEventDeliveryModes(
|
|||
.nth(0)
|
||||
.fill(process.env.TAVILY_API_KEY ?? "");
|
||||
|
||||
//* TODO: Remove these 5 steps once the template is updated *//
|
||||
await page.getByTestId("dropdown-output-openaimodel").click();
|
||||
|
||||
await page
|
||||
.getByTestId("dropdown-item-output-openaimodel-language model")
|
||||
.click();
|
||||
|
||||
await page
|
||||
.getByTestId("handle-structuredoutput-shownode-structured output-right")
|
||||
.click();
|
||||
|
||||
await page
|
||||
.getByTestId("handle-parser-shownode-data or dataframe-left")
|
||||
.click();
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue