refactor: move API key input into Cohere and NVIDIA rerank components (#6339)
Remove generic API key input from base compressor model and add specific API key inputs for Cohere and NVIDIA Rerank components
This commit is contained in:
parent
339f06571b
commit
63294b9b08
3 changed files with 11 additions and 5 deletions
|
|
@ -2,7 +2,7 @@ from abc import abstractmethod
|
||||||
|
|
||||||
from langflow.custom import Component
|
from langflow.custom import Component
|
||||||
from langflow.field_typing import BaseDocumentCompressor
|
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 import Data
|
||||||
from langflow.schema.dataframe import DataFrame
|
from langflow.schema.dataframe import DataFrame
|
||||||
from langflow.template.field.base import Output
|
from langflow.template.field.base import Output
|
||||||
|
|
@ -15,10 +15,6 @@ class LCCompressorComponent(Component):
|
||||||
display_name="Search Query",
|
display_name="Search Query",
|
||||||
tool_mode=True,
|
tool_mode=True,
|
||||||
),
|
),
|
||||||
SecretStrInput(
|
|
||||||
name="api_key",
|
|
||||||
display_name="API Key",
|
|
||||||
),
|
|
||||||
DataInput(
|
DataInput(
|
||||||
name="search_results",
|
name="search_results",
|
||||||
display_name="Search Results",
|
display_name="Search Results",
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,6 @@
|
||||||
from langflow.base.compressors.model import LCCompressorComponent
|
from langflow.base.compressors.model import LCCompressorComponent
|
||||||
from langflow.field_typing import BaseDocumentCompressor
|
from langflow.field_typing import BaseDocumentCompressor
|
||||||
|
from langflow.inputs.inputs import SecretStrInput
|
||||||
from langflow.io import DropdownInput
|
from langflow.io import DropdownInput
|
||||||
from langflow.template.field.base import Output
|
from langflow.template.field.base import Output
|
||||||
|
|
||||||
|
|
@ -12,6 +13,10 @@ class CohereRerankComponent(LCCompressorComponent):
|
||||||
|
|
||||||
inputs = [
|
inputs = [
|
||||||
*LCCompressorComponent.inputs,
|
*LCCompressorComponent.inputs,
|
||||||
|
SecretStrInput(
|
||||||
|
name="api_key",
|
||||||
|
display_name="Cohere API Key",
|
||||||
|
),
|
||||||
DropdownInput(
|
DropdownInput(
|
||||||
name="model",
|
name="model",
|
||||||
display_name="Model",
|
display_name="Model",
|
||||||
|
|
|
||||||
|
|
@ -2,6 +2,7 @@ from typing import Any
|
||||||
|
|
||||||
from langflow.base.compressors.model import LCCompressorComponent
|
from langflow.base.compressors.model import LCCompressorComponent
|
||||||
from langflow.field_typing import BaseDocumentCompressor
|
from langflow.field_typing import BaseDocumentCompressor
|
||||||
|
from langflow.inputs.inputs import SecretStrInput
|
||||||
from langflow.io import DropdownInput, StrInput
|
from langflow.io import DropdownInput, StrInput
|
||||||
from langflow.schema.dotdict import dotdict
|
from langflow.schema.dotdict import dotdict
|
||||||
from langflow.template.field.base import Output
|
from langflow.template.field.base import Output
|
||||||
|
|
@ -14,6 +15,10 @@ class NvidiaRerankComponent(LCCompressorComponent):
|
||||||
|
|
||||||
inputs = [
|
inputs = [
|
||||||
*LCCompressorComponent.inputs,
|
*LCCompressorComponent.inputs,
|
||||||
|
SecretStrInput(
|
||||||
|
name="api_key",
|
||||||
|
display_name="NVIDIA API Key",
|
||||||
|
),
|
||||||
StrInput(
|
StrInput(
|
||||||
name="base_url",
|
name="base_url",
|
||||||
display_name="Base URL",
|
display_name="Base URL",
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue