From 0a77490f8c5c4064bca96d11a7e7862551649ba1 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 27 Feb 2024 22:03:40 -0300 Subject: [PATCH] Update input field names in model components --- .../langflow/components/chains/RetrievalQA.py | 5 ++++- src/backend/langflow/components/io/TextInput.py | 12 ++++++------ .../langflow/components/models/AmazonBedrockModel.py | 2 +- .../langflow/components/models/AnthropicModel.py | 2 +- .../langflow/components/models/AzureOpenAIModel.py | 2 +- .../components/models/BaiduQianfanChatModel.py | 2 +- .../langflow/components/models/CTransformersModel.py | 2 +- .../langflow/components/models/CohereModel.py | 2 +- .../components/models/GoogleGenerativeAIModel.py | 7 ++----- .../langflow/components/models/HuggingFaceModel.py | 2 +- .../langflow/components/models/LlamaCppModel.py | 2 +- .../langflow/components/models/OllamaModel.py | 2 +- .../langflow/components/models/OpenAIModel.py | 2 +- .../langflow/components/models/VertexAiModel.py | 2 +- .../components/utilities/RunnableExecutor.py | 2 +- .../langflow/components/vectorstores/ChromaSearch.py | 2 +- 16 files changed, 25 insertions(+), 25 deletions(-) diff --git a/src/backend/langflow/components/chains/RetrievalQA.py b/src/backend/langflow/components/chains/RetrievalQA.py index 4968afe87..53fa24f15 100644 --- a/src/backend/langflow/components/chains/RetrievalQA.py +++ b/src/backend/langflow/components/chains/RetrievalQA.py @@ -20,7 +20,10 @@ class RetrievalQAComponent(CustomComponent): "input_key": {"display_name": "Input Key", "advanced": True}, "output_key": {"display_name": "Output Key", "advanced": True}, "return_source_documents": {"display_name": "Return Source Documents"}, - "inputs": {"display_name": "Input", "input_types": ["Text", "Document"]}, + "input_value": { + "display_name": "Input", + "input_types": ["Text", "Document"], + }, } def build( diff --git a/src/backend/langflow/components/io/TextInput.py b/src/backend/langflow/components/io/TextInput.py index 3fba54fda..f8c1ad606 100644 --- a/src/backend/langflow/components/io/TextInput.py +++ b/src/backend/langflow/components/io/TextInput.py @@ -9,11 +9,11 @@ class TextInput(CustomComponent): description = "Used to pass text input to the next component." field_config = { - "value": {"display_name": "Value", "multiline": True}, + "input_value": {"display_name": "Value", "multiline": True}, } - def build(self, value: Optional[str] = "") -> Text: - self.status = value - if not value: - value = "" - return value + def build(self, input_value: Optional[str] = "") -> Text: + self.status = input_value + if not input_value: + input_value = "" + return input_value diff --git a/src/backend/langflow/components/models/AmazonBedrockModel.py b/src/backend/langflow/components/models/AmazonBedrockModel.py index 478bf8e9a..70daa2b4c 100644 --- a/src/backend/langflow/components/models/AmazonBedrockModel.py +++ b/src/backend/langflow/components/models/AmazonBedrockModel.py @@ -34,7 +34,7 @@ class AmazonBedrockComponent(CustomComponent): "model_kwargs": {"display_name": "Model Kwargs"}, "cache": {"display_name": "Cache"}, "code": {"advanced": True}, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/AnthropicModel.py b/src/backend/langflow/components/models/AnthropicModel.py index cb8e55194..fe3d63d34 100644 --- a/src/backend/langflow/components/models/AnthropicModel.py +++ b/src/backend/langflow/components/models/AnthropicModel.py @@ -49,7 +49,7 @@ class AnthropicLLM(CustomComponent): "info": "Endpoint of the Anthropic API. Defaults to 'https://api.anthropic.com' if not specified.", }, "code": {"show": False}, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/AzureOpenAIModel.py b/src/backend/langflow/components/models/AzureOpenAIModel.py index df8413870..4abd7fcac 100644 --- a/src/backend/langflow/components/models/AzureOpenAIModel.py +++ b/src/backend/langflow/components/models/AzureOpenAIModel.py @@ -73,7 +73,7 @@ class AzureChatOpenAIComponent(CustomComponent): "info": "Maximum number of tokens to generate.", }, "code": {"show": False}, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/BaiduQianfanChatModel.py b/src/backend/langflow/components/models/BaiduQianfanChatModel.py index 0075316a2..af76262f8 100644 --- a/src/backend/langflow/components/models/BaiduQianfanChatModel.py +++ b/src/backend/langflow/components/models/BaiduQianfanChatModel.py @@ -68,7 +68,7 @@ class QianfanChatEndpointComponent(CustomComponent): "info": "Endpoint of the Qianfan LLM, required if custom model used.", }, "code": {"show": False}, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/CTransformersModel.py b/src/backend/langflow/components/models/CTransformersModel.py index fb292d9cf..8a556a954 100644 --- a/src/backend/langflow/components/models/CTransformersModel.py +++ b/src/backend/langflow/components/models/CTransformersModel.py @@ -28,7 +28,7 @@ class CTransformersComponent(CustomComponent): "field_type": "dict", "value": '{"top_k":40,"top_p":0.95,"temperature":0.8,"repetition_penalty":1.1,"last_n_tokens":64,"seed":-1,"max_new_tokens":256,"stop":"","stream":"False","reset":"True","batch_size":8,"threads":-1,"context_length":-1,"gpu_layers":0}', }, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/CohereModel.py b/src/backend/langflow/components/models/CohereModel.py index c2a004c38..2a6bf613b 100644 --- a/src/backend/langflow/components/models/CohereModel.py +++ b/src/backend/langflow/components/models/CohereModel.py @@ -28,7 +28,7 @@ class CohereComponent(CustomComponent): "type": "float", "show": True, }, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/GoogleGenerativeAIModel.py b/src/backend/langflow/components/models/GoogleGenerativeAIModel.py index 3b0c758e8..192fef8da 100644 --- a/src/backend/langflow/components/models/GoogleGenerativeAIModel.py +++ b/src/backend/langflow/components/models/GoogleGenerativeAIModel.py @@ -1,10 +1,7 @@ -from typing import Optional -from langchain_google_genai import ChatGoogleGenerativeAI # type: ignore -from pydantic.v1.types import SecretStr from langflow import CustomComponent -from langflow.field_typing import RangeSpec, Text +from langflow.field_typing import RangeSpec class GoogleGenerativeAIComponent(CustomComponent): @@ -50,7 +47,7 @@ class GoogleGenerativeAIComponent(CustomComponent): "code": { "advanced": True, }, - "inputs": {"display_name": "Input"}, + "input_value": {e": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/HuggingFaceModel.py b/src/backend/langflow/components/models/HuggingFaceModel.py index 99fcc6ab0..3fe97aca1 100644 --- a/src/backend/langflow/components/models/HuggingFaceModel.py +++ b/src/backend/langflow/components/models/HuggingFaceModel.py @@ -24,7 +24,7 @@ class HuggingFaceEndpointsComponent(CustomComponent): "field_type": "code", }, "code": {"show": False}, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/LlamaCppModel.py b/src/backend/langflow/components/models/LlamaCppModel.py index 5941f213b..d3080b8b5 100644 --- a/src/backend/langflow/components/models/LlamaCppModel.py +++ b/src/backend/langflow/components/models/LlamaCppModel.py @@ -56,7 +56,7 @@ class LlamaCppComponent(CustomComponent): "use_mmap": {"display_name": "Use Mmap", "advanced": True}, "verbose": {"display_name": "Verbose", "advanced": True}, "vocab_only": {"display_name": "Vocab Only", "advanced": True}, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/OllamaModel.py b/src/backend/langflow/components/models/OllamaModel.py index 7ae896532..72e3617df 100644 --- a/src/backend/langflow/components/models/OllamaModel.py +++ b/src/backend/langflow/components/models/OllamaModel.py @@ -164,7 +164,7 @@ class ChatOllamaComponent(CustomComponent): "info": "Template to use for generating text.", "advanced": True, }, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/models/OpenAIModel.py b/src/backend/langflow/components/models/OpenAIModel.py index ee6809e0b..6a8d5c19e 100644 --- a/src/backend/langflow/components/models/OpenAIModel.py +++ b/src/backend/langflow/components/models/OpenAIModel.py @@ -12,7 +12,7 @@ class OpenAIModelComponent(CustomComponent): def build_config(self): return { - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, "max_tokens": { "display_name": "Max Tokens", "advanced": False, diff --git a/src/backend/langflow/components/models/VertexAiModel.py b/src/backend/langflow/components/models/VertexAiModel.py index c05dc5e94..a8fd0eb1f 100644 --- a/src/backend/langflow/components/models/VertexAiModel.py +++ b/src/backend/langflow/components/models/VertexAiModel.py @@ -57,7 +57,7 @@ class ChatVertexAIComponent(CustomComponent): "value": False, "advanced": True, }, - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, } def build( diff --git a/src/backend/langflow/components/utilities/RunnableExecutor.py b/src/backend/langflow/components/utilities/RunnableExecutor.py index 5533e6d1d..502e1eec6 100644 --- a/src/backend/langflow/components/utilities/RunnableExecutor.py +++ b/src/backend/langflow/components/utilities/RunnableExecutor.py @@ -15,7 +15,7 @@ class RunnableExecComponent(CustomComponent): "display_name": "Input Key", "info": "The key to use for the input.", }, - "inputs": { + "input_value": { "display_name": "Inputs", "info": "The inputs to pass to the runnable.", }, diff --git a/src/backend/langflow/components/vectorstores/ChromaSearch.py b/src/backend/langflow/components/vectorstores/ChromaSearch.py index 5dd33abf2..6cb3e89df 100644 --- a/src/backend/langflow/components/vectorstores/ChromaSearch.py +++ b/src/backend/langflow/components/vectorstores/ChromaSearch.py @@ -26,7 +26,7 @@ class ChromaSearchComponent(CustomComponent): - dict: A dictionary containing the configuration options for the component. """ return { - "inputs": {"display_name": "Input"}, + "input_value": {"display_name": "Input"}, "search_type": { "display_name": "Search Type", "options": ["Similarity", "MMR"],