From 32379092212993abd26eabba05633cfe4d271018 Mon Sep 17 00:00:00 2001 From: Naka Masato Date: Tue, 11 Jun 2024 05:58:30 +0900 Subject: [PATCH] langchain[patch]: allow to use partial variables in create_sql_query_chain (#22688) - **Description:** allow to use partial variables to pass `top_k` and `table_info` - **Issue:** no - **Dependencies:** no - **Twitter handle:** @gymnstcs --------- Co-authored-by: Bagatur --- libs/langchain/langchain/chains/sql_database/query.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/langchain/langchain/chains/sql_database/query.py b/libs/langchain/langchain/chains/sql_database/query.py index bf5feef483b..d424adb4c5d 100644 --- a/libs/langchain/langchain/chains/sql_database/query.py +++ b/libs/langchain/langchain/chains/sql_database/query.py @@ -116,7 +116,9 @@ def create_sql_query_chain( prompt_to_use = SQL_PROMPTS[db.dialect] else: prompt_to_use = PROMPT - if {"input", "top_k", "table_info"}.difference(prompt_to_use.input_variables): + if {"input", "top_k", "table_info"}.difference( + prompt_to_use.input_variables + list(prompt_to_use.partial_variables) + ): raise ValueError( f"Prompt must have input variables: 'input', 'top_k', " f"'table_info'. Received prompt with input variables: "