jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor SQLGeneratorComponent class to handle prompts without {question} variable

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-26 16:22:02 -03:00
commit 0111c62c4e
1 changed files with 4 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/backend/langflow/components/chains/SQLGenerator.py
View file

@ -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