From 4767606fb7de2d8f01739d607045356f3994cbc1 Mon Sep 17 00:00:00 2001 From: Juan Carlos Jose Camacho Date: Tue, 7 Nov 2023 14:52:06 -0600 Subject: [PATCH] Send db_connection object as parameter --- dataherald/sql_generator/generates_nl_answer.py | 1 + dataherald/sql_generator/langchain_sqlagent.py | 1 + dataherald/sql_generator/langchain_sqlchain.py | 1 + dataherald/sql_generator/llamaindex.py | 1 + 4 files changed, 4 insertions(+) diff --git a/dataherald/sql_generator/generates_nl_answer.py b/dataherald/sql_generator/generates_nl_answer.py index de82b69e..89500f0d 100644 --- a/dataherald/sql_generator/generates_nl_answer.py +++ b/dataherald/sql_generator/generates_nl_answer.py @@ -58,6 +58,7 @@ def execute( query_response, top_k=int(os.getenv("UPPER_LIMIT_QUERY_RETURN_ROWS", "50")), generate_csv=generate_csv, + database_connection=database_connection, ) if query_response.csv_file_path: diff --git a/dataherald/sql_generator/langchain_sqlagent.py b/dataherald/sql_generator/langchain_sqlagent.py index 5e6813f4..14a3ffcf 100644 --- a/dataherald/sql_generator/langchain_sqlagent.py +++ b/dataherald/sql_generator/langchain_sqlagent.py @@ -91,4 +91,5 @@ def generate_response( response.sql_query, response, generate_csv=generate_csv, + database_connection=database_connection, ) diff --git a/dataherald/sql_generator/langchain_sqlchain.py b/dataherald/sql_generator/langchain_sqlchain.py index 90516080..4abaef4a 100644 --- a/dataherald/sql_generator/langchain_sqlchain.py +++ b/dataherald/sql_generator/langchain_sqlchain.py @@ -100,4 +100,5 @@ def generate_response( response.sql_query, response, generate_csv=generate_csv, + database_connection=database_connection, ) diff --git a/dataherald/sql_generator/llamaindex.py b/dataherald/sql_generator/llamaindex.py index 3d5ee4fb..689729af 100644 --- a/dataherald/sql_generator/llamaindex.py +++ b/dataherald/sql_generator/llamaindex.py @@ -115,4 +115,5 @@ def generate_response( response.sql_query, response, generate_csv=generate_csv, + database_connection=database_connection, )