import { expect, test } from "@playwright/test"; test( "dropDownComponent", { tag: ["@release", "@workspace"] }, async ({ page }) => { await page.goto("/"); await page.waitForSelector('[data-testid="mainpage_title"]', { timeout: 30000, }); await page.waitForSelector('[id="new-project-btn"]', { timeout: 30000, }); let modalCount = 0; try { const modalTitleElement = await page?.getByTestId("modal-title"); if (modalTitleElement) { modalCount = await modalTitleElement.count(); } } catch (error) { modalCount = 0; } while (modalCount === 0) { await page.getByText("New Flow", { exact: true }).click(); await page.waitForSelector('[data-testid="modal-title"]', { timeout: 3000, }); modalCount = await page.getByTestId("modal-title")?.count(); } await page.waitForSelector('[data-testid="blank-flow"]', { timeout: 30000, }); await page.getByTestId("blank-flow").click(); await page.getByTestId("sidebar-search-input").click(); await page.getByTestId("sidebar-search-input").fill("amazon"); await page.waitForSelector('[data-testid="modelsAmazon Bedrock"]', { timeout: 3000, }); await page .getByTestId("modelsAmazon Bedrock") .first() .dragTo(page.locator('//*[@id="react-flow-id"]')); await page.mouse.up(); await page.mouse.down(); await page.getByTestId("fit_view").click(); await page.getByTestId("zoom_out").click(); await page.getByTestId("zoom_out").click(); await page.getByTestId("zoom_out").click(); await page.getByTestId("title-Amazon Bedrock").click(); await page.getByTestId("dropdown_str_model_id").click(); await page .getByTestId(/anthropic\.claude-3-haiku-20240307-v1:0.*option/) .click(); let value = await page .getByTestId(/anthropic\.claude-3-haiku-20240307-v1:0.*option/) .first() .innerText(); if (value !== "anthropic.claude-3-haiku-20240307-v1:0") { expect(false).toBeTruthy(); } await page.getByTestId("dropdown_str_model_id").click(); await page.getByText("anthropic.claude-v2").last().click(); value = await page.getByTestId("dropdown_str_model_id").innerText(); if (value !== "anthropic.claude-v2:1") { expect(false).toBeTruthy(); } await page.waitForSelector('[data-testid="more-options-modal"]', { timeout: 3000, }); await page.getByTestId("more-options-modal").click(); await page.getByTestId("advanced-button-modal").click(); value = await page .getByTestId("value-dropdown-dropdown_str_edit_model_id") .innerText(); if (value !== "anthropic.claude-v2:1") { expect(false).toBeTruthy(); } await page.locator('//*[@id="showregion_name"]').click(); expect( await page.locator('//*[@id="showregion_name"]').isChecked(), ).toBeFalsy(); await page.locator('//*[@id="showregion_name"]').click(); expect( await page.locator('//*[@id="showregion_name"]').isChecked(), ).toBeTruthy(); // showmodel_id await page.locator('//*[@id="showmodel_id"]').click(); expect( await page.locator('//*[@id="showmodel_id"]').isChecked(), ).toBeFalsy(); // showmodel_id await page.locator('//*[@id="showmodel_id"]').click(); expect( await page.locator('//*[@id="showmodel_id"]').isChecked(), ).toBeTruthy(); await page.locator('//*[@id="showregion_name"]').click(); expect( await page.locator('//*[@id="showregion_name"]').isChecked(), ).toBeFalsy(); await page.locator('//*[@id="showregion_name"]').click(); expect( await page.locator('//*[@id="showregion_name"]').isChecked(), ).toBeTruthy(); // showmodel_id await page.locator('//*[@id="showmodel_id"]').click(); expect( await page.locator('//*[@id="showmodel_id"]').isChecked(), ).toBeFalsy(); // showmodel_id await page.locator('//*[@id="showmodel_id"]').click(); expect( await page.locator('//*[@id="showmodel_id"]').isChecked(), ).toBeTruthy(); await page.getByTestId("value-dropdown-dropdown_str_edit_model_id").click(); await page.getByText("cohere").last().click(); value = await page .getByTestId("value-dropdown-dropdown_str_edit_model_id") .innerText(); if (value !== "cohere.command-r-plus-v1:0") { expect(false).toBeTruthy(); } await page.getByText("Close").last().click(); value = await page .getByTestId("value-dropdown-dropdown_str_model_id") .innerText(); if (value !== "cohere.command-r-plus-v1:0") { expect(false).toBeTruthy(); } await page.getByTestId("code-button-modal").click(); await page.locator("textarea").press("Control+a"); const emptyOptionsCode = `from langchain_community.chat_models.bedrock import BedrockChat from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent from langflow.field_typing import BaseLanguageModel, Text from langflow.io import BoolInput, DictInput, DropdownInput, StrInput from langflow.io import MessageInput from langflow.io import Output class AmazonBedrockComponent(LCModelComponent): display_name: str = "Amazon Bedrock" description: str = "Generate text using Amazon Bedrock LLMs." icon = "Amazon" inputs = [ MessageInput(name="input_value", display_name="Input"), DropdownInput( name="model_id", display_name="Model Id", options=[ "amazon.titan-text-express-v1", "amazon.titan-text-lite-v1", "amazon.titan-text-premier-v1:0", "amazon.titan-embed-text-v1", "amazon.titan-embed-text-v2:0", "amazon.titan-embed-image-v1", "amazon.titan-image-generator-v1", "anthropic.claude-v2", "anthropic.claude-v2:1", "anthropic.claude-3-sonnet-20240229-v1:0", "anthropic.claude-3-haiku-20240307-v1:0", "anthropic.claude-3-opus-20240229-v1:0", "anthropic.claude-instant-v1", "ai21.j2-mid-v1", "ai21.j2-ultra-v1", "cohere.command-text-v14", "cohere.command-light-text-v14", "cohere.command-r-v1:0", "cohere.command-r-plus-v1:0", "cohere.embed-english-v3", "cohere.embed-multilingual-v3", "meta.llama2-13b-chat-v1", "meta.llama2-70b-chat-v1", "meta.llama3-8b-instruct-v1:0", "meta.llama3-70b-instruct-v1:0", "mistral.mistral-7b-instruct-v0:2", "mistral.mixtral-8x7b-instruct-v0:1", "mistral.mistral-large-2402-v1:0", "mistral.mistral-small-2402-v1:0", "stability.stable-diffusion-xl-v0", "stability.stable-diffusion-xl-v1", ], value="anthropic.claude-3-haiku-20240307-v1:0", ), StrInput(name="credentials_profile_name", display_name="Credentials Profile Name"), StrInput(name="region_name", display_name="Region Name"), DictInput(name="model_kwargs", display_name="Model Kwargs", advanced=True), StrInput(name="endpoint_url", display_name="Endpoint URL"), BoolInput(name="cache", display_name="Cache"), StrInput( name="system_message", display_name="System Message", info="System message to pass to the model.", advanced=True, ), BoolInput(name="stream", display_name="Stream", info=STREAM_INFO_TEXT, advanced=True), ] outputs = [ Output(display_name="Text", name="text_output", method="text_response"), Output(display_name="Language Model", name="model_output", method="build_model"), ] def text_response(self) -> Text: input_value = self.input_value stream = self.stream system_message = self.system_message output = self.build_model() result = self.get_chat_result(output, stream, input_value, system_message) self.status = result return result def build_model(self) -> BaseLanguageModel: model_id = self.model_id credentials_profile_name = self.credentials_profile_name region_name = self.region_name model_kwargs = self.model_kwargs endpoint_url = self.endpoint_url cache = self.cache stream = self.stream try: output = BedrockChat( credentials_profile_name=credentials_profile_name, model_id=model_id, region_name=region_name, model_kwargs=model_kwargs, endpoint_url=endpoint_url, streaming=stream, cache=cache, ) except Exception as e: raise ValueError("Could not connect to AmazonBedrock API.") from e return output `; await page.locator("textarea").fill(emptyOptionsCode); await page.getByRole("button", { name: "Check & Save" }).click(); await page .getByText("No parameters are available for display.") .isVisible(); }, );