Send db_connection object as parameter

Dataherald · Nov 7, 2023 · 4767606 · 4767606
1 parent 147f207
commit 4767606
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -115,4 +115,5 @@ def generate_response(
             response.sql_query,
             response,
             generate_csv=generate_csv,
+            database_connection=database_connection,
         )