DH-5584/updating the timeout

Dataherald · Mar 13, 2024 · 6093d2f · 6093d2f
1 parent e0ec7c3
commit 6093d2f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/dataherald/sql_generator/create_sql_query_status.py b/dataherald/sql_generator/create_sql_query_status.py
@@ -34,7 +34,7 @@ def create_sql_query_status(
             run_with_timeout(
                 db.run_sql,
                 args=(query,),
-                timeout_duration=int(os.getenv("SQL_EXECUTION_TIMEOUT", "30")),
+                timeout_duration=int(os.getenv("SQL_EXECUTION_TIMEOUT", "60")),
             )
             sql_generation.status = "VALID"
             sql_generation.error = None

diff --git a/dataherald/sql_generator/dataherald_finetuning_agent.py b/dataherald/sql_generator/dataherald_finetuning_agent.py
@@ -248,7 +248,7 @@ def _run(
                 self.db.run_sql,
                 args=(query,),
                 kwargs={"top_k": TOP_K},
-                timeout_duration=int(os.getenv("SQL_EXECUTION_TIMEOUT", "30")),
+                timeout_duration=int(os.getenv("SQL_EXECUTION_TIMEOUT", "60")),
             )
         except TimeoutError:
             return "SQL query execution time exceeded, proceed without query execution"

diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py
@@ -166,7 +166,7 @@ def _run(
                 self.db.run_sql,
                 args=(query,),
                 kwargs={"top_k": top_k},
-                timeout_duration=int(os.getenv("SQL_EXECUTION_TIMEOUT", "30")),
+                timeout_duration=int(os.getenv("SQL_EXECUTION_TIMEOUT", "60")),
             )
         except TimeoutError:
             return "SQL query execution time exceeded, proceed without query execution"