feat: add models category (#8231)
* models category * [autofix.ci] apply automated fixes * change paths * [autofix.ci] apply automated fixes * test: comment out visibility check for disclosure-embeddings * test: comment out visibility check for disclosure-embeddings * test: update selector for OpenAI language models in decision flow integration test * test: update selector for NVIDIA models to use language model identifier * test: update selector for OpenAI language models in regression test * models category (#8238) * Update styleUtils.ts * Update test_xai.py * test: update selectors for OpenAI language models to use 'languagemodelsOpenAI' * Update test_embedding_model_component.py * test: update test IDs for NVIDIA and Ollama language models to use 'languagemodels' * test: update test IDs to use 'languagemodels' prefix for consistency --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Deon Sanchez <69873175+deon-sanchez@users.noreply.github.com> Co-authored-by: deon-sanchez <deon.sanchez@datastax.com>
This commit is contained in:
parent
09a65f6f41
commit
b14189ed05
64 changed files with 152 additions and 143 deletions
|
|
@ -31,12 +31,12 @@ test(
|
|||
await page.getByTestId("sidebar-search-input").click();
|
||||
await page.getByTestId("sidebar-search-input").fill("NVIDIA");
|
||||
|
||||
await page.waitForSelector('[data-testid="modelsNVIDIA"]', {
|
||||
await page.waitForSelector('[data-testid="languagemodelsNVIDIA"]', {
|
||||
timeout: 3000,
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("modelsNVIDIA")
|
||||
.getByTestId("languagemodelsNVIDIA")
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'));
|
||||
await page.mouse.up();
|
||||
await page.mouse.down();
|
||||
|
|
@ -85,7 +85,7 @@ test(
|
|||
await page.keyboard.press("Escape");
|
||||
await page.locator('//*[@id="react-flow-id"]').click();
|
||||
|
||||
const lastNvidiaModel = page.getByTestId("modelsNVIDIA").last();
|
||||
const lastNvidiaModel = page.getByTestId("languagemodelsNVIDIA").last();
|
||||
await lastNvidiaModel.scrollIntoViewIfNeeded();
|
||||
|
||||
try {
|
||||
|
|
@ -128,12 +128,12 @@ test(
|
|||
await page.getByTestId("sidebar-search-input").click();
|
||||
await page.getByTestId("sidebar-search-input").fill("NVIDIA");
|
||||
|
||||
await page.waitForSelector('[data-testid="modelsNVIDIA"]', {
|
||||
await page.waitForSelector('[data-testid="languagemodelsNVIDIA"]', {
|
||||
timeout: 3000,
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("modelsNVIDIA")
|
||||
.getByTestId("languagemodelsNVIDIA")
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'));
|
||||
await page.mouse.up();
|
||||
await page.mouse.down();
|
||||
|
|
|
|||
|
|
@ -62,7 +62,6 @@ test(
|
|||
"disclosure-models",
|
||||
"disclosure-helpers",
|
||||
"disclosure-vector stores",
|
||||
"disclosure-embeddings",
|
||||
"disclosure-agents",
|
||||
"disclosure-memories",
|
||||
"disclosure-logic",
|
||||
|
|
@ -99,23 +98,23 @@ test(
|
|||
await page.getByTestId("sidebar-search-input").click();
|
||||
|
||||
const visibleModelSpecsTestIds = [
|
||||
"modelsAIML",
|
||||
"modelsAnthropic",
|
||||
"modelsAzure OpenAI",
|
||||
"modelsCohere",
|
||||
"modelsGoogle Generative AI",
|
||||
"modelsGroq",
|
||||
"modelsHuggingFace",
|
||||
"modelsLM Studio",
|
||||
"modelsMaritalk",
|
||||
"modelsMistralAI",
|
||||
"modelsNVIDIA",
|
||||
"modelsOllama",
|
||||
"modelsOpenAI",
|
||||
"modelsPerplexity",
|
||||
"modelsQianfan",
|
||||
"modelsSambaNova",
|
||||
"modelsVertex AI",
|
||||
"languagemodelsAIML",
|
||||
"languagemodelsAnthropic",
|
||||
"languagemodelsAzure OpenAI",
|
||||
"languagemodelsCohere",
|
||||
"languagemodelsGoogle Generative AI",
|
||||
"languagemodelsGroq",
|
||||
"languagemodelsHuggingFace",
|
||||
"languagemodelsLM Studio",
|
||||
"languagemodelsMaritalk",
|
||||
"languagemodelsMistralAI",
|
||||
"languagemodelsNVIDIA",
|
||||
"languagemodelsOllama",
|
||||
"languagemodelsOpenAI",
|
||||
"languagemodelsPerplexity",
|
||||
"languagemodelsQianfan",
|
||||
"languagemodelsSambaNova",
|
||||
"languagemodelsVertex AI",
|
||||
];
|
||||
|
||||
await Promise.all(
|
||||
|
|
|
|||
|
|
@ -16,12 +16,12 @@ test(
|
|||
await page.getByTestId("sidebar-search-input").click();
|
||||
await page.getByTestId("sidebar-search-input").fill("ollama");
|
||||
|
||||
await page.waitForSelector('[data-testid="modelsOllama"]', {
|
||||
await page.waitForSelector('[data-testid="languagemodelsOllama"]', {
|
||||
timeout: 3000,
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("modelsOllama")
|
||||
.getByTestId("languagemodelsOllama")
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'));
|
||||
await page.mouse.up();
|
||||
await page.mouse.down();
|
||||
|
|
|
|||
|
|
@ -14,12 +14,12 @@ test(
|
|||
await page.getByTestId("sidebar-search-input").click();
|
||||
await page.getByTestId("sidebar-search-input").fill("openai");
|
||||
|
||||
await page.waitForSelector('[data-testid="modelsOpenAI"]', {
|
||||
await page.waitForSelector('[data-testid="languagemodelsOpenAI"]', {
|
||||
timeout: 1000,
|
||||
});
|
||||
|
||||
await page
|
||||
.getByTestId("modelsOpenAI")
|
||||
.getByTestId("languagemodelsOpenAI")
|
||||
.hover()
|
||||
.then(async () => {
|
||||
await page.getByTestId("add-component-button-openai").last().click();
|
||||
|
|
|
|||
|
|
@ -55,7 +55,7 @@ test(
|
|||
await page.getByTestId("sidebar-search-input").fill("openai");
|
||||
|
||||
await page
|
||||
.getByTestId("modelsOpenAI")
|
||||
.getByTestId("languagemodelsOpenAI")
|
||||
.dragTo(page.locator('//*[@id="react-flow-id"]'), {
|
||||
targetPosition: { x: 100, y: 200 },
|
||||
});
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue