LangChain SQL - Template

template.py

# Create the prompt template
template = PromptTemplate(
    input_variables=["input_question", "context_explanation"],
    template="""{context_explanation}
    Question: {input_question}
    """
)

# Agent
agent_executor = create_sql_agent(
    llm=llm,
    toolkit=toolkit,
    verbose=True,
    handle_parsing_errors=True,
    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
    return_intermediate_steps=True,
)

try:
    print("Question: ", template.format(context_explanation = context_explanation, input_question = input_question))
    result = agent_executor.run(template.format(context_explanation = context_explanation, input_question = input_question))
    print("Result: ", result)
except Exception as e:
    print(f"An error occurred: {e}")