diff --git a/src/backend/base/langflow/base/compressors/model.py b/src/backend/base/langflow/base/compressors/model.py index de3731cde3cf..fb52e641aa5d 100644 --- a/src/backend/base/langflow/base/compressors/model.py +++ b/src/backend/base/langflow/base/compressors/model.py @@ -2,7 +2,7 @@ from langflow.custom import Component from langflow.field_typing import BaseDocumentCompressor -from langflow.io import DataInput, IntInput, MultilineInput, SecretStrInput +from langflow.io import DataInput, IntInput, MultilineInput from langflow.schema import Data from langflow.schema.dataframe import DataFrame from langflow.template.field.base import Output @@ -15,10 +15,6 @@ class LCCompressorComponent(Component): display_name="Search Query", tool_mode=True, ), - SecretStrInput( - name="api_key", - display_name="API Key", - ), DataInput( name="search_results", display_name="Search Results", diff --git a/src/backend/base/langflow/components/cohere/cohere_rerank.py b/src/backend/base/langflow/components/cohere/cohere_rerank.py index 20a68925b181..0f6838fd4713 100644 --- a/src/backend/base/langflow/components/cohere/cohere_rerank.py +++ b/src/backend/base/langflow/components/cohere/cohere_rerank.py @@ -1,5 +1,6 @@ from langflow.base.compressors.model import LCCompressorComponent from langflow.field_typing import BaseDocumentCompressor +from langflow.inputs.inputs import SecretStrInput from langflow.io import DropdownInput from langflow.template.field.base import Output @@ -12,6 +13,10 @@ class CohereRerankComponent(LCCompressorComponent): inputs = [ *LCCompressorComponent.inputs, + SecretStrInput( + name="api_key", + display_name="Cohere API Key", + ), DropdownInput( name="model", display_name="Model", diff --git a/src/backend/base/langflow/components/nvidia/nvidia_rerank.py b/src/backend/base/langflow/components/nvidia/nvidia_rerank.py index db32b1a31713..122e0ec47373 100644 --- a/src/backend/base/langflow/components/nvidia/nvidia_rerank.py +++ b/src/backend/base/langflow/components/nvidia/nvidia_rerank.py @@ -2,6 +2,7 @@ from langflow.base.compressors.model import LCCompressorComponent from langflow.field_typing import BaseDocumentCompressor +from langflow.inputs.inputs import SecretStrInput from langflow.io import DropdownInput, StrInput from langflow.schema.dotdict import dotdict from langflow.template.field.base import Output @@ -14,6 +15,10 @@ class NvidiaRerankComponent(LCCompressorComponent): inputs = [ *LCCompressorComponent.inputs, + SecretStrInput( + name="api_key", + display_name="NVIDIA API Key", + ), StrInput( name="base_url", display_name="Base URL",