From 7c6ee78c4af8413597f67960a1d48eeacc227d56 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 3 Jan 2024 11:20:15 -0300 Subject: [PATCH] Refactor AmazonBedrockComponent cache parameter --- src/backend/langflow/components/llms/AmazonBedrock.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/backend/langflow/components/llms/AmazonBedrock.py b/src/backend/langflow/components/llms/AmazonBedrock.py index 306f84e30..5b931bb94 100644 --- a/src/backend/langflow/components/llms/AmazonBedrock.py +++ b/src/backend/langflow/components/llms/AmazonBedrock.py @@ -2,7 +2,6 @@ from typing import Optional from langchain.llms.base import BaseLLM from langchain.llms.bedrock import Bedrock - from langflow import CustomComponent @@ -44,7 +43,7 @@ class AmazonBedrockComponent(CustomComponent): model_kwargs: Optional[dict] = None, endpoint_url: Optional[str] = None, streaming: bool = False, - cache: bool | None = None, + cache: Optional[bool] = None, ) -> BaseLLM: try: output = Bedrock(