From 8c2bca31ff33c0d621ddb4d5010c1ebe90ed6afd Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 25 Mar 2024 13:55:09 -0300
Subject: [PATCH] Add input_value parameter to chain components

---
 src/backend/langflow/components/chains/ConversationChain.py  | 5 ++++-
 src/backend/langflow/components/chains/LLMChain.py           | 1 -
 src/backend/langflow/components/chains/LLMCheckerChain.py    | 4 ++++
 src/backend/langflow/components/chains/LLMMathChain.py       | 4 ++++
 .../components/chains/RetrievalQAWithSourcesChain.py         | 4 ++++
 src/backend/langflow/components/chains/SQLGenerator.py       | 4 ++++
 6 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/components/chains/ConversationChain.py b/src/backend/langflow/components/chains/ConversationChain.py
index 71599ef2e..a6d66d8ab 100644
--- a/src/backend/langflow/components/chains/ConversationChain.py
+++ b/src/backend/langflow/components/chains/ConversationChain.py
@@ -18,7 +18,10 @@ class ConversationChainComponent(CustomComponent):
                 "display_name": "Memory",
                 "info": "Memory to load context from. If none is provided, a ConversationBufferMemory will be used.",
             },
-            "code": {"show": False},
+            "input_value": {
+                "display_name": "Input Value",
+                "info": "The input value to pass to the chain.",
+            },
         }
 
     def build(
diff --git a/src/backend/langflow/components/chains/LLMChain.py b/src/backend/langflow/components/chains/LLMChain.py
index 5d660cd4b..0354082c9 100644
--- a/src/backend/langflow/components/chains/LLMChain.py
+++ b/src/backend/langflow/components/chains/LLMChain.py
@@ -20,7 +20,6 @@ class LLMChainComponent(CustomComponent):
             "prompt": {"display_name": "Prompt"},
             "llm": {"display_name": "LLM"},
             "memory": {"display_name": "Memory"},
-            "code": {"show": False},
         }
 
     def build(
diff --git a/src/backend/langflow/components/chains/LLMCheckerChain.py b/src/backend/langflow/components/chains/LLMCheckerChain.py
index 5e7af9af5..ee869caef 100644
--- a/src/backend/langflow/components/chains/LLMCheckerChain.py
+++ b/src/backend/langflow/components/chains/LLMCheckerChain.py
@@ -12,6 +12,10 @@ class LLMCheckerChainComponent(CustomComponent):
     def build_config(self):
         return {
             "llm": {"display_name": "LLM"},
+            "input_value": {
+                "display_name": "Input Value",
+                "info": "The input value to pass to the chain.",
+            },
         }
 
     def build(
diff --git a/src/backend/langflow/components/chains/LLMMathChain.py b/src/backend/langflow/components/chains/LLMMathChain.py
index 52fb5e1ee..26453b597 100644
--- a/src/backend/langflow/components/chains/LLMMathChain.py
+++ b/src/backend/langflow/components/chains/LLMMathChain.py
@@ -18,6 +18,10 @@ class LLMMathChainComponent(CustomComponent):
             "memory": {"display_name": "Memory"},
             "input_key": {"display_name": "Input Key"},
             "output_key": {"display_name": "Output Key"},
+            "input_value": {
+                "display_name": "Input Value",
+                "info": "The input value to pass to the chain.",
+            },
         }
 
     def build(
diff --git a/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py b/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py
index 9a1b63756..48f918a19 100644
--- a/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py
+++ b/src/backend/langflow/components/chains/RetrievalQAWithSourcesChain.py
@@ -22,6 +22,10 @@ class RetrievalQAWithSourcesChainComponent(CustomComponent):
             "memory": {"display_name": "Memory"},
             "return_source_documents": {"display_name": "Return Source Documents"},
             "retriever": {"display_name": "Retriever"},
+            "input_value": {
+                "display_name": "Input Value",
+                "info": "The input value to pass to the chain.",
+            },
         }
 
     def build(
diff --git a/src/backend/langflow/components/chains/SQLGenerator.py b/src/backend/langflow/components/chains/SQLGenerator.py
index ecd9eb248..d96db7a7f 100644
--- a/src/backend/langflow/components/chains/SQLGenerator.py
+++ b/src/backend/langflow/components/chains/SQLGenerator.py
@@ -25,6 +25,10 @@ class SQLGeneratorComponent(CustomComponent):
                 "display_name": "Top K",
                 "info": "The number of results per select statement to return. If 0, no limit.",
             },
+            "input_value": {
+                "display_name": "Input Value",
+                "info": "The input value to pass to the chain.",
+            },
         }
 
     def build(