diff --git a/src/backend/langflow/components/chains/SQLGenerator.py b/src/backend/langflow/components/chains/SQLGenerator.py index 3b6347ca4..5efb0f738 100644 --- a/src/backend/langflow/components/chains/SQLGenerator.py +++ b/src/backend/langflow/components/chains/SQLGenerator.py @@ -43,19 +43,10 @@ class SQLGeneratorComponent(CustomComponent): else: template = prompt.template if hasattr(prompt, "template") else prompt # Check if {question} is in the prompt - if ( - "{question}" not in template - or "question" not in template.input_variables - ): - raise ValueError( - "Prompt must contain `{question}` to be used with Natural Language to SQL." - ) - sql_query_chain = create_sql_query_chain( - llm=llm, db=db, prompt=prompt, **kwargs - ) - query_writer = sql_query_chain | { - "query": lambda x: x.replace("SQLQuery:", "").strip() - } + if "{question}" not in template or "question" not in template.input_variables: + raise ValueError("Prompt must contain `{question}` to be used with Natural Language to SQL.") + sql_query_chain = create_sql_query_chain(llm=llm, db=db, prompt=prompt, **kwargs) + query_writer = sql_query_chain | {"query": lambda x: x.replace("SQLQuery:", "").strip()} response = query_writer.invoke({"question": inputs}) query = response.get("query") self.status = query