added dev mode to execute query, to incorporate it in the retry method

defog-ai · Apr 9, 2024 · 71fe600 · 71fe600
1 parent 6c402b0
commit 71fe600
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/defog/query.py b/defog/query.py
@@ -119,6 +119,7 @@ def execute_query(
     hard_filters: str = "",
     retries: int = 3,
     schema: dict = None,
+    dev: bool = False,
 ):
     """
     Execute the query and retry with adaptive learning if there is an error.
@@ -144,6 +145,7 @@ def execute_query(
                     "db_type": db_type,
                     "question": question,
                     "query": query,
+                    "dev": dev,
                 },
                 timeout=1,
             )
@@ -162,6 +164,7 @@ def execute_query(
                     "db_type": db_type,
                     "hard_filters": hard_filters,
                     "question": question,
+                    "dev": dev,
                 }
                 if schema is not None:
                     retry["schema"] = schema

diff --git a/defog/query_methods.py b/defog/query_methods.py
@@ -113,6 +113,7 @@ def run_query(
                 question,
                 hard_filters,
                 retries,
+                dev=dev,
             )
             tend = datetime.now()
             time_taken = (tend - tstart).total_seconds()